[Mlir-commits] [mlir] [mlir][vector] Add support for multi-dim reduction vector distribution (PR #71193)
Lei Zhang
llvmlistbot at llvm.org
Thu Nov 9 22:48:03 PST 2023
================
@@ -475,16 +472,22 @@ static VectorType getDistributedType(VectorType originalType, AffineMap map,
/// }
/// vector.transfer_write %v, %A[%id] : vector<1xf32>, memref<128xf32>
struct WarpOpTransferWrite : public OpRewritePattern<vector::TransferWriteOp> {
- WarpOpTransferWrite(MLIRContext *ctx, DistributionMapFn fn,
- PatternBenefit b = 1)
- : OpRewritePattern<vector::TransferWriteOp>(ctx, b),
- distributionMapFn(std::move(fn)) {}
+ WarpOpTransferWrite(MLIRContext *ctx, PatternBenefit b = 1)
----------------
antiagainst wrote:
This can just be `using OpRewritePattern::OpRewritePattern` now?
https://github.com/llvm/llvm-project/pull/71193
More information about the Mlir-commits
mailing list