commit c59c4b0d778b445cd016c77dc42ee024ef150bf7
author Your Name <you@example.com> Wed Oct 07 20:38:19 2020 -0700
committer Facebook GitHub Bot <facebook-github-bot@users.noreply.github.com> Wed Oct 07 20:42:06 2020 -0700
tree 33d228209833e32bb05430bb3934c7564164c8d8
parent 903acc6b83e058cb6ed1cb7faa5938b425e695fd

Fix cholesky TF32 tests (#45492)

Summary:
This test is changed one day before the landing of the tf32 tests PR, therefore the fix for this is not included in that PR.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/45492

Reviewed By: ezyang

Differential Revision: D24101876

Pulled By: ngimel

fbshipit-source-id: cb3615b2fb8acf17abe54cd18b1faec26582d6b6

test/test_torch.py

diff --git a/test/test_torch.py b/test/test_torch.py
index fd89dff..312943d 100644
--- a/test/test_torch.py
+++ b/test/test_torch.py
@@ -7794,6 +7794,7 @@
     @skipCUDAIfNoMagma
     @skipCPUIfNoLapack
     @dtypes(torch.float32, torch.float64, torch.complex64, torch.complex128)
+    @tf32_on_and_off(0.01)
     def test_cholesky(self, device, dtype):
         from torch.testing._internal.common_utils import \
             (random_symmetric_pd_matrix,