Skip to content

Commit 6023ea3

Browse files
committed
Add testing against CUDA 12.9
1 parent 7945e6a commit 6023ea3

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

.github/workflows/linux_cuda_wheel.yaml

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -67,7 +67,9 @@ jobs:
6767
# For the actual release we should add that label and change this to
6868
# include more python versions.
6969
python-version: ['3.9']
70-
cuda-version: ['12.6', '12.8']
70+
# We test against 12.6 and 12.9 to avoid having too big of a CI matrix,
71+
# but for releases we should add 12.8.
72+
cuda-version: ['12.6', '12.9']
7173
# TODO: put back ffmpeg 5 https://github.com/pytorch/torchcodec/issues/325
7274
ffmpeg-version-for-tests: ['4.4.2', '6', '7']
7375

0 commit comments

Comments
 (0)