Searched refs:comm_stream (Results 1 – 1 of 1) sorted by relevance
542 se::Stream* comm_stream = nccl_stream->stream.get(); in LoopKernelLaunches() local545 comm_stream->implementation()->GpuStreamMemberHack()); in LoopKernelLaunches()577 << nccl_comm << " comm_stream " << comm_stream in LoopKernelLaunches()609 << " comm_stream " << comm_stream << " cuda_stream " in LoopKernelLaunches()637 p->event_mgr->ThenExecute(comm_stream, done_callback); in LoopKernelLaunches()