Skip to content

Commit d2313a1

Browse files
committed
Unset CUDA_PATH instead
1 parent 1d68f59 commit d2313a1

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

ci/test_cuda_cccl_examples_python.sh

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -27,11 +27,12 @@ fi
2727
CUDA_CCCL_WHEEL_PATH="$(ls /home/coder/cccl/wheelhouse/cuda_cccl-*.whl)"
2828
python -m pip install "${CUDA_CCCL_WHEEL_PATH}[test-cu${cuda_major_version}]"
2929

30-
# On CUDA 12.0, install cuda-toolkit components to supplement the environment
30+
# Workaround for https://github.com/cupy/cupy/issues/9719,
31+
# which is caused by some complex interaction of cupy, pathfinder and the environment.
3132
if [[ "${cuda_major_version}" == "12" && "${cuda_minor_version}" == "0" ]]; then
32-
python -m pip install "cuda-toolkit[cudart,nvcc,nvrtc,nvjitlink]==12.0.*"
33+
unset CUDA_PATH
3334
fi
3435

35-
# Run tests for parallel module
36+
# Run examples
3637
cd "/home/coder/cccl/python/cuda_cccl/tests/"
3738
python -m pytest -n 6 test_examples.py

ci/test_cuda_compute_python.sh

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,9 +29,8 @@ python -m pip install "${CUDA_CCCL_WHEEL_PATH}[test-cu${cuda_major_version}]"
2929

3030
# Workaround for https://github.com/cupy/cupy/issues/9719,
3131
# which is caused by some complex interaction of cupy, pathfinder and the environment.
32-
# Ensuring the environment has all CUDA components fixes it:
3332
if [[ "${cuda_major_version}" == "12" && "${cuda_minor_version}" == "0" ]]; then
34-
python -m pip install "cuda-toolkit[cudart,nvcc,nvrtc,nvjitlink]==12.0.*"
33+
unset CUDA_PATH
3534
fi
3635

3736
# Run tests for compute module

0 commit comments

Comments
 (0)