From edcd77dbf9adced39de9f6fb09dc42e892ec8553 Mon Sep 17 00:00:00 2001
From: Zheng Jiang <jiangzbox@gmail.com>
Date: Sat, 25 Jul 2020 00:09:50 +0800
Subject: [PATCH] [Relay]Port eliminate_common_subexpr to non-recursive form
 (#6134)

Co-authored-by: Zheng Jiang <zhejiang@amazon.com>
---
 src/relay/transforms/eliminate_common_subexpr.cc | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/src/relay/transforms/eliminate_common_subexpr.cc b/src/relay/transforms/eliminate_common_subexpr.cc
index dc3f77e..92cc64d 100644
--- a/src/relay/transforms/eliminate_common_subexpr.cc
+++ b/src/relay/transforms/eliminate_common_subexpr.cc
@@ -37,13 +37,13 @@
 namespace tvm {
 namespace relay {
 
-class CommonSubexprEliminator : public ExprMutator {
+class CommonSubexprEliminator : public MixedModeMutator {
  public:
   explicit CommonSubexprEliminator(runtime::TypedPackedFunc<bool(Expr)> fskip) : fskip_(fskip) {}
 
-  Expr VisitExpr_(const CallNode* call) final {
+  Expr Rewrite_(const CallNode* call, const Expr& post) final {
     static auto op_stateful = Op::GetAttrMap<TOpIsStateful>("TOpIsStateful");
-    Expr new_expr = ExprMutator::VisitExpr_(call);
+    Expr new_expr = post;
     const CallNode* new_call = new_expr.as<CallNode>();
     CHECK(new_call);
     const OpNode* op = new_call->op.as<OpNode>();
@@ -80,8 +80,8 @@ class CommonSubexprEliminator : public ExprMutator {
     return new_expr;
   }
 
-  Expr VisitExpr_(const TupleGetItemNode* op) final {
-    Expr new_expr = ExprMutator::VisitExpr_(op);
+  Expr Rewrite_(const TupleGetItemNode* op, const Expr& post) final {
+    Expr new_expr = post;
     const TupleGetItemNode* new_tuple_item = new_expr.as<TupleGetItemNode>();
     CHECK(new_tuple_item);
 
-- 
2.7.4