From 9d164be8805534db9313b4ceff647b21c9814210 Mon Sep 17 00:00:00 2001 From: Peter Andreas Entschev Date: Tue, 2 May 2023 09:42:57 -0700 Subject: [PATCH] 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. --- ci/test_python.sh | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/ci/test_python.sh b/ci/test_python.sh index 82be0e8a..167dc658 100755 --- 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