MLIR  17.0.0git
Public Member Functions | Static Public Member Functions | Protected Attributes | Friends | List of all members
mlir::LLVMTypeConverter Class Reference

Conversion from types to the LLVM IR dialect. More...

#include "mlir/Conversion/LLVMCommon/TypeConverter.h"

+ Inheritance diagram for mlir::LLVMTypeConverter:

Public Member Functions

 LLVMTypeConverter (MLIRContext *ctx, const DataLayoutAnalysis *analysis=nullptr)
 Create an LLVMTypeConverter using the default LowerToLLVMOptions. More...
 
 LLVMTypeConverter (MLIRContext *ctx, const LowerToLLVMOptions &options, const DataLayoutAnalysis *analysis=nullptr)
 Create an LLVMTypeConverter using custom LowerToLLVMOptions. More...
 
Type convertFunctionSignature (FunctionType funcTy, bool isVariadic, bool useBarePtrCallConv, SignatureConversion &result)
 Convert a function type. More...
 
Type packFunctionResults (TypeRange types, bool useBarePointerCallConv=false)
 Convert a non-empty list of types to be returned from a function into an LLVM-compatible type. More...
 
Type packOperationResults (TypeRange types)
 Convert a non-empty list of types of values produced by an operation into an LLVM-compatible type. More...
 
Type convertCallingConventionType (Type type, bool useBarePointerCallConv=false)
 Convert a type in the context of the default or bare pointer calling convention. More...
 
void promoteBarePtrsToDescriptors (ConversionPatternRewriter &rewriter, Location loc, ArrayRef< Type > stdTypes, SmallVectorImpl< Value > &values)
 Promote the bare pointers in 'values' that resulted from memrefs to descriptors. More...
 
MLIRContextgetContext ()
 Returns the MLIR context. More...
 
LLVM::LLVMDialect * getDialect ()
 Returns the LLVM dialect. More...
 
const LowerToLLVMOptionsgetOptions () const
 
void dangerousSetOptions (LowerToLLVMOptions newOptions)
 Set the lowering options to newOptions. More...
 
SmallVector< Value, 4 > promoteOperands (Location loc, ValueRange opOperands, ValueRange operands, OpBuilder &builder, bool useBarePtrCallConv=false)
 Promote the LLVM representation of all operands including promoting MemRef descriptors to stack and use pointers to struct to avoid the complexity of the platform-specific C/C++ ABI lowering related to struct argument passing. More...
 
Value promoteOneMemRefDescriptor (Location loc, Value operand, OpBuilder &builder)
 Promote the LLVM struct representation of one MemRef descriptor to stack and use pointer to struct to avoid the complexity of the platform-specific C/C++ ABI lowering related to struct argument passing. More...
 
std::pair< LLVM::LLVMFunctionType, LLVM::LLVMStructTypeconvertFunctionTypeCWrapper (FunctionType type)
 Converts the function type to a C-compatible format, in particular using pointers to memref descriptors for arguments. More...
 
const llvm::DataLayout & getDataLayout ()
 Returns the data layout to use during and after conversion. More...
 
const DataLayoutAnalysisgetDataLayoutAnalysis () const
 Returns the data layout analysis to query during conversion. More...
 
Type getIndexType ()
 Gets the LLVM representation of the index type. More...
 
bool useOpaquePointers () const
 Returns true if using opaque pointers was enabled in the lowering options. More...
 
LLVM::LLVMPointerType getPointerType (Type elementType, unsigned addressSpace=0)
 Creates an LLVM pointer type with the given element type and address space. More...
 
unsigned getIndexTypeBitwidth ()
 Gets the bitwidth of the index type when converted to LLVM. More...
 
unsigned getPointerBitwidth (unsigned addressSpace=0)
 Gets the pointer bitwidth. More...
 
unsigned getMemRefDescriptorSize (MemRefType type, const DataLayout &layout)
 Returns the size of the memref descriptor object in bytes. More...
 
unsigned getUnrankedMemRefDescriptorSize (UnrankedMemRefType type, const DataLayout &layout)
 Returns the size of the unranked memref descriptor object in bytes. More...
 
FailureOr< unsigned > getMemRefAddressSpace (BaseMemRefType type)
 Return the LLVM address space corresponding to the memory space of the memref type type or failure if the memory space cannot be converted to an integer. More...
 
