From 24de22d0dee699530bddbaa111614f3e1bc0bb4b Mon Sep 17 00:00:00 2001
From: Simon Perretta <simon.perretta@imgtec.com>
Date: Mon, 24 Feb 2025 13:38:05 +0000
Subject: [PATCH] pco: amend index register mapping

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_map.h.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/src/imagination/pco/pco_map.h.py b/src/imagination/pco/pco_map.h.py
index 0cac4d8d66f..818ac4b8792 100644
--- a/src/imagination/pco/pco_map.h.py
+++ b/src/imagination/pco/pco_map.h.py
@@ -73,8 +73,11 @@ ${enum_map.type_to} ${enum_map.name}(${enum_map.type_from} val)
 static inline
 enum pco_regbank pco_map_reg_bank(pco_ref ref)
 {
-   enum pco_regbank regbank = pco_map_reg_class_to_regbank(pco_ref_get_reg_class(ref));
-   return pco_ref_is_idx_reg(ref) ? regbank + ref.idx_reg.num : regbank;
+   if (pco_ref_is_idx_reg(ref)) {
+      return PCO_REGBANK_IDX0 + ref.idx_reg.num;
+   }
+
+   return pco_map_reg_class_to_regbank(pco_ref_get_reg_class(ref));
 }
 
 static inline