Update

malfet · malfet · commit f2323e02d7cd · 2025-06-12T15:25:35.000-07:00
[ghstack-poisoned]
diff --git a/.ci/pytorch/win-build.sh b/.ci/pytorch/win-build.sh
@@ -40,7 +40,7 @@ if [[ $PYLONG_API_CHECK == 0 ]]; then
 fi
 set -ex -o pipefail
 
-"$SCRIPT_HELPERS_DIR"/build_pytorch.ps1
+"$SCRIPT_HELPERS_DIR"/build_pytorch.bat
 
 assert_git_not_dirty
 
diff --git a/.ci/pytorch/windows/internal/build_wheels.bat b/.ci/pytorch/windows/internal/build_wheels.bat
diff --git a/.ci/pytorch/windows/internal/build_wheels.ps1 b/.ci/pytorch/windows/internal/build_wheels.ps1
@@ -0,0 +1,10 @@
+& "windows/internal/vc_install_helper.bat"
+if ($LASTEXITCODE -ne 0) { exit 1 }
+
+& "windows/internal/cuda_install.bat"
+if ($LASTEXITCODE -ne 0) { exit 1 }
+
+& "windows/internal/xpu_install.bat"
+if ($LASTEXITCODE -ne 0) { exit 1 }
+
+. "windows/build_pytorch.ps1" $env:CUDA_VERSION $env:PYTORCH_BUILD_VERSION $env:PYTORCH_BUILD_NUMBER
diff --git a/.circleci/scripts/binary_windows_build.sh b/.circleci/scripts/binary_windows_build.sh
@@ -32,7 +32,7 @@ if [[ "$OS" == "windows-arm64" ]]; then
         ./windows/arm64/build_pytorch.bat
     fi
 else
-    ./windows/internal/build_wheels.bat
+    ./windows/internal/build_wheels.ps1
 fi
 
 echo "Free space on filesystem after build:"
