1cde4d5a6SJacques Pienaar //===- ModuleTranslation.cpp - MLIR to LLVM conversion --------------------===// 25d7231d8SStephan Herhut // 330857107SMehdi Amini // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 456222a06SMehdi Amini // See https://llvm.org/LICENSE.txt for license information. 556222a06SMehdi Amini // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 65d7231d8SStephan Herhut // 756222a06SMehdi Amini //===----------------------------------------------------------------------===// 85d7231d8SStephan Herhut // 95d7231d8SStephan Herhut // This file implements the translation between an MLIR LLVM dialect module and 105d7231d8SStephan Herhut // the corresponding LLVMIR module. It only handles core LLVM IR operations. 115d7231d8SStephan Herhut // 125d7231d8SStephan Herhut //===----------------------------------------------------------------------===// 135d7231d8SStephan Herhut 145d7231d8SStephan Herhut #include "mlir/Target/LLVMIR/ModuleTranslation.h" 155d7231d8SStephan Herhut 16c33d6970SRiver Riddle #include "DebugTranslation.h" 17ea998709SAlex Zinenko #include "mlir/Dialect/DLTI/DLTI.h" 18ba0fa925SRiver Riddle #include "mlir/Dialect/LLVMIR/LLVMDialect.h" 19ce8f10d6SAlex Zinenko #include "mlir/Dialect/LLVMIR/Transforms/LegalizeForExport.h" 2092a295ebSKiran Chandramohan #include "mlir/Dialect/OpenMP/OpenMPDialect.h" 215d7231d8SStephan Herhut #include "mlir/IR/Attributes.h" 2265fcddffSRiver Riddle #include "mlir/IR/BuiltinOps.h" 2309f7a55fSRiver Riddle #include "mlir/IR/BuiltinTypes.h" 24d4568ed7SGeorge Mitenkov #include "mlir/IR/RegionGraphTraits.h" 255d7231d8SStephan Herhut #include "mlir/Support/LLVM.h" 26b77bac05SAlex Zinenko #include "mlir/Target/LLVMIR/LLVMTranslationInterface.h" 27929189a4SWilliam S. Moses #include "mlir/Target/LLVMIR/TypeToLLVM.h" 28ebf190fcSRiver Riddle #include "llvm/ADT/TypeSwitch.h" 295d7231d8SStephan Herhut 30d4568ed7SGeorge Mitenkov #include "llvm/ADT/PostOrderIterator.h" 315d7231d8SStephan Herhut #include "llvm/ADT/SetVector.h" 3292a295ebSKiran Chandramohan #include "llvm/Frontend/OpenMP/OMPIRBuilder.h" 335d7231d8SStephan Herhut #include "llvm/IR/BasicBlock.h" 34d9067dcaSKiran Chandramohan #include "llvm/IR/CFG.h" 355d7231d8SStephan Herhut #include "llvm/IR/Constants.h" 365d7231d8SStephan Herhut #include "llvm/IR/DerivedTypes.h" 375d7231d8SStephan Herhut #include "llvm/IR/IRBuilder.h" 38047400edSNicolas Vasilache #include "llvm/IR/InlineAsm.h" 39875eb523SNavdeep Kumar #include "llvm/IR/IntrinsicsNVPTX.h" 405d7231d8SStephan Herhut #include "llvm/IR/LLVMContext.h" 4199d03f03SGeorge Mitenkov #include "llvm/IR/MDBuilder.h" 425d7231d8SStephan Herhut #include "llvm/IR/Module.h" 43ce8f10d6SAlex Zinenko #include "llvm/IR/Verifier.h" 44d9067dcaSKiran Chandramohan #include "llvm/Transforms/Utils/BasicBlockUtils.h" 455d7231d8SStephan Herhut #include "llvm/Transforms/Utils/Cloning.h" 4657b9b296SUday Bondhugula #include "llvm/Transforms/Utils/ModuleUtils.h" 475d7231d8SStephan Herhut 482666b973SRiver Riddle using namespace mlir; 492666b973SRiver Riddle using namespace mlir::LLVM; 50c33d6970SRiver Riddle using namespace mlir::LLVM::detail; 515d7231d8SStephan Herhut 52eb67bd78SAlex Zinenko #include "mlir/Dialect/LLVMIR/LLVMConversionEnumsToLLVM.inc" 53eb67bd78SAlex Zinenko 54ea998709SAlex Zinenko /// Translates the given data layout spec attribute to the LLVM IR data layout. 55ea998709SAlex Zinenko /// Only integer, float and endianness entries are currently supported. 56ea998709SAlex Zinenko FailureOr<llvm::DataLayout> 57ea998709SAlex Zinenko translateDataLayout(DataLayoutSpecInterface attribute, 58ea998709SAlex Zinenko const DataLayout &dataLayout, 59ea998709SAlex Zinenko Optional<Location> loc = llvm::None) { 60ea998709SAlex Zinenko if (!loc) 61ea998709SAlex Zinenko loc = UnknownLoc::get(attribute.getContext()); 62ea998709SAlex Zinenko 63ea998709SAlex Zinenko // Translate the endianness attribute. 64ea998709SAlex Zinenko std::string llvmDataLayout; 65ea998709SAlex Zinenko llvm::raw_string_ostream layoutStream(llvmDataLayout); 66ea998709SAlex Zinenko for (DataLayoutEntryInterface entry : attribute.getEntries()) { 67ea998709SAlex Zinenko auto key = entry.getKey().dyn_cast<StringAttr>(); 68ea998709SAlex Zinenko if (!key) 69ea998709SAlex Zinenko continue; 70ea998709SAlex Zinenko if (key.getValue() == DLTIDialect::kDataLayoutEndiannessKey) { 71ea998709SAlex Zinenko auto value = entry.getValue().cast<StringAttr>(); 72ea998709SAlex Zinenko bool isLittleEndian = 73ea998709SAlex Zinenko value.getValue() == DLTIDialect::kDataLayoutEndiannessLittle; 74ea998709SAlex Zinenko layoutStream << (isLittleEndian ? "e" : "E"); 75ea998709SAlex Zinenko layoutStream.flush(); 76ea998709SAlex Zinenko continue; 77ea998709SAlex Zinenko } 78ea998709SAlex Zinenko emitError(*loc) << "unsupported data layout key " << key; 79ea998709SAlex Zinenko return failure(); 80ea998709SAlex Zinenko } 81ea998709SAlex Zinenko 82ea998709SAlex Zinenko // Go through the list of entries to check which types are explicitly 83ea998709SAlex Zinenko // specified in entries. Don't use the entries directly though but query the 84ea998709SAlex Zinenko // data from the layout. 85ea998709SAlex Zinenko for (DataLayoutEntryInterface entry : attribute.getEntries()) { 86ea998709SAlex Zinenko auto type = entry.getKey().dyn_cast<Type>(); 87ea998709SAlex Zinenko if (!type) 88ea998709SAlex Zinenko continue; 89eb27da7dSAlex Zinenko // Data layout for the index type is irrelevant at this point. 90eb27da7dSAlex Zinenko if (type.isa<IndexType>()) 91eb27da7dSAlex Zinenko continue; 92ea998709SAlex Zinenko FailureOr<std::string> prefix = 93ea998709SAlex Zinenko llvm::TypeSwitch<Type, FailureOr<std::string>>(type) 94ea998709SAlex Zinenko .Case<IntegerType>( 95ea998709SAlex Zinenko [loc](IntegerType integerType) -> FailureOr<std::string> { 96ea998709SAlex Zinenko if (integerType.getSignedness() == IntegerType::Signless) 97ea998709SAlex Zinenko return std::string("i"); 98ea998709SAlex Zinenko emitError(*loc) 99ea998709SAlex Zinenko << "unsupported data layout for non-signless integer " 100ea998709SAlex Zinenko << integerType; 101ea998709SAlex Zinenko return failure(); 102ea998709SAlex Zinenko }) 103ea998709SAlex Zinenko .Case<Float16Type, Float32Type, Float64Type, Float80Type, 104ea998709SAlex Zinenko Float128Type>([](Type) { return std::string("f"); }) 105ea998709SAlex Zinenko .Default([loc](Type type) -> FailureOr<std::string> { 106ea998709SAlex Zinenko emitError(*loc) << "unsupported type in data layout: " << type; 107ea998709SAlex Zinenko return failure(); 108ea998709SAlex Zinenko }); 109ea998709SAlex Zinenko if (failed(prefix)) 110ea998709SAlex Zinenko return failure(); 111ea998709SAlex Zinenko 112ea998709SAlex Zinenko unsigned size = dataLayout.getTypeSizeInBits(type); 113ea998709SAlex Zinenko unsigned abi = dataLayout.getTypeABIAlignment(type) * 8u; 114ea998709SAlex Zinenko unsigned preferred = dataLayout.getTypePreferredAlignment(type) * 8u; 115ea998709SAlex Zinenko layoutStream << "-" << *prefix << size << ":" << abi; 116ea998709SAlex Zinenko if (abi != preferred) 117ea998709SAlex Zinenko layoutStream << ":" << preferred; 118ea998709SAlex Zinenko } 119ea998709SAlex Zinenko layoutStream.flush(); 120ea998709SAlex Zinenko StringRef layoutSpec(llvmDataLayout); 121ea998709SAlex Zinenko if (layoutSpec.startswith("-")) 122ea998709SAlex Zinenko layoutSpec = layoutSpec.drop_front(); 123ea998709SAlex Zinenko 124ea998709SAlex Zinenko return llvm::DataLayout(layoutSpec); 125ea998709SAlex Zinenko } 126ea998709SAlex Zinenko 127a922e231SAlex Zinenko /// Builds a constant of a sequential LLVM type `type`, potentially containing 128a922e231SAlex Zinenko /// other sequential types recursively, from the individual constant values 129a922e231SAlex Zinenko /// provided in `constants`. `shape` contains the number of elements in nested 130a922e231SAlex Zinenko /// sequential types. Reports errors at `loc` and returns nullptr on error. 131a4a42160SAlex Zinenko static llvm::Constant * 132a4a42160SAlex Zinenko buildSequentialConstant(ArrayRef<llvm::Constant *> &constants, 133a4a42160SAlex Zinenko ArrayRef<int64_t> shape, llvm::Type *type, 134a4a42160SAlex Zinenko Location loc) { 135a4a42160SAlex Zinenko if (shape.empty()) { 136a4a42160SAlex Zinenko llvm::Constant *result = constants.front(); 137a4a42160SAlex Zinenko constants = constants.drop_front(); 138a4a42160SAlex Zinenko return result; 139a4a42160SAlex Zinenko } 140a4a42160SAlex Zinenko 14168b03aeeSEli Friedman llvm::Type *elementType; 14268b03aeeSEli Friedman if (auto *arrayTy = dyn_cast<llvm::ArrayType>(type)) { 14368b03aeeSEli Friedman elementType = arrayTy->getElementType(); 14468b03aeeSEli Friedman } else if (auto *vectorTy = dyn_cast<llvm::VectorType>(type)) { 14568b03aeeSEli Friedman elementType = vectorTy->getElementType(); 14668b03aeeSEli Friedman } else { 147a4a42160SAlex Zinenko emitError(loc) << "expected sequential LLVM types wrapping a scalar"; 148a4a42160SAlex Zinenko return nullptr; 149a4a42160SAlex Zinenko } 150a4a42160SAlex Zinenko 151a4a42160SAlex Zinenko SmallVector<llvm::Constant *, 8> nested; 152a4a42160SAlex Zinenko nested.reserve(shape.front()); 153a4a42160SAlex Zinenko for (int64_t i = 0; i < shape.front(); ++i) { 154a4a42160SAlex Zinenko nested.push_back(buildSequentialConstant(constants, shape.drop_front(), 155a4a42160SAlex Zinenko elementType, loc)); 156a4a42160SAlex Zinenko if (!nested.back()) 157a4a42160SAlex Zinenko return nullptr; 158a4a42160SAlex Zinenko } 159a4a42160SAlex Zinenko 160a4a42160SAlex Zinenko if (shape.size() == 1 && type->isVectorTy()) 161a4a42160SAlex Zinenko return llvm::ConstantVector::get(nested); 162a4a42160SAlex Zinenko return llvm::ConstantArray::get( 163a4a42160SAlex Zinenko llvm::ArrayType::get(elementType, shape.front()), nested); 164a4a42160SAlex Zinenko } 165a4a42160SAlex Zinenko 166fc817b09SKazuaki Ishizaki /// Returns the first non-sequential type nested in sequential types. 167a4a42160SAlex Zinenko static llvm::Type *getInnermostElementType(llvm::Type *type) { 16868b03aeeSEli Friedman do { 16968b03aeeSEli Friedman if (auto *arrayTy = dyn_cast<llvm::ArrayType>(type)) { 17068b03aeeSEli Friedman type = arrayTy->getElementType(); 17168b03aeeSEli Friedman } else if (auto *vectorTy = dyn_cast<llvm::VectorType>(type)) { 17268b03aeeSEli Friedman type = vectorTy->getElementType(); 17368b03aeeSEli Friedman } else { 174a4a42160SAlex Zinenko return type; 175a4a42160SAlex Zinenko } 1760881a4f1SAlex Zinenko } while (true); 17768b03aeeSEli Friedman } 178a4a42160SAlex Zinenko 179f9be7a7aSAlex Zinenko /// Convert a dense elements attribute to an LLVM IR constant using its raw data 180f9be7a7aSAlex Zinenko /// storage if possible. This supports elements attributes of tensor or vector 181f9be7a7aSAlex Zinenko /// type and avoids constructing separate objects for individual values of the 182f9be7a7aSAlex Zinenko /// innermost dimension. Constants for other dimensions are still constructed 183f9be7a7aSAlex Zinenko /// recursively. Returns null if constructing from raw data is not supported for 184f9be7a7aSAlex Zinenko /// this type, e.g., element type is not a power-of-two-sized primitive. Reports 185f9be7a7aSAlex Zinenko /// other errors at `loc`. 186f9be7a7aSAlex Zinenko static llvm::Constant * 187f9be7a7aSAlex Zinenko convertDenseElementsAttr(Location loc, DenseElementsAttr denseElementsAttr, 188f9be7a7aSAlex Zinenko llvm::Type *llvmType, 189f9be7a7aSAlex Zinenko const ModuleTranslation &moduleTranslation) { 190f9be7a7aSAlex Zinenko if (!denseElementsAttr) 191f9be7a7aSAlex Zinenko return nullptr; 192f9be7a7aSAlex Zinenko 193f9be7a7aSAlex Zinenko llvm::Type *innermostLLVMType = getInnermostElementType(llvmType); 194f9be7a7aSAlex Zinenko if (!llvm::ConstantDataSequential::isElementTypeCompatible(innermostLLVMType)) 195f9be7a7aSAlex Zinenko return nullptr; 196f9be7a7aSAlex Zinenko 197898e8096SBenjamin Kramer ShapedType type = denseElementsAttr.getType(); 198898e8096SBenjamin Kramer if (type.getNumElements() == 0) 199898e8096SBenjamin Kramer return nullptr; 200898e8096SBenjamin Kramer 201f9be7a7aSAlex Zinenko // Compute the shape of all dimensions but the innermost. Note that the 202f9be7a7aSAlex Zinenko // innermost dimension may be that of the vector element type. 203f9be7a7aSAlex Zinenko bool hasVectorElementType = type.getElementType().isa<VectorType>(); 204f9be7a7aSAlex Zinenko unsigned numAggregates = 205f9be7a7aSAlex Zinenko denseElementsAttr.getNumElements() / 206f9be7a7aSAlex Zinenko (hasVectorElementType ? 1 207f9be7a7aSAlex Zinenko : denseElementsAttr.getType().getShape().back()); 208f9be7a7aSAlex Zinenko ArrayRef<int64_t> outerShape = type.getShape(); 209f9be7a7aSAlex Zinenko if (!hasVectorElementType) 210f9be7a7aSAlex Zinenko outerShape = outerShape.drop_back(); 211f9be7a7aSAlex Zinenko 212f9be7a7aSAlex Zinenko // Handle the case of vector splat, LLVM has special support for it. 213f9be7a7aSAlex Zinenko if (denseElementsAttr.isSplat() && 214f9be7a7aSAlex Zinenko (type.isa<VectorType>() || hasVectorElementType)) { 215f9be7a7aSAlex Zinenko llvm::Constant *splatValue = LLVM::detail::getLLVMConstant( 216937e40a8SRiver Riddle innermostLLVMType, denseElementsAttr.getSplatValue<Attribute>(), loc, 217f9be7a7aSAlex Zinenko moduleTranslation, /*isTopLevel=*/false); 218f9be7a7aSAlex Zinenko llvm::Constant *splatVector = 219f9be7a7aSAlex Zinenko llvm::ConstantDataVector::getSplat(0, splatValue); 220f9be7a7aSAlex Zinenko SmallVector<llvm::Constant *> constants(numAggregates, splatVector); 221f9be7a7aSAlex Zinenko ArrayRef<llvm::Constant *> constantsRef = constants; 222f9be7a7aSAlex Zinenko return buildSequentialConstant(constantsRef, outerShape, llvmType, loc); 223f9be7a7aSAlex Zinenko } 224f9be7a7aSAlex Zinenko if (denseElementsAttr.isSplat()) 225f9be7a7aSAlex Zinenko return nullptr; 226f9be7a7aSAlex Zinenko 227f9be7a7aSAlex Zinenko // In case of non-splat, create a constructor for the innermost constant from 228f9be7a7aSAlex Zinenko // a piece of raw data. 229f9be7a7aSAlex Zinenko std::function<llvm::Constant *(StringRef)> buildCstData; 230f9be7a7aSAlex Zinenko if (type.isa<TensorType>()) { 231f9be7a7aSAlex Zinenko auto vectorElementType = type.getElementType().dyn_cast<VectorType>(); 232f9be7a7aSAlex Zinenko if (vectorElementType && vectorElementType.getRank() == 1) { 233f9be7a7aSAlex Zinenko buildCstData = [&](StringRef data) { 234f9be7a7aSAlex Zinenko return llvm::ConstantDataVector::getRaw( 235f9be7a7aSAlex Zinenko data, vectorElementType.getShape().back(), innermostLLVMType); 236f9be7a7aSAlex Zinenko }; 237f9be7a7aSAlex Zinenko } else if (!vectorElementType) { 238f9be7a7aSAlex Zinenko buildCstData = [&](StringRef data) { 239f9be7a7aSAlex Zinenko return llvm::ConstantDataArray::getRaw(data, type.getShape().back(), 240f9be7a7aSAlex Zinenko innermostLLVMType); 241f9be7a7aSAlex Zinenko }; 242f9be7a7aSAlex Zinenko } 243f9be7a7aSAlex Zinenko } else if (type.isa<VectorType>()) { 244f9be7a7aSAlex Zinenko buildCstData = [&](StringRef data) { 245f9be7a7aSAlex Zinenko return llvm::ConstantDataVector::getRaw(data, type.getShape().back(), 246f9be7a7aSAlex Zinenko innermostLLVMType); 247f9be7a7aSAlex Zinenko }; 248f9be7a7aSAlex Zinenko } 249f9be7a7aSAlex Zinenko if (!buildCstData) 250f9be7a7aSAlex Zinenko return nullptr; 251f9be7a7aSAlex Zinenko 252f9be7a7aSAlex Zinenko // Create innermost constants and defer to the default constant creation 253f9be7a7aSAlex Zinenko // mechanism for other dimensions. 254f9be7a7aSAlex Zinenko SmallVector<llvm::Constant *> constants; 255f9be7a7aSAlex Zinenko unsigned aggregateSize = denseElementsAttr.getType().getShape().back() * 256f9be7a7aSAlex Zinenko (innermostLLVMType->getScalarSizeInBits() / 8); 257f9be7a7aSAlex Zinenko constants.reserve(numAggregates); 258f9be7a7aSAlex Zinenko for (unsigned i = 0; i < numAggregates; ++i) { 259f9be7a7aSAlex Zinenko StringRef data(denseElementsAttr.getRawData().data() + i * aggregateSize, 260f9be7a7aSAlex Zinenko aggregateSize); 261f9be7a7aSAlex Zinenko constants.push_back(buildCstData(data)); 262f9be7a7aSAlex Zinenko } 263f9be7a7aSAlex Zinenko 264f9be7a7aSAlex Zinenko ArrayRef<llvm::Constant *> constantsRef = constants; 265f9be7a7aSAlex Zinenko return buildSequentialConstant(constantsRef, outerShape, llvmType, loc); 266f9be7a7aSAlex Zinenko } 267f9be7a7aSAlex Zinenko 2682666b973SRiver Riddle /// Create an LLVM IR constant of `llvmType` from the MLIR attribute `attr`. 2692666b973SRiver Riddle /// This currently supports integer, floating point, splat and dense element 2705ef21506SAdrian Kuegel /// attributes and combinations thereof. Also, an array attribute with two 2715ef21506SAdrian Kuegel /// elements is supported to represent a complex constant. In case of error, 2725ef21506SAdrian Kuegel /// report it to `loc` and return nullptr. 273176379e0SAlex Zinenko llvm::Constant *mlir::LLVM::detail::getLLVMConstant( 274176379e0SAlex Zinenko llvm::Type *llvmType, Attribute attr, Location loc, 2755ef21506SAdrian Kuegel const ModuleTranslation &moduleTranslation, bool isTopLevel) { 27633a3a91bSChristian Sigg if (!attr) 27733a3a91bSChristian Sigg return llvm::UndefValue::get(llvmType); 2785ef21506SAdrian Kuegel if (auto *structType = dyn_cast<::llvm::StructType>(llvmType)) { 2795ef21506SAdrian Kuegel if (!isTopLevel) { 2805ef21506SAdrian Kuegel emitError(loc, "nested struct types are not supported in constants"); 281a4a42160SAlex Zinenko return nullptr; 282a4a42160SAlex Zinenko } 2835ef21506SAdrian Kuegel auto arrayAttr = attr.cast<ArrayAttr>(); 2845ef21506SAdrian Kuegel llvm::Type *elementType = structType->getElementType(0); 2855ef21506SAdrian Kuegel llvm::Constant *real = getLLVMConstant(elementType, arrayAttr[0], loc, 2865ef21506SAdrian Kuegel moduleTranslation, false); 2875ef21506SAdrian Kuegel if (!real) 2885ef21506SAdrian Kuegel return nullptr; 2895ef21506SAdrian Kuegel llvm::Constant *imag = getLLVMConstant(elementType, arrayAttr[1], loc, 2905ef21506SAdrian Kuegel moduleTranslation, false); 2915ef21506SAdrian Kuegel if (!imag) 2925ef21506SAdrian Kuegel return nullptr; 2935ef21506SAdrian Kuegel return llvm::ConstantStruct::get(structType, {real, imag}); 2945ef21506SAdrian Kuegel } 295ac9d742bSStephan Herhut // For integer types, we allow a mismatch in sizes as the index type in 296ac9d742bSStephan Herhut // MLIR might have a different size than the index type in the LLVM module. 2975d7231d8SStephan Herhut if (auto intAttr = attr.dyn_cast<IntegerAttr>()) 298ac9d742bSStephan Herhut return llvm::ConstantInt::get( 299ac9d742bSStephan Herhut llvmType, 300ac9d742bSStephan Herhut intAttr.getValue().sextOrTrunc(llvmType->getIntegerBitWidth())); 3015ef21506SAdrian Kuegel if (auto floatAttr = attr.dyn_cast<FloatAttr>()) { 3025ef21506SAdrian Kuegel if (llvmType != 3035ef21506SAdrian Kuegel llvm::Type::getFloatingPointTy(llvmType->getContext(), 3045ef21506SAdrian Kuegel floatAttr.getValue().getSemantics())) { 3055ef21506SAdrian Kuegel emitError(loc, "FloatAttr does not match expected type of the constant"); 3065ef21506SAdrian Kuegel return nullptr; 3075ef21506SAdrian Kuegel } 3085d7231d8SStephan Herhut return llvm::ConstantFP::get(llvmType, floatAttr.getValue()); 3095ef21506SAdrian Kuegel } 3109b9c647cSRiver Riddle if (auto funcAttr = attr.dyn_cast<FlatSymbolRefAttr>()) 311176379e0SAlex Zinenko return llvm::ConstantExpr::getBitCast( 312176379e0SAlex Zinenko moduleTranslation.lookupFunction(funcAttr.getValue()), llvmType); 3135d7231d8SStephan Herhut if (auto splatAttr = attr.dyn_cast<SplatElementsAttr>()) { 31468b03aeeSEli Friedman llvm::Type *elementType; 31568b03aeeSEli Friedman uint64_t numElements; 3167c564586SJavier Setoain bool isScalable = false; 31768b03aeeSEli Friedman if (auto *arrayTy = dyn_cast<llvm::ArrayType>(llvmType)) { 31868b03aeeSEli Friedman elementType = arrayTy->getElementType(); 31968b03aeeSEli Friedman numElements = arrayTy->getNumElements(); 32002b6fb21SMehdi Amini } else if (auto *fVectorTy = dyn_cast<llvm::FixedVectorType>(llvmType)) { 321a4830d14SJavier Setoain elementType = fVectorTy->getElementType(); 322a4830d14SJavier Setoain numElements = fVectorTy->getNumElements(); 32302b6fb21SMehdi Amini } else if (auto *sVectorTy = dyn_cast<llvm::ScalableVectorType>(llvmType)) { 324a4830d14SJavier Setoain elementType = sVectorTy->getElementType(); 325a4830d14SJavier Setoain numElements = sVectorTy->getMinNumElements(); 3267c564586SJavier Setoain isScalable = true; 32768b03aeeSEli Friedman } else { 328a4830d14SJavier Setoain llvm_unreachable("unrecognized constant vector type"); 32968b03aeeSEli Friedman } 330d6ea8ff0SAlex Zinenko // Splat value is a scalar. Extract it only if the element type is not 331d6ea8ff0SAlex Zinenko // another sequence type. The recursion terminates because each step removes 332d6ea8ff0SAlex Zinenko // one outer sequential type. 33368b03aeeSEli Friedman bool elementTypeSequential = 334d891d738SRahul Joshi isa<llvm::ArrayType, llvm::VectorType>(elementType); 335d6ea8ff0SAlex Zinenko llvm::Constant *child = getLLVMConstant( 336d6ea8ff0SAlex Zinenko elementType, 337937e40a8SRiver Riddle elementTypeSequential ? splatAttr 338937e40a8SRiver Riddle : splatAttr.getSplatValue<Attribute>(), 339937e40a8SRiver Riddle loc, moduleTranslation, false); 340a4a42160SAlex Zinenko if (!child) 341a4a42160SAlex Zinenko return nullptr; 3422f13df13SMLIR Team if (llvmType->isVectorTy()) 343396a42d9SRiver Riddle return llvm::ConstantVector::getSplat( 3447c564586SJavier Setoain llvm::ElementCount::get(numElements, /*Scalable=*/isScalable), child); 3452f13df13SMLIR Team if (llvmType->isArrayTy()) { 346ac9d742bSStephan Herhut auto *arrayType = llvm::ArrayType::get(elementType, numElements); 3472f13df13SMLIR Team SmallVector<llvm::Constant *, 8> constants(numElements, child); 3482f13df13SMLIR Team return llvm::ConstantArray::get(arrayType, constants); 3492f13df13SMLIR Team } 3505d7231d8SStephan Herhut } 351a4a42160SAlex Zinenko 352f9be7a7aSAlex Zinenko // Try using raw elements data if possible. 353f9be7a7aSAlex Zinenko if (llvm::Constant *result = 354f9be7a7aSAlex Zinenko convertDenseElementsAttr(loc, attr.dyn_cast<DenseElementsAttr>(), 355f9be7a7aSAlex Zinenko llvmType, moduleTranslation)) { 356f9be7a7aSAlex Zinenko return result; 357f9be7a7aSAlex Zinenko } 358f9be7a7aSAlex Zinenko 359f9be7a7aSAlex Zinenko // Fall back to element-by-element construction otherwise. 360d906f84bSRiver Riddle if (auto elementsAttr = attr.dyn_cast<ElementsAttr>()) { 361a4a42160SAlex Zinenko assert(elementsAttr.getType().hasStaticShape()); 362a4a42160SAlex Zinenko assert(!elementsAttr.getType().getShape().empty() && 363a4a42160SAlex Zinenko "unexpected empty elements attribute shape"); 364a4a42160SAlex Zinenko 3655d7231d8SStephan Herhut SmallVector<llvm::Constant *, 8> constants; 366a4a42160SAlex Zinenko constants.reserve(elementsAttr.getNumElements()); 367a4a42160SAlex Zinenko llvm::Type *innermostType = getInnermostElementType(llvmType); 368d906f84bSRiver Riddle for (auto n : elementsAttr.getValues<Attribute>()) { 369176379e0SAlex Zinenko constants.push_back( 3705ef21506SAdrian Kuegel getLLVMConstant(innermostType, n, loc, moduleTranslation, false)); 3715d7231d8SStephan Herhut if (!constants.back()) 3725d7231d8SStephan Herhut return nullptr; 3735d7231d8SStephan Herhut } 374a4a42160SAlex Zinenko ArrayRef<llvm::Constant *> constantsRef = constants; 375a4a42160SAlex Zinenko llvm::Constant *result = buildSequentialConstant( 376a4a42160SAlex Zinenko constantsRef, elementsAttr.getType().getShape(), llvmType, loc); 377a4a42160SAlex Zinenko assert(constantsRef.empty() && "did not consume all elemental constants"); 378a4a42160SAlex Zinenko return result; 3792f13df13SMLIR Team } 380a4a42160SAlex Zinenko 381cb348dffSStephan Herhut if (auto stringAttr = attr.dyn_cast<StringAttr>()) { 382cb348dffSStephan Herhut return llvm::ConstantDataArray::get( 383176379e0SAlex Zinenko moduleTranslation.getLLVMContext(), 384176379e0SAlex Zinenko ArrayRef<char>{stringAttr.getValue().data(), 385cb348dffSStephan Herhut stringAttr.getValue().size()}); 386cb348dffSStephan Herhut } 387a4c3a645SRiver Riddle emitError(loc, "unsupported constant value"); 3885d7231d8SStephan Herhut return nullptr; 3895d7231d8SStephan Herhut } 3905d7231d8SStephan Herhut 391c33d6970SRiver Riddle ModuleTranslation::ModuleTranslation(Operation *module, 392c33d6970SRiver Riddle std::unique_ptr<llvm::Module> llvmModule) 393c33d6970SRiver Riddle : mlirModule(module), llvmModule(std::move(llvmModule)), 394c33d6970SRiver Riddle debugTranslation( 39592a295ebSKiran Chandramohan std::make_unique<DebugTranslation>(module, *this->llvmModule)), 396b77bac05SAlex Zinenko typeTranslator(this->llvmModule->getContext()), 397b77bac05SAlex Zinenko iface(module->getContext()) { 398c33d6970SRiver Riddle assert(satisfiesLLVMModule(mlirModule) && 399c33d6970SRiver Riddle "mlirModule should honor LLVM's module semantics."); 400c33d6970SRiver Riddle } 401d9067dcaSKiran Chandramohan ModuleTranslation::~ModuleTranslation() { 402d9067dcaSKiran Chandramohan if (ompBuilder) 403d9067dcaSKiran Chandramohan ompBuilder->finalize(); 404d9067dcaSKiran Chandramohan } 405d9067dcaSKiran Chandramohan 4068647e4c3SAlex Zinenko void ModuleTranslation::forgetMapping(Region ®ion) { 4078647e4c3SAlex Zinenko SmallVector<Region *> toProcess; 4088647e4c3SAlex Zinenko toProcess.push_back(®ion); 4098647e4c3SAlex Zinenko while (!toProcess.empty()) { 4108647e4c3SAlex Zinenko Region *current = toProcess.pop_back_val(); 4118647e4c3SAlex Zinenko for (Block &block : *current) { 4128647e4c3SAlex Zinenko blockMapping.erase(&block); 4138647e4c3SAlex Zinenko for (Value arg : block.getArguments()) 4148647e4c3SAlex Zinenko valueMapping.erase(arg); 4158647e4c3SAlex Zinenko for (Operation &op : block) { 4168647e4c3SAlex Zinenko for (Value value : op.getResults()) 4178647e4c3SAlex Zinenko valueMapping.erase(value); 4188647e4c3SAlex Zinenko if (op.hasSuccessors()) 4198647e4c3SAlex Zinenko branchMapping.erase(&op); 4208647e4c3SAlex Zinenko if (isa<LLVM::GlobalOp>(op)) 4218647e4c3SAlex Zinenko globalsMapping.erase(&op); 4228647e4c3SAlex Zinenko accessGroupMetadataMapping.erase(&op); 4238647e4c3SAlex Zinenko llvm::append_range( 4248647e4c3SAlex Zinenko toProcess, 4258647e4c3SAlex Zinenko llvm::map_range(op.getRegions(), [](Region &r) { return &r; })); 4268647e4c3SAlex Zinenko } 4278647e4c3SAlex Zinenko } 4288647e4c3SAlex Zinenko } 4298647e4c3SAlex Zinenko } 4308647e4c3SAlex Zinenko 431d9067dcaSKiran Chandramohan /// Get the SSA value passed to the current block from the terminator operation 432d9067dcaSKiran Chandramohan /// of its predecessor. 433d9067dcaSKiran Chandramohan static Value getPHISourceValue(Block *current, Block *pred, 434d9067dcaSKiran Chandramohan unsigned numArguments, unsigned index) { 435d9067dcaSKiran Chandramohan Operation &terminator = *pred->getTerminator(); 436d9067dcaSKiran Chandramohan if (isa<LLVM::BrOp>(terminator)) 437d9067dcaSKiran Chandramohan return terminator.getOperand(index); 438d9067dcaSKiran Chandramohan 439bea16e72SAlex Zinenko #ifndef NDEBUG 440bea16e72SAlex Zinenko llvm::SmallPtrSet<Block *, 4> seenSuccessors; 441bea16e72SAlex Zinenko for (unsigned i = 0, e = terminator.getNumSuccessors(); i < e; ++i) { 442bea16e72SAlex Zinenko Block *successor = terminator.getSuccessor(i); 443bea16e72SAlex Zinenko auto branch = cast<BranchOpInterface>(terminator); 4440c789db5SMarkus Böck SuccessorOperands successorOperands = branch.getSuccessorOperands(i); 445bea16e72SAlex Zinenko assert( 4460c789db5SMarkus Böck (!seenSuccessors.contains(successor) || successorOperands.empty()) && 44714f24155SBrian Gesiak "successors with arguments in LLVM branches must be different blocks"); 448bea16e72SAlex Zinenko seenSuccessors.insert(successor); 449bea16e72SAlex Zinenko } 450bea16e72SAlex Zinenko #endif 451d9067dcaSKiran Chandramohan 45214f24155SBrian Gesiak // For instructions that branch based on a condition value, we need to take 45314f24155SBrian Gesiak // the operands for the branch that was taken. 45414f24155SBrian Gesiak if (auto condBranchOp = dyn_cast<LLVM::CondBrOp>(terminator)) { 45514f24155SBrian Gesiak // For conditional branches, we take the operands from either the "true" or 45614f24155SBrian Gesiak // the "false" branch. 457d9067dcaSKiran Chandramohan return condBranchOp.getSuccessor(0) == current 458cfb72fd3SJacques Pienaar ? condBranchOp.getTrueDestOperands()[index] 459cfb72fd3SJacques Pienaar : condBranchOp.getFalseDestOperands()[index]; 4600881a4f1SAlex Zinenko } 4610881a4f1SAlex Zinenko 4620881a4f1SAlex Zinenko if (auto switchOp = dyn_cast<LLVM::SwitchOp>(terminator)) { 46314f24155SBrian Gesiak // For switches, we take the operands from either the default case, or from 46414f24155SBrian Gesiak // the case branch that was taken. 465dde96363SJacques Pienaar if (switchOp.getDefaultDestination() == current) 466dde96363SJacques Pienaar return switchOp.getDefaultOperands()[index]; 467e4853be2SMehdi Amini for (const auto &i : llvm::enumerate(switchOp.getCaseDestinations())) 46814f24155SBrian Gesiak if (i.value() == current) 46914f24155SBrian Gesiak return switchOp.getCaseOperands(i.index())[index]; 47014f24155SBrian Gesiak } 47114f24155SBrian Gesiak 47256097205SMarkus Böck if (auto invokeOp = dyn_cast<LLVM::InvokeOp>(terminator)) { 47356097205SMarkus Böck return invokeOp.getNormalDest() == current 47456097205SMarkus Böck ? invokeOp.getNormalDestOperands()[index] 47556097205SMarkus Böck : invokeOp.getUnwindDestOperands()[index]; 47656097205SMarkus Böck } 47756097205SMarkus Böck 47856097205SMarkus Böck llvm_unreachable( 47956097205SMarkus Böck "only branch, switch or invoke operations can be terminators " 48056097205SMarkus Böck "of a block that has successors"); 481d9067dcaSKiran Chandramohan } 482d9067dcaSKiran Chandramohan 483d9067dcaSKiran Chandramohan /// Connect the PHI nodes to the results of preceding blocks. 48466900b3eSAlex Zinenko void mlir::LLVM::detail::connectPHINodes(Region ®ion, 48566900b3eSAlex Zinenko const ModuleTranslation &state) { 486d9067dcaSKiran Chandramohan // Skip the first block, it cannot be branched to and its arguments correspond 487d9067dcaSKiran Chandramohan // to the arguments of the LLVM function. 48866900b3eSAlex Zinenko for (auto it = std::next(region.begin()), eit = region.end(); it != eit; 48966900b3eSAlex Zinenko ++it) { 490d9067dcaSKiran Chandramohan Block *bb = &*it; 4910881a4f1SAlex Zinenko llvm::BasicBlock *llvmBB = state.lookupBlock(bb); 492d9067dcaSKiran Chandramohan auto phis = llvmBB->phis(); 493d9067dcaSKiran Chandramohan auto numArguments = bb->getNumArguments(); 494d9067dcaSKiran Chandramohan assert(numArguments == std::distance(phis.begin(), phis.end())); 495d9067dcaSKiran Chandramohan for (auto &numberedPhiNode : llvm::enumerate(phis)) { 496d9067dcaSKiran Chandramohan auto &phiNode = numberedPhiNode.value(); 497d9067dcaSKiran Chandramohan unsigned index = numberedPhiNode.index(); 498d9067dcaSKiran Chandramohan for (auto *pred : bb->getPredecessors()) { 499db884dafSAlex Zinenko // Find the LLVM IR block that contains the converted terminator 500db884dafSAlex Zinenko // instruction and use it in the PHI node. Note that this block is not 5010881a4f1SAlex Zinenko // necessarily the same as state.lookupBlock(pred), some operations 502db884dafSAlex Zinenko // (in particular, OpenMP operations using OpenMPIRBuilder) may have 503db884dafSAlex Zinenko // split the blocks. 504db884dafSAlex Zinenko llvm::Instruction *terminator = 5050881a4f1SAlex Zinenko state.lookupBranch(pred->getTerminator()); 506db884dafSAlex Zinenko assert(terminator && "missing the mapping for a terminator"); 5070881a4f1SAlex Zinenko phiNode.addIncoming( 5080881a4f1SAlex Zinenko state.lookupValue(getPHISourceValue(bb, pred, numArguments, index)), 509db884dafSAlex Zinenko terminator->getParent()); 510d9067dcaSKiran Chandramohan } 511d9067dcaSKiran Chandramohan } 512d9067dcaSKiran Chandramohan } 513d9067dcaSKiran Chandramohan } 514d9067dcaSKiran Chandramohan 515d9067dcaSKiran Chandramohan /// Sort function blocks topologically. 5164efb7754SRiver Riddle SetVector<Block *> 51766900b3eSAlex Zinenko mlir::LLVM::detail::getTopologicallySortedBlocks(Region ®ion) { 518d4568ed7SGeorge Mitenkov // For each block that has not been visited yet (i.e. that has no 519d4568ed7SGeorge Mitenkov // predecessors), add it to the list as well as its successors. 5204efb7754SRiver Riddle SetVector<Block *> blocks; 52166900b3eSAlex Zinenko for (Block &b : region) { 522d4568ed7SGeorge Mitenkov if (blocks.count(&b) == 0) { 523d4568ed7SGeorge Mitenkov llvm::ReversePostOrderTraversal<Block *> traversal(&b); 524d4568ed7SGeorge Mitenkov blocks.insert(traversal.begin(), traversal.end()); 525d4568ed7SGeorge Mitenkov } 526d9067dcaSKiran Chandramohan } 52766900b3eSAlex Zinenko assert(blocks.size() == region.getBlocks().size() && 52866900b3eSAlex Zinenko "some blocks are not sorted"); 529d9067dcaSKiran Chandramohan 530d9067dcaSKiran Chandramohan return blocks; 531d9067dcaSKiran Chandramohan } 532d9067dcaSKiran Chandramohan 533176379e0SAlex Zinenko llvm::Value *mlir::LLVM::detail::createIntrinsicCall( 534176379e0SAlex Zinenko llvm::IRBuilderBase &builder, llvm::Intrinsic::ID intrinsic, 535176379e0SAlex Zinenko ArrayRef<llvm::Value *> args, ArrayRef<llvm::Type *> tys) { 536176379e0SAlex Zinenko llvm::Module *module = builder.GetInsertBlock()->getModule(); 537176379e0SAlex Zinenko llvm::Function *fn = llvm::Intrinsic::getDeclaration(module, intrinsic, tys); 538176379e0SAlex Zinenko return builder.CreateCall(fn, args); 539176379e0SAlex Zinenko } 540176379e0SAlex Zinenko 5412666b973SRiver Riddle /// Given a single MLIR operation, create the corresponding LLVM IR operation 542176379e0SAlex Zinenko /// using the `builder`. 543ce8f10d6SAlex Zinenko LogicalResult 54438b106f6SMehdi Amini ModuleTranslation::convertOperation(Operation &op, 545ce8f10d6SAlex Zinenko llvm::IRBuilderBase &builder) { 54638b106f6SMehdi Amini const LLVMTranslationDialectInterface *opIface = iface.getInterfaceFor(&op); 54738b106f6SMehdi Amini if (!opIface) 54838b106f6SMehdi Amini return op.emitError("cannot be converted to LLVM IR: missing " 54938b106f6SMehdi Amini "`LLVMTranslationDialectInterface` registration for " 55038b106f6SMehdi Amini "dialect for op: ") 55138b106f6SMehdi Amini << op.getName(); 552176379e0SAlex Zinenko 55338b106f6SMehdi Amini if (failed(opIface->convertOperation(&op, builder, *this))) 55438b106f6SMehdi Amini return op.emitError("LLVM Translation failed for operation: ") 55538b106f6SMehdi Amini << op.getName(); 55638b106f6SMehdi Amini 55738b106f6SMehdi Amini return convertDialectAttributes(&op); 5585d7231d8SStephan Herhut } 5595d7231d8SStephan Herhut 5602666b973SRiver Riddle /// Convert block to LLVM IR. Unless `ignoreArguments` is set, emit PHI nodes 5612666b973SRiver Riddle /// to define values corresponding to the MLIR block arguments. These nodes 56210164a2eSAlex Zinenko /// are not connected to the source basic blocks, which may not exist yet. Uses 56310164a2eSAlex Zinenko /// `builder` to construct the LLVM IR. Expects the LLVM IR basic block to have 56410164a2eSAlex Zinenko /// been created for `bb` and included in the block mapping. Inserts new 56510164a2eSAlex Zinenko /// instructions at the end of the block and leaves `builder` in a state 56610164a2eSAlex Zinenko /// suitable for further insertion into the end of the block. 56710164a2eSAlex Zinenko LogicalResult ModuleTranslation::convertBlock(Block &bb, bool ignoreArguments, 568ce8f10d6SAlex Zinenko llvm::IRBuilderBase &builder) { 5690881a4f1SAlex Zinenko builder.SetInsertPoint(lookupBlock(&bb)); 570c33d6970SRiver Riddle auto *subprogram = builder.GetInsertBlock()->getParent()->getSubprogram(); 5715d7231d8SStephan Herhut 5725d7231d8SStephan Herhut // Before traversing operations, make block arguments available through 5735d7231d8SStephan Herhut // value remapping and PHI nodes, but do not add incoming edges for the PHI 5745d7231d8SStephan Herhut // nodes just yet: those values may be defined by this or following blocks. 5755d7231d8SStephan Herhut // This step is omitted if "ignoreArguments" is set. The arguments of the 5765d7231d8SStephan Herhut // first block have been already made available through the remapping of 5775d7231d8SStephan Herhut // LLVM function arguments. 5785d7231d8SStephan Herhut if (!ignoreArguments) { 5795d7231d8SStephan Herhut auto predecessors = bb.getPredecessors(); 5805d7231d8SStephan Herhut unsigned numPredecessors = 5815d7231d8SStephan Herhut std::distance(predecessors.begin(), predecessors.end()); 58235807bc4SRiver Riddle for (auto arg : bb.getArguments()) { 583c69c9e0fSAlex Zinenko auto wrappedType = arg.getType(); 584c69c9e0fSAlex Zinenko if (!isCompatibleType(wrappedType)) 585baa1ec22SAlex Zinenko return emitError(bb.front().getLoc(), 586a4c3a645SRiver Riddle "block argument does not have an LLVM type"); 587aec38c61SAlex Zinenko llvm::Type *type = convertType(wrappedType); 5885d7231d8SStephan Herhut llvm::PHINode *phi = builder.CreatePHI(type, numPredecessors); 5890881a4f1SAlex Zinenko mapValue(arg, phi); 5905d7231d8SStephan Herhut } 5915d7231d8SStephan Herhut } 5925d7231d8SStephan Herhut 5935d7231d8SStephan Herhut // Traverse operations. 5945d7231d8SStephan Herhut for (auto &op : bb) { 595c33d6970SRiver Riddle // Set the current debug location within the builder. 596c33d6970SRiver Riddle builder.SetCurrentDebugLocation( 597c33d6970SRiver Riddle debugTranslation->translateLoc(op.getLoc(), subprogram)); 598c33d6970SRiver Riddle 599baa1ec22SAlex Zinenko if (failed(convertOperation(op, builder))) 600baa1ec22SAlex Zinenko return failure(); 6015d7231d8SStephan Herhut } 6025d7231d8SStephan Herhut 603baa1ec22SAlex Zinenko return success(); 6045d7231d8SStephan Herhut } 6055d7231d8SStephan Herhut 606ce8f10d6SAlex Zinenko /// A helper method to get the single Block in an operation honoring LLVM's 607ce8f10d6SAlex Zinenko /// module requirements. 608ce8f10d6SAlex Zinenko static Block &getModuleBody(Operation *module) { 609ce8f10d6SAlex Zinenko return module->getRegion(0).front(); 610ce8f10d6SAlex Zinenko } 611ce8f10d6SAlex Zinenko 612ffa455d4SJean Perier /// A helper method to decide if a constant must not be set as a global variable 613d4df3825SAlex Zinenko /// initializer. For an external linkage variable, the variable with an 614d4df3825SAlex Zinenko /// initializer is considered externally visible and defined in this module, the 615d4df3825SAlex Zinenko /// variable without an initializer is externally available and is defined 616d4df3825SAlex Zinenko /// elsewhere. 617ffa455d4SJean Perier static bool shouldDropGlobalInitializer(llvm::GlobalValue::LinkageTypes linkage, 618ffa455d4SJean Perier llvm::Constant *cst) { 619d4df3825SAlex Zinenko return (linkage == llvm::GlobalVariable::ExternalLinkage && !cst) || 620ffa455d4SJean Perier linkage == llvm::GlobalVariable::ExternalWeakLinkage; 621ffa455d4SJean Perier } 622ffa455d4SJean Perier 6238ca04b05SFelipe de Azevedo Piovezan /// Sets the runtime preemption specifier of `gv` to dso_local if 6248ca04b05SFelipe de Azevedo Piovezan /// `dsoLocalRequested` is true, otherwise it is left unchanged. 6258ca04b05SFelipe de Azevedo Piovezan static void addRuntimePreemptionSpecifier(bool dsoLocalRequested, 6268ca04b05SFelipe de Azevedo Piovezan llvm::GlobalValue *gv) { 6278ca04b05SFelipe de Azevedo Piovezan if (dsoLocalRequested) 6288ca04b05SFelipe de Azevedo Piovezan gv->setDSOLocal(true); 6298ca04b05SFelipe de Azevedo Piovezan } 6308ca04b05SFelipe de Azevedo Piovezan 6312666b973SRiver Riddle /// Create named global variables that correspond to llvm.mlir.global 63257b9b296SUday Bondhugula /// definitions. Convert llvm.global_ctors and global_dtors ops. 633efa2d533SAlex Zinenko LogicalResult ModuleTranslation::convertGlobals() { 63444fc7d72STres Popp for (auto op : getModuleBody(mlirModule).getOps<LLVM::GlobalOp>()) { 635aec38c61SAlex Zinenko llvm::Type *type = convertType(op.getType()); 636d4df3825SAlex Zinenko llvm::Constant *cst = nullptr; 637250a11aeSJames Molloy if (op.getValueOrNull()) { 63868451df2SAlex Zinenko // String attributes are treated separately because they cannot appear as 63968451df2SAlex Zinenko // in-function constants and are thus not supported by getLLVMConstant. 64033a3a91bSChristian Sigg if (auto strAttr = op.getValueOrNull().dyn_cast_or_null<StringAttr>()) { 6412dd38b09SAlex Zinenko cst = llvm::ConstantDataArray::getString( 64268451df2SAlex Zinenko llvmModule->getContext(), strAttr.getValue(), /*AddNull=*/false); 6432dd38b09SAlex Zinenko type = cst->getType(); 644176379e0SAlex Zinenko } else if (!(cst = getLLVMConstant(type, op.getValueOrNull(), op.getLoc(), 645176379e0SAlex Zinenko *this))) { 646efa2d533SAlex Zinenko return failure(); 64768451df2SAlex Zinenko } 648ffa455d4SJean Perier } 649ffa455d4SJean Perier 650cfb72fd3SJacques Pienaar auto linkage = convertLinkageToLLVM(op.getLinkage()); 651cfb72fd3SJacques Pienaar auto addrSpace = op.getAddrSpace(); 652d4df3825SAlex Zinenko 653d4df3825SAlex Zinenko // LLVM IR requires constant with linkage other than external or weak 654d4df3825SAlex Zinenko // external to have initializers. If MLIR does not provide an initializer, 655d4df3825SAlex Zinenko // default to undef. 656d4df3825SAlex Zinenko bool dropInitializer = shouldDropGlobalInitializer(linkage, cst); 657d4df3825SAlex Zinenko if (!dropInitializer && !cst) 658d4df3825SAlex Zinenko cst = llvm::UndefValue::get(type); 659d4df3825SAlex Zinenko else if (dropInitializer && cst) 660d4df3825SAlex Zinenko cst = nullptr; 661d4df3825SAlex Zinenko 662ffa455d4SJean Perier auto *var = new llvm::GlobalVariable( 663cfb72fd3SJacques Pienaar *llvmModule, type, op.getConstant(), linkage, cst, op.getSymName(), 664*f0ba32d6SShraiysh Vaishay /*InsertBefore=*/nullptr, 665*f0ba32d6SShraiysh Vaishay op.getThreadLocal_() ? llvm::GlobalValue::GeneralDynamicTLSModel 666*f0ba32d6SShraiysh Vaishay : llvm::GlobalValue::NotThreadLocal, 667*f0ba32d6SShraiysh Vaishay addrSpace); 668ffa455d4SJean Perier 669cfb72fd3SJacques Pienaar if (op.getUnnamedAddr().hasValue()) 670cfb72fd3SJacques Pienaar var->setUnnamedAddr(convertUnnamedAddrToLLVM(*op.getUnnamedAddr())); 671c46a8862Sclementval 672cfb72fd3SJacques Pienaar if (op.getSection().hasValue()) 673cfb72fd3SJacques Pienaar var->setSection(*op.getSection()); 674b65472d6SRanjith Kumar H 675cfb72fd3SJacques Pienaar addRuntimePreemptionSpecifier(op.getDsoLocal(), var); 6768ca04b05SFelipe de Azevedo Piovezan 677cfb72fd3SJacques Pienaar Optional<uint64_t> alignment = op.getAlignment(); 6789a0ea599SDumitru Potop if (alignment.hasValue()) 6799a0ea599SDumitru Potop var->setAlignment(llvm::MaybeAlign(alignment.getValue())); 6809a0ea599SDumitru Potop 681ffa455d4SJean Perier globalsMapping.try_emplace(op, var); 682ffa455d4SJean Perier } 683ffa455d4SJean Perier 684ffa455d4SJean Perier // Convert global variable bodies. This is done after all global variables 685ffa455d4SJean Perier // have been created in LLVM IR because a global body may refer to another 686ffa455d4SJean Perier // global or itself. So all global variables need to be mapped first. 687ffa455d4SJean Perier for (auto op : getModuleBody(mlirModule).getOps<LLVM::GlobalOp>()) { 688ffa455d4SJean Perier if (Block *initializer = op.getInitializerBlock()) { 689250a11aeSJames Molloy llvm::IRBuilder<> builder(llvmModule->getContext()); 690250a11aeSJames Molloy for (auto &op : initializer->without_terminator()) { 691250a11aeSJames Molloy if (failed(convertOperation(op, builder)) || 6920881a4f1SAlex Zinenko !isa<llvm::Constant>(lookupValue(op.getResult(0)))) 693efa2d533SAlex Zinenko return emitError(op.getLoc(), "unemittable constant value"); 694250a11aeSJames Molloy } 695250a11aeSJames Molloy ReturnOp ret = cast<ReturnOp>(initializer->getTerminator()); 696ffa455d4SJean Perier llvm::Constant *cst = 697ffa455d4SJean Perier cast<llvm::Constant>(lookupValue(ret.getOperand(0))); 698ffa455d4SJean Perier auto *global = cast<llvm::GlobalVariable>(lookupGlobal(op)); 699ffa455d4SJean Perier if (!shouldDropGlobalInitializer(global->getLinkage(), cst)) 700ffa455d4SJean Perier global->setInitializer(cst); 701250a11aeSJames Molloy } 702b9ff2dd8SAlex Zinenko } 703efa2d533SAlex Zinenko 70457b9b296SUday Bondhugula // Convert llvm.mlir.global_ctors and dtors. 70557b9b296SUday Bondhugula for (Operation &op : getModuleBody(mlirModule)) { 70657b9b296SUday Bondhugula auto ctorOp = dyn_cast<GlobalCtorsOp>(op); 70757b9b296SUday Bondhugula auto dtorOp = dyn_cast<GlobalDtorsOp>(op); 70857b9b296SUday Bondhugula if (!ctorOp && !dtorOp) 70957b9b296SUday Bondhugula continue; 71062fea88bSJacques Pienaar auto range = ctorOp ? llvm::zip(ctorOp.getCtors(), ctorOp.getPriorities()) 71162fea88bSJacques Pienaar : llvm::zip(dtorOp.getDtors(), dtorOp.getPriorities()); 71257b9b296SUday Bondhugula auto appendGlobalFn = 71357b9b296SUday Bondhugula ctorOp ? llvm::appendToGlobalCtors : llvm::appendToGlobalDtors; 71457b9b296SUday Bondhugula for (auto symbolAndPriority : range) { 71557b9b296SUday Bondhugula llvm::Function *f = lookupFunction( 71657b9b296SUday Bondhugula std::get<0>(symbolAndPriority).cast<FlatSymbolRefAttr>().getValue()); 71757b9b296SUday Bondhugula appendGlobalFn( 71818eb6818SMehdi Amini *llvmModule, f, 71957b9b296SUday Bondhugula std::get<1>(symbolAndPriority).cast<IntegerAttr>().getInt(), 72057b9b296SUday Bondhugula /*Data=*/nullptr); 72157b9b296SUday Bondhugula } 72257b9b296SUday Bondhugula } 72357b9b296SUday Bondhugula 724efa2d533SAlex Zinenko return success(); 725b9ff2dd8SAlex Zinenko } 726b9ff2dd8SAlex Zinenko 7270a2131b7SAlex Zinenko /// Attempts to add an attribute identified by `key`, optionally with the given 7280a2131b7SAlex Zinenko /// `value` to LLVM function `llvmFunc`. Reports errors at `loc` if any. If the 7290a2131b7SAlex Zinenko /// attribute has a kind known to LLVM IR, create the attribute of this kind, 7300a2131b7SAlex Zinenko /// otherwise keep it as a string attribute. Performs additional checks for 7310a2131b7SAlex Zinenko /// attributes known to have or not have a value in order to avoid assertions 7320a2131b7SAlex Zinenko /// inside LLVM upon construction. 7330a2131b7SAlex Zinenko static LogicalResult checkedAddLLVMFnAttribute(Location loc, 7340a2131b7SAlex Zinenko llvm::Function *llvmFunc, 7350a2131b7SAlex Zinenko StringRef key, 7360a2131b7SAlex Zinenko StringRef value = StringRef()) { 7370a2131b7SAlex Zinenko auto kind = llvm::Attribute::getAttrKindFromName(key); 7380a2131b7SAlex Zinenko if (kind == llvm::Attribute::None) { 7390a2131b7SAlex Zinenko llvmFunc->addFnAttr(key, value); 7400a2131b7SAlex Zinenko return success(); 7410a2131b7SAlex Zinenko } 7420a2131b7SAlex Zinenko 7436ac32872SNikita Popov if (llvm::Attribute::isIntAttrKind(kind)) { 7440a2131b7SAlex Zinenko if (value.empty()) 7450a2131b7SAlex Zinenko return emitError(loc) << "LLVM attribute '" << key << "' expects a value"; 7460a2131b7SAlex Zinenko 7470a2131b7SAlex Zinenko int result; 7480a2131b7SAlex Zinenko if (!value.getAsInteger(/*Radix=*/0, result)) 7490a2131b7SAlex Zinenko llvmFunc->addFnAttr( 7500a2131b7SAlex Zinenko llvm::Attribute::get(llvmFunc->getContext(), kind, result)); 7510a2131b7SAlex Zinenko else 7520a2131b7SAlex Zinenko llvmFunc->addFnAttr(key, value); 7530a2131b7SAlex Zinenko return success(); 7540a2131b7SAlex Zinenko } 7550a2131b7SAlex Zinenko 7560a2131b7SAlex Zinenko if (!value.empty()) 7570a2131b7SAlex Zinenko return emitError(loc) << "LLVM attribute '" << key 7580a2131b7SAlex Zinenko << "' does not expect a value, found '" << value 7590a2131b7SAlex Zinenko << "'"; 7600a2131b7SAlex Zinenko 7610a2131b7SAlex Zinenko llvmFunc->addFnAttr(kind); 7620a2131b7SAlex Zinenko return success(); 7630a2131b7SAlex Zinenko } 7640a2131b7SAlex Zinenko 7650a2131b7SAlex Zinenko /// Attaches the attributes listed in the given array attribute to `llvmFunc`. 7660a2131b7SAlex Zinenko /// Reports error to `loc` if any and returns immediately. Expects `attributes` 7670a2131b7SAlex Zinenko /// to be an array attribute containing either string attributes, treated as 7680a2131b7SAlex Zinenko /// value-less LLVM attributes, or array attributes containing two string 7690a2131b7SAlex Zinenko /// attributes, with the first string being the name of the corresponding LLVM 7700a2131b7SAlex Zinenko /// attribute and the second string beings its value. Note that even integer 7710a2131b7SAlex Zinenko /// attributes are expected to have their values expressed as strings. 7720a2131b7SAlex Zinenko static LogicalResult 7730a2131b7SAlex Zinenko forwardPassthroughAttributes(Location loc, Optional<ArrayAttr> attributes, 7740a2131b7SAlex Zinenko llvm::Function *llvmFunc) { 7750a2131b7SAlex Zinenko if (!attributes) 7760a2131b7SAlex Zinenko return success(); 7770a2131b7SAlex Zinenko 7780a2131b7SAlex Zinenko for (Attribute attr : *attributes) { 7790a2131b7SAlex Zinenko if (auto stringAttr = attr.dyn_cast<StringAttr>()) { 7800a2131b7SAlex Zinenko if (failed( 7810a2131b7SAlex Zinenko checkedAddLLVMFnAttribute(loc, llvmFunc, stringAttr.getValue()))) 7820a2131b7SAlex Zinenko return failure(); 7830a2131b7SAlex Zinenko continue; 7840a2131b7SAlex Zinenko } 7850a2131b7SAlex Zinenko 7860a2131b7SAlex Zinenko auto arrayAttr = attr.dyn_cast<ArrayAttr>(); 7870a2131b7SAlex Zinenko if (!arrayAttr || arrayAttr.size() != 2) 7880a2131b7SAlex Zinenko return emitError(loc) 7890a2131b7SAlex Zinenko << "expected 'passthrough' to contain string or array attributes"; 7900a2131b7SAlex Zinenko 7910a2131b7SAlex Zinenko auto keyAttr = arrayAttr[0].dyn_cast<StringAttr>(); 7920a2131b7SAlex Zinenko auto valueAttr = arrayAttr[1].dyn_cast<StringAttr>(); 7930a2131b7SAlex Zinenko if (!keyAttr || !valueAttr) 7940a2131b7SAlex Zinenko return emitError(loc) 7950a2131b7SAlex Zinenko << "expected arrays within 'passthrough' to contain two strings"; 7960a2131b7SAlex Zinenko 7970a2131b7SAlex Zinenko if (failed(checkedAddLLVMFnAttribute(loc, llvmFunc, keyAttr.getValue(), 7980a2131b7SAlex Zinenko valueAttr.getValue()))) 7990a2131b7SAlex Zinenko return failure(); 8000a2131b7SAlex Zinenko } 8010a2131b7SAlex Zinenko return success(); 8020a2131b7SAlex Zinenko } 8030a2131b7SAlex Zinenko 8045e7959a3SAlex Zinenko LogicalResult ModuleTranslation::convertOneFunction(LLVMFuncOp func) { 805db884dafSAlex Zinenko // Clear the block, branch value mappings, they are only relevant within one 8065d7231d8SStephan Herhut // function. 8075d7231d8SStephan Herhut blockMapping.clear(); 8085d7231d8SStephan Herhut valueMapping.clear(); 809db884dafSAlex Zinenko branchMapping.clear(); 8100881a4f1SAlex Zinenko llvm::Function *llvmFunc = lookupFunction(func.getName()); 811c33d6970SRiver Riddle 812c33d6970SRiver Riddle // Translate the debug information for this function. 813c33d6970SRiver Riddle debugTranslation->translate(func, *llvmFunc); 814c33d6970SRiver Riddle 8155d7231d8SStephan Herhut // Add function arguments to the value remapping table. 8165d7231d8SStephan Herhut // If there was noalias info then we decorate each argument accordingly. 8175d7231d8SStephan Herhut unsigned int argIdx = 0; 818eeef50b1SFangrui Song for (auto kvp : llvm::zip(func.getArguments(), llvmFunc->args())) { 8195d7231d8SStephan Herhut llvm::Argument &llvmArg = std::get<1>(kvp); 820e62a6956SRiver Riddle BlockArgument mlirArg = std::get<0>(kvp); 8215d7231d8SStephan Herhut 8221c777ab4SUday Bondhugula if (auto attr = func.getArgAttrOfType<UnitAttr>( 82367cc5cecSStephan Herhut argIdx, LLVMDialect::getNoAliasAttrName())) { 8245d7231d8SStephan Herhut // NB: Attribute already verified to be boolean, so check if we can indeed 8255d7231d8SStephan Herhut // attach the attribute to this argument, based on its type. 826c69c9e0fSAlex Zinenko auto argTy = mlirArg.getType(); 8278de43b92SAlex Zinenko if (!argTy.isa<LLVM::LLVMPointerType>()) 828baa1ec22SAlex Zinenko return func.emitError( 8295d7231d8SStephan Herhut "llvm.noalias attribute attached to LLVM non-pointer argument"); 8305d7231d8SStephan Herhut llvmArg.addAttr(llvm::Attribute::AttrKind::NoAlias); 8315d7231d8SStephan Herhut } 8322416e28cSStephan Herhut 83367cc5cecSStephan Herhut if (auto attr = func.getArgAttrOfType<IntegerAttr>( 83467cc5cecSStephan Herhut argIdx, LLVMDialect::getAlignAttrName())) { 8352416e28cSStephan Herhut // NB: Attribute already verified to be int, so check if we can indeed 8362416e28cSStephan Herhut // attach the attribute to this argument, based on its type. 837c69c9e0fSAlex Zinenko auto argTy = mlirArg.getType(); 8388de43b92SAlex Zinenko if (!argTy.isa<LLVM::LLVMPointerType>()) 8392416e28cSStephan Herhut return func.emitError( 8402416e28cSStephan Herhut "llvm.align attribute attached to LLVM non-pointer argument"); 841d50298ddSEric Schweitz llvmArg.addAttrs(llvm::AttrBuilder(llvmArg.getContext()) 842d50298ddSEric Schweitz .addAlignmentAttr(llvm::Align(attr.getInt()))); 8432416e28cSStephan Herhut } 8442416e28cSStephan Herhut 84570b841acSEric Schweitz if (auto attr = func.getArgAttrOfType<UnitAttr>(argIdx, "llvm.sret")) { 846a4ad79c5SNikita Popov auto argTy = mlirArg.getType().dyn_cast<LLVM::LLVMPointerType>(); 847a4ad79c5SNikita Popov if (!argTy) 84870b841acSEric Schweitz return func.emitError( 84970b841acSEric Schweitz "llvm.sret attribute attached to LLVM non-pointer argument"); 850a4ad79c5SNikita Popov llvmArg.addAttrs( 851a4ad79c5SNikita Popov llvm::AttrBuilder(llvmArg.getContext()) 852a4ad79c5SNikita Popov .addStructRetAttr(convertType(argTy.getElementType()))); 85370b841acSEric Schweitz } 85470b841acSEric Schweitz 85570b841acSEric Schweitz if (auto attr = func.getArgAttrOfType<UnitAttr>(argIdx, "llvm.byval")) { 856a4ad79c5SNikita Popov auto argTy = mlirArg.getType().dyn_cast<LLVM::LLVMPointerType>(); 857a4ad79c5SNikita Popov if (!argTy) 85870b841acSEric Schweitz return func.emitError( 85970b841acSEric Schweitz "llvm.byval attribute attached to LLVM non-pointer argument"); 860a4ad79c5SNikita Popov llvmArg.addAttrs(llvm::AttrBuilder(llvmArg.getContext()) 861a4ad79c5SNikita Popov .addByValAttr(convertType(argTy.getElementType()))); 86270b841acSEric Schweitz } 86370b841acSEric Schweitz 8641c083e69SEric Schweitz if (auto attr = func.getArgAttrOfType<UnitAttr>(argIdx, "llvm.nest")) { 8651c083e69SEric Schweitz auto argTy = mlirArg.getType(); 8661c083e69SEric Schweitz if (!argTy.isa<LLVM::LLVMPointerType>()) 8671c083e69SEric Schweitz return func.emitError( 8681c083e69SEric Schweitz "llvm.nest attribute attached to LLVM non-pointer argument"); 8691c083e69SEric Schweitz llvmArg.addAttrs(llvm::AttrBuilder(llvmArg.getContext()) 8701c083e69SEric Schweitz .addAttribute(llvm::Attribute::Nest)); 8711c083e69SEric Schweitz } 8721c083e69SEric Schweitz 8730881a4f1SAlex Zinenko mapValue(mlirArg, &llvmArg); 8745d7231d8SStephan Herhut argIdx++; 8755d7231d8SStephan Herhut } 8765d7231d8SStephan Herhut 877ff77397fSShraiysh Vaishay // Check the personality and set it. 878dde96363SJacques Pienaar if (func.getPersonality().hasValue()) { 879ff77397fSShraiysh Vaishay llvm::Type *ty = llvm::Type::getInt8PtrTy(llvmFunc->getContext()); 880dde96363SJacques Pienaar if (llvm::Constant *pfunc = getLLVMConstant(ty, func.getPersonalityAttr(), 881dde96363SJacques Pienaar func.getLoc(), *this)) 882ff77397fSShraiysh Vaishay llvmFunc->setPersonalityFn(pfunc); 883ff77397fSShraiysh Vaishay } 884ff77397fSShraiysh Vaishay 8851bf79213SMarkus Böck if (auto gc = func.getGarbageCollector()) 8861bf79213SMarkus Böck llvmFunc->setGC(gc->str()); 8871bf79213SMarkus Böck 8885d7231d8SStephan Herhut // First, create all blocks so we can jump to them. 8895d7231d8SStephan Herhut llvm::LLVMContext &llvmContext = llvmFunc->getContext(); 8905d7231d8SStephan Herhut for (auto &bb : func) { 8915d7231d8SStephan Herhut auto *llvmBB = llvm::BasicBlock::Create(llvmContext); 8925d7231d8SStephan Herhut llvmBB->insertInto(llvmFunc); 8930881a4f1SAlex Zinenko mapBlock(&bb, llvmBB); 8945d7231d8SStephan Herhut } 8955d7231d8SStephan Herhut 8965d7231d8SStephan Herhut // Then, convert blocks one by one in topological order to ensure defs are 8975d7231d8SStephan Herhut // converted before uses. 89866900b3eSAlex Zinenko auto blocks = detail::getTopologicallySortedBlocks(func.getBody()); 89910164a2eSAlex Zinenko for (Block *bb : blocks) { 90010164a2eSAlex Zinenko llvm::IRBuilder<> builder(llvmContext); 90110164a2eSAlex Zinenko if (failed(convertBlock(*bb, bb->isEntryBlock(), builder))) 902baa1ec22SAlex Zinenko return failure(); 9035d7231d8SStephan Herhut } 9045d7231d8SStephan Herhut 905176379e0SAlex Zinenko // After all blocks have been traversed and values mapped, connect the PHI 906176379e0SAlex Zinenko // nodes to the results of preceding blocks. 90766900b3eSAlex Zinenko detail::connectPHINodes(func.getBody(), *this); 908176379e0SAlex Zinenko 909176379e0SAlex Zinenko // Finally, convert dialect attributes attached to the function. 910176379e0SAlex Zinenko return convertDialectAttributes(func); 911176379e0SAlex Zinenko } 912176379e0SAlex Zinenko 913176379e0SAlex Zinenko LogicalResult ModuleTranslation::convertDialectAttributes(Operation *op) { 914176379e0SAlex Zinenko for (NamedAttribute attribute : op->getDialectAttrs()) 915176379e0SAlex Zinenko if (failed(iface.amendOperation(op, attribute, *this))) 916176379e0SAlex Zinenko return failure(); 917baa1ec22SAlex Zinenko return success(); 9185d7231d8SStephan Herhut } 9195d7231d8SStephan Herhut 920a084b94fSSean Silva LogicalResult ModuleTranslation::convertFunctionSignatures() { 9215d7231d8SStephan Herhut // Declare all functions first because there may be function calls that form a 922a084b94fSSean Silva // call graph with cycles, or global initializers that reference functions. 92344fc7d72STres Popp for (auto function : getModuleBody(mlirModule).getOps<LLVMFuncOp>()) { 9245e7959a3SAlex Zinenko llvm::FunctionCallee llvmFuncCst = llvmModule->getOrInsertFunction( 9255e7959a3SAlex Zinenko function.getName(), 9264a3460a7SRiver Riddle cast<llvm::FunctionType>(convertType(function.getFunctionType()))); 9270a2131b7SAlex Zinenko llvm::Function *llvmFunc = cast<llvm::Function>(llvmFuncCst.getCallee()); 928dde96363SJacques Pienaar llvmFunc->setLinkage(convertLinkageToLLVM(function.getLinkage())); 9290881a4f1SAlex Zinenko mapFunction(function.getName(), llvmFunc); 930dde96363SJacques Pienaar addRuntimePreemptionSpecifier(function.getDsoLocal(), llvmFunc); 9310a2131b7SAlex Zinenko 9320a2131b7SAlex Zinenko // Forward the pass-through attributes to LLVM. 933dde96363SJacques Pienaar if (failed(forwardPassthroughAttributes( 934dde96363SJacques Pienaar function.getLoc(), function.getPassthrough(), llvmFunc))) 9350a2131b7SAlex Zinenko return failure(); 9365d7231d8SStephan Herhut } 9375d7231d8SStephan Herhut 938a084b94fSSean Silva return success(); 939a084b94fSSean Silva } 940a084b94fSSean Silva 941a084b94fSSean Silva LogicalResult ModuleTranslation::convertFunctions() { 9425d7231d8SStephan Herhut // Convert functions. 94344fc7d72STres Popp for (auto function : getModuleBody(mlirModule).getOps<LLVMFuncOp>()) { 9445d7231d8SStephan Herhut // Ignore external functions. 9455d7231d8SStephan Herhut if (function.isExternal()) 9465d7231d8SStephan Herhut continue; 9475d7231d8SStephan Herhut 948baa1ec22SAlex Zinenko if (failed(convertOneFunction(function))) 949baa1ec22SAlex Zinenko return failure(); 9505d7231d8SStephan Herhut } 9515d7231d8SStephan Herhut 952baa1ec22SAlex Zinenko return success(); 9535d7231d8SStephan Herhut } 9545d7231d8SStephan Herhut 9554a2930f4SArpith C. Jacob llvm::MDNode * 9564a2930f4SArpith C. Jacob ModuleTranslation::getAccessGroup(Operation &opInst, 9574a2930f4SArpith C. Jacob SymbolRefAttr accessGroupRef) const { 9584a2930f4SArpith C. Jacob auto metadataName = accessGroupRef.getRootReference(); 9594a2930f4SArpith C. Jacob auto accessGroupName = accessGroupRef.getLeafReference(); 9604a2930f4SArpith C. Jacob auto metadataOp = SymbolTable::lookupNearestSymbolFrom<LLVM::MetadataOp>( 9614a2930f4SArpith C. Jacob opInst.getParentOp(), metadataName); 9624a2930f4SArpith C. Jacob auto *accessGroupOp = 9634a2930f4SArpith C. Jacob SymbolTable::lookupNearestSymbolFrom(metadataOp, accessGroupName); 9644a2930f4SArpith C. Jacob return accessGroupMetadataMapping.lookup(accessGroupOp); 9654a2930f4SArpith C. Jacob } 9664a2930f4SArpith C. Jacob 9674a2930f4SArpith C. Jacob LogicalResult ModuleTranslation::createAccessGroupMetadata() { 9684a2930f4SArpith C. Jacob mlirModule->walk([&](LLVM::MetadataOp metadatas) { 9694a2930f4SArpith C. Jacob metadatas.walk([&](LLVM::AccessGroupMetadataOp op) { 9704a2930f4SArpith C. Jacob llvm::LLVMContext &ctx = llvmModule->getContext(); 9714a2930f4SArpith C. Jacob llvm::MDNode *accessGroup = llvm::MDNode::getDistinct(ctx, {}); 9724a2930f4SArpith C. Jacob accessGroupMetadataMapping.insert({op, accessGroup}); 9734a2930f4SArpith C. Jacob }); 9744a2930f4SArpith C. Jacob }); 9754a2930f4SArpith C. Jacob return success(); 9764a2930f4SArpith C. Jacob } 9774a2930f4SArpith C. Jacob 9784e393350SArpith C. Jacob void ModuleTranslation::setAccessGroupsMetadata(Operation *op, 9794e393350SArpith C. Jacob llvm::Instruction *inst) { 9804e393350SArpith C. Jacob auto accessGroups = 9814e393350SArpith C. Jacob op->getAttrOfType<ArrayAttr>(LLVMDialect::getAccessGroupsAttrName()); 9824e393350SArpith C. Jacob if (accessGroups && !accessGroups.empty()) { 9834e393350SArpith C. Jacob llvm::Module *module = inst->getModule(); 9844e393350SArpith C. Jacob SmallVector<llvm::Metadata *> metadatas; 9854e393350SArpith C. Jacob for (SymbolRefAttr accessGroupRef : 9864e393350SArpith C. Jacob accessGroups.getAsRange<SymbolRefAttr>()) 9874e393350SArpith C. Jacob metadatas.push_back(getAccessGroup(*op, accessGroupRef)); 9884e393350SArpith C. Jacob 9894e393350SArpith C. Jacob llvm::MDNode *unionMD = nullptr; 9904e393350SArpith C. Jacob if (metadatas.size() == 1) 9914e393350SArpith C. Jacob unionMD = llvm::cast<llvm::MDNode>(metadatas.front()); 9924e393350SArpith C. Jacob else if (metadatas.size() >= 2) 9934e393350SArpith C. Jacob unionMD = llvm::MDNode::get(module->getContext(), metadatas); 9944e393350SArpith C. Jacob 9954e393350SArpith C. Jacob inst->setMetadata(module->getMDKindID("llvm.access.group"), unionMD); 9964e393350SArpith C. Jacob } 9974e393350SArpith C. Jacob } 9984e393350SArpith C. Jacob 999d25e91d7STyler Augustine LogicalResult ModuleTranslation::createAliasScopeMetadata() { 1000d25e91d7STyler Augustine mlirModule->walk([&](LLVM::MetadataOp metadatas) { 1001d25e91d7STyler Augustine // Create the domains first, so they can be reference below in the scopes. 1002d25e91d7STyler Augustine DenseMap<Operation *, llvm::MDNode *> aliasScopeDomainMetadataMapping; 1003d25e91d7STyler Augustine metadatas.walk([&](LLVM::AliasScopeDomainMetadataOp op) { 1004d25e91d7STyler Augustine llvm::LLVMContext &ctx = llvmModule->getContext(); 1005d25e91d7STyler Augustine llvm::SmallVector<llvm::Metadata *, 2> operands; 1006d25e91d7STyler Augustine operands.push_back({}); // Placeholder for self-reference 1007cfb72fd3SJacques Pienaar if (Optional<StringRef> description = op.getDescription()) 1008d25e91d7STyler Augustine operands.push_back(llvm::MDString::get(ctx, description.getValue())); 1009d25e91d7STyler Augustine llvm::MDNode *domain = llvm::MDNode::get(ctx, operands); 1010d25e91d7STyler Augustine domain->replaceOperandWith(0, domain); // Self-reference for uniqueness 1011d25e91d7STyler Augustine aliasScopeDomainMetadataMapping.insert({op, domain}); 1012d25e91d7STyler Augustine }); 1013d25e91d7STyler Augustine 1014d25e91d7STyler Augustine // Now create the scopes, referencing the domains created above. 1015d25e91d7STyler Augustine metadatas.walk([&](LLVM::AliasScopeMetadataOp op) { 1016d25e91d7STyler Augustine llvm::LLVMContext &ctx = llvmModule->getContext(); 1017d25e91d7STyler Augustine assert(isa<LLVM::MetadataOp>(op->getParentOp())); 1018d25e91d7STyler Augustine auto metadataOp = dyn_cast<LLVM::MetadataOp>(op->getParentOp()); 1019d25e91d7STyler Augustine Operation *domainOp = 1020cfb72fd3SJacques Pienaar SymbolTable::lookupNearestSymbolFrom(metadataOp, op.getDomainAttr()); 1021d25e91d7STyler Augustine llvm::MDNode *domain = aliasScopeDomainMetadataMapping.lookup(domainOp); 1022d25e91d7STyler Augustine assert(domain && "Scope's domain should already be valid"); 1023d25e91d7STyler Augustine llvm::SmallVector<llvm::Metadata *, 3> operands; 1024d25e91d7STyler Augustine operands.push_back({}); // Placeholder for self-reference 1025d25e91d7STyler Augustine operands.push_back(domain); 1026cfb72fd3SJacques Pienaar if (Optional<StringRef> description = op.getDescription()) 1027d25e91d7STyler Augustine operands.push_back(llvm::MDString::get(ctx, description.getValue())); 1028d25e91d7STyler Augustine llvm::MDNode *scope = llvm::MDNode::get(ctx, operands); 1029d25e91d7STyler Augustine scope->replaceOperandWith(0, scope); // Self-reference for uniqueness 1030d25e91d7STyler Augustine aliasScopeMetadataMapping.insert({op, scope}); 1031d25e91d7STyler Augustine }); 1032d25e91d7STyler Augustine }); 1033d25e91d7STyler Augustine return success(); 1034d25e91d7STyler Augustine } 1035d25e91d7STyler Augustine 1036d25e91d7STyler Augustine llvm::MDNode * 1037d25e91d7STyler Augustine ModuleTranslation::getAliasScope(Operation &opInst, 1038d25e91d7STyler Augustine SymbolRefAttr aliasScopeRef) const { 103941d4aa7dSChris Lattner StringAttr metadataName = aliasScopeRef.getRootReference(); 104041d4aa7dSChris Lattner StringAttr scopeName = aliasScopeRef.getLeafReference(); 1041d25e91d7STyler Augustine auto metadataOp = SymbolTable::lookupNearestSymbolFrom<LLVM::MetadataOp>( 1042d25e91d7STyler Augustine opInst.getParentOp(), metadataName); 1043d25e91d7STyler Augustine Operation *aliasScopeOp = 1044d25e91d7STyler Augustine SymbolTable::lookupNearestSymbolFrom(metadataOp, scopeName); 1045d25e91d7STyler Augustine return aliasScopeMetadataMapping.lookup(aliasScopeOp); 1046d25e91d7STyler Augustine } 1047d25e91d7STyler Augustine 1048d25e91d7STyler Augustine void ModuleTranslation::setAliasScopeMetadata(Operation *op, 1049d25e91d7STyler Augustine llvm::Instruction *inst) { 1050d25e91d7STyler Augustine auto populateScopeMetadata = [this, op, inst](StringRef attrName, 1051d25e91d7STyler Augustine StringRef llvmMetadataName) { 1052d25e91d7STyler Augustine auto scopes = op->getAttrOfType<ArrayAttr>(attrName); 1053d25e91d7STyler Augustine if (!scopes || scopes.empty()) 1054d25e91d7STyler Augustine return; 1055d25e91d7STyler Augustine llvm::Module *module = inst->getModule(); 1056d25e91d7STyler Augustine SmallVector<llvm::Metadata *> scopeMDs; 1057d25e91d7STyler Augustine for (SymbolRefAttr scopeRef : scopes.getAsRange<SymbolRefAttr>()) 1058d25e91d7STyler Augustine scopeMDs.push_back(getAliasScope(*op, scopeRef)); 1059bdaf0382SAlex Zinenko llvm::MDNode *unionMD = llvm::MDNode::get(module->getContext(), scopeMDs); 1060d25e91d7STyler Augustine inst->setMetadata(module->getMDKindID(llvmMetadataName), unionMD); 1061d25e91d7STyler Augustine }; 1062d25e91d7STyler Augustine 1063d25e91d7STyler Augustine populateScopeMetadata(LLVMDialect::getAliasScopesAttrName(), "alias.scope"); 1064d25e91d7STyler Augustine populateScopeMetadata(LLVMDialect::getNoAliasScopesAttrName(), "noalias"); 1065d25e91d7STyler Augustine } 1066d25e91d7STyler Augustine 1067c69c9e0fSAlex Zinenko llvm::Type *ModuleTranslation::convertType(Type type) { 1068b2ab375dSAlex Zinenko return typeTranslator.translateType(type); 1069aec38c61SAlex Zinenko } 1070aec38c61SAlex Zinenko 10718647e4c3SAlex Zinenko /// A helper to look up remapped operands in the value remapping table. 10728647e4c3SAlex Zinenko SmallVector<llvm::Value *> ModuleTranslation::lookupValues(ValueRange values) { 10738647e4c3SAlex Zinenko SmallVector<llvm::Value *> remapped; 1074efadb6b8SAlex Zinenko remapped.reserve(values.size()); 10750881a4f1SAlex Zinenko for (Value v : values) 10760881a4f1SAlex Zinenko remapped.push_back(lookupValue(v)); 1077efadb6b8SAlex Zinenko return remapped; 1078efadb6b8SAlex Zinenko } 1079efadb6b8SAlex Zinenko 108066900b3eSAlex Zinenko const llvm::DILocation * 108166900b3eSAlex Zinenko ModuleTranslation::translateLoc(Location loc, llvm::DILocalScope *scope) { 108266900b3eSAlex Zinenko return debugTranslation->translateLoc(loc, scope); 108366900b3eSAlex Zinenko } 108466900b3eSAlex Zinenko 1085176379e0SAlex Zinenko llvm::NamedMDNode * 1086176379e0SAlex Zinenko ModuleTranslation::getOrInsertNamedModuleMetadata(StringRef name) { 1087176379e0SAlex Zinenko return llvmModule->getOrInsertNamedMetadata(name); 1088176379e0SAlex Zinenko } 1089176379e0SAlex Zinenko 109072d013ddSAlex Zinenko void ModuleTranslation::StackFrame::anchor() {} 109172d013ddSAlex Zinenko 1092ce8f10d6SAlex Zinenko static std::unique_ptr<llvm::Module> 1093ce8f10d6SAlex Zinenko prepareLLVMModule(Operation *m, llvm::LLVMContext &llvmContext, 1094ce8f10d6SAlex Zinenko StringRef name) { 1095f9dc2b70SMehdi Amini m->getContext()->getOrLoadDialect<LLVM::LLVMDialect>(); 1096db1c197bSAlex Zinenko auto llvmModule = std::make_unique<llvm::Module>(name, llvmContext); 1097168213f9SAlex Zinenko if (auto dataLayoutAttr = 1098ea998709SAlex Zinenko m->getAttr(LLVM::LLVMDialect::getDataLayoutAttrName())) { 1099168213f9SAlex Zinenko llvmModule->setDataLayout(dataLayoutAttr.cast<StringAttr>().getValue()); 1100ea998709SAlex Zinenko } else { 1101ea998709SAlex Zinenko FailureOr<llvm::DataLayout> llvmDataLayout(llvm::DataLayout("")); 1102ea998709SAlex Zinenko if (auto iface = dyn_cast<DataLayoutOpInterface>(m)) { 1103ea998709SAlex Zinenko if (DataLayoutSpecInterface spec = iface.getDataLayoutSpec()) { 1104ea998709SAlex Zinenko llvmDataLayout = 1105ea998709SAlex Zinenko translateDataLayout(spec, DataLayout(iface), m->getLoc()); 1106ea998709SAlex Zinenko } 1107ea998709SAlex Zinenko } else if (auto mod = dyn_cast<ModuleOp>(m)) { 1108ea998709SAlex Zinenko if (DataLayoutSpecInterface spec = mod.getDataLayoutSpec()) { 1109ea998709SAlex Zinenko llvmDataLayout = 1110ea998709SAlex Zinenko translateDataLayout(spec, DataLayout(mod), m->getLoc()); 1111ea998709SAlex Zinenko } 1112ea998709SAlex Zinenko } 1113ea998709SAlex Zinenko if (failed(llvmDataLayout)) 1114ea998709SAlex Zinenko return nullptr; 1115ea998709SAlex Zinenko llvmModule->setDataLayout(*llvmDataLayout); 1116ea998709SAlex Zinenko } 11175dd5a083SNicolas Vasilache if (auto targetTripleAttr = 11185dd5a083SNicolas Vasilache m->getAttr(LLVM::LLVMDialect::getTargetTripleAttrName())) 11195dd5a083SNicolas Vasilache llvmModule->setTargetTriple(targetTripleAttr.cast<StringAttr>().getValue()); 11205d7231d8SStephan Herhut 11215d7231d8SStephan Herhut // Inject declarations for `malloc` and `free` functions that can be used in 11225d7231d8SStephan Herhut // memref allocation/deallocation coming from standard ops lowering. 1123db1c197bSAlex Zinenko llvm::IRBuilder<> builder(llvmContext); 11245d7231d8SStephan Herhut llvmModule->getOrInsertFunction("malloc", builder.getInt8PtrTy(), 11255d7231d8SStephan Herhut builder.getInt64Ty()); 11265d7231d8SStephan Herhut llvmModule->getOrInsertFunction("free", builder.getVoidTy(), 11275d7231d8SStephan Herhut builder.getInt8PtrTy()); 11285d7231d8SStephan Herhut 11295d7231d8SStephan Herhut return llvmModule; 11305d7231d8SStephan Herhut } 1131ce8f10d6SAlex Zinenko 1132ce8f10d6SAlex Zinenko std::unique_ptr<llvm::Module> 1133ce8f10d6SAlex Zinenko mlir::translateModuleToLLVMIR(Operation *module, llvm::LLVMContext &llvmContext, 1134ce8f10d6SAlex Zinenko StringRef name) { 1135ce8f10d6SAlex Zinenko if (!satisfiesLLVMModule(module)) 1136ce8f10d6SAlex Zinenko return nullptr; 1137ea998709SAlex Zinenko 1138ce8f10d6SAlex Zinenko std::unique_ptr<llvm::Module> llvmModule = 1139ce8f10d6SAlex Zinenko prepareLLVMModule(module, llvmContext, name); 1140ea998709SAlex Zinenko if (!llvmModule) 1141ea998709SAlex Zinenko return nullptr; 1142ce8f10d6SAlex Zinenko 1143ce8f10d6SAlex Zinenko LLVM::ensureDistinctSuccessors(module); 1144ce8f10d6SAlex Zinenko 1145ce8f10d6SAlex Zinenko ModuleTranslation translator(module, std::move(llvmModule)); 1146ce8f10d6SAlex Zinenko if (failed(translator.convertFunctionSignatures())) 1147ce8f10d6SAlex Zinenko return nullptr; 1148ce8f10d6SAlex Zinenko if (failed(translator.convertGlobals())) 1149ce8f10d6SAlex Zinenko return nullptr; 11504a2930f4SArpith C. Jacob if (failed(translator.createAccessGroupMetadata())) 11514a2930f4SArpith C. Jacob return nullptr; 1152d25e91d7STyler Augustine if (failed(translator.createAliasScopeMetadata())) 1153d25e91d7STyler Augustine return nullptr; 1154ce8f10d6SAlex Zinenko if (failed(translator.convertFunctions())) 1155ce8f10d6SAlex Zinenko return nullptr; 11568647e4c3SAlex Zinenko 11578647e4c3SAlex Zinenko // Convert other top-level operations if possible. 11588647e4c3SAlex Zinenko llvm::IRBuilder<> llvmBuilder(llvmContext); 11598647e4c3SAlex Zinenko for (Operation &o : getModuleBody(module).getOperations()) { 116057b9b296SUday Bondhugula if (!isa<LLVM::LLVMFuncOp, LLVM::GlobalOp, LLVM::GlobalCtorsOp, 116157b9b296SUday Bondhugula LLVM::GlobalDtorsOp, LLVM::MetadataOp>(&o) && 11628647e4c3SAlex Zinenko !o.hasTrait<OpTrait::IsTerminator>() && 11638647e4c3SAlex Zinenko failed(translator.convertOperation(o, llvmBuilder))) { 11648647e4c3SAlex Zinenko return nullptr; 11658647e4c3SAlex Zinenko } 11668647e4c3SAlex Zinenko } 11678647e4c3SAlex Zinenko 1168ce8f10d6SAlex Zinenko if (llvm::verifyModule(*translator.llvmModule, &llvm::errs())) 1169ce8f10d6SAlex Zinenko return nullptr; 1170ce8f10d6SAlex Zinenko 1171ce8f10d6SAlex Zinenko return std::move(translator.llvmModule); 1172ce8f10d6SAlex Zinenko } 1173