spirv/nir/opencl: handle some multiply instructions.

This adds support for some missing 24-bit and hi multiply
variants.

Reviewed-by: Jason Ekstrand <jason@jlekstrand.net>
This commit is contained in:
Dave Airlie
2019-04-30 06:57:11 +10:00
parent 5375c30234
commit 12913bcf86
2 changed files with 55 additions and 0 deletions

View File

@@ -82,6 +82,43 @@ nir_uabs_diff(nir_builder *b, nir_ssa_def *x, nir_ssa_def *y)
return nir_bcsel(b, cond, res0, res1);
}
static inline nir_ssa_def *
nir_umul24(nir_builder *b, nir_ssa_def *x, nir_ssa_def *y)
{
nir_ssa_def *mask = nir_imm_int(b, 0xffffff);
nir_ssa_def *x_24 = nir_iand(b, x, mask);
nir_ssa_def *y_24 = nir_iand(b, y, mask);
return nir_imul(b, x_24, y_24);
}
static inline nir_ssa_def *
nir_umad24(nir_builder *b, nir_ssa_def *x, nir_ssa_def *y, nir_ssa_def *z)
{
nir_ssa_def *temp = nir_umul24(b, x, y);
return nir_iadd(b, temp, z);
}
static inline nir_ssa_def *
nir_imad24(nir_builder *b, nir_ssa_def *x, nir_ssa_def *y, nir_ssa_def *z)
{
nir_ssa_def *temp = nir_imul24(b, x, y);
return nir_iadd(b, temp, z);
}
static inline nir_ssa_def *
nir_imad_hi(nir_builder *b, nir_ssa_def *x, nir_ssa_def *y, nir_ssa_def *z)
{
nir_ssa_def *temp = nir_imul_high(b, x, y);
return nir_iadd(b, temp, z);
}
static inline nir_ssa_def *
nir_umad_hi(nir_builder *b, nir_ssa_def *x, nir_ssa_def *y, nir_ssa_def *z)
{
nir_ssa_def *temp = nir_umul_high(b, x, y);
return nir_iadd(b, temp, z);
}
static inline nir_ssa_def *
nir_bitselect(nir_builder *b, nir_ssa_def *x, nir_ssa_def *y, nir_ssa_def *s)
{