fix memeory_optimize_pass bug (PaddlePaddle#37324)

JZZ-NOTE · Nov 22, 2021 · 670761a · 670761a
1 parent 9ffb43b
commit 670761a
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();