SWDEV-270013 - Assign Kern_args segment only for gfx90a.

Change-Id: Icc890041eb27800cb78ef9e3d7bbe02658f53821 [ROCm/clr commit: a7abe44436]
2021-02-26 15:39:53 -05:00
parent 3c836ff64c
commit 5ee300da46
1 changed files with 7 additions and 1 deletions
@@ -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)) {