[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:
@@ -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):
|
||||
|
||||
Reference in New Issue
Block a user