Skip to content

Commit 75622e3

Browse files
committed
[MLIR] Define getArgument() for Toy tutorial passes
This is important during debugging to be able to dump a pass pipeline. It is also what is used by `--mlir-print-ir-tree-dir` to compute filenames during dumps.
1 parent d459784 commit 75622e3

File tree

9 files changed

+9
-0
lines changed

9 files changed

+9
-0
lines changed

mlir/examples/toy/Ch4/mlir/ShapeInferencePass.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -55,6 +55,7 @@ namespace {
5555
struct ShapeInferencePass
5656
: public mlir::PassWrapper<ShapeInferencePass, OperationPass<toy::FuncOp>> {
5757
MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(ShapeInferencePass)
58+
StringRef getArgument() const override { return "toy-shape-inference"; }
5859

5960
void runOnOperation() override {
6061
auto f = getOperation();

mlir/examples/toy/Ch5/mlir/LowerToAffineLoops.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -328,6 +328,7 @@ namespace {
328328
struct ToyToAffineLoweringPass
329329
: public PassWrapper<ToyToAffineLoweringPass, OperationPass<ModuleOp>> {
330330
MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(ToyToAffineLoweringPass)
331+
StringRef getArgument() const override { return "toy-to-affine"; }
331332

332333
void getDependentDialects(DialectRegistry &registry) const override {
333334
registry.insert<affine::AffineDialect, func::FuncDialect,

mlir/examples/toy/Ch5/mlir/ShapeInferencePass.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -55,6 +55,7 @@ namespace {
5555
struct ShapeInferencePass
5656
: public mlir::PassWrapper<ShapeInferencePass, OperationPass<toy::FuncOp>> {
5757
MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(ShapeInferencePass)
58+
StringRef getArgument() const override { return "toy-shape-inference"; }
5859

5960
void runOnOperation() override {
6061
auto f = getOperation();

mlir/examples/toy/Ch6/mlir/LowerToAffineLoops.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -328,6 +328,7 @@ namespace {
328328
struct ToyToAffineLoweringPass
329329
: public PassWrapper<ToyToAffineLoweringPass, OperationPass<ModuleOp>> {
330330
MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(ToyToAffineLoweringPass)
331+
StringRef getArgument() const override { return "toy-to-affine"; }
331332

332333
void getDependentDialects(DialectRegistry &registry) const override {
333334
registry.insert<affine::AffineDialect, func::FuncDialect,

mlir/examples/toy/Ch6/mlir/LowerToLLVM.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -184,6 +184,7 @@ namespace {
184184
struct ToyToLLVMLoweringPass
185185
: public PassWrapper<ToyToLLVMLoweringPass, OperationPass<ModuleOp>> {
186186
MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(ToyToLLVMLoweringPass)
187+
StringRef getArgument() const override { return "toy-to-llvm"; }
187188

188189
void getDependentDialects(DialectRegistry &registry) const override {
189190
registry.insert<LLVM::LLVMDialect, scf::SCFDialect>();

mlir/examples/toy/Ch6/mlir/ShapeInferencePass.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -55,6 +55,7 @@ namespace {
5555
struct ShapeInferencePass
5656
: public mlir::PassWrapper<ShapeInferencePass, OperationPass<toy::FuncOp>> {
5757
MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(ShapeInferencePass)
58+
StringRef getArgument() const override { return "toy-shape-inference"; }
5859

5960
void runOnOperation() override {
6061
auto f = getOperation();

mlir/examples/toy/Ch7/mlir/LowerToAffineLoops.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -328,6 +328,7 @@ namespace {
328328
struct ToyToAffineLoweringPass
329329
: public PassWrapper<ToyToAffineLoweringPass, OperationPass<ModuleOp>> {
330330
MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(ToyToAffineLoweringPass)
331+
StringRef getArgument() const override { return "toy-to-affine"; }
331332

332333
void getDependentDialects(DialectRegistry &registry) const override {
333334
registry.insert<affine::AffineDialect, func::FuncDialect,

mlir/examples/toy/Ch7/mlir/LowerToLLVM.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -184,6 +184,7 @@ namespace {
184184
struct ToyToLLVMLoweringPass
185185
: public PassWrapper<ToyToLLVMLoweringPass, OperationPass<ModuleOp>> {
186186
MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(ToyToLLVMLoweringPass)
187+
StringRef getArgument() const override { return "toy-to-llvm"; }
187188

188189
void getDependentDialects(DialectRegistry &registry) const override {
189190
registry.insert<LLVM::LLVMDialect, scf::SCFDialect>();

mlir/examples/toy/Ch7/mlir/ShapeInferencePass.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -55,6 +55,7 @@ namespace {
5555
struct ShapeInferencePass
5656
: public mlir::PassWrapper<ShapeInferencePass, OperationPass<toy::FuncOp>> {
5757
MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(ShapeInferencePass)
58+
StringRef getArgument() const override { return "toy-shape-inference"; }
5859

5960
void runOnOperation() override {
6061
auto f = getOperation();

0 commit comments

Comments
 (0)