fdo-mirrors/mesa
1
0
Fork 0
mirror of https://gitlab.freedesktop.org/mesa/mesa.git synced 2026-05-05 05:18:08 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

agx: Fuse ubitfield_extract

Browse source 
Similarly, let's get the win everywhere.

Signed-off-by: Alyssa Rosenzweig <alyssa@rosenzweig.io>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/25498>
This commit is contained in:
Alyssa Rosenzweig 2023-10-01 09:41:57 -04:00 • committed by Marge Bot
parent 5500e02a61
commit 7193849f30
1 changed files with 8 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
src/asahi/compiler/agx_nir_algebraic.py
View file

@ -90,6 +90,13 @@ for start in range(32):
('extr_agx', a, b, start, 0)),
])
fuse_ubfe = []
for bits in range(1, 32):
fuse_ubfe.extend([
(('iand', ('ushr', 'a@32', b), (1 << bits) - 1),
('ubitfield_extract', a, b, bits))
])
# (x * y) + s = (x * y) + (s << 0)
def imad(x, y, z):
return ('imadshl_agx', x, y, z, 0)
@ -165,7 +172,7 @@ def run():
print(nir_algebraic.AlgebraicPass("agx_nir_lower_algebraic_late",
lower_sm5_shift + lower_pack).render())
print(nir_algebraic.AlgebraicPass("agx_nir_fuse_algebraic_late",
fuse_extr + fuse_imad).render())
fuse_extr + fuse_ubfe + fuse_imad).render())
print(nir_algebraic.AlgebraicPass("agx_nir_opt_ixor_bcsel",
ixor_bcsel).render())