Skip to content

Commit 89dbd19

Browse files
committed
rename pattern
1 parent aaf353d commit 89dbd19

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

mlir/lib/Dialect/Vector/Transforms/VectorTransforms.cpp

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1196,7 +1196,8 @@ class ExtractOpFromLoad final : public OpRewritePattern<vector::ExtractOp> {
11961196
/// ```
11971197
/// memref.store %arg2, %arg0[%arg1] : memref<?xf32>
11981198
/// ```
1199-
class StoreFromSplat final : public OpRewritePattern<vector::StoreOp> {
1199+
class StoreFromSplatOrBroadcast final
1200+
: public OpRewritePattern<vector::StoreOp> {
12001201
public:
12011202
using OpRewritePattern::OpRewritePattern;
12021203

@@ -2311,8 +2312,8 @@ void mlir::vector::populateSinkVectorOpsPatterns(RewritePatternSet &patterns,
23112312

23122313
void mlir::vector::populateSinkVectorMemOpsPatterns(RewritePatternSet &patterns,
23132314
PatternBenefit benefit) {
2314-
patterns.add<ExtractOpFromLoad, StoreFromSplat>(patterns.getContext(),
2315-
benefit);
2315+
patterns.add<ExtractOpFromLoad, StoreFromSplatOrBroadcast>(
2316+
patterns.getContext(), benefit);
23162317
}
23172318

23182319
void mlir::vector::populateChainedVectorReductionFoldingPatterns(

0 commit comments

Comments
 (0)