search

nod-ai / iree-amd-aie

IREE plugin repository for the AMD AIE accelerator
Apache License 2.0
46 stars 23 forks source link

Vendor AIE/AIEX passes (5/n) #423

Closed makslevental closed 1 week ago

makslevental commented 2 weeks ago

This PR is part of a stack of PRs that refactor the dependency on MLIR-AIE. See https://github.com/nod-ai/iree-amd-aie/issues/430 for more information.

This PR vendors all of the AIE/AIEX passes under our own namespace mlir::iree_compiler::AMDAIE.

Notes:

  1. We still depend (until subsequent PRs) on #include "aie/Dialect/AIE/IR/AIEDialect.h" and #include "aie/Dialect/AIEX/IR/AIEXDialect.h" for AIE/AIEX ops (this PR specifically removes all Transforms/AIEPassses.h etc);
  2. It's not worth bikeshedding/auditing this code yet - subsequent PRs refactor many things dramatically (e.g., removing AIEXToStandard and AIEToStandard passes and joining AIEDmaToNpu with AIETargetNPU). This was the point of merging the mlir-aie lit/unit tests early :)

@nirvedhmeshram @newling this is now ready for review.