| /external/llvm-project/mlir/lib/Transforms/ |
| D | NormalizeMemRefs.cpp | 213 MemRefType memrefType = opType.dyn_cast<MemRefType>(); in updateFunctionSignature() local
216 if (!memrefType || memrefType == resultTypes[operandEn.index()]) in updateFunctionSignature()
226 if (memrefType.getAffineMaps().empty()) in updateFunctionSignature()
227 resultTypes[operandEn.index()] = memrefType; in updateFunctionSignature()
344 MemRefType memrefType = argType.dyn_cast<MemRefType>(); in normalizeFuncOpMemRefs() local
347 if (!memrefType) { in normalizeFuncOpMemRefs()
353 MemRefType newMemRefType = normalizeMemRefType(memrefType, b, in normalizeFuncOpMemRefs()
355 if (newMemRefType == memrefType || funcOp.isExternal()) { in normalizeFuncOpMemRefs()
366 AffineMap layoutMap = memrefType.getAffineMaps().front(); in normalizeFuncOpMemRefs()
451 MemRefType memrefType = resType.dyn_cast<MemRefType>(); in normalizeFuncOpMemRefs() local
[all …]
|
| /external/llvm-project/mlir/test/EDSC/ |
| D | builder-api-test.cpp | 315 auto memrefType = in TEST_FUNC() local
320 makeFunction("builder_helpers", {}, {memrefType, memrefType, memrefType}); in TEST_FUNC()
395 auto memrefType = MemRefType::get({}, i1Type, {}, 0); in TEST_FUNC() local
397 {memrefType, memrefType}); in TEST_FUNC()
540 auto memrefType = MemRefType::get( in TEST_FUNC() local
542 auto f = makeFunction("select_op", {}, {memrefType}); in TEST_FUNC()
615 auto memrefType = MemRefType::get( in TEST_FUNC() local
617 auto f = makeFunction("select_op", {}, {memrefType, memrefType}); in TEST_FUNC()
722 auto memrefType = in TEST_FUNC() local
726 auto f = makeFunction("tile_2d", {}, {memrefType, memrefType, memrefType}); in TEST_FUNC()
[all …]
|
| /external/llvm-project/mlir/lib/Transforms/Utils/ |
| D | Utils.cpp | 397 MemRefType memrefType = allocOp.getType(); in normalizeMemRef() local
403 normalizeMemRefType(memrefType, b, allocOp.symbolOperands().size()); in normalizeMemRef()
404 if (newMemRefType == memrefType) in normalizeMemRef()
414 AffineMap layoutMap = memrefType.getAffineMaps().front(); in normalizeMemRef()
437 MemRefType mlir::normalizeMemRefType(MemRefType memrefType, OpBuilder b, in normalizeMemRefType() argument
439 unsigned rank = memrefType.getRank(); in normalizeMemRefType()
441 return memrefType; in normalizeMemRefType()
443 ArrayRef<AffineMap> layoutMaps = memrefType.getAffineMaps(); in normalizeMemRefType()
448 return memrefType; in normalizeMemRefType()
455 if (memrefType.getNumDynamicDims() > 0) in normalizeMemRefType()
[all …]
|
| /external/llvm-project/mlir/test/lib/Transforms/ |
| D | TestMemRefStrideCalculation.cpp | 27 auto memrefType = allocOp.getResult().getType().cast<MemRefType>(); in runOnFunction() local
30 if (failed(getStridesAndOffset(memrefType, strides, offset))) { in runOnFunction()
31 llvm::outs() << "MemRefType " << memrefType << " cannot be converted to " in runOnFunction()
|
| /external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| D | Bufferize.cpp | 31 auto memrefType = memref.getType().cast<MemRefType>(); in cloneMemref() local
33 for (auto dim : llvm::enumerate(memrefType.getShape())) { in cloneMemref()
38 auto alloc = b.create<AllocOp>(loc, memrefType, dynOperands); in cloneMemref()
62 auto memrefType = MemRefType::get(tensorShape, tensorType.getElementType()); in allocateBuffersForResults() local
76 if (memrefType.hasStaticShape()) { in allocateBuffersForResults()
77 resultBuffers.push_back(b.create<AllocOp>(loc, memrefType)); in allocateBuffersForResults()
102 resultBuffers.push_back(b.create<AllocOp>(loc, memrefType, dynOperands)); in allocateBuffersForResults()
|
| D | Vectorization.cpp | 131 auto memrefType = memref.getType().cast<MemRefType>(); in transferReadVector() local
133 SmallVector<Value, 4> indices(memrefType.getRank(), std_constant_index(0)); in transferReadVector()
141 auto memrefType = memref.getType().cast<MemRefType>(); in transferWriteVector() local
143 SmallVector<Value, 4> indices(memrefType.getRank(), std_constant_index(0)); in transferWriteVector()
|
| /external/llvm-project/mlir/lib/Conversion/LinalgToStandard/ |
| D | LinalgToStandard.cpp | 38 if (auto memrefType = type.dyn_cast<MemRefType>()) in extractOperandTypes() local
39 result.push_back(eraseStridedLayout(memrefType)); in extractOperandTypes()
90 auto memrefType = op.getType().dyn_cast<MemRefType>(); in createTypeCanonicalizedMemRefOperands() local
91 if (!memrefType) { in createTypeCanonicalizedMemRefOperands()
96 b.create<MemRefCastOp>(loc, eraseStridedLayout(memrefType), op); in createTypeCanonicalizedMemRefOperands()
|
| /external/llvm-project/mlir/lib/Conversion/StandardToSPIRV/ |
| D | ConvertStandardToSPIRV.cpp | 837 auto memrefType = loadOp.memref().getType().cast<MemRefType>(); in matchAndRewrite() local
838 if (!memrefType.getElementType().isSignlessInteger()) in matchAndRewrite()
841 spirv::getElementPtr(typeConverter, memrefType, loadOperands.memref(), in matchAndRewrite()
844 int srcBits = memrefType.getElementType().getIntOrFloatBitWidth(); in matchAndRewrite()
845 auto dstType = typeConverter.convertType(memrefType) in matchAndRewrite()
910 auto memrefType = loadOp.memref().getType().cast<MemRefType>(); in matchAndRewrite() local
911 if (memrefType.getElementType().isSignlessInteger()) in matchAndRewrite()
914 spirv::getElementPtr(typeConverter, memrefType, loadOperands.memref(), in matchAndRewrite()
956 auto memrefType = storeOp.memref().getType().cast<MemRefType>(); in matchAndRewrite() local
957 if (!memrefType.getElementType().isSignlessInteger()) in matchAndRewrite()
[all …]
|
| /external/llvm-project/mlir/lib/Dialect/Affine/IR/ |
| D | AffineOps.cpp | 2076 auto memrefType = operands[0].getType().cast<MemRefType>(); in build() local
2077 result.types.push_back(memrefType.getElementType()); in build()
2085 auto memrefType = memref.getType().cast<MemRefType>(); in build() local
2087 result.types.push_back(memrefType.getElementType()); in build()
2092 auto memrefType = memref.getType().cast<MemRefType>(); in build() local
2093 int64_t rank = memrefType.getRank(); in build()
2137 MemRefType memrefType, unsigned numIndexOperands) { in verifyMemoryOpIndexing() argument
2140 if (map.getNumResults() != memrefType.getRank()) in verifyMemoryOpIndexing()
2145 if (memrefType.getRank() != numIndexOperands) in verifyMemoryOpIndexing()
2162 auto memrefType = op.getMemRefType(); in verify() local
[all …]
|
| /external/llvm-project/mlir/lib/Dialect/StandardOps/Transforms/ |
| D | Bufferize.cpp | 51 MemRefType memrefType = in matchAndRewrite() local
54 rewriter.create<AllocOp>(loc, memrefType, transformed.dynamicExtents()); in matchAndRewrite()
68 : rewriter.create<ConstantIndexOp>(loc, memrefType.getDimSize(i)); in matchAndRewrite()
|
| /external/llvm-project/mlir/lib/Dialect/StandardOps/IR/ |
| D | Ops.cpp | 312 auto memrefType = alloc.getType(); in matchAndRewrite() local
317 newShapeConstants.reserve(memrefType.getRank()); in matchAndRewrite()
321 for (unsigned dim = 0, e = memrefType.getRank(); dim < e; ++dim) { in matchAndRewrite()
322 int64_t dimSize = memrefType.getDimSize(dim); in matchAndRewrite()
342 MemRefType::Builder(memrefType).setShape(newShapeConstants); in matchAndRewrite()
485 if (auto memrefType = memref.getType().dyn_cast<MemRefType>()) { in build() local
486 Type elementType = memrefType.getElementType(); in build()
519 Type memrefType; in parseGenericAtomicRMWOp() local
525 parser.parseColonType(memrefType) || in parseGenericAtomicRMWOp()
526 parser.resolveOperand(memref, memrefType, result.operands) || in parseGenericAtomicRMWOp()
[all …]
|
| /external/llvm-project/mlir/include/mlir/Transforms/ |
| D | Utils.h | 98 MemRefType normalizeMemRefType(MemRefType memrefType, OpBuilder builder,
|
| /external/llvm-project/mlir/lib/Dialect/Vector/ |
| D | VectorOps.cpp | 1893 static LogicalResult verifyTransferOp(Operation *op, MemRefType memrefType, in verifyTransferOp() argument
1897 auto memrefElementType = memrefType.getElementType(); in verifyTransferOp()
1937 if (permutationMap.getNumInputs() != memrefType.getRank()) in verifyTransferOp()
2043 MemRefType memrefType = op.getMemRefType(); in verify() local
2047 auto memrefElementType = memrefType.getElementType(); in verify()
2049 if (static_cast<int64_t>(op.indices().size()) != memrefType.getRank()) in verify()
2050 return op.emitOpError("requires ") << memrefType.getRank() << " indices"; in verify()
2052 if (failed(verifyTransferOp(op.getOperation(), memrefType, vectorType, in verify()
2223 MemRefType memrefType = op.getMemRefType(); in verify() local
2227 if (llvm::size(op.indices()) != memrefType.getRank()) in verify()
[all …]
|
| /external/llvm-project/mlir/test/CAPI/ |
| D | ir.c | 78 MlirType memrefType = in makeAndDumpAdd() local
80 MlirType funcBodyArgTypes[] = {memrefType, memrefType}; in makeAndDumpAdd()
|
| /external/llvm-project/mlir/include/mlir/Dialect/StandardOps/IR/ |
| D | Ops.td | 182 OpBuilderDAG<(ins "MemRefType":$memrefType,
184 return build($_builder, $_state, memrefType, {}, alignment);
186 OpBuilderDAG<(ins "MemRefType":$memrefType, "ValueRange":$dynamicSizes,
188 return build($_builder, $_state, memrefType, dynamicSizes, {}, alignment);
190 OpBuilderDAG<(ins "MemRefType":$memrefType, "ValueRange":$dynamicSizes,
193 $_state.types.push_back(memrefType);
2045 auto memrefType = memref.getType().cast<MemRefType>();
2048 $_state.types.push_back(memrefType.getElementType());
|
| /external/llvm-project/mlir/lib/Conversion/StandardToLLVM/ |
| D | StandardToLLVM.cpp | 1247 if (auto memrefType = en.value().dyn_cast<MemRefType>()) { in wrapForExternalCallers() local
1249 MemRefDescriptor::unpack(rewriter, loc, loaded, memrefType, args); in wrapForExternalCallers()
4088 if (auto memrefType = operand.getType().dyn_cast<MemRefType>()) { in promoteOperands() local
4100 if (auto memrefType = operand.getType().dyn_cast<MemRefType>()) { in promoteOperands() local
|