OpenHarmony/third_party_mesa3d
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

agx: optimize first_invocation

Browse Source 
Signed-off-by: Alyssa Rosenzweig <alyssa@rosenzweig.io>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/27616>
This commit is contained in:
Alyssa Rosenzweig
2024-01-12 17:39:12 -04:00
committed by Marge Bot
parent fb81201a34
commit 0de7018879
1 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/asahi/compiler/agx_nir_lower_subgroups.c
View File

@@ -28,6 +28,14 @@ lower(nir_builder *b, nir_intrinsic_instr *intr, void *data)
return true; return true;
} }
case nir_intrinsic_first_invocation: {
nir_def *active_id = nir_load_active_subgroup_invocation_agx(b);
nir_def *is_first = nir_ieq_imm(b, active_id, 0);
nir_def *first_bit = nir_ballot(b, 1, 32, is_first);
nir_def_rewrite_uses(&intr->def, nir_ufind_msb(b, first_bit));
return true;
}
default: default:
return false; return false;
} }
@@ -41,7 +49,6 @@ agx_nir_lower_subgroups(nir_shader *s)
.lower_vote_eq = true, .lower_vote_eq = true,
.lower_vote_bool_eq = true, .lower_vote_bool_eq = true,
.lower_read_first_invocation = true, .lower_read_first_invocation = true,
.lower_first_invocation_to_ballot = true,
.lower_to_scalar = true, .lower_to_scalar = true,
.lower_subgroup_masks = true, .lower_subgroup_masks = true,
.ballot_components = 1, .ballot_components = 1,