Use CUDA 13.0 on CI (#911)

Co-authored-by: Nicolas Hug <[email protected]>
Co-authored-by: Daniel Flores <[email protected]>

Author: 3 people

.github/workflows/linux_cuda_wheel.yaml

@@ -67,10 +67,9 @@ jobs:
           # For the actual release we should add that label and change this to
           # include more python versions.
         python-version: ['3.10']
-        # We test against 12.6 to avoid having too big of a CI matrix,
+        # We test against 12.6 and 13.0 to avoid having too big of a CI matrix,
         # but for releases we should add 12.8.
-        # TODO add 13.0!
-        cuda-version: ['12.6']
+        cuda-version: ['12.6', '13.0']
         # TODO: put back ffmpeg 5 https://github.com/pytorch/torchcodec/issues/325
         ffmpeg-version-for-tests: ['4.4.2', '6', '7']
 

test/test_decoders.py

@@ -1223,7 +1223,9 @@ def test_full_and_studio_range_bt709_video(self, asset):
             gpu_frame = decoder_gpu.get_frame_at(frame_index).data.cpu()
             cpu_frame = decoder_cpu.get_frame_at(frame_index).data
 
-            if cuda_version_used_for_building_torch() >= (12, 9):
+            if cuda_version_used_for_building_torch() >= (13, 0):
+                torch.testing.assert_close(gpu_frame, cpu_frame, rtol=0, atol=3)
+            elif cuda_version_used_for_building_torch() >= (12, 9):
                 torch.testing.assert_close(gpu_frame, cpu_frame, rtol=0, atol=2)
             elif cuda_version_used_for_building_torch() == (12, 8):
                 assert psnr(gpu_frame, cpu_frame) > 20

test/utils.py

@@ -73,7 +73,7 @@ def psnr(a, b, max_val=255) -> float:
 def assert_frames_equal(*args, **kwargs):
     if sys.platform == "linux":
         if args[0].device.type == "cuda":
-            atol = 2
+            atol = 3 if cuda_version_used_for_building_torch() >= (13, 0) else 2
             if get_ffmpeg_major_version() == 4:
                 assert_tensor_close_on_at_least(
                     args[0], args[1], percentage=95, atol=atol