Searched refs:ncclComm_t (Results 1 – 11 of 11) sorted by relevance
54 using ncclComm_t = ncclComm*; variable127 ncclComm_t comm) = 0;
83 using NcclComm = std::unique_ptr<ncclComm, void (*)(ncclComm_t)>;95 ncclComm_t GetCommForDeviceOrdinal(int device_ordinal) const;
104 ncclComm_t NcclClique::GetCommForDeviceOrdinal(int device_ordinal) const { in GetCommForDeviceOrdinal()121 void DestroyNcclComm(ncclComm_t comm) { in DestroyNcclComm()179 std::vector<ncclComm_t> raw_comms(local_participants.size(), nullptr); in CreateNcclClique()
49 ncclComm_t comm) override;
51 ncclComm_t comm) override;
99 ncclComm_t comm = in ExecuteOnStream()
71 ncclComm_t comm) { in RunNcclCollective()
63 ncclComm_t comm) { in RunNcclCollective()
120 ncclComm_t comm) { in RunNcclCollective()
106 ncclComm_t nccl_comm = nullptr;390 std::vector<ncclComm_t> nccl_comms(collective->num_local_devices); in GetCommunicator()