Searched refs:nccl_result (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_manager.cc | 734 ncclResult_t nccl_result = ncclSuccess; in LoopKernelLaunches() local
752 nccl_result = ncclAllReduce(sendbuff, recvbuff, p->input->NumElements(), in LoopKernelLaunches()
789 nccl_result = in LoopKernelLaunches()
805 nccl_result = ncclReduce(sendbuff, recvbuff, p->input->NumElements(), in LoopKernelLaunches()
827 nccl_result = ncclAllGather(sendbuff, recvbuff, p->input->NumElements(), in LoopKernelLaunches()
834 auto done_callback = [collective, p_idx, nccl_result]() { in LoopKernelLaunches()
837 << " ncclResult " << nccl_result; in LoopKernelLaunches()
838 if (nccl_result == ncclSuccess) { in LoopKernelLaunches()
844 "Error invoking NCCL: ", ncclGetErrorString(nccl_result))); in LoopKernelLaunches()
|