add xpu skip (#3119)

huggingface · Sep 18, 2024 · 4305033 · 4305033
1 parent 4617be3
commit 4305033
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git 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
@@ -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