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

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


The following commit(s) were added to refs/heads/unity by this push:
     new 43e0f66234 [Unity][Op][Docs] Update comment for `call_tir_dyn` (#14441)
43e0f66234 is described below

commit 43e0f6623402a2d90ea3f803868c69befb71eea8
Author: Steven S. Lyubomirsky <[email protected]>
AuthorDate: Fri Mar 31 09:30:28 2023 -0400

    [Unity][Op][Docs] Update comment for `call_tir_dyn` (#14441)
    
    Update comment for call_tir_dyn
---
 python/tvm/relax/op/vm/vm.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/python/tvm/relax/op/vm/vm.py b/python/tvm/relax/op/vm/vm.py
index 89d31b6581..a20407a4c9 100644
--- a/python/tvm/relax/op/vm/vm.py
+++ b/python/tvm/relax/op/vm/vm.py
@@ -88,12 +88,13 @@ def alloc_tensor(
 
 @args_converter.auto
 def call_tir_dyn(func: Expr, args: Tuple) -> Call:
-    """Construct a Call to kill a storage.
+    """Construct a Call to call_tir_dyn (invoke the given TIR PrimFunc)
+    consisting of the input tensors and the shape of the result.
 
     Parameters
     ----------
     func : Expr
-        The storage to be killed.
+        An expression evaluating to a TIR PrimFunc.
 
     args : Tuple
         The input args, includes a list of tensors, and a ShapeExpr.

Reply via email to