14 #include "llvm/ADT/TypeSwitch.h"
15 #include "llvm/IR/DataLayout.h"
16 #include "llvm/IR/DerivedTypes.h"
17 #include "llvm/IR/Type.h"
33 if (knownTranslations.count(type))
34 return knownTranslations.lookup(type);
37 llvm::Type *translated =
39 .Case([
this](LLVM::LLVMVoidType) {
40 return llvm::Type::getVoidTy(context);
43 [
this](Float16Type) {
return llvm::Type::getHalfTy(context); })
44 .Case([
this](BFloat16Type) {
45 return llvm::Type::getBFloatTy(context);
48 [
this](Float32Type) {
return llvm::Type::getFloatTy(context); })
49 .Case([
this](Float64Type) {
50 return llvm::Type::getDoubleTy(context);
52 .Case([
this](Float80Type) {
53 return llvm::Type::getX86_FP80Ty(context);
55 .Case([
this](Float128Type) {
56 return llvm::Type::getFP128Ty(context);
58 .Case([
this](LLVM::LLVMPPCFP128Type) {
59 return llvm::Type::getPPC_FP128Ty(context);
61 .Case([
this](LLVM::LLVMTokenType) {
62 return llvm::Type::getTokenTy(context);
64 .Case([
this](LLVM::LLVMLabelType) {
65 return llvm::Type::getLabelTy(context);
67 .Case([
this](LLVM::LLVMMetadataType) {
68 return llvm::Type::getMetadataTy(context);
70 .Case<LLVM::LLVMArrayType, IntegerType, LLVM::LLVMFunctionType,
72 LLVM::LLVMFixedVectorType, LLVM::LLVMScalableVectorType,
73 VectorType, LLVM::LLVMTargetExtType>(
74 [
this](
auto type) {
return this->translate(type); })
75 .Default([](
Type t) -> llvm::Type * {
76 llvm_unreachable(
"unknown LLVM dialect type");
80 knownTranslations.try_emplace(type, translated);
86 llvm::Type *translate(LLVM::LLVMArrayType type) {
88 type.getNumElements());
92 llvm::Type *translate(LLVM::LLVMFunctionType type) {
94 translateTypes(type.getParams(), paramTypes);
96 paramTypes, type.isVarArg());
100 llvm::Type *translate(IntegerType type) {
105 llvm::Type *translate(LLVM::LLVMPointerType type) {
115 translateTypes(type.
getBody(), subtypes);
119 llvm::StructType *structType =
120 llvm::StructType::create(context, type.
getName());
123 knownTranslations.try_emplace(type, structType);
127 translateTypes(type.
getBody(), subtypes);
128 structType->setBody(subtypes, type.
isPacked());
133 llvm::Type *translate(VectorType type) {
135 "expected compatible with LLVM vector type");
136 if (type.isScalable())
138 type.getNumElements());
140 type.getNumElements());
144 llvm::Type *translate(LLVM::LLVMFixedVectorType type) {
146 type.getNumElements());
150 llvm::Type *translate(LLVM::LLVMScalableVectorType type) {
152 type.getMinNumElements());
156 llvm::Type *translate(LLVM::LLVMTargetExtType type) {
158 translateTypes(type.getTypeParams(), typeParams);
160 type.getIntParams());
166 result.reserve(result.size() + types.size());
167 for (
auto type : types)
172 llvm::LLVMContext &context;
185 :
impl(new detail::TypeToLLVMIRTranslatorImpl(context)) {}
190 return impl->translateType(type);
194 Type type,
const llvm::DataLayout &layout) {
195 return layout.getPrefTypeAlign(translateType(type)).value();
LLVM dialect structure type representing a collection of different-typed elements manipulated togethe...
StringRef getName()
Returns the name of an identified struct.
bool isPacked() const
Checks if a struct is packed.
ArrayRef< Type > getBody() const
Returns the list of element types contained in a non-opaque struct.
bool isOpaque()
Checks if a struct is opaque.
bool isIdentified() const
Checks if a struct is identified.
unsigned getPreferredAlignment(Type type, const llvm::DataLayout &layout)
Returns the preferred alignment for the type given the data layout.
llvm::Type * translateType(Type type)
Translates the given MLIR LLVM dialect type to LLVM IR.
TypeToLLVMIRTranslator(llvm::LLVMContext &context)
~TypeToLLVMIRTranslator()
Support for translating MLIR LLVM dialect types to LLVM IR.
TypeToLLVMIRTranslatorImpl(llvm::LLVMContext &context)
Constructs a class creating types in the given LLVM context.
llvm::Type * translateType(Type type)
Translates a single type.
Instances of the Type class are uniqued, have an immutable identifier and an optional mutable compone...
bool isCompatibleVectorType(Type type)
Returns true if the given type is a vector type compatible with the LLVM dialect.
Include the generated interface declarations.
auto get(MLIRContext *context, Ts &&...params)
Helper method that injects context only if needed, this helps unify some of the attribute constructio...