xref: /aosp_15_r20/external/pytorch/.ci/pytorch/win-test-helpers/installation-helpers/install_magma.bat (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1if "%CUDA_VERSION%" == "cpu" (
2  echo skip magma installation for cpu builds
3  exit /b 0
4)
5
6rem remove dot in cuda_version, fox example 11.1 to 111
7
8if not "%USE_CUDA%"=="1" (
9    exit /b 0
10)
11
12if x%CUDA_VERSION:.=%==x%CUDA_VERSION% (
13    echo CUDA version %CUDA_VERSION% format isn't correct, which doesn't contain '.'
14    exit /b 1
15)
16
17set VERSION_SUFFIX=%CUDA_VERSION:.=%
18set CUDA_SUFFIX=cuda%VERSION_SUFFIX%
19
20if "%CUDA_SUFFIX%" == "" (
21  echo unknown CUDA version, please set `CUDA_VERSION` higher than 10.2
22  exit /b 1
23)
24
25if "%REBUILD%"=="" (
26  if "%BUILD_ENVIRONMENT%"=="" (
27    curl --retry 3 --retry-all-errors -k https://s3.amazonaws.com/ossci-windows/magma_2.5.4_%CUDA_SUFFIX%_%BUILD_TYPE%.7z --output %TMP_DIR_WIN%\magma_2.5.4_%CUDA_SUFFIX%_%BUILD_TYPE%.7z
28  ) else (
29    aws s3 cp s3://ossci-windows/magma_2.5.4_%CUDA_SUFFIX%_%BUILD_TYPE%.7z %TMP_DIR_WIN%\magma_2.5.4_%CUDA_SUFFIX%_%BUILD_TYPE%.7z --quiet
30  )
31  if errorlevel 1 exit /b
32  if not errorlevel 0 exit /b
33  7z x -aoa %TMP_DIR_WIN%\magma_2.5.4_%CUDA_SUFFIX%_%BUILD_TYPE%.7z -o%TMP_DIR_WIN%\magma
34  if errorlevel 1 exit /b
35  if not errorlevel 0 exit /b
36)
37set MAGMA_HOME=%TMP_DIR_WIN%\magma
38