add starter poly-to-llvm pipeline

j2kun · Oct 30, 2023 · a54f23f · a54f23f
1 parent 48f8909
commit a54f23f
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/tools/tutorial-opt.cpp b/tools/tutorial-opt.cpp
@@ -7,6 +7,13 @@
 #include "mlir/include/mlir/Pass/PassManager.h"
 #include "mlir/include/mlir/Pass/PassRegistry.h"
 #include "mlir/include/mlir/Tools/mlir-opt/MlirOptMain.h"
+#include "mlir/include/mlir/Transforms/Passes.h"
+
+void polyToLLVMPipelineBuilder(mlir::OpPassManager &manager) {
+  // Poly
+  manager.addPass(mlir::tutorial::poly::createPolyToStandard());
+  manager.addPass(mlir::createCanonicalizerPass());
+}
 
 int main(int argc, char **argv) {
   mlir::DialectRegistry registry;
@@ -20,6 +27,10 @@ int main(int argc, char **argv) {
   // Dialect conversion passes
   mlir::tutorial::poly::registerPolyToStandardPasses();
 
+  mlir::PassPipelineRegistration<>("poly-to-llvm",
+                             "Run passes to lower the poly dialect to LLVM",
+                             polyToLLVMPipelineBuilder);
+
   return mlir::asMainReturnCode(
       mlir::MlirOptMain(argc, argv, "Tutorial Pass Driver", registry));
 }