From 3bb7d929f4de31acce36675ab93fbc1e0d46a814 Mon Sep 17 00:00:00 2001
From: Faith Ekstrand <faith.ekstrand@collabora.com>
Date: Thu, 12 Feb 2026 17:13:48 -0500
Subject: [PATCH] panvk/preload: Stop assuming 32 registers

cc: mesa-stable

Reviewed-by: Lars-Ivar Hesselberg Simonsen <lars-ivar.simonsen@arm.com>
Acked-by: Boris Brezillon <boris.brezillon@collabora.com>
Acked-by: Eric R. Smith <eric.smith@collabora.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/39759>
---
 src/panfrost/vulkan/panvk_vX_cmd_fb_preload.c | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/panfrost/vulkan/panvk_vX_cmd_fb_preload.c b/src/panfrost/vulkan/panvk_vX_cmd_fb_preload.c
index 5a3ed7fbe9b..499e3ee99ae 100644
--- a/src/panfrost/vulkan/panvk_vX_cmd_fb_preload.c
+++ b/src/panfrost/vulkan/panvk_vX_cmd_fb_preload.c
@@ -183,7 +183,8 @@ get_preload_shader(struct panvk_device *dev,
    panvk_priv_mem_write_desc(shader->spd, 0, SHADER_PROGRAM, cfg) {
       cfg.stage = MALI_SHADER_STAGE_FRAGMENT;
       cfg.fragment_coverage_bitmask_type = MALI_COVERAGE_BITMASK_TYPE_GL;
-      cfg.register_allocation = MALI_SHADER_REGISTER_ALLOCATION_32_PER_THREAD;
+      cfg.register_allocation =
+         pan_register_allocation(shader->info.work_reg_count);
       cfg.binary = panvk_priv_mem_dev_addr(shader->code_mem);
       cfg.preload.r48_r63 = shader->info.preload >> 48;
    }