xref: /aosp_15_r20/external/mesa3d/src/panfrost/vulkan/panvk_buffer.c (revision 6104692788411f58d303aa86923a9ff6ecaded22)
1 /*
2  * Copyright © 2021 Collabora Ltd.
3  * SPDX-License-Identifier: MIT
4  */
5 
6 #include "panvk_buffer.h"
7 #include "panvk_device.h"
8 #include "panvk_device_memory.h"
9 #include "panvk_entrypoints.h"
10 
11 #include "vk_log.h"
12 
13 #define PANVK_MAX_BUFFER_SIZE (1 << 30)
14 
15 VKAPI_ATTR VkDeviceAddress VKAPI_CALL
panvk_GetBufferDeviceAddress(VkDevice _device,const VkBufferDeviceAddressInfo * pInfo)16 panvk_GetBufferDeviceAddress(VkDevice _device,
17                              const VkBufferDeviceAddressInfo *pInfo)
18 {
19    VK_FROM_HANDLE(panvk_buffer, buffer, pInfo->buffer);
20 
21    return buffer->dev_addr;
22 }
23 
24 VKAPI_ATTR uint64_t VKAPI_CALL
panvk_GetBufferOpaqueCaptureAddress(VkDevice _device,const VkBufferDeviceAddressInfo * pInfo)25 panvk_GetBufferOpaqueCaptureAddress(VkDevice _device,
26                                     const VkBufferDeviceAddressInfo *pInfo)
27 {
28    return panvk_GetBufferDeviceAddress(_device, pInfo);
29 }
30 
31 VKAPI_ATTR void VKAPI_CALL
panvk_GetBufferMemoryRequirements2(VkDevice device,const VkBufferMemoryRequirementsInfo2 * pInfo,VkMemoryRequirements2 * pMemoryRequirements)32 panvk_GetBufferMemoryRequirements2(VkDevice device,
33                                    const VkBufferMemoryRequirementsInfo2 *pInfo,
34                                    VkMemoryRequirements2 *pMemoryRequirements)
35 {
36    VK_FROM_HANDLE(panvk_buffer, buffer, pInfo->buffer);
37 
38    const uint64_t align = 64;
39    const uint64_t size = align64(buffer->vk.size, align);
40 
41    pMemoryRequirements->memoryRequirements.memoryTypeBits = 1;
42    pMemoryRequirements->memoryRequirements.alignment = align;
43    pMemoryRequirements->memoryRequirements.size = size;
44 }
45 
46 VKAPI_ATTR VkResult VKAPI_CALL
panvk_BindBufferMemory2(VkDevice device,uint32_t bindInfoCount,const VkBindBufferMemoryInfo * pBindInfos)47 panvk_BindBufferMemory2(VkDevice device, uint32_t bindInfoCount,
48                         const VkBindBufferMemoryInfo *pBindInfos)
49 {
50    for (uint32_t i = 0; i < bindInfoCount; ++i) {
51       VK_FROM_HANDLE(panvk_device_memory, mem, pBindInfos[i].memory);
52       VK_FROM_HANDLE(panvk_buffer, buffer, pBindInfos[i].buffer);
53       struct pan_kmod_bo *old_bo = buffer->bo;
54 
55       assert(mem != NULL);
56 
57       buffer->bo = pan_kmod_bo_get(mem->bo);
58       buffer->dev_addr = mem->addr.dev + pBindInfos[i].memoryOffset;
59 
60       /* FIXME: Only host map for index buffers so we can do the min/max
61        * index retrieval on the CPU. This is all broken anyway and the
62        * min/max search should be done with a compute shader that also
63        * patches the job descriptor accordingly (basically an indirect draw).
64        *
65        * Make sure this goes away as soon as we fixed indirect draws.
66        */
67       if (buffer->vk.usage & VK_BUFFER_USAGE_INDEX_BUFFER_BIT) {
68          VkDeviceSize offset = pBindInfos[i].memoryOffset;
69          VkDeviceSize pgsize = getpagesize();
70          off_t map_start = offset & ~(pgsize - 1);
71          off_t map_end = offset + buffer->vk.size;
72          void *map_addr =
73             pan_kmod_bo_mmap(mem->bo, map_start, map_end - map_start,
74                              PROT_WRITE, MAP_SHARED, NULL);
75 
76          assert(map_addr != MAP_FAILED);
77          buffer->host_ptr = map_addr + (offset & pgsize);
78       }
79 
80       pan_kmod_bo_put(old_bo);
81    }
82    return VK_SUCCESS;
83 }
84 
85 VKAPI_ATTR VkResult VKAPI_CALL
panvk_CreateBuffer(VkDevice _device,const VkBufferCreateInfo * pCreateInfo,const VkAllocationCallbacks * pAllocator,VkBuffer * pBuffer)86 panvk_CreateBuffer(VkDevice _device, const VkBufferCreateInfo *pCreateInfo,
87                    const VkAllocationCallbacks *pAllocator, VkBuffer *pBuffer)
88 {
89    VK_FROM_HANDLE(panvk_device, device, _device);
90    struct panvk_buffer *buffer;
91 
92    assert(pCreateInfo->sType == VK_STRUCTURE_TYPE_BUFFER_CREATE_INFO);
93 
94    if (pCreateInfo->size > PANVK_MAX_BUFFER_SIZE)
95       return vk_error(device, VK_ERROR_OUT_OF_DEVICE_MEMORY);
96 
97    buffer =
98       vk_buffer_create(&device->vk, pCreateInfo, pAllocator, sizeof(*buffer));
99    if (buffer == NULL)
100       return vk_error(device, VK_ERROR_OUT_OF_HOST_MEMORY);
101 
102    *pBuffer = panvk_buffer_to_handle(buffer);
103 
104    return VK_SUCCESS;
105 }
106 
107 VKAPI_ATTR void VKAPI_CALL
panvk_DestroyBuffer(VkDevice _device,VkBuffer _buffer,const VkAllocationCallbacks * pAllocator)108 panvk_DestroyBuffer(VkDevice _device, VkBuffer _buffer,
109                     const VkAllocationCallbacks *pAllocator)
110 {
111    VK_FROM_HANDLE(panvk_device, device, _device);
112    VK_FROM_HANDLE(panvk_buffer, buffer, _buffer);
113 
114    if (!buffer)
115       return;
116 
117    if (buffer->host_ptr) {
118       VkDeviceSize pgsize = getpagesize();
119       uintptr_t map_start = (uintptr_t)buffer->host_ptr & ~(pgsize - 1);
120       uintptr_t map_end =
121          ALIGN_POT((uintptr_t)buffer->host_ptr + buffer->vk.size, pgsize);
122       ASSERTED int ret = os_munmap((void *)map_start, map_end - map_start);
123 
124       assert(!ret);
125       buffer->host_ptr = NULL;
126    }
127 
128    pan_kmod_bo_put(buffer->bo);
129    vk_buffer_destroy(&device->vk, pAllocator, &buffer->vk);
130 }
131