nod-ai / iree-amd-aie

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

[POC][WIP] Merge prologue and epilogue into main loop #843

Closed newling closed 3 days ago

newling commented 1 month ago

Inserting this pass after tiling is finished, but before vectorization (and before bufferization). This fails only in iree-amdaie-distribute-cores-and-objectfifos I don't think it'd be hard to get past that. I agree that making this robust to pattern matching is hard though. That said, if it were a pass in the tiling pipeline, it wouldn't need to be super robust, because the tiling pipeline isn't general purpose (we control what goes in and can annotate things).

I'm not suggesting this is a better approach than moving vectorization after bufferization and doing function outlining, I think that's good. Just throwing this option out there for future consideration

@Abhishek-Varma @jtuyls @MaheshRavishankar