提交 38d75c91 编写于 作者: Y Yu Yang

Follow comments

上级 35ebe2ec
...@@ -143,10 +143,8 @@ void MultiDevSSAGraphBuilder::InsertNCCLAllReduceOp( ...@@ -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);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册