[Mlir-commits] [mlir] [mlir][vector] Add unroll patterns for vector.load and vector.store (PR #143420)
James Newling
llvmlistbot at llvm.org
Fri Jun 20 13:12:04 PDT 2025
================
@@ -631,6 +653,100 @@ struct UnrollGatherPattern : public OpRewritePattern<vector::GatherOp> {
vector::UnrollVectorOptions options;
};
+struct UnrollLoadPattern : public OpRewritePattern<vector::LoadOp> {
+ UnrollLoadPattern(MLIRContext *context,
+ const vector::UnrollVectorOptions &options,
+ PatternBenefit benefit = 1)
+ : OpRewritePattern<vector::LoadOp>(context, benefit), options(options) {}
+
+ LogicalResult matchAndRewrite(vector::LoadOp loadOp,
+ PatternRewriter &rewriter) const override {
+ VectorType vecType = loadOp.getVectorType();
+ if (vecType.getRank() <= 1)
+ return failure();
+
+ auto targetShape = getTargetShape(options, loadOp);
+ if (!targetShape)
+ return failure();
+
+ Location loc = loadOp.getLoc();
+ ArrayRef<int64_t> originalShape = vecType.getShape();
+ SmallVector<int64_t> strides(targetShape->size(), 1);
+
+ Value result = rewriter.create<arith::ConstantOp>(
+ loc, vecType, rewriter.getZeroAttr(vecType));
+
+ SmallVector<Value> originalIndices(loadOp.getIndices().begin(),
+ loadOp.getIndices().end());
+
+ SmallVector<int64_t> loopOrder =
+ getUnrollOrder(originalShape.size(), loadOp, options);
+
+ auto targetVecType =
+ VectorType::get(*targetShape, vecType.getElementType());
+
+ for (SmallVector<int64_t> offsets :
+ StaticTileOffsetRange(originalShape, *targetShape, loopOrder)) {
+ SmallVector<Value> indices =
+ computeIndices(rewriter, loc, originalIndices, offsets);
+ Value slice = rewriter.create<vector::LoadOp>(loc, targetVecType,
----------------
newling wrote:
```suggestion
Value slicedLoad = rewriter.create<vector::LoadOp>(loc, targetVecType,
```
'slice' makes me think if a extract_strided_slice which this isn't
https://github.com/llvm/llvm-project/pull/143420
More information about the Mlir-commits
mailing list