Searched refs:group_ranks (Results 1 – 8 of 8) sorted by relevance
77 std::vector<int> group_ranks(ranks.begin(), ranks.end()); in CreateCommGroup() local85 …CHECK_RET(MPI_Group_incl(comm_group_world_, group_ranks.size(), group_ranks.data(), &mpi_group), M… in CreateCommGroup()
43 …per::CreateCommGroup(const std::string &group_name, const std::vector<unsigned int> &group_ranks) { in CreateCommGroup() argument44 std::vector<int> ranks(group_ranks.begin(), group_ranks.end()); in CreateCommGroup()
35 std::vector<int> group_ranks; member
122 return group_info_[group_name].group_ranks; in GetGroupRanks()
37 RankList group_ranks; member
238 parameter_slice_info.group_ranks = GetGroupByTensorInfo(tensor_info); in GetParameterSliceInfo()241 … << tensor_info.shape() << ", the group rank list is " << parameter_slice_info.group_ranks; in GetParameterSliceInfo()258 RankList first_user_group_list = parameter_slice_info.group_ranks; in CheckParameterSplit()264 RankList user_group_list = user_slice_info.group_ranks; in CheckParameterSplit()
130 std::vector<int> group_ranks = (*get_group_size_funcptr)(group_name_); in Init() local131 device_count_ = group_ranks.size(); in Init()
146 std::vector<int> group_ranks = (*get_group_size_funcptr)(group_name_); in Init() local147 group_size_ = group_ranks.size(); in Init()