Skip to content

Commit

Permalink
Add enum options to target lowering tablegen (nod-ai#121)
Browse files Browse the repository at this point in the history
-- Add enum options to target lowering tablegen.

Signed-off-by: Jorn Tuyls <[email protected]>
  • Loading branch information
jtuyls authored Feb 6, 2024
1 parent 049a943 commit 714c015
Showing 1 changed file with 18 additions and 2 deletions.
20 changes: 18 additions & 2 deletions compiler/plugins/target/AMD-AIE/iree-amd-aie/Transforms/Passes.td
Original file line number Diff line number Diff line change
Expand Up @@ -66,7 +66,15 @@ def AMDAIELowerExecutableTarget :
Option<"usePassPipeline", "use-pass-pipeline",
"mlir::iree_compiler::AMDAIE::AIEPassPipeline",
/*default=*/"mlir::iree_compiler::AMDAIE::AIEPassPipeline::PadPipeline",
"Pass pipeline to use while lowering to AIR dialect">
"Pass pipeline to use while lowering to AIR dialect",
[{::llvm::cl::values(
clEnumValN(mlir::iree_compiler::AMDAIE::AIEPassPipeline::PadPipeline, "pad",
"Use the pad-based lowering strategy."),
clEnumValN(mlir::iree_compiler::AMDAIE::AIEPassPipeline::PackPipeline, "pack",
"Use the more advanced pack-based lowering strategy, including peeling and double-buffering."),
clEnumValN(mlir::iree_compiler::AMDAIE::AIEPassPipeline::SimplePackPipeline, "simple-pack",
"Use the simple-pack based lowering strategy.")
)}]>
];
}

Expand All @@ -78,7 +86,15 @@ def AMDAIELoweringStrategy :
Option<"usePassPipeline", "use-pass-pipeline",
"mlir::iree_compiler::AMDAIE::AIEPassPipeline",
/*default=*/"mlir::iree_compiler::AMDAIE::AIEPassPipeline::PadPipeline",
"Pass pipeline to use while lowering to AIR dialect">
"Pass pipeline to use while lowering to AIR dialect",
[{::llvm::cl::values(
clEnumValN(mlir::iree_compiler::AMDAIE::AIEPassPipeline::PadPipeline, "pad",
"Use the pad-based lowering strategy."),
clEnumValN(mlir::iree_compiler::AMDAIE::AIEPassPipeline::PackPipeline, "pack",
"Use the more advanced pack-based lowering strategy, including peeling and double-buffering."),
clEnumValN(mlir::iree_compiler::AMDAIE::AIEPassPipeline::SimplePackPipeline, "simple-pack",
"Use the simple-pack based lowering strategy.")
)}]>
];
}

Expand Down

0 comments on commit 714c015

Please sign in to comment.