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;
     }
   }