File tree 2 files changed +5
-0
lines changed
2 files changed +5
-0
lines changed Original file line number Diff line number Diff line change 30
30
require_huggingface_suite ,
31
31
require_multi_device ,
32
32
require_multi_gpu ,
33
+ require_non_xpu ,
33
34
require_pippy ,
34
35
require_schedulefree ,
35
36
require_trackers ,
@@ -274,12 +275,14 @@ def test_distributed_inference_examples_phi2(self):
274
275
testargs = ["examples/inference/distributed/phi2.py" ]
275
276
run_command (self .launch_args + testargs )
276
277
278
+ @require_non_xpu
277
279
@require_pippy
278
280
@require_multi_gpu
279
281
def test_pippy_examples_bert (self ):
280
282
testargs = ["examples/inference/pippy/bert.py" ]
281
283
run_command (self .launch_args + testargs )
282
284
285
+ @require_non_xpu
283
286
@require_pippy
284
287
@require_multi_gpu
285
288
def test_pippy_examples_gpt2 (self ):
Original file line number Diff line number Diff line change 30
30
require_multi_device ,
31
31
require_multi_gpu ,
32
32
require_non_torch_xla ,
33
+ require_non_xpu ,
33
34
require_pippy ,
34
35
require_torchvision ,
35
36
torch_device ,
@@ -93,6 +94,7 @@ def test_distributed_data_loop(self):
93
94
with patch_environment (** env_kwargs ):
94
95
execute_subprocess_async (cmd )
95
96
97
+ @require_non_xpu
96
98
@require_multi_gpu
97
99
@require_pippy
98
100
@require_torchvision
You can’t perform that action at this time.
0 commit comments