SWDEV-483315 - Add hipGraphNodeGetDependentNodes_v2

Change-Id: I418d0c2a5a77b07603e4e481fd9547cf43b03a20


[ROCm/hipother commit: e22853eb4d]
This commit is contained in:
Branislav Brzak
2024-09-30 11:36:43 +00:00
کامیت شده توسط Branislav Brzak
والد 927cc12c33
کامیت 2896b2c2d9
@@ -3711,6 +3711,14 @@ inline static hipError_t hipStreamBeginCaptureToGraph(hipStream_t stream, hipGra
return hipCUDAErrorTohipError(cudaStreamBeginCaptureToGraph(
stream, graph, dependencies, dependencyData, numDependencies, mode));
}
inline static hipError_t hipGraphNodeGetDependentNodes_v2(hipGraphNode_t node,
hipGraphNode_t* pDependentNodes,
hipGraphEdgeData* edgeData,
size_t* pNumDependentNodes) {
return hipCUDAErrorTohipError(
cudaGraphNodeGetDependentNodes_v2(node, pDependentNodes, edgeData, pNumDependentNodes));
}
#endif
inline static hipError_t hipStreamEndCapture(hipStream_t stream, hipGraph_t* pGraph) {
return hipCUDAErrorTohipError(cudaStreamEndCapture(stream, pGraph));