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

[CI] Fixes CI for CUDA Version > 12.9 (#157385)

Browse Source 
Compute capabilities older than volta (inclusive) is no longer supported in CUDA Version > 12.9
Pull Request resolved: https://github.com/pytorch/pytorch/pull/157385
Approved by: https://github.com/huydhn
This commit is contained in:
AaronWang04
2025-07-15 07:04:54 +00:00
committed by PyTorch MergeBot
parent c8c221c0b3
commit 6c5227ba00
1 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
test/test_cpp_extensions_jit.py
View File

@ -322,12 +322,15 @@ class TestCppExtensionJIT(common.TestCase):
[f"{capability[0]}{capability[1]}" for capability in capabilities],
None,
),
"Maxwell+Tegra;6.1": (["53", "61"], None),
"Volta": (["70"], ["70"]),
}
archflags["7.5+PTX"] = (["75"], ["75"])
archflags["5.0;6.0+PTX;7.0;7.5"] = (["50", "60", "70", "75"], ["60"])
if int(torch.version.cuda.split(".")[0]) < 12:
major, minor = map(int, torch.version.cuda.split(".")[:2])
if major < 12 or (major == 12 and minor <= 9):
# Compute capability <= 7.0 is only supported up to CUDA 12.9
archflags["Maxwell+Tegra;6.1"] = (["53", "61"], None)
archflags["Volta"] = ((["70"], ["70"]),)
archflags["5.0;6.0+PTX;7.0;7.5"] = (["50", "60", "70", "75"], ["60"])
if major < 12:
# CUDA 12 drops compute capability < 5.0
archflags["Pascal 3.5"] = (["35", "60", "61"], None)