From ab8d7c1928ec752b328e3860c8f249a6396dc833 Mon Sep 17 00:00:00 2001
From: Manfei <41607353+ManfeiBai@users.noreply.github.com>
Date: Mon, 13 Nov 2023 10:41:23 -0800
Subject: [PATCH] update_only_test_debug_on_tpu

---
 test/tpu/xla_test_job.yaml | 24 ++++++++++++------------
 1 file changed, 12 insertions(+), 12 deletions(-)

diff --git a/test/tpu/xla_test_job.yaml b/test/tpu/xla_test_job.yaml
index b02356064db..b7d37726837 100644
--- a/test/tpu/xla_test_job.yaml
+++ b/test/tpu/xla_test_job.yaml
@@ -43,18 +43,18 @@ spec:
     - |
       pip install expecttest
 
-      python3 /src/pytorch/xla/test/test_operations.py -v
-      python3 /src/pytorch/xla/test/pjrt/test_runtime_tpu.py
-      python3 /src/pytorch/xla/test/pjrt/test_collective_ops_tpu.py
-      python3 /src/pytorch/xla/test/spmd/test_xla_sharding.py
-      python3 /src/pytorch/xla/test/spmd/test_xla_virtual_device.py
-      python3 /src/pytorch/xla/test/spmd/test_xla_distributed_checkpoint.py
-      python3 /src/pytorch/xla/test/spmd/test_train_spmd_linear_model.py
-      python3 /src/pytorch/xla/test/spmd/test_xla_spmd_python_api_interaction.py
-      XLA_EXPERIMENTAL=nonzero:masked_select python3 /src/pytorch/xla/test/ds/test_dynamic_shape_models.py -v
-      XLA_EXPERIMENTAL=nonzero:masked_select python3 /src/pytorch/xla/test/ds/test_dynamic_shapes.py -v
-      python3 /src/pytorch/xla/test/test_autocast.py
-      python3 /src/pytorch/xla/test/dynamo/test_dynamo.py
+      # python3 /src/pytorch/xla/test/test_operations.py -v
+      # python3 /src/pytorch/xla/test/pjrt/test_runtime_tpu.py
+      # python3 /src/pytorch/xla/test/pjrt/test_collective_ops_tpu.py
+      # python3 /src/pytorch/xla/test/spmd/test_xla_sharding.py
+      # python3 /src/pytorch/xla/test/spmd/test_xla_virtual_device.py
+      # python3 /src/pytorch/xla/test/spmd/test_xla_distributed_checkpoint.py
+      # python3 /src/pytorch/xla/test/spmd/test_train_spmd_linear_model.py
+      # python3 /src/pytorch/xla/test/spmd/test_xla_spmd_python_api_interaction.py
+      # XLA_EXPERIMENTAL=nonzero:masked_select python3 /src/pytorch/xla/test/ds/test_dynamic_shape_models.py -v
+      # XLA_EXPERIMENTAL=nonzero:masked_select python3 /src/pytorch/xla/test/ds/test_dynamic_shapes.py -v
+      # python3 /src/pytorch/xla/test/test_autocast.py
+      # python3 /src/pytorch/xla/test/dynamo/test_dynamo.py
       python3 /src/pytorch/xla/test/spmd/test_spmd_debugging.py
     volumeMounts:
     - mountPath: /dev/shm