Searched refs:nccl_comms (Results 1 – 1 of 1) sorted by relevance
301 std::vector<ncclComm_t> nccl_comms(collective->num_local_devices); in GetCommunicator() local321 nccl_comms.data() + i, collective->num_global_devices, nccl_id, rank)); in GetCommunicator()331 nccl_comms.data(), collective->num_local_devices, devices.data())); in GetCommunicator()335 members[i].nccl_comm = nccl_comms[i]; in GetCommunicator()