Этот коммит содержится в:
BertanDogancay
2024-02-11 22:32:55 -08:00
родитель d999d9ad21
Коммит 32cca51894
+9 -13
Просмотреть файл
@@ -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);