Migrate smoke tests to pytorch/pytorch (pytorch#142482)

Related to pytorch/builder#2054 This should fix nightly xpu failure: https://github.com/pytorch/pytorch/actions/runs/12251477588/job/34180135207 and rocm failure: https://github.com/pytorch/pytorch/actions/runs/12251477588/job/34182185374 due to missing : `` /builder/check_binary.sh`` Builder Scripts revision: pytorch/builder@3468139 Pull Request resolved: pytorch#142482 Approved by: https://github.com/chuanqi129, https://github.com/kit1980, https://github.com/malfet, https://github.com/jeffdaily, https://github.com/huydhn
mori360 · Dec 11, 2024 · 868ba10 · 868ba10
1 parent ec23816
commit 868ba10
Show file tree

Hide file tree

Showing 10 changed files with 1,554 additions and 76 deletions.
diff --git a/.ci/manywheel/build_common.sh b/.ci/manywheel/build_common.sh
@@ -490,9 +490,9 @@ if [[ -z "$BUILD_PYTHONLESS" ]]; then
   echo "$(date) :: Running tests"
   pushd "$PYTORCH_ROOT"
 
-  #TODO: run_tests.sh and check_binary.sh should be moved to pytorch/pytorch project
+
   LD_LIBRARY_PATH=/usr/local/nvidia/lib64 \
-          "/builder/run_tests.sh" manywheel "${py_majmin}" "$DESIRED_CUDA"
+          "${PYTORCH_ROOT}/.ci/pytorch/run_tests.sh" manywheel "${py_majmin}" "$DESIRED_CUDA"
   popd
   echo "$(date) :: Finished tests"
 fi