fix lint

apache · Mar 10, 2020 · 96e460f · 96e460f
1 parent 102ea86
commit 96e460f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/tvm/autotvm/task/relay_integration.py b/python/tvm/autotvm/task/relay_integration.py
@@ -46,7 +46,8 @@ def _lower(mod,
             with vta.build_config():
                 mod, _ = relay.optimize(mod, target, params)
                 grc = graph_runtime_codegen.GraphRuntimeCodegen(None, target)
-                return grc.codegen(mod["main"])
+                grc.codegen(mod["main"])
+                return
     # default case
     # Try graph codegen first to extract autotvm tasks.
     # If failed to compile, then fallback to use VM compiler.