Fix linalg testing

pytorch · Dec 10, 2023 · 5d7e8e1 · 5d7e8e1
1 parent 70d5c5f
commit 5d7e8e1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/check_binary.sh b/check_binary.sh
@@ -412,7 +412,7 @@ fi # if cuda
 # Run parts of smoke tests
 ##########################
 if [[ "$PACKAGE_TYPE" != 'libtorch' ]]; then
-  pushd test/smoke_test
+  pushd "$(dirname ${BASH_SOURCE[0]})/test/smoke_test"
   python -c "from smoke_test import test_linalg; test_linalg()"
   if [[ "$DESIRED_CUDA" == *cuda* ]]; then
     python -c "from smoke_test import test_linalg; test_linalg('cuda')"