Mesa (main): radv: Set wave size for task shaders.

[GitLab Mirror]

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

Author: Timur Kristóf <timur.kristof at gmail.com>
Date:   Tue Jan 18 16:37:16 2022 +0100

radv: Set wave size for task shaders.

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/14929>

---

 src/amd/vulkan/radv_pipeline.c | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/amd/vulkan/radv_pipeline.c b/src/amd/vulkan/radv_pipeline.c
index a752658496a..bd23ec721e1 100644
--- a/src/amd/vulkan/radv_pipeline.c
+++ b/src/amd/vulkan/radv_pipeline.c
@@ -3167,6 +3167,8 @@ radv_get_wave_size(struct radv_device *device,  gl_shader_stage stage,
       return info->cs.subgroup_size;
    } else if (stage == MESA_SHADER_FRAGMENT)
       return device->physical_device->ps_wave_size;
+   else if (stage == MESA_SHADER_TASK)
+      return device->physical_device->cs_wave_size;
    else
       return device->physical_device->ge_wave_size;
 }