fix memeory_optimize_pass bug

PaddlePaddle · Nov 18, 2021 · e1269cc · e1269cc · paddle-bot-old · Nov 22, 2021
1 parent 34a44d5
commit e1269cc
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();