From 5ee300da460b6d45d8717063ad5fa1b51bca454a Mon Sep 17 00:00:00 2001
From: kjayapra-amd <karthik.jayaprakash@amd.com>
Date: Fri, 26 Feb 2021 15:39:53 -0500
Subject: [PATCH] SWDEV-270013 - Assign Kern_args segment only for gfx90a.

Change-Id: Icc890041eb27800cb78ef9e3d7bbe02658f53821


[ROCm/clr commit: a7abe44436193f790a3964d9ee86a1eb9c5ffa92]
---
 projects/clr/rocclr/device/rocm/rocdevice.cpp | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/projects/clr/rocclr/device/rocm/rocdevice.cpp b/projects/clr/rocclr/device/rocm/rocdevice.cpp
index 85b1877ec6..c660a54796 100644
--- a/projects/clr/rocclr/device/rocm/rocdevice.cpp
+++ b/projects/clr/rocclr/device/rocm/rocdevice.cpp
@@ -1859,7 +1859,13 @@ void* Device::hostAlloc(size_t size, size_t alignment, MemorySegment mem_seg) co
 
   hsa_amd_memory_pool_t segment{0};
   switch (mem_seg) {
-    case kKernArg :
+    case kKernArg : {
+      if (::strcmp(isa().processorName().c_str(), "gfx90a") == 0) {
+        segment = system_kernarg_segment_;
+        break;
+      }
+      // Falls through on else case.
+    }
     case kNoAtomics :
       // If runtime disables barrier, then all host allocations must have L2 disabled
       if ((settings().barrier_sync_) && (system_coarse_segment_.handle != 0)) {