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

spirv,nir: Move the SPIR-V vector insert code to NIR

Browse Source 
This also makes spirv_to_nir a bit simpler because the new
nir_vector_insert helper automatically handles a constant component
selector like nir_vector_extract does.

Reviewed-by: Caio Marcelo de Oliveira Filho <caio.oliveira@intel.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/4495>
This commit is contained in:
Jason Ekstrand
2020-04-09 17:09:10 -05:00
committed by Marge Bot
parent feca439697
commit f5deed138a
4 changed files with 61 additions and 53 deletions
Download Patch File Download Diff File Expand all files Collapse all files

55
src/compiler/nir/nir_builder.h
View File

@@ -601,6 +601,61 @@ nir_vector_extract(nir_builder *b, nir_ssa_def *vec, nir_ssa_def *c)
}
}
/** Replaces the component of `vec` specified by `c` with `scalar` */
static inline nir_ssa_def *
nir_vector_insert_imm(nir_builder *b, nir_ssa_def *vec,
nir_ssa_def *scalar, unsigned c)
{
assert(scalar->num_components == 1);
assert(c < vec->num_components);
nir_op vec_op = nir_op_vec(vec->num_components);
nir_alu_instr *vec_instr = nir_alu_instr_create(b->shader, vec_op);
for (unsigned i = 0; i < vec->num_components; i++) {
if (i == c) {
vec_instr->src[i].src = nir_src_for_ssa(scalar);
vec_instr->src[i].swizzle[0] = 0;
} else {
vec_instr->src[i].src = nir_src_for_ssa(vec);
vec_instr->src[i].swizzle[0] = i;
}
}
return nir_builder_alu_instr_finish_and_insert(b, vec_instr);
}
/** Replaces the component of `vec` specified by `c` with `scalar` */
static inline nir_ssa_def *
nir_vector_insert(nir_builder *b, nir_ssa_def *vec, nir_ssa_def *scalar,
nir_ssa_def *c)
{
assert(scalar->num_components == 1);
assert(c->num_components == 1);
nir_src c_src = nir_src_for_ssa(c);
if (nir_src_is_const(c_src)) {
uint64_t c_const = nir_src_as_uint(c_src);
if (c_const < vec->num_components)
return nir_vector_insert_imm(b, vec, scalar, c_const);
else
return vec;
} else {
nir_const_value per_comp_idx_const[NIR_MAX_VEC_COMPONENTS];
for (unsigned i = 0; i < NIR_MAX_VEC_COMPONENTS; i++)
per_comp_idx_const[i] = nir_const_value_for_int(i, c->bit_size);
nir_ssa_def *per_comp_idx =
nir_build_imm(b, vec->num_components,
c->bit_size, per_comp_idx_const);
/* nir_builder will automatically splat out scalars to vectors so an
* insert is as simple as "if I'm the channel, replace me with the
* scalar."
*/
return nir_bcsel(b, nir_ieq(b, c, per_comp_idx), scalar, vec);
}
}
static inline nir_ssa_def *
nir_i2i(nir_builder *build, nir_ssa_def *x, unsigned dest_bit_size)
{