.

Author: ahmadsharif1

.github/workflows/linux_cuda_wheel.yaml

@@ -60,6 +60,11 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
+          # 3.9 corresponds to the minimum python version for which we build
+          # the wheel unless the label cliflow/binaries/all is present in the
+          # PR.
+          # For the actual release we should add that label and change this to
+          # include more python versions.
         python-version: ['3.9']
         cuda-version: ['12.4']
         ffmpeg-version-for-tests: ['5', '6', '7']
@@ -75,8 +80,6 @@ jobs:
           echo cuda_version_without_periods=${cuda_version_without_periods} >> $GITHUB_ENV
       - uses: actions/download-artifact@v3
         with:
-          # We use 3.9 because it's the minimum version we build.
-          # We test version 3.9 against all other python versions in the matrix.
           name: pytorch_torchcodec__3.9_cu${{ env.cuda_version_without_periods }}_x86_64
           path: pytorch/torchcodec/dist/
       - name: Setup miniconda using test-infra
@@ -133,4 +136,4 @@ jobs:
       - name: Run Python tests
         run: |
           ${CONDA_RUN} FAIL_WITHOUT_CUDA=1 pytest test -vvv
-          ${CONDA_RUN} python benchmarks/decoders/gpu_benchmark.py --devices=cuda:0,cpu --resize_devices=none
+          ${CONDA_RUN} time python benchmarks/decoders/gpu_benchmark.py --devices=cuda:0,cpu --resize_devices=none

src/torchcodec/decoders/_core/CMakeLists.txt

@@ -77,17 +77,13 @@ if(DEFINED ENV{BUILD_AGAINST_ALL_FFMPEG_FROM_S3})
     )
 
 
-    if(ENABLE_CUDA)
-        # TODO: Enable more ffmpeg versions for cuda.
-        make_torchcodec_library(libtorchcodec7 ffmpeg7)
-        make_torchcodec_library(libtorchcodec6 ffmpeg6)
-        make_torchcodec_library(libtorchcodec5 ffmpeg5)
-    else()
-        make_torchcodec_library(libtorchcodec7 ffmpeg7)
-        make_torchcodec_library(libtorchcodec6 ffmpeg6)
-        make_torchcodec_library(libtorchcodec5 ffmpeg5)
-        make_torchcodec_library(libtorchcodec4 ffmpeg4)
-    endif()
+    if(NOT ENABLE_CUDA)
+	    # TODO: Enable more ffmpeg versions for cuda.
+	    make_torchcodec_library(libtorchcodec4 ffmpeg4)
+	endif()
+	make_torchcodec_library(libtorchcodec7 ffmpeg7)
+	make_torchcodec_library(libtorchcodec6 ffmpeg6)
+	make_torchcodec_library(libtorchcodec5 ffmpeg5)
 
 else()
     message(