[Parser] Typo in mod creation (apache#6165)

trevor-m · Aug 26, 2020 · 2038fd6 · 2038fd6
1 parent ff856d4
commit 2038fd6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/tvm/relay/frontend/common.py b/python/tvm/relay/frontend/common.py
@@ -457,7 +457,7 @@ def get_name(node):
 def infer_type(node, mod=None):
     """A method to infer the type of an intermediate node in the relay graph."""
     if isinstance(mod, IRModule):
-        mod["main"] = _function.Function([], node)
+        mod["main"] = _function.Function(tvm.relay.analysis.free_vars(node), node)
         mod = _transform.InferType()(mod)
         entry = mod["main"]
         ret = entry.body