Searched refs:comm_stream (Results 1 – 1 of 1) sorted by relevance
700 se::Stream* comm_stream = nccl_stream->stream; in LoopKernelLaunches() local702 se::Stream* comm_stream = nccl_stream->stream.get(); in LoopKernelLaunches()706 comm_stream->implementation()->GpuStreamMemberHack()); in LoopKernelLaunches()744 << " nccl_comm " << nccl_comm << " comm_stream " << comm_stream in LoopKernelLaunches()781 << " nccl_comm " << nccl_comm << " comm_stream " << comm_stream in LoopKernelLaunches()819 << nccl_comm << " comm_stream " << comm_stream in LoopKernelLaunches()848 p->event_mgr->ThenExecute(comm_stream, done_callback); in LoopKernelLaunches()