Lines Matching refs:allocOp
51 memref::AllocOp allocOp = cast<memref::AllocOp>(op); in allocateBuffer() local
52 MemRefType memRefType = allocOp.getType(); in allocateBuffer()
55 if (auto alignmentAttr = allocOp.getAlignment()) { in allocateBuffer()
73 auto allocFuncOp = getAllocFn(allocOp->getParentOfType<ModuleOp>()); in allocateBuffer()
134 int64_t getAllocationAlignment(memref::AllocOp allocOp) const { in getAllocationAlignment()
135 if (Optional<uint64_t> alignment = allocOp.getAlignment()) in getAllocationAlignment()
141 auto eltSizeBytes = getMemRefEltSizeInBytes(allocOp.getType(), allocOp); in getAllocationAlignment()
159 memref::AllocOp allocOp = cast<memref::AllocOp>(op); in allocateBuffer() local
160 MemRefType memRefType = allocOp.getType(); in allocateBuffer()
161 int64_t alignment = getAllocationAlignment(allocOp); in allocateBuffer()
170 auto allocFuncOp = getAllocFn(allocOp->getParentOfType<ModuleOp>()); in allocateBuffer()