From f62ee55b023828224580407110c8ea0cf8ce0e17 Mon Sep 17 00:00:00 2001
From: Leonardo-Ding <48182083+Leonardo-Ding@users.noreply.github.com>
Date: Mon, 15 Mar 2021 14:27:56 +0800
Subject: [PATCH] [cherry-pick]fix identity_scale_pass,test=develop (#5708)

---
 lite/core/mir/elimination/identity_scale_eliminate_pass.cc | 1 +
 1 file changed, 1 insertion(+)

diff --git a/lite/core/mir/elimination/identity_scale_eliminate_pass.cc b/lite/core/mir/elimination/identity_scale_eliminate_pass.cc
index 2e522214bfa..0c96858d56e 100644
--- 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");