Skip to content

Commit e8c908c

Browse files
authored
add sm_100 and sm_120 to cuda 12.8 domain builds (#6273)
current torchvision/audio builds missing sm_100 and sm_120 adding to cu128 builds `cuobjdump -ptx /usr/local/lib/python3.12/dist-packages/torchvision/_C.so | grep "arch = sm_" | sort --unique arch = sm_50 arch = sm_60 arch = sm_70 arch = sm_75 arch = sm_80 arch = sm_86 arch = sm_90` cc @atalman @ptrblck @nWEIdia
1 parent dbde372 commit e8c908c

File tree

1 file changed

+4
-1
lines changed
  • tools/pkg-helpers/pytorch_pkg_helpers

1 file changed

+4
-1
lines changed

tools/pkg-helpers/pytorch_pkg_helpers/cuda.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,8 +8,11 @@
88

99

1010
def get_cuda_arch_list(sanitized_version: str) -> str:
11+
base_arch_list = "5.0+PTX;6.0;7.0;7.5;8.0;8.6;9.0"
1112
if float(sanitized_version) >= 12.0:
12-
return "5.0+PTX;6.0;7.0;7.5;8.0;8.6;9.0"
13+
if sanitized_version == "12.8":
14+
return base_arch_list + ";10.0;12.0"
15+
return base_arch_list
1316
if float(sanitized_version) > 11.3:
1417
return "3.5;5.0+PTX;6.0;7.0;7.5;8.0;8.6"
1518
# mainly for cuda 10.2

0 commit comments

Comments
 (0)