AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

SWDEV-318833 - Get and Set attribute for kernel nodes (#2533)

Browse Source 
Signed-off-by: sdashmiz <shadi.dashmiz@amd.com>
Change-Id: Ieacd1fd9aa6d6e98a3a5574ba374c201f05d0e51

[ROCm/hip commit: c52a81e96f]
This commit is contained in:
ROCm CI Service Account
2022-03-10 16:47:03 +05:30
committed by GitHub
parent f0187ca1ae
commit 1621668fe9
1 changed files with 49 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
projects/hip/include/hip/hip_runtime_api.h
+49
View File
@@ -165,6 +165,31 @@ typedef enum hipMemoryType {
hipMemoryTypeUnified ///< Not used currently
} hipMemoryType;
/**
* @brief hipKernelNodeAttrID
* @enum
*
*/
typedef enum hipKernelNodeAttrID {
hipKernelNodeAttributeAccessPolicyWindow = 1,
hipKernelNodeAttributeCooperative = 2,
} hipKernelNodeAttrID;
typedef enum hipAccessProperty {
hipAccessPropertyNormal = 0,
hipAccessPropertyStreaming = 1,
hipAccessPropertyPersisting = 2,
} hipAccessProperty;
typedef struct hipAccessPolicyWindow {
void* base_ptr;
hipAccessProperty hitProp;
float hitRatio;
hipAccessProperty missProp;
size_t num_bytes;
} hipAccessPolicyWindow;
typedef union hipKernelNodeAttrValue {
hipAccessPolicyWindow accessPolicyWindow;
int cooperative;
} hipKernelNodeAttrValue;
/**
* Pointer attributes
@@ -4887,6 +4912,30 @@ hipError_t hipGraphMemcpyNodeGetParams(hipGraphNode_t node, hipMemcpy3DParms* pN
*/
hipError_t hipGraphMemcpyNodeSetParams(hipGraphNode_t node, const hipMemcpy3DParms* pNodeParams);
/**
* @brief Sets a node attribute.
*
* @param [in] hNode - instance of the node to set parameters to.
* @param [in] attr - the attribute node is set to.
* @param [in] value - const pointer to the parameters.
* @returns #hipSuccess, #hipErrorInvalidValue
* @warning : This API is marked as beta, meaning, while this is feature complete,
* it is still open to changes and may have outstanding issues.
*/
hipError_t hipGraphKernelNodeSetAttribute(hipGraphNode_t hNode, hipKernelNodeAttrID attr,
const hipKernelNodeAttrValue* value);
/**
* @brief Gets a node attribute.
*
* @param [in] hNode - instance of the node to set parameters to.
* @param [in] attr - the attribute node is set to.
* @param [in] value - const pointer to the parameters.
* @returns #hipSuccess, #hipErrorInvalidValue
* @warning : This API is marked as beta, meaning, while this is feature complete,
* it is still open to changes and may have outstanding issues.
*/
hipError_t hipGraphKernelNodeGetAttribute(hipGraphNode_t hNode, hipKernelNodeAttrID attr,
hipKernelNodeAttrValue* value);
/**
* @brief Sets the parameters for a memcpy node in the given graphExec.
*