From 2d2d33dd9c2355214d9d7704d6962bc2dd64f1ba Mon Sep 17 00:00:00 2001
From: German Andryeyev <German.Andryeyev@amd.com>
Date: Wed, 27 Oct 2021 17:13:52 -0400
Subject: [PATCH] SWDEV-303567 - reset kernel arg buffer to chunk 0

Change-Id: I2974e31af9700705554b0f274ede6f8b9a9d6e7b


[ROCm/clr commit: 30bba18a060d6fff53b98984d85cd66c9292c0a9]
---
 projects/clr/rocclr/device/rocm/rocvirtual.hpp | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/projects/clr/rocclr/device/rocm/rocvirtual.hpp b/projects/clr/rocclr/device/rocm/rocvirtual.hpp
index f80e75d207..30cb1c8016 100644
--- a/projects/clr/rocclr/device/rocm/rocvirtual.hpp
+++ b/projects/clr/rocclr/device/rocm/rocvirtual.hpp
@@ -401,7 +401,11 @@ class VirtualGPU : public device::VirtualDevice {
   void destroyPool();
 
   void* allocKernArg(size_t size, size_t alignment);
-  void resetKernArgPool() { kernarg_pool_cur_offset_ = 0; }
+  void resetKernArgPool() {
+    kernarg_pool_cur_offset_ = 0;
+    kernarg_pool_chunk_end_ = kernarg_pool_size_ / KernelArgPoolNumSignal;
+    active_chunk_ = 0;
+  }
 
   uint64_t getVQVirtualAddress();