Searched refs:local_participants (Results 1 – 3 of 3) sorted by relevance
140 const std::vector<LocalParticipant>& local_participants) { in LocalParticipantsToString() argument142 for (const LocalParticipant& local_participant : local_participants) { in LocalParticipantsToString()152 const std::vector<LocalParticipant>& local_participants, in CreateNcclClique() argument160 TF_RET_CHECK((num_participants == local_participants.size()) || in CreateNcclClique()168 << LocalParticipantsToString(local_participants); in CreateNcclClique()179 std::vector<ncclComm_t> raw_comms(local_participants.size(), nullptr); in CreateNcclClique()182 for (int i = 0; i < local_participants.size(); ++i) { in CreateNcclClique()184 cudaSetDevice(local_participants[i].device_ordinal)); in CreateNcclClique()187 local_participants[i].rank)); in CreateNcclClique()197 int device_ordinal = local_participants[i].device_ordinal; in CreateNcclClique()[all …]
85 std::vector<LocalParticipant> local_participants, in ExecuteOnStream()90 local_participants.size(), in ExecuteOnStream()98 local_participants, params.nccl_unique_id_callback)); in ExecuteOnStream()
152 se::Stream* stream, const std::vector<LocalParticipant>& local_participants,