cubecl_hip_sys

Function hipMemGetInfo

Source 
pub unsafe extern "C" fn hipMemGetInfo(
    free: *mut usize,
    total: *mut usize,
) -> hipError_t
Expand description

@brief Query memory info.

On ROCM, this function gets the actual free memory left on the current device, so supports the cases while running multi-workload (such as multiple processes, multiple threads, and multiple GPUs).

@warning On Windows, the free memory only accounts for memory allocated by this process and may be optimistic.

@param[out] free Returns free memory on the current device in bytes @param[out] total Returns total allocatable memory on the current device in bytes

@return #hipSuccess, #hipErrorInvalidDevice, #hipErrorInvalidValue