mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
d0e70c4fd3
https://github.com/pytorch/pytorch/issues/145570 removing `.ci/pytorch/windows/internal/cuda_install.bat` as it is a duplicate with` .github/scripts/windows/cuda_install.bat`. The later one is the one in use - https://github.com/pytorch/pytorch/pull/146653/files#diff-613791f266f2f7b81148ca8f447b0cd6c6544f824f5f46a78a2794006c78957bR8 Pull Request resolved: https://github.com/pytorch/pytorch/pull/146653 Approved by: https://github.com/atalman Co-authored-by: atalman <atalman@fb.com>
69 lines
2.7 KiB
Batchfile
69 lines
2.7 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%" == "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:~0,2%" == "12" if NOT "%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
|
|
)
|
|
if "%CUVER_NODOT%" == "118" (
|
|
set CUDA_ARCH_LIST= -gencode arch=compute_37,code=sm_37 -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%"
|
|
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
|