From 2b3da4f663a895f220312a26801413808c4a192b Mon Sep 17 00:00:00 2001
From: Steven Lyubomirsky <slyubomirsky@octoml.ai>
Date: Thu, 30 Mar 2023 23:12:24 -0400
Subject: [PATCH] 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 89d31b6581ee..a20407a4c94e 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.