From ae0a08923192a88fa8b7cf4b7dff92a581ff50f9 Mon Sep 17 00:00:00 2001 From: masahi Date: Fri, 22 Jan 2021 11:28:16 +0900 Subject: [PATCH] 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 661e284c299f..6ff2529f7570 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