Build Windows ARM64 binaries cross-compiled on x64

Author: Blackhex

windows/build_pytorch.bat

@@ -1,5 +1,3 @@
-@echo off
-
 :: This script parses args, installs required libraries (miniconda, MKL,
 :: Magma), and then delegates to cpu.bat, cuda80.bat, etc.
 

windows/cpu.bat

@@ -1,5 +1,3 @@
-@echo off
-
 set MODULE_NAME=pytorch
 
 IF NOT EXIST "setup.py" IF NOT EXIST "%MODULE_NAME%" (

windows/internal/setup.bat

@@ -1,19 +1,39 @@
-@echo off
+@echo on
 
 echo The flags after configuring:
 echo USE_CUDA=%USE_CUDA%
 echo CMAKE_GENERATOR=%CMAKE_GENERATOR%
+echo CROSS_COMPILE_ARM64=%CROSS_COMPILE_ARM64%
+
 if "%USE_CUDA%"==""  echo CUDA_PATH=%CUDA_PATH%
 if NOT "%CC%"==""   echo CC=%CC%
 if NOT "%CXX%"==""  echo CXX=%CXX%
 if NOT "%DISTUTILS_USE_SDK%"==""  echo DISTUTILS_USE_SDK=%DISTUTILS_USE_SDK%
 
 set SRC_DIR=%~dp0\..
 
+set
+
+if "%CROSS_COMPILE_ARM64%" == "" (
+    set VSDEVCMD_ARCH=x64
+) else (
+    set VSDEVCMD_ARCH=x64_arm64
+    set CMAKE_TOOLCHAIN_FILE=cmake\Toolchains\msvc2022-x64-arm64.cmake
+    set USE_HOST_PROTOC=1
+    set PROTOC=%CONDA_HOME%\Library\bin\protoc.exe
+)
+
+if EXIST %PROTOC% (
+    echo "%PROTOC% is installed"
+) else (
+    echo "%PROTOC% is not installed"
+    call protoc.exe
+)
+
 IF "%VSDEVCMD_ARGS%" == "" (
-    call "%VS15VCVARSALL%" x64
+    call "%VS15VCVARSALL%" %VSDEVCMD_ARCH%
 ) ELSE (
-    call "%VS15VCVARSALL%" x64 %VSDEVCMD_ARGS%
+    call "%VS15VCVARSALL%" %VSDEVCMD_ARCH% %VSDEVCMD_ARGS%
 )
 
 pushd %SRC_DIR%

windows/internal/vs2022_install.ps1

@@ -4,15 +4,21 @@
 # 17.4.3 BuildTools
 $VS_DOWNLOAD_LINK = "https://download.visualstudio.microsoft.com/download/pr/8f480125-28b8-4a2c-847c-c2b02a8cdd1b/64be21d4ada005d7d07896ed0b004c322409bd04d6e8eba4c03c9fa39c928e7a/vs_BuildTools.exe"
 $COLLECT_DOWNLOAD_LINK = "https://aka.ms/vscollect.exe"
-$VS_INSTALL_ARGS = @("--nocache","--quiet","--wait", "--add Microsoft.VisualStudio.Workload.VCTools",
-                                                     "--add Microsoft.Component.MSBuild",
-                                                     "--add Microsoft.VisualStudio.Component.Roslyn.Compiler",
-                                                     "--add Microsoft.VisualStudio.Component.TextTemplating",
-                                                     "--add Microsoft.VisualStudio.Component.VC.CoreIde",
-                                                     "--add Microsoft.VisualStudio.Component.VC.Redist.14.Latest",
-                                                     "--add Microsoft.VisualStudio.ComponentGroup.NativeDesktop.Core",
-                                                     "--add Microsoft.VisualStudio.Component.VC.Tools.x86.x64",
-                                                     "--add Microsoft.VisualStudio.ComponentGroup.NativeDesktop.Win81")
+$VS_INSTALL_ARGS = @(
+    "--nocache",
+    "--quiet",
+    "--wait",
+    "--add Microsoft.VisualStudio.Workload.VCTools",
+    "--add Microsoft.Component.MSBuild",
+    "--add Microsoft.VisualStudio.Component.Roslyn.Compiler",
+    "--add Microsoft.VisualStudio.Component.TextTemplating",
+    "--add Microsoft.VisualStudio.Component.VC.CoreIde",
+    "--add Microsoft.VisualStudio.Component.VC.Redist.14.Latest",
+    "--add Microsoft.VisualStudio.ComponentGroup.NativeDesktop.Core",
+    "--add Microsoft.VisualStudio.Component.VC.Tools.x86.x64",
+    "--add Microsoft.VisualStudio.Component.VC.Tools.ARM64",
+    "--add Microsoft.VisualStudio.Component.Windows10SDK.20348"
+)
 
 curl.exe --retry 3 -kL $VS_DOWNLOAD_LINK --output vs_installer.exe
 if ($LASTEXITCODE -ne 0) {