Add support for hipExtStreamGetCUMask API

Change-Id: I0fa67ad581dd75556f17c7410af2c1d5cb6ae99a
このコミットが含まれているのは:
Aryan Salmanpour
2020-11-03 12:24:04 -05:00
committed by Aryan Salmanpour
コミット d9a335bccf
5個のファイルの変更102行の追加0行の削除
+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
*/