[TEST] Fix flaky topi/tests/python/test_topi_pooling.py:test_adaptive…

…_pool (apache#5736)
trevor-m · Jun 9, 2020 · 0078773 · 0078773
1 parent 6996fb6
commit 0078773
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/topi/tests/python/test_topi_pooling.py b/topi/tests/python/test_topi_pooling.py
@@ -284,7 +284,7 @@ def check_device(device):
         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)