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.