Skip to content

Commit d6e8411

Browse files
pytorchbotatalman
andauthored
Make sure Windows CUDA 12.8 build follow same arches as Linux builds (pytorch#164477)
Make sure Windows CUDA 12.8 build follow same arches as Linux builds (pytorch#164470) I believe ``set TORCH_CUDA_ARCH_LIST=7.0;7.5;8.0;8.6;9.0;10.0;12.0`` is the one thats actually used. Hence remove 6.1 to align the support with Linux support. Pull Request resolved: pytorch#164470 Approved by: https://github.com/tinglvv, https://github.com/nWEIdia, https://github.com/Camyll (cherry picked from commit 235b995) Co-authored-by: Andrey Talman <[email protected]>
1 parent 10b501f commit d6e8411

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

.ci/pytorch/windows/cuda128.bat

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -37,10 +37,10 @@ IF "%CUDA_PATH_V128%"=="" (
3737
)
3838

3939
IF "%BUILD_VISION%" == "" (
40-
set TORCH_CUDA_ARCH_LIST=6.1;7.0;7.5;8.0;8.6;9.0;10.0;12.0
40+
set TORCH_CUDA_ARCH_LIST=7.0;7.5;8.0;8.6;9.0;10.0;12.0
4141
set TORCH_NVCC_FLAGS=-Xfatbin -compress-all
4242
) ELSE (
43-
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
43+
set NVCC_FLAGS=-D__CUDA_NO_HALF_OPERATORS__ --expt-relaxed-constexpr -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

4646
set "CUDA_PATH=%CUDA_PATH_V128%"

0 commit comments

Comments
 (0)