WIP: Raise to linalg

include/polygeist/Ops.h

@@ -57,6 +57,8 @@ bool mayAlias(mlir::MemoryEffects::EffectInstance a,
 
 bool mayAlias(mlir::MemoryEffects::EffectInstance a, mlir::Value b);
 
+bool mayAlias(mlir::Value v, mlir::Value v2);
+
 extern llvm::cl::opt<bool> BarrierOpt;
 
 template <bool NotTopLevel = false>

include/polygeist/Passes/Passes.h

@@ -31,6 +31,7 @@ std::unique_ptr<Pass> replaceAffineCFGPass();
 std::unique_ptr<Pass> createOpenMPOptPass();
 std::unique_ptr<Pass> createCanonicalizeForPass();
 std::unique_ptr<Pass> createRaiseSCFToAffinePass();
+std::unique_ptr<Pass> createRaiseAffineToLinalgPass();
 std::unique_ptr<Pass> createCPUifyPass(StringRef method = "");
 std::unique_ptr<Pass> createBarrierRemovalContinuation();
 std::unique_ptr<Pass> detectReductionPass();
@@ -123,6 +124,10 @@ namespace affine {
 class AffineDialect;
 }
 
+namespace linalg {
+class LinalgDialect;
+}
+
 namespace LLVM {
 class LLVMDialect;
 }

include/polygeist/Passes/Passes.td

@@ -151,6 +151,15 @@ def SCFRaiseToAffine : Pass<"raise-scf-to-affine"> {
   ];
 }
 
+def AffineRaiseToLinalg : Pass<"raise-affine-to-linalg"> {
+  let summary = "Raise affine to linalg";
+  let constructor = "mlir::polygeist::createRaiseAffineToLinalgPass()";
+  let dependentDialects = [
+    "affine::AffineDialect",
+    "linalg::LinalgDialect",
+  ];
+}
+
 def SCFCanonicalizeFor : Pass<"canonicalize-scf-for"> {
   let summary = "Run some additional canonicalization for scf::for";
   let constructor = "mlir::polygeist::createCanonicalizeForPass()";

lib/polygeist/Ops.cpp

@@ -784,7 +784,8 @@ bool isStackAlloca(Value v) {
          v.getDefiningOp<memref::AllocOp>() ||
          v.getDefiningOp<LLVM::AllocaOp>();
 }
-static bool mayAlias(Value v, Value v2) {
+
+bool mayAlias(Value v, Value v2) {
   v = getBase(v);
   v2 = getBase(v2);
   if (v == v2)

lib/polygeist/Passes/CMakeLists.txt

@@ -11,6 +11,7 @@ add_mlir_dialect_library(MLIRPolygeistTransforms
   OpenMPOpt.cpp
   BarrierRemovalContinuation.cpp
   RaiseToAffine.cpp
+  RaiseToLinalg.cpp
   ParallelLower.cpp
   TrivialUse.cpp
   ConvertPolygeistToLLVM.cpp