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

[BE] Update cudnn to 9.10.1.4 (#155122)

Browse Source 
Follow up to #152782
Pull Request resolved: https://github.com/pytorch/pytorch/pull/155122
Approved by: https://github.com/malfet, https://github.com/atalman, https://github.com/eqy
This commit is contained in:
Aaron Gokaslan
2025-06-10 16:59:00 +00:00
committed by PyTorch MergeBot
parent 38c4d05535
commit 73220d52fd
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/test_nn.py
View File

@ -11662,6 +11662,8 @@ class TestNNDeviceType(NNTestCase):
prec = dtype2prec_DONTUSE[dtype]
if dtype == torch.float16:
prec = 4e-2
elif dtype == torch.float32:
prec = 2e-5
self.assertEqual(p1.grad, p2.grad, atol=prec, rtol=0)
tests = [