[Mlir-commits] [mlir] [mlir][gpu] Add patterns to break down subgroup reduce (PR #76271)
Kunwar Grover
llvmlistbot at llvm.org
Thu Dec 28 10:38:36 PST 2023
================
@@ -62,6 +63,12 @@ void populateGpuShufflePatterns(RewritePatternSet &patterns);
/// Collect a set of patterns to rewrite all-reduce ops within the GPU dialect.
void populateGpuAllReducePatterns(RewritePatternSet &patterns);
+/// Collect a set of patterns to break down subgroup_reduce ops into smaller
+/// ones supported by the target of size <= `maxShuffleBitwidth`.
----------------
Groverkss wrote:
Can you mention what "size" here is? It looks like it's the number of bits used by the vector?
https://github.com/llvm/llvm-project/pull/76271
More information about the Mlir-commits
mailing list