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

spirv: Add support for the CL Round instruction

Browse Source 
Add a round() implementation that's conformant with the CL spec.

Signed-off-by: Boris Brezillon <boris.brezillon@collabora.com>
Reviewed-by: Karol Herbst <kherbst@redhat.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/6256>
This commit is contained in:
Boris Brezillon
2020-07-28 08:42:53 +02:00
parent 6d28270968
commit 165009bc70
1 changed files with 18 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
src/compiler/spirv/vtn_opencl.c
View File

@@ -281,6 +281,21 @@ handle_printf(struct vtn_builder *b, enum OpenCLstd_Entrypoints opcode,
return nir_imm_int(&b->nb, -1);
}
static nir_ssa_def *
handle_round(struct vtn_builder *b, enum OpenCLstd_Entrypoints opcode,
unsigned num_srcs, nir_ssa_def **srcs,
const struct glsl_type *dest_type)
{
nir_ssa_def *src = srcs[0];
nir_builder *nb = &b->nb;
nir_ssa_def *half = nir_imm_floatN_t(nb, 0.5, src->bit_size);
nir_ssa_def *truncated = nir_ftrunc(nb, src);
nir_ssa_def *remainder = nir_fsub(nb, src, truncated);
return nir_bcsel(nb, nir_fge(nb, nir_fabs(nb, remainder), half),
nir_fadd(nb, truncated, nir_fsign(nb, src)), truncated);
}
static nir_ssa_def *
handle_shuffle(struct vtn_builder *b, enum OpenCLstd_Entrypoints opcode, unsigned num_srcs,
nir_ssa_def **srcs, const struct glsl_type *dest_type)
@@ -435,6 +450,9 @@ vtn_handle_opencl_instruction(struct vtn_builder *b, SpvOp ext_opcode,
case OpenCLstd_Shuffle2:
handle_instr(b, cl_opcode, w, count, handle_shuffle2);
return true;
case OpenCLstd_Round:
handle_instr(b, cl_opcode, w, count, handle_round);
return true;
case OpenCLstd_Printf:
handle_instr(b, cl_opcode, w, count, handle_printf);
return true;