revert changes about using LTC_TS_CUDA flag in rocm tests

hongxiayang · hongxiayang · commit 969b5b654526 · 2023-09-25T16:31:43.000Z
diff --git a/.ci/pytorch/test.sh b/.ci/pytorch/test.sh
@@ -606,9 +606,6 @@ test_libtorch_jit() {
   # Run jit and lazy tensor cpp tests together to finish them faster
   if [[ "$BUILD_ENVIRONMENT" == *cuda* && "$TEST_CONFIG" != *nogpu* ]]; then
     LTC_TS_CUDA=1 python test/run_test.py --cpp --verbose -i cpp/test_jit cpp/nvfuser_tests cpp/test_lazy
-  elif [[ "$BUILD_ENVIRONMENT" == *rocm* && "$TEST_CONFIG" != *nogpu* ]]; then
-    LTC_TS_CUDA=1 python test/run_test.py --cpp --verbose -i cpp/test_lazy
-    python test/run_test.py --cpp --verbose -i cpp/test_jit
   else
     # CUDA tests have already been skipped when CUDA is not available
     python test/run_test.py --cpp --verbose -i cpp/test_jit cpp/test_lazy -k "not CUDA"
