Module: Mesa Branch: main Commit: a8c1f10294db954ce60964dadb9f219b2dee95a3 URL: http://cgit.freedesktop.org/mesa/mesa/commit/?id=a8c1f10294db954ce60964dadb9f219b2dee95a3
Author: Timur Kristóf <[email protected]> Date: Mon Jan 24 11:33:55 2022 +0100 radv: Add task ring entry argument for mesh shaders. Signed-off-by: Timur Kristóf <[email protected]> Reviewed-by: Rhys Perry <[email protected]> Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/14929> --- src/amd/vulkan/radv_shader_args.c | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/src/amd/vulkan/radv_shader_args.c b/src/amd/vulkan/radv_shader_args.c index 084c779934b..0326ec45509 100644 --- a/src/amd/vulkan/radv_shader_args.c +++ b/src/amd/vulkan/radv_shader_args.c @@ -97,6 +97,8 @@ count_ms_user_sgprs(const struct radv_shader_info *info) if (info->vs.needs_draw_id) count++; + if (info->cs.uses_task_rings) + count++; return count; } @@ -380,6 +382,9 @@ declare_ms_input_sgprs(const struct radv_shader_info *info, struct radv_shader_a if (info->vs.needs_draw_id) { ac_add_arg(&args->ac, AC_ARG_SGPR, 1, AC_ARG_INT, &args->ac.draw_id); } + if (info->cs.uses_task_rings) { + ac_add_arg(&args->ac, AC_ARG_SGPR, 1, AC_ARG_INT, &args->ac.task_ring_entry); + } } static void @@ -507,6 +512,9 @@ set_ms_input_locs(struct radv_shader_args *args, uint8_t *user_sgpr_idx) unsigned vs_num = args->ac.base_vertex.used + 3 * args->ac.num_work_groups.used + args->ac.draw_id.used; set_loc_shader(args, AC_UD_VS_BASE_VERTEX_START_INSTANCE, user_sgpr_idx, vs_num); + + if (args->ac.task_ring_entry.used) + set_loc_shader(args, AC_UD_TASK_RING_ENTRY, user_sgpr_idx, 1); } void
