- for (unsigned stage = 0; stage < MESA_SHADER_STAGES; stage++) {
- struct gl_linked_shader *sh = prog->_LinkedShaders[stage];
- if (!sh)
- continue;
-
- struct hash_entry *entry =
- _mesa_hash_table_search(referenced_uniforms[stage], var);
- if (entry) {
- struct uniform_array_info *ainfo =
- (struct uniform_array_info *) entry->data;
- util_dynarray_foreach(ainfo->deref_list, nir_deref_instr *, deref) {
- (*deref)->type = var->type;
- }
+ struct hash_entry *entry =
+ _mesa_hash_table_search(referenced_uniforms[current_var_stage], var->name);
+ if (entry) {
+ struct uniform_array_info *ainfo =
+ (struct uniform_array_info *) entry->data;
+ util_dynarray_foreach(ainfo->deref_list, nir_deref_instr *, deref) {
+ (*deref)->type = var->type;