SWDEV-318833 - Get and Set attribute for kernel nodes
Signed-off-by: sdashmiz <shadi.dashmiz@amd.com> Change-Id: Ie51aa4b56661cbb8c5b4eb1dbaad327377084ffb Signed-off-by: sdashmiz <shadi.dashmiz@amd.com>
Tá an tiomantas seo le fáil i:
tiomanta ag
Shadi Dashmiz
tuismitheoir
43c650cf48
tiomantas
efd2d55e9d
@@ -1131,6 +1131,10 @@ typedef enum cudaStreamCaptureStatus hipStreamCaptureStatus;
|
||||
#define hipStreamCaptureStatusActive cudaStreamCaptureStatusActive
|
||||
#define hipStreamCaptureStatusInvalidated cudaStreamCaptureStatusInvalidated
|
||||
|
||||
typedef union cudaKernelNodeAttrValue hipKernelNodeAttrValue;
|
||||
typedef enum cudaKernelNodeAttrID hipKernelNodeAttrID;
|
||||
|
||||
|
||||
#if CUDA_VERSION >= CUDA_11030
|
||||
typedef enum cudaStreamUpdateCaptureDependenciesFlags hipStreamUpdateCaptureDependenciesFlags;
|
||||
#define hipStreamAddCaptureDependencies cudaStreamAddCaptureDependencies
|
||||
@@ -2677,6 +2681,16 @@ inline static hipError_t hipGraphKernelNodeSetParams(hipGraphNode_t node,
|
||||
return hipCUDAErrorTohipError(cudaGraphKernelNodeSetParams(node, pNodeParams));
|
||||
}
|
||||
|
||||
inline static hipError_t hipGraphKernelNodeSetAttribute(hipGraphNode_t hNode, hipKernelNodeAttrID attr,
|
||||
const hipKernelNodeAttrValue* value) {
|
||||
return hipCUDAErrorTohipError(cudaGraphKernelNodeSetAttribute(hNode, attr, value));
|
||||
}
|
||||
|
||||
inline static hipError_t hipGraphKernelNodeGetAttribute(hipGraphNode_t hNode, hipKernelNodeAttrID attr,
|
||||
hipKernelNodeAttrValue* value) {
|
||||
return hipCUDAErrorTohipError(cudaGraphKernelNodeGetAttribute(hNode, attr, value));
|
||||
}
|
||||
|
||||
inline static hipError_t hipGraphMemcpyNodeGetParams(hipGraphNode_t node,
|
||||
hipMemcpy3DParms* pNodeParams) {
|
||||
return hipCUDAErrorTohipError(cudaGraphMemcpyNodeGetParams(node, pNodeParams));
|
||||
|
||||
Tagairt in Eagrán Nua
Cuir bac ar úsáideoir