mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
a68f63e331
Add magma build 13.0 for Windows Add cuda_install.bat 13.0 for Windows build https://github.com/pytorch/pytorch/issues/159779 Pull Request resolved: https://github.com/pytorch/pytorch/pull/161073 Approved by: https://github.com/atalman Co-authored-by: Andrey Talman <atalman@fb.com>
74 lines
2.9 KiB
Batchfile
74 lines
2.9 KiB
Batchfile
@setlocal
|
|
|
|
set MAGMA_VERSION=2.5.4
|
|
|
|
set CUVER_NODOT=%CUDA_VERSION%
|
|
set CUVER=%CUVER_NODOT:~0,-1%.%CUVER_NODOT:~-1,1%
|
|
|
|
set CONFIG_LOWERCASE=%CONFIG:D=d%
|
|
set CONFIG_LOWERCASE=%CONFIG_LOWERCASE:R=r%
|
|
set CONFIG_LOWERCASE=%CONFIG_LOWERCASE:M=m%
|
|
|
|
echo Building for configuration: %CONFIG_LOWERCASE%, %CUVER%
|
|
|
|
:: Download Ninja
|
|
curl -k https://s3.amazonaws.com/ossci-windows/ninja_1.8.2.exe --output C:\Tools\ninja.exe
|
|
if errorlevel 1 exit /b 1
|
|
|
|
set "PATH=C:\Tools;C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v%CUVER%\bin;C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v%CUVER%\libnvvp;%PATH%"
|
|
set CUDA_PATH=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v%CUVER%
|
|
set NVTOOLSEXT_PATH=C:\Program Files\NVIDIA Corporation\NvToolsExt
|
|
|
|
mkdir magma_cuda%CUVER_NODOT%
|
|
cd magma_cuda%CUVER_NODOT%
|
|
|
|
if not exist magma (
|
|
:: MAGMA 2.5.4 from http://icl.utk.edu/projectsfiles/magma/downloads/ with applied patches from our magma folder
|
|
git clone https://github.com/ptrblck/magma_win.git magma
|
|
if errorlevel 1 exit /b 1
|
|
) else (
|
|
rmdir /S /Q magma\build
|
|
rmdir /S /Q magma\install
|
|
)
|
|
|
|
cd magma
|
|
mkdir build && cd build
|
|
|
|
set GPU_TARGET=All
|
|
if "%CUVER_NODOT%" == "130" (
|
|
set CUDA_ARCH_LIST=-gencode=arch=compute_75,code=sm_75 -gencode arch=compute_80,code=sm_80 -gencode arch=compute_86,code=sm_86 -gencode arch=compute_90,code=sm_90 -gencode arch=compute_100,code=sm_100 -gencode arch=compute_120,code=sm_120
|
|
)
|
|
if "%CUVER_NODOT%" == "129" (
|
|
set CUDA_ARCH_LIST=-gencode=arch=compute_75,code=sm_75 -gencode arch=compute_80,code=sm_80 -gencode arch=compute_86,code=sm_86 -gencode arch=compute_90,code=sm_90 -gencode arch=compute_100,code=sm_100 -gencode arch=compute_120,code=sm_120
|
|
)
|
|
if "%CUVER_NODOT%" == "128" (
|
|
set CUDA_ARCH_LIST=-gencode arch=compute_50,code=sm_50 -gencode arch=compute_60,code=sm_60 -gencode arch=compute_70,code=sm_70 -gencode arch=compute_80,code=sm_80 -gencode arch=compute_86,code=sm_86 -gencode arch=compute_90,code=sm_90 -gencode arch=compute_100,code=sm_100 -gencode arch=compute_120,code=sm_120
|
|
)
|
|
if "%CUVER_NODOT%" == "126" (
|
|
set CUDA_ARCH_LIST=-gencode arch=compute_50,code=sm_50 -gencode arch=compute_60,code=sm_60 -gencode arch=compute_70,code=sm_70 -gencode arch=compute_80,code=sm_80 -gencode arch=compute_86,code=sm_86 -gencode arch=compute_90,code=sm_90
|
|
)
|
|
|
|
set CC=cl.exe
|
|
set CXX=cl.exe
|
|
|
|
cmake .. -DGPU_TARGET="%GPU_TARGET%" ^
|
|
-DUSE_FORTRAN=0 ^
|
|
-DCMAKE_CXX_FLAGS="/FS /Zf" ^
|
|
-DCMAKE_BUILD_TYPE=%CONFIG% ^
|
|
-DCMAKE_GENERATOR=Ninja ^
|
|
-DCMAKE_INSTALL_PREFIX=..\install\ ^
|
|
-DCUDA_ARCH_LIST="%CUDA_ARCH_LIST%" ^
|
|
-DCMAKE_POLICY_VERSION_MINIMUM=3.5
|
|
if errorlevel 1 exit /b 1
|
|
|
|
cmake --build . --target install --config %CONFIG% -- -j%NUMBER_OF_PROCESSORS%
|
|
if errorlevel 1 exit /b 1
|
|
|
|
cd ..\..\..
|
|
|
|
:: Create
|
|
7z a magma_%MAGMA_VERSION%_cuda%CUVER_NODOT%_%CONFIG_LOWERCASE%.7z %cd%\magma_cuda%CUVER_NODOT%\magma\install\*
|
|
|
|
rmdir /S /Q magma_cuda%CUVER_NODOT%\
|
|
@endlocal
|