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

agx: Translate simple subgroup ops

Browse Source 
We'll use these for optimizing parallel prefix sums.

Signed-off-by: Alyssa Rosenzweig <alyssa@rosenzweig.io>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/26056>
This commit is contained in:
Alyssa Rosenzweig
2023-08-28 20:48:10 -04:00
committed by Marge Bot
parent 77bb446e90
commit a74fbb3840
1 changed files with 20 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
src/asahi/compiler/agx_compile.c
View File

@@ -1207,6 +1207,26 @@ agx_emit_intrinsic(agx_builder *b, nir_intrinsic_instr *instr)
return NULL; return NULL;
} }
case nir_intrinsic_reduce: {
assert(nir_intrinsic_reduction_op(instr) == nir_op_iadd &&
"other reductions todo");
return agx_simd_iadd_to(b, dst, agx_src_index(&instr->src[0]));
}
case nir_intrinsic_exclusive_scan: {
assert(nir_intrinsic_reduction_op(instr) == nir_op_iadd &&
"other reductions todo");
return agx_simd_prefix_iadd_to(b, dst, agx_src_index(&instr->src[0]));
}
case nir_intrinsic_read_invocation: {
/* Lane ID guaranteed to be uniform */
return agx_simd_shuffle_to(b, dst, agx_src_index(&instr->src[0]),
agx_src_index(&instr->src[1]));
}
case nir_intrinsic_load_barycentric_sample: case nir_intrinsic_load_barycentric_sample:
case nir_intrinsic_load_sample_id: case nir_intrinsic_load_sample_id:
case nir_intrinsic_load_sample_pos: case nir_intrinsic_load_sample_pos: