Searched refs:nccl_result (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_manager.cc | 569 ncclResult_t nccl_result = ncclSuccess; in LoopKernelLaunches() local
579 nccl_result = ncclAllReduce(sendbuff, recvbuff, p->input->NumElements(), in LoopKernelLaunches()
587 nccl_result = ncclBcast(buf, buf_t->NumElements(), data_type, in LoopKernelLaunches()
596 nccl_result = ncclReduce(sendbuff, recvbuff, p->input->NumElements(), in LoopKernelLaunches()
611 nccl_result = ncclAllGather(sendbuff, recvbuff, p->input->NumElements(), in LoopKernelLaunches()
618 auto done_callback = [collective, p_idx, nccl_result]() { in LoopKernelLaunches()
619 if (nccl_result == ncclSuccess) { in LoopKernelLaunches()
625 "Error invoking NCCL: ", ncclGetErrorString(nccl_result))); in LoopKernelLaunches()
|