vk/cmd_queue: generate CmdPushDescriptorSet

Reviewed-by: Konstantin Seurer <konstantin.seurer@gmail.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/40268>
This commit is contained in:
Mike Blumenkrantz 2026-03-06 10:38:26 -05:00 committed by Marge Bot
parent 5f8b244ce6
commit dbbd5360f0
2 changed files with 52 additions and 84 deletions

View file

@ -317,82 +317,6 @@ vk_cmd_enqueue_CmdDrawMultiIndexedEXT(VkCommandBuffer commandBuffer,
}
}
VKAPI_ATTR void VKAPI_CALL
vk_cmd_enqueue_CmdPushDescriptorSet(VkCommandBuffer commandBuffer,
VkPipelineBindPoint pipelineBindPoint,
VkPipelineLayout layout,
uint32_t set,
uint32_t descriptorWriteCount,
const VkWriteDescriptorSet *pDescriptorWrites)
{
VK_FROM_HANDLE(vk_command_buffer, cmd_buffer, commandBuffer);
struct vk_cmd_push_descriptor_set *pds;
struct vk_cmd_queue_entry *cmd =
linear_zalloc_child(cmd_buffer->cmd_queue.ctx, sizeof(*cmd));
if (!cmd)
return;
pds = &cmd->u.push_descriptor_set;
cmd->type = VK_CMD_PUSH_DESCRIPTOR_SET;
list_addtail(&cmd->cmd_link, &cmd_buffer->cmd_queue.cmds);
pds->pipeline_bind_point = pipelineBindPoint;
pds->set = set;
pds->descriptor_write_count = descriptorWriteCount;
pds->layout = layout;
enqueue_pipeline_layout(&cmd_buffer->cmd_queue, layout);
if (pDescriptorWrites) {
pds->descriptor_writes =
linear_alloc_child(cmd_buffer->cmd_queue.ctx,
sizeof(*pds->descriptor_writes) * descriptorWriteCount);
memcpy(pds->descriptor_writes,
pDescriptorWrites,
sizeof(*pds->descriptor_writes) * descriptorWriteCount);
for (unsigned i = 0; i < descriptorWriteCount; i++) {
switch (pds->descriptor_writes[i].descriptorType) {
case VK_DESCRIPTOR_TYPE_SAMPLER:
case VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER:
case VK_DESCRIPTOR_TYPE_SAMPLED_IMAGE:
case VK_DESCRIPTOR_TYPE_STORAGE_IMAGE:
case VK_DESCRIPTOR_TYPE_INPUT_ATTACHMENT:
pds->descriptor_writes[i].pImageInfo =
linear_alloc_child(cmd_buffer->cmd_queue.ctx,
sizeof(VkDescriptorImageInfo) * pds->descriptor_writes[i].descriptorCount);
memcpy((VkDescriptorImageInfo *)pds->descriptor_writes[i].pImageInfo,
pDescriptorWrites[i].pImageInfo,
sizeof(VkDescriptorImageInfo) * pds->descriptor_writes[i].descriptorCount);
break;
case VK_DESCRIPTOR_TYPE_UNIFORM_TEXEL_BUFFER:
case VK_DESCRIPTOR_TYPE_STORAGE_TEXEL_BUFFER:
pds->descriptor_writes[i].pTexelBufferView =
linear_alloc_child(cmd_buffer->cmd_queue.ctx,
sizeof(VkBufferView) * pds->descriptor_writes[i].descriptorCount);
memcpy((VkBufferView *)pds->descriptor_writes[i].pTexelBufferView,
pDescriptorWrites[i].pTexelBufferView,
sizeof(VkBufferView) * pds->descriptor_writes[i].descriptorCount);
break;
case VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER:
case VK_DESCRIPTOR_TYPE_STORAGE_BUFFER:
case VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER_DYNAMIC:
case VK_DESCRIPTOR_TYPE_STORAGE_BUFFER_DYNAMIC:
default:
pds->descriptor_writes[i].pBufferInfo =
linear_zalloc_child(cmd_buffer->cmd_queue.ctx,
sizeof(VkDescriptorBufferInfo) * pds->descriptor_writes[i].descriptorCount);
memcpy((VkDescriptorBufferInfo *)pds->descriptor_writes[i].pBufferInfo,
pDescriptorWrites[i].pBufferInfo,
sizeof(VkDescriptorBufferInfo) * pds->descriptor_writes[i].descriptorCount);
break;
}
}
}
}
#ifdef VK_ENABLE_BETA_EXTENSIONS
VKAPI_ATTR void VKAPI_CALL

View file

