Skip to content

Commit 2c66b6e

Browse files
[mlir][Linalg] NFC - Expose tiling canonicalization patterns through a populate method
1 parent 507bbc4 commit 2c66b6e

File tree

2 files changed

+8
-1
lines changed

2 files changed

+8
-1
lines changed

mlir/include/mlir/Dialect/Linalg/Transforms/Transforms.h

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -387,6 +387,8 @@ struct LinalgTilingOptions {
387387
/// when tiling is called programmatically.
388388
OwningRewritePatternList
389389
getLinalgTilingCanonicalizationPatterns(MLIRContext *ctx);
390+
void populateLinalgTilingCanonicalizationPatterns(
391+
OwningRewritePatternList &patterns, MLIRContext *ctx);
390392

391393
struct LinalgBaseTilingPattern : public RewritePattern {
392394
// Entry point to match any LinalgOp OpInterface.

mlir/lib/Dialect/Linalg/Transforms/Tiling.cpp

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -554,6 +554,12 @@ class RewritePatternList<OpTy, OpTypes...> {
554554
OwningRewritePatternList
555555
mlir::linalg::getLinalgTilingCanonicalizationPatterns(MLIRContext *ctx) {
556556
OwningRewritePatternList patterns;
557+
populateLinalgTilingCanonicalizationPatterns(patterns, ctx);
558+
return patterns;
559+
}
560+
561+
void mlir::linalg::populateLinalgTilingCanonicalizationPatterns(
562+
OwningRewritePatternList &patterns, MLIRContext *ctx) {
557563
AffineApplyOp::getCanonicalizationPatterns(patterns, ctx);
558564
AffineForOp::getCanonicalizationPatterns(patterns, ctx);
559565
AffineMinOp::getCanonicalizationPatterns(patterns, ctx);
@@ -569,7 +575,6 @@ mlir::linalg::getLinalgTilingCanonicalizationPatterns(MLIRContext *ctx) {
569575
#define GET_OP_LIST
570576
#include "mlir/Dialect/Linalg/IR/LinalgStructuredOps.cpp.inc"
571577
>::insert(patterns, ctx);
572-
return patterns;
573578
}
574579

575580
/// Populate the given list with patterns that apply Linalg tiling.

0 commit comments

Comments
 (0)