update test markers

ROCm · Jan 21, 2025 · af673a5 · af673a5
1 parent 4a264fc
commit af673a5
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 3 deletions.
diff --git a/tests/unit_tests/dist_checkpointing/models/test_moe_experts.py b/tests/unit_tests/dist_checkpointing/models/test_moe_experts.py
@@ -291,7 +291,7 @@ def test_sequential_grouped_mlp_interchangeable(
             ('te_grouped', 'sequential', (1, 1, 4), (1, 1, 1)),
         ],
     )
-    @pytest.mark.failing_on_rocm
+    @pytest.mark.failing_on_upstream
     def test_sequential_grouped_mlp_extra_state(
         self, tmp_path_dist_ckpt, src_tp_pp_exp, dest_tp_pp_exp, src_module, dst_module
     ):

diff --git a/tests/unit_tests/inference/model_inference_wrappers/t5/test_t5_inference_wrapper.py b/tests/unit_tests/inference/model_inference_wrappers/t5/test_t5_inference_wrapper.py
@@ -90,7 +90,6 @@ def setup_model(self, tensor_parallel_size, pipeline_parallel_size):
     def teardown_method(self, method):
         Utils.destroy_model_parallel()
 
-    @pytest.mark.failing_on_rocm
     def test_inference_only_tensor_parallel(self):
         self.setup_model(tensor_parallel_size=4, pipeline_parallel_size=1)
 

diff --git a/.../inference/text_generation_controllers/test_encoder_decoder_text_generation_controller.py b/.../inference/text_generation_controllers/test_encoder_decoder_text_generation_controller.py
@@ -103,7 +103,6 @@ def setup_method(self, method):
     def teardown_method(self, method):
         Utils.destroy_model_parallel()
 
-    @pytest.mark.failing_on_rocm
     def test_generate_all_output_tokens_static_batch(self):
         self.mock_tokenizer.vocab_size = self.vocab_size
         self.mock_tokenizer.eod = self.vocab_size - 1