Skip to content

Commit 8b4dd51

Browse files
committed
Test if h265 can work on other GPU runners
1 parent 81de40e commit 8b4dd51

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

.github/workflows/linux_cuda_wheel.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,7 @@ jobs:
5656
build-command: "BUILD_AGAINST_ALL_FFMPEG_FROM_S3=1 ENABLE_CUDA=1 python -m build --wheel -vvv --no-isolation"
5757

5858
install-and-test:
59-
runs-on: linux.4xlarge.nvidia.gpu
59+
runs-on: linux.g5.4xlarge.nvidia.gpu
6060
strategy:
6161
fail-fast: false
6262
matrix:

test/decoders/test_video_decoder.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -426,6 +426,7 @@ def test_get_frame_played_at(self, device):
426426
assert isinstance(decoder.get_frame_played_at(6.02).pts_seconds, float)
427427
assert isinstance(decoder.get_frame_played_at(6.02).duration_seconds, float)
428428

429+
@pytest.mark.parametrize("device", cpu_and_cuda())
429430
def test_get_frame_played_at_h265(self):
430431
# Non-regression test for https://github.com/pytorch/torchcodec/issues/179
431432
# We don't parametrize with CUDA because the current GPUs on CI do not

0 commit comments

Comments
 (0)