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

fdb719a1 · liuwei1031 · GitHub · ed971564 · fdb719a1
隐藏空白更改
内联并排

Showing with 3 addition and 0 deletion

paddle/fluid/framework/details/inplace_op_pass.cc paddle/fluid/framework/details/inplace_op_pass.cc +3 -0

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