From c6fe6807e8b9edc3aa5d34e06444677b8a3421f3 Mon Sep 17 00:00:00 2001
From: Longlong Yao <Longlong.Yao@amd.com>
Date: Tue, 10 Sep 2024 06:01:18 -0700
Subject: [PATCH] wsl/hsakmt: Set uncached for kernel arg region

Signed-off-by: Longlong Yao <Longlong.Yao@amd.com>
Reviewed-by: Flora Cui <flora.cui@amd.com>
Part-of: <http://10.67.69.192/wsl/libhsakmt/-/merge_requests/5>
---
 memory.cpp | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/memory.cpp b/memory.cpp
index 9d4c1679d8..a244be19e9 100644
--- a/memory.cpp
+++ b/memory.cpp
@@ -152,7 +152,10 @@ HSAKMT_STATUS HSAKMTAPI hsaKmtAllocMemoryAlign(HSAuint32 PreferredNode,
   if (!MemFlags.ui32.CoarseGrain)
     create_info.mem_flags = rocr_proxy::kFineGrain;
 
-  // create_info.mem_flags |= rocr_proxy::kKernarg;
+  //In hsa-runtime, only kernarg region set Uncached.
+  if (MemFlags.ui32.Uncached)
+    create_info.mem_flags |= rocr_proxy::kKernarg;
+
   create_info.flags.physical_only = MemFlags.ui32.NoAddress;
   create_info.flags.interprocess = MemFlags.ui32.NoAddress;
   create_info.flags.locked = MemFlags.ui32.NoSubstitute;//AllocatePinned