From c17fcbaa2fe39083e8a72e44ec6290ba130b44f9 Mon Sep 17 00:00:00 2001
From: Alyssa Rosenzweig <alyssa@rosenzweig.io>
Date: Sun, 11 Sep 2022 12:03:15 -0400
Subject: [PATCH] agx: Account for mask when writing registers

To use fewer registers.

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

diff --git a/src/asahi/compiler/agx_register_allocate.c b/src/asahi/compiler/agx_register_allocate.c
index 1b9780a3f74..0c6cbd0a700 100644
--- a/src/asahi/compiler/agx_register_allocate.c
+++ b/src/asahi/compiler/agx_register_allocate.c
@@ -41,8 +41,7 @@ agx_write_registers(agx_instr *I, unsigned d)
    case AGX_OPCODE_TEXTURE_LOAD:
    case AGX_OPCODE_TEXTURE_SAMPLE:
    case AGX_OPCODE_LD_TILE:
-      /* TODO: mask */
-      return 4 * size;
+      return util_bitcount(I->mask) * size;
 
    case AGX_OPCODE_LDCF:
       return 6;