nod-ai / iree-amd-aie

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

[DmaLoopSubsumption] Add checks for all dims after inserting new access pattern dimension #819

Closed yzhang93 closed 1 month ago

yzhang93 commented 1 month ago

This PR adds checks for all dimensions of new sizes/strides after insertion to make sure they don't exceed the max value. Without this PR, it only checks the inserted dimension, and ignores the case that inter size becomes intra size after insertion which could exceed the max value.