Skip to content

Commit e5aa053

Browse files
committed
add xpu skip
1 parent 4617be3 commit e5aa053

File tree

2 files changed

+5
-0
lines changed

2 files changed

+5
-0
lines changed

tests/test_examples.py

+3
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,7 @@
3030
require_huggingface_suite,
3131
require_multi_device,
3232
require_multi_gpu,
33+
require_non_xpu,
3334
require_pippy,
3435
require_schedulefree,
3536
require_trackers,
@@ -274,12 +275,14 @@ def test_distributed_inference_examples_phi2(self):
274275
testargs = ["examples/inference/distributed/phi2.py"]
275276
run_command(self.launch_args + testargs)
276277

278+
@require_non_xpu
277279
@require_pippy
278280
@require_multi_gpu
279281
def test_pippy_examples_bert(self):
280282
testargs = ["examples/inference/pippy/bert.py"]
281283
run_command(self.launch_args + testargs)
282284

285+
@require_non_xpu
283286
@require_pippy
284287
@require_multi_gpu
285288
def test_pippy_examples_gpt2(self):

tests/test_multigpu.py

+2
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,7 @@
3030
require_multi_device,
3131
require_multi_gpu,
3232
require_non_torch_xla,
33+
require_non_xpu,
3334
require_pippy,
3435
require_torchvision,
3536
torch_device,
@@ -93,6 +94,7 @@ def test_distributed_data_loop(self):
9394
with patch_environment(**env_kwargs):
9495
execute_subprocess_async(cmd)
9596

97+
@require_non_xpu
9698
@require_multi_gpu
9799
@require_pippy
98100
@require_torchvision

0 commit comments

Comments
 (0)