fix bug of constant folding pass (PaddlePaddle#55556)

cqulilujia · Jul 24, 2023 · a1ab8ad · a1ab8ad
1 parent 21f161a
commit a1ab8ad
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/paddle/fluid/framework/ir/constant_folding_pass.cc b/paddle/fluid/framework/ir/constant_folding_pass.cc
@@ -83,6 +83,8 @@ void ConstantFoldingPass::ApplyImpl(ir::Graph *graph) const {
       map[in_node->Name()] = 0;
       if (!in_node->Var()->Persistable()) {
         input_persis = false;
+      } else if (!in_node->inputs.empty()) {
+        input_persis = false;
       }
     }
     for (auto out_node : op_node->outputs) {