@@ -109,24 +109,28 @@ rapids-logger "Python Async Tests"
109
109
# run_tests_async PROGRESS_MODE ENABLE_DELAYED_SUBMISSION ENABLE_PYTHON_FUTURE SKIP
110
110
run_tests_async thread 0 0 0
111
111
run_tests_async thread 1 1 0
112
+ run_tests_async blocking 1 1 0
112
113
113
114
rapids-logger " Python Benchmarks"
114
115
# run_py_benchmark BACKEND PROGRESS_MODE ASYNCIO_WAIT ENABLE_DELAYED_SUBMISSION ENABLE_PYTHON_FUTURE NBUFFERS SLOW
115
116
run_py_benchmark ucxx-core thread 0 0 0 1 0
116
117
run_py_benchmark ucxx-core thread 1 0 0 1 0
117
118
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
126
129
done
127
130
128
131
rapids-logger " Distributed Tests"
129
132
# run_distributed_ucxx_tests PROGRESS_MODE ENABLE_DELAYED_SUBMISSION ENABLE_PYTHON_FUTURE
133
+ run_distributed_ucxx_tests blocking 0 0
130
134
run_distributed_ucxx_tests polling 0 0
131
135
run_distributed_ucxx_tests thread 0 0
132
136
run_distributed_ucxx_tests thread 0 1
0 commit comments