relax tolerance for dlpack test (apache#7325)

Lokiiiiii · Mar 1, 2021 · 237a9b8 · 237a9b8
1 parent f680642
commit 237a9b8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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