You are viewing a plain text version of this content. The canonical link for it is here.

Posted to commits@tvm.apache.org by tq...@apache.org on 2025/04/22 14:43:28 UTC

(tvm) 05/06: fix python shape obj reg

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

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

commit 304dd0667af71c935225d4d8f091c4abdcf84b18
Author: tqchen <ti...@gmail.com>
AuthorDate: Tue Apr 22 10:23:12 2025 -0400

    fix python shape obj reg
---
 python/tvm/runtime/container.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/tvm/runtime/container.py b/python/tvm/runtime/container.py
index 5fe5a303fe..715cf1f385 100644
--- a/python/tvm/runtime/container.py
+++ b/python/tvm/runtime/container.py
@@ -87,7 +87,7 @@ class String(str, PyNativeObject):
         return val
 
 
-@tvm._ffi.register_object("runtime.ShapeTuple")
+@tvm._ffi.register_object("object.Shape")
 class ShapeTuple(Object):
     """TVM runtime ShapeTuple object.
     Parameters