From 49897f99d65b48adc8fe15f9c85bdd5c0fd6566b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Marek=20Ol=C5=A1=C3=A1k?= <marek.olsak@amd.com>
Date: Sun, 5 Jan 2025 08:56:55 -0500
Subject: [PATCH] st/mesa: recompute IO bases for ARB_vp/fp

This is a prerequisite for generating lowered IO for ARB_vp/fp.

Reviewed-by: Alyssa Rosenzweig <alyssa@rosenzweig.io>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/32779>
---
 src/mesa/state_tracker/st_program.c | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/mesa/state_tracker/st_program.c b/src/mesa/state_tracker/st_program.c
index 7af6ea92cef..182cb925014 100644
--- a/src/mesa/state_tracker/st_program.c
+++ b/src/mesa/state_tracker/st_program.c
@@ -395,6 +395,9 @@ st_prog_to_nir_postprocess(struct st_context *st, nir_shader *nir,
    /* Optimise NIR */
    NIR_PASS(_, nir, nir_opt_constant_folding);
    gl_nir_opts(nir);
+
+   /* This must be done after optimizations to assign IO bases. */
+   nir_recompute_io_bases(nir, nir_var_shader_in | nir_var_shader_out);
    st_finalize_nir_before_variants(nir);
 
    if (st->allow_st_finalize_nir_twice) {