From 6023ea395edf83b2162c85cf32a3f6c0466cd221 Mon Sep 17 00:00:00 2001 From: Nicolas Hug Date: Mon, 7 Jul 2025 11:34:50 +0100 Subject: [PATCH] Add testing against CUDA 12.9 --- .github/workflows/linux_cuda_wheel.yaml | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/.github/workflows/linux_cuda_wheel.yaml b/.github/workflows/linux_cuda_wheel.yaml index bd57cac5e..8bc8c3bc1 100644 --- a/.github/workflows/linux_cuda_wheel.yaml +++ b/.github/workflows/linux_cuda_wheel.yaml @@ -67,7 +67,9 @@ jobs: # For the actual release we should add that label and change this to # include more python versions. python-version: ['3.9'] - cuda-version: ['12.6', '12.8'] + # We test against 12.6 and 12.9 to avoid having too big of a CI matrix, + # but for releases we should add 12.8. + cuda-version: ['12.6', '12.9'] # TODO: put back ffmpeg 5 https://github.com/pytorch/torchcodec/issues/325 ffmpeg-version-for-tests: ['4.4.2', '6', '7']