LogicalResult convertType (Type t, SmallVectorImpl< Type > &results)
 Convert the given type. More...
 
Type convertType (Type t)
 This hook simplifies defining 1-1 type conversions. More...
 
template<typename TargetType >
TargetType convertType (Type t)
 Attempts a 1-1 type conversion, expecting the result type to be TargetType. More...
 
- Public Member Functions inherited from mlir::TypeConverter
template<typename FnT , typename T = typename llvm::function_traits< std::decay_t<FnT>>::template arg_t<0>>
void addConversion (FnT &&callback)
 Register a conversion function. More...
 
template<typename FnT , typename T = typename llvm::function_traits< std::decay_t<FnT>>::template arg_t<1>>
void addArgumentMaterialization (FnT &&callback)
 Register a materialization function, which must be convertible to the following form: std::optional<Value>(OpBuilder &, T, ValueRange, Location), where T is any subclass of Type. More...
 
template<typename FnT , typename T = typename llvm::function_traits< std::decay_t<FnT>>::template arg_t<1>>
void addSourceMaterialization (FnT &&callback)
 This method registers a materialization that will be called when converting a legal type to an illegal source type. More...
 
template<typename FnT , typename T = typename llvm::function_traits< std::decay_t<FnT>>::template arg_t<1>>
void addTargetMaterialization (FnT &&callback)
 This method registers a materialization that will be called when converting type from an illegal, or source, type to a legal type. More...
 
template<typename FnT , typename T = typename llvm::function_traits<std::decay_t<FnT>>::template arg_t<0>, typename A = typename llvm::function_traits<std::decay_t<FnT>>::template arg_t<1>>
void addTypeAttributeConversion (FnT &&callback)
 Register a conversion function for attributes within types. More...
 
LogicalResult convertType (Type t, SmallVectorImpl< Type > &results)
 Convert the given type. More...
 
Type convertType (Type t)
 This hook simplifies defining 1-1 type conversions. More...
 
template<typename TargetType >
TargetType convertType (Type t)
 Attempts a 1-1 type conversion, expecting the result type to be TargetType. More...
 
LogicalResult convertTypes (TypeRange types, SmallVectorImpl< Type > &results)
 Convert the given set of types, filling 'results' as necessary. More...
 
bool isLegal (Type type)
 Return true if the given type is legal for this type converter, i.e. More...
 
template<typename RangeT >
std::enable_if_t<!std::is_convertible< RangeT, Type >::value &&!std::is_convertible< RangeT, Operation * >::value, bool > isLegal (RangeT &&range)
 Return true if all of the given types are legal for this type converter. More...
 
bool isLegal (Operation *op)
 Return true if the given operation has legal operand and result types. More...
 
bool isLegal (Region *region)
 Return true if the types of block arguments within the region are legal. More...
 
bool isSignatureLegal (FunctionType ty)
 Return true if the inputs and outputs of the given function type are legal. More...
 
LogicalResult convertSignatureArg (unsigned inputNo, Type type, SignatureConversion &result)
 This method allows for converting a specific argument of a signature. More...
 
LogicalResult convertSignatureArgs (TypeRange types, SignatureConversion &result, unsigned origInputOffset=0)
 
std::optional< SignatureConversionconvertBlockSignature (Block *block)
 This function converts the type signature of the given block, by invoking 'convertSignatureArg' for each argument. More...
 
Value materializeArgumentConversion (OpBuilder &builder, Location loc, Type resultType, ValueRange inputs)
 Materialize a conversion from a set of types into one result type by generating a cast sequence of some kind. More...
 
Value materializeSourceConversion (OpBuilder &builder, Location loc, Type resultType, ValueRange inputs)
 
Value materializeTargetConversion (OpBuilder &builder, Location loc, Type resultType, ValueRange inputs)
 
std::optional< AttributeconvertTypeAttribute (Type type, Attribute attr)
 Convert an attribute present attr from within the type type using the registered conversion functions. More...
 

Static Public Member Functions

static bool canConvertToBarePtr (BaseMemRefType type)
 Check if a memref type can be converted to a bare pointer. More...
 

Protected Attributes

LLVM::LLVMDialect * llvmDialect
 Pointer to the LLVM dialect. More...
 

