* [UnitTests][Topi] Updated test_topi_conv2d_nchw.py to have parametrized tests.
- Better error messages, displays which workloads/targets failed and why.
- Fixed bug in topi.nn.conv2d._get_workload exposed by the
parametrized tests. Incorrect padding if the "SAME" parameter is
used with dilation>1.
- Fixed bug in tvm.topi.x86.group_conv2d._get_default_config, missing
dilation parameter in call to _get_conv2d_workload.
* [UnitTests][Topi] Parametrized the tests in test_topi_depthwise_conv2d.py
In preparation for parametrizing to test on float16 as well.
- Single test_conv2d test with parameters for layout/input sizes.
- Extended the support for NCHWc layouts, so that they could be
included in the parametrization. (Implemented
topi.testing.depthwise_conv2d_python_nchwc and
topi.nn.scale_shift_nchwc, added layout argument to
topi.nn.depthwise_conv2d._get_workload).
Co-authored-by: Eric Lunderberg <elunderberg@octoml.ai>
Lunderberg
45497bd362