fix test_cholesky TF32 test

pytorch · Nov 20, 2020 · c70c837 · c70c837
1 parent c0723a0
commit c70c837
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/test/test_linalg.py b/test/test_linalg.py
@@ -62,6 +62,7 @@ def run_test_case(a, b):
     @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_hermitian_pd_matrix