From 853b293983edd9921e75d3eacccf76da4c9a5afc Mon Sep 17 00:00:00 2001
From: Ioannis Assiouras <Ioannis.Assiouras@amd.com>
Date: Mon, 22 Jan 2024 16:58:12 +0000
Subject: [PATCH] SWDEV-433312 - Return invalidArgument in cuMemSetAccess for
 hipMemAccessFlagsProtRead

Change-Id: I8e5c460ac310ad32849359720a40fc5131d4f32c
---
 hipamd/src/hip_vm.cpp | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/hipamd/src/hip_vm.cpp b/hipamd/src/hip_vm.cpp
index 2335c426bc..031c001834 100644
--- a/hipamd/src/hip_vm.cpp
+++ b/hipamd/src/hip_vm.cpp
@@ -288,6 +288,10 @@ hipError_t hipMemSetAccess(void* ptr, size_t size, const hipMemAccessDesc* desc,
       HIP_RETURN(hipErrorInvalidValue)
     }
 
+    if (desc[desc_idx].flags == hipMemAccessFlagsProtRead) {
+      HIP_RETURN(hipErrorInvalidValue)
+    }
+
     auto& dev = g_devices[desc[desc_idx].location.id];
     amd::Device::VmmAccess access_flags = static_cast<amd::Device::VmmAccess>(desc[desc_idx].flags);