SWDEV-461072 - Extend AQL Optimization for child graph nodes

Change-Id: I6baf906add7240b29ea653020a9a0b56206ee2a7

hipamd/src/hip_graph.cpp

@@ -1760,7 +1760,24 @@ hipError_t hipGraphExecChildGraphNodeSetParams(hipGraphExec_t hGraphExec, hipGra
   if (clonedNode == nullptr) {
     HIP_RETURN(hipErrorInvalidValue);
   }
-  HIP_RETURN(reinterpret_cast<hip::ChildGraphNode*>(clonedNode)->SetParams(cg));
+  hipError_t status = reinterpret_cast<hip::ChildGraphNode*>(clonedNode)->SetParams(cg);
+  if (status != hipSuccess) {
+    return status;
+  }
+  if (reinterpret_cast<hip::ChildGraphNode*>(clonedNode)->GetGraphCaptureStatus()) {
+    std::vector<hip::GraphNode*> childGraphNodes;
+    reinterpret_cast<hip::ChildGraphNode*>(clonedNode)->TopologicalOrder(childGraphNodes);
+    for (std::vector<hip::GraphNode*>::size_type i = 0; i != childGraphNodes.size(); i++) {
+      if (childGraphNodes[i]->GetType() == hipGraphNodeTypeKernel) {
+        status = reinterpret_cast<hip::GraphExec*>(hGraphExec)
+                     ->UpdateAQLPacket(reinterpret_cast<hip::GraphKernelNode*>(childGraphNodes[i]));
+        if (status != hipSuccess) {
+          return status;
+        }
+      }
+    }
+  }
+  return status;
 }
 
 hipError_t hipStreamGetCaptureInfo_common(hipStream_t stream,