From 592711ea1ff55acff85edea9bb9e89556600e21d Mon Sep 17 00:00:00 2001 From: Tianqi Chen Date: Fri, 5 Jun 2020 12:13:17 -0700 Subject: [PATCH] [TEST] Fix flaky topi/tests/python/test_topi_pooling.py:test_adaptive_pool (#5736) --- topi/tests/python/test_topi_pooling.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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) -- 2.7.4