Skip to content

Commit 3668826

Browse files
committed
rename pattern
1 parent abf51af commit 3668826

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
@@ -1136,7 +1136,8 @@ class ExtractOpFromLoad final : public OpRewritePattern<vector::ExtractOp> {
11361136
/// ```
11371137
/// memref.store %arg2, %arg0[%arg1] : memref<?xf32>
11381138
/// ```
1139-
class StoreFromSplat final : public OpRewritePattern<vector::StoreOp> {
1139+
class StoreFromSplatOrBroadcast final
1140+
: public OpRewritePattern<vector::StoreOp> {
11401141
public:
11411142
using OpRewritePattern::OpRewritePattern;
11421143

@@ -2245,8 +2246,8 @@ void mlir::vector::populateSinkVectorOpsPatterns(RewritePatternSet &patterns,
22452246

22462247
void mlir::vector::populateSinkVectorMemOpsPatterns(RewritePatternSet &patterns,
22472248
PatternBenefit benefit) {
2248-
patterns.add<ExtractOpFromLoad, StoreFromSplat>(patterns.getContext(),
2249-
benefit);
2249+
patterns.add<ExtractOpFromLoad, StoreFromSplatOrBroadcast>(
2250+
patterns.getContext(), benefit);
22502251
}
22512252

22522253
void mlir::vector::populateChainedVectorReductionFoldingPatterns(

0 commit comments

Comments
 (0)