We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 73ce1e2 commit 716b2a7Copy full SHA for 716b2a7
1 file changed
.github/workflows/buildcupy.yaml
@@ -43,12 +43,11 @@ jobs:
43
- name: Check cuda versions
44
run: |
45
srun --account=yuxilab -n 1 -N 1 --gres=gpu:L40S:1 bash -c 'ls /u/yuxihong/workspace/cuda/'
46
+ - name: Install CUDA 12.8 toolkit
47
+ run: uv add nvidia-cuda-nvcc-cu12==12.8.*
48
- name: Test with pytest
49
50
srun --account=yuxilab -n 1 -N 1 --gres=gpu:L40S:1 bash -c '
- uv add nvidia-cuda-nvcc-cu12==12.8.\* &&
- export NUMBA_FORCE_CUDA_CC=8.6
51
- export NUMBA_CUDA_DEFAULT_PTX_CC=8.6
52
export CUPY_PYLOPS=1
53
export TEST_CUPY_PYLOPS=1
54
uv run pytest --color=yes pytests/
0 commit comments