Introducing task level computation abstraction

include/Conversion/ConversionPasses.h

@@ -21,7 +21,7 @@ std::unique_ptr<mlir::Pass> createLowerBuiltinToNeuraPass();
 std::unique_ptr<mlir::Pass> createLowerAffineToNeuraPass();
 
 // TaskFlow Conversion Passes.
-std::unique_ptr<mlir::Pass> createConvertLinalgToTaskflowPass();
+std::unique_ptr<mlir::Pass> createConvertAffineToTaskflowPass();
 #define GEN_PASS_REGISTRATION
 #include "Conversion/ConversionPasses.h.inc"
 

include/Conversion/ConversionPasses.td

@@ -47,16 +47,29 @@ def LowerAffineToNeura : Pass<"lower-affine-to-neura", "func::FuncOp">{
 //=========================================================//
 // TaskFlow Conversion Passes.
 //=========================================================//
-def ConvertLinalgToTaskflow : Pass<"convert-linalg-to-taskflow", "ModuleOp">{
-  let summary = "Convert Linalg operations to Taskflow dialect";
+
+def ConvertAffineToTaskflow : Pass<"convert-affine-to-taskflow", "ModuleOp">{
+  let summary = "Convert top-level affine.for operations to Taskflow dialect";
   let description = [{
-    Extracts compute-intensive linalg operations and wraps them into a Taskflow graph for spatial architecture execution.
+    This pass converts top-level affine.for loops in a function into
+    taskflow.task operations within a taskflow.graph. Each top-level loop
+    becomes a separate task, and data dependencies between tasks are made
+    explicit through taskflow.channel operations.
+
+    The pass:
+    1. Identifies all top-level affine.for operations
+    2. Analyzes data dependencies (RAW, WAR, WAW) between loops
+    3. Creates a taskflow.graph containing the loops
+    4. Converts each loop to a taskflow.task
+    5. Inserts taskflow.channel operations for data dependencies
   }];
-  let constructor = "mlir::createConvertLinalgToTaskflowPass()";
+
+  let constructor = "mlir::createConvertAffineToTaskflowPass()";
   let dependentDialects = [
     "mlir::taskflow::TaskflowDialect",
-    "mlir::linalg::LinalgDialect",
+    "mlir::affine::AffineDialect",
     "mlir::func::FuncDialect",
+    "mlir::memref::MemRefDialect",
     "mlir::arith::ArithDialect"
   ];
 }

include/NeuraDialect/NeuraPasses.h

@@ -15,7 +15,7 @@ namespace neura {
 
 void registerNeuraConversionPassPipeline();
 
-// Passes defined in GraphPasses.td
+// Passes defined in NeuraPasses.td
 #define GEN_PASS_DECL
 #include "NeuraDialect/NeuraPasses.h.inc"
 std::unique_ptr<mlir::Pass> createInsertDataMovPass();

include/TaskflowDialect/CMakeLists.txt

@@ -1 +1,5 @@
 add_mlir_dialect(Taskflow taskflow)
+
+set(LLVM_TARGET_DEFINITIONS TaskflowPasses.td)
+mlir_tablegen(TaskflowPasses.h.inc --gen-pass-decls)
+add_public_tablegen_target(MLIRTaskflowTransformsIncGen)

include/TaskflowDialect/Taskflow.td

@@ -3,5 +3,7 @@
 
 include "TaskflowDialect.td"
 include "TaskflowOps.td"
+include "TaskflowPasses.td"
+include "TaskflowTypes.td"
 
 #endif // TASKFLOW_TD

include/TaskflowDialect/TaskflowOps.h

@@ -8,6 +8,7 @@
 #include "mlir/IR/DialectImplementation.h"
 #include "mlir/IR/OpDefinition.h"
 #include "mlir/Interfaces/ControlFlowInterfaces.h"
+#include "mlir/Interfaces/InferTypeOpInterface.h"
 #include "mlir/Interfaces/SideEffectInterfaces.h"
 
 // First includes the interface declarations.