From f4a5ac35541b06393b61fc41d07a9715903271a7 Mon Sep 17 00:00:00 2001 From: Alyssa Rosenzweig Date: Sat, 20 Apr 2024 12:47:15 -0400 Subject: [PATCH] agx: handle quad_broadcast Signed-off-by: Alyssa Rosenzweig Part-of: --- src/asahi/compiler/agx_compile.c | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/src/asahi/compiler/agx_compile.c b/src/asahi/compiler/agx_compile.c index ee6fe1b3eb9..c69cf3559e8 100644 --- a/src/asahi/compiler/agx_compile.c +++ b/src/asahi/compiler/agx_compile.c @@ -1496,6 +1496,15 @@ agx_emit_intrinsic(agx_builder *b, nir_intrinsic_instr *instr) agx_src_index(&instr->src[1])); } + case nir_intrinsic_quad_broadcast: { + /* TODO: Check if we're actually inside divergent control flow */ + b->shader->any_quad_divergent_shuffle |= b->shader->any_cf; + + /* Lane ID guaranteed to be uniform */ + return agx_quad_shuffle_to(b, dst, agx_src_index(&instr->src[0]), + agx_src_index(&instr->src[1])); + } + case nir_intrinsic_ballot: { return agx_ballot_to(b, dst, agx_src_index(&instr->src[0])); }