diff --git a/tests/unit_tests/pipeline_parallel/test_fine_grained_activation_offloading.py b/tests/unit_tests/pipeline_parallel/test_fine_grained_activation_offloading.py index 8508d8c14b5..5b3209841ae 100644 --- a/tests/unit_tests/pipeline_parallel/test_fine_grained_activation_offloading.py +++ b/tests/unit_tests/pipeline_parallel/test_fine_grained_activation_offloading.py @@ -334,6 +334,7 @@ def test_gpt_fine_grained_activation_offloading_correctness_and_memory( Utils.destroy_model_parallel() +@pytest.mark.flaky_in_dev @pytest.mark.skipif(not torch.cuda.is_available(), reason="CUDA is required for offloading tests.") @pytest.mark.skipif( not is_te_min_version("1.9.0.dev0"),