From da5efa735a3db297131421e9b7dab703f1d5d2ae Mon Sep 17 00:00:00 2001 From: Xin Pan Date: Tue, 17 Jul 2018 21:34:00 +0800 Subject: [PATCH] fix --- paddle/fluid/framework/details/var_handle.h | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/paddle/fluid/framework/details/var_handle.h b/paddle/fluid/framework/details/var_handle.h index ba8b38d1e7..d8c2bc40b9 100644 --- a/paddle/fluid/framework/details/var_handle.h +++ b/paddle/fluid/framework/details/var_handle.h @@ -57,7 +57,8 @@ struct VarHandleBase { void RemoveOutput(OpHandleBase* out, ir::Node* node) { pending_ops_.erase(out); node_->outputs.erase( - std::remove(node_->outputs.begin(), node_->outputs.end(), node)); + std::remove(node_->outputs.begin(), node_->outputs.end(), node), + node_->outputs.end()); } void ClearGeneratedOp() {