From 32cca518943b0a12543ca76545ea6376174cf427 Mon Sep 17 00:00:00 2001 From: BertanDogancay Date: Sun, 11 Feb 2024 22:32:55 -0800 Subject: [PATCH] Fix docs --- src/nccl.h.in | 22 +++++++++------------- 1 file changed, 9 insertions(+), 13 deletions(-) diff --git a/src/nccl.h.in b/src/nccl.h.in index 3cabb9976f..f9be58cb22 100644 --- a/src/nccl.h.in +++ b/src/nccl.h.in @@ -98,6 +98,15 @@ typedef struct ncclConfig_v21700 { } /*! @} */ +/* NCCL malloc and free function for all types of NCCL optimizations + * (e.g. user buffer registration). The actual allocated size might + * be larger than requested due to granularity requirement. */ +ncclResult_t ncclMemAlloc(void** ptr, size_t size); +ncclResult_t pncclMemAlloc(void** ptr, size_t size); + +ncclResult_t ncclMemFree(void *ptr); +ncclResult_t pncclMemFree(void *ptr); + /*! @defgroup rccl_api_version Version Information @details API call that returns RCCL version @{ */ @@ -108,19 +117,6 @@ typedef struct ncclConfig_v21700 { @param[out] version Pointer to where version will be stored */ -/* NCCL malloc and free function for all types of NCCL optimizations - * (e.g. user buffer registration). The actual allocated size might - * be larger than requested due to granularity requirement. */ -ncclResult_t ncclMemAlloc(void** ptr, size_t size); -ncclResult_t pncclMemAlloc(void** ptr, size_t size); - -ncclResult_t ncclMemFree(void *ptr); -ncclResult_t pncclMemFree(void *ptr); - -/* Return the NCCL_VERSION_CODE of the NCCL library in the supplied integer. - * This integer is coded with the MAJOR, MINOR and PATCH level of the - * NCCL library - */ ncclResult_t ncclGetVersion(int *version); /*! @cond include_hidden */ ncclResult_t pncclGetVersion(int *version);