Skip to content

Commit 912d6a2

Browse files
eqypytorchmergebot
authored andcommitted
[CUDA] Bump tolerances in test_svd_lowrank_cuda_float64 (pytorch#143049)
pre-emptive bump for apparent noisy failure Pull Request resolved: pytorch#143049 Approved by: https://github.com/Skylion007, https://github.com/lezcano, https://github.com/nikitaved
1 parent 8960cb5 commit 912d6a2

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/test_linalg.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2506,7 +2506,7 @@ def run_subtest(actual_rank, matrix_size, batches, device, svd_lowrank, **option
25062506
# check if svd_lowrank produces same singular values as linalg.svdvals
25072507
U, S, Vh = torch.linalg.svd(a, full_matrices=False)
25082508
V = Vh.mH
2509-
self.assertEqual(s, S)
2509+
self.assertEqual(s, S, rtol=5e-7, atol=1e-7)
25102510

25112511
if density == 1:
25122512
# actual_rank is known only for dense inputs

0 commit comments

Comments
 (0)