Home
last modified time | relevance | path

Searched refs:AllocOp (Results 1 – 25 of 41) sorted by relevance

12

/llvm-project-15.0.7/mlir/test/lib/Dialect/MemRef/
H A DTestMultiBuffer.cpp41 SmallVector<memref::AllocOp> allocs; in runOnOperation()
43 [&allocs](memref::AllocOp alloc) { allocs.push_back(alloc); }); in runOnOperation()
44 for (memref::AllocOp alloc : allocs) in runOnOperation()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/MemRef/Transforms/
H A DPasses.h34 class AllocOp; variable
83 LogicalResult multiBuffer(memref::AllocOp allocOp, unsigned multiplier);
/llvm-project-15.0.7/mlir/test/Dialect/Bufferization/Transforms/
H A Dbuffer-hoisting.mlir9 // bb1 bb2 <- Initial position of AllocOp
37 // bb1 bb2 <- Initial position of AllocOp
73 // bb1 bb2 <- Initial position of AllocOp
123 // | bb1 <- Initial position of AllocOp
147 // bb0 <- Initial position of the first AllocOp
151 // bb3 <- Initial position of the second AllocOp
185 // bb0 <- Initial position of AllocOp
214 // bb0 <- Initial position of the first AllocOp
221 // bb5 <- Initial position of the second AllocOp
279 // Initial pos of the 1st AllocOp -> bb1 bb2 <- Initial pos of the 2nd AllocOp
[all …]
H A Dbuffer-deallocation.mlir10 // bb1 bb2 <- Initial position of AllocOp
49 // bb1 bb2 <- Initial position of AllocOp
133 // bb1 bb2 <- Initial position of AllocOp
201 // Test Case: Existing AllocOp with no users.
219 // | bb1 <- Initial position of AllocOp
251 // bb0 <- Initial position of AllocOp
277 // bb0 <- Initial position of the first AllocOp
281 // bb3 <- Initial position of the second AllocOp
317 // bb0 <- Initial position of AllocOp
420 // Initial pos of the 1st AllocOp -> bb1 bb2 <- Initial pos of the 2nd AllocOp
[all …]
H A Dbuffer-loop-hoisting.mlir9 // bb1 bb2 <- Initial position of AllocOp
12 // BufferLoopHoisting expected behavior: It should not move the AllocOp.
36 // bb1 bb2 <- Initial position of AllocOp
39 // BufferLoopHoisting expected behavior: It should not move the existing AllocOp
71 // BufferLoopHoisting expected behavior: The AllocOp for the BufferBasedOp
72 // should remain inside the region of the RegionBufferBasedOp. The AllocOp of
/llvm-project-15.0.7/mlir/test/Transforms/
H A Dpromote-buffers-to-stack.mlir11 // bb1 bb2 <- Initial position of AllocOp
42 // bb1 bb2 <- Initial position of AllocOp
113 // Test Case: Existing AllocOp with no users.
130 // | bb1 <- Initial position of AllocOp
157 // bb0 <- Initial position of AllocOp
184 // bb0 <- Initial position of the first AllocOp
188 // bb3 <- Initial position of the second AllocOp
219 // bb0 <- Initial position of AllocOp
248 // bb0 <- Initial position of the first AllocOp
315 // Initial pos of the 1st AllocOp -> bb1 bb2 <- Initial pos of the 2nd AllocOp
[all …]
/llvm-project-15.0.7/flang/lib/Optimizer/Transforms/
H A DAffineDemotion.cpp127 class StdAllocConversion : public mlir::OpRewritePattern<memref::AllocOp> {
131 matchAndRewrite(memref::AllocOp op, in matchAndRewrite()
154 target.addIllegalOp<memref::AllocOp>(); in runOnOperation()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Affine/
H A DUtils.h32 class AllocOp; variable
241 LogicalResult normalizeMemRef(memref::AllocOp *op);
/llvm-project-15.0.7/mlir/lib/Dialect/MemRef/Transforms/
H A DNormalizeMemRefs.cpp158 .walk([&](memref::AllocOp allocOp) -> WalkResult { in areMemRefsNormalizable()
331 SmallVector<memref::AllocOp, 4> allocOps; in normalizeFuncOpMemRefs()
332 funcOp.walk([&](memref::AllocOp op) { allocOps.push_back(op); }); in normalizeFuncOpMemRefs()
333 for (memref::AllocOp allocOp : allocOps) in normalizeFuncOpMemRefs()
H A DMultiBuffer.cpp81 LogicalResult mlir::memref::multiBuffer(memref::AllocOp allocOp, in multiBuffer()
120 auto newAlloc = builder.create<memref::AllocOp>(loc, newMemref); in multiBuffer()
/llvm-project-15.0.7/mlir/lib/Dialect/Bufferization/Transforms/
H A DBufferOptimizations.cpp37 if (!type || !alloc.getDefiningOp<memref::AllocOp>()) in defaultIsSmallAlloc()
270 return llvm::isa<memref::AllocOp>(op); in shouldHoistOpType()
307 return llvm::isa<memref::AllocOp, memref::AllocaOp>(op); in shouldHoistOpType()
H A DBufferDeallocation.cpp622 DefaultAllocationInterface, memref::AllocOp> {
694 memref::AllocOp::attachInterface<DefaultAllocationInterface>(*ctx); in registerAllocationOpInterfaceExternalModels()
H A DBufferResultsToOutParams.cpp139 Value outParam = builder.create<memref::AllocOp>(op.getLoc(), allocType); in updateCalls()
/llvm-project-15.0.7/mlir/lib/Conversion/MemRefToSPIRV/
H A DMemRefToSPIRV.cpp92 if (isa<memref::AllocOp, memref::DeallocOp>(allocOp)) { in isAllocationSupported()
179 class AllocOpPattern final : public OpConversionPattern<memref::AllocOp> {
181 using OpConversionPattern<memref::AllocOp>::OpConversionPattern;
184 matchAndRewrite(memref::AllocOp operation, OpAdaptor adaptor,
265 AllocOpPattern::matchAndRewrite(memref::AllocOp operation, OpAdaptor adaptor, in matchAndRewrite()
/llvm-project-15.0.7/mlir/test/lib/Analysis/
H A DTestMemRefStrideCalculation.cpp34 getOperation().walk([&](memref::AllocOp allocOp) { in runOnOperation()
/llvm-project-15.0.7/mlir/lib/Dialect/NVGPU/Transforms/
H A DOptimizeSharedMemory.cpp251 SmallVector<memref::AllocOp> shmAllocOps; in runOnOperation()
252 op->walk([&](memref::AllocOp allocOp) { in runOnOperation()
/llvm-project-15.0.7/mlir/lib/Conversion/BufferizationToMemRef/
H A DBufferizationToMemRef.cpp66 Value alloc = rewriter.create<memref::AllocOp>(op->getLoc(), allocType, in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Affine/Utils/
H A DUtils.cpp664 memref.getDefiningOp<memref::AllocOp>(); in hasNoInterveningEffect()
687 effect.getValue().getDefiningOp<memref::AllocOp>())) { in hasNoInterveningEffect()
1059 if (!defOp || !isa<memref::AllocOp>(defOp)) in affineScalarReplace()
1629 memref::AllocOp *allocOp, OpBuilder b, in createNewDynamicSizes()
1677 LogicalResult mlir::normalizeMemRef(memref::AllocOp *allocOp) { in normalizeMemRef()
1694 memref::AllocOp newAlloc; in normalizeMemRef()
1706 b.create<memref::AllocOp>(allocOp->getLoc(), newMemRefType, in normalizeMemRef()
1709 newAlloc = b.create<memref::AllocOp>(allocOp->getLoc(), newMemRefType, in normalizeMemRef()
/llvm-project-15.0.7/mlir/lib/Conversion/GPUCommon/
H A DGPUToLLVMConversion.cpp211 : public ConvertOpToGpuRuntimeCallPattern<gpu::AllocOp> {
214 : ConvertOpToGpuRuntimeCallPattern<gpu::AllocOp>(typeConverter) {} in ConvertAllocOpToGpuRuntimeCallPattern()
218 matchAndRewrite(gpu::AllocOp allocOp, OpAdaptor adaptor,
446 gpu::AllocOp allocOp, OpAdaptor adaptor, in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DPromotion.cpp63 return b.createOrFold<memref::AllocOp>(staticBufferType, ValueRange{}, in allocBuffer()
73 return b.create<memref::AllocOp>(dynamicBufferType, mul, alignmentAttr); in allocBuffer()
/llvm-project-15.0.7/mlir/lib/Dialect/MemRef/IR/
H A DMemRefOps.cpp118 static_assert(llvm::is_one_of<AllocLikeOp, AllocOp, AllocaOp>::value, in verifyAllocLikeOp()
140 LogicalResult AllocOp::verify() { return verifyAllocLikeOp(*this); } in verify()
238 void AllocOp::getCanonicalizationPatterns(RewritePatternSet &results, in getCanonicalizationPatterns()
240 results.add<SimplifyAllocConst<AllocOp>, SimplifyDeadAlloc<AllocOp>>(context); in getCanonicalizationPatterns()
896 if (auto alloc = dyn_cast_or_null<AllocOp>(definingOp)) in fold()
2832 AllocOp allocOp = allocOperand.getDefiningOp<AllocOp>(); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Conversion/MemRefToLLVM/
H A DMemRefToLLVM.cpp35 : AllocLikeOpLLVMLowering(memref::AllocOp::getOperationName(), in AllocOpLowering()
51 memref::AllocOp allocOp = cast<memref::AllocOp>(op); in allocateBuffer()
96 : AllocLikeOpLLVMLowering(memref::AllocOp::getOperationName(), in AlignedAllocOpLowering()
134 int64_t getAllocationAlignment(memref::AllocOp allocOp) const { in getAllocationAlignment()
159 memref::AllocOp allocOp = cast<memref::AllocOp>(op); in allocateBuffer()
/llvm-project-15.0.7/mlir/lib/Dialect/Bufferization/IR/
H A DBufferizableOpInterface.cpp658 .create<memref::AllocOp>(loc, type, dynShape, in createAlloc()
661 return b.create<memref::AllocOp>(loc, type, dynShape).getResult(); in createAlloc()
/llvm-project-15.0.7/mlir/lib/Dialect/Affine/Transforms/
H A DPipelineDataTransfer.cpp94 Value newMemRef = bOuter.create<memref::AllocOp>( in doubleBuffer()
/llvm-project-15.0.7/mlir/docs/Tutorials/
H A DUnderstandingTheIRStructure.md202 `AllocOp` nested inside the function and interrupt the traversal if one of them
206 WalkResult result = getFunction().walk([&](AllocOp allocOp) {

12