nvk: Use the correct root descriptor table size for CmdDispatch

Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/24326>

diff --git a/src/nouveau/vulkan/nvk_cmd_dispatch.c b/src/nouveau/vulkan/nvk_cmd_dispatch.c
index dd26db3..4e758b5 100644 (file)
--- a/src/nouveau/vulkan/nvk_cmd_dispatch.c
+++ b/src/nouveau/vulkan/nvk_cmd_dispatch.c
@@ -118,8 +118,8 @@ nvk_CmdDispatch(VkCommandBuffer commandBuffer,
    memcpy(qmd, pipeline->qmd_template, 256);
    gv100_compute_setup_launch_desc(qmd, groupCountX, groupCountY, groupCountZ);
 
-   gp100_cp_launch_desc_set_cb(qmd, 0, 256, root_table_addr);
-   gp100_cp_launch_desc_set_cb(qmd, 1, 256, root_table_addr);
+   gp100_cp_launch_desc_set_cb(qmd, 0, root_table_size, root_table_addr);
+   gp100_cp_launch_desc_set_cb(qmd, 1, root_table_size, root_table_addr);
 
    P_MTHD(cmd->push, NVA0C0, INVALIDATE_SHADER_CACHES_NO_WFI);
    P_NVA0C0_INVALIDATE_SHADER_CACHES_NO_WFI(cmd->push, { .constant = CONSTANT_TRUE });