From 24c57cb984bfcdf2bc1968d1949ac8a276ecbe96 Mon Sep 17 00:00:00 2001
From: Jaydeep Patel <jaydeepkumar.patel@amd.com>
Date: Mon, 11 Nov 2024 11:02:56 +0000
Subject: [PATCH] SWDEV-496544 - Reset mem alloc node count for
 AutoFreeOnLaunch.

Change-Id: Ib32b04584548a46632606ecd85b58c6ce4a5894d
---
 hipamd/src/hip_graph_internal.cpp | 1 +
 1 file changed, 1 insertion(+)

diff --git a/hipamd/src/hip_graph_internal.cpp b/hipamd/src/hip_graph_internal.cpp
index 252d1e7952..64f4c7be77 100644
--- a/hipamd/src/hip_graph_internal.cpp
+++ b/hipamd/src/hip_graph_internal.cpp
@@ -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;
     }
   }