diff options
-rw-r--r-- | src/gallium/drivers/radeonsi/si_shader_nir.c | 21 |
1 files changed, 11 insertions, 10 deletions
diff --git a/src/gallium/drivers/radeonsi/si_shader_nir.c b/src/gallium/drivers/radeonsi/si_shader_nir.c index 9c0379a997b..ea842452ca0 100644 --- a/src/gallium/drivers/radeonsi/si_shader_nir.c +++ b/src/gallium/drivers/radeonsi/si_shader_nir.c @@ -496,7 +496,6 @@ void si_nir_scan_shader(const struct nir_shader *nir, i = 0; uint64_t processed_inputs = 0; - unsigned num_inputs = 0; nir_foreach_variable(variable, &nir->inputs) { unsigned semantic_name, semantic_index; @@ -516,10 +515,10 @@ void si_nir_scan_shader(const struct nir_shader *nir, * variable->data.driver_location. */ if (nir->info.stage == MESA_SHADER_VERTEX) { - if (glsl_type_is_dual_slot(glsl_without_array(variable->type))) - num_inputs++; + processed_inputs |= 1ull << i; - num_inputs++; + if (glsl_type_is_dual_slot(glsl_without_array(variable->type))) + processed_inputs |= 2ull << i; continue; } @@ -529,7 +528,6 @@ void si_nir_scan_shader(const struct nir_shader *nir, continue; processed_inputs |= ((uint64_t)1 << i); - num_inputs++; tgsi_get_gl_varying_semantic(variable->data.location + j, true, &semantic_name, &semantic_index); @@ -585,11 +583,8 @@ void si_nir_scan_shader(const struct nir_shader *nir, } } - info->num_inputs = num_inputs; - i = 0; uint64_t processed_outputs = 0; - unsigned num_outputs = 0; nir_foreach_variable(variable, &nir->outputs) { unsigned semantic_name, semantic_index; @@ -691,7 +686,6 @@ void si_nir_scan_shader(const struct nir_shader *nir, continue; processed_outputs |= ((uint64_t)1 << i); - num_outputs++; info->output_semantic_name[i] = semantic_name; info->output_semantic_index[i] = semantic_index; @@ -742,7 +736,14 @@ void si_nir_scan_shader(const struct nir_shader *nir, } } - info->num_outputs = num_outputs; + info->num_inputs = util_last_bit64(processed_inputs); + info->num_outputs = util_last_bit64(processed_outputs); + + /* Inputs and outputs can't have holes. If this fails, use + * nir_assign_io_var_locations to re-assign driver_location. + */ + assert(processed_inputs == u_bit_consecutive64(0, info->num_inputs)); + assert(processed_outputs == u_bit_consecutive64(0, info->num_outputs)); struct set *ubo_set = _mesa_set_create(NULL, _mesa_hash_pointer, _mesa_key_pointer_equal); |