Home
last modified time | relevance | path

Searched refs:oldMemRefType (Results 1 – 4 of 4) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Affine/Transforms/
H A DPipelineDataTransfer.cpp71 auto doubleShape = [&](MemRefType oldMemRefType) -> MemRefType { in doubleBuffer() argument
73 ArrayRef<int64_t> oldShape = oldMemRefType.getShape(); in doubleBuffer()
74 SmallVector<int64_t, 4> newShape(1 + oldMemRefType.getRank()); in doubleBuffer()
77 return MemRefType::Builder(oldMemRefType).setShape(newShape).setLayout({}); in doubleBuffer()
80 auto oldMemRefType = oldMemRef.getType().cast<MemRefType>(); in doubleBuffer() local
81 auto newMemRefType = doubleShape(oldMemRefType); in doubleBuffer()
87 for (const auto &dim : llvm::enumerate(oldMemRefType.getShape())) { in doubleBuffer()
H A DLoopFusion.cpp902 auto oldMemRefType = oldMemRef.getType().cast<MemRefType>(); in createPrivateMemRef() local
903 unsigned rank = oldMemRefType.getRank(); in createPrivateMemRef()
946 getMemRefEltSizeInBytes(oldMemRefType) * numElements.value(); in createPrivateMemRef()
951 newMemSpace = oldMemRefType.getMemorySpaceAsInt(); in createPrivateMemRef()
953 auto newMemRefType = MemRefType::get(newShape, oldMemRefType.getElementType(), in createPrivateMemRef()
/llvm-project-15.0.7/mlir/lib/Dialect/MemRef/Transforms/
H A DNormalizeMemRefs.cpp411 MemRefType oldMemRefType = oldMemRef.getType().dyn_cast<MemRefType>(); in normalizeFuncOpMemRefs() local
413 if (!oldMemRefType) in normalizeFuncOpMemRefs()
416 if (oldMemRefType == newMemRefType) in normalizeFuncOpMemRefs()
419 AffineMap layoutMap = oldMemRefType.getLayout().getAffineMap(); in normalizeFuncOpMemRefs()
/llvm-project-15.0.7/mlir/lib/Dialect/Affine/Utils/
H A DUtils.cpp1627 static void createNewDynamicSizes(MemRefType oldMemRefType, in createNewDynamicSizes() argument
1633 ArrayRef<int64_t> oldMemRefShape = oldMemRefType.getShape(); in createNewDynamicSizes()
1635 for (unsigned d = 0; d < oldMemRefType.getRank(); ++d) { in createNewDynamicSizes()
1700 MemRefType oldMemRefType = oldMemRef.getType().cast<MemRefType>(); in normalizeMemRef() local
1702 createNewDynamicSizes(oldMemRefType, newMemRefType, layoutMap, allocOp, b, in normalizeMemRef()