Skip to content

P
Paddle

机器未来 / Paddle
与 Fork 源项目一致

Fork自 PaddlePaddle / Paddle

通知 1
Star 1
Fork 0
P
Paddle
提交 6bc261b9 编写于 作者: D Dong Zhihong
浏览文件
操作

fix ci

上级 4b9cf0e8
隐藏空白更改
内联 并排
Showing with 2 addition and 0 deletion
paddle/operators/nccl/nccl_gpu_common.h
浏览文件 @ 6bc261b9
...@@ -34,6 +34,8 @@ struct Communicator { ...@@ -34,6 +34,8 @@ struct Communicator {
std::vector<ncclComm_t> comms_; std::vector<ncclComm_t> comms_;
std::unordered_map<int, int> comm_id_map_; std::unordered_map<int, int> comm_id_map_;
Communicator() {}
int GetCommId(int device_id) const { return comm_id_map_.at(device_id); } int GetCommId(int device_id) const { return comm_id_map_.at(device_id); }
void InitAll(const std::vector<int>& gpus) { void InitAll(const std::vector<int>& gpus) {
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或