enhanced print message. test=develop

6 years ago · 4ef34916a4
parent ce0394bcd0
commit 4ef34916a4
1 changed files with 2 additions and 1 deletions
--- a/paddle/fluid/framework/details/inplace_op_pass.cc
+++ b/paddle/fluid/framework/details/inplace_op_pass.cc
@ -408,7 +408,8 @@ void GraphView::Build(ir::Graph* g) {
      if (in->IsVar() && in->Var() != nullptr) dup_nodes_.emplace(in->Name());
    }
    for (auto& out : node->outputs) {
-      if (in->IsVar() && in->Var() != nullptr) dup_nodes_.emplace(in->Name());
+      if (out->IsVar() && out->Var() != nullptr)
+        dup_nodes_.emplace(out->Name());
    }
  };
  for (auto& node : g->Nodes()) {