Friends

LogicalResult structFuncArgTypeConverter (LLVMTypeConverter &converter, Type type, SmallVectorImpl< Type > &result)
 Give structFuncArgTypeConverter access to memref-specific functions. More...
 

Detailed Description

Conversion from types to the LLVM IR dialect.

Definition at line 33 of file TypeConverter.h.

Constructor & Destructor Documentation

◆ LLVMTypeConverter() [1/2]

LLVMTypeConverter::LLVMTypeConverter ( MLIRContext ctx,
const DataLayoutAnalysis analysis = nullptr 
)

Create an LLVMTypeConverter using the default LowerToLLVMOptions.

Create an LLVMTypeConverter using default LowerToLLVMOptions.

Optionally takes a data layout analysis to use in conversions.

Definition at line 19 of file TypeConverter.cpp.

◆ LLVMTypeConverter() [2/2]

LLVMTypeConverter::LLVMTypeConverter ( MLIRContext ctx,
const LowerToLLVMOptions options,
const DataLayoutAnalysis analysis = nullptr 
)

Member Function Documentation

◆ canConvertToBarePtr()

bool LLVMTypeConverter::canConvertToBarePtr ( BaseMemRefType  type)
static

Check if a memref type can be converted to a bare pointer.

Definition at line 414 of file TypeConverter.cpp.

References mlir::failed(), and mlir::getStridesAndOffset().

Referenced by canBeCalledWithBarePointers().

◆ convertCallingConventionType()

Type LLVMTypeConverter::convertCallingConventionType ( Type  type,
bool  useBarePtrCallConv = false 
)

Convert a type in the context of the default or bare pointer calling convention.

Calling convention sensitive types, such as MemRefType and UnrankedMemRefType, are converted following the specific rules for the calling convention. Calling convention independent types are converted following the default LLVM type conversions.

Definition at line 476 of file TypeConverter.cpp.

References convertType().

Referenced by mlir::barePtrFuncArgTypeConverter(), and packFunctionResults().

◆ convertFunctionSignature()

Type LLVMTypeConverter::convertFunctionSignature ( FunctionType  funcTy,
bool  isVariadic,
bool  useBarePtrCallConv,
LLVMTypeConverter::SignatureConversion result 
)

Convert a function type.

The arguments and results are converted one by one and results are packed into a wrapped LLVM IR structure type. result is populated with argument mapping.

Definition at line 223 of file TypeConverter.cpp.

References mlir::TypeConverter::SignatureConversion::addInputs(), mlir::barePtrFuncArgTypeConverter(), mlir::detail::enumerate(), mlir::failed(), mlir::get(), getContext(), mlir::TypeConverter::SignatureConversion::getConvertedTypes(), packFunctionResults(), structFuncArgTypeConverter, and mlir::LowerToLLVMOptions::useBarePtrCallConv.

Referenced by mlir::GPUFuncOpLowering::matchAndRewrite().

◆ convertFunctionTypeCWrapper()

std::pair< LLVM::LLVMFunctionType, LLVM::LLVMStructType > LLVMTypeConverter::convertFunctionTypeCWrapper ( FunctionType  type)

Converts the function type to a C-compatible format, in particular using pointers to memref descriptors for arguments.

Also converts the return type to a pointer argument if it is a struct. Returns true if this was the case.

Definition at line 254 of file TypeConverter.cpp.

References convertType(), mlir::get(), getContext(), getPointerType(), mlir::LLVM::isCompatibleType(), and packFunctionResults().

Referenced by wrapExternalFunction(), and wrapForExternalCallers().

◆ convertType() [1/3]

Type TypeConverter::convertType

This hook simplifies defining 1-1 type conversions.

This function returns the type to convert to on success, and a null type on failure.

Definition at line 224 of file DialectConversion.cpp.

◆ convertType() [2/3]

template<typename TargetType >
TargetType mlir::TypeConverter::convertType ( typename TargetType  )
inline

Attempts a 1-1 type conversion, expecting the result type to be TargetType.

Returns the converted type cast to TargetType on success, and a null type on conversion or cast failure.

Definition at line 230 of file DialectConversion.h.

◆ convertType() [3/3]

LogicalResult TypeConverter::convertType

◆ dangerousSetOptions()

