[cherry-pick]fix identity_scale_pass,test=develop (PaddlePaddle#5708)

BillDior · Mar 15, 2021 · f62ee55 · f62ee55
1 parent 9709e4f
commit f62ee55
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/lite/core/mir/elimination/identity_scale_eliminate_pass.cc b/lite/core/mir/elimination/identity_scale_eliminate_pass.cc
@@ -28,6 +28,7 @@ class Eliminator : public FuseBase {
     // the previous op's output need updat
     auto* pre_op = OpNode("preop")
                        ->assert_is_not_op_type("conditional_block")
+                       ->assert_is_not_op_type("while")
                        ->assert_is_not_op_type("scale");
     // TODO(Superjomn) check has only one output
     auto* x = VarNode("x")->assert_is_op_input("scale", "X");