|
|
@ -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);
|
|
|
|