fix memeory_optimize_pass bug (#37324) (#37413)

0fa96e91 · JingZhuangzhuang · GitHub · 6b3ffe99 · 0fa96e91
显示空白变更内容
内联并排

Showing with 2 addition and 1 deletion

paddle/fluid/inference/analysis/passes/memory_optimize_pass.cc ...e/fluid/inference/analysis/passes/memory_optimize_pass.cc +2 -1

未找到文件。
--- 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();