Skip to content

Commit fd4effd

Browse files
committed
[fix] Add stringdoc for new param in create_executor
1 parent bf172d9 commit fd4effd

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

python/tvm/relay/build_module.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -544,6 +544,10 @@ def create_executor(kind="debug", mod=None, device=None, target="llvm", params=N
544544
target : :py:class:`tvm.Target`
545545
The corresponding context
546546
547+
params : dict of str to NDArray
548+
Input parameters to the graph that do not change
549+
during inference time.
550+
547551
Returns
548552
-------
549553
executor : :py:class:`~tvm.relay.backend.interpreter.Executor`
@@ -555,10 +559,9 @@ def create_executor(kind="debug", mod=None, device=None, target="llvm", params=N
555559
else:
556560
device = _nd.device(str(target), 0)
557561

558-
# print("params " + params)
559562
if params is not None:
560-
mod = IRModule.from_expr(bind_params_by_name(mod["main"], params))
561-
563+
mod = IRModule.from_expr(bind_params_by_name(mod["main"], params))
564+
562565
if isinstance(target, str):
563566
target = Target(target)
564567
if kind == "debug":

0 commit comments

Comments
 (0)