OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

[cuSPARSE][B200] Bump tolerances for test_sparse_csr matvec (#148721)

Browse Source 
Small tolerance bump for blackwell (appears to use same kernel as prev. arches)

Pull Request resolved: https://github.com/pytorch/pytorch/pull/148721
Approved by: https://github.com/nWEIdia, https://github.com/ngimel
This commit is contained in:
eqy 2025-04-16 18:44:18 +00:00 committed by PyTorch MergeBot
parent 1f29190b59
commit 17bf59340c
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
test/test_sparse_csr.py
View File

@ -1495,7 +1495,8 @@ class TestSparseCSR(TestCase):
res = csr.matmul(vec) res = csr.matmul(vec)
expected = csr.to_dense().matmul(vec) expected = csr.to_dense().matmul(vec)
self.assertEqual(res, expected) atol, rtol = (2e-3, 1e-3) if dtype == torch.half else (None, None)
self.assertEqual(res, expected, atol=atol, rtol=rtol)
bad_vec = torch.randn(side + 10, dtype=dtype, device=device) bad_vec = torch.randn(side + 10, dtype=dtype, device=device)
err_msg = "size mismatch, got" err_msg = "size mismatch, got"