From 9157087d4393302c78daad13694c2965dbfc6784 Mon Sep 17 00:00:00 2001
From: Samuel Pitoiset <samuel.pitoiset@gmail.com>
Date: Fri, 19 Jan 2024 14:18:08 +0100
Subject: [PATCH] radv: determine next stage for mesh/task with ESO

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 18b67499a88..475de93ce71 100644
--- a/src/amd/vulkan/radv_shader_object.c
+++ b/src/amd/vulkan/radv_shader_object.c
@@ -471,6 +471,7 @@ radv_shader_object_create_linked(VkDevice _device, uint32_t createInfoCount, con
          }
          break;
       case MESA_SHADER_GEOMETRY:
+      case MESA_SHADER_MESH:
          if (stages[MESA_SHADER_FRAGMENT].entrypoint) {
             stages[i].next_stage = MESA_SHADER_FRAGMENT;
          }
@@ -478,6 +479,9 @@ radv_shader_object_create_linked(VkDevice _device, uint32_t createInfoCount, con
       case MESA_SHADER_FRAGMENT:
          stages[i].next_stage = MESA_SHADER_NONE;
          break;
+      case MESA_SHADER_TASK:
+         stages[i].next_stage = MESA_SHADER_MESH;
+         break;
       default:
          assert(0);
       }