Searched refs:NcclComm (Results 1 – 2 of 2) sorted by relevance
83 using NcclComm = std::unique_ptr<ncclComm, void (*)(ncclComm_t)>; variable93 absl::flat_hash_map<int, NcclComm> comms_by_device_ordinal);99 absl::flat_hash_map<int, NcclComm> comms_by_device_ordinal_;
101 absl::flat_hash_map<int, NcclComm> comms_by_device_ordinal) in NcclClique()195 absl::flat_hash_map<int, NcclComm> comms_by_device_ordinal(raw_comms.size()); in CreateNcclClique()202 NcclComm(raw_comms[i], &DestroyNcclComm)); in CreateNcclClique()