fdo-mirrors/mesa
1
0
Fork 0
mirror of https://gitlab.freedesktop.org/mesa/mesa.git synced 2026-04-18 10:20:39 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

nir: add nir_shader_alu_pass

Browse source 
after the smashing success of nir_shader_intrinsics_pass, let's add the ALU
version to help the odd non-algebraic ALU lowering pass.

Signed-off-by: Alyssa Rosenzweig <alyssa@rosenzweig.io>
Reviewed-by: Reviewed-by: Christian Gmeiner <cgmeiner@igalia.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/30582>
This commit is contained in:
Alyssa Rosenzweig 2024-08-08 22:21:53 -04:00 • committed by Marge Bot
parent d330870f9c
commit cc1f092b62
1 changed files with 35 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

35
src/compiler/nir/nir_builder.h
View file

@ -81,6 +81,8 @@ nir_builder MUST_CHECK PRINTFLIKE(3, 4)
typedef bool (*nir_instr_pass_cb)(struct nir_builder *, nir_instr *, void *);
typedef bool (*nir_intrinsic_pass_cb)(struct nir_builder *,
nir_intrinsic_instr *, void *);
typedef bool (*nir_alu_pass_cb)(struct nir_builder *,
nir_alu_instr *, void *);
/**
* Iterates over all the instructions in a NIR function and calls the given pass
@ -184,6 +186,39 @@ nir_shader_intrinsics_pass(nir_shader *shader,
return progress;
}
/* As above, but for ALU */
static inline bool
nir_shader_alu_pass(nir_shader *shader,
nir_alu_pass_cb pass,
nir_metadata preserved,
void *cb_data)
{
bool progress = false;
nir_foreach_function_impl(impl, shader) {
bool func_progress = false;
nir_builder b = nir_builder_create(impl);
nir_foreach_block_safe(block, impl) {
nir_foreach_instr_safe(instr, block) {
if (instr->type == nir_instr_type_alu) {
nir_alu_instr *intr = nir_instr_as_alu(instr);
func_progress |= pass(&b, intr, cb_data);
}
}
}
if (func_progress) {
nir_metadata_preserve(impl, preserved);
progress = true;
} else {
nir_metadata_preserve(impl, nir_metadata_all);
}
}
return progress;
}
void nir_builder_instr_insert(nir_builder *build, nir_instr *instr);
void nir_builder_instr_insert_at_top(nir_builder *build, nir_instr *instr);