Fix ci tests

nod-ai · Aug 30, 2024 · b3196b5 · b3196b5
1 parent 64df148
commit b3196b5
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/build_tools/ci/print_ir_aie2xclbin/print_ir_aie2xclbin.sh b/build_tools/ci/print_ir_aie2xclbin/print_ir_aie2xclbin.sh
@@ -146,6 +146,7 @@ IREE_COMPILE_COMMAND="${IREE_COMPILE_EXE} \
 ${SOURCE_MLIR_FILE} \
 --compile-mode=hal-executable \
 --iree-hal-target-backends=amd-aie \
+--iree-amdaie-lower-to-aie-pipeline=air \
 --iree-amd-aie-peano-install-dir=${PEANO} \
 --iree-amd-aie-install-dir=${IREE_INSTALL_DIR} \
 --iree-hal-dump-executable-intermediates-to=${OUTPUT} \
@@ -168,6 +169,7 @@ IREE_COMPILE_COMMAND="${IREE_COMPILE_EXE} \
 ${SOURCE_MLIR_FILE} \
 --compile-mode=hal-executable \
 --iree-hal-target-backends=amd-aie \
+--iree-amdaie-lower-to-aie-pipeline=air \
 --iree-amd-aie-peano-install-dir=${PEANO} \
 --iree-amd-aie-install-dir=${IREE_INSTALL_DIR} \
 --iree-hal-dump-executable-intermediates-to=${OUTPUT} \

diff --git a/compiler/plugins/target/AMD-AIE/iree-amd-aie/Transforms/Passes.td b/compiler/plugins/target/AMD-AIE/iree-amd-aie/Transforms/Passes.td
@@ -299,6 +299,16 @@ def AMDAIELowerExecutableTarget :
                    "Use the pad-pack based lowering strategy for matmul-like ops."),
         clEnumValN(mlir::iree_compiler::AMDAIE::TilePassPipeline::ConvDecomposePipeline, "conv-decompose",
                    "Use the conv-decompose based lowering strategy for convolution interface ops.")
+      )}]>,
+    Option<"useLowerToAIEPipeline", "use-lower-to-aie-pipeline",
+      "mlir::iree_compiler::AMDAIE::LowerToAIEPassPipeline",
+      /*default=*/"mlir::iree_compiler::AMDAIE::LowerToAIEPassPipeline::ObjectFifo",
+      "Lowering pass pipeline to use",
+      [{::llvm::cl::values(
+        clEnumValN(mlir::iree_compiler::AMDAIE::LowerToAIEPassPipeline::ObjectFifo, "objectFifo",
+                   "Use the IREE lowering to objectFifos"),
+        clEnumValN(mlir::iree_compiler::AMDAIE::LowerToAIEPassPipeline::AIR, "air",
+                   "Use the IREE lowering through AIR")
       )}]>
   ];
 }