diff options
Diffstat (limited to 'src/include/group.h')
-rw-r--r-- | src/include/group.h | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/include/group.h b/src/include/group.h index 76da30f..239b05f 100644 --- a/src/include/group.h +++ b/src/include/group.h @@ -8,14 +8,14 @@ #define NCCL_GROUP_H_ #include "nccl.h" -#include "core.h" +#include "comm.h" bool ncclAsyncMode(); ncclResult_t ncclAsyncErrCheck(ncclResult_t ret); -typedef ncclResult_t(*ncclInitFunc_t)(ncclComm_t* newcomm, int ndev, ncclUniqueId commId, int myrank); +typedef ncclResult_t(*ncclInitFunc_t)(ncclComm_t* newcomm, int ndev, ncclUniqueId commId, int myrank, int cudaDev); -ncclResult_t ncclAsyncInit(ncclInitFunc_t func, int cudaDev, ncclComm_t* newcomm, int ndev, ncclUniqueId commId, int myrank); +ncclResult_t ncclAsyncInit(ncclInitFunc_t func, ncclComm_t* newcomm, int ndev, ncclUniqueId commId, int myrank, int cudaDev); typedef ncclResult_t(*ncclCollFunc_t)(const void* sendbuff, void* recvbuff, size_t count, ncclDataType_t type, ncclRedOp_t op, int root, ncclComm_t comm, cudaStream_t stream); |