void mlir::LLVMTypeConverter::dangerousSetOptions ( LowerToLLVMOptions  newOptions)
inline

Set the lowering options to newOptions.

Note: using this after some some conversions have been performed can lead to inconsistencies in the IR.

Definition at line 98 of file TypeConverter.h.

◆ getContext()

MLIRContext & LLVMTypeConverter::getContext ( )

◆ getDataLayout()

const llvm::DataLayout& mlir::LLVMTypeConverter::getDataLayout ( )
inline

Returns the data layout to use during and after conversion.

Definition at line 124 of file TypeConverter.h.

References mlir::LowerToLLVMOptions::dataLayout.

Referenced by getMemRefAlignment().

◆ getDataLayoutAnalysis()

const DataLayoutAnalysis* mlir::LLVMTypeConverter::getDataLayoutAnalysis ( ) const
inline

Returns the data layout analysis to query during conversion.

Definition at line 127 of file TypeConverter.h.

◆ getDialect()

LLVM::LLVMDialect* mlir::LLVMTypeConverter::getDialect ( )
inline

Returns the LLVM dialect.

Definition at line 91 of file TypeConverter.h.

References llvmDialect.

Referenced by getContext(), mlir::ConvertToLLVMPattern::getDialect(), and mlir::populateVectorToLLVMConversionPatterns().

◆ getIndexType()

Type LLVMTypeConverter::getIndexType ( )

◆ getIndexTypeBitwidth()

unsigned mlir::LLVMTypeConverter::getIndexTypeBitwidth ( )
inline

Gets the bitwidth of the index type when converted to LLVM.

Definition at line 147 of file TypeConverter.h.

References mlir::LowerToLLVMOptions::getIndexBitwidth().

Referenced by mlir::UnrankedMemRefDescriptor::computeSizes(), and getIndexType().

◆ getMemRefAddressSpace()

FailureOr< unsigned > LLVMTypeConverter::getMemRefAddressSpace ( BaseMemRefType  type)

Return the LLVM address space corresponding to the memory space of the memref type type or failure if the memory space cannot be converted to an integer.

Definition at line 399 of file TypeConverter.cpp.

References mlir::TypeConverter::convertTypeAttribute(), mlir::failure(), and mlir::BaseMemRefType::getMemorySpace().

Referenced by castAllocFuncResult(), castDataPtr(), mlir::ConvertToLLVMPattern::copyUnrankedDescriptors(), mlir::ConvertToLLVMPattern::getElementPtrType(), getMemRefDescriptorSize(), getUnrankedMemRefDescriptorSize(), and isMemRefTypeSupported().

◆ getMemRefDescriptorSize()

unsigned LLVMTypeConverter::getMemRefDescriptorSize ( MemRefType  type,
const DataLayout layout 
)

Returns the size of the memref descriptor object in bytes.

Definition at line 351 of file TypeConverter.cpp.

References getIndexType(), getMemRefAddressSpace(), getPointerBitwidth(), and mlir::DataLayout::getTypeSize().

◆ getOptions()

const LowerToLLVMOptions& mlir::LLVMTypeConverter::getOptions ( ) const
inline

◆ getPointerBitwidth()

unsigned LLVMTypeConverter::getPointerBitwidth ( unsigned  addressSpace = 0)

◆ getPointerType()

LLVM::LLVMPointerType LLVMTypeConverter::getPointerType ( Type  elementType,
unsigned  addressSpace = 0 
)

◆ getUnrankedMemRefDescriptorSize()

unsigned LLVMTypeConverter::getUnrankedMemRefDescriptorSize ( UnrankedMemRefType  type,
const DataLayout layout 
)

Returns the size of the unranked memref descriptor object in bytes.

Definition at line 383 of file TypeConverter.cpp.

References getIndexType(), getMemRefAddressSpace(), getPointerBitwidth(), and mlir::DataLayout::getTypeSize().

◆ packFunctionResults()

Type LLVMTypeConverter::packFunctionResults ( TypeRange  types,
bool  useBarePtrCallConv = false 
)

Convert a non-empty list of types to be returned from a function into an LLVM-compatible type.

In particular, if more than one value is returned, create an LLVM dialect structure type with elements that correspond to each of the types converted with convertCallingConventionType.

