Home
last modified time | relevance | path

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

/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DPromotion.cpp82 OpBuilder &builder, memref::SubViewOp subView, in defaultAllocBufferCallBack() argument
85 ShapedType viewType = subView.getType(); in defaultAllocBufferCallBack()
86 ImplicitLocOpBuilder b(subView.getLoc(), builder); in defaultAllocBufferCallBack()
212 OpBuilder &b, Location loc, memref::SubViewOp subView, in promoteSubviewAsNewBuffer() argument
214 auto viewType = subView.getType(); in promoteSubviewAsNewBuffer()
220 llvm::SmallBitVector droppedDims = subView.getDroppedDims(); in promoteSubviewAsNewBuffer()
222 for (const auto &en : llvm::enumerate(subView.getOrCreateRanges(b, loc))) { in promoteSubviewAsNewBuffer()
237 b.createOrFold<memref::DimOp>(loc, subView, resultDimIdx++)); in promoteSubviewAsNewBuffer()
261 memref::SubViewOp subView = in promoteSubViews() local
264 b, b.getLoc(), subView, options.allocationFn, layout); in promoteSubViews()
[all …]
H A DVectorization.cpp1161 Value subView = subViewOp.getResult(); in matchAndRewrite() local
1162 LDBG("with subView " << subView); in matchAndRewrite()
1166 for (auto &u : subView.getUses()) { in matchAndRewrite()
1169 if (newCopyOp.getTarget() != subView) in matchAndRewrite()
1172 if (mayExistInterleavedUses(newCopyOp, xferOp, {viewOrAlloc, subView})) in matchAndRewrite()
1191 if (mayExistInterleavedUses(newFillOp, copyOp, {viewOrAlloc, subView})) in matchAndRewrite()
1242 Value subView = subViewOp.getResult(); in matchAndRewrite() local
1248 if (newCopyOp.getSource() != subView) in matchAndRewrite()
1250 if (mayExistInterleavedUses(xferOp, newCopyOp, {viewOrAlloc, subView})) in matchAndRewrite()
H A DFusion.cpp357 auto subView = consumerOpOperand.get().getDefiningOp<memref::SubViewOp>(); in fuseProducerOfBuffer() local
358 if (!subView) { in fuseProducerOfBuffer()
/llvm-project-15.0.7/mlir/lib/Dialect/Tensor/Transforms/
H A DBufferizableOpInterfaceImpl.cpp299 Value subView = rewriter.create<memref::SubViewOp>( in bufferize() local
303 replaceOpWithBufferizedValues(rewriter, op, subView); in bufferize()
724 Value subView = rewriter.create<memref::SubViewOp>( in bufferize() local
734 if (failed(options.createMemCpy(rewriter, loc, *srcMemref, subView))) in bufferize()
/llvm-project-15.0.7/mlir/lib/Dialect/NVGPU/Transforms/
H A DOptimizeSharedMemory.cpp183 parentOp->walk([&](memref::SubViewOp subView) { hasSubView = true; }); in optimizeSharedMemoryReadsAndWrites() argument
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/Transforms/
H A DTransforms.h199 OpBuilder &b, memref::SubViewOp subView,
290 promoteSubviewAsNewBuffer(OpBuilder &b, Location loc, memref::SubViewOp subView,