From da132a2e2841f1b7a52ceb94d57d592f99f0275b Mon Sep 17 00:00:00 2001
From: Anusha GodavarthySurya <anusha.godavarthysurya@amd.com>
Date: Thu, 18 Jul 2024 10:58:08 +0000
Subject: [PATCH] SWDEV-468424 - Fix kernelArg mgr release and clear commands
 after capture

Creation of ReferenceCountedObject will increase reference count by 1.
Clear the commands from Node after capture so that they wont be reference later.

Change-Id: I1cc4085939cf65218ec2aa2e25ab6d737f7cacd3


[ROCm/clr commit: 6ae5d6896cd10bf287afb6ed68fc2cbdde247cd5]
---
 projects/clr/hipamd/src/hip_graph_internal.hpp | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/projects/clr/hipamd/src/hip_graph_internal.hpp b/projects/clr/hipamd/src/hip_graph_internal.hpp
index 171bfb24da..eee2c45811 100644
--- a/projects/clr/hipamd/src/hip_graph_internal.hpp
+++ b/projects/clr/hipamd/src/hip_graph_internal.hpp
@@ -252,6 +252,8 @@ struct GraphNode : public hipGraphNodeDOTAttribute {
       command->submit(*(command->queue())->vdev());
       command->release();
     }
+    // Commands are captured and released. Clear them from the object.
+    commands_.clear();
   }
   hip::Stream* GetQueue() const { return stream_; }
 
@@ -704,7 +706,6 @@ struct GraphExec : public amd::ReferenceCountedObject {
   // child graph nodes and each child graph has only one node.
   void SetKernelArgManager(GraphKernelArgManager* kernArgManager) {
     kernArgManager_ = kernArgManager;
-    kernArgManager_->retain();
   }
   GraphKernelArgManager* GetKernelArgManager() {
     return kernArgManager_;