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

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

Browse Source 
This reverts commit 6c5227ba00a2904365af566c24b4681cd01a041c.

Reverted https://github.com/pytorch/pytorch/pull/157385 on behalf of https://github.com/clee2000 due to broke some slow tests test_cpp_extensions_jit.py::TestCppExtensionJIT::test_jit_cuda_archflags [GH job link](https://github.com/pytorch/pytorch/actions/runs/16286465717/job/45986677885) [HUD commit link](6c5227ba00) ([comment](https://github.com/pytorch/pytorch/pull/157385#issuecomment-3074737541))
This commit is contained in:
PyTorch MergeBot
2025-07-15 18:06:52 +00:00
parent 243b12e565
commit b26da7741b
1 changed files with 4 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
test/test_cpp_extensions_jit.py
View File

@ -322,15 +322,12 @@ 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"])
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:
archflags["5.0;6.0+PTX;7.0;7.5"] = (["50", "60", "70", "75"], ["60"])
if int(torch.version.cuda.split(".")[0]) < 12:
# CUDA 12 drops compute capability < 5.0
archflags["Pascal 3.5"] = (["35", "60", "61"], None)