Searched refs:nccl_id (Results 1 – 1 of 1) sorted by relevance
113 void StringToNcclUniqueId(const string& str_id, ncclUniqueId* nccl_id) { in StringToNcclUniqueId() argument115 memcpy(nccl_id->internal, str_id.data(), NCCL_UNIQUE_ID_BYTES); in StringToNcclUniqueId()190 ncclUniqueId nccl_id; in GenerateCommunicatorKey() local191 ncclGetUniqueId(&nccl_id); in GenerateCommunicatorKey()192 return string(nccl_id.internal, NCCL_UNIQUE_ID_BYTES); in GenerateCommunicatorKey()305 ncclUniqueId nccl_id; in GetCommunicator() local307 NCCL_RETURN_IF_ERROR(ncclGetUniqueId(&nccl_id)); in GetCommunicator()309 StringToNcclUniqueId(collective->communicator_key, &nccl_id); in GetCommunicator()321 nccl_comms.data() + i, collective->num_global_devices, nccl_id, rank)); in GetCommunicator()