Skip to content

Commit bbde763

Browse files
committed
remove
1 parent fbe3bd1 commit bbde763

File tree

2 files changed

+6
-3
lines changed

2 files changed

+6
-3
lines changed

mlir/include/mlir/Dialect/GPU/Transforms/Passes.h

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -88,11 +88,14 @@ void populateGpuLowerClusteredSubgroupReduceToDPPPatterns(
8888
PatternBenefit benefit = 1);
8989

9090
/// 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) {
9293
populateGpuAllReducePatterns(patterns);
9394
populateGpuGlobalIdPatterns(patterns);
9495
populateGpuShufflePatterns(patterns);
95-
populateGpuSubgroupIdPatterns(patterns);
96+
if (includeSubgroupIdRewrite) {
97+
populateGpuSubgroupIdPatterns(patterns);
98+
}
9699
}
97100

98101
namespace gpu {

mlir/lib/Dialect/GPU/Transforms/SubgroupIdRewriter.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
//===- SubgroupIdRewriter.cpp - Implementation of SubgroupId rewriting ----===//
1+
//===- SubgroupIdRewriter.cpp - Implementation of SubgroupId rewriting ----===//
22
//
33
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
44
// See https://llvm.org/LICENSE.txt for license information.

0 commit comments

Comments
 (0)