diff --git a/paddle/fluid/framework/details/inplace_op_pass.cc b/paddle/fluid/framework/details/inplace_op_pass.cc index cf16ac1264973e631580991244f565560f2c9a1b..78c5d5b50e606daa963e728355dc1bce83cd5484 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()) {