From 9b5ea351583b3192392f1b48112c0eb92c4805cf Mon Sep 17 00:00:00 2001
From: Samuel Pitoiset <samuel.pitoiset@gmail.com>
Date: Fri, 19 Jan 2024 16:54:10 +0100
Subject: [PATCH] radv: add support for VK_SHADER_CREATE_NO_TASK_SHADER_BIT_EXT

Signed-off-by: Samuel Pitoiset <samuel.pitoiset@gmail.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/27237>
---
 src/amd/vulkan/radv_shader_object.c | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/amd/vulkan/radv_shader_object.c b/src/amd/vulkan/radv_shader_object.c
index bfa0a363913..dc2169c8460 100644
--- a/src/amd/vulkan/radv_shader_object.c
+++ b/src/amd/vulkan/radv_shader_object.c
@@ -99,6 +99,10 @@ radv_shader_stage_init(const VkShaderCreateInfoEXT *sinfo, struct radv_shader_st
    if (sinfo->flags & VK_SHADER_CREATE_REQUIRE_FULL_SUBGROUPS_BIT_EXT) {
       out_stage->key.subgroup_require_full = 1;
    }
+
+   if (out_stage->stage == MESA_SHADER_MESH) {
+      out_stage->key.has_task_shader = !(sinfo->flags & VK_SHADER_CREATE_NO_TASK_SHADER_BIT_EXT);
+   }
 }
 
 static VkResult