提交 feb569f8 编写于 作者: Y Yu Yang

Add log

上级 260cfe3b
...@@ -376,7 +376,7 @@ struct NCCLAllReduceOpHandle : public OpHandle { ...@@ -376,7 +376,7 @@ struct NCCLAllReduceOpHandle : public OpHandle {
return; // No need to all reduce when GPU count = 1; return; // No need to all reduce when GPU count = 1;
} else { } else {
auto &var_name = static_cast<VarHandle *>(this->inputs_[0])->name_; auto &var_name = static_cast<VarHandle *>(this->inputs_[0])->name_;
VLOG(3) << "Invoke NCCL AllReduce";
int dtype = -1; int dtype = -1;
size_t numel = 0; size_t numel = 0;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册