Searched refs:rank_num (Results 1 – 5 of 5) sorted by relevance
43 def check_rank_num(rank_num): argument50 if isinstance(rank_num, (int)):51 if rank_num > MAX_RANK_NUM or rank_num <= 0:96 def create_group(group, rank_num, rank_ids): argument110 check_rank_num(rank_num)112 if rank_num != len(rank_ids):118 c_rank_num = ctypes.c_uint(rank_num)
86 size_t rank_num = LongToSize(parallel_group_ranks.dim_size()); in LoadGroupInfo() local88 for (size_t j = 0; j < rank_num; ++j) { in LoadGroupInfo()
52 HcclResult HcclCreateGroup(const std::string &group, uint32_t rank_num, uint32_t *rank_ids) const;
434 HcclResult HcclAdapter::HcclCreateGroup(const std::string &group, uint32_t rank_num, uint32_t *rank… in HcclCreateGroup() argument436 return hccl_create_group_(group.c_str(), rank_num, rank_ids); in HcclCreateGroup()
1397 for rank_num in param_index:1398 param_stride.append(param_total_dict[param.name][rank_num].data.asnumpy())