From: Tianqi Chen <tqchen@users.noreply.github.com>
Date: Fri, 5 Jun 2020 19:13:17 +0000 (-0700)
Subject: [TEST] Fix flaky topi/tests/python/test_topi_pooling.py:test_adaptive_pool (#5736)
X-Git-Tag: upstream/0.7.0~604
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=592711ea1ff55acff85edea9bb9e89556600e21d;p=platform%2Fupstream%2Ftvm.git

[TEST] Fix flaky topi/tests/python/test_topi_pooling.py:test_adaptive_pool (#5736)
---

diff --git a/topi/tests/python/test_topi_pooling.py b/topi/tests/python/test_topi_pooling.py
index 9f71a31..048de81 100644
--- a/topi/tests/python/test_topi_pooling.py
+++ b/topi/tests/python/test_topi_pooling.py
@@ -284,7 +284,7 @@ def verify_adaptive_pool(dshape, out_size, pool_type, layout="NCHW", dtype="floa
         b = tvm.nd.array(np.zeros(get_const_tuple(oshape), dtype=out.dtype), ctx)
         f = tvm.build(s, [data, out], device)
         f(a, b)
-        tvm.testing.assert_allclose(b.asnumpy(), np_out, rtol=1e-5)
+        tvm.testing.assert_allclose(b.asnumpy(), np_out, rtol=4e-5, atol=1e-6)
 
     for device in get_all_backend():
         check_device(device)