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

@@ -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);
       }