Follow comments

wangkuiyi-patch-2
Yu Yang 7 years ago
parent 35ebe2eca7
commit 38d75c912c

@ -143,10 +143,8 @@ void MultiDevSSAGraphBuilder::InsertNCCLAllReduceOp(
for (size_t i = 0; i < places_.size(); ++i) { for (size_t i = 0; i < places_.size(); ++i) {
auto &p = places_[i]; auto &p = places_[i];
auto &vars = result->vars_[i][og]; auto &vars = result->vars_[i][og];
if (vars.empty()) { // This device has no data. continue. PADDLE_ENFORCE(!vars.empty());
continue; auto &prev_grad = vars.back();
}
auto &prev_grad = vars[vars.size() - 1];
op_handle->AddInput(prev_grad.get()); op_handle->AddInput(prev_grad.get());
auto var = new VarHandle(vars.size() - 1, i, og, p); auto var = new VarHandle(vars.size() - 1, i, og, p);

Loading…
Cancel
Save