From cd72d8e36608d726ef8f012bcfbed46460256d91 Mon Sep 17 00:00:00 2001
From: Job Noorman <jnoorman@igalia.com>
Date: Tue, 19 Aug 2025 10:41:46 +0200
Subject: [PATCH] nir/opt_shrink_vectors: add support for offset_shift

Signed-off-by: Job Noorman <jnoorman@igalia.com>
Reviewed-by: Emma Anholt <emma@anholt.net>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/35092>
---
 src/compiler/nir/nir_opt_shrink_vectors.c | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/compiler/nir/nir_opt_shrink_vectors.c b/src/compiler/nir/nir_opt_shrink_vectors.c
index 235499d62b8..f3add2b4be6 100644
--- a/src/compiler/nir/nir_opt_shrink_vectors.c
+++ b/src/compiler/nir/nir_opt_shrink_vectors.c
@@ -130,7 +130,7 @@ shrink_dest_to_read_mask(nir_def *def, bool shrink_start)
             }
 
             nir_builder b = nir_builder_at(nir_before_instr(&intr->instr));
-            nir_src_rewrite(offset_src, nir_iadd_imm(&b, offset_src->ssa, offset));
+            nir_add_io_offset(&b, intr, offset);
          }
 
          /* Reswizzle sources, which must be ALU since they have swizzle */