提交 6bc261b9 编写于 作者: D Dong Zhihong

fix ci

上级 4b9cf0e8
...@@ -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.
先完成此消息的编辑!
想要评论请 注册