From d12bfbbb4d5d23796d2ee57d8be65e513a1048aa Mon Sep 17 00:00:00 2001 From: Jaydeep Patel Date: Thu, 5 Jan 2023 12:55:26 +0000 Subject: [PATCH] SWDEV-374383 - Check from and to if no of dependencies are more than 0, Check from and to node are belongs to graph pass in. Change-Id: I0cdafb497147f9ac7f84d40203a21e9eefd595a3 [ROCm/clr commit: bb486b04c6ecb92386d0a0d815248c15c54bc0a0] --- projects/clr/hipamd/src/hip_graph.cpp | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/projects/clr/hipamd/src/hip_graph.cpp b/projects/clr/hipamd/src/hip_graph.cpp index 0d7387e7b0..43e833bdf2 100644 --- a/projects/clr/hipamd/src/hip_graph.cpp +++ b/projects/clr/hipamd/src/hip_graph.cpp @@ -1615,11 +1615,12 @@ hipError_t hipStreamUpdateCaptureDependencies(hipStream_t stream, hipGraphNode_t hipError_t hipGraphRemoveDependencies(hipGraph_t graph, const hipGraphNode_t* from, const hipGraphNode_t* to, size_t numDependencies) { HIP_INIT_API(hipGraphRemoveDependencies, graph, from, to, numDependencies); - if (graph == nullptr || from == nullptr || to == nullptr) { + if (graph == nullptr || (numDependencies > 0 && (from == nullptr || to == nullptr))) { HIP_RETURN(hipErrorInvalidValue); } for (size_t i = 0; i < numDependencies; i++) { - if (from[i]->RemoveEdge(to[i]) == false) { + if (to[i]->GetParentGraph() != graph || from[i]->GetParentGraph() != graph || + from[i]->RemoveEdge(to[i]) == false) { HIP_RETURN(hipErrorInvalidValue); } }