fix extract entry_func

apache · Jun 12, 2019 · ec8e91a · ec8e91a
1 parent 3c54299
commit ec8e91a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/tvm/relay/backend/interpreter.py b/python/tvm/relay/backend/interpreter.py
@@ -309,6 +309,6 @@ def _interp_wrapper(*args, **kwargs):
                     self.mod = module.Module.from_expr(func)
 
             mod = self.optimize()
-            opt_expr = Call(mod[mod.entry_func], relay_args)
+            opt_expr = Call(mod[self.mod.entry_func.name_hint], relay_args)
             return self._intrp(opt_expr)
         return _interp_wrapper