Run smoke_test_linalg during check_binary

This is a regression test for pytorch/pytorch#114862
pytorch · Dec 10, 2023 · 70d5c5f · 70d5c5f
1 parent 4f298cb
commit 70d5c5f
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/check_binary.sh b/check_binary.sh
@@ -408,6 +408,18 @@ if [[ "$DESIRED_CUDA" != 'cpu' && "$DESIRED_CUDA" != 'cpu-cxx11-abi' && "$DESIRE
   fi # if libtorch
 fi # if cuda
 
+##########################
+# Run parts of smoke tests
+##########################
+if [[ "$PACKAGE_TYPE" != 'libtorch' ]]; then
+  pushd 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')"
+  fi
+  popd
+fi
+
 ###############################################################################
 # Check PyTorch supports TCP_TLS gloo transport
 ###############################################################################