vulkan: Add vk_shader_module_init

This will be used for allocating shader modules using ralloc by RADV.

Reviewed-by: Samuel Pitoiset <samuel.pitoiset@gmail.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/21707>
This commit is contained in:
Konstantin Seurer
2023-03-04 17:17:13 +01:00
committed by Marge Bot
parent 0fc8335ccb
commit ef5cba56a0
2 changed files with 23 additions and 8 deletions

View File

@@ -24,6 +24,7 @@
#include "vk_shader_module.h"
#include "util/mesa-sha1.h"
#include "vk_alloc.h"
#include "vk_common_entrypoints.h"
#include "vk_device.h"
#include "vk_log.h"
@@ -31,6 +32,20 @@
#include "vk_pipeline.h"
#include "vk_util.h"
void vk_shader_module_init(struct vk_device *device,
struct vk_shader_module *module,
const VkShaderModuleCreateInfo *create_info)
{
vk_object_base_init(device, &module->base, VK_OBJECT_TYPE_SHADER_MODULE);
module->nir = NULL;
module->size = create_info->codeSize;
memcpy(module->data, create_info->pCode, module->size);
_mesa_sha1_compute(module->data, module->size, module->sha1);
}
VKAPI_ATTR VkResult VKAPI_CALL
vk_common_CreateShaderModule(VkDevice _device,
const VkShaderModuleCreateInfo *pCreateInfo,
@@ -43,17 +58,13 @@ vk_common_CreateShaderModule(VkDevice _device,
assert(pCreateInfo->sType == VK_STRUCTURE_TYPE_SHADER_MODULE_CREATE_INFO);
assert(pCreateInfo->flags == 0);
module = vk_object_alloc(device, pAllocator,
sizeof(*module) + pCreateInfo->codeSize,
VK_OBJECT_TYPE_SHADER_MODULE);
module = vk_alloc2(&device->alloc, pAllocator,
sizeof(*module) + pCreateInfo->codeSize, 8,
VK_SYSTEM_ALLOCATION_SCOPE_OBJECT);
if (module == NULL)
return VK_ERROR_OUT_OF_HOST_MEMORY;
module->size = pCreateInfo->codeSize;
module->nir = NULL;
memcpy(module->data, pCreateInfo->pCode, module->size);
_mesa_sha1_compute(module->data, module->size, module->sha1);
vk_shader_module_init(device, module, pCreateInfo);
*pShaderModule = vk_shader_module_to_handle(module);

View File

@@ -50,6 +50,10 @@ extern const uint8_t vk_shaderModuleIdentifierAlgorithmUUID[VK_UUID_SIZE];
VK_DEFINE_NONDISP_HANDLE_CASTS(vk_shader_module, base, VkShaderModule,
VK_OBJECT_TYPE_SHADER_MODULE)
void vk_shader_module_init(struct vk_device *device,
struct vk_shader_module *module,
const VkShaderModuleCreateInfo *create_info);
uint32_t vk_shader_module_spirv_version(const struct vk_shader_module *mod);
VkResult