enable 6 modeling cases on XPU (#37571)

Signed-off-by: YAO Matrix <matrix.yao@intel.com>
2025-04-18 18:28:08 +08:00
parent 3cd6627cd7
commit a1b82563f1
4 changed files with 55 additions and 21 deletions
--- a/tests/models/gemma/test_modeling_gemma.py
+++ b/tests/models/gemma/test_modeling_gemma.py
@@ -643,7 +643,7 @@ class GemmaIntegrationTest(unittest.TestCase):
        self.assertEqual(output_text, EXPECTED_TEXTS)

    @slow
-    @require_torch_gpu
+    @require_torch_accelerator
    @require_read_token
    def test_compile_static_cache(self):
        # `torch==2.2` will throw an error on this test (as in other compilation tests), but torch==2.1.2 and torch>2.2