Avoid running tpu_info cli test on Trillium (#8257)

pytorch · Oct 14, 2024 · ecdc68b · ecdc68b
1 parent 7760d87
commit ecdc68b
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/test/tpu/run_tests.sh b/test/tpu/run_tests.sh
@@ -56,5 +56,7 @@ if [[ -n "$TPU_VERSION" && "$TPU_VERSION" == "4" ]]; then
     XLA_EXPERIMENTAL=nonzero:masked_select:nms python3 test/ds/test_dynamic_shapes.py -v
 fi
 
-# Test `tpu-info` CLI compatibility
-python3 test/tpu/tpu_info/test_cli.py
+if [[ -n "$TPU_VERSION" && "$TPU_VERSION" != "6" ]]; then
+    # Test `tpu-info` CLI compatibility
+    python3 test/tpu/tpu_info/test_cli.py
+fi