Home
last modified time | relevance | path

Searched defs:allocOp (Results 1 – 13 of 13) sorted by relevance

/llvm-project/mlir/test/lib/Analysis/
H A DTestMemRefStrideCalculation.cpp34 getOperation().walk([&](memref::AllocOp allocOp) { in runOnOperation() argument
/llvm-project/mlir/lib/Dialect/MemRef/Transforms/
H A DMultiBuffer.cpp99 mlir::memref::multiBuffer(RewriterBase &rewriter, memref::AllocOp allocOp, in multiBuffer()
246 mlir::memref::multiBuffer(memref::AllocOp allocOp, in multiBuffer()
H A DNormalizeMemRefs.cpp167 .walk([&](memref::AllocOp allocOp) -> WalkResult { in areMemRefsNormalizable() argument
347 for (memref::AllocOp allocOp : allocOps) normalizeFuncOpMemRefs() local
/llvm-project/mlir/lib/Dialect/NVGPU/Transforms/
H A DOptimizeSharedMemory.cpp232 if (!NVGPUDialect::hasSharedMemoryAddressSpace(allocOp.getType())) in runOnOperation() local
227 __anonaf47cb3f0602(memref::AllocOp allocOp) runOnOperation() argument
/llvm-project/mlir/lib/Dialect/Bufferization/Transforms/
H A DBufferOptimizations.cpp71 auto allocOp = dyn_cast<AllocationOpInterface>(op); in allowAllocDominateBlockHoisting() local
79 auto allocOp = dyn_cast<AllocationOpInterface>(op); in allowAllocLoopHoisting() local
398 Operation *allocOp = alloc.getDefiningOp(); in promote() local
/llvm-project/mlir/lib/Dialect/MemRef/IR/
H A DValueBoundsOpInterfaceImpl.cpp26 auto allocOp = cast<OpTy>(op); in populateBoundsForShapedValueDim() local
H A DMemRefOps.cpp3523 AllocOp allocOp = allocOperand.getDefiningOp<AllocOp>(); matchAndRewrite() local
/llvm-project/mlir/lib/Dialect/Affine/Transforms/
H A DPipelineDataTransfer.cpp287 if (auto *allocOp = oldMemRef.getDefiningOp()) { in runOnAffineForOp() local
/llvm-project/mlir/lib/Dialect/Bufferization/IR/
H A DBufferizationOps.cpp1105 if (!allocOp) in matchAndRewrite() local
1142 if (auto allocOp = memref.getDefiningOp<MemoryEffectOpInterface>()) { matchAndRewrite() local
/llvm-project/mlir/lib/Dialect/Affine/Utils/
H A DUtils.cpp1644 createNewDynamicSizes(MemRefType oldMemRefType,MemRefType newMemRefType,AffineMap map,memref::AllocOp * allocOp,OpBuilder b,SmallVectorImpl<Value> & newDynamicSizes) createNewDynamicSizes() argument
1691 normalizeMemRef(memref::AllocOp * allocOp) normalizeMemRef() argument
[all...]
/llvm-project/mlir/lib/Conversion/MemRefToSPIRV/
H A DMemRefToSPIRV.cpp123 static bool isAllocationSupported(Operation *allocOp, MemRefType type) { in isAllocationSupported() argument
/llvm-project/mlir/include/mlir/Dialect/Linalg/Transforms/
H A DTransforms.h52 AllocOp allocOp = AllocOp::MemrefAlloc; member
/llvm-project/mlir/lib/Conversion/GPUCommon/
H A DGPUToLLVMConversion.cpp738 matchAndRewrite(gpu::AllocOp allocOp,OpAdaptor adaptor,ConversionPatternRewriter & rewriter) const matchAndRewrite() argument
[all...]