fix bug of constant folding pass (#55556)

bc61c796 · ming1753 · GitHub · 806f8d2b · bc61c796
隐藏空白更改
内联并排

Showing with 2 addition and 0 deletion

paddle/fluid/framework/ir/constant_folding_pass.cc paddle/fluid/framework/ir/constant_folding_pass.cc +2 -0

未找到文件。
--- 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) {