From c3f60f77d6bdc6f6cc9327216b5db9cdddc6b219 Mon Sep 17 00:00:00 2001
From: Gert Wollny <gert.wollny@collabora.com>
Date: Fri, 10 Feb 2023 16:03:07 +0100
Subject: [PATCH] r600/sfn: handle AR and IDX register in shader from string

This is needed for testing

Signed-off-by: Gert Wollny <gert.wollny@collabora.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/21347>
---
 src/gallium/drivers/r600/sfn/sfn_shader.cpp | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/gallium/drivers/r600/sfn/sfn_shader.cpp b/src/gallium/drivers/r600/sfn/sfn_shader.cpp
index 238171a..d54c952 100644
--- a/src/gallium/drivers/r600/sfn/sfn_shader.cpp
+++ b/src/gallium/drivers/r600/sfn/sfn_shader.cpp
@@ -330,7 +330,9 @@ Shader::allocate_registers_from_string(std::istream& is, Pin pin)
       if (reg_str.empty())
          break;
 
-      if (strchr(reg_str.c_str(), '@')) {
+      if (strchr(reg_str.c_str(), '@') ||
+          reg_str == "AR" ||
+          reg_str.substr(0,3) == "IDX") {
          value_factory().dest_from_string(reg_str);
       } else {
          RegisterVec4::Swizzle swz = {0, 1, 2, 3};
-- 
2.7.4