[Mlir-commits] [mlir] [mlir][gpu] Allow subgroup reductions over 1-d vector types (PR #76015)
Jakub Kuderski
llvmlistbot at llvm.org
Wed Dec 20 08:01:25 PST 2023
================
@@ -591,10 +592,12 @@ class GPUSubgroupReduceConversion final
LogicalResult
matchAndRewrite(gpu::SubgroupReduceOp op, OpAdaptor adaptor,
ConversionPatternRewriter &rewriter) const override {
- auto opType = op.getOp();
- auto result =
- createGroupReduceOp(rewriter, op.getLoc(), adaptor.getValue(), opType,
- /*isGroup*/ false, op.getUniform());
+ if (isa<VectorType>(adaptor.getValue().getType()))
----------------
kuhar wrote:
Done, thanks
https://github.com/llvm/llvm-project/pull/76015
More information about the Mlir-commits
mailing list