Searched refs:nccl_comm (Results 1 – 1 of 1) sorted by relevance
103 if (nccl_comm != nullptr) ncclCommDestroy(nccl_comm); in ~CommunicatorMember()106 ncclComm_t nccl_comm = nullptr; member424 members[i].nccl_comm = nccl_comms[i]; in GetCommunicator()733 auto nccl_comm = collective->communicator->members[p_idx].nccl_comm; in LoopKernelLaunches() local744 << " nccl_comm " << nccl_comm << " comm_stream " << comm_stream in LoopKernelLaunches()754 nccl_comm, *cu_stream); in LoopKernelLaunches()781 << " nccl_comm " << nccl_comm << " comm_stream " << comm_stream in LoopKernelLaunches()791 collective->root_rank, nccl_comm, *cu_stream); in LoopKernelLaunches()807 collective->root_rank, nccl_comm, *cu_stream); in LoopKernelLaunches()819 << nccl_comm << " comm_stream " << comm_stream in LoopKernelLaunches()[all …]