[tests] skip tests for xpu (#33553)

* enable

* fix

* add xpu skip

* add marker

* skip for xpu

* add more

* add one more
This commit is contained in:
Fanli Lin
2024-09-20 02:28:04 +08:00
committed by GitHub
parent f111d5b783
commit b87755aa6d
6 changed files with 26 additions and 1 deletions

View File

@@ -31,6 +31,7 @@ from transformers.testing_utils import (
get_torch_dist_unique_port,
require_apex,
require_bitsandbytes,
require_non_xpu,
require_torch,
require_torch_gpu,
require_torch_multi_accelerator,
@@ -106,6 +107,7 @@ class TestTrainerExt(TestCasePlus):
def test_run_seq2seq_ddp(self):
self.run_seq2seq_quick(distributed=True)
@require_non_xpu
@require_apex
@require_torch_gpu
def test_run_seq2seq_apex(self):