Skip to content

Commit

Permalink
Temporarily disable Python async benchmark on CUDA 11.2 (#37)
Browse files Browse the repository at this point in the history
Python async benchmarks are frequently hanging only in CUDA 11.2, disabling those temporarily will help in reducing CI failures.

Authors:
  - Peter Andreas Entschev (https://github.com/pentschev)

Approvers:
  - AJ Schmidt (https://github.com/ajschmidt8)

URL: #37
  • Loading branch information
pentschev authored May 2, 2023
1 parent 1bc38c9 commit c96d347
Showing 1 changed file with 6 additions and 4 deletions.
10 changes: 6 additions & 4 deletions ci/test_python.sh
Original file line number Diff line number Diff line change
Expand Up @@ -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

0 comments on commit c96d347

Please sign in to comment.