AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source 
Change-Id: Ib32b04584548a46632606ecd85b58c6ce4a5894d
This commit is contained in:
Jaydeep Patel
2024-11-11 11:02:56 +00:00
parent cc25c5d646
commit 24c57cb984
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
hipamd/src/hip_graph_internal.cpp
+1
View File
@@ -832,6 +832,7 @@ hipError_t GraphExec::Run(hipStream_t graph_launch_stream) {
if (flags_ & hipGraphInstantiateFlagAutoFreeOnLaunch) { if (flags_ & hipGraphInstantiateFlagAutoFreeOnLaunch) {
if (!topoOrder_.empty()) { if (!topoOrder_.empty()) {
topoOrder_[0]->GetParentGraph()->FreeAllMemory(launch_stream); topoOrder_[0]->GetParentGraph()->FreeAllMemory(launch_stream);
topoOrder_[0]->GetParentGraph()->memalloc_nodes_ = 0;
} }
} }