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>
[ROCm/hipother commit: f874762659]
This commit is contained in:
@@ -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));
|
||||
|
||||
مرجع در شماره جدید
Block a user