radeonsi: use wave64 for KHR_shader_subgroup enabled shader

Signed-off-by: Qiang Yu <yuq825@gmail.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/30610>
This commit is contained in:
Qiang Yu
2024-08-06 15:11:58 +08:00
parent a78d1d49e6
commit 1ee612e1ac

View File

@@ -46,6 +46,11 @@ unsigned si_determine_wave_size(struct si_screen *sscreen, struct si_shader *sha
(stage == MESA_SHADER_GEOMETRY && !shader->key.ge.as_ngg))
return 64;
/* For KHR_shader_subgroup which require a constant subgroup size known by user. */
if (info->base.subgroup_size == SUBGROUP_SIZE_API_CONSTANT ||
(prev_sel && prev_sel->info.base.subgroup_size == SUBGROUP_SIZE_API_CONSTANT))
return 64;
/* Workgroup sizes that are not divisible by 64 use Wave32. */
if (stage == MESA_SHADER_COMPUTE && !info->base.workgroup_size_variable &&
(info->base.workgroup_size[0] *