From 106991d25a35338c28b0766f9ad360d1d226b9cc Mon Sep 17 00:00:00 2001
From: Junru Shao <junrushao1994@gmail.com>
Date: Mon, 1 Oct 2018 12:21:39 -0400
Subject: [PATCH] Add atol=1e-5 to test_topi_matmul.test_matmul (#1791)
---
topi/tests/python/test_topi_matmul.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/topi/tests/python/test_topi_matmul.py b/topi/tests/python/test_topi_matmul.py
index a2902e17d..407a2859b 100644
--- a/topi/tests/python/test_topi_matmul.py
+++ b/topi/tests/python/test_topi_matmul.py
@@ -27,7 +27,7 @@ def verify_matmul(sa, sb, transp_a, transp_b):
c1 = np.matmul(np.transpose(a) if transp_a else a,
np.transpose(b) if transp_b else b)
c2 = with_tvm(lambda A,B: topi.matmul(A,B,transp_a,transp_b), a,b)
- np.testing.assert_allclose(c1, c2, rtol=1e-5)
+ np.testing.assert_allclose(c1, c2, rtol=1e-5, atol=1e-5)
def test_matmul():
verify_matmul((1,1),(1,1),False,False)
--
GitLab