From 63c8c8c2ebf04de276de1ab6bca5a23e9872d6c0 Mon Sep 17 00:00:00 2001 From: JingZhuangzhuang <75348594+JZZ-NOTE@users.noreply.github.com> Date: Thu, 11 Nov 2021 23:16:51 -0600 Subject: [PATCH] add fetch to black list (#37123) --- 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 71531638728..357c2bc4231 100644 --- a/paddle/fluid/inference/analysis/passes/memory_optimize_pass.cc +++ b/paddle/fluid/inference/analysis/passes/memory_optimize_pass.cc @@ -104,7 +104,8 @@ void MemoryOptimizePass::CollectVarMemorySize( "equal", "sequence_pool", "recurrent", - "lod_reset"}; + "lod_reset", + "fetch"}; for (auto* tmp : node->inputs) { CHECK(tmp->IsOp()); std::string op_type = tmp->Op()->Type(); -- GitLab