mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
Update MAGMA to 2.5.2 for Windows (#34205)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/34205 Differential Revision: D20248224 Pulled By: soumith fbshipit-source-id: f5e0fe06aa8f8ee551abe45db1d55d06e95ab928
This commit is contained in:
parent
f6c883ccea
commit
5f4a01b2ea
2 changed files with 4 additions and 3 deletions
|
|
@ -8,10 +8,10 @@ if "%CUDA_SUFFIX%" == "" (
|
|||
|
||||
if "%REBUILD%"=="" (
|
||||
if "%BUILD_ENVIRONMENT%"=="" (
|
||||
curl --retry 3 -k https://s3.amazonaws.com/ossci-windows/magma_2.5.1_%CUDA_SUFFIX%_%BUILD_TYPE%.7z --output %TMP_DIR_WIN%\magma_2.5.1_%CUDA_SUFFIX%_%BUILD_TYPE%.7z
|
||||
curl --retry 3 -k https://s3.amazonaws.com/ossci-windows/magma_2.5.2_%CUDA_SUFFIX%_%BUILD_TYPE%.7z --output %TMP_DIR_WIN%\magma_2.5.2_%CUDA_SUFFIX%_%BUILD_TYPE%.7z
|
||||
) else (
|
||||
aws s3 cp s3://ossci-windows/magma_2.5.1_%CUDA_SUFFIX%_%BUILD_TYPE%.7z %TMP_DIR_WIN%\magma_2.5.1_%CUDA_SUFFIX%_%BUILD_TYPE%.7z --quiet
|
||||
aws s3 cp s3://ossci-windows/magma_2.5.2_%CUDA_SUFFIX%_%BUILD_TYPE%.7z %TMP_DIR_WIN%\magma_2.5.2_%CUDA_SUFFIX%_%BUILD_TYPE%.7z --quiet
|
||||
)
|
||||
7z x -aoa %TMP_DIR_WIN%\magma_2.5.1_%CUDA_SUFFIX%_%BUILD_TYPE%.7z -o%TMP_DIR_WIN%\magma
|
||||
7z x -aoa %TMP_DIR_WIN%\magma_2.5.2_%CUDA_SUFFIX%_%BUILD_TYPE%.7z -o%TMP_DIR_WIN%\magma
|
||||
)
|
||||
set MAGMA_HOME=%TMP_DIR_WIN%\magma
|
||||
|
|
|
|||
|
|
@ -20,6 +20,7 @@ MKL and MAGMA. Here are the steps to build with them.
|
|||
|
||||
REM Download MAGMA files
|
||||
REM version available:
|
||||
REM 2.5.2 (CUDA 9.2 10.0 10.1 10.2) x (Debug Release)
|
||||
REM 2.5.1 (CUDA 9.2 10.0 10.1 10.2) x (Debug Release)
|
||||
REM 2.5.0 (CUDA 9.0 9.2 10.0 10.1) x (Debug Release)
|
||||
REM 2.4.0 (CUDA 8.0 9.2) x (Release)
|
||||
|
|
|
|||
Loading…
Reference in a new issue