Skip to content

Commit d5d3bdb

Browse files
atalmanpytorchmergebot
authored andcommitted
Fix var CUDA_PATH_V128 in cuda128.bat file (pytorch#146906)
Followup after: pytorch#146653 This should fix upcoming CUDA 12.8 windows builds. Issue found during pytorch-canary Windows AMI test. Pull Request resolved: pytorch#146906 Approved by: https://github.com/malfet, https://github.com/tinglvv
1 parent c7515da commit d5d3bdb

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

.ci/pytorch/windows/cuda128.bat

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -43,8 +43,8 @@ IF "%BUILD_VISION%" == "" (
4343
set NVCC_FLAGS=-D__CUDA_NO_HALF_OPERATORS__ --expt-relaxed-constexpr -gencode=arch=compute_50,code=sm_50 -gencode=arch=compute_60,code=sm_60 -gencode=arch=compute_70,code=sm_70 -gencode=arch=compute_75,code=sm_75 -gencode=arch=compute_80,code=compute_80 -gencode=arch=compute_86,code=compute_86 -gencode=arch=compute_90,code=compute_90 -gencode=arch=compute_100,code=compute_100 -gencode=arch=compute_120,code=compute_120
4444
)
4545

46-
set "CUDA_PATH=%CUDA_PATH_V126%"
47-
set "PATH=%CUDA_PATH_V126%\bin;%PATH%"
46+
set "CUDA_PATH=%CUDA_PATH_V128%"
47+
set "PATH=%CUDA_PATH_V128%\bin;%PATH%"
4848

4949
:optcheck
5050

@@ -56,4 +56,4 @@ call %~dp0\internal\copy.bat
5656
IF ERRORLEVEL 1 goto :eof
5757

5858
call %~dp0\internal\setup.bat
59-
IF ERRORLEVEL 1 goto :eof
59+
IF ERRORLEVEL 1 goto :eof

0 commit comments

Comments
 (0)