nir: lower int16 and uint16 in nir_lower_mediump_outputs

Reviewed-by: Alyssa Rosenzweig <alyssa.rosenzweig@collabora.com>
Reviewed-by: Rob Clark <robdclark@chromium.org>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/5002>
This commit is contained in:
Marek Olšák 2020-05-10 23:04:23 -04:00 committed by Marge Bot
parent 6f2e95f24d
commit 92333c6d1a

View file

@ -24,24 +24,7 @@
#include "nir.h"
#include "nir_builder.h"
/* Lower mediump FS outputs to fp16 */
static bool
lower_output_var(nir_shader *nir, int location)
{
nir_foreach_variable (var, &nir->outputs) {
if (var->data.driver_location == location &&
((var->data.precision == GLSL_PRECISION_MEDIUM) ||
(var->data.precision == GLSL_PRECISION_LOW))) {
if (glsl_get_base_type(var->type) == GLSL_TYPE_FLOAT)
var->type = glsl_float16_type(var->type);
return glsl_get_base_type(var->type) == GLSL_TYPE_FLOAT16;
}
}
return false;
}
/* Lower mediump outputs to float16, int16, or uint16. */
void
nir_lower_mediump_outputs(nir_shader *nir)
@ -64,14 +47,43 @@ nir_lower_mediump_outputs(nir_shader *nir)
if (intr->intrinsic != nir_intrinsic_store_output)
continue;
if (!lower_output_var(nir, nir_intrinsic_base(intr)))
continue;
nir_foreach_variable (var, &nir->outputs) {
if (var->data.driver_location != nir_intrinsic_base(intr))
continue; /* not found yet */
b.cursor = nir_before_instr(&intr->instr);
nir_instr_rewrite_src(&intr->instr, &intr->src[0],
nir_src_for_ssa(nir_f2f16(&b, intr->src[0].ssa)));
if (var->data.precision != GLSL_PRECISION_MEDIUM &&
var->data.precision != GLSL_PRECISION_LOW)
break; /* can't lower */
nir_intrinsic_set_type(intr, nir_type_float16);
switch (glsl_get_base_type(var->type)) {
case GLSL_TYPE_FLOAT:
var->type = glsl_float16_type(var->type);
b.cursor = nir_before_instr(&intr->instr);
nir_instr_rewrite_src(&intr->instr, &intr->src[0],
nir_src_for_ssa(nir_f2f16(&b, intr->src[0].ssa)));
nir_intrinsic_set_type(intr, nir_type_float16);
break;
case GLSL_TYPE_INT:
var->type = glsl_int16_type(var->type);
b.cursor = nir_before_instr(&intr->instr);
nir_instr_rewrite_src(&intr->instr, &intr->src[0],
nir_src_for_ssa(nir_i2i16(&b, intr->src[0].ssa)));
nir_intrinsic_set_type(intr, nir_type_int16);
break;
case GLSL_TYPE_UINT:
var->type = glsl_uint16_type(var->type);
b.cursor = nir_before_instr(&intr->instr);
nir_instr_rewrite_src(&intr->instr, &intr->src[0],
nir_src_for_ssa(nir_u2u16(&b, intr->src[0].ssa)));
nir_intrinsic_set_type(intr, nir_type_uint16);
break;
default:;
}
break;
}
}
}
}