[tests] make more tests device-agnostic (#33580)

* enable * fix * add xpu skip * add marker * skip for xpu * add more * enable on accelerator * add more cases * add more tests * add more
2024-09-20 17:16:43 +08:00
parent 31650a53a1
commit 8bd1f2f338
7 changed files with 24 additions and 17 deletions
--- a/tests/test_modeling_common.py
+++ b/tests/test_modeling_common.py
@@ -4751,7 +4751,7 @@ class ModelTesterMixin:

    # For now, Let's focus only on GPU for `torch.compile`
    @slow
-    @require_torch_gpu
+    @require_torch_accelerator
    @require_read_token
    def test_torch_compile(self):
        if version.parse(torch.__version__) < version.parse("2.3"):