cubecl_hip_sys

Function hipDeviceGetMemPool

Source 
pub unsafe extern "C" fn hipDeviceGetMemPool(
    mem_pool: *mut hipMemPool_t,
    device: c_int,
) -> hipError_t
Expand description

@brief Gets the current memory pool for the specified device

Returns the last pool provided to @p hipDeviceSetMemPool for this device or the device’s default memory pool if @p hipDeviceSetMemPool has never been called. By default the current mempool is the default mempool for a device, otherwise the returned pool must have been set with @p hipDeviceSetMemPool.

@param [out] mem_pool Current memory pool on the specified device @param [in] device Device index to query the current memory pool

@returns #hipSuccess, #hipErrorInvalidValue, #hipErrorNotSupported

@see hipDeviceGetDefaultMemPool, hipMallocAsync, hipMemPoolTrimTo, hipMemPoolGetAttribute, hipDeviceSetMemPool, hipMemPoolSetAttribute, hipMemPoolSetAccess, hipMemPoolGetAccess

@warning : This API is marked as beta, meaning, while this is feature complete, it is still open to changes and may have outstanding issues.