projects
/
platform
/
upstream
/
tvm.git
/ commitdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
| commitdiff |
tree
raw
|
patch
| inline |
side by side
(parent:
24597f0
)
[TEST] Fix flaky topi/tests/python/test_topi_pooling.py:test_adaptive_pool (#5736)
author
Tianqi Chen
<tqchen@users.noreply.github.com>
Fri, 5 Jun 2020 19:13:17 +0000
(12:13 -0700)
committer
GitHub
<noreply@github.com>
Fri, 5 Jun 2020 19:13:17 +0000
(12:13 -0700)
topi/tests/python/test_topi_pooling.py
patch
|
blob
|
history
diff --git
a/topi/tests/python/test_topi_pooling.py
b/topi/tests/python/test_topi_pooling.py
index
9f71a31
..
048de81
100644
(file)
--- 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)