Upgrade the OS and ROCM versions. (#411)

* upgrade the OS and ROCM versions in CK docker

* add cxx flags to link code with rocm5.2 and ck-9110 compiler

* rename the docker image

* run ONNX gemms using init=1
This commit is contained in:
Illia Silin
2022-09-13 08:39:14 -07:00
committed by GitHub
parent efd1d25733
commit b22ebd4485
3 changed files with 22 additions and 19 deletions

View File

@@ -138,5 +138,5 @@ print_log_header $splitK_gemm_log $env_type $branch $host_name
#run ONNX gemm tests
export onnx_log="perf_onnx_gemm_${gpu_arch}.log"
print_log_header $onnx_log $env_type $branch $host_name
./profile_onnx_gemm.sh gemm 0 0 $verify 2 0 1 2>&1 | tee -a $onnx_log
./profile_onnx_gemm.sh gemm 1 0 $verify 2 0 1 2>&1 | tee -a $onnx_log
./profile_onnx_gemm.sh gemm 0 0 $verify 1 0 1 2>&1 | tee -a $onnx_log
./profile_onnx_gemm.sh gemm 1 0 $verify 1 0 1 2>&1 | tee -a $onnx_log