From 0fa96e9105fc9037ad22d3fee8e09a8cfeb4b76e Mon Sep 17 00:00:00 2001 From: JingZhuangzhuang <75348594+JZZ-NOTE@users.noreply.github.com> Date: Tue, 23 Nov 2021 10:01:48 +0800 Subject: [PATCH] fix memeory_optimize_pass bug (#37324) (#37413) --- paddle/fluid/inference/analysis/passes/memory_optimize_pass.cc | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/paddle/fluid/inference/analysis/passes/memory_optimize_pass.cc b/paddle/fluid/inference/analysis/passes/memory_optimize_pass.cc index 357c2bc4231..2202b94bee7 100644 --- a/paddle/fluid/inference/analysis/passes/memory_optimize_pass.cc +++ b/paddle/fluid/inference/analysis/passes/memory_optimize_pass.cc @@ -105,7 +105,8 @@ void MemoryOptimizePass::CollectVarMemorySize( "sequence_pool", "recurrent", "lod_reset", - "fetch"}; + "fetch", + "share_data"}; for (auto* tmp : node->inputs) { CHECK(tmp->IsOp()); std::string op_type = tmp->Op()->Type(); -- GitLab