Add simplify pass in the Relay interpreter (#2417)

312802f3 · Haichen Shen · Tianqi Chen · 547a0913 · 312802f3
Commit 312802f3 authored Jan 11, 2019 by Haichen Shen Committed by Tianqi Chen Jan 11, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletions

python/tvm/relay/backend/interpreter.py
+3 -1

No files found.
--- a/python/tvm/relay/backend/interpreter.py
+++ b/python/tvm/relay/backend/interpreter.py
@@ -218,7 +218,9 @@ class Interpreter(Executor):
        """
        # TODO: We need to move this optimization code into the optimizer/pass manager
        ck_expr = ir_pass.infer_type(expr, mod=self.mod)
-        fused_expr = ir_pass.fuse_ops(ck_expr)
+        simp_expr = ir_pass.simplify_inference(ck_expr)
+        ck_simp = ir_pass.infer_type(simp_expr, mod=self.mod)
+        fused_expr = ir_pass.fuse_ops(ck_simp)
        ck_fused = ir_pass.infer_type(fused_expr, mod=self.mod)
        return ck_fused