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>

diff --git a/src/gallium/drivers/r600/sfn/sfn_shader.cpp b/src/gallium/drivers/r600/sfn/sfn_shader.cpp
index 238171a..d54c952 100644 (file)
--- 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};