Commit 026162ad by Balint Cristian Committed by Thierry Moreau

Hotfix for issue #3641. (#3644)

parent 9a542e37
......@@ -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:
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment