Temporarily disable Python async benchmark on CUDA 11.2

Python async benchmarks are frequently hanging only in CUDA 11.2, disabling those temporarily will help in reducing CI failures.
rapidsai · May 2, 2023 · 9d164be · 9d164be
1 parent 1bc38c9
commit 9d164be
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/ci/test_python.sh b/ci/test_python.sh
@@ -92,8 +92,10 @@ run_py_benchmark    ucxx-core thread          1             0
 
 for nbuf in 1 8; do
   # run_py_benchmark  BACKEND     PROGRESS_MODE   ASYNCIO_WAIT  ENABLE_DELAYED_SUBMISSION ENABLE_PYTHON_FUTURE NBUFFERS SLOW
-  run_py_benchmark    ucxx-async  thread          0             0                         0                    ${nbuf}  0
-  run_py_benchmark    ucxx-async  thread          0             0                         1                    ${nbuf}  0
-  run_py_benchmark    ucxx-async  thread          0             1                         0                    ${nbuf}  0
-  run_py_benchmark    ucxx-async  thread          0             1                         1                    ${nbuf}  0
+  if [[ $RAPIDS_CUDA_VERSION != "11.2.*" ]]; then
+    run_py_benchmark    ucxx-async  thread          0             0                         0                    ${nbuf}  0
+    run_py_benchmark    ucxx-async  thread          0             0                         1                    ${nbuf}  0
+    run_py_benchmark    ucxx-async  thread          0             1                         0                    ${nbuf}  0
+    run_py_benchmark    ucxx-async  thread          0             1                         1                    ${nbuf}  0
+  fi
 done