Searched refs:GetCommForDeviceOrdinal (Results 1 – 3 of 3) sorted by relevance
100 locked_clique.clique.GetCommForDeviceOrdinal(device_ordinal); in ExecuteOnStream()
95 ncclComm_t GetCommForDeviceOrdinal(int device_ordinal) const;
104 ncclComm_t NcclClique::GetCommForDeviceOrdinal(int device_ordinal) const { in GetCommForDeviceOrdinal() function in xla::gpu::NcclClique