Mesa (main): radv: Always use 3D block ID and grid size in task shaders.

[GitLab Mirror]

Module: Mesa
Branch: main
Commit: 08435b1c6f9062ba4cd4e56803dd9e19ea119b3a
URL:    http://cgit.freedesktop.org/mesa/mesa/commit/?id=08435b1c6f9062ba4cd4e56803dd9e19ea119b3a

Author: Timur Kristóf <timur.kristof at gmail.com>
Date:   Mon Feb 28 14:21:36 2022 +0100

radv: Always use 3D block ID and grid size in task shaders.

These are needed to address the task draw and payload ring buffers
when the task shader dispatch is 3 dimensional.

Signed-off-by: Timur Kristóf <timur.kristof at gmail.com>
Reviewed-by: Rhys Perry <pendingchaos02 at gmail.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/17023>

---

 src/amd/vulkan/radv_shader_info.c | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/amd/vulkan/radv_shader_info.c b/src/amd/vulkan/radv_shader_info.c
index eabee36bb64..825b50d7245 100644
--- a/src/amd/vulkan/radv_shader_info.c
+++ b/src/amd/vulkan/radv_shader_info.c
@@ -607,6 +607,9 @@ radv_nir_shader_info_pass(struct radv_device *device, const struct nir_shader *n
 
          /* Needed to address the task draw/payload rings. */
          info->cs.uses_block_id[0] = true;
+         info->cs.uses_block_id[1] = true;
+         info->cs.uses_block_id[2] = true;
+         info->cs.uses_grid_size = true;
 
          /* Needed for storing draw ready only on the 1st thread. */
          info->cs.uses_local_invocation_idx = true;