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

Remove C++ and test branches for CUDA<12 (#163443)

Browse Source 
Remove conditional branches for CUDA<12.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/163443
Approved by: https://github.com/eqy
This commit is contained in:
Yuanyuan Chen
2025-09-22 18:20:04 +00:00
committed by PyTorch MergeBot
parent d279a6a6f1
commit bec967eaa4
6 changed files with 3 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/test_sparse.py
View File

@ -64,7 +64,7 @@ load_tests = load_tests
gradcheck = functools.partial(gradcheck, check_batched_grad=False)
CUSPARSE_SPMM_COMPLEX128_SUPPORTED = (
IS_WINDOWS and torch.version.cuda and version.parse(torch.version.cuda) > version.parse("11.2")
IS_WINDOWS and torch.version.cuda
) or (not IS_WINDOWS and not TEST_WITH_ROCM)
HIPSPARSE_SPMM_COMPLEX128_SUPPORTED = torch.version.hip and version.parse(torch.version.hip.split("-")[0]) >= version.parse("6.0")