Add support for hipExtStreamGetCUMask API

Change-Id: I0fa67ad581dd75556f17c7410af2c1d5cb6ae99a
Этот коммит содержится в:
Aryan Salmanpour authored and Aryan Salmanpour committed 2020-11-20 13:41:38 -05:00
1 родитель 5e43a6defb
Коммит d9a335bccf
5 files changed
+102

No files matched your search

+14
Просмотреть файл
@@ -1196,6 +1196,20 @@ hipError_t hipStreamGetPriority(hipStream_t stream, int* priority);
hipError_t hipExtStreamCreateWithCUMask(hipStream_t* stream, uint32_t cuMaskSize, const uint32_t* cuMask);
/**
* @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
*/
hipError_t hipExtStreamGetCUMask(hipStream_t stream, uint32_t cuMaskSize, uint32_t* cuMask);
/**
* Stream CallBack struct
*/