Spatial Task Placement for Multi-CGRA

include/TaskflowDialect/TaskflowPasses.h

@@ -17,6 +17,7 @@ namespace taskflow {
 #include "TaskflowDialect/TaskflowPasses.h.inc"
 std::unique_ptr<mlir::Pass> createConstructHyperblockFromTaskPass();
 std::unique_ptr<mlir::Pass> createClassifyCountersPass();
+std::unique_ptr<mlir::Pass> createMapCTOnCGRAArrayPass();
 
 //=========================================================//
 // Optimization Passes

include/TaskflowDialect/TaskflowPasses.td

@@ -45,4 +45,16 @@ def ClassifyCounters : Pass<"classify-counters", "ModuleOp">{
   }];
   let constructor = "taskflow::createClassifyCountersPass()";
 }
+
+def MapCTOnCGRAArray : Pass<"map-ct-on-cgra-array", "func::FuncOp"> {
+  let summary = "Maps Canonical Tasks (CTs) onto a 2D CGRA grid array";
+  let description = [{
+    This pass maps Canonical Tasks (CTs) onto a 2D CGRA grid array.
+    Fusion candidates (same-header SSA dependencies) are placed on adjacent
+    CGRAs to enable direct data forwarding.
+
+    Uses a default 4x4 CGRA grid.
+  }];
+  let constructor = "taskflow::createMapCTOnCGRAArrayPass()";
+}
 #endif // TASKFLOW_PASSES_TD

lib/TaskflowDialect/Transforms/CMakeLists.txt

@@ -3,6 +3,7 @@ get_property(dialect_libs GLOBAL PROPERTY MLIR_DIALECT_LIBS)
 add_mlir_library(MLIRTaskflowTransforms
     ConstructHyperblockFromTaskPass.cpp
     ClassifyCountersPass.cpp
+    MapCTOnCGRAArrayPass.cpp
 
     DEPENDS
     MLIRTaskflowTransformsIncGen