Skip to content

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torch/utils/cpp_extension.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1995,7 +1995,7 @@ def _get_cuda_arch_flags(cflags: Optional[List[str]] = None) -> List[str]:
19951995
])
19961996

19971997
supported_arches = ['3.5', '3.7', '5.0', '5.2', '5.3', '6.0', '6.1', '6.2',
1998-
'7.0', '7.2', '7.5', '8.0', '8.6', '8.7', '8.9', '9.0', '9.0a'
1998+
'7.0', '7.2', '7.5', '8.0', '8.6', '8.7', '8.9', '9.0', '9.0a',
19991999
'10.0']
20002000
valid_arch_strings = supported_arches + [s + "+PTX" for s in supported_arches]
20012001

0 commit comments

Comments
 (0)