diff --git a/src/compiler/nir/nir.c b/src/compiler/nir/nir.c index 66bfb399225..bf1315abeae 100644 --- a/src/compiler/nir/nir.c +++ b/src/compiler/nir/nir.c @@ -332,6 +332,20 @@ nir_alu_dest_copy(nir_alu_dest *dest, const nir_alu_dest *src, dest->saturate = src->saturate; } +bool +nir_alu_src_is_trivial_ssa(const nir_alu_instr *alu, unsigned srcn) +{ + static uint8_t trivial_swizzle[] = { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15 }; + STATIC_ASSERT(ARRAY_SIZE(trivial_swizzle) == NIR_MAX_VEC_COMPONENTS); + + const nir_alu_src *src = &alu->src[srcn]; + unsigned num_components = nir_ssa_alu_instr_src_components(alu, srcn); + + return src->src.is_ssa && (src->src.ssa->num_components == num_components) && + !src->abs && !src->negate && + (memcmp(src->swizzle, trivial_swizzle, num_components) == 0); +} + static void cf_init(nir_cf_node *node, nir_cf_node_type type) diff --git a/src/compiler/nir/nir.h b/src/compiler/nir/nir.h index eb706a19bf5..ba835cd221d 100644 --- a/src/compiler/nir/nir.h +++ b/src/compiler/nir/nir.h @@ -1432,6 +1432,8 @@ bool nir_alu_srcs_negative_equal(const nir_alu_instr *alu1, const nir_alu_instr *alu2, unsigned src1, unsigned src2); +bool nir_alu_src_is_trivial_ssa(const nir_alu_instr *alu, unsigned srcn); + typedef enum { nir_deref_type_var, nir_deref_type_array,