Lunderberg opened a new pull request #8433: URL: https://github.com/apache/tvm/pull/8433
Along similar lines to #8336, parametrizing and cleaning up unit tests for conv2d and depthwise_conv2d unit tests. [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. [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). -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected]
