From 4ef34916a41a98ebbd7cd560e350d2690a2c9c1e Mon Sep 17 00:00:00 2001 From: dzhwinter Date: Fri, 1 Feb 2019 15:36:07 +0800 Subject: [PATCH] enhanced print message. test=develop --- paddle/fluid/framework/details/inplace_op_pass.cc | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/paddle/fluid/framework/details/inplace_op_pass.cc b/paddle/fluid/framework/details/inplace_op_pass.cc index cf16ac126..78c5d5b50 100644 --- 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()) { -- GitLab