[TEST] test_cuddn flaky (apache#4846)

neo-ai · Mar 2, 2020 · 627dfc6 · 627dfc6
1 parent f54865f
commit 627dfc6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/python/contrib/test_cudnn.py b/tests/python/contrib/test_cudnn.py
@@ -78,7 +78,7 @@ def verify():
             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():
             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()