commit 00d71b3e865a6a2d27c9d7a273764390548cf0b8
author Joel Schlosser <jbschlosser@meta.com> Wed Jul 10 12:43:09 2024 -0400
committer PyTorch MergeBot <pytorchmergebot@users.noreply.github.com> Mon Jul 15 17:35:34 2024 +0000
tree 155515a0047f6b143c1ac62ca2a51557ef23dfe6
parent 9e161af179a781acfb839942aacb12c1f09395af

Tweak tolerances for test_vjp_linalg_tensorsolve_cuda_float32 to pass in Windows / debug builds (#130449)

Pull Request resolved: https://github.com/pytorch/pytorch/pull/130449
Approved by: https://github.com/zou3519, https://github.com/malfet
ghstack dependencies: #128238, #130360

test/functorch/test_ops.py

diff --git a/test/functorch/test_ops.py b/test/functorch/test_ops.py
index c876841..f047250 100644
--- a/test/functorch/test_ops.py
+++ b/test/functorch/test_ops.py
@@ -790,7 +790,7 @@
             tol2(
                 "linalg.pinv", "hermitian", {torch.float32: tol(atol=1e-05, rtol=1e-05)}
             ),
-            tol1("linalg.tensorsolve", {torch.float32: tol(atol=4e-05, rtol=5e-05)}),
+            tol1("linalg.tensorsolve", {torch.float32: tol(atol=9e-03, rtol=2e-04)}),
             tol1("linalg.multi_dot", {torch.float32: tol(atol=1e-04, rtol=1e-04)}),
             tol1("svd_lowrank", {torch.float32: tol(atol=1e-04, rtol=1e-04)}),
             tol1("pca_lowrank", {torch.float32: tol(atol=1e-04, rtol=1e-04)}),