From: Mike Blumenkrantz <michael.blumenkrantz@gmail.com>
Date: Mon, 17 Oct 2022 14:11:08 +0000 (-0400)
Subject: zink: pass KERNEL shaders through successfully
X-Git-Tag: upstream/23.3.3~2778
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=939845e424da7b2433902a9954cb6b615cd06d41;p=platform%2Fupstream%2Fmesa.git

zink: pass KERNEL shaders through successfully

Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/24868>
---

diff --git a/src/gallium/drivers/zink/zink_compiler.c b/src/gallium/drivers/zink/zink_compiler.c
index a29a0a0..fab15ab 100644
--- a/src/gallium/drivers/zink/zink_compiler.c
+++ b/src/gallium/drivers/zink/zink_compiler.c
@@ -3052,6 +3052,7 @@ zink_get_next_stage(gl_shader_stage stage)
       return VK_SHADER_STAGE_FRAGMENT_BIT;
    case MESA_SHADER_FRAGMENT:
    case MESA_SHADER_COMPUTE:
+   case MESA_SHADER_KERNEL:
       return 0;
    default:
       unreachable("invalid shader stage");
@@ -5340,7 +5341,7 @@ zink_shader_create(struct zink_screen *screen, struct nir_shader *nir)
          } else if (var->data.mode == nir_var_mem_ssbo) {
             ztype = ZINK_DESCRIPTOR_TYPE_SSBO;
             var->data.descriptor_set = screen->desc_set_id[ztype];
-            var->data.binding = zink_binding(nir->info.stage,
+            var->data.binding = zink_binding(clamp_stage(&nir->info),
                                              VK_DESCRIPTOR_TYPE_STORAGE_BUFFER,
                                              var->data.driver_location,
                                              screen->compact_descriptors);