zink: flag shader modules as default

Reviewed-by: Dave Airlie <airlied@redhat.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/11740>

diff --git a/src/gallium/drivers/zink/zink_program.c b/src/gallium/drivers/zink/zink_program.c
index 798b534..52d85f9 100644 (file)
--- a/src/gallium/drivers/zink/zink_program.c
+++ b/src/gallium/drivers/zink/zink_program.c
@@ -250,6 +250,7 @@ get_shader_module_for_stage(struct zink_context *ctx, struct zink_shader *zs, st
       if (key.is_default_variant) {
          /* previously returned */
          *default_zm = zm;
+         zm->default_variant = true;
          prog->default_variant_key[pstage] = keybox;
       }
    }

diff --git a/src/gallium/drivers/zink/zink_program.h b/src/gallium/drivers/zink/zink_program.h
index 7548b45..a406002 100644 (file)
--- a/src/gallium/drivers/zink/zink_program.h
+++ b/src/gallium/drivers/zink/zink_program.h
@@ -65,6 +65,7 @@ struct zink_cs_push_constant {
  */
 struct zink_shader_module {
    VkShaderModule shader;
+   bool default_variant;
 };
 
 struct zink_program {