[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