nir: Add builder helpers for MOVs with ALU sources and swizzling MOVs.
These will be useful for prog->nir and tgsi->nir. v2: Don't forget to mark nir_swizzle as inline (Eric). Signed-off-by: Kenneth Graunke <kenneth@whitecape.org> Reviewed-by: Connor Abbott <cwabbott0@gmail.com> Reviewed-by: Eric Anholt <eric@anholt.net>
This commit is contained in:
parent
75c922e0fe
commit
06f7bea96a
|
@ -162,4 +162,48 @@ nir_##op(nir_builder *build, nir_ssa_def *src0, \
|
||||||
|
|
||||||
#include "nir_builder_opcodes.h"
|
#include "nir_builder_opcodes.h"
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Similar to nir_fmov, but takes a nir_alu_src instead of a nir_ssa_def.
|
||||||
|
*/
|
||||||
|
static inline nir_ssa_def *
|
||||||
|
nir_fmov_alu(nir_builder *build, nir_alu_src src, unsigned num_components)
|
||||||
|
{
|
||||||
|
nir_alu_instr *mov = nir_alu_instr_create(build->shader, nir_op_fmov);
|
||||||
|
nir_ssa_dest_init(&mov->instr, &mov->dest.dest, num_components, NULL);
|
||||||
|
mov->dest.write_mask = (1 << num_components) - 1;
|
||||||
|
mov->src[0] = src;
|
||||||
|
nir_instr_insert_after_cf_list(build->cf_node_list, &mov->instr);
|
||||||
|
|
||||||
|
return &mov->dest.dest.ssa;
|
||||||
|
}
|
||||||
|
|
||||||
|
static inline nir_ssa_def *
|
||||||
|
nir_imov_alu(nir_builder *build, nir_alu_src src, unsigned num_components)
|
||||||
|
{
|
||||||
|
nir_alu_instr *mov = nir_alu_instr_create(build->shader, nir_op_imov);
|
||||||
|
nir_ssa_dest_init(&mov->instr, &mov->dest.dest, num_components, NULL);
|
||||||
|
mov->dest.write_mask = (1 << num_components) - 1;
|
||||||
|
mov->src[0] = src;
|
||||||
|
nir_instr_insert_after_cf_list(build->cf_node_list, &mov->instr);
|
||||||
|
|
||||||
|
return &mov->dest.dest.ssa;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Construct an fmov or imov that reswizzles the source's components.
|
||||||
|
*/
|
||||||
|
static inline nir_ssa_def *
|
||||||
|
nir_swizzle(nir_builder *build, nir_ssa_def *src, unsigned swiz[4],
|
||||||
|
unsigned num_components, bool use_fmov)
|
||||||
|
{
|
||||||
|
nir_alu_src alu_src;
|
||||||
|
memset(&alu_src, 0, sizeof(alu_src));
|
||||||
|
alu_src.src = nir_src_for_ssa(src);
|
||||||
|
for (int i = 0; i < 4; i++)
|
||||||
|
alu_src.swizzle[i] = swiz[i];
|
||||||
|
|
||||||
|
return use_fmov ? nir_fmov_alu(build, alu_src, num_components) :
|
||||||
|
nir_imov_alu(build, alu_src, num_components);
|
||||||
|
}
|
||||||
|
|
||||||
#endif /* NIR_BUILDER_H */
|
#endif /* NIR_BUILDER_H */
|
||||||
|
|
Loading…
Reference in New Issue