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

nir: use nir_shader_instructions_pass in nir_lower_frexp

Browse Source 
Changes:
- nir_metadata_preserve(..., nir_metadata_all) is called when pass doesn't
  make progress

Reviewed-by: Alyssa Rosenzweig <alyssa@collabora.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/12282>
This commit is contained in:
Marcin Ślusarz
2021-08-09 16:43:13 +02:00
committed by Marge Bot
parent a87070937d
commit dd51dedefd
1 changed files with 22 additions and 43 deletions
Download Patch File Download Diff File Expand all files Collapse all files

65
src/compiler/nir/nir_lower_frexp.c
View File

@@ -163,58 +163,37 @@ lower_frexp_exp(nir_builder *b, nir_ssa_def *x)
}
static bool
lower_frexp_impl(nir_function_impl *impl)
lower_frexp_instr(nir_builder *b, nir_instr *instr, UNUSED void *cb_data)
{
bool progress = false;
if (instr->type != nir_instr_type_alu)
return false;
nir_builder b;
nir_builder_init(&b, impl);
nir_alu_instr *alu_instr = nir_instr_as_alu(instr);
nir_ssa_def *lower;
nir_foreach_block(block, impl) {
nir_foreach_instr_safe(instr, block) {
if (instr->type != nir_instr_type_alu)
continue;
b->cursor = nir_before_instr(instr);
nir_alu_instr *alu_instr = nir_instr_as_alu(instr);
nir_ssa_def *lower;
b.cursor = nir_before_instr(instr);
switch (alu_instr->op) {
case nir_op_frexp_sig:
lower = lower_frexp_sig(&b, nir_ssa_for_alu_src(&b, alu_instr, 0));
break;
case nir_op_frexp_exp:
lower = lower_frexp_exp(&b, nir_ssa_for_alu_src(&b, alu_instr, 0));
break;
default:
continue;
}
nir_ssa_def_rewrite_uses(&alu_instr->dest.dest.ssa,
lower);
nir_instr_remove(instr);
progress = true;
}
switch (alu_instr->op) {
case nir_op_frexp_sig:
lower = lower_frexp_sig(b, nir_ssa_for_alu_src(b, alu_instr, 0));
break;
case nir_op_frexp_exp:
lower = lower_frexp_exp(b, nir_ssa_for_alu_src(b, alu_instr, 0));
break;
default:
return false;
}
if (progress) {
nir_metadata_preserve(impl, nir_metadata_block_index |
nir_metadata_dominance);
}
return progress;
nir_ssa_def_rewrite_uses(&alu_instr->dest.dest.ssa, lower);
nir_instr_remove(instr);
return true;
}
bool
nir_lower_frexp(nir_shader *shader)
{
bool progress = false;
nir_foreach_function(function, shader) {
if (function->impl)
progress |= lower_frexp_impl(function->impl);
}
return progress;
return nir_shader_instructions_pass(shader, lower_frexp_instr,
nir_metadata_block_index |
nir_metadata_dominance,
NULL);
}