@ -39,9 +39,6 @@ from vk_extensions import filter_api, get_all_required
# These have hand-typed implementations in vk_cmd_enqueue.c
MANUAL_COMMANDS = [
# This script doesn't know how to copy arrays in structs in arrays
'CmdPushDescriptorSet',
# The size of the elements is specified in a stride param
'CmdDrawMultiEXT',
'CmdDrawMultiIndexedEXT',
@ -274,6 +271,42 @@ enqueue_pipeline_layout(struct vk_cmd_queue *queue, VkPipelineLayout layout)
util_dynarray_append(&queue->pipeline_layouts, vklayout);
}
static void
enqueue_VkWriteDescriptorSet(struct vk_cmd_queue *queue, VkWriteDescriptorSet *dst, const VkWriteDescriptorSet *src)
{
switch (dst->descriptorType) {
case VK_DESCRIPTOR_TYPE_SAMPLER:
case VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER:
case VK_DESCRIPTOR_TYPE_SAMPLED_IMAGE:
case VK_DESCRIPTOR_TYPE_STORAGE_IMAGE:
case VK_DESCRIPTOR_TYPE_INPUT_ATTACHMENT:
dst->pImageInfo = linear_alloc_child(queue->ctx, sizeof(VkDescriptorImageInfo) * dst->descriptorCount);
memcpy((VkDescriptorImageInfo *)dst->pImageInfo,
src->pImageInfo,
sizeof(VkDescriptorImageInfo) * dst->descriptorCount);
break;
case VK_DESCRIPTOR_TYPE_UNIFORM_TEXEL_BUFFER:
case VK_DESCRIPTOR_TYPE_STORAGE_TEXEL_BUFFER:
dst->pTexelBufferView = linear_alloc_child(queue->ctx, sizeof(VkBufferView) * dst->descriptorCount);
memcpy((VkBufferView *)dst->pTexelBufferView,
src->pTexelBufferView,
sizeof(VkBufferView) * dst->descriptorCount);
break;
case VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER:
case VK_DESCRIPTOR_TYPE_STORAGE_BUFFER:
case VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER_DYNAMIC:
case VK_DESCRIPTOR_TYPE_STORAGE_BUFFER_DYNAMIC:
dst->pBufferInfo = linear_zalloc_child(queue->ctx, sizeof(VkDescriptorBufferInfo) * dst->descriptorCount);
memcpy((VkDescriptorBufferInfo *)dst->pBufferInfo,
src->pBufferInfo,
sizeof(VkDescriptorBufferInfo) * dst->descriptorCount);
break;
default:
break;
}
}
% for c in commands:
% if c.guard is not None:
#ifdef ${c.guard}
@ -371,7 +404,7 @@ vk_cmd_enqueue_${c.name}(${c.decl_params()})
% endif
if (unlikely(!cmd))
vk_command_buffer_set_error(cmd_buffer, VK_ERROR_OUT_OF_HOST_MEMORY);
% if 'CmdBindDescriptorSets' == c.name:
% if 'CmdBindDescriptorSets' == c.name or 'CmdPushDescriptorSet' == c.name:
else
enqueue_pipeline_layout(&cmd_buffer->cmd_queue, layout);
% endif
@ -474,6 +507,10 @@ def categorize_param(types, parent_type, param):
return ParamCategory.STRUCT
EXPLICIT_PARAM_COPIES = [
'VkWriteDescriptorSet',
]
def get_pnext_copy(builder, command, types, parent_type, src, dst):
if not types[parent_type].extended_by:
return
@ -536,7 +573,8 @@ def get_param_copy(builder, command, types, src_parent_access, dst_parent_access
builder.add("memcpy((void *)%s, %s, %s);" % (dst, src, size))
if param.type in types:
needs_member_copy = False
has_explicit_copy = param.type in EXPLICIT_PARAM_COPIES
needs_member_copy = has_explicit_copy
for member in types[param.type].members:
category = categorize_param(types, param.type, member)
if category == ParamCategory.PNEXT or category == ParamCategory.STRUCT or category == ParamCategory.STRING:
@ -563,11 +601,17 @@ def get_param_copy(builder, command, types, src_parent_access, dst_parent_access
for member in types[param.type].members:
category = categorize_param(types, param.type, member)
if category == ParamCategory.STRUCT or category == ParamCategory.STRING:
get_param_copy(builder, command, types, "%s->" % (tmp_src_name), "%s->" % (tmp_dst_name), member, dst_initialized=True)
elif category == ParamCategory.PNEXT:
if category == ParamCategory.PNEXT:
get_pnext_copy(builder, command, types, param.type, "%s->pNext" % (tmp_src_name), "%s->pNext" % (tmp_dst_name))
if has_explicit_copy:
builder.add("enqueue_%s(queue, %s, %s);" % (param.type, tmp_dst_name, tmp_src_name))
else:
for member in types[param.type].members:
category = categorize_param(types, param.type, member)
if category == ParamCategory.STRUCT or category == ParamCategory.STRING:
get_param_copy(builder, command, types, "%s->" % (tmp_src_name), "%s->" % (tmp_dst_name), member, dst_initialized=True)
if struct_array_copy:
builder.level -= 1
builder.add("}")