diff --git a/projects/hipother/hipnv/include/hip/nvidia_detail/nvidia_hip_runtime_api.h b/projects/hipother/hipnv/include/hip/nvidia_detail/nvidia_hip_runtime_api.h index b88548630a..20d73aec6c 100644 --- a/projects/hipother/hipnv/include/hip/nvidia_detail/nvidia_hip_runtime_api.h +++ b/projects/hipother/hipnv/include/hip/nvidia_detail/nvidia_hip_runtime_api.h @@ -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));