This is an automated email from the ASF dual-hosted git repository.
tqchen pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-tvm.git
The following commit(s) were added to refs/heads/master by this push:
new b46c254 [TEST] test_cuddn flaky (#4846)
b46c254 is described below
commit b46c2548f9a6b8be305d41cc28381e1e218e22d6
Author: Tianqi Chen <[email protected]>
AuthorDate: Sat Feb 8 14:05:21 2020 -0800
[TEST] test_cuddn flaky (#4846)
---
tests/python/contrib/test_cudnn.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/tests/python/contrib/test_cudnn.py
b/tests/python/contrib/test_cudnn.py
index a28f433..1a22f90 100644
--- a/tests/python/contrib/test_cudnn.py
+++ b/tests/python/contrib/test_cudnn.py
@@ -78,7 +78,7 @@ def verify_conv2d(data_dtype, conv_dtype, tensor_format=0):
c_np = topi.testing.conv2d_nhwc_python(x_np, wt, 1, 1)
f(x, w, y)
- tvm.testing.assert_allclose(y.asnumpy(), c_np, atol=1e-5, rtol=1e-3)
+ tvm.testing.assert_allclose(y.asnumpy(), c_np, atol=3e-5, rtol=1e-3)
verify()
@@ -149,7 +149,7 @@ def verify_conv3d(data_dtype, conv_dtype, tensor_format=0):
raise AssertionError("For now, conv3d tensor format only support:
0(NCHW)")
f(x, w, y)
- tvm.testing.assert_allclose(y.asnumpy(), c_np, atol=1e-5, rtol=1e-4)
+ tvm.testing.assert_allclose(y.asnumpy(), c_np, atol=3e-5, rtol=1e-4)
verify()
