AlexIndustrial/mesa
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

nir/lower_subgroups: add build_cluster_mask helper

Browse Source 
This functionality will become more complex in the next commit so
separate it into a helper function.

Signed-off-by: Job Noorman <jnoorman@igalia.com>
Reviewed-by: Alyssa Rosenzweig <alyssa@rosenzweig.io>
Reviewed-by: Faith Ekstrand <faith.ekstrand@collabora.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/31587>
This commit is contained in:
Job Noorman
2024-10-17 21:44:39 +02:00
committed by Marge Bot
parent e0cb4a94a3
commit 58b199f7ed
1 changed files with 11 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/compiler/nir/nir_lower_subgroups.c
+11 -6
View File
@@ -803,6 +803,16 @@ build_scan_reduce(nir_builder *b, nir_intrinsic_op op, nir_op red_op,
}
}
static nir_def *
build_cluster_mask(nir_builder *b, unsigned cluster_size)
{
nir_def *idx = nir_load_subgroup_invocation(b);
nir_def *cluster = nir_iand_imm(b, idx, ~(uint64_t)(cluster_size - 1));
nir_def *cluster_mask = nir_imm_int(b, BITFIELD_MASK(cluster_size));
return nir_ishl(b, cluster_mask, cluster);
}
static nir_def *
lower_scan_reduce(nir_builder *b, nir_intrinsic_instr *intrin,
unsigned subgroup_size)
@@ -830,12 +840,7 @@ lower_scan_reduce(nir_builder *b, nir_intrinsic_instr *intrin,
{
/* Mask according to the cluster size */
if (cluster_size < subgroup_size) {
nir_def *idx = nir_load_subgroup_invocation(b);
nir_def *cluster = nir_iand_imm(b, idx, ~(uint64_t)(cluster_size - 1));
nir_def *cluster_mask = nir_imm_int(b, BITFIELD_MASK(cluster_size));
cluster_mask = nir_ishl(b, cluster_mask, cluster);
nir_def *cluster_mask = build_cluster_mask(b, cluster_size);
mask = nir_iand(b, mask, cluster_mask);
}