search

nod-ai / iree-amd-aie

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

[DmaLoopSubsumption] Relax circular dma loop subsumption condition #826

Closed yzhang93 closed 1 month ago

yzhang93 commented 1 month ago
  1. This PR relaxes the condition for circular dma ops loop subsumption, so that npu.circular_dma_cpy_nd ops can be hoisted out of the loop even if there is other npu.dma_cpy_nd user of the same connection op after it.
  2. With this change, we can further subsume loops and hoist npu.dma_cpy_nd ops out of the loop. This PR makes use of https://github.com/nod-ai/iree-amd-aie/pull/812 and brings the dma optimizations in Passes.cpp.