search

nod-ai / iree-amd-aie

IREE plugin repository for the AMD AIE accelerator
Apache License 2.0
68 stars 29 forks source link

Add passes to convert `memref::CopyOp` and `LinalgExt::PackOp` into `AMDAIE::DmaCpyNdOp` #480

Open jtuyls opened 4 months ago

jtuyls commented 4 months ago

From @nirvedhmeshram:

Dont have to do this in this PR, but it would be good to just have a pass that lowers memcopy to dma because the air::createCopyToDmaPass does a lot of other stuff looking for subviews that we (hopefully) dont need.

(https://github.com/nod-ai/iree-amd-aie/pull/473#discussion_r1657679808)

Additionally, a new pass can be created/the existing AMDAIEPackToDmaPass can be updated to convert LinalgExt::PackOp into AMDAIE::DmaCpyNdOp.

makslevental commented 4 months ago

I'll take this one.