This is an automated email from the ASF dual-hosted git repository. liuyizhi pushed a commit to branch v0.6 in repository https://gitbox.apache.org/repos/asf/incubator-tvm.git
The following commit(s) were added to refs/heads/v0.6 by this push: new c03d97c [Fix] Fix conv2d alter op for arm cpu (#5906) c03d97c is described below commit c03d97c7871b6b67a07bec2adb64ae2cf68d2b6e Author: Haichen Shen <shenhaic...@gmail.com> AuthorDate: Tue Jun 23 18:39:59 2020 -0700 [Fix] Fix conv2d alter op for arm cpu (#5906) --- topi/python/topi/arm_cpu/conv2d.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/topi/python/topi/arm_cpu/conv2d.py b/topi/python/topi/arm_cpu/conv2d.py index cbb6085..587a2f8 100644 --- a/topi/python/topi/arm_cpu/conv2d.py +++ b/topi/python/topi/arm_cpu/conv2d.py @@ -615,6 +615,7 @@ def _alter_conv2d_layout_arm(attrs, inputs, tinfos, F): copy_inputs[1] = weight new_attrs['tile_size'] = tile_size new_attrs[data_layout_key] = 'NCHW' + new_attrs['channels'] = CO # Store the same config for the altered operator (workload) new_data = tvm.placeholder((N, CI, H, W), dtype=data.dtype) @@ -645,6 +646,7 @@ def _alter_conv2d_layout_arm(attrs, inputs, tinfos, F): new_kernel = tvm.placeholder((CO, CI, 8, 8), "float32") bias = tvm.placeholder((CO, ), "float32") new_attrs[data_layout_key] = 'NCHW' + new_attrs['channels'] = CO new_workload = autotvm.task.args_to_workload( [new_data, new_kernel, bias, strides, padding, dilation, new_attrs[data_layout_key], out_dtype]