Skipping failing CI tests for now.

pytorch · Feb 11, 2025 · bbceeb3 · bbceeb3
1 parent 960bbf4
commit bbceeb3
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 1 deletion.
diff --git a/test/test_triton.py b/test/test_triton.py
@@ -6,6 +6,7 @@
 import torch_xla.experimental.triton as xla_triton
 import torch_xla
 from torch_xla import runtime as xr
+from torch_xla.test.test_utils import skipIfCUDA
 
 import triton
 import triton.language as tl
@@ -241,6 +242,8 @@ def _attn_fwd(
   tl.store(O_block_ptr, acc.to(Out.type.element_ty))
 
 
+# Ref: https://github.com/pytorch/xla/pull/8593
+@skipIfCUDA("GPU CI is failing")
 class TritonTest(unittest.TestCase):
 
   @unittest.skipIf(xr.device_type() != 'CUDA', "This test only works on GPU.")

diff --git a/test/test_utils.py b/test/test_utils.py
@@ -13,6 +13,12 @@
 import torch_xla
 import torch_xla.core.xla_model as xm
 import torch_xla.utils.utils as xu
+import torch_xla.runtime as xr
+
+
+def skipIfCUDA(reason):
+  accelerator = xr.device_type() or ""
+  return lambda f: unittest.skipIf(accelerator.lower() == "cuda", reason)(f)
 
 
 def _set_rng_seed(seed):

diff --git a/test/torch_distributed/test_ddp.py b/test/torch_distributed/test_ddp.py
@@ -3,6 +3,7 @@
 import sys
 import torch_xla
 import torch_xla.core.xla_model as xm
+from torch_xla.test.test_utils import skipIfCUDA
 
 # Setup import folders.
 xla_test_folder = os.path.dirname(os.path.dirname(os.path.abspath(sys.argv[0])))
@@ -13,7 +14,6 @@
 
 FLAGS = args_parse.parse_common_options()
 
-
 class TestXrtDistributedDataParallel(parameterized.TestCase):
 
   @staticmethod
@@ -38,6 +38,8 @@ def _ddp_correctness(rank,
   def test_ddp_correctness(self):
     torch_xla.launch(self._ddp_correctness, args=(False, FLAGS.debug))
 
+  # Ref: https://github.com/pytorch/xla/pull/8593
+  @skipIfCUDA("GPU CI is failing")
   def test_ddp_correctness_with_gradient_as_bucket_view(self):
     torch_xla.launch(self._ddp_correctness, args=(False, FLAGS.debug, True))