Родитель
9a542e37de
Коммит
026162ad62
|
@ -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:
|
||||
|
|
Загрузка…
Ссылка в новой задаче