From fb81201a34fbc526ca49614d293f541fdb428452 Mon Sep 17 00:00:00 2001
From: Alyssa Rosenzweig <alyssa@rosenzweig.io>
Date: Fri, 12 Jan 2024 17:34:44 -0400
Subject: [PATCH] agx: implement active_subgroup_invocation_agx

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 | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/asahi/compiler/agx_compile.c b/src/asahi/compiler/agx_compile.c
index 7684430174e..cf2b84ef3c6 100644
--- a/src/asahi/compiler/agx_compile.c
+++ b/src/asahi/compiler/agx_compile.c
@@ -1260,6 +1260,10 @@ agx_emit_intrinsic(agx_builder *b, nir_intrinsic_instr *instr)
    case nir_intrinsic_load_subgroup_invocation:
       return agx_get_sr_to(b, dst, AGX_SR_THREAD_INDEX_IN_SUBGROUP);
 
+   case nir_intrinsic_load_active_subgroup_invocation_agx:
+      return agx_get_sr_coverage_to(b, dst,
+                                    AGX_SR_ACTIVE_THREAD_INDEX_IN_SUBGROUP);
+
    case nir_intrinsic_reduce: {
       assert(nir_intrinsic_reduction_op(instr) == nir_op_iadd &&
              "other reductions todo");