radv: do not load the local invocation index when it's unused

[mesa.git] / src / amd / common / ac_shader_info.c

diff --git a/src/amd/common/ac_shader_info.c b/src/amd/common/ac_shader_info.c
index 979b120b73113b51757edc86bfbc008364ccde84..3299b47e6b879f97b7934dd7fa4e947eec3df040 100644 (file)
--- a/src/amd/common/ac_shader_info.c
+++ b/src/amd/common/ac_shader_info.c
@@ -43,7 +43,23 @@ gather_intrinsic_info(nir_intrinsic_instr *instr, struct ac_shader_info *info)
                info->vs.needs_instance_id = true;
                break;
        case nir_intrinsic_load_num_work_groups:
-               info->cs.grid_components_used = instr->num_components;
+               info->cs.uses_grid_size = true;
+               break;
+       case nir_intrinsic_load_local_invocation_id:
+       case nir_intrinsic_load_work_group_id: {
+               unsigned mask = nir_ssa_def_components_read(&instr->dest.ssa);
+               while (mask) {
+                       unsigned i = u_bit_scan(&mask);
+
+                       if (instr->intrinsic == nir_intrinsic_load_work_group_id)
+                               info->cs.uses_block_id[i] = true;
+                       else
+                               info->cs.uses_thread_id[i] = true;
+               }
+               break;
+       }
+       case nir_intrinsic_load_local_invocation_index:
+               info->cs.uses_local_invocation_idx = true;
                break;
        case nir_intrinsic_load_sample_id:
                info->ps.force_persample = true;
@@ -116,7 +132,7 @@ gather_info_input_decl(nir_shader *nir,
                       nir_variable *var,
                       struct ac_shader_info *info)
 {
-       switch (nir->stage) {
+       switch (nir->info.stage) {
        case MESA_SHADER_VERTEX:
                info->vs.has_vertex_buffers = true;
                break;