Add relative threshold while testing inversing (#8973)

Oneflow-Inc · Aug 20, 2022 · 0d91cf6 · 0d91cf6
1 parent 8f5745b
commit 0d91cf6
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/python/oneflow/test/modules/test_inv.py b/python/oneflow/test/modules/test_inv.py
@@ -23,19 +23,19 @@
 
 @flow.unittest.skip_unless_1n1d()
 class TestLinalgInv(flow.unittest.TestCase):
-    @autotest(n=5)
+    @autotest(n=5, rtol=1e-2)
     def test_inv_3by3_with_random_data(test_case):
         device = random_device()
         x = random_tensor(ndim=2, dim0=3, dim1=3, low=-1).to(device)
         return torch.linalg.inv(x)
 
-    @autotest(n=5)
+    @autotest(n=5, rtol=1e-2)
     def test_inv_batch_3by3_with_random_data(test_case):
         device = random_device()
         x = random_tensor(ndim=3, dim0=random(), dim1=3, dim2=3, low=-1).to(device)
         return torch.linalg.inv(x)
 
-    @autotest(n=5, rtol=1e-3, atol=1e-3)
+    @autotest(n=5, rtol=1e-2)
     def test_inv_random_square_with_random_data(test_case):
         device = random_device()
         square_dim = random()