From 193c5e387ea0f3c54cfd286a0bf734766f0c1fd0 Mon Sep 17 00:00:00 2001
From: kjayapra-amd <karthik.jayaprakash@amd.com>
Date: Mon, 9 Jan 2023 15:05:03 -0800
Subject: [PATCH] SWDEV-376697 - Pass uncached memory flag when
 hipDeviceMallocUncached memory is requested.

Change-Id: Idc12948047fdb69876e31edf34a8bcda46c9c303
---
 hipamd/src/hip_memory.cpp | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/hipamd/src/hip_memory.cpp b/hipamd/src/hip_memory.cpp
index 3eb2ad1e0e..da7ed30a55 100644
--- a/hipamd/src/hip_memory.cpp
+++ b/hipamd/src/hip_memory.cpp
@@ -527,7 +527,9 @@ hipError_t hipExtMallocWithFlags(void** ptr, size_t sizeBytes, unsigned int flag
   if (flags == hipDeviceMallocDefault) {
     ihipFlags = 0;
   } else if (flags == hipDeviceMallocFinegrained) {
-    ihipFlags = CL_MEM_SVM_ATOMICS | ROCCLR_MEM_HSA_PSEUDO_FINE_GRAIN;
+    ihipFlags = CL_MEM_SVM_ATOMICS;
+  } else if (flags == hipDeviceMallocUncached) {
+    ihipFlags = CL_MEM_SVM_ATOMICS | ROCCLR_MEM_HSA_UNCACHED;
   } else if (flags == hipMallocSignalMemory) {
     ihipFlags = CL_MEM_SVM_ATOMICS | CL_MEM_SVM_FINE_GRAIN_BUFFER | ROCCLR_MEM_HSA_SIGNAL_MEMORY;
     if (sizeBytes != 8) {