diff --git a/tests/unittests/image/test_ms_ssim.py b/tests/unittests/image/test_ms_ssim.py index 89b9b5778bd..53989382519 100644 --- a/tests/unittests/image/test_ms_ssim.py +++ b/tests/unittests/image/test_ms_ssim.py @@ -127,6 +127,7 @@ def _run_ms_ssim_ddp(rank: int, world_size: int, free_port: int): cleanup_ddp() +@pytest.mark.DDP @pytest.mark.skipif(not torch.cuda.is_available(), reason="test requires cuda") @pytest.mark.skipif(_IS_WINDOWS, reason="DDP not supported on Windows") def test_ms_ssim_reduction_none_ddp(): diff --git a/tests/unittests/image/test_ssim.py b/tests/unittests/image/test_ssim.py index e327e7d7f70..4f433d54d9c 100644 --- a/tests/unittests/image/test_ssim.py +++ b/tests/unittests/image/test_ssim.py @@ -382,6 +382,7 @@ def _run_ssim_ddp(rank: int, world_size: int, free_port: int): cleanup_ddp() +@pytest.mark.DDP @pytest.mark.skipif(not torch.cuda.is_available(), reason="test requires cuda") @pytest.mark.skipif(_IS_WINDOWS, reason="DDP not supported on Windows") def test_ssim_reduction_none_ddp():