From fc4e53370954e5cd48528f12882bd016f5c281d5 Mon Sep 17 00:00:00 2001
From: Alyssa Rosenzweig <alyssa@rosenzweig.io>
Date: Tue, 26 Nov 2024 22:45:44 -0500
Subject: [PATCH] agx: add more 8-bit address fusing rules

helps kernel I'm writing

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

diff --git a/src/asahi/compiler/agx_nir_algebraic.py b/src/asahi/compiler/agx_nir_algebraic.py
index 3af4fb903cf..5041640beb6 100644
--- a/src/asahi/compiler/agx_nir_algebraic.py
+++ b/src/asahi/compiler/agx_nir_algebraic.py
@@ -292,6 +292,11 @@ for s_ in range(1, 5):
 fuse_lea += [
     (('iadd', a, ('u2u64', 'b@32')), ('ulea_agx', a, b, 0)),
     (('iadd', a, ('i2i64', 'b@32')), ('ilea_agx', a, b, 0)),
+
+    (('iadd', a, ('iadd', ('u2u64', 'b@32'), c)),
+     ('ulea_agx', ('iadd', a, c), b, 0)),
+    (('iadd', a, ('iadd', ('i2i64', 'b@32'), c)),
+     ('ilea_agx', ('iadd', a, c), b, 0)),
 ]
 
 # After lowering address arithmetic, the various address arithmetic opcodes are