Searched refs:nccl_id (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_manager.cc | 145 void StringToNcclUniqueId(const string& str_id, ncclUniqueId* nccl_id) { in StringToNcclUniqueId() argument
147 memcpy(nccl_id->internal, str_id.data(), NCCL_UNIQUE_ID_BYTES); in StringToNcclUniqueId()
262 ncclUniqueId nccl_id; in GenerateCommunicatorKey() local
263 ncclGetUniqueId(&nccl_id); in GenerateCommunicatorKey()
264 return string(nccl_id.internal, NCCL_UNIQUE_ID_BYTES); in GenerateCommunicatorKey()
394 ncclUniqueId nccl_id; in GetCommunicator() local
396 NCCL_RETURN_IF_ERROR(ncclGetUniqueId(&nccl_id)); in GetCommunicator()
398 StringToNcclUniqueId(collective->communicator_key, &nccl_id); in GetCommunicator()
410 nccl_comms.data() + i, collective->num_global_devices, nccl_id, rank)); in GetCommunicator()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | nccl_utils.cc | 126 Status ToNcclUniqueId(const std::string& str_id, ncclUniqueId* nccl_id) { in ToNcclUniqueId() argument
135 std::memcpy(static_cast<void*>(nccl_id), str_id.data(), NCCL_UNIQUE_ID_BYTES); in ToNcclUniqueId()
|