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

[CI][CUDA][cuSPARSELt] cusparselt 0.6.3 and cu121 related cleanups (#145793)

Browse Source 
Make ci cusparselt installation be consistent with nightly binary
Remove cu121 related docker build jobs and inductor runs Update test failures relating to cu121

Retry of https://github.com/pytorch/pytorch/pull/145696
Pull Request resolved: https://github.com/pytorch/pytorch/pull/145793
Approved by: https://github.com/eqy, https://github.com/tinglvv
This commit is contained in:
Wei Wang
2025-01-28 21:01:55 +00:00
committed by PyTorch MergeBot
parent ccc2878c97
commit 6bcb545d9c
11 changed files with 33 additions and 222 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
test/test_sparse_semi_structured.py
View File

@ -1209,12 +1209,12 @@ class TestSparseSemiStructuredCUSPARSELT(TestCase):
# CUDA 11.8 has cuSPARSELt v0.4.0 support
if version == (11, 8):
assert torch.backends.cusparselt.version() == 400
# CUDA 12.1 has cuSPARSELt v0.5.2 support
elif version == (12, 1):
assert torch.backends.cusparselt.version() == 502
# CUDA 12.4+ has cuSPARSELt v0.6.2 support
# PyTorch CUDA 12.4 using cuSPARSELt v0.6.2
elif version >= (12, 4):
assert torch.backends.cusparselt.version() == 602
# PyTorch CUDA 12.6+ using cuSPARSELt v0.6.3
elif version >= (12, 6):
assert torch.backends.cusparselt.version() == 603
else:
assert torch.backends.cusparselt.version() is None