tree f6f4b008b8e0d73d6cd2846439068339b696dab3
parent 21b010130bf1eff01466decf2bf096b367da4dfa
author Manupa Karunaratne <manupa.karunaratne@amd.com> 1736846192 +0000
committer GitHub <noreply@github.com> 1736846192 +0000
gpgsig -----BEGIN PGP SIGNATURE-----
 
 wsFcBAABCAAQBQJnhitwCRC1aQ7uu5UhlAAApN8QAKTVzIph28JBRN1gu6MJ+wwq
 tEiMZgOc1ku1s13RMgk007uIspntTVfgbNJerL/BD0edp/405n//s94Y8+a3yt9G
 OG+Fv742t6DbGU8mQW4OLOguVl356955VfhXVbrx+N2XmqpHVIUe2tjlnyw+cbFh
 hKf3telytwB0ZPBSJtLIo1g9Vh8ajPDJyteIlSSPeTWqXsnZ2hN0nSwI0U/JwXv/
 eE4B+0NqSY4nBjpqDOotIlsPgeY2bj8lY/q3OQquS1zMvXpfKm6LkYLVEAKQ50ba
 tu6BLQqanW53uIjARo98XpnAJURDwjyCeJueySDnDCiRRWQQEE35D6bdL3oB7+sr
 yA0C3ufh+LsCto3ykZJwt862kzIURBfqZUfPWIZCOX93cwE+6J8ajdjmVwGwfVzr
 0Fl6kUgZiB8Lab/nHtHvN++0N+eGyooex0x+fXAMezjm7aOnta1MUlqx+83OOGyD
 kNlAKPB4CPE56xuUQkZoA0nWwJg5OioMU689jfxFA7WXxSyNgoowoRVgAfYaXfaP
 XVc7a9dHiCZBU97r8bZe6IxdK6Erh96jqVsB/Ce0blRQmQXTZ4GdoVkLDkqRMNWJ
 FjF7kPD/fBpUCLXO88lUUhD2XpxngAhsMGPMSKkM29uhYulWYQsw6HbxjgWHzLua
 nRH3pEZV2LaSzEXvkQBD
 =g7EF
 -----END PGP SIGNATURE-----
 

[LLVMGPUVectorDistribute] Add support for inter-subgroup multi_reduction (#19596)

This commit adds support for distribute multi_reductions where the
reduction dimension(s) is/are distributed across subgroups.

We perform the existing reduction distribution, however, we are left
with partial reductions accross subgroups.

Thereafter, we insert tranfer_write / transfer_read to shared memory to
achieve a layout change where
we re-distribute reduction subgroup tiles into element tile. Finally, we
do another multi_reduction to complete the reduction.

closes: https://github.com/iree-org/iree/issues/19578

---------

Signed-off-by: Manupa Karunaratne <manupa.karunaratne@amd.com>