Skip to content

Commit f62ee55

Browse files
[cherry-pick]fix identity_scale_pass,test=develop (#5708)
1 parent 9709e4f commit f62ee55

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

lite/core/mir/elimination/identity_scale_eliminate_pass.cc

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,7 @@ class Eliminator : public FuseBase {
2828
// the previous op's output need updat
2929
auto* pre_op = OpNode("preop")
3030
->assert_is_not_op_type("conditional_block")
31+
->assert_is_not_op_type("while")
3132
->assert_is_not_op_type("scale");
3233
// TODO(Superjomn) check has only one output
3334
auto* x = VarNode("x")->assert_is_op_input("scale", "X");

0 commit comments

Comments
 (0)