Bump mlir-air; more generic air compiler pipeline (#855)

nod-ai · Oct 23, 2024 · c198536 · c198536
1 parent bf6505a
commit c198536
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 7 deletions.
diff --git a/compiler/plugins/target/AMD-AIE/iree-amd-aie/Transforms/Passes.cpp b/compiler/plugins/target/AMD-AIE/iree-amd-aie/Transforms/Passes.cpp
@@ -539,7 +539,6 @@ void buildAMDAIETransformPassPipeline(
   });
 }
 
-
 void addAMDAIEObjectFifoLoweringPasses(OpPassManager &passManager,
                                        bool enablePacketFlow,
                                        TilePassPipeline useTilePipeline) {
@@ -728,11 +727,6 @@ void addMLIRAIRLoweringPasses(OpPassManager &passManager, AMDAIEDevice device,
   passManager.addPass(xilinx::air::createAIRDependencyCanonicalizePass());
   passManager.addPass(createCanonicalizerPass());
   passManager.addPass(createCSEPass());
-  // TODO (Erwei): This pass currently doesn't support pack-peel pipeline. This
-  // pass needs to work in order to get multiple AIE columns to work.
-  if (useTilePipeline != TilePassPipeline::PackPeelPipeline)
-    passManager.addNestedPass<func::FuncOp>(
-        xilinx::air::createAIRSplitL2MemrefForBufferConstraintPass());
   passManager.addPass(xilinx::air::createAIRIsolateAsyncDmaLoopNests());
   passManager.addPass(createCanonicalizerPass());
   passManager.addPass(createCSEPass());
@@ -747,6 +741,13 @@ void addMLIRAIRLoweringPasses(OpPassManager &passManager, AMDAIEDevice device,
   }
   passManager.addPass(createCanonicalizerPass());
   passManager.addPass(createCSEPass());
+  passManager.addNestedPass<func::FuncOp>(
+      xilinx::air::createAIRSplitL2MemrefForBufferConstraintPass());
+  passManager.addPass(xilinx::air::createAIRIsolateAsyncDmaLoopNests());
+  passManager.addPass(createCanonicalizerPass());
+  passManager.addPass(createCSEPass());
+  passManager.addPass(createCanonicalizerPass());
+  passManager.addPass(createCSEPass());
   passManager.addNestedPass<func::FuncOp>(
       xilinx::air::createAIRSegmentLoopFusion());
 

diff --git a/third_party/mlir-air b/third_party/mlir-air