Skip to content

Commit 957cc12

Browse files
authored
[Relay] Modify create_executor to pass params (#8418)
* Overload create_executor to accept params * [fix] Add stringdoc for new param in create_executor
1 parent 80f48c7 commit 957cc12

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

python/tvm/relay/build_module.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -511,7 +511,7 @@ def _graph_wrapper(*args, **kwargs):
511511
return _graph_wrapper
512512

513513

514-
def create_executor(kind="debug", mod=None, device=None, target="llvm"):
514+
def create_executor(kind="debug", mod=None, device=None, target="llvm", params=None):
515515
"""Factory function to create an executor.
516516
517517
Example
@@ -544,6 +544,10 @@ def create_executor(kind="debug", mod=None, device=None, target="llvm"):
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,6 +559,9 @@ def create_executor(kind="debug", mod=None, device=None, target="llvm"):
555559
else:
556560
device = _nd.device(str(target), 0)
557561

562+
if params is not None:
563+
mod = IRModule.from_expr(bind_params_by_name(mod["main"], params))
564+
558565
if isinstance(target, str):
559566
target = Target(target)
560567
if kind == "debug":

0 commit comments

Comments
 (0)