We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent dbde372 commit e8c908cCopy full SHA for e8c908c
tools/pkg-helpers/pytorch_pkg_helpers/cuda.py
@@ -8,8 +8,11 @@
8
9
10
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"
12
if float(sanitized_version) >= 12.0:
- 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
16
if float(sanitized_version) > 11.3:
17
return "3.5;5.0+PTX;6.0;7.0;7.5;8.0;8.6"
18
# mainly for cuda 10.2
0 commit comments