From b4497e54be0870dfadc93ae2b6a101a9065ed485 Mon Sep 17 00:00:00 2001
From: Faith Ekstrand <faith.ekstrand@collabora.com>
Date: Mon, 20 Mar 2023 17:02:36 -0500
Subject: [PATCH] vulkan: Provide wrappers for VK_EXT_map_memory2 functions
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Reviewed-by: Samuel Pitoiset <samuel.pitoiset@gmail.com>
Reviewed-by: Iván Briano <ivan.briano@intel.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/22031>
---
 src/vulkan/runtime/vk_device.c | 37 ++++++++++++++++++++++++++++++++++
 1 file changed, 37 insertions(+)

diff --git a/src/vulkan/runtime/vk_device.c b/src/vulkan/runtime/vk_device.c
index 3a183746cb6..9d7b92be232 100644
--- a/src/vulkan/runtime/vk_device.c
+++ b/src/vulkan/runtime/vk_device.c
@@ -387,6 +387,43 @@ vk_common_GetDeviceQueue2(VkDevice _device,
       *pQueue = VK_NULL_HANDLE;
 }
 
+VKAPI_ATTR VkResult VKAPI_CALL
+vk_common_MapMemory(VkDevice _device,
+                    VkDeviceMemory memory,
+                    VkDeviceSize offset,
+                    VkDeviceSize size,
+                    VkMemoryMapFlags flags,
+                    void **ppData)
+{
+   VK_FROM_HANDLE(vk_device, device, _device);
+
+   const VkMemoryMapInfoKHR info = {
+      .sType = VK_STRUCTURE_TYPE_MEMORY_MAP_INFO_KHR,
+      .flags = flags,
+      .memory = memory,
+      .offset = offset,
+      .size = size,
+   };
+
+   return device->dispatch_table.MapMemory2KHR(_device, &info, ppData);
+}
+
+VKAPI_ATTR void VKAPI_CALL
+vk_common_UnmapMemory(VkDevice _device,
+                      VkDeviceMemory memory)
+{
+   VK_FROM_HANDLE(vk_device, device, _device);
+   ASSERTED VkResult result;
+
+   const VkMemoryUnmapInfoKHR info = {
+      .sType = VK_STRUCTURE_TYPE_MEMORY_UNMAP_INFO_KHR,
+      .memory = memory,
+   };
+
+   result = device->dispatch_table.UnmapMemory2KHR(_device, &info);
+   assert(result == VK_SUCCESS);
+}
+
 VKAPI_ATTR void VKAPI_CALL
 vk_common_GetDeviceGroupPeerMemoryFeatures(
    VkDevice device,