OpenHarmony/third_party_mesa3d
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

nir/loop_analyze: Use nir_loop_variable::init_src instead of nir_basic_induction_var::def_outside_loop

Browse Source 
These track the same information in a slightly different way. Since
nir_loop_variable::init_src is visible outside this module, it cannot
be eliminated.

As an intentional side effect, induction variables with constant
initializers will now have their nir_loop_induction_variable::init_src
field point to the load_const source. Previously this pointer would be
NULL.

v2: Update unit tests and commit message. Remove the now unused ind_var
variable in find_trip_count.

Reviewed-by: Caio Oliveira <caio.oliveira@intel.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/21289>
This commit is contained in:
Ian Romanick
2023-02-07 13:10:22 -08:00
committed by Marge Bot
parent 72e763650c
commit e444ed9210
2 changed files with 44 additions and 47 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
src/compiler/nir/nir_loop_analyze.c
View File

@@ -423,7 +423,6 @@ compute_induction_information(loop_info_state *state)
nir_phi_instr *phi = nir_instr_as_phi(var->def->parent_instr);
nir_basic_induction_var *biv = rzalloc(state, nir_basic_induction_var);
nir_src *init_src = NULL;
nir_loop_variable *alu_src_var = NULL;
nir_foreach_phi_src(src, phi) {
nir_loop_variable *src_var = get_loop_var(src->src.ssa, state);
@@ -448,9 +447,8 @@ compute_induction_information(loop_info_state *state)
}
}
if (!src_var->in_loop && !biv->def_outside_loop) {
biv->def_outside_loop = src_var->def;
init_src = &src->src;
if (!src_var->in_loop && !var->init_src) {
var->init_src = &src->src;
} else if (is_var_alu(src_var) && !var->update_src) {
alu_src_var = src_var;
nir_alu_instr *alu = nir_instr_as_alu(src_var->def->parent_instr);
@@ -480,8 +478,8 @@ compute_induction_information(loop_info_state *state)
}
}
if (var->update_src && biv->def_outside_loop) {
nir_instr *inst = biv->def_outside_loop->parent_instr;
if (var->update_src && var->init_src) {
nir_instr *inst = var->init_src->ssa->parent_instr;
if (inst->type == nir_instr_type_load_const) {
/* Initial value of induction variable is a constant */
alu_src_var->init_src = var->init_src;
@@ -493,20 +491,20 @@ compute_induction_information(loop_info_state *state)
found_induction_var = true;
num_induction_vars += 2;
} else if (is_only_uniform_src(init_src)) {
} else if (is_only_uniform_src(var->init_src)) {
/* Initial value of induction variable is a uniform */
var->init_src = init_src;
alu_src_var->init_src = var->init_src;
alu_src_var->update_src = var->update_src;
num_induction_vars += 2;
ralloc_free(biv);
} else {
var->init_src = NULL;
var->update_src = NULL;
ralloc_free(biv);
}
} else {
var->init_src = NULL;
var->update_src = NULL;
ralloc_free(biv);
}
@@ -1195,13 +1193,12 @@ find_trip_count(loop_info_state *state, unsigned execution_mode)
*/
nir_loop_variable *lv = get_loop_var(basic_ind.def, state);
nir_basic_induction_var *ind_var = lv->ind;
/* The basic induction var might be a vector but, because we guarantee
* earlier that the phi source has a scalar swizzle, we can take the
* component from basic_ind.
*/
nir_ssa_scalar initial_s = { ind_var->def_outside_loop, basic_ind.comp };
nir_ssa_scalar initial_s = { lv->init_src->ssa, basic_ind.comp };
nir_ssa_scalar alu_s = {
lv->update_src->src.ssa,
lv->update_src->swizzle[basic_ind.comp]