frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix dumb typo (#142079)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/142079
Approved by: https://github.com/jainapurva, https://github.com/soulitzer
This commit is contained in:
drisspg
2024-12-04 17:15:34 -05:00
committed by PyTorch MergeBot
parent 60a192036b
commit 3fdc74ae29
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
torch/utils/cpp_extension.py
View File

@ -1995,7 +1995,7 @@ def _get_cuda_arch_flags(cflags: Optional[List[str]] = None) -> List[str]:
])
supported_arches = ['3.5', '3.7', '5.0', '5.2', '5.3', '6.0', '6.1', '6.2',
'7.0', '7.2', '7.5', '8.0', '8.6', '8.7', '8.9', '9.0', '9.0a'
'7.0', '7.2', '7.5', '8.0', '8.6', '8.7', '8.9', '9.0', '9.0a',
'10.0']
valid_arch_strings = supported_arches + [s + "+PTX" for s in supported_arches]