File tree 3 files changed +10
-10
lines changed
include/mlir/Dialect/GPU/Transforms
lib/Dialect/GPU/Transforms
3 files changed +10
-10
lines changed Original file line number Diff line number Diff line change @@ -66,14 +66,14 @@ void populateGpuAllReducePatterns(RewritePatternSet &patterns);
66
66
// / Collect a set of patterns to break down subgroup_reduce ops into smaller
67
67
// / ones supported by the target of `size <= maxShuffleBitwidth`, where `size`
68
68
// / is the subgroup_reduce value bitwidth.
69
- void populateGpuBreakDownSubgrupReducePatterns (RewritePatternSet &patterns,
70
- unsigned maxShuffleBitwidth = 32 ,
71
- PatternBenefit benefit = 1 );
69
+ void populateGpuBreakDownSubgroupReducePatterns (
70
+ RewritePatternSet &patterns, unsigned maxShuffleBitwidth = 32 ,
71
+ PatternBenefit benefit = 1 );
72
72
73
73
// / Collect a set of patterns to lower `gpu.subgroup_reduce` into `gpu.shuffle`
74
74
// / ops over `shuffleBitwidth` scalar types. Assumes that the subgroup has
75
75
// / `subgroupSize` lanes. Uses the butterfly shuffle algorithm.
76
- void populateGpuLowerSubgroupReduceToShufflePattenrs (
76
+ void populateGpuLowerSubgroupReduceToShufflePatterns (
77
77
RewritePatternSet &patterns, unsigned subgroupSize,
78
78
unsigned shuffleBitwidth = 32 , PatternBenefit benefit = 1 );
79
79
Original file line number Diff line number Diff line change @@ -346,15 +346,15 @@ struct VectorSubgroupReduceToShuffles final
346
346
};
347
347
} // namespace
348
348
349
- void mlir::populateGpuBreakDownSubgrupReducePatterns (
349
+ void mlir::populateGpuBreakDownSubgroupReducePatterns (
350
350
RewritePatternSet &patterns, unsigned maxShuffleBitwidth,
351
351
PatternBenefit benefit) {
352
352
patterns.add <BreakDownSubgroupReduce>(patterns.getContext (),
353
353
maxShuffleBitwidth, benefit);
354
354
patterns.add <ScalarizeSingleElementReduce>(patterns.getContext (), benefit);
355
355
}
356
356
357
- void mlir::populateGpuLowerSubgroupReduceToShufflePattenrs (
357
+ void mlir::populateGpuLowerSubgroupReduceToShufflePatterns (
358
358
RewritePatternSet &patterns, unsigned subgroupSize,
359
359
unsigned shuffleBitwidth, PatternBenefit benefit) {
360
360
patterns.add <ScalarSubgroupReduceToShuffles, VectorSubgroupReduceToShuffles>(
Original file line number Diff line number Diff line change @@ -75,11 +75,11 @@ struct TestGpuSubgroupReduceLoweringPass
75
75
76
76
// Since both pattern sets match on the same ops, set higher benefit to
77
77
// perform fewer failing matches.
78
- populateGpuBreakDownSubgrupReducePatterns (patterns,
79
- /* maxShuffleBitwidth=*/ 32 ,
80
- PatternBenefit (2 ));
78
+ populateGpuBreakDownSubgroupReducePatterns (patterns,
79
+ /* maxShuffleBitwidth=*/ 32 ,
80
+ PatternBenefit (2 ));
81
81
if (expandToShuffles)
82
- populateGpuLowerSubgroupReduceToShufflePattenrs (
82
+ populateGpuLowerSubgroupReduceToShufflePatterns (
83
83
patterns, /* subgroupSize=*/ 32 , /* shuffleBitwidth=*/ 32 );
84
84
85
85
(void )applyPatternsAndFoldGreedily (getOperation (), std::move (patterns));
You can’t perform that action at this time.
0 commit comments