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;