From c3cd8502bcd50a40834622b21ec3d1fb9549798e Mon Sep 17 00:00:00 2001 From: Zhang Jun Date: Tue, 31 Jan 2023 17:03:16 +0800 Subject: [PATCH] [pass] Upgrade Constant Folding Pass (#49908) --- paddle/fluid/framework/ir/constant_folding_pass.cc | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/paddle/fluid/framework/ir/constant_folding_pass.cc b/paddle/fluid/framework/ir/constant_folding_pass.cc index cd069e474e..0bcd7a733d 100644 --- a/paddle/fluid/framework/ir/constant_folding_pass.cc +++ b/paddle/fluid/framework/ir/constant_folding_pass.cc @@ -143,6 +143,10 @@ void ConstantFoldingPass::ApplyImpl(ir::Graph *graph) const { } out_desc->SetShape(out_shape); out_desc->SetPersistable(true); + auto *var_desc_out = op_node->Op()->Block()->Var(out_name); + var_desc_out->SetShape(out_shape); + var_desc_out->SetPersistable(true); + var_desc_out->Flush(); auto *global_out_tensor = scope->Var(out_name)->GetMutable(); *global_out_tensor = *local_out_tensor; -- GitLab