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

SWDEV-325711: Add userobject functions for graph

Browse Source 
- add user obj APIs for creating release and retain of user onbjects

Signed-off-by: sdashmiz <shadi.dashmiz@amd.com>
Change-Id: I0bf2999c77e44269565b27c31c7c1461f8a160a2
This commit is contained in:
sdashmiz
2022-04-25 13:42:17 -04:00
committed by Shadi Dashmiz
parent 5a485e3393
commit d6cee2332a
1 changed files with 24 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
hipnv/include/hip/nvidia_detail/nvidia_hip_runtime_api.h
+24
View File
@@ -1107,6 +1107,7 @@ inline static enum cudaChannelFormatKind hipChannelFormatKindToCudaChannelFormat
typedef cudaGraph_t hipGraph_t;
typedef cudaGraphNode_t hipGraphNode_t;
typedef cudaGraphExec_t hipGraphExec_t;
typedef cudaUserObject_t hipUserObject_t;
typedef enum cudaGraphNodeType hipGraphNodeType;
#define hipGraphNodeTypeKernel cudaGraphNodeTypeKernel
@@ -3232,6 +3233,29 @@ inline static hipError_t hipDeviceSetGraphMemAttribute(int device, hipGraphMemAt
inline static hipError_t hipDeviceGraphMemTrim(int device) {
return hipCUDAErrorTohipError(cudaDeviceGraphMemTrim(device));
}
inline static hipError_t hipUserObjectCreate(hipUserObject_t* object_out, void* ptr, hipHostFn_t destroy,
unsigned int initialRefcount, unsigned int flags) {
return hipCUDAErrorTohipError(cudaUserObjectCreate(object_out, ptr, destroy, initialRefcount, flags));
}
inline static hipError_t hipUserObjectRelease(hipUserObject_t object, unsigned int count __dparm(1)) {
return hipCUDAErrorTohipError(cudaUserObjectRelease(object, count));
}
inline static hipError_t hipUserObjectRetain(hipUserObject_t object, unsigned int count __dparm(1)) {
return hipCUDAErrorTohipError(cudaUserObjectRelease(object, count));
}
inline static hipError_t hipGraphRetainUserObject(hipGraph_t graph, hipUserObject_t object, unsigned int count __dparm(1), unsigned int flags __dparm(0)) {
return hipCUDAErrorTohipError(cudaGraphRetainUserObject(graph, object, count, flags));
}
inline static hipError_t hipGraphReleaseUserObject(hipGraph_t graph, hipUserObject_t object, unsigned int count __dparm(1)) {
return hipCUDAErrorTohipError(cudaGraphReleaseUserObject(graph, object, count));
}
#endif
inline static hipError_t hipGraphHostNodeSetParams(hipGraphNode_t node,