[tvm] branch jcf94-patch-1 created (now 2b81abc)

[jc...@apache.org]

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

jcf94 pushed a change to branch jcf94-patch-1
in repository https://gitbox.apache.org/repos/asf/tvm.git.


      at 2b81abc  [Dyn] Use SizeVar instead of Var in the GetShape function

This branch includes the following new commits:

     new 2b81abc  [Dyn] Use SizeVar instead of Var in the GetShape function

The 1 revisions listed above as "new" are entirely new to this
repository and will be described in separate emails.  The revisions
listed as "add" were already present in the repository and have only
been added to this reference.

[tvm] 01/01: [Dyn] Use SizeVar instead of Var in the GetShape function

[jc...@apache.org]

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

jcf94 pushed a commit to branch jcf94-patch-1
in repository https://gitbox.apache.org/repos/asf/tvm.git

commit 2b81abcb0b23356cbaa845d78cb3854babd82cb0
Author: Chenfan <jc...@outlook.com>
AuthorDate: Sun Dec 5 18:07:15 2021 +0800

    [Dyn] Use SizeVar instead of Var in the GetShape function
---
 python/tvm/relay/backend/te_compiler.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/tvm/relay/backend/te_compiler.py b/python/tvm/relay/backend/te_compiler.py
index db75049..def827c 100644
--- a/python/tvm/relay/backend/te_compiler.py
+++ b/python/tvm/relay/backend/te_compiler.py
@@ -268,7 +268,7 @@ def get_shape(shape):
                 assert val <= np.iinfo(np.int32).max
                 ret.append(tvm.tir.IntImm("int32", val))
         elif isinstance(dim, tvm.tir.Any):
-            ret.append(te.var("any_dim", "int32"))
+            ret.append(te.size_var("any_dim", "int32"))
         else:
             ret.append(dim)
     return ret