diff --git a/test/test_decoders.py b/test/test_decoders.py index 6995161fd..951da0c40 100644 --- a/test/test_decoders.py +++ b/test/test_decoders.py @@ -591,7 +591,7 @@ def test_get_frame_at_av1(self, device): if device == "cuda" and get_ffmpeg_major_version() == 4: return - if device == "cuda" and in_fbcode(): + if "cuda" in device and in_fbcode(): pytest.skip("decoding on CUDA is not supported internally") decoder = VideoDecoder(AV1_VIDEO.path, device=device) @@ -1477,6 +1477,9 @@ def test_beta_cuda_interface_get_frame_at( if seek_mode == "approximate" and not supports_approximate_mode(asset): pytest.skip("asset doesn't work with approximate mode") + if in_fbcode() and asset is AV1_VIDEO: + pytest.skip("AV1 CUDA not supported internally") + ref_decoder = VideoDecoder(asset.path, device="cuda", seek_mode=seek_mode) beta_decoder = VideoDecoder( asset.path, device="cuda:0:beta", seek_mode=seek_mode @@ -1522,6 +1525,8 @@ def test_beta_cuda_interface_get_frames_at( ): if seek_mode == "approximate" and not supports_approximate_mode(asset): pytest.skip("asset doesn't work with approximate mode") + if in_fbcode() and asset is AV1_VIDEO: + pytest.skip("AV1 CUDA not supported internally") ref_decoder = VideoDecoder(asset.path, device="cuda", seek_mode=seek_mode) beta_decoder = VideoDecoder( @@ -1566,6 +1571,8 @@ def test_beta_cuda_interface_get_frames_at( def test_beta_cuda_interface_get_frame_played_at(self, asset, seek_mode): if seek_mode == "approximate" and not supports_approximate_mode(asset): pytest.skip("asset doesn't work with approximate mode") + if in_fbcode() and asset is AV1_VIDEO: + pytest.skip("AV1 CUDA not supported internally") ref_decoder = VideoDecoder(asset.path, device="cuda", seek_mode=seek_mode) beta_decoder = VideoDecoder( @@ -1607,6 +1614,8 @@ def test_beta_cuda_interface_get_frame_played_at(self, asset, seek_mode): def test_beta_cuda_interface_get_frames_played_at(self, asset, seek_mode): if seek_mode == "approximate" and not supports_approximate_mode(asset): pytest.skip("asset doesn't work with approximate mode") + if in_fbcode() and asset is AV1_VIDEO: + pytest.skip("AV1 CUDA not supported internally") ref_decoder = VideoDecoder(asset.path, device="cuda", seek_mode=seek_mode) beta_decoder = VideoDecoder( @@ -1649,6 +1658,8 @@ def test_beta_cuda_interface_get_frames_played_at(self, asset, seek_mode): def test_beta_cuda_interface_backwards(self, asset, seek_mode): if seek_mode == "approximate" and not supports_approximate_mode(asset): pytest.skip("asset doesn't work with approximate mode") + if in_fbcode() and asset is AV1_VIDEO: + pytest.skip("AV1 CUDA not supported internally") ref_decoder = VideoDecoder(asset.path, device="cuda", seek_mode=seek_mode) beta_decoder = VideoDecoder(