frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-21 05:34:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

Increase tolerance for test_corrcoef_cuda_int32 (#157206)

Browse Source 
Fixes #156988
Pull Request resolved: https://github.com/pytorch/pytorch/pull/157206
Approved by: https://github.com/Skylion007
This commit is contained in:
Jason Ansel
2025-06-28 13:29:02 -07:00
committed by PyTorch MergeBot
parent e959dd017d
commit b147b6c0e3
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/test_torch.py
View File

@ -2324,7 +2324,7 @@ else:
for x in self._generate_correlation_tensors(device, dtype):
res = torch.corrcoef(x)
ref = np.corrcoef(x.cpu().numpy())
self.assertEqual(res, ref, exact_dtype=False)
self.assertEqual(res, ref, atol=1e-04, rtol=1e-03, exact_dtype=False)
@skipRocmIfTorchInductor
@dtypes(torch.int, torch.float, torch.cfloat)