Home
last modified time | relevance | path

Searched refs:UnrankedMemRefType (Results 1 – 25 of 34) sorted by relevance

12

/llvm-project-15.0.7/mlir/lib/ExecutionEngine/
H A DRunnerUtils.cpp28 _mlir_ciface_printMemrefShapeI32(UnrankedMemRefType<int32_t> *M) { in _mlir_ciface_printMemrefShapeI32()
35 _mlir_ciface_printMemrefShapeI64(UnrankedMemRefType<int64_t> *M) { in _mlir_ciface_printMemrefShapeI64()
48 _mlir_ciface_printMemrefShapeF64(UnrankedMemRefType<double> *M) { in _mlir_ciface_printMemrefShapeF64()
88 UnrankedMemRefType<int32_t> descriptor = {rank, ptr}; in printMemrefI32()
93 UnrankedMemRefType<int64_t> descriptor = {rank, ptr}; in printMemrefI64()
98 UnrankedMemRefType<float> descriptor = {rank, ptr}; in printMemrefF32()
103 UnrankedMemRefType<double> descriptor = {rank, ptr}; in printMemrefF64()
132 _mlir_ciface_verifyMemRefF32(UnrankedMemRefType<float> *actual, in _mlir_ciface_verifyMemRefF32()
145 UnrankedMemRefType<int32_t> actualDesc = {rank, actualPtr}; in verifyMemRefI32()
152 UnrankedMemRefType<float> actualDesc = {rank, actualPtr}; in verifyMemRefF32()
[all …]
H A DCRunnerUtils.cpp49 extern "C" void memrefCopy(int64_t elemSize, UnrankedMemRefType<char> *srcArg, in memrefCopy()
50 UnrankedMemRefType<char> *dstArg) { in memrefCopy()
/llvm-project-15.0.7/mlir/include/mlir/ExecutionEngine/
H A DRunnerUtils.h202 void printMemRefShape(UnrankedMemRefType<T> &m) {
227 void printMemRef(UnrankedMemRefType<T> &m) {
332 int64_t verifyMemRef(UnrankedMemRefType<T> &actual,
333 UnrankedMemRefType<T> &expected) {
355 _mlir_ciface_printMemrefI8(UnrankedMemRefType<int8_t> *m);
357 _mlir_ciface_printMemrefI32(UnrankedMemRefType<int32_t> *m);
361 _mlir_ciface_printMemrefF32(UnrankedMemRefType<float> *m);
363 _mlir_ciface_printMemrefF64(UnrankedMemRefType<double> *m);
388 UnrankedMemRefType<int32_t> *actual, UnrankedMemRefType<int32_t> *expected);
390 UnrankedMemRefType<float> *actual, UnrankedMemRefType<float> *expected);
[all …]
H A DCRunnerUtils.h296 struct UnrankedMemRefType {
315 explicit DynamicMemRefType(const ::UnrankedMemRefType<T> &memRef)
337 memrefCopy(int64_t elemSize, ::UnrankedMemRefType<char> *src,
338 ::UnrankedMemRefType<char> *dst);
/llvm-project-15.0.7/mlir/lib/Conversion/LLVMCommon/
H A DTypeConverter.cpp38 [&](UnrankedMemRefType type) { return convertUnrankedMemRefType(type); }); in LLVMTypeConverter()
125 [&](OpBuilder &builder, UnrankedMemRefType resultType, ValueRange inputs, in LLVMTypeConverter()
262 if (t.isa<MemRefType, UnrankedMemRefType>()) in convertFunctionTypeCWrapper()
356 LLVMTypeConverter::getUnrankedMemRefDescriptorSize(UnrankedMemRefType type, in getUnrankedMemRefDescriptorSize()
364 Type LLVMTypeConverter::convertUnrankedMemRefType(UnrankedMemRefType type) { in convertUnrankedMemRefType()
373 if (type.isa<UnrankedMemRefType>()) in canConvertToBarePtr()
508 } else if (operand.getType().isa<UnrankedMemRefType>()) { in promoteOperands()
512 if (operand.getType().isa<UnrankedMemRefType>()) { in promoteOperands()
546 if (type.isa<UnrankedMemRefType>()) { in structFuncArgTypeConverter()
H A DPattern.cpp236 if (origTypes[i].isa<UnrankedMemRefType>()) in copyUnrankedDescriptors()
268 if (!type.isa<UnrankedMemRefType>()) in copyUnrankedDescriptors()
/llvm-project-15.0.7/cross-project-tests/debuginfo-tests/llvm-prettyprinters/gdb/
H A Dmlir-support.cpp18 mlir::Type UnrankedMemRefType = mlir::UnrankedMemRefType::get(IntegerType, 6); variable
H A Dmlir-support.gdb43 # CHECK-LABEL: +print UnrankedMemRefType
44 print UnrankedMemRefType
45 # CHECK: typeID = mlir::TypeID::get<mlir::UnrankedMemRefType>()
/llvm-project-15.0.7/mlir/lib/IR/
H A DBuiltinTypes.cpp386 .Case<MemRefType, UnrankedMemRefType>( in getElementType()
390 bool BaseMemRefType::hasRank() const { return !isa<UnrankedMemRefType>(); } in hasRank()
398 if (auto unrankedTy = dyn_cast<UnrankedMemRefType>()) { in cloneWith()
400 return UnrankedMemRefType::get(elementType, getMemorySpace()); in cloneWith()
416 return cast<UnrankedMemRefType>().getMemorySpace(); in getMemorySpace()
422 return cast<UnrankedMemRefType>().getMemorySpaceAsInt(); in getMemorySpaceAsInt()
702 unsigned UnrankedMemRefType::getMemorySpaceAsInt() const { in getMemorySpaceAsInt()
707 UnrankedMemRefType::verify(function_ref<InFlightDiagnostic()> emitError, in verify()
857 void UnrankedMemRefType::walkImmediateSubElements( in walkImmediateSubElements()
864 Type UnrankedMemRefType::replaceImmediateSubElements( in replaceImmediateSubElements()
/llvm-project-15.0.7/mlir/include/mlir/Conversion/LLVMCommon/
H A DTypeConverter.h125 unsigned getUnrankedMemRefDescriptorSize(UnrankedMemRefType type,
199 Type convertUnrankedMemRefType(UnrankedMemRefType type);
H A DMemRefBuilder.h24 class UnrankedMemRefType; variable
167 LLVMTypeConverter &converter, UnrankedMemRefType type,
/llvm-project-15.0.7/mlir/lib/CAPI/IR/
H A DBuiltinTypes.cpp289 return unwrap(type).isa<UnrankedMemRefType>(); in mlirTypeIsAUnrankedMemRef()
295 UnrankedMemRefType::get(unwrap(elementType), unwrap(memorySpace))); in mlirUnrankedMemRefTypeGet()
301 return wrap(UnrankedMemRefType::getChecked(unwrap(loc), unwrap(elementType), in mlirUnrankedMemRefTypeGetChecked()
306 return wrap(unwrap(type).cast<UnrankedMemRefType>().getMemorySpace()); in mlirUnrankedMemrefGetMemorySpace()
/llvm-project-15.0.7/mlir/unittests/IR/
H A DShapedTypeTest.cpp58 UnrankedMemRefType::get(memrefOriginalType, memSpace); in TEST()
63 UnrankedMemRefType::get(memrefNewType, memSpace)); in TEST()
/llvm-project-15.0.7/mlir/lib/Dialect/Arithmetic/Transforms/
H A DBufferizableOpInterfaceImpl.cpp104 auto unrankedMemrefType = sourceType.cast<UnrankedMemRefType>(); in bufferize()
105 resultType = UnrankedMemRefType::get(resultTensorType.getElementType(), in bufferize()
/llvm-project-15.0.7/mlir/lib/Conversion/MemRefToLLVM/
H A DMemRefToLLVM.cpp112 if (auto memRefElementType = elementType.dyn_cast<UnrankedMemRefType>()) in getMemRefEltSizeInBytes()
351 if (operandType.isa<UnrankedMemRefType>()) { in matchAndRewrite()
373 auto unrankedMemRefType = operandType.cast<UnrankedMemRefType>(); in extractSizeOfUnrankedMemRef()
793 assert(srcType.isa<UnrankedMemRefType>() || in match()
794 dstType.isa<UnrankedMemRefType>()); in match()
797 return !(srcType.isa<UnrankedMemRefType>() && in match()
798 dstType.isa<UnrankedMemRefType>()) in match()
814 if (srcType.isa<MemRefType>() && dstType.isa<UnrankedMemRefType>()) { in rewrite()
1014 operandType.cast<UnrankedMemRefType>().getMemorySpaceAsInt(); in extractPointersAndOffset()
1015 Type elementType = operandType.cast<UnrankedMemRefType>().getElementType(); in extractPointersAndOffset()
[all …]
/llvm-project-15.0.7/mlir/lib/Dialect/Bufferization/IR/
H A DBufferizableOpInterface.cpp629 replacement.getType().isa<UnrankedMemRefType>()) && in replaceOpWithBufferizedValues()
708 return UnrankedMemRefType::get(unrankedTensorType.getElementType(), in getMemRefType()
728 return UnrankedMemRefType::get(unrankedTensorType.getElementType(), in getMemRefTypeWithFullyDynamicLayout()
753 return UnrankedMemRefType::get(unrankedTensorType.getElementType(), in getMemRefTypeWithStaticIdentityLayout()
/llvm-project-15.0.7/mlir/test/python/ir/
H A Dbuiltin_types.py411 unranked_memref = UnrankedMemRefType.get(f32, Attribute.parse("2"))
438 memref_invalid = UnrankedMemRefType.get(none, Attribute.parse("2"))
/llvm-project-15.0.7/mlir/include/mlir/IR/
H A DBuiltinTypes.td808 // UnrankedMemRefType
855 return UnrankedMemRefType::get(elementType, memorySpaceAttr);
859 using ShapedType::Trait<UnrankedMemRefType>::clone;
860 using ShapedType::Trait<UnrankedMemRefType>::getElementTypeBitWidth;
861 using ShapedType::Trait<UnrankedMemRefType>::getRank;
862 using ShapedType::Trait<UnrankedMemRefType>::getNumElements;
863 using ShapedType::Trait<UnrankedMemRefType>::isDynamicDim;
864 using ShapedType::Trait<UnrankedMemRefType>::hasStaticShape;
865 using ShapedType::Trait<UnrankedMemRefType>::getNumDynamicDims;
866 using ShapedType::Trait<UnrankedMemRefType>::getDimSize;
[all …]
H A DBuiltinTypes.h370 return type.isa<MemRefType, UnrankedMemRefType>(); in classof()
375 type.isa<ComplexType, MemRefType, VectorType, UnrankedMemRefType>() || in isValidElementType()
/llvm-project-15.0.7/mlir/lib/Conversion/BufferizationToMemRef/
H A DBufferizationToMemRef.cpp38 if (type.isa<UnrankedMemRefType>()) { in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Conversion/FuncToLLVM/
H A DFuncToLLVM.cpp157 if (en.value().isa<UnrankedMemRefType>()) { in wrapForExternalCallers()
238 auto unrankedMemRefType = en.value().dyn_cast<UnrankedMemRefType>(); in wrapExternalFunction()
437 assert(!argTy.isa<UnrankedMemRefType>() && in matchAndRewrite()
612 } else if (oldTy.isa<UnrankedMemRefType>()) { in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Vector/Utils/
H A DVectorUtils.cpp39 if (source.getType().isa<UnrankedMemRefType, MemRefType>()) in createOrFoldDimOp()
/llvm-project-15.0.7/mlir/lib/Dialect/Bufferization/Transforms/
H A DBufferize.cpp49 return UnrankedMemRefType::get(type.getElementType(), 0); in BufferizeTypeConverter()
/llvm-project-15.0.7/mlir/lib/AsmParser/
H A DTypeParser.cpp272 return getChecked<UnrankedMemRefType>(loc, elementType, memorySpace); in parseMemRefType()
/llvm-project-15.0.7/mlir/lib/Dialect/MemRef/IR/
H A DMemRefOps.cpp94 !cast.getOperand().getType().isa<UnrankedMemRefType>()) { in foldMemRefCast()
107 if (auto memref = type.dyn_cast<UnrankedMemRefType>()) in getTensorTypeFromMemRefType()
572 auto uaT = a.dyn_cast<UnrankedMemRefType>(); in areCastCompatible()
573 auto ubT = b.dyn_cast<UnrankedMemRefType>(); in areCastCompatible()
769 } else if (type.isa<UnrankedMemRefType>()) { in verify()

12