There was an error while loading. Please reload this page.
1 parent 6b61a67 commit 2834c46Copy full SHA for 2834c46
1 file changed
tools/pkg-helpers/pytorch_pkg_helpers/cuda.py
@@ -9,7 +9,9 @@
9
10
def get_cuda_arch_list(sanitized_version: str, package_type: str = "") -> str:
11
# Fallback for other versions - maintain backward compatibility
12
- base_arch_list = "5.0;6.0;7.0;7.5;8.0;8.6;9.0" # default arch list for the stable versions
+ base_arch_list = (
13
+ "5.0;6.0;7.0;7.5;8.0;8.6;9.0" # default arch list for the stable versions
14
+ )
15
# removing sm_50-sm_60 as these architectures are deprecated in CUDA 12.8/9 and will be removed in future releases
16
# however we would like to keep sm_70 architecture see: https://github.com/pytorch/pytorch/issues/157517
17
if sanitized_version == "12.8":
0 commit comments