cubecl_hip_sys

Function hipExtStreamGetCUMask

Source
pub unsafe extern "C" fn hipExtStreamGetCUMask(
    stream: hipStream_t,
    cuMaskSize: u32,
    cuMask: *mut u32,
) -> hipError_t
Expand description

@brief Get CU mask associated with an asynchronous stream

@param[in] stream stream to be queried @param[in] cuMaskSize number of the block of memories (uint32_t *) allocated by user @param[out] cuMask Pointer to a pre-allocated block of memories (uint32_t *) in which the stream’s CU mask is returned. The CU mask is returned in a chunck of 32 bits where each active bit represents one active CU @return #hipSuccess, #hipErrorInvalidHandle, #hipErrorInvalidValue

@see hipStreamCreate, hipStreamSynchronize, hipStreamWaitEvent, hipStreamDestroy