.

pytorch · Oct 23, 2024 · 3a51218 · 3a51218
1 parent 250c3a8
commit 3a51218
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/.github/workflows/linux_cuda_wheel.yaml b/.github/workflows/linux_cuda_wheel.yaml
@@ -139,4 +139,6 @@ jobs:
         run: |
           # We skip test_get_ffmpeg_version because it may not have a micro version.
           ${CONDA_RUN} FAIL_WITHOUT_CUDA=1 pytest test -k "not test_get_ffmpeg_version" -vvv
+      - name: Run Python benchmark
+        run: |
           ${CONDA_RUN} time python benchmarks/decoders/gpu_benchmark.py --devices=cuda:0,cpu --resize_devices=none