[incubator-tvm] branch v0.6 updated: [Fix] Fix conv2d alter op for arm cpu (#5906)

[li...@apache.org]

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 <sh...@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]