diff --git a/src/imagination/pco/pco_nir.c b/src/imagination/pco/pco_nir.c index 275b6bfc3a4..15da3af67b4 100644 --- a/src/imagination/pco/pco_nir.c +++ b/src/imagination/pco/pco_nir.c @@ -898,6 +898,7 @@ void pco_lower_nir(pco_ctx *ctx, nir_shader *nir, pco_data *data) nir_lower_tex, &(nir_lower_tex_options){ .lower_txd_cube_map = true, + .lower_tg4_offsets = true, }); NIR_PASS(_, nir, pco_nir_lower_tex, data, ctx); diff --git a/src/imagination/pco/pco_nir_tex.c b/src/imagination/pco/pco_nir_tex.c index fd07449c943..eef7a42886c 100644 --- a/src/imagination/pco/pco_nir_tex.c +++ b/src/imagination/pco/pco_nir_tex.c @@ -417,19 +417,13 @@ nir_intrinsic_instr *pco_emit_nir_smp(nir_builder *b, pco_smp_params *params) static nir_def * lower_tex_gather(nir_builder *b, nir_tex_instr *tex, nir_def *raw_data) { - unsigned swiz[ARRAY_SIZE(tex->tg4_offsets)]; - for (unsigned u = 0; u < ARRAY_SIZE(tex->tg4_offsets); ++u) { - unsigned offset = ARRAY_SIZE(*tex->tg4_offsets) * tex->tg4_offsets[u][0]; - offset += tex->tg4_offsets[u][1]; - offset *= ARRAY_SIZE(tex->tg4_offsets); - offset += tex->component; + assert(!nir_tex_instr_has_explicit_tg4_offsets(tex)); - swiz[u] = offset; - } +#define TG4_SEL(sample) (((sample) * 4) + tex->component) + unsigned swiz[] = { TG4_SEL(2), TG4_SEL(3), TG4_SEL(1), TG4_SEL(0) }; +#undef TG4_SEL - nir_def *result = nir_swizzle(b, raw_data, swiz, ARRAY_SIZE(swiz)); - - return result; + return nir_swizzle(b, raw_data, swiz, ARRAY_SIZE(swiz)); } static nir_def *lower_tex_shadow(nir_builder *b,