nod-ai / iree-amd-aie

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

Refactor: lower DMAs to connection + controlcode DMA ops #745

Closed jtuyls closed 2 months ago

jtuyls commented 2 months ago

In the effort to enable control packets, a first step is to extract aie.flow logic from AMDAIEStatefulTransform to enable packet flow ops on amdaie. This PR adds amdaie.connection which resembles aie.flow and lowers circular DMA ops (amdaie.circular_dma_cpy_nd) to a connection (amdaie.connection) + control code (amdaie.npu.circular_dma_cpy_nd). Note that this also benefits the effort to move all/most DMA/PM/... configuration logic into the control code instead of the xclbin.