polish code for reading. test=develop

6 years ago · 591ad33e32
parent 18afb77e78
commit 591ad33e32
1 changed files with 2 additions and 1 deletions
--- a/paddle/fluid/framework/details/memory_optimize_pass.cc
+++ b/paddle/fluid/framework/details/memory_optimize_pass.cc
@ -46,6 +46,7 @@ namespace details {
 std::unique_ptr<ir::Graph> MemoryOptimizePass::ApplyImpl(
    std::unique_ptr<ir::Graph> graph) const {
  auto nodes = graph->Nodes();
+  ClearControlDepVars(graph.get());
  CollectSkipVarsSet(nodes);

  cfg_.reset(new details::ControlFlowGraph(*graph));
@ -128,7 +129,7 @@ std::unique_ptr<ir::Graph> MemoryOptimizePass::ApplyImpl(
      }
    }
  }
-  // graph->ResolveHazard(var_nodes_);
+  graph->ResolveHazard(var_nodes_);

  return graph;
 }