fix memeory_optimize_pass bug (#37324) (#37413)

PaddlePaddle · Nov 23, 2021 · 0fa96e9 · 0fa96e9
1 parent 6b3ffe9
commit 0fa96e9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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();