Searched refs:kRankSize (Results 1 – 3 of 3) sorted by relevance
142 const int kRankSize = rank_size_; in AssignLocalRankID() local143 size_t all_host_hashs[kRankSize]; in AssignLocalRankID()144 CHECK_RET((rank_id_ < kRankSize), true, "The rank id is not less than rank size."); in AssignLocalRankID()148 for (int global_rank = 0; global_rank < kRankSize; global_rank++) { in AssignLocalRankID()
115 const int kRankSize = rank_size_; in AssignLocalRankID() local116 size_t all_host_hashs[kRankSize]; in AssignLocalRankID()120 for (int global_rank = 0; global_rank < kRankSize; global_rank++) { in AssignLocalRankID()
31 constexpr auto kRankSize = "rank_size"; variable465 if (!primitive->HasAttr(kRankSize)) { in InferImplAllGather()468 auto rank_size = GetValue<int>(primitive->GetAttr(kRankSize)); in InferImplAllGather()489 if (!primitive->HasAttr(kRankSize)) { in InferImplReduceScatter()492 auto rank_size = GetValue<int>(primitive->GetAttr(kRankSize)); in InferImplReduceScatter()