From 5c5555a862754a5b43fee2abf4fc34e888d22a06 Mon Sep 17 00:00:00 2001 From: Jason Ekstrand Date: Wed, 22 Jul 2020 23:37:27 -0500 Subject: [PATCH] nir: Add a find_variable_with_[driver_]location helper We've hand-rolled this loop 10 places and those are just the ones I found easily. Reviewed-by: Eric Anholt Reviewed-by: Iago Toral Quiroga Part-of: --- src/amd/vulkan/radv_shader.c | 12 ++++----- src/compiler/nir/nir.c | 26 +++++++++++++++++++ src/compiler/nir/nir.h | 8 ++++++ src/compiler/nir/nir_lower_bitmap.c | 13 +++------- .../nir/nir_lower_input_attachments.c | 14 +++++----- src/compiler/nir/nir_lower_point_size_mov.c | 10 +++---- src/compiler/nir/nir_lower_two_sided_color.c | 23 ++++++++-------- src/compiler/nir/nir_range_analysis.c | 7 ++--- src/gallium/drivers/iris/iris_program.c | 10 ++----- src/gallium/drivers/vc4/vc4_nir_lower_io.c | 20 +++++--------- .../brw_nir_lower_alpha_to_coverage.c | 24 ++++------------- 11 files changed, 79 insertions(+), 88 deletions(-) diff --git a/src/amd/vulkan/radv_shader.c b/src/amd/vulkan/radv_shader.c index 749167e1723..810d4055186 100644 --- a/src/amd/vulkan/radv_shader.c +++ b/src/amd/vulkan/radv_shader.c @@ -586,14 +586,12 @@ type_size_vec4(const struct glsl_type *type, bool bindless) static nir_variable * find_layer_in_var(nir_shader *nir) { - nir_foreach_shader_in_variable(var, nir) { - if (var->data.location == VARYING_SLOT_LAYER) { - return var; - } - } - nir_variable *var = - nir_variable_create(nir, nir_var_shader_in, glsl_int_type(), "layer id"); + nir_find_variable_with_location(nir, nir_var_shader_in, VARYING_SLOT_LAYER); + if (var != NULL) + return var; + + var = nir_variable_create(nir, nir_var_shader_in, glsl_int_type(), "layer id"); var->data.location = VARYING_SLOT_LAYER; var->data.interpolation = INTERP_MODE_FLAT; return var; diff --git a/src/compiler/nir/nir.c b/src/compiler/nir/nir.c index dc56c4922fc..e17b11cdd4a 100644 --- a/src/compiler/nir/nir.c +++ b/src/compiler/nir/nir.c @@ -171,6 +171,32 @@ nir_local_variable_create(nir_function_impl *impl, return var; } +nir_variable * +nir_find_variable_with_location(nir_shader *shader, + nir_variable_mode mode, + unsigned location) +{ + assert(util_bitcount(mode) == 1 && mode != nir_var_function_temp); + nir_foreach_variable_with_modes(var, shader, mode) { + if (var->data.location == location) + return var; + } + return NULL; +} + +nir_variable * +nir_find_variable_with_driver_location(nir_shader *shader, + nir_variable_mode mode, + unsigned location) +{ + assert(util_bitcount(mode) == 1 && mode != nir_var_function_temp); + nir_foreach_variable_with_modes(var, shader, mode) { + if (var->data.driver_location == location) + return var; + } + return NULL; +} + nir_function * nir_function_create(nir_shader *shader, const char *name) { diff --git a/src/compiler/nir/nir.h b/src/compiler/nir/nir.h index b216d293b3e..52544b9f06c 100644 --- a/src/compiler/nir/nir.h +++ b/src/compiler/nir/nir.h @@ -3308,6 +3308,14 @@ nir_variable *nir_local_variable_create(nir_function_impl *impl, const struct glsl_type *type, const char *name); +nir_variable *nir_find_variable_with_location(nir_shader *shader, + nir_variable_mode mode, + unsigned location); + +nir_variable *nir_find_variable_with_driver_location(nir_shader *shader, + nir_variable_mode mode, + unsigned location); + /** creates a function and adds it to the shader's list of functions */ nir_function *nir_function_create(nir_shader *shader, const char *name); diff --git a/src/compiler/nir/nir_lower_bitmap.c b/src/compiler/nir/nir_lower_bitmap.c index f9173a64796..4d361f5e2df 100644 --- a/src/compiler/nir/nir_lower_bitmap.c +++ b/src/compiler/nir/nir_lower_bitmap.c @@ -55,16 +55,9 @@ static nir_variable * get_texcoord(nir_shader *shader) { - nir_variable *texcoord = NULL; - - /* find gl_TexCoord, if it exists: */ - nir_foreach_shader_in_variable(var, shader) { - if (var->data.location == VARYING_SLOT_TEX0) { - texcoord = var; - break; - } - } - + nir_variable *texcoord = + nir_find_variable_with_location(shader, nir_var_shader_in, + VARYING_SLOT_TEX0); /* otherwise create it: */ if (texcoord == NULL) { texcoord = nir_variable_create(shader, diff --git a/src/compiler/nir/nir_lower_input_attachments.c b/src/compiler/nir/nir_lower_input_attachments.c index 75b9e1faff9..20fa801b45d 100644 --- a/src/compiler/nir/nir_lower_input_attachments.c +++ b/src/compiler/nir/nir_lower_input_attachments.c @@ -27,14 +27,14 @@ static nir_ssa_def * load_frag_coord(nir_builder *b) { - nir_foreach_shader_in_variable(var, b->shader) { - if (var->data.location == VARYING_SLOT_POS) - return nir_load_var(b, var); + nir_variable *pos = + nir_find_variable_with_location(b->shader, nir_var_shader_in, + VARYING_SLOT_POS); + if (pos == NULL) { + pos = nir_variable_create(b->shader, nir_var_shader_in, + glsl_vec4_type(), NULL); + pos->data.location = VARYING_SLOT_POS; } - - nir_variable *pos = nir_variable_create(b->shader, nir_var_shader_in, - glsl_vec4_type(), NULL); - pos->data.location = VARYING_SLOT_POS; /** * From Vulkan spec: * "The OriginLowerLeft execution mode must not be used; fragment entry diff --git a/src/compiler/nir/nir_lower_point_size_mov.c b/src/compiler/nir/nir_lower_point_size_mov.c index efe162c8d8f..f0526ba55fb 100644 --- a/src/compiler/nir/nir_lower_point_size_mov.c +++ b/src/compiler/nir/nir_lower_point_size_mov.c @@ -71,13 +71,9 @@ nir_lower_point_size_mov(nir_shader *shader, assert(shader->info.stage != MESA_SHADER_FRAGMENT && shader->info.stage != MESA_SHADER_COMPUTE); - nir_variable *out = NULL; - nir_foreach_shader_out_variable(var, shader) { - if (var->data.location == VARYING_SLOT_PSIZ) { - out = var; - break; - } - } + nir_variable *out = + nir_find_variable_with_location(shader, nir_var_shader_out, + VARYING_SLOT_PSIZ); lower_impl(nir_shader_get_entrypoint(shader), pointsize_state_tokens, out); diff --git a/src/compiler/nir/nir_lower_two_sided_color.c b/src/compiler/nir/nir_lower_two_sided_color.c index aa0d9fe84b2..0507c4d2317 100644 --- a/src/compiler/nir/nir_lower_two_sided_color.c +++ b/src/compiler/nir/nir_lower_two_sided_color.c @@ -66,19 +66,20 @@ create_input(nir_shader *shader, gl_varying_slot slot, static nir_variable * create_face_input(nir_shader *shader) { - nir_foreach_shader_in_variable(var, shader) { - if (var->data.location == VARYING_SLOT_FACE) - return var; + nir_variable *var = + nir_find_variable_with_location(shader, nir_var_shader_in, + VARYING_SLOT_FACE); + + if (var == NULL) { + var = nir_variable_create(shader, nir_var_shader_in, + glsl_bool_type(), "gl_FrontFacing"); + + var->data.driver_location = shader->num_inputs++; + var->data.index = 0; + var->data.location = VARYING_SLOT_FACE; + var->data.interpolation = INTERP_MODE_FLAT; } - nir_variable *var = nir_variable_create(shader, nir_var_shader_in, - glsl_bool_type(), "gl_FrontFacing"); - - var->data.driver_location = shader->num_inputs++; - var->data.index = 0; - var->data.location = VARYING_SLOT_FACE; - var->data.interpolation = INTERP_MODE_FLAT; - return var; } diff --git a/src/compiler/nir/nir_range_analysis.c b/src/compiler/nir/nir_range_analysis.c index 3d2ff8b2c84..14f93061040 100644 --- a/src/compiler/nir/nir_range_analysis.c +++ b/src/compiler/nir/nir_range_analysis.c @@ -1141,11 +1141,8 @@ search_phi_bcsel(nir_ssa_scalar scalar, nir_ssa_scalar *buf, unsigned buf_size, static nir_variable * lookup_input(nir_shader *shader, unsigned driver_location) { - nir_foreach_shader_in_variable(var, shader) { - if (driver_location == var->data.driver_location) - return var; - } - return NULL; + return nir_find_variable_with_driver_location(shader, nir_var_shader_in, + driver_location); } uint32_t diff --git a/src/gallium/drivers/iris/iris_program.c b/src/gallium/drivers/iris/iris_program.c index f6c2ab67518..ce037f76323 100644 --- a/src/gallium/drivers/iris/iris_program.c +++ b/src/gallium/drivers/iris/iris_program.c @@ -273,14 +273,8 @@ iris_fix_edge_flags(nir_shader *nir) return false; } - nir_variable *var = NULL; - nir_foreach_shader_out_variable(v, nir) { - if (v->data.location == VARYING_SLOT_EDGE) { - var = v; - break; - } - } - + nir_variable *var = nir_find_variable_with_location(nir, nir_var_shader_out, + VARYING_SLOT_EDGE); if (!var) { nir_shader_preserve_all_metadata(nir); return false; diff --git a/src/gallium/drivers/vc4/vc4_nir_lower_io.c b/src/gallium/drivers/vc4/vc4_nir_lower_io.c index eb9bcfffd66..119ccf276d5 100644 --- a/src/gallium/drivers/vc4/vc4_nir_lower_io.c +++ b/src/gallium/drivers/vc4/vc4_nir_lower_io.c @@ -239,13 +239,9 @@ vc4_nir_lower_fs_input(struct vc4_compile *c, nir_builder *b, return; } - nir_variable *input_var = NULL; - nir_foreach_shader_in_variable(var, c->s) { - if (var->data.driver_location == nir_intrinsic_base(intr)) { - input_var = var; - break; - } - } + nir_variable *input_var = + nir_find_variable_with_driver_location(c->s, nir_var_shader_in, + nir_intrinsic_base(intr)); assert(input_var); int comp = nir_intrinsic_component(intr); @@ -290,13 +286,9 @@ static void vc4_nir_lower_output(struct vc4_compile *c, nir_builder *b, nir_intrinsic_instr *intr) { - nir_variable *output_var = NULL; - nir_foreach_shader_out_variable(var, c->s) { - if (var->data.driver_location == nir_intrinsic_base(intr)) { - output_var = var; - break; - } - } + nir_variable *output_var = + nir_find_variable_with_driver_location(c->s, nir_var_shader_out, + nir_intrinsic_base(intr)); assert(output_var); if (c->stage == QSTAGE_COORD && diff --git a/src/intel/compiler/brw_nir_lower_alpha_to_coverage.c b/src/intel/compiler/brw_nir_lower_alpha_to_coverage.c index 1eed9cd62b4..113c2266d9c 100644 --- a/src/intel/compiler/brw_nir_lower_alpha_to_coverage.c +++ b/src/intel/compiler/brw_nir_lower_alpha_to_coverage.c @@ -88,15 +88,8 @@ brw_nir_lower_alpha_to_coverage(nir_shader *shader) assert(shader->info.stage == MESA_SHADER_FRAGMENT); /* Bail out early if we don't have gl_SampleMask */ - bool is_sample_mask = false; - nir_foreach_shader_out_variable(var, shader) { - if (var->data.location == FRAG_RESULT_SAMPLE_MASK) { - is_sample_mask = true; - break; - } - } - - if (!is_sample_mask) + if (!nir_find_variable_with_location(shader, nir_var_shader_out, + FRAG_RESULT_SAMPLE_MASK)) return; nir_foreach_function(function, shader) { @@ -115,16 +108,9 @@ brw_nir_lower_alpha_to_coverage(nir_shader *shader) switch (intr->intrinsic) { case nir_intrinsic_store_output: - nir_foreach_shader_out_variable(var, shader) { - int drvloc = var->data.driver_location; - if (nir_intrinsic_base(intr) == drvloc) { - out = var; - break; - } - } - - if (out->data.mode != nir_var_shader_out) - continue; + out = nir_find_variable_with_driver_location(shader, nir_var_shader_out, + nir_intrinsic_base(intr)); + assert(out->data.mode == nir_var_shader_out); /* save gl_SampleMask instruction pointer */ if (out->data.location == FRAG_RESULT_SAMPLE_MASK) { -- 2.30.2