radv: add set_loc_shader_ptr() helper

[mesa.git] / src / amd / vulkan / radv_shader_info.c

diff --git a/src/amd/vulkan/radv_shader_info.c b/src/amd/vulkan/radv_shader_info.c
index a436bd753430a525f38b972b6352f7f3616b5d54..b45b4c0c95b723be30fe5341b97748db28aa1868 100644 (file)
--- a/src/amd/vulkan/radv_shader_info.c
+++ b/src/amd/vulkan/radv_shader_info.c
@@ -134,6 +134,12 @@ gather_intrinsic_store_var_info(const nir_shader *nir,
                                        instr->const_index[0] << comp;
                        }
                        break;
+               case MESA_SHADER_GEOMETRY:
+                       for (unsigned i = 0; i < attrib_count; i++) {
+                               info->gs.output_usage_mask[idx + i + const_offset] |=
+                                       instr->const_index[0] << comp;
+                       }
+                       break;
                case MESA_SHADER_TESS_EVAL:
                        for (unsigned i = 0; i < attrib_count; i++) {
                                info->tes.output_usage_mask[idx + i + const_offset] |=