SWDEV-556588 - Handle graph node set params and disabled nodes for AQL packet batching. (#1099)

2025-10-06 13:26:12 +05:30
parent 02883c3d8d
commit 98d6d268a0
4 changed files with 175 additions and 40 deletions
@@ -3084,6 +3084,11 @@ hipError_t hipGraphNodeSetEnabled(hipGraphExec_t hGraphExec, hipGraphNode_t hNod
    HIP_RETURN(hipErrorInvalidValue);
  }
  clonedNode->SetEnabled(isEnabled);
+  // Update packet batches when node is enabled/disabled
+  hipError_t status = graphExec->UpdatePacketBatchesForNodeEnableDisable(clonedNode, isEnabled != 0);
+  if (status != hipSuccess) {
+    HIP_RETURN(status);
+  }
  HIP_RETURN(hipSuccess);
 }