update_only_test_debug_on_tpu

pytorch · Nov 13, 2023 · ab8d7c1 · ab8d7c1
1 parent 711fc1f
commit ab8d7c1
Showing 1 changed file with 12 additions and 12 deletions.
diff --git 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