Skip to content

Commit 2834c46

Browse files
committed
fix lint
1 parent 6b61a67 commit 2834c46

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

  • tools/pkg-helpers/pytorch_pkg_helpers

tools/pkg-helpers/pytorch_pkg_helpers/cuda.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,9 @@
99

1010
def get_cuda_arch_list(sanitized_version: str, package_type: str = "") -> str:
1111
# 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
12+
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+
)
1315
# removing sm_50-sm_60 as these architectures are deprecated in CUDA 12.8/9 and will be removed in future releases
1416
# however we would like to keep sm_70 architecture see: https://github.com/pytorch/pytorch/issues/157517
1517
if sanitized_version == "12.8":

0 commit comments

Comments
 (0)