mirror of
https://gitlab.freedesktop.org/mesa/mesa.git
synced 2025-12-28 23:20:08 +01:00
nir: Add a function for rewriting the condition of an if statement
Reviewed-by: Connor Abbott <cwabbott0@gmail.com>
This commit is contained in:
parent
300d729436
commit
f72a8d1cf0
2 changed files with 23 additions and 0 deletions
|
|
@ -1894,6 +1894,28 @@ nir_instr_rewrite_src(nir_instr *instr, nir_src *src, nir_src new_src)
|
|||
}
|
||||
}
|
||||
|
||||
void
|
||||
nir_if_rewrite_condition(nir_if *if_stmt, nir_src new_src)
|
||||
{
|
||||
for (nir_src *src = &if_stmt->condition; src;
|
||||
src = src->is_ssa ? NULL : src->reg.indirect) {
|
||||
struct set *uses = src->is_ssa ? src->ssa->if_uses
|
||||
: src->reg.reg->if_uses;
|
||||
struct set_entry *entry = _mesa_set_search(uses, if_stmt);
|
||||
assert(entry);
|
||||
_mesa_set_remove(uses, entry);
|
||||
}
|
||||
|
||||
if_stmt->condition = new_src;
|
||||
|
||||
for (nir_src *src = &if_stmt->condition; src;
|
||||
src = src->is_ssa ? NULL : src->reg.indirect) {
|
||||
struct set *uses = src->is_ssa ? src->ssa->if_uses
|
||||
: src->reg.reg->if_uses;
|
||||
_mesa_set_add(uses, if_stmt);
|
||||
}
|
||||
}
|
||||
|
||||
void
|
||||
nir_ssa_def_init(nir_instr *instr, nir_ssa_def *def,
|
||||
unsigned num_components, const char *name)
|
||||
|
|
|
|||
|
|
@ -1549,6 +1549,7 @@ bool nir_foreach_src(nir_instr *instr, nir_foreach_src_cb cb, void *state);
|
|||
nir_const_value *nir_src_as_const_value(nir_src src);
|
||||
bool nir_srcs_equal(nir_src src1, nir_src src2);
|
||||
void nir_instr_rewrite_src(nir_instr *instr, nir_src *src, nir_src new_src);
|
||||
void nir_if_rewrite_condition(nir_if *if_stmt, nir_src new_src);
|
||||
|
||||
void nir_ssa_dest_init(nir_instr *instr, nir_dest *dest,
|
||||
unsigned num_components, const char *name);
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue