From 1f961ad4959b06f3f1000baa203c46fd23379a0e Mon Sep 17 00:00:00 2001
From: Syed Tousif Ahmed <syeahmed@nvidia.com>
Date: Fri, 19 Jul 2024 12:35:33 +0000
Subject: [PATCH] Runs aten cuda cpp tests in CI (#131061)

It seems like these tests are never run because https://github.com/pytorch/pytorch/pull/99956 got rid of the `pushd $1` which would make the if conditions true in CUDA builds.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/131061
Approved by: https://github.com/malfet, https://github.com/eqy
---
 aten/tools/run_tests.sh | 30 +++++++++++++++---------------
 1 file changed, 15 insertions(+), 15 deletions(-)

diff --git a/aten/tools/run_tests.sh b/aten/tools/run_tests.sh
index 343ec4fdfe35d..32b220505b8e3 100755
--- a/aten/tools/run_tests.sh
+++ b/aten/tools/run_tests.sh
@@ -27,46 +27,46 @@ python test/run_test.py --cpp --verbose -i \
   cpp/legacy_vmap_test \
   cpp/operators_test
 
-if [[ -x ./tensor_interop_test ]]; then
+if [[ -x ${CPP_TESTS_DIR}/tensor_interop_test ]]; then
   python test/run_test.py --cpp --verbose -i cpp/tensor_interop_test
 fi
-if [[ -x ./cudnn_test ]]; then
+if [[ -x ${CPP_TESTS_DIR}/cudnn_test ]]; then
   python test/run_test.py --cpp --verbose -i cpp/cudnn_test
 fi
-if [[ -x ./cuda_generator_test ]]; then
+if [[ -x ${CPP_TESTS_DIR}/cuda_generator_test ]]; then
   python test/run_test.py --cpp --verbose -i cpp/cuda_generator_test
 fi
-if [[ -x ./apply_test ]]; then
+if [[ -x ${CPP_TESTS_DIR}/apply_test ]]; then
   python test/run_test.py --cpp --verbose -i cpp/apply_test
 fi
-if [[ -x ./stream_test ]]; then
+if [[ -x ${CPP_TESTS_DIR}/stream_test ]]; then
   python test/run_test.py --cpp --verbose -i cpp/stream_test
 fi
-if [[ -x ./cuda_half_test ]]; then
+if [[ -x ${CPP_TESTS_DIR}/cuda_half_test ]]; then
   python test/run_test.py --cpp --verbose -i cpp/cuda_half_test
 fi
-if [[ -x ./cuda_vectorized_test ]]; then
+if [[ -x ${CPP_TESTS_DIR}/cuda_vectorized_test ]]; then
   python test/run_test.py --cpp --verbose -i cpp/cuda_vectorized_test
 fi
-if [[ -x ./cuda_distributions_test ]]; then
+if [[ -x ${CPP_TESTS_DIR}/cuda_distributions_test ]]; then
   python test/run_test.py --cpp --verbose -i cpp/cuda_distributions_test
 fi
-if [[ -x ./cuda_optional_test ]]; then
+if [[ -x ${CPP_TESTS_DIR}/cuda_optional_test ]]; then
   python test/run_test.py --cpp --verbose -i cpp/cuda_optional_test
 fi
-if [[ -x ./cuda_tensor_interop_test ]]; then
+if [[ -x ${CPP_TESTS_DIR}/cuda_tensor_interop_test ]]; then
   python test/run_test.py --cpp --verbose -i cpp/cuda_tensor_interop_test
 fi
-if [[ -x ./cuda_complex_test ]]; then
+if [[ -x ${CPP_TESTS_DIR}/cuda_complex_test ]]; then
   python test/run_test.py --cpp --verbose -i cpp/cuda_complex_test
 fi
-if [[ -x ./cuda_complex_math_test ]]; then
+if [[ -x ${CPP_TESTS_DIR}/cuda_complex_math_test ]]; then
   python test/run_test.py --cpp --verbose -i cpp/cuda_complex_math_test
 fi
-if [[ -x ./cuda_cub_test ]]; then
+if [[ -x ${CPP_TESTS_DIR}/cuda_cub_test ]]; then
   python test/run_test.py --cpp --verbose -i cpp/cuda_cub_test
 fi
-if [[ -x ./cuda_atomic_ops_test ]]; then
+if [[ -x ${CPP_TESTS_DIR}/cuda_atomic_ops_test ]]; then
   python test/run_test.py --cpp --verbose -i cpp/cuda_atomic_ops_test
 fi
 
@@ -74,7 +74,7 @@ if [ "$VALGRIND" == "ON" ]; then
   # NB: As these tests are invoked by valgrind, let's leave them for now as it's
   # unclear if valgrind -> python -> gtest would work
   valgrind --suppressions="$VALGRIND_SUP" --error-exitcode=1 "${CPP_TESTS_DIR}/basic" --gtest_filter='-*CUDA'
-  if [[ -x ./tensor_interop_test ]]; then
+  if [[ -x ${CPP_TESTS_DIR}/tensor_interop_test ]]; then
     valgrind --suppressions="$VALGRIND_SUP" --error-exitcode=1 "${CPP_TESTS_DIR}/tensor_interop_test"
   fi
 fi