spirv: Track when a shader has a cooperative matrix

Reviewed-by: Caio Oliveira <caio.oliveira@intel.com>
Reviewed-by: Bas Nieuwenhuizen <bas@basnieuwenhuizen.nl>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/23825>

diff --git a/src/compiler/shader_info.h b/src/compiler/shader_info.h
index 51000b2d8fc0019b9ffdc9dea659affba0cdbe34..ad2c7afcc8ccb692ffab0ad1cb147cba9a71b0cd 100644 (file)
--- a/src/compiler/shader_info.h
+++ b/src/compiler/shader_info.h
@@ -524,6 +524,12 @@ typedef struct shader_info {
           */
          bool has_variable_shared_mem:1;
 
+         /**
+          * If the shader has any use of a cooperative matrix. From
+          * SPV_KHR_cooperative_matrix.
+          */
+         bool has_cooperative_matrix:1;
+
          /**
           * pointer size is:
           *   AddressingModelLogical:    0    (default)

diff --git a/src/compiler/spirv/vtn_cmat.c b/src/compiler/spirv/vtn_cmat.c
index 2e2898429bcbf75767ba4e736b4827ac42faf1f6..e902b219a6a8dd19ecdd3816c96b091460ba992c 100644 (file)
--- a/src/compiler/spirv/vtn_cmat.c
+++ b/src/compiler/spirv/vtn_cmat.c
@@ -29,6 +29,8 @@ vtn_handle_cooperative_type(struct vtn_builder *b, struct vtn_value *val,
 {
    vtn_assert(opcode == SpvOpTypeCooperativeMatrixKHR);
 
+   b->shader->info.cs.has_cooperative_matrix = true;
+
    struct vtn_type *component_type = vtn_get_type(b, w[2]);
 
    const mesa_scope scope = vtn_translate_scope(b, vtn_constant_uint(b, w[3]));