[tvm] branch main updated: relax tolerance for dlpack test (#7325)

[ma...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

masahi pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git


The following commit(s) were added to refs/heads/main by this push:
     new 790344c  relax tolerance for dlpack test (#7325)
790344c is described below

commit 790344c6ef035947caaaf1cd812ade8d862802aa
Author: masahi <ma...@gmail.com>
AuthorDate: Fri Jan 22 11:28:16 2021 +0900

    relax tolerance for dlpack test (#7325)
---
 tests/python/contrib/test_dlpack.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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 @@ def test():
         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