Definition at line 524 of file TypeConverter.cpp.

References convertCallingConventionType(), getContext(), mlir::LLVM::LLVMStructType::getLiteral(), mlir::LLVM::isCompatibleType(), and mlir::LowerToLLVMOptions::useBarePtrCallConv.

Referenced by convertFunctionSignature(), and convertFunctionTypeCWrapper().

◆ packOperationResults()

Type LLVMTypeConverter::packOperationResults ( TypeRange  types)

Convert a non-empty list of types of values produced by an operation into an LLVM-compatible type.

In particular, if more than one value is produced, create a literal structure with elements that correspond to each of the LLVM-compatible types converted with convertType.

In particular, if more than one value is produced, create a literal structure with elements that correspond to each of the types converted with convertType.

Definition at line 503 of file TypeConverter.cpp.

References convertType(), getContext(), mlir::LLVM::LLVMStructType::getLiteral(), and mlir::LLVM::isCompatibleType().

◆ promoteBarePtrsToDescriptors()

void LLVMTypeConverter::promoteBarePtrsToDescriptors ( ConversionPatternRewriter rewriter,
Location  loc,
ArrayRef< Type stdTypes,
SmallVectorImpl< Value > &  values 
)

Promote the bare pointers in 'values' that resulted from memrefs to descriptors.

'stdTypes' holds the types of 'values' before the conversion to the LLVM-IR dialect (i.e., MemRefType, or any other builtin type).

'stdTypes' holds they types of 'values' before the conversion to the LLVM-IR dialect (i.e., MemRefType, or any other builtin type).

Definition at line 488 of file TypeConverter.cpp.

References mlir::MemRefDescriptor::fromStaticShape().

◆ promoteOneMemRefDescriptor()

Value LLVMTypeConverter::promoteOneMemRefDescriptor ( Location  loc,
Value  operand,
OpBuilder builder 
)

Promote the LLVM struct representation of one MemRef descriptor to stack and use pointer to struct to avoid the complexity of the platform-specific C/C++ ABI lowering related to struct argument passing.

Definition at line 544 of file TypeConverter.cpp.

References mlir::OpBuilder::create(), mlir::Builder::getI64Type(), mlir::Builder::getIndexAttr(), getPointerType(), and mlir::Value::getType().

◆ promoteOperands()

SmallVector< Value, 4 > LLVMTypeConverter::promoteOperands ( Location  loc,
ValueRange  opOperands,
ValueRange  operands,
OpBuilder builder,
bool  useBarePtrCallConv = false 
)

Promote the LLVM representation of all operands including promoting MemRef descriptors to stack and use pointers to struct to avoid the complexity of the platform-specific C/C++ ABI lowering related to struct argument passing.

Definition at line 559 of file TypeConverter.cpp.

References mlir::MemRefDescriptor::alignedPtr(), mlir::MemRefDescriptor::unpack(), mlir::UnrankedMemRefDescriptor::unpack(), and mlir::LowerToLLVMOptions::useBarePtrCallConv.

◆ useOpaquePointers()

bool mlir::LLVMTypeConverter::useOpaquePointers ( ) const
inline

Returns true if using opaque pointers was enabled in the lowering options.

Definition at line 136 of file TypeConverter.h.

References getOptions(), and mlir::LowerToLLVMOptions::useOpaquePointers.

Referenced by bitAndAddrspaceCast(), castAllocFuncResult(), castDataPtr(), createPrintMsg(), and getPointerType().

Friends And Related Function Documentation

◆ structFuncArgTypeConverter

LogicalResult structFuncArgTypeConverter ( LLVMTypeConverter converter,
Type  type,
SmallVectorImpl< Type > &  result 
)
friend

Give structFuncArgTypeConverter access to memref-specific functions.

It converts a MemRef function argument to a list of non-aggregate types containing descriptor information, and an UnrankedmemRef function argument to a list containing the rank and a pointer to a descriptor struct.

Referenced by convertFunctionSignature().

Member Data Documentation

◆ llvmDialect

LLVM::LLVMDialect* mlir::LLVMTypeConverter::llvmDialect
protected

Pointer to the LLVM dialect.

Definition at line 169 of file TypeConverter.h.

Referenced by getDialect(), and LLVMTypeConverter().


The documentation for this class was generated from the following files: