Searched refs:raw_comms (Results 1 – 1 of 1) sorted by relevance
179 std::vector<ncclComm_t> raw_comms(local_participants.size(), nullptr); in CreateNcclClique() local185 XLA_CUDA_RETURN_IF_ERROR(ncclCommInitRank(&raw_comms[i], num_participants, in CreateNcclClique()195 absl::flat_hash_map<int, NcclComm> comms_by_device_ordinal(raw_comms.size()); in CreateNcclClique()196 for (int i = 0; i < raw_comms.size(); ++i) { in CreateNcclClique()199 device_ordinal, raw_comms[i]); in CreateNcclClique()200 CHECK(raw_comms[i] != nullptr || !status.ok()); in CreateNcclClique()202 NcclComm(raw_comms[i], &DestroyNcclComm)); in CreateNcclClique()