Skip to content

Commit 37ea715

Browse files
committed
Test blocking mode in CI
1 parent f965258 commit 37ea715

File tree

1 file changed

+12
-8
lines changed

1 file changed

+12
-8
lines changed

ci/test_python.sh

Lines changed: 12 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -109,24 +109,28 @@ rapids-logger "Python Async Tests"
109109
# run_tests_async PROGRESS_MODE ENABLE_DELAYED_SUBMISSION ENABLE_PYTHON_FUTURE SKIP
110110
run_tests_async thread 0 0 0
111111
run_tests_async thread 1 1 0
112+
run_tests_async blocking 1 1 0
112113

113114
rapids-logger "Python Benchmarks"
114115
# run_py_benchmark BACKEND PROGRESS_MODE ASYNCIO_WAIT ENABLE_DELAYED_SUBMISSION ENABLE_PYTHON_FUTURE NBUFFERS SLOW
115116
run_py_benchmark ucxx-core thread 0 0 0 1 0
116117
run_py_benchmark ucxx-core thread 1 0 0 1 0
117118

118-
for nbuf in 1 8; do
119-
if [[ ! $RAPIDS_CUDA_VERSION =~ 11.2.* ]]; then
120-
# run_py_benchmark BACKEND PROGRESS_MODE ASYNCIO_WAIT ENABLE_DELAYED_SUBMISSION ENABLE_PYTHON_FUTURE NBUFFERS SLOW
121-
run_py_benchmark ucxx-async thread 0 0 0 ${nbuf} 0
122-
run_py_benchmark ucxx-async thread 0 0 1 ${nbuf} 0
123-
run_py_benchmark ucxx-async thread 0 1 0 ${nbuf} 0
124-
run_py_benchmark ucxx-async thread 0 1 1 ${nbuf} 0
125-
fi
119+
for progress_mode in "blocking" "thread"; do
120+
for nbuf in 1 8; do
121+
if [[ ! $RAPIDS_CUDA_VERSION =~ 11.2.* ]]; then
122+
# run_py_benchmark BACKEND PROGRESS_MODE ASYNCIO_WAIT ENABLE_DELAYED_SUBMISSION ENABLE_PYTHON_FUTURE NBUFFERS SLOW
123+
run_py_benchmark ucxx-async ${progress_mode} 0 0 0 ${nbuf} 0
124+
run_py_benchmark ucxx-async ${progress_mode} 0 0 1 ${nbuf} 0
125+
run_py_benchmark ucxx-async ${progress_mode} 0 1 0 ${nbuf} 0
126+
run_py_benchmark ucxx-async ${progress_mode} 0 1 1 ${nbuf} 0
127+
fi
128+
done
126129
done
127130

128131
rapids-logger "Distributed Tests"
129132
# run_distributed_ucxx_tests PROGRESS_MODE ENABLE_DELAYED_SUBMISSION ENABLE_PYTHON_FUTURE
133+
run_distributed_ucxx_tests blocking 0 0
130134
run_distributed_ucxx_tests polling 0 0
131135
run_distributed_ucxx_tests thread 0 0
132136
run_distributed_ucxx_tests thread 0 1

0 commit comments

Comments
 (0)