Add scf-to-affine pass

UIUC-ChenLab · Jan 14, 2024 · 496c4c0 · 496c4c0
1 parent 6e9453a
commit 496c4c0
Show file tree

Hide file tree

Showing 4 changed files with 1,037 additions and 0 deletions.
diff --git a/include/scalehls/Transforms/Passes.h b/include/scalehls/Transforms/Passes.h
@@ -28,6 +28,7 @@ namespace scalehls {
 
 std::unique_ptr<Pass> createConvertLinalgToDataflowPass();
 std::unique_ptr<Pass> createConvertDataflowToFuncPass();
+std::unique_ptr<Pass> createRaiseSCFToAffinePass();
 
 //===----------------------------------------------------------------------===//
 // Transform Passes

diff --git a/include/scalehls/Transforms/Passes.td b/include/scalehls/Transforms/Passes.td
@@ -25,6 +25,11 @@ def ConvertDataflowToFunc :
   let constructor = "mlir::scalehls::createConvertDataflowToFuncPass()";
 }
 
+def RaiseSCFToAffine : Pass<"scalehls-raise-scf-to-affine"> {
+  let summary = "Raise SCF to affine";
+  let constructor = "mlir::scalehls::createRaiseSCFToAffinePass()";
+}
+
 //===----------------------------------------------------------------------===//
 // Transform Passes
 //===----------------------------------------------------------------------===//

diff --git a/lib/Transforms/CMakeLists.txt b/lib/Transforms/CMakeLists.txt
@@ -5,6 +5,7 @@ add_mlir_library(MLIRScaleHLSTransforms
   GenerateRuntimeFunc.cpp
   LowerCopyToAffineLoops.cpp
   Pipelines.cpp
+  RaiseSCFToAffine.cpp
 
   DEPENDS
   MLIRScaleHLSTransformsIncGen