Searched refs:nccl_comm (Results 1 – 1 of 1) sorted by relevance
75 if (nccl_comm != nullptr) ncclCommDestroy(nccl_comm); in ~CommunicatorMember()77 ncclComm_t nccl_comm; member335 members[i].nccl_comm = nccl_comms[i]; in GetCommunicator()568 auto nccl_comm = collective->communicator->members[p_idx].nccl_comm; in LoopKernelLaunches() local577 << nccl_comm << " comm_stream " << comm_stream in LoopKernelLaunches()581 nccl_comm, *cu_stream); in LoopKernelLaunches()588 collective->root_rank, nccl_comm, *cu_stream); in LoopKernelLaunches()598 collective->root_rank, nccl_comm, *cu_stream); in LoopKernelLaunches()608 << p->output->NumElements() << " nccl_comm " << nccl_comm in LoopKernelLaunches()612 data_type, nccl_comm, *cu_stream); in LoopKernelLaunches()