diff --git a/paddle/fluid/inference/analysis/passes/memory_optimize_pass.cc b/paddle/fluid/inference/analysis/passes/memory_optimize_pass.cc index 715316387289ccbba788aa000e175856010c4451..357c2bc4231fd0909dbec4c0600dadc7facda8c4 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();