commit 790344c6ef035947caaaf1cd812ade8d862802aa
author masahi <masahi129@gmail.com> Fri Jan 22 11:28:16 2021 +0900
committer GitHub <noreply@github.com> Fri Jan 22 11:28:16 2021 +0900
tree 177b97a7fd2fad39f98ca0b9afd5b1346780eb7c
parent 17ae44dbb8ce65bd72af1e2400d5796ed14277e5

relax tolerance for dlpack test (#7325)

tests/python/contrib/test_dlpack.py

diff --git a/tests/python/contrib/test_dlpack.py b/tests/python/contrib/test_dlpack.py
index 661e284..6ff2529 100644
--- a/tests/python/contrib/test_dlpack.py
+++ b/tests/python/contrib/test_dlpack.py
@@ -54,7 +54,7 @@
         f_pytorch = to_pytorch_func(f)
         zz2 = torch.empty(137, 137)
         f_pytorch(xx, yy, zz2)
-        tvm.testing.assert_allclose(zz.numpy(), zz2.numpy(), rtol=1e-6)
+        tvm.testing.assert_allclose(zz.numpy(), zz2.numpy(), rtol=1e-4, atol=1e-4)
 
     except ImportError:
         pass