Home
last modified time | relevance | path

Searched refs:group_info_ (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/distribution/
Dnccl_wrapper.cc39 for (auto group : group_info_) { in InitNCCLComm()
44 group_info_[group_name].comm = group_info.comm; in InitNCCLComm()
51 CHECK_RET(group_info_.count(group_name), 1, in AllReduce()
53 ncclComm_t group_comm = group_info_[group_name].comm; in AllReduce()
59 CHECK_RET(group_info_.count(group_name), 1, in AllGather()
61 ncclComm_t group_comm = group_info_[group_name].comm; in AllGather()
68 CHECK_RET(group_info_.count(group_name), 1, in ReduceScatter()
70 ncclComm_t group_comm = group_info_[group_name].comm; in ReduceScatter()
76 CHECK_RET(group_info_.count(group_name), 1, in Broadcast()
78 ncclComm_t group_comm = group_info_[group_name].comm; in Broadcast()
[all …]
Dnccl_wrapper.h62 std::map<std::string, NcclGroupInfo> group_info_; variable
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/distribute/
Dmpi_collective_group.cc26 group_info_.clear(); in FinalizeMPI()
44 CHECK_RET(group_info_.count(name), 1, "Failed to get MPI group rank by group name " + name); in GetRankIdByGroup()
45 return group_info_[name].first; in GetRankIdByGroup()
48 CHECK_RET(group_info_.count(name), 1, "Failed to get MPI group size by group name " + name); in GetGroupSize()
49 return group_info_[name].second; in GetGroupSize()
69 group_info_["hccl_world_group"] = {rank_id_, rank_size_}; in Init()
107 group_info_[name] = {group_rank[0], static_cast<int>(ranks.size())}; in CreateCommGroup()
Dmpi_collective_group.h58 std::map<std::string, std::pair<int, int>> group_info_; variable
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dgroup_manager.h66 …d::vector<std::pair<std::string, std::vector<uint32_t>>> group_info() const { return group_info_; } in group_info()
77 std::vector<std::pair<std::string, std::vector<uint32_t>>> group_info_; variable
Dgroup_manager.cc201 group_info_.push_back(group_info); in CreateGroup()