[tvm] branch main updated: Fix the shape check for vta dense strategy (#6983)

[tq...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

tqchen pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git


The following commit(s) were added to refs/heads/main by this push:
     new f65d0de  Fix the shape check for vta dense strategy (#6983)
f65d0de is described below

commit f65d0de11fe3deac78f58a81657993cf8ef0c6f3
Author: isong <in...@gmail.com>
AuthorDate: Sun Nov 29 07:06:46 2020 -0800

    Fix the shape check for vta dense strategy (#6983)
---
 vta/python/vta/top/op.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/vta/python/vta/top/op.py b/vta/python/vta/top/op.py
index f3b808a..a217104 100644
--- a/vta/python/vta/top/op.py
+++ b/vta/python/vta/top/op.py
@@ -123,7 +123,7 @@ def conv2d_transpose_strategy_vta(attrs, inputs, out_type, target):
 @_strategy.dense_strategy.register("vta")
 def dense_strategy_vta(attrs, inputs, out_type, target):
     """dense vta strategy"""
-    if inputs[0].shape == 4:  # this implies the layout is packed
+    if len(inputs[0].shape) == 4:  # this implies the layout is packed
         strategy = OpStrategy()
         strategy.add_implementation(
             _strategy.wrap_compute_dense(dense_packed),