#include "mlir/Dialect/LLVMIR/LLVMDialect.h"#include "TypeDetail.h"#include "mlir/Dialect/LLVMIR/LLVMTypes.h"#include "mlir/IR/Builders.h"#include "mlir/IR/BuiltinOps.h"#include "mlir/IR/BuiltinTypes.h"#include "mlir/IR/DialectImplementation.h"#include "mlir/IR/FunctionImplementation.h"#include "mlir/IR/MLIRContext.h"#include "mlir/IR/Matchers.h"#include "llvm/ADT/TypeSwitch.h"#include "llvm/AsmParser/Parser.h"#include "llvm/Bitcode/BitcodeReader.h"#include "llvm/Bitcode/BitcodeWriter.h"#include "llvm/IR/Attributes.h"#include "llvm/IR/Function.h"#include "llvm/IR/Type.h"#include "llvm/Support/Error.h"#include "llvm/Support/Mutex.h"#include "llvm/Support/SourceMgr.h"#include <numeric>#include "mlir/Dialect/LLVMIR/LLVMOpsDialect.cpp.inc"#include "mlir/Dialect/LLVMIR/LLVMOpsInterfaces.cpp.inc"#include "mlir/Dialect/LLVMIR/LLVMOps.cpp.inc"#include "mlir/Dialect/LLVMIR/LLVMIntrinsicOps.cpp.inc"
Include dependency graph for LLVMDialect.cpp:
Go to the source code of this file.
Macros | |
| #define | REGISTER_ENUM_TYPE(Ty) |
| #define | GET_OP_LIST |
| #define | GET_OP_LIST |
| #define | GET_OP_CLASSES |
| #define | GET_OP_CLASSES |
Functions | |
| static auto | processFMFAttr (ArrayRef< NamedAttribute > attrs) |
| static ParseResult | parseLLVMOpAttrs (OpAsmParser &parser, NamedAttrList &result) |
| static void | printLLVMOpAttrs (OpAsmPrinter &printer, Operation *op, DictionaryAttr attrs) |
| static LogicalResult | verifySymbolAttrUse (FlatSymbolRefAttr symbol, Operation *op, SymbolTableCollection &symbolTable) |
Verifies symbol's use in op to ensure the symbol is a valid and fully defined llvm.func. More... | |
| static Type | getI1SameShape (Type type) |
Returns a boolean type that has the same shape as type. More... | |
| template<typename CmpPredicateType > | |
| static ParseResult | parseCmpOp (OpAsmParser &parser, OperationState &result) |
| static LogicalResult | verifyOpaquePtr (Operation *op, LLVMPointerType ptrType, Optional< Type > ptrElementType) |
| Checks that the elemental type is present in either the pointer type or the attribute, but not both. More... | |
| static ParseResult | parseSwitchOpCases (OpAsmParser &parser, Type flagType, ElementsAttr &caseValues, SmallVectorImpl< Block * > &caseDestinations, SmallVectorImpl< SmallVector< OpAsmParser::UnresolvedOperand >> &caseOperands, SmallVectorImpl< SmallVector< Type >> &caseOperandTypes) |
<cases> ::= integer : bb-id (( ssa-use-and-type-list ))? ( , integer : bb-id (( ssa-use-and-type-list ))? )? More... | |
| static void | printSwitchOpCases (OpAsmPrinter &p, SwitchOp op, Type flagType, ElementsAttr caseValues, SuccessorRange caseDestinations, OperandRangeRange caseOperands, const TypeRangeRange &caseOperandTypes) |
| static Type | extractVectorElementType (Type type) |
| Returns the elemental type of any LLVM-compatible vector type or self. More... | |
| static void | destructureIndices (Type currType, ArrayRef< GEPArg > indices, SmallVectorImpl< int32_t > &rawConstantIndices, SmallVectorImpl< Value > &dynamicIndices) |
| Destructures the 'indices' parameter into 'rawConstantIndices' and 'dynamicIndices', encoding the former in the process. More... | |
| static ParseResult | parseGEPIndices (OpAsmParser &parser, SmallVectorImpl< OpAsmParser::UnresolvedOperand > &indices, DenseI32ArrayAttr &rawConstantIndices) |
| static void | printGEPIndices (OpAsmPrinter &printer, LLVM::GEPOp gepOp, OperandRange indices, DenseI32ArrayAttr rawConstantIndices) |
| static llvm::Error | verifyStructIndices (Type baseGEPType, unsigned indexPos, GEPIndicesAdaptor< ValueRange > indices) |
For the given structIndices and indices, check if they're complied with baseGEPType, especially check against LLVMStructTypes nested within. More... | |
| static llvm::Error | verifyStructIndices (Type baseGEPType, GEPIndicesAdaptor< ValueRange > indices) |
Driver function around recordStructIndices. More... | |
| LogicalResult | verifySymbolAttribute (Operation *op, StringRef attributeName, llvm::function_ref< LogicalResult(Operation *, SymbolRefAttr)> verifySymbolType) |
| template<typename OpTy > | |
| static LogicalResult | verifyOpMetadata (Operation *op, StringRef attributeName) |
| static LogicalResult | verifyMemoryOpMetadata (Operation *op) |
| static Optional< Type > | getLoadStoreElementType (OpAsmParser &parser, Type type, SMLoc trailingTypeLoc) |
| static Type | getInsertExtractValueElementType (function_ref< InFlightDiagnostic(StringRef)> emitError, Type containerType, ArrayRef< int64_t > position) |
Extract the type at position in the LLVM IR aggregate type containerType. More... | |
| static Type | getInsertExtractValueElementType (Type llvmType, ArrayRef< int64_t > position) |
Extract the type at position in the wrapped LLVM IR aggregate type containerType. More... | |
| static ParseResult | parseInsertExtractValueElementType (AsmParser &parser, Type &valueType, Type containerType, DenseI64ArrayAttr position) |
| Infer the value type from the container type and position. More... | |
| static void | printInsertExtractValueElementType (AsmPrinter &printer, Operation *op, Type valueType, Type containerType, DenseI64ArrayAttr position) |
| Nothing to print for an inferred type. More... | |
| static Operation * | parentLLVMModule (Operation *op) |
| static int | parseOptionalKeywordAlternative (OpAsmParser &parser, ArrayRef< StringRef > keywords) |
| template<typename EnumTy , typename RetTy = EnumTy> | |
| static RetTy | parseOptionalLLVMKeyword (OpAsmParser &parser, OperationState &result, EnumTy defaultValue) |
| Parse an enum from the keyword, or default to the provided default value. More... | |
| static bool | isZeroAttribute (Attribute value) |
| static ParseResult | parseShuffleType (AsmParser &parser, Type v1Type, Type &resType, DenseI32ArrayAttr mask) |
| Build the result type of a shuffle vector operation. More... | |
| static void | printShuffleType (AsmPrinter &printer, Operation *op, Type v1Type, Type resType, DenseI32ArrayAttr mask) |
| Nothing to do when the result type is inferred. More... | |
| static Type | buildLLVMFunctionType (OpAsmParser &parser, SMLoc loc, ArrayRef< Type > inputs, ArrayRef< Type > outputs, function_interface_impl::VariadicFlag variadicFlag) |
| static ParseResult | parseAtomicBinOp (OpAsmParser &parser, OperationState &result, StringRef attrName) |
| static ParseResult | parseAtomicOrdering (OpAsmParser &parser, OperationState &result, StringRef attrName) |
| static LogicalResult | verifyFuncOpInterfaceStructAttr (Operation *op, Attribute attr, const std::function< Type(FunctionOpInterface)> &getAnnotatedType) |
Variables | |
| static constexpr const char | kVolatileAttrName [] = "volatile_" |
| static constexpr const char | kNonTemporalAttrName [] = "nontemporal" |
| static constexpr const char | kElemTypeAttrName [] = "elem_type" |
Macro Definition Documentation
◆ GET_OP_CLASSES [1/2]
| #define GET_OP_CLASSES |
Definition at line 2607 of file LLVMDialect.cpp.
◆ GET_OP_CLASSES [2/2]
| #define GET_OP_CLASSES |
Definition at line 2607 of file LLVMDialect.cpp.
◆ GET_OP_LIST [1/2]
| #define GET_OP_LIST |
◆ GET_OP_LIST [2/2]
| #define GET_OP_LIST |
◆ REGISTER_ENUM_TYPE
| #define REGISTER_ENUM_TYPE | ( | Ty | ) |
Definition at line 1698 of file LLVMDialect.cpp.
Function Documentation
◆ buildLLVMFunctionType()
|
static |
Definition at line 2017 of file LLVMDialect.cpp.
References mlir::AsmParser::emitError(), mlir::AsmParser::getBuilder(), mlir::Builder::getContext(), mlir::LLVM::isCompatibleType(), and mlir::function_interface_impl::VariadicFlag::isVariadic().
◆ destructureIndices()
|
static |
Destructures the 'indices' parameter into 'rawConstantIndices' and 'dynamicIndices', encoding the former in the process.
In the process, dynamic indices which are used to index into a structure type are converted to constant indices when possible. To do this, the GEPs element type should be passed as first parameter.
Definition at line 437 of file LLVMDialect.cpp.
References mlir::Value::dyn_cast(), mlir::LLVM::LLVMStructType::getBody(), mlir::Type::isa_and_nonnull(), mlir::LLVM::kGEPConstantBitWidth, mlir::m_ConstantInt(), and mlir::matchPattern().
◆ extractVectorElementType()
Returns the elemental type of any LLVM-compatible vector type or self.
Definition at line 410 of file LLVMDialect.cpp.
References mlir::Type::dyn_cast().
◆ getI1SameShape()
Returns a boolean type that has the same shape as type.
It supports both fixed size vectors as well as scalable vectors.
Definition at line 91 of file LLVMDialect.cpp.
References mlir::Type::getContext(), mlir::LLVM::getVectorNumElements(), mlir::LLVM::getVectorType(), and mlir::LLVM::isCompatibleVectorType().
◆ getInsertExtractValueElementType() [1/2]
|
static |
Extract the type at position in the LLVM IR aggregate type containerType.
Each element of position is an index into a nested aggregate type. Return the resulting type or emit an error.
Definition at line 1369 of file LLVMDialect.cpp.
References mlir::Type::dyn_cast(), mlir::emitError(), and mlir::LLVM::isCompatibleType().
Referenced by parseInsertExtractValueElementType().
◆ getInsertExtractValueElementType() [2/2]
Extract the type at position in the wrapped LLVM IR aggregate type containerType.
Definition at line 1406 of file LLVMDialect.cpp.
References mlir::Type::cast(), mlir::Type::dyn_cast(), and getElementType().
◆ getLoadStoreElementType()
|
static |
Definition at line 774 of file LLVMDialect.cpp.
References mlir::Type::dyn_cast(), and mlir::AsmParser::emitError().
◆ isZeroAttribute()
|
static |
Definition at line 1801 of file LLVMDialect.cpp.
References value.
◆ parentLLVMModule()
Definition at line 1556 of file LLVMDialect.cpp.
References mlir::Operation::getParentOp(), and mlir::LLVM::satisfiesLLVMModule().
◆ parseAtomicBinOp()
|
static |
Definition at line 2247 of file LLVMDialect.cpp.
References mlir::failure(), mlir::AsmParser::getCurrentLocation(), and mlir::AsmParser::parseKeyword().
◆ parseAtomicOrdering()
|
static |
Definition at line 2273 of file LLVMDialect.cpp.
References mlir::failure(), mlir::AsmParser::getCurrentLocation(), and mlir::AsmParser::parseKeyword().
◆ parseCmpOp()
|
static |
Definition at line 131 of file LLVMDialect.cpp.
◆ parseGEPIndices()
|
static |
Definition at line 516 of file LLVMDialect.cpp.
References mlir::sparse_tensor::constantIndex(), mlir::failed(), mlir::failure(), mlir::detail::DenseArrayAttrImpl< int32_t >::get(), mlir::AsmParser::getContext(), mlir::OptionalParseResult::has_value(), mlir::AsmParser::parseCommaSeparatedList(), mlir::OpAsmParser::parseOperand(), mlir::AsmParser::parseOptionalInteger(), mlir::success(), and mlir::OptionalParseResult::value().
◆ parseInsertExtractValueElementType()
|
static |
Infer the value type from the container type and position.
Definition at line 1477 of file LLVMDialect.cpp.
References mlir::detail::DenseArrayAttrImpl< T >::asArrayRef(), mlir::AsmParser::emitError(), mlir::AsmParser::getCurrentLocation(), getInsertExtractValueElementType(), and mlir::success().
◆ parseLLVMOpAttrs()
|
static |
Definition at line 63 of file LLVMDialect.cpp.
References mlir::AsmParser::parseOptionalAttrDict().
◆ parseOptionalKeywordAlternative()
|
static |
Definition at line 1685 of file LLVMDialect.cpp.
References mlir::detail::enumerate(), mlir::AsmParser::parseOptionalKeyword(), and mlir::succeeded().
Referenced by parseOptionalLLVMKeyword().
◆ parseOptionalLLVMKeyword()
|
static |
Parse an enum from the keyword, or default to the provided default value.
The return type is the enum type by default, unless overriden with the second template argument.
Definition at line 1714 of file LLVMDialect.cpp.
References parseOptionalKeywordAlternative(), and mlir::tblgen::stringify().
◆ parseShuffleType()
|
static |
Build the result type of a shuffle vector operation.
Definition at line 1948 of file LLVMDialect.cpp.
References mlir::AsmParser::emitError(), mlir::AsmParser::getCurrentLocation(), mlir::LLVM::getVectorElementType(), mlir::LLVM::getVectorType(), mlir::LLVM::isCompatibleVectorType(), mlir::LLVM::isScalableVectorType(), and mlir::success().
◆ parseSwitchOpCases()
|
static |
<cases> ::= integer : bb-id (( ssa-use-and-type-list ))? ( , integer : bb-id (( ssa-use-and-type-list ))? )?
Definition at line 315 of file LLVMDialect.cpp.
References mlir::failure(), mlir::DenseIntElementsAttr::get(), mlir::Type::getIntOrFloatBitWidth(), mlir::OptionalParseResult::has_value(), mlir::AsmParser::None, mlir::AsmParser::parseColon(), mlir::AsmParser::parseColonTypeList(), mlir::OpAsmParser::parseOperandList(), mlir::AsmParser::parseOptionalComma(), mlir::AsmParser::parseOptionalInteger(), mlir::AsmParser::parseOptionalLParen(), mlir::AsmParser::parseRParen(), mlir::OpAsmParser::parseSuccessor(), mlir::success(), mlir::OptionalParseResult::value(), and value.
◆ printGEPIndices()
|
static |
Definition at line 543 of file LLVMDialect.cpp.
References mlir::OpAsmPrinter::printOperand().
◆ printInsertExtractValueElementType()
|
static |
Nothing to print for an inferred type.
Definition at line 1489 of file LLVMDialect.cpp.
◆ printLLVMOpAttrs()
|
static |
Definition at line 68 of file LLVMDialect.cpp.
References mlir::OpAsmPrinter::printOptionalAttrDict(), and processFMFAttr().
◆ printShuffleType()
|
static |
Nothing to do when the result type is inferred.
Definition at line 1959 of file LLVMDialect.cpp.
◆ printSwitchOpCases()
|
static |
Definition at line 354 of file LLVMDialect.cpp.
References mlir::OpAsmPrinter::printNewline().
◆ processFMFAttr()
|
static |
Definition at line 50 of file LLVMDialect.cpp.
References mlir::NamedAttribute::getName(), and mlir::NamedAttribute::getValue().
Referenced by printLLVMOpAttrs().
◆ verifyFuncOpInterfaceStructAttr()
|
static |
Definition at line 2734 of file LLVMDialect.cpp.
References mlir::Operation::emitError().
◆ verifyMemoryOpMetadata()
|
static |
Definition at line 725 of file LLVMDialect.cpp.
References mlir::failed(), mlir::failure(), and mlir::success().
◆ verifyOpaquePtr()
|
static |
Checks that the elemental type is present in either the pointer type or the attribute, but not both.
Definition at line 254 of file LLVMDialect.cpp.
References mlir::Operation::emitOpError(), kElemTypeAttrName, and mlir::success().
◆ verifyOpMetadata()
|
static |
Definition at line 711 of file LLVMDialect.cpp.
References mlir::Operation::emitOpError(), mlir::success(), and verifySymbolAttribute().
◆ verifyStructIndices() [1/2]
|
static |
Driver function around recordStructIndices.
Note that we always check from the second GEP index since the first one is always dynamic.
Definition at line 637 of file LLVMDialect.cpp.
References verifyStructIndices().
◆ verifyStructIndices() [2/2]
|
static |
For the given structIndices and indices, check if they're complied with baseGEPType, especially check against LLVMStructTypes nested within.
Definition at line 604 of file LLVMDialect.cpp.
References mlir::LLVM::GEPIndicesAdaptor< DynamicRange >::size(), and mlir::success().
Referenced by verifyStructIndices().
◆ verifySymbolAttribute()
| LogicalResult verifySymbolAttribute | ( | Operation * | op, |
| StringRef | attributeName, | ||
| llvm::function_ref< LogicalResult(Operation *, SymbolRefAttr)> | verifySymbolType | ||
| ) |
Definition at line 675 of file LLVMDialect.cpp.
References mlir::Operation::emitOpError(), mlir::failed(), mlir::failure(), mlir::Operation::getAttr(), mlir::Operation::getParentOp(), mlir::SymbolTable::lookupNearestSymbolFrom(), and mlir::success().
Referenced by verifyOpMetadata().
◆ verifySymbolAttrUse()
|
static |
Verifies symbol's use in op to ensure the symbol is a valid and fully defined llvm.func.
Definition at line 75 of file LLVMDialect.cpp.
References mlir::Operation::emitOpError(), mlir::FlatSymbolRefAttr::getAttr(), mlir::FlatSymbolRefAttr::getValue(), mlir::SymbolTableCollection::lookupNearestSymbolFrom(), and mlir::success().
Variable Documentation
◆ kElemTypeAttrName
|
staticconstexpr |
Definition at line 46 of file LLVMDialect.cpp.
Referenced by verifyOpaquePtr().
◆ kNonTemporalAttrName
|
staticconstexpr |
Definition at line 45 of file LLVMDialect.cpp.
◆ kVolatileAttrName
|
staticconstexpr |
Definition at line 44 of file LLVMDialect.cpp.
