Module: Mesa Branch: main Commit: c34aa784961aac34732e9f190b20769b2af98c35 URL: http://cgit.freedesktop.org/mesa/mesa/commit/?id=c34aa784961aac34732e9f190b20769b2af98c35
Author: Timur Kristóf <[email protected]> Date: Tue Jan 18 16:37:16 2022 +0100 radv: Set wave size for task shaders. Signed-off-by: Timur Kristóf <[email protected]> Reviewed-by: Rhys Perry <[email protected]> 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; }
