From 02287fe42dcf4514449c67f0036e5cea3ebf213f Mon Sep 17 00:00:00 2001
From: Mike Blumenkrantz <michael.blumenkrantz@gmail.com>
Date: Fri, 14 May 2021 18:17:23 -0400
Subject: [PATCH] zink: flag shader modules as default

Reviewed-by: Dave Airlie <airlied@redhat.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/11740>
---
 src/gallium/drivers/zink/zink_program.c | 1 +
 src/gallium/drivers/zink/zink_program.h | 1 +
 2 files changed, 2 insertions(+)

diff --git a/src/gallium/drivers/zink/zink_program.c b/src/gallium/drivers/zink/zink_program.c
index 798b53488e6..52d85f987d6 100644
--- 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 7548b45f213..a406002288b 100644
--- 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 {