diff --git a/src/alltoall.cu b/src/alltoall.cu index c81f2ff..c6fa782 100644 --- a/src/alltoall.cu +++ b/src/alltoall.cu @@ -48,7 +48,7 @@ void AlltoAllGetBw(size_t count, int typesize, double sec, double* algBw, double } testResult_t AlltoAllRunColl(void* sendbuff, void* recvbuff, size_t count, ncclDataType_t type, ncclRedOp_t op, int root, ncclComm_t comm, cudaStream_t stream, void* bias = nullptr) { - NCCLCHECK(ncclAllToAll(sendbuff, recvbuff, count, type, comm, stream)); + NCCLCHECK(ncclAlltoAll(sendbuff, recvbuff, count, type, comm, stream)); return testSuccess; } diff --git a/src/common.h b/src/common.h index 4c6af89..bdeab43 100644 --- a/src/common.h +++ b/src/common.h @@ -387,7 +387,7 @@ typedef enum { ncclFuncSendRecv = 6, ncclFuncSend = 7, ncclFuncRecv = 8, - ncclFuncAllToAllPivot = 9, + ncclFuncAlltoAll = 9, ncclNumFuncs = 10 } ncclFunc_t;