Skip to content
Closed
Show file tree
Hide file tree
Changes from 5 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 6 additions & 0 deletions include/TaskflowDialect/TaskflowPasses.h
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,7 @@

#include "TaskflowDialect/TaskflowDialect.h"
#include "TaskflowDialect/TaskflowOps.h"
#include "mlir/Dialect/Func/IR/FuncOps.h"
#include "mlir/Pass/Pass.h"
#include "mlir/Pass/PassManager.h"
#include "mlir/Pass/PassRegistry.h"
Expand All @@ -23,6 +24,11 @@ std::unique_ptr<mlir::Pass> createConstructHyperblockFromTaskPass();
std::unique_ptr<mlir::Pass> createClassifyCountersPass();
std::unique_ptr<mlir::Pass> createMapTaskOnCgraPass();

// Runs the CGRA task placement logic directly on a function.
// grid_rows/grid_cols default to 4x4 (kCgraGridRows/kCgraGridCols).
void runMapTaskOnCgra(mlir::func::FuncOp func,
int grid_rows = 4, int grid_cols = 4);

//=========================================================//
// Optimization Passes
//=========================================================//
Expand Down
Loading