From 367d211aa29782aec425f2f54cfe916b91acd081 Mon Sep 17 00:00:00 2001 From: Jaydeep Patel Date: Tue, 7 Nov 2023 08:21:31 +0000 Subject: [PATCH] SWDEV-430966 - Size to unmap should be same as mapped memory size. Change-Id: Id59990c655e7bb4bb980efe587275296103e1e73 [ROCm/clr commit: 2415701080406cd8faece0e4db48f7422e78be27] --- projects/clr/hipamd/src/hip_vm.cpp | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/projects/clr/hipamd/src/hip_vm.cpp b/projects/clr/hipamd/src/hip_vm.cpp index 249f224ede..d68dc2501c 100644 --- a/projects/clr/hipamd/src/hip_vm.cpp +++ b/projects/clr/hipamd/src/hip_vm.cpp @@ -305,6 +305,9 @@ hipError_t hipMemUnmap(void* ptr, size_t size) { } amd::Memory* va = amd::MemObjMap::FindMemObj(ptr); + if (va && va->getSize() != size) { + HIP_RETURN(hipErrorInvalidValue); + } auto& queue = *g_devices[va->getUserData().deviceId]->NullStream(); amd::Command* cmd = new amd::VirtualMapCommand(queue, amd::Command::EventWaitList{}, ptr, size, @@ -320,4 +323,4 @@ hipError_t hipMemUnmap(void* ptr, size_t size) { ga->release(); HIP_RETURN(hipSuccess); -} \ No newline at end of file +}