Debug prints for RaiseTolinalg and 2. SelectFunc pass to process just a given function

Author: arpitj1

include/polygeist/Passes/Passes.h

@@ -22,6 +22,7 @@ class PatternRewriter;
 class RewritePatternSet;
 class DominanceInfo;
 namespace polygeist {
+std::unique_ptr<Pass> createSelectFuncPass();
 std::unique_ptr<Pass> createParallelLICMPass();
 std::unique_ptr<Pass> createPolygeistMem2RegPass();
 std::unique_ptr<Pass> createLoopRestructurePass();

include/polygeist/Passes/Passes.td

@@ -4,6 +4,17 @@
 include "mlir/Pass/PassBase.td"
 include "mlir/Rewrite/PassUtil.td"
 
+def SelectFunc : Pass<"select-func"> {
+  let summary = "Run a pass pipeline on selected functions by name";
+  let constructor = "mlir::polygeist::createSelectFuncPass()";
+  let options = [
+    Option<"pipeline", "pipeline", "std::string", /*default=*/"\"\"",
+           "The pass pipeline to run on filtered functions">,
+    ListOption<"funcNames", "func-name", "std::string",
+               "Function names to process (if empty, process all)">
+  ];
+}
+
 def AffineCFG : Pass<"affine-cfg"> {
   let summary = "Replace scf.if and similar with affine.if";
   let constructor = "mlir::polygeist::replaceAffineCFGPass()";

lib/polygeist/Passes/CMakeLists.txt

@@ -1,5 +1,6 @@
 add_mlir_dialect_library(MLIRPolygeistTransforms
   ConvertToOpaquePtr.cpp
+  SelectFunc.cpp
   AffineCFG.cpp
   AffineReduction.cpp
   CanonicalizeFor.cpp