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