Test

mbzomowski-test-org · Jan 3, 2024 · 8c62d3b · 8c62d3b
1 parent 1f7f01a
commit 8c62d3b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/.github/workflows/tpu_ci.yml b/.github/workflows/tpu_ci.yml
@@ -45,4 +45,5 @@ jobs:
           XLA_EXPERIMENTAL=nonzero:masked_select python3 -u test/ds/test_dynamic_shapes.py -v
           python3 -u test/test_autocast.py
           python3 -u test/dynamo/test_dynamo.py
-          python3 -u test/spmd/test_spmd_debugging.py
+          # TODO(mbzomowski): Uncomment once https://github.com/pytorch/xla/issues/6252 is fixed
+          # python3 -u test/spmd/test_spmd_debugging.py