Home
last modified time | relevance | path

Searched refs:subViewOp (Results 1 – 6 of 6) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/MemRef/Transforms/
H A DFoldSubViewOps.cpp86 Location loc = subViewOp.getLoc(); in resolveSourceIndices()
115 int64_t sourceRank = subViewOp.getSourceType().getRank(); in getPermutationMapAttr()
141 void replaceOp(OpTy loadOp, memref::SubViewOp subViewOp,
156 void replaceOp(OpTy storeOp, memref::SubViewOp subViewOp,
187 StoreOpTy storeOp, memref::SubViewOp subViewOp, in replaceOp() argument
213 auto subViewOp = in matchAndRewrite() local
215 if (!subViewOp) in matchAndRewrite()
223 replaceOp(loadOp, subViewOp, sourceIndices, rewriter); in matchAndRewrite()
231 auto subViewOp = in matchAndRewrite() local
233 if (!subViewOp) in matchAndRewrite()
[all …]
/llvm-project-15.0.7/mlir/lib/Dialect/MemRef/IR/
H A DMemRefOps.cpp2480 if (subViewOp.getSourceType().getRank() != subViewOp.getType().getRank()) in isTrivialSubViewOp()
2554 subViewOp.getType(), subViewOp.getSourceType(), in matchAndRewrite()
2556 subViewOp.getMixedOffsets(), subViewOp.getMixedSizes(), in matchAndRewrite()
2557 subViewOp.getMixedStrides()); in matchAndRewrite()
2563 subViewOp.getOffsets(), subViewOp.getSizes(), subViewOp.getStrides(), in matchAndRewrite()
2564 subViewOp.getStaticOffsets(), subViewOp.getStaticSizes(), in matchAndRewrite()
2565 subViewOp.getStaticStrides()); in matchAndRewrite()
2566 rewriter.replaceOpWithNewOp<CastOp>(subViewOp, subViewOp.getType(), in matchAndRewrite()
2582 if (subViewOp.getSourceType() == subViewOp.getType()) { in matchAndRewrite()
2583 rewriter.replaceOp(subViewOp, subViewOp.getSource()); in matchAndRewrite()
[all …]
/llvm-project-15.0.7/mlir/lib/Conversion/MemRefToLLVM/
H A DMemRefToLLVM.cpp1642 matchAndRewrite(memref::SubViewOp subViewOp, OpAdaptor adaptor, in matchAndRewrite()
1644 auto loc = subViewOp.getLoc(); in matchAndRewrite()
1650 auto viewMemRefType = subViewOp.getType(); in matchAndRewrite()
1653 subViewOp.getSourceType(), in matchAndRewrite()
1655 extractFromI64ArrayAttr(subViewOp.getStaticSizes()), in matchAndRewrite()
1721 subViewOp.isDynamicOffset(i) in matchAndRewrite()
1737 llvm::SmallBitVector unusedDims = subViewOp.getDroppedDims(); in matchAndRewrite()
1771 subViewOp.isDynamicSize(i) in matchAndRewrite()
1781 subViewOp.isDynamicStride(i) in matchAndRewrite()
1786 subViewOp.getStaticStride(i))); in matchAndRewrite()
[all …]
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DVectorization.cpp1129 memref::SubViewOp subViewOp; in getSubViewUseIfUnique() local
1132 if (subViewOp) in getSubViewUseIfUnique()
1134 subViewOp = newSubViewOp; in getSubViewUseIfUnique()
1137 return subViewOp; in getSubViewUseIfUnique()
1158 memref::SubViewOp subViewOp = getSubViewUseIfUnique(viewOrAlloc); in matchAndRewrite() local
1159 if (!subViewOp) in matchAndRewrite()
1161 Value subView = subViewOp.getResult(); in matchAndRewrite()
1239 memref::SubViewOp subViewOp = getSubViewUseIfUnique(viewOrAlloc); in matchAndRewrite() local
1240 if (!subViewOp) in matchAndRewrite()
1242 Value subView = subViewOp.getResult(); in matchAndRewrite()
[all …]
H A DFusion.cpp183 if (auto subViewOp = dyn_cast<memref::SubViewOp>(shapeProducingOp)) in getRangeFromOperandShape() local
184 return subViewOp.getOrCreateRanges(b, loc)[dim]; in getRangeFromOperandShape()
H A DPromotion.cpp167 allocationFn = [&](OpBuilder &b, memref::SubViewOp subViewOp, in LinalgOpInstancePromotionOptions()
170 return defaultAllocBufferCallBack(options, b, subViewOp, in LinalgOpInstancePromotionOptions()