From 38d75c912cf7ca3e64bedc1446334a1a33d8e776 Mon Sep 17 00:00:00 2001 From: Yu Yang Date: Wed, 18 Apr 2018 17:23:33 +0800 Subject: [PATCH] Follow comments --- .../fluid/framework/details/multi_devices_graph_builder.cc | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/paddle/fluid/framework/details/multi_devices_graph_builder.cc b/paddle/fluid/framework/details/multi_devices_graph_builder.cc index 882d4389349..d2b6a35a5d5 100644 --- a/paddle/fluid/framework/details/multi_devices_graph_builder.cc +++ b/paddle/fluid/framework/details/multi_devices_graph_builder.cc @@ -143,10 +143,8 @@ void MultiDevSSAGraphBuilder::InsertNCCLAllReduceOp( for (size_t i = 0; i < places_.size(); ++i) { auto &p = places_[i]; auto &vars = result->vars_[i][og]; - if (vars.empty()) { // This device has no data. continue. - continue; - } - auto &prev_grad = vars[vars.size() - 1]; + PADDLE_ENFORCE(!vars.empty()); + auto &prev_grad = vars.back(); op_handle->AddInput(prev_grad.get()); auto var = new VarHandle(vars.size() - 1, i, og, p); -- GitLab