From: Balint Cristian Date: Sun, 28 Jul 2019 08:05:37 +0000 (+0300) Subject: Hotfix for issue #3641. (#3644) X-Git-Tag: upstream/0.7.0~2118 X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=026162ad62ebdfddeb50d8d45278b3c9c781f01c;p=platform%2Fupstream%2Ftvm.git Hotfix for issue #3641. (#3644) --- diff --git a/topi/tests/python/test_topi_conv2d_winograd.py b/topi/tests/python/test_topi_conv2d_winograd.py index cf176a8..a42d61d 100644 --- a/topi/tests/python/test_topi_conv2d_winograd.py +++ b/topi/tests/python/test_topi_conv2d_winograd.py @@ -81,7 +81,12 @@ def verify_conv2d_nchw(batch, in_channel, in_size, num_filter, kernel, stride, p else: func = tvm.build(s, [A, W, C], device, name="relu_%d_%d_%d_%d_%d_%d_%d_%d" % (batch, in_channel, in_size, num_filter, kernel, stride, padding, dilation)) func(a, w, c) - tvm.testing.assert_allclose(c.asnumpy(), c_np, rtol=1e-5) + + rtol = 1e-5 + if (kernel > 3): + rtol = 2e-5 + + tvm.testing.assert_allclose(c.asnumpy(), c_np, rtol=rtol) for device in devices: