Update test_modeling_common.py

ROCm · Nov 9, 2024 · 5fba946 · 5fba946
1 parent 0048de1
commit 5fba946
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/test_modeling_common.py b/tests/test_modeling_common.py
@@ -2885,6 +2885,7 @@ def test_inputs_embeds_matches_input_ids_with_generate(self):
                 )
             self.assertTrue(torch.allclose(out_embeds, out_ids))
 
+    @skipIfRocm
     @require_torch_multi_gpu
     def test_multi_gpu_data_parallel_forward(self):
         config, inputs_dict = self.model_tester.prepare_config_and_inputs_for_common()