Skip to content

Commit

Permalink
Switch pytest-xdist algorithm to worksteal (#708)
Browse files Browse the repository at this point in the history
This PR uses the `worksteal` algorithm to accelerate parallel pytests. See rapidsai/cudf#15207.

Authors:
  - Bradley Dice (https://github.com/bdice)

Approvers:
  - GALI PREM SAGAR (https://github.com/galipremsagar)
  - Jake Awe (https://github.com/AyodeAwe)

URL: #708
  • Loading branch information
bdice authored Mar 6, 2024
1 parent 46da10e commit 5ffe82b
Show file tree
Hide file tree
Showing 2 changed files with 11 additions and 3 deletions.
2 changes: 1 addition & 1 deletion ci/test_python.sh
Original file line number Diff line number Diff line change
Expand Up @@ -51,7 +51,7 @@ timeout 20m pytest \
--cache-clear \
--junitxml="${RAPIDS_TESTS_DIR}/junit-cucim.xml" \
--numprocesses=8 \
--dist=loadscope \
--dist=worksteal \
--cov-config=.coveragerc \
--cov=cucim \
--cov-report=xml:"${RAPIDS_COVERAGE_DIR}/cucim-coverage.xml" \
Expand Down
12 changes: 10 additions & 2 deletions ci/test_wheel.sh
Original file line number Diff line number Diff line change
Expand Up @@ -22,8 +22,16 @@ else
if [[ ${CUDA_MAJOR_VERSION} == "11" ]]; then
# Omit I/O-related tests in ./python/cucim/tests due to known CUDA bug
# with dynamic loading of libcufile.
python -m pytest ./python/cucim/src/
python -m pytest \
--junitxml="${RAPIDS_TESTS_DIR}/junit-cucim.xml" \
--numprocesses=8 \
--dist=worksteal \
./python/cucim/src/
else
python -m pytest ./python/cucim
python -m pytest \
--junitxml="${RAPIDS_TESTS_DIR}/junit-cucim.xml" \
--numprocesses=8 \
--dist=worksteal \
./python/cucim
fi
fi

0 comments on commit 5ffe82b

Please sign in to comment.