diff options
Diffstat (limited to 'src/nccl.h.in')
| -rw-r--r-- | src/nccl.h.in | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/src/nccl.h.in b/src/nccl.h.in index 985274e..f07e0a4 100644 --- a/src/nccl.h.in +++ b/src/nccl.h.in @@ -41,7 +41,7 @@ typedef enum { ncclSuccess = 0, * This integer is coded with the MAJOR, MINOR and PATCH level of the * NCCL library */ -ncclResult_t ncclGetVersion(int *version); +ncclResult_t ncclGetVersion(int *version); ncclResult_t pncclGetVersion(int *version); /* Generates an Id to be used in ncclCommInitRank. ncclGetUniqueId should be @@ -244,7 +244,8 @@ ncclResult_t pncclAllGather(const void* sendbuff, void* recvbuff, size_t sendcou * Start a group call. All subsequent calls to NCCL may not block due to * inter-CPU synchronization. */ -ncclResult_t ncclGroupStart(); +ncclResult_t ncclGroupStart(); +ncclResult_t pncclGroupStart(); /* * Group End @@ -252,7 +253,8 @@ ncclResult_t ncclGroupStart(); * End a group call. Wait for all calls since ncclGroupStart to complete * before returning. */ -ncclResult_t ncclGroupEnd(); +ncclResult_t ncclGroupEnd(); +ncclResult_t pncclGroupEnd(); #ifdef __cplusplus } // end extern "C" |