mirror of
https://gitlab.freedesktop.org/mesa/mesa.git
synced 2026-05-05 20:28:04 +02:00
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:
parent
6f2e95f24d
commit
92333c6d1a
1 changed files with 36 additions and 24 deletions
|
|
@ -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;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue