File tree 2 files changed +6
-3
lines changed
include/mlir/Dialect/GPU/Transforms
lib/Dialect/GPU/Transforms
2 files changed +6
-3
lines changed Original file line number Diff line number Diff line change @@ -88,11 +88,14 @@ void populateGpuLowerClusteredSubgroupReduceToDPPPatterns(
88
88
PatternBenefit benefit = 1 );
89
89
90
90
// / Collect all patterns to rewrite ops within the GPU dialect.
91
- inline void populateGpuRewritePatterns (RewritePatternSet &patterns) {
91
+ inline void populateGpuRewritePatterns (RewritePatternSet &patterns,
92
+ bool includeSubgroupIdRewrite = false ) {
92
93
populateGpuAllReducePatterns (patterns);
93
94
populateGpuGlobalIdPatterns (patterns);
94
95
populateGpuShufflePatterns (patterns);
95
- populateGpuSubgroupIdPatterns (patterns);
96
+ if (includeSubgroupIdRewrite) {
97
+ populateGpuSubgroupIdPatterns (patterns);
98
+ }
96
99
}
97
100
98
101
namespace gpu {
Original file line number Diff line number Diff line change 1
- // ===- SubgroupIdRewriter.cpp - Implementation of SubgroupId rewriting ----===//
1
+ // ===- SubgroupIdRewriter.cpp - Implementation of SubgroupId rewriting ----===//
2
2
//
3
3
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4
4
// See https://llvm.org/LICENSE.txt for license information.
You can’t perform that action at this time.
0 commit comments