From 2424b31870fd19ad1456ac82df58dcc15ef5613f Mon Sep 17 00:00:00 2001
From: Simon Perretta <simon.perretta@imgtec.com>
Date: Wed, 9 Apr 2025 00:59:24 +0100
Subject: [PATCH] pco: add lower_io_array_vars_to_elements_no_indirects to
 preprocessing

Signed-off-by: Simon Perretta <simon.perretta@imgtec.com>
Acked-by: Erik Faye-Lund <erik.faye-lund@collabora.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/36412>
---
 src/imagination/pco/pco_nir.c | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/src/imagination/pco/pco_nir.c b/src/imagination/pco/pco_nir.c
index 123d55ac2b1..270356a16fa 100644
--- a/src/imagination/pco/pco_nir.c
+++ b/src/imagination/pco/pco_nir.c
@@ -551,6 +551,16 @@ void pco_preprocess_nir(pco_ctx *ctx, nir_shader *nir)
             nir_var_function_temp | nir_var_shader_temp,
             NULL);
 
+   NIR_PASS(_,
+            nir,
+            nir_io_add_const_offset_to_base,
+            nir_var_shader_in | nir_var_shader_out);
+
+   NIR_PASS(_,
+            nir,
+            nir_lower_io_array_vars_to_elements_no_indirects,
+            nir->info.stage == MESA_SHADER_VERTEX);
+
    pco_nir_opt(ctx, nir);
 
    if (pco_should_print_nir(nir)) {