mi300 skips

ROCm · Jul 26, 2024 · df67b09 · df67b09
1 parent 1be2e0b
commit df67b09
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tests/models/layoutlmv2/test_modeling_layoutlmv2.py b/tests/models/layoutlmv2/test_modeling_layoutlmv2.py
@@ -16,7 +16,7 @@
 
 import unittest
 
-from transformers.testing_utils import require_detectron2, require_torch, require_torch_multi_gpu, slow, torch_device
+from transformers.testing_utils import require_detectron2, require_torch, require_torch_multi_gpu, slow, torch_device, skipIfRocm
 from transformers.utils import is_detectron2_available, is_torch_available
 
 from ...test_configuration_common import ConfigTester
@@ -274,6 +274,10 @@ class LayoutLMv2ModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCa
         else {}
     )
 
+    @skipIfRocm
+    def test_inputs_embeds_matches_input_ids(self):
+        super().test_inputs_embeds_matches_input_ids()  
+
     def setUp(self):
         self.model_tester = LayoutLMv2ModelTester(self)
         self.config_tester = ConfigTester(self, config_class=LayoutLMv2Config, hidden_size=37)