Searched refs:recvbuff (Results 1 – 1 of 1) sorted by relevance
573 void* recvbuff = const_cast<char*>(p->output->tensor_data().data()); in LoopKernelLaunches() local576 << sendbuff << " recvbuff " << recvbuff << " nccl_comm " in LoopKernelLaunches()579 nccl_result = ncclAllReduce(sendbuff, recvbuff, p->input->NumElements(), in LoopKernelLaunches()593 void* recvbuff = in LoopKernelLaunches() local596 nccl_result = ncclReduce(sendbuff, recvbuff, p->input->NumElements(), in LoopKernelLaunches()603 void* recvbuff = const_cast<char*>(p->output->tensor_data().data()); in LoopKernelLaunches() local607 << " recvbuff " << recvbuff << " recvcount " in LoopKernelLaunches()611 nccl_result = ncclAllGather(sendbuff, recvbuff, p->input->NumElements(), in LoopKernelLaunches()