SWDEV-316819 - adding graph checks

hipGraphChildGraphNodeGetGraph - Scenario 3: Pass Parent node instead of the child node Change-Id: Ia55921bca1e2745c9c97f2b7eb36244ffb719eb4
2022-03-10 18:01:51 -05:00
parent e383c5fca0
commit 5dd91d698b
2 changed files with 4 additions and 1 deletions
@@ -1200,7 +1200,9 @@ hipError_t hipGraphChildGraphNodeGetGraph(hipGraphNode_t node, hipGraph_t* pGrap
    HIP_RETURN(hipErrorInvalidValue);
  }
  *pGraph = reinterpret_cast<hipChildGraphNode*>(node)->GetChildGraph();
-  if (pGraph == nullptr) {
+
+  //if the node count is larger than 0, the current node is a parent
+  if (*pGraph == nullptr || reinterpret_cast<ihipGraph*>(pGraph)->GetNodeCount() > 0) {
    HIP_RETURN(hipErrorInvalidValue);
  }
  HIP_RETURN(hipSuccess);