nod-ai / iree-amd-aie

IREE plugin repository for the AMD AIE accelerator
Apache License 2.0
69 stars 30 forks source link

[AMDAIEDistributeCoresAndObjectFifos] Factorize out memory privatization sub-pass #841

Closed newling closed 1 month ago

newling commented 1 month ago

By running this sub-pass earlier, it can be made simpler because it doesn't need to handle anything DMA related. Also this sub-pass is quite modular and doesn't need to be part of the larger distribution pass. My initial motivation for this factorization was that it makes running aievec passes before the cores are materialized easier (so can run aievec passes on just a single function, instead of 16 cores).