avoid optimize variable used in subblock, test=develop (#16739)

6 years ago · fdb719a1bf
parent ed97156461
commit fdb719a1bf
1 changed files with 3 additions and 0 deletions
--- a/paddle/fluid/framework/details/inplace_op_pass.cc
+++ b/paddle/fluid/framework/details/inplace_op_pass.cc
@ -533,6 +533,9 @@ void GraphView::Build(ir::Graph* g) {
  };
  for (auto& node : g->Nodes()) {
    if (!node->IsOp()) continue;
+    // avoid optimize the variable used in sub-blocks
+    if (OpHasSubBlock(node->Op())) update_skip_set(node);
+
    if (node->Name() == "send") update_skip_set(node);
    if (node->Name() == "recv") update_skip_set(node);
    if (node->Name() == "prefetch") update_skip_set(node);