SWDEV-496544 - Reset mem alloc node count for AutoFreeOnLaunch.

Change-Id: Ib32b04584548a46632606ecd85b58c6ce4a5894d
2024-11-11 11:02:56 +00:00
parent cc25c5d646
commit 24c57cb984
1 changed files with 1 additions and 0 deletions
@@ -832,6 +832,7 @@ hipError_t GraphExec::Run(hipStream_t graph_launch_stream) {
  if (flags_ & hipGraphInstantiateFlagAutoFreeOnLaunch) {
    if (!topoOrder_.empty()) {
      topoOrder_[0]->GetParentGraph()->FreeAllMemory(launch_stream);
+      topoOrder_[0]->GetParentGraph()->memalloc_nodes_ = 0;
    }
  }