nir: Support deref instructions in lower_alpha_test

Reviewed-by: Caio Marcelo de Oliveira Filho <caio.oliveira@intel.com>
Acked-by: Rob Clark <robdclark@gmail.com>
Acked-by: Bas Nieuwenhuizen <bas@basnieuwenhuizen.nl>
Acked-by: Dave Airlie <airlied@redhat.com>
Reviewed-by: Kenneth Graunke <kenneth@whitecape.org>
This commit is contained in:
Jason Ekstrand 2018-03-26 17:06:35 -07:00
parent 414148cdc1
commit 076b6627c2
1 changed files with 6 additions and 2 deletions

View File

@ -41,8 +41,6 @@ nir_lower_alpha_test(nir_shader *shader, enum compare_func func,
{
assert(shader->info.stage == MESA_SHADER_FRAGMENT);
nir_assert_lowered_derefs(shader, nir_lower_load_store_derefs);
nir_foreach_function(function, shader) {
nir_function_impl *impl = function->impl;
nir_builder b;
@ -60,6 +58,9 @@ nir_lower_alpha_test(nir_shader *shader, enum compare_func func,
case nir_intrinsic_store_var:
out = intr->variables[0]->var;
break;
case nir_intrinsic_store_deref:
out = nir_deref_instr_get_variable(nir_src_as_deref(intr->src[0]));
break;
case nir_intrinsic_store_output:
/* already had i/o lowered.. lookup the matching output var: */
nir_foreach_variable(var, &shader->outputs) {
@ -87,6 +88,9 @@ nir_lower_alpha_test(nir_shader *shader, enum compare_func func,
nir_ssa_def *alpha;
if (alpha_to_one) {
alpha = nir_imm_float(&b, 1.0);
} else if (intr->intrinsic == nir_intrinsic_store_deref) {
alpha = nir_channel(&b, nir_ssa_for_src(&b, intr->src[1], 4),
3);
} else {
alpha = nir_channel(&b, nir_ssa_for_src(&b, intr->src[0], 4),
3);