diff --git a/tests/test_examples.py b/tests/test_examples.py index efb5e7bbc3a..a16dce52f92 100644 --- a/tests/test_examples.py +++ b/tests/test_examples.py @@ -30,6 +30,7 @@ require_huggingface_suite, require_multi_device, require_multi_gpu, + require_non_xpu, require_pippy, require_schedulefree, require_trackers, @@ -274,12 +275,14 @@ def test_distributed_inference_examples_phi2(self): testargs = ["examples/inference/distributed/phi2.py"] run_command(self.launch_args + testargs) + @require_non_xpu @require_pippy @require_multi_gpu def test_pippy_examples_bert(self): testargs = ["examples/inference/pippy/bert.py"] run_command(self.launch_args + testargs) + @require_non_xpu @require_pippy @require_multi_gpu def test_pippy_examples_gpt2(self): diff --git a/tests/test_multigpu.py b/tests/test_multigpu.py index d4063e520ba..76652e76ff5 100644 --- a/tests/test_multigpu.py +++ b/tests/test_multigpu.py @@ -30,6 +30,7 @@ require_multi_device, require_multi_gpu, require_non_torch_xla, + require_non_xpu, require_pippy, require_torchvision, torch_device, @@ -93,6 +94,7 @@ def test_distributed_data_loop(self): with patch_environment(**env_kwargs): execute_subprocess_async(cmd) + @require_non_xpu @require_multi_gpu @require_pippy @require_torchvision