Update run_tests.sh for r2.4

pytorch · Jun 18, 2024 · 76c4855 · 76c4855
1 parent 0563cb9
commit 76c4855
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/test/run_tests.sh b/test/run_tests.sh
@@ -204,9 +204,7 @@ function run_xla_op_tests2 {
   run_downcast_bf16 "$CDIR/test_data_type.py"
   run_test "$CDIR/pjrt/test_dtypes.py"
   run_test "$CDIR/test_while_loop.py"
-  run_test "$CDIR/test_autocast.py"
-  run_test "$CDIR/eager/test_eager_with_xla_compile.py"
-  run_test "$CDIR/eager/test_eager_with_torch_compile.py"
+  run_test "$CDIR/test_autocast.py"  # TODO(yeounoh) this is expensive on GPU
 }
 
 # All the new xla op tests should go to run_xla_op_tests3