File tree Expand file tree Collapse file tree 1 file changed +3
-4
lines changed
mlir/lib/Dialect/Linalg/IR Expand file tree Collapse file tree 1 file changed +3
-4
lines changed Original file line number Diff line number Diff 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
23252325void 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// ===----------------------------------------------------------------------===//
You can’t perform that action at this time.
0 commit comments