Skip to content

Commit ba7f3ea

Browse files
committed
rename pattern
1 parent 23f53b5 commit ba7f3ea

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

mlir/lib/Dialect/Linalg/IR/LinalgOps.cpp

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2292,8 +2292,8 @@ Speculation::Speculatability BroadcastOp::getSpeculatability() {
22922292
return getGenericSpeculatabilityImpl(cast<LinalgOp>(getOperation()));
22932293
}
22942294

2295-
/// Fold broadcast with broadcast.
2296-
struct FoldBroadcastWithBroadcast : OpRewritePattern<linalg::BroadcastOp> {
2295+
/// Fold back-to-back broadcasts together.
2296+
struct FoldBroadcasts : OpRewritePattern<linalg::BroadcastOp> {
22972297
using OpRewritePattern<linalg::BroadcastOp>::OpRewritePattern;
22982298

22992299
LogicalResult matchAndRewrite(linalg::BroadcastOp broadcastOp,
@@ -2324,8 +2324,7 @@ struct FoldBroadcastWithBroadcast : OpRewritePattern<linalg::BroadcastOp> {
23242324

23252325
void BroadcastOp::getCanonicalizationPatterns(RewritePatternSet &results,
23262326
MLIRContext *context) {
2327-
results.add<EraseIdentityLinalgOp<BroadcastOp>, FoldBroadcastWithBroadcast>(
2328-
context);
2327+
results.add<EraseIdentityLinalgOp<BroadcastOp>, FoldBroadcasts>(context);
23292328
}
23302329

23312330
//===----------------------------------------------------------------------===//

0 commit comments

Comments
 (0)