[Mlir-commits] [mlir] [mlir][gpu] Add 'cluster_stride' attribute to gpu.subgroup_reduce (PR #107142)

Jakub Kuderski llvmlistbot at llvm.org
Tue Sep 3 14:16:08 PDT 2024


================
@@ -579,7 +579,7 @@ class GPUSubgroupReduceConversion final
   LogicalResult
   matchAndRewrite(gpu::SubgroupReduceOp op, OpAdaptor adaptor,
                   ConversionPatternRewriter &rewriter) const override {
-    if (op.getClusterSize())
+    if (op.getClusterSize() || op.getClusterStride() != 1)
----------------
kuhar wrote:

Also here

https://github.com/llvm/llvm-project/pull/107142


More information about the Mlir-commits mailing list