From f52e0bd70ba2b3031324fdc55d4ccd9d8d9919df Mon Sep 17 00:00:00 2001
From: Alyssa Rosenzweig <alyssa@rosenzweig.io>
Date: Fri, 12 Jan 2024 15:33:41 -0400
Subject: [PATCH] agx: implement load_subgroup_invocation

There's an sr.

Signed-off-by: Alyssa Rosenzweig <alyssa@rosenzweig.io>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/27616>
---
 src/asahi/compiler/agx_compile.c | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/asahi/compiler/agx_compile.c b/src/asahi/compiler/agx_compile.c
index 207f22f133a..cfcee946550 100644
--- a/src/asahi/compiler/agx_compile.c
+++ b/src/asahi/compiler/agx_compile.c
@@ -1257,6 +1257,9 @@ agx_emit_intrinsic(agx_builder *b, nir_intrinsic_instr *instr)
       return NULL;
    }
 
+   case nir_intrinsic_load_subgroup_invocation:
+      return agx_get_sr_to(b, dst, AGX_SR_THREAD_INDEX_IN_SUBGROUP);
+
    case nir_intrinsic_reduce: {
       assert(nir_intrinsic_reduction_op(instr) == nir_op_iadd &&
              "other reductions todo");