1044d5b5dSValentin Clement //===-- CodeGen.cpp -- bridge to lower to LLVM ----------------------------===// 2044d5b5dSValentin Clement // 3044d5b5dSValentin Clement // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4044d5b5dSValentin Clement // See https://llvm.org/LICENSE.txt for license information. 5044d5b5dSValentin Clement // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6044d5b5dSValentin Clement // 7044d5b5dSValentin Clement //===----------------------------------------------------------------------===// 8044d5b5dSValentin Clement // 9044d5b5dSValentin Clement // Coding style: https://mlir.llvm.org/getting_started/DeveloperGuide/ 10044d5b5dSValentin Clement // 11044d5b5dSValentin Clement //===----------------------------------------------------------------------===// 12044d5b5dSValentin Clement 13044d5b5dSValentin Clement #include "flang/Optimizer/CodeGen/CodeGen.h" 14044d5b5dSValentin Clement #include "PassDetail.h" 15b6e44ecdSValentin Clement #include "flang/ISO_Fortran_binding.h" 1639f4ef81SValentin Clement #include "flang/Optimizer/Dialect/FIRAttr.h" 17044d5b5dSValentin Clement #include "flang/Optimizer/Dialect/FIROps.h" 18044d5b5dSValentin Clement #include "mlir/Conversion/ArithmeticToLLVM/ArithmeticToLLVM.h" 19044d5b5dSValentin Clement #include "mlir/Conversion/LLVMCommon/Pattern.h" 20044d5b5dSValentin Clement #include "mlir/Conversion/StandardToLLVM/ConvertStandardToLLVM.h" 21044d5b5dSValentin Clement #include "mlir/IR/BuiltinTypes.h" 223ae8e442SValentin Clement #include "mlir/IR/Matchers.h" 23044d5b5dSValentin Clement #include "mlir/Pass/Pass.h" 24044d5b5dSValentin Clement #include "llvm/ADT/ArrayRef.h" 25044d5b5dSValentin Clement 26044d5b5dSValentin Clement #define DEBUG_TYPE "flang-codegen" 27044d5b5dSValentin Clement 28044d5b5dSValentin Clement // fir::LLVMTypeConverter for converting to LLVM IR dialect types. 29044d5b5dSValentin Clement #include "TypeConverter.h" 30044d5b5dSValentin Clement 31b6e44ecdSValentin Clement /// `fir.box` attribute values as defined for CFI_attribute_t in 32b6e44ecdSValentin Clement /// flang/ISO_Fortran_binding.h. 33b6e44ecdSValentin Clement static constexpr unsigned kAttrPointer = CFI_attribute_pointer; 34b6e44ecdSValentin Clement static constexpr unsigned kAttrAllocatable = CFI_attribute_allocatable; 35b6e44ecdSValentin Clement 361e6d9c06SDiana Picus static mlir::LLVM::ConstantOp 371e6d9c06SDiana Picus genConstantIndex(mlir::Location loc, mlir::Type ity, 381e6d9c06SDiana Picus mlir::ConversionPatternRewriter &rewriter, 391e6d9c06SDiana Picus std::int64_t offset) { 401e6d9c06SDiana Picus auto cattr = rewriter.getI64IntegerAttr(offset); 411e6d9c06SDiana Picus return rewriter.create<mlir::LLVM::ConstantOp>(loc, ity, cattr); 421e6d9c06SDiana Picus } 431e6d9c06SDiana Picus 4439f4ef81SValentin Clement static Block *createBlock(mlir::ConversionPatternRewriter &rewriter, 4539f4ef81SValentin Clement mlir::Block *insertBefore) { 4639f4ef81SValentin Clement assert(insertBefore && "expected valid insertion block"); 4739f4ef81SValentin Clement return rewriter.createBlock(insertBefore->getParent(), 4839f4ef81SValentin Clement mlir::Region::iterator(insertBefore)); 4939f4ef81SValentin Clement } 5039f4ef81SValentin Clement 51044d5b5dSValentin Clement namespace { 52044d5b5dSValentin Clement /// FIR conversion pattern template 53044d5b5dSValentin Clement template <typename FromOp> 54044d5b5dSValentin Clement class FIROpConversion : public mlir::ConvertOpToLLVMPattern<FromOp> { 55044d5b5dSValentin Clement public: 56044d5b5dSValentin Clement explicit FIROpConversion(fir::LLVMTypeConverter &lowering) 57044d5b5dSValentin Clement : mlir::ConvertOpToLLVMPattern<FromOp>(lowering) {} 58044d5b5dSValentin Clement 59044d5b5dSValentin Clement protected: 60044d5b5dSValentin Clement mlir::Type convertType(mlir::Type ty) const { 61044d5b5dSValentin Clement return lowerTy().convertType(ty); 62044d5b5dSValentin Clement } 63044d5b5dSValentin Clement 64df3b9810SValentin Clement mlir::LLVM::ConstantOp 65df3b9810SValentin Clement genConstantOffset(mlir::Location loc, 66df3b9810SValentin Clement mlir::ConversionPatternRewriter &rewriter, 67df3b9810SValentin Clement int offset) const { 68df3b9810SValentin Clement auto ity = lowerTy().offsetType(); 69df3b9810SValentin Clement auto cattr = rewriter.getI32IntegerAttr(offset); 70df3b9810SValentin Clement return rewriter.create<mlir::LLVM::ConstantOp>(loc, ity, cattr); 71df3b9810SValentin Clement } 72df3b9810SValentin Clement 73b6e44ecdSValentin Clement /// Construct code sequence to extract the specifc value from a `fir.box`. 74b6e44ecdSValentin Clement mlir::Value getValueFromBox(mlir::Location loc, mlir::Value box, 75df3b9810SValentin Clement mlir::Type resultTy, 76b6e44ecdSValentin Clement mlir::ConversionPatternRewriter &rewriter, 77b6e44ecdSValentin Clement unsigned boxValue) const { 78df3b9810SValentin Clement mlir::LLVM::ConstantOp c0 = genConstantOffset(loc, rewriter, 0); 79b6e44ecdSValentin Clement mlir::LLVM::ConstantOp cValuePos = 80b6e44ecdSValentin Clement genConstantOffset(loc, rewriter, boxValue); 81df3b9810SValentin Clement auto pty = mlir::LLVM::LLVMPointerType::get(resultTy); 82df3b9810SValentin Clement auto p = rewriter.create<mlir::LLVM::GEPOp>( 83b6e44ecdSValentin Clement loc, pty, mlir::ValueRange{box, c0, cValuePos}); 84df3b9810SValentin Clement return rewriter.create<mlir::LLVM::LoadOp>(loc, resultTy, p); 85df3b9810SValentin Clement } 86df3b9810SValentin Clement 87df3b9810SValentin Clement /// Method to construct code sequence to get the triple for dimension `dim` 88df3b9810SValentin Clement /// from a box. 89df3b9810SValentin Clement SmallVector<mlir::Value, 3> 90df3b9810SValentin Clement getDimsFromBox(mlir::Location loc, ArrayRef<mlir::Type> retTys, 91df3b9810SValentin Clement mlir::Value box, mlir::Value dim, 92df3b9810SValentin Clement mlir::ConversionPatternRewriter &rewriter) const { 93df3b9810SValentin Clement mlir::LLVM::ConstantOp c0 = genConstantOffset(loc, rewriter, 0); 94df3b9810SValentin Clement mlir::LLVM::ConstantOp cDims = 95df3b9810SValentin Clement genConstantOffset(loc, rewriter, kDimsPosInBox); 96df3b9810SValentin Clement mlir::LLVM::LoadOp l0 = 97df3b9810SValentin Clement loadFromOffset(loc, box, c0, cDims, dim, 0, retTys[0], rewriter); 98df3b9810SValentin Clement mlir::LLVM::LoadOp l1 = 99df3b9810SValentin Clement loadFromOffset(loc, box, c0, cDims, dim, 1, retTys[1], rewriter); 100df3b9810SValentin Clement mlir::LLVM::LoadOp l2 = 101df3b9810SValentin Clement loadFromOffset(loc, box, c0, cDims, dim, 2, retTys[2], rewriter); 102df3b9810SValentin Clement return {l0.getResult(), l1.getResult(), l2.getResult()}; 103df3b9810SValentin Clement } 104df3b9810SValentin Clement 105df3b9810SValentin Clement mlir::LLVM::LoadOp 106df3b9810SValentin Clement loadFromOffset(mlir::Location loc, mlir::Value a, mlir::LLVM::ConstantOp c0, 107df3b9810SValentin Clement mlir::LLVM::ConstantOp cDims, mlir::Value dim, int off, 108df3b9810SValentin Clement mlir::Type ty, 109df3b9810SValentin Clement mlir::ConversionPatternRewriter &rewriter) const { 110df3b9810SValentin Clement auto pty = mlir::LLVM::LLVMPointerType::get(ty); 111df3b9810SValentin Clement mlir::LLVM::ConstantOp c = genConstantOffset(loc, rewriter, off); 112df3b9810SValentin Clement mlir::LLVM::GEPOp p = genGEP(loc, pty, rewriter, a, c0, cDims, dim, c); 113df3b9810SValentin Clement return rewriter.create<mlir::LLVM::LoadOp>(loc, ty, p); 114df3b9810SValentin Clement } 115df3b9810SValentin Clement 116df3b9810SValentin Clement /// Read base address from a fir.box. Returned address has type ty. 117df3b9810SValentin Clement mlir::Value 118df3b9810SValentin Clement loadBaseAddrFromBox(mlir::Location loc, mlir::Type ty, mlir::Value box, 119df3b9810SValentin Clement mlir::ConversionPatternRewriter &rewriter) const { 120df3b9810SValentin Clement mlir::LLVM::ConstantOp c0 = genConstantOffset(loc, rewriter, 0); 121df3b9810SValentin Clement mlir::LLVM::ConstantOp cAddr = 122df3b9810SValentin Clement genConstantOffset(loc, rewriter, kAddrPosInBox); 123df3b9810SValentin Clement auto pty = mlir::LLVM::LLVMPointerType::get(ty); 124df3b9810SValentin Clement mlir::LLVM::GEPOp p = genGEP(loc, pty, rewriter, box, c0, cAddr); 125df3b9810SValentin Clement return rewriter.create<mlir::LLVM::LoadOp>(loc, ty, p); 126df3b9810SValentin Clement } 127df3b9810SValentin Clement 128df3b9810SValentin Clement mlir::Value 129df3b9810SValentin Clement loadElementSizeFromBox(mlir::Location loc, mlir::Type ty, mlir::Value box, 130df3b9810SValentin Clement mlir::ConversionPatternRewriter &rewriter) const { 131df3b9810SValentin Clement mlir::LLVM::ConstantOp c0 = genConstantOffset(loc, rewriter, 0); 132df3b9810SValentin Clement mlir::LLVM::ConstantOp cElemLen = 133df3b9810SValentin Clement genConstantOffset(loc, rewriter, kElemLenPosInBox); 134df3b9810SValentin Clement auto pty = mlir::LLVM::LLVMPointerType::get(ty); 135df3b9810SValentin Clement mlir::LLVM::GEPOp p = genGEP(loc, pty, rewriter, box, c0, cElemLen); 136df3b9810SValentin Clement return rewriter.create<mlir::LLVM::LoadOp>(loc, ty, p); 137df3b9810SValentin Clement } 138df3b9810SValentin Clement 139b6e44ecdSValentin Clement // Load the attribute from the \p box and perform a check against \p maskValue 140b6e44ecdSValentin Clement // The final comparison is implemented as `(attribute & maskValue) != 0`. 141b6e44ecdSValentin Clement mlir::Value genBoxAttributeCheck(mlir::Location loc, mlir::Value box, 142b6e44ecdSValentin Clement mlir::ConversionPatternRewriter &rewriter, 143b6e44ecdSValentin Clement unsigned maskValue) const { 144b6e44ecdSValentin Clement mlir::Type attrTy = rewriter.getI32Type(); 145b6e44ecdSValentin Clement mlir::Value attribute = 146b6e44ecdSValentin Clement getValueFromBox(loc, box, attrTy, rewriter, kAttributePosInBox); 147b6e44ecdSValentin Clement mlir::LLVM::ConstantOp attrMask = 148b6e44ecdSValentin Clement genConstantOffset(loc, rewriter, maskValue); 149b6e44ecdSValentin Clement auto maskRes = 150b6e44ecdSValentin Clement rewriter.create<mlir::LLVM::AndOp>(loc, attrTy, attribute, attrMask); 151b6e44ecdSValentin Clement mlir::LLVM::ConstantOp c0 = genConstantOffset(loc, rewriter, 0); 152b6e44ecdSValentin Clement return rewriter.create<mlir::LLVM::ICmpOp>( 153b6e44ecdSValentin Clement loc, mlir::LLVM::ICmpPredicate::ne, maskRes, c0); 154b6e44ecdSValentin Clement } 155b6e44ecdSValentin Clement 156df3b9810SValentin Clement template <typename... ARGS> 157df3b9810SValentin Clement mlir::LLVM::GEPOp genGEP(mlir::Location loc, mlir::Type ty, 158df3b9810SValentin Clement mlir::ConversionPatternRewriter &rewriter, 159df3b9810SValentin Clement mlir::Value base, ARGS... args) const { 160df3b9810SValentin Clement SmallVector<mlir::Value> cv{args...}; 161df3b9810SValentin Clement return rewriter.create<mlir::LLVM::GEPOp>(loc, ty, base, cv); 162df3b9810SValentin Clement } 163df3b9810SValentin Clement 1641e6d9c06SDiana Picus /// Perform an extension or truncation as needed on an integer value. Lowering 1651e6d9c06SDiana Picus /// to the specific target may involve some sign-extending or truncation of 1661e6d9c06SDiana Picus /// values, particularly to fit them from abstract box types to the 1671e6d9c06SDiana Picus /// appropriate reified structures. 1681e6d9c06SDiana Picus mlir::Value integerCast(mlir::Location loc, 1691e6d9c06SDiana Picus mlir::ConversionPatternRewriter &rewriter, 1701e6d9c06SDiana Picus mlir::Type ty, mlir::Value val) const { 1711e6d9c06SDiana Picus auto valTy = val.getType(); 1721e6d9c06SDiana Picus // If the value was not yet lowered, lower its type so that it can 1731e6d9c06SDiana Picus // be used in getPrimitiveTypeSizeInBits. 1741e6d9c06SDiana Picus if (!valTy.isa<mlir::IntegerType>()) 1751e6d9c06SDiana Picus valTy = convertType(valTy); 1761e6d9c06SDiana Picus auto toSize = mlir::LLVM::getPrimitiveTypeSizeInBits(ty); 1771e6d9c06SDiana Picus auto fromSize = mlir::LLVM::getPrimitiveTypeSizeInBits(valTy); 1781e6d9c06SDiana Picus if (toSize < fromSize) 1791e6d9c06SDiana Picus return rewriter.create<mlir::LLVM::TruncOp>(loc, ty, val); 1801e6d9c06SDiana Picus if (toSize > fromSize) 1811e6d9c06SDiana Picus return rewriter.create<mlir::LLVM::SExtOp>(loc, ty, val); 1821e6d9c06SDiana Picus return val; 1831e6d9c06SDiana Picus } 1841e6d9c06SDiana Picus 185044d5b5dSValentin Clement fir::LLVMTypeConverter &lowerTy() const { 186044d5b5dSValentin Clement return *static_cast<fir::LLVMTypeConverter *>(this->getTypeConverter()); 187044d5b5dSValentin Clement } 188044d5b5dSValentin Clement }; 189044d5b5dSValentin Clement 1903ae8e442SValentin Clement /// FIR conversion pattern template 1913ae8e442SValentin Clement template <typename FromOp> 1923ae8e442SValentin Clement class FIROpAndTypeConversion : public FIROpConversion<FromOp> { 1933ae8e442SValentin Clement public: 1943ae8e442SValentin Clement using FIROpConversion<FromOp>::FIROpConversion; 1953ae8e442SValentin Clement using OpAdaptor = typename FromOp::Adaptor; 1963ae8e442SValentin Clement 1973ae8e442SValentin Clement mlir::LogicalResult 1983ae8e442SValentin Clement matchAndRewrite(FromOp op, OpAdaptor adaptor, 1993ae8e442SValentin Clement mlir::ConversionPatternRewriter &rewriter) const final { 2003ae8e442SValentin Clement mlir::Type ty = this->convertType(op.getType()); 2013ae8e442SValentin Clement return doRewrite(op, ty, adaptor, rewriter); 2023ae8e442SValentin Clement } 2033ae8e442SValentin Clement 2043ae8e442SValentin Clement virtual mlir::LogicalResult 2053ae8e442SValentin Clement doRewrite(FromOp addr, mlir::Type ty, OpAdaptor adaptor, 2063ae8e442SValentin Clement mlir::ConversionPatternRewriter &rewriter) const = 0; 2073ae8e442SValentin Clement }; 2083ae8e442SValentin Clement 209420ad7ceSAndrzej Warzynski /// Create value signaling an absent optional argument in a call, e.g. 210420ad7ceSAndrzej Warzynski /// `fir.absent !fir.ref<i64>` --> `llvm.mlir.null : !llvm.ptr<i64>` 211420ad7ceSAndrzej Warzynski struct AbsentOpConversion : public FIROpConversion<fir::AbsentOp> { 212420ad7ceSAndrzej Warzynski using FIROpConversion::FIROpConversion; 213420ad7ceSAndrzej Warzynski 214420ad7ceSAndrzej Warzynski mlir::LogicalResult 215420ad7ceSAndrzej Warzynski matchAndRewrite(fir::AbsentOp absent, OpAdaptor, 216420ad7ceSAndrzej Warzynski mlir::ConversionPatternRewriter &rewriter) const override { 217420ad7ceSAndrzej Warzynski mlir::Type ty = convertType(absent.getType()); 218420ad7ceSAndrzej Warzynski mlir::Location loc = absent.getLoc(); 219420ad7ceSAndrzej Warzynski 220420ad7ceSAndrzej Warzynski if (absent.getType().isa<fir::BoxCharType>()) { 221420ad7ceSAndrzej Warzynski auto structTy = ty.cast<mlir::LLVM::LLVMStructType>(); 222420ad7ceSAndrzej Warzynski assert(!structTy.isOpaque() && !structTy.getBody().empty()); 223420ad7ceSAndrzej Warzynski auto undefStruct = rewriter.create<mlir::LLVM::UndefOp>(loc, ty); 224420ad7ceSAndrzej Warzynski auto nullField = 225420ad7ceSAndrzej Warzynski rewriter.create<mlir::LLVM::NullOp>(loc, structTy.getBody()[0]); 226420ad7ceSAndrzej Warzynski mlir::MLIRContext *ctx = absent.getContext(); 227420ad7ceSAndrzej Warzynski auto c0 = mlir::ArrayAttr::get(ctx, rewriter.getI32IntegerAttr(0)); 228420ad7ceSAndrzej Warzynski rewriter.replaceOpWithNewOp<mlir::LLVM::InsertValueOp>( 229420ad7ceSAndrzej Warzynski absent, ty, undefStruct, nullField, c0); 230420ad7ceSAndrzej Warzynski } else { 231420ad7ceSAndrzej Warzynski rewriter.replaceOpWithNewOp<mlir::LLVM::NullOp>(absent, ty); 232420ad7ceSAndrzej Warzynski } 233420ad7ceSAndrzej Warzynski return success(); 234420ad7ceSAndrzej Warzynski } 235420ad7ceSAndrzej Warzynski }; 236420ad7ceSAndrzej Warzynski 2370c4a7a52SValentin Clement // Lower `fir.address_of` operation to `llvm.address_of` operation. 238044d5b5dSValentin Clement struct AddrOfOpConversion : public FIROpConversion<fir::AddrOfOp> { 239044d5b5dSValentin Clement using FIROpConversion::FIROpConversion; 240044d5b5dSValentin Clement 241044d5b5dSValentin Clement mlir::LogicalResult 242044d5b5dSValentin Clement matchAndRewrite(fir::AddrOfOp addr, OpAdaptor adaptor, 243044d5b5dSValentin Clement mlir::ConversionPatternRewriter &rewriter) const override { 244044d5b5dSValentin Clement auto ty = convertType(addr.getType()); 245044d5b5dSValentin Clement rewriter.replaceOpWithNewOp<mlir::LLVM::AddressOfOp>( 246044d5b5dSValentin Clement addr, ty, addr.symbol().getRootReference().getValue()); 247044d5b5dSValentin Clement return success(); 248044d5b5dSValentin Clement } 249044d5b5dSValentin Clement }; 2501e6d9c06SDiana Picus } // namespace 2511e6d9c06SDiana Picus 2521e6d9c06SDiana Picus /// Lookup the function to compute the memory size of this parametric derived 2531e6d9c06SDiana Picus /// type. The size of the object may depend on the LEN type parameters of the 2541e6d9c06SDiana Picus /// derived type. 2551e6d9c06SDiana Picus static mlir::LLVM::LLVMFuncOp 2561e6d9c06SDiana Picus getDependentTypeMemSizeFn(fir::RecordType recTy, fir::AllocaOp op, 2571e6d9c06SDiana Picus mlir::ConversionPatternRewriter &rewriter) { 2581e6d9c06SDiana Picus auto module = op->getParentOfType<mlir::ModuleOp>(); 2591e6d9c06SDiana Picus std::string name = recTy.getName().str() + "P.mem.size"; 2601e6d9c06SDiana Picus return module.lookupSymbol<mlir::LLVM::LLVMFuncOp>(name); 2611e6d9c06SDiana Picus } 2621e6d9c06SDiana Picus 2631e6d9c06SDiana Picus namespace { 2641e6d9c06SDiana Picus /// convert to LLVM IR dialect `alloca` 2651e6d9c06SDiana Picus struct AllocaOpConversion : public FIROpConversion<fir::AllocaOp> { 2661e6d9c06SDiana Picus using FIROpConversion::FIROpConversion; 2671e6d9c06SDiana Picus 2681e6d9c06SDiana Picus mlir::LogicalResult 2691e6d9c06SDiana Picus matchAndRewrite(fir::AllocaOp alloc, OpAdaptor adaptor, 2701e6d9c06SDiana Picus mlir::ConversionPatternRewriter &rewriter) const override { 2711e6d9c06SDiana Picus mlir::ValueRange operands = adaptor.getOperands(); 2721e6d9c06SDiana Picus auto loc = alloc.getLoc(); 2731e6d9c06SDiana Picus mlir::Type ity = lowerTy().indexType(); 2741e6d9c06SDiana Picus unsigned i = 0; 2751e6d9c06SDiana Picus mlir::Value size = genConstantIndex(loc, ity, rewriter, 1).getResult(); 2761e6d9c06SDiana Picus mlir::Type ty = convertType(alloc.getType()); 2771e6d9c06SDiana Picus mlir::Type resultTy = ty; 2781e6d9c06SDiana Picus if (alloc.hasLenParams()) { 2791e6d9c06SDiana Picus unsigned end = alloc.numLenParams(); 2801e6d9c06SDiana Picus llvm::SmallVector<mlir::Value> lenParams; 2811e6d9c06SDiana Picus for (; i < end; ++i) 2821e6d9c06SDiana Picus lenParams.push_back(operands[i]); 2831e6d9c06SDiana Picus mlir::Type scalarType = fir::unwrapSequenceType(alloc.getInType()); 2841e6d9c06SDiana Picus if (auto chrTy = scalarType.dyn_cast<fir::CharacterType>()) { 2851e6d9c06SDiana Picus fir::CharacterType rawCharTy = fir::CharacterType::getUnknownLen( 2861e6d9c06SDiana Picus chrTy.getContext(), chrTy.getFKind()); 2871e6d9c06SDiana Picus ty = mlir::LLVM::LLVMPointerType::get(convertType(rawCharTy)); 2881e6d9c06SDiana Picus assert(end == 1); 2891e6d9c06SDiana Picus size = integerCast(loc, rewriter, ity, lenParams[0]); 2901e6d9c06SDiana Picus } else if (auto recTy = scalarType.dyn_cast<fir::RecordType>()) { 2911e6d9c06SDiana Picus mlir::LLVM::LLVMFuncOp memSizeFn = 2921e6d9c06SDiana Picus getDependentTypeMemSizeFn(recTy, alloc, rewriter); 2931e6d9c06SDiana Picus if (!memSizeFn) 2941e6d9c06SDiana Picus emitError(loc, "did not find allocation function"); 2951e6d9c06SDiana Picus mlir::NamedAttribute attr = rewriter.getNamedAttr( 2961e6d9c06SDiana Picus "callee", mlir::SymbolRefAttr::get(memSizeFn)); 2971e6d9c06SDiana Picus auto call = rewriter.create<mlir::LLVM::CallOp>( 2981e6d9c06SDiana Picus loc, ity, lenParams, llvm::ArrayRef<mlir::NamedAttribute>{attr}); 2991e6d9c06SDiana Picus size = call.getResult(0); 3001e6d9c06SDiana Picus ty = mlir::LLVM::LLVMPointerType::get( 3011e6d9c06SDiana Picus mlir::IntegerType::get(alloc.getContext(), 8)); 3021e6d9c06SDiana Picus } else { 3031e6d9c06SDiana Picus return emitError(loc, "unexpected type ") 3041e6d9c06SDiana Picus << scalarType << " with type parameters"; 3051e6d9c06SDiana Picus } 3061e6d9c06SDiana Picus } 3071e6d9c06SDiana Picus if (alloc.hasShapeOperands()) { 3081e6d9c06SDiana Picus mlir::Type allocEleTy = fir::unwrapRefType(alloc.getType()); 3091e6d9c06SDiana Picus // Scale the size by constant factors encoded in the array type. 3101e6d9c06SDiana Picus if (auto seqTy = allocEleTy.dyn_cast<fir::SequenceType>()) { 3111e6d9c06SDiana Picus fir::SequenceType::Extent constSize = 1; 3121e6d9c06SDiana Picus for (auto extent : seqTy.getShape()) 3131e6d9c06SDiana Picus if (extent != fir::SequenceType::getUnknownExtent()) 3141e6d9c06SDiana Picus constSize *= extent; 3151e6d9c06SDiana Picus mlir::Value constVal{ 3161e6d9c06SDiana Picus genConstantIndex(loc, ity, rewriter, constSize).getResult()}; 3171e6d9c06SDiana Picus size = rewriter.create<mlir::LLVM::MulOp>(loc, ity, size, constVal); 3181e6d9c06SDiana Picus } 3191e6d9c06SDiana Picus unsigned end = operands.size(); 3201e6d9c06SDiana Picus for (; i < end; ++i) 3211e6d9c06SDiana Picus size = rewriter.create<mlir::LLVM::MulOp>( 3221e6d9c06SDiana Picus loc, ity, size, integerCast(loc, rewriter, ity, operands[i])); 3231e6d9c06SDiana Picus } 3241e6d9c06SDiana Picus if (ty == resultTy) { 3251e6d9c06SDiana Picus // Do not emit the bitcast if ty and resultTy are the same. 3261e6d9c06SDiana Picus rewriter.replaceOpWithNewOp<mlir::LLVM::AllocaOp>(alloc, ty, size, 3271e6d9c06SDiana Picus alloc->getAttrs()); 3281e6d9c06SDiana Picus } else { 3291e6d9c06SDiana Picus auto al = rewriter.create<mlir::LLVM::AllocaOp>(loc, ty, size, 3301e6d9c06SDiana Picus alloc->getAttrs()); 3311e6d9c06SDiana Picus rewriter.replaceOpWithNewOp<mlir::LLVM::BitcastOp>(alloc, resultTy, al); 3321e6d9c06SDiana Picus } 3331e6d9c06SDiana Picus return success(); 3341e6d9c06SDiana Picus } 3351e6d9c06SDiana Picus }; 336044d5b5dSValentin Clement 337df3b9810SValentin Clement /// Lower `fir.box_addr` to the sequence of operations to extract the first 338df3b9810SValentin Clement /// element of the box. 339df3b9810SValentin Clement struct BoxAddrOpConversion : public FIROpConversion<fir::BoxAddrOp> { 340df3b9810SValentin Clement using FIROpConversion::FIROpConversion; 341df3b9810SValentin Clement 342df3b9810SValentin Clement mlir::LogicalResult 343df3b9810SValentin Clement matchAndRewrite(fir::BoxAddrOp boxaddr, OpAdaptor adaptor, 344df3b9810SValentin Clement mlir::ConversionPatternRewriter &rewriter) const override { 345df3b9810SValentin Clement mlir::Value a = adaptor.getOperands()[0]; 346df3b9810SValentin Clement auto loc = boxaddr.getLoc(); 347df3b9810SValentin Clement mlir::Type ty = convertType(boxaddr.getType()); 348df3b9810SValentin Clement if (auto argty = boxaddr.val().getType().dyn_cast<fir::BoxType>()) { 349df3b9810SValentin Clement rewriter.replaceOp(boxaddr, loadBaseAddrFromBox(loc, ty, a, rewriter)); 350df3b9810SValentin Clement } else { 351df3b9810SValentin Clement auto c0attr = rewriter.getI32IntegerAttr(0); 352df3b9810SValentin Clement auto c0 = mlir::ArrayAttr::get(boxaddr.getContext(), c0attr); 353df3b9810SValentin Clement rewriter.replaceOpWithNewOp<mlir::LLVM::ExtractValueOp>(boxaddr, ty, a, 354df3b9810SValentin Clement c0); 355df3b9810SValentin Clement } 356df3b9810SValentin Clement return success(); 357df3b9810SValentin Clement } 358df3b9810SValentin Clement }; 359df3b9810SValentin Clement 360df3b9810SValentin Clement /// Lower `fir.box_dims` to a sequence of operations to extract the requested 361df3b9810SValentin Clement /// dimension infomartion from the boxed value. 362df3b9810SValentin Clement /// Result in a triple set of GEPs and loads. 363df3b9810SValentin Clement struct BoxDimsOpConversion : public FIROpConversion<fir::BoxDimsOp> { 364df3b9810SValentin Clement using FIROpConversion::FIROpConversion; 365df3b9810SValentin Clement 366df3b9810SValentin Clement mlir::LogicalResult 367df3b9810SValentin Clement matchAndRewrite(fir::BoxDimsOp boxdims, OpAdaptor adaptor, 368df3b9810SValentin Clement mlir::ConversionPatternRewriter &rewriter) const override { 369df3b9810SValentin Clement SmallVector<mlir::Type, 3> resultTypes = { 370df3b9810SValentin Clement convertType(boxdims.getResult(0).getType()), 371df3b9810SValentin Clement convertType(boxdims.getResult(1).getType()), 372df3b9810SValentin Clement convertType(boxdims.getResult(2).getType()), 373df3b9810SValentin Clement }; 374df3b9810SValentin Clement auto results = 375df3b9810SValentin Clement getDimsFromBox(boxdims.getLoc(), resultTypes, adaptor.getOperands()[0], 376df3b9810SValentin Clement adaptor.getOperands()[1], rewriter); 377df3b9810SValentin Clement rewriter.replaceOp(boxdims, results); 378df3b9810SValentin Clement return success(); 379df3b9810SValentin Clement } 380df3b9810SValentin Clement }; 381df3b9810SValentin Clement 382df3b9810SValentin Clement /// Lower `fir.box_elesize` to a sequence of operations ro extract the size of 383df3b9810SValentin Clement /// an element in the boxed value. 384df3b9810SValentin Clement struct BoxEleSizeOpConversion : public FIROpConversion<fir::BoxEleSizeOp> { 385df3b9810SValentin Clement using FIROpConversion::FIROpConversion; 386df3b9810SValentin Clement 387df3b9810SValentin Clement mlir::LogicalResult 388df3b9810SValentin Clement matchAndRewrite(fir::BoxEleSizeOp boxelesz, OpAdaptor adaptor, 389df3b9810SValentin Clement mlir::ConversionPatternRewriter &rewriter) const override { 390df3b9810SValentin Clement mlir::Value a = adaptor.getOperands()[0]; 391df3b9810SValentin Clement auto loc = boxelesz.getLoc(); 392df3b9810SValentin Clement auto ty = convertType(boxelesz.getType()); 393b6e44ecdSValentin Clement auto elemSize = getValueFromBox(loc, a, ty, rewriter, kElemLenPosInBox); 394b6e44ecdSValentin Clement rewriter.replaceOp(boxelesz, elemSize); 395b6e44ecdSValentin Clement return success(); 396b6e44ecdSValentin Clement } 397b6e44ecdSValentin Clement }; 398b6e44ecdSValentin Clement 399b6e44ecdSValentin Clement /// Lower `fir.box_isalloc` to a sequence of operations to determine if the 400b6e44ecdSValentin Clement /// boxed value was from an ALLOCATABLE entity. 401b6e44ecdSValentin Clement struct BoxIsAllocOpConversion : public FIROpConversion<fir::BoxIsAllocOp> { 402b6e44ecdSValentin Clement using FIROpConversion::FIROpConversion; 403b6e44ecdSValentin Clement 404b6e44ecdSValentin Clement mlir::LogicalResult 405b6e44ecdSValentin Clement matchAndRewrite(fir::BoxIsAllocOp boxisalloc, OpAdaptor adaptor, 406b6e44ecdSValentin Clement mlir::ConversionPatternRewriter &rewriter) const override { 407b6e44ecdSValentin Clement mlir::Value box = adaptor.getOperands()[0]; 408b6e44ecdSValentin Clement auto loc = boxisalloc.getLoc(); 409b6e44ecdSValentin Clement mlir::Value check = 410b6e44ecdSValentin Clement genBoxAttributeCheck(loc, box, rewriter, kAttrAllocatable); 411b6e44ecdSValentin Clement rewriter.replaceOp(boxisalloc, check); 412b6e44ecdSValentin Clement return success(); 413b6e44ecdSValentin Clement } 414b6e44ecdSValentin Clement }; 415b6e44ecdSValentin Clement 416b6e44ecdSValentin Clement /// Lower `fir.box_isarray` to a sequence of operations to determine if the 417b6e44ecdSValentin Clement /// boxed is an array. 418b6e44ecdSValentin Clement struct BoxIsArrayOpConversion : public FIROpConversion<fir::BoxIsArrayOp> { 419b6e44ecdSValentin Clement using FIROpConversion::FIROpConversion; 420b6e44ecdSValentin Clement 421b6e44ecdSValentin Clement mlir::LogicalResult 422b6e44ecdSValentin Clement matchAndRewrite(fir::BoxIsArrayOp boxisarray, OpAdaptor adaptor, 423b6e44ecdSValentin Clement mlir::ConversionPatternRewriter &rewriter) const override { 424b6e44ecdSValentin Clement mlir::Value a = adaptor.getOperands()[0]; 425b6e44ecdSValentin Clement auto loc = boxisarray.getLoc(); 426b6e44ecdSValentin Clement auto rank = 427b6e44ecdSValentin Clement getValueFromBox(loc, a, rewriter.getI32Type(), rewriter, kRankPosInBox); 428b6e44ecdSValentin Clement auto c0 = genConstantOffset(loc, rewriter, 0); 429b6e44ecdSValentin Clement rewriter.replaceOpWithNewOp<mlir::LLVM::ICmpOp>( 430b6e44ecdSValentin Clement boxisarray, mlir::LLVM::ICmpPredicate::ne, rank, c0); 431b6e44ecdSValentin Clement return success(); 432b6e44ecdSValentin Clement } 433b6e44ecdSValentin Clement }; 434b6e44ecdSValentin Clement 435b6e44ecdSValentin Clement /// Lower `fir.box_isptr` to a sequence of operations to determined if the 436b6e44ecdSValentin Clement /// boxed value was from a POINTER entity. 437b6e44ecdSValentin Clement struct BoxIsPtrOpConversion : public FIROpConversion<fir::BoxIsPtrOp> { 438b6e44ecdSValentin Clement using FIROpConversion::FIROpConversion; 439b6e44ecdSValentin Clement 440b6e44ecdSValentin Clement mlir::LogicalResult 441b6e44ecdSValentin Clement matchAndRewrite(fir::BoxIsPtrOp boxisptr, OpAdaptor adaptor, 442b6e44ecdSValentin Clement mlir::ConversionPatternRewriter &rewriter) const override { 443b6e44ecdSValentin Clement mlir::Value box = adaptor.getOperands()[0]; 444b6e44ecdSValentin Clement auto loc = boxisptr.getLoc(); 445b6e44ecdSValentin Clement mlir::Value check = genBoxAttributeCheck(loc, box, rewriter, kAttrPointer); 446b6e44ecdSValentin Clement rewriter.replaceOp(boxisptr, check); 447df3b9810SValentin Clement return success(); 448df3b9810SValentin Clement } 449df3b9810SValentin Clement }; 450df3b9810SValentin Clement 451df3b9810SValentin Clement /// Lower `fir.box_rank` to the sequence of operation to extract the rank from 452df3b9810SValentin Clement /// the box. 453df3b9810SValentin Clement struct BoxRankOpConversion : public FIROpConversion<fir::BoxRankOp> { 454df3b9810SValentin Clement using FIROpConversion::FIROpConversion; 455df3b9810SValentin Clement 456df3b9810SValentin Clement mlir::LogicalResult 457df3b9810SValentin Clement matchAndRewrite(fir::BoxRankOp boxrank, OpAdaptor adaptor, 458df3b9810SValentin Clement mlir::ConversionPatternRewriter &rewriter) const override { 459df3b9810SValentin Clement mlir::Value a = adaptor.getOperands()[0]; 460df3b9810SValentin Clement auto loc = boxrank.getLoc(); 461df3b9810SValentin Clement mlir::Type ty = convertType(boxrank.getType()); 462b6e44ecdSValentin Clement auto result = getValueFromBox(loc, a, ty, rewriter, kRankPosInBox); 463df3b9810SValentin Clement rewriter.replaceOp(boxrank, result); 464df3b9810SValentin Clement return success(); 465df3b9810SValentin Clement } 466df3b9810SValentin Clement }; 467df3b9810SValentin Clement 468ddd11b9aSAndrzej Warzynski // `fir.call` -> `llvm.call` 469ddd11b9aSAndrzej Warzynski struct CallOpConversion : public FIROpConversion<fir::CallOp> { 470ddd11b9aSAndrzej Warzynski using FIROpConversion::FIROpConversion; 471ddd11b9aSAndrzej Warzynski 472ddd11b9aSAndrzej Warzynski mlir::LogicalResult 473ddd11b9aSAndrzej Warzynski matchAndRewrite(fir::CallOp call, OpAdaptor adaptor, 474ddd11b9aSAndrzej Warzynski mlir::ConversionPatternRewriter &rewriter) const override { 475ddd11b9aSAndrzej Warzynski SmallVector<mlir::Type> resultTys; 476ddd11b9aSAndrzej Warzynski for (auto r : call.getResults()) 477ddd11b9aSAndrzej Warzynski resultTys.push_back(convertType(r.getType())); 478ddd11b9aSAndrzej Warzynski rewriter.replaceOpWithNewOp<mlir::LLVM::CallOp>( 479ddd11b9aSAndrzej Warzynski call, resultTys, adaptor.getOperands(), call->getAttrs()); 480ddd11b9aSAndrzej Warzynski return success(); 481ddd11b9aSAndrzej Warzynski } 482ddd11b9aSAndrzej Warzynski }; 483ddd11b9aSAndrzej Warzynski 484092cee5fSValentin Clement static mlir::Type getComplexEleTy(mlir::Type complex) { 485092cee5fSValentin Clement if (auto cc = complex.dyn_cast<mlir::ComplexType>()) 486092cee5fSValentin Clement return cc.getElementType(); 487092cee5fSValentin Clement return complex.cast<fir::ComplexType>().getElementType(); 488092cee5fSValentin Clement } 489092cee5fSValentin Clement 490f1dfc027SDiana Picus /// Compare complex values 491f1dfc027SDiana Picus /// 492f1dfc027SDiana Picus /// Per 10.1, the only comparisons available are .EQ. (oeq) and .NE. (une). 493f1dfc027SDiana Picus /// 494f1dfc027SDiana Picus /// For completeness, all other comparison are done on the real component only. 495f1dfc027SDiana Picus struct CmpcOpConversion : public FIROpConversion<fir::CmpcOp> { 496f1dfc027SDiana Picus using FIROpConversion::FIROpConversion; 497f1dfc027SDiana Picus 498f1dfc027SDiana Picus mlir::LogicalResult 499f1dfc027SDiana Picus matchAndRewrite(fir::CmpcOp cmp, OpAdaptor adaptor, 500f1dfc027SDiana Picus mlir::ConversionPatternRewriter &rewriter) const override { 501f1dfc027SDiana Picus mlir::ValueRange operands = adaptor.getOperands(); 502f1dfc027SDiana Picus mlir::MLIRContext *ctxt = cmp.getContext(); 503f1dfc027SDiana Picus mlir::Type eleTy = convertType(getComplexEleTy(cmp.lhs().getType())); 504f1dfc027SDiana Picus mlir::Type resTy = convertType(cmp.getType()); 505f1dfc027SDiana Picus mlir::Location loc = cmp.getLoc(); 506f1dfc027SDiana Picus auto pos0 = mlir::ArrayAttr::get(ctxt, rewriter.getI32IntegerAttr(0)); 507f1dfc027SDiana Picus SmallVector<mlir::Value, 2> rp{rewriter.create<mlir::LLVM::ExtractValueOp>( 508f1dfc027SDiana Picus loc, eleTy, operands[0], pos0), 509f1dfc027SDiana Picus rewriter.create<mlir::LLVM::ExtractValueOp>( 510f1dfc027SDiana Picus loc, eleTy, operands[1], pos0)}; 511f1dfc027SDiana Picus auto rcp = 512f1dfc027SDiana Picus rewriter.create<mlir::LLVM::FCmpOp>(loc, resTy, rp, cmp->getAttrs()); 513f1dfc027SDiana Picus auto pos1 = mlir::ArrayAttr::get(ctxt, rewriter.getI32IntegerAttr(1)); 514f1dfc027SDiana Picus SmallVector<mlir::Value, 2> ip{rewriter.create<mlir::LLVM::ExtractValueOp>( 515f1dfc027SDiana Picus loc, eleTy, operands[0], pos1), 516f1dfc027SDiana Picus rewriter.create<mlir::LLVM::ExtractValueOp>( 517f1dfc027SDiana Picus loc, eleTy, operands[1], pos1)}; 518f1dfc027SDiana Picus auto icp = 519f1dfc027SDiana Picus rewriter.create<mlir::LLVM::FCmpOp>(loc, resTy, ip, cmp->getAttrs()); 520f1dfc027SDiana Picus SmallVector<mlir::Value, 2> cp{rcp, icp}; 521f1dfc027SDiana Picus switch (cmp.getPredicate()) { 522f1dfc027SDiana Picus case mlir::arith::CmpFPredicate::OEQ: // .EQ. 523f1dfc027SDiana Picus rewriter.replaceOpWithNewOp<mlir::LLVM::AndOp>(cmp, resTy, cp); 524f1dfc027SDiana Picus break; 525f1dfc027SDiana Picus case mlir::arith::CmpFPredicate::UNE: // .NE. 526f1dfc027SDiana Picus rewriter.replaceOpWithNewOp<mlir::LLVM::OrOp>(cmp, resTy, cp); 527f1dfc027SDiana Picus break; 528f1dfc027SDiana Picus default: 529f1dfc027SDiana Picus rewriter.replaceOp(cmp, rcp.getResult()); 530f1dfc027SDiana Picus break; 531f1dfc027SDiana Picus } 532f1dfc027SDiana Picus return success(); 533f1dfc027SDiana Picus } 534f1dfc027SDiana Picus }; 535f1dfc027SDiana Picus 536092cee5fSValentin Clement /// convert value of from-type to value of to-type 537092cee5fSValentin Clement struct ConvertOpConversion : public FIROpConversion<fir::ConvertOp> { 538092cee5fSValentin Clement using FIROpConversion::FIROpConversion; 539092cee5fSValentin Clement 540092cee5fSValentin Clement static bool isFloatingPointTy(mlir::Type ty) { 541092cee5fSValentin Clement return ty.isa<mlir::FloatType>(); 542092cee5fSValentin Clement } 543092cee5fSValentin Clement 544092cee5fSValentin Clement mlir::LogicalResult 545092cee5fSValentin Clement matchAndRewrite(fir::ConvertOp convert, OpAdaptor adaptor, 546092cee5fSValentin Clement mlir::ConversionPatternRewriter &rewriter) const override { 547092cee5fSValentin Clement auto fromTy = convertType(convert.value().getType()); 548092cee5fSValentin Clement auto toTy = convertType(convert.res().getType()); 549092cee5fSValentin Clement mlir::Value op0 = adaptor.getOperands()[0]; 550092cee5fSValentin Clement if (fromTy == toTy) { 551092cee5fSValentin Clement rewriter.replaceOp(convert, op0); 552092cee5fSValentin Clement return success(); 553092cee5fSValentin Clement } 554092cee5fSValentin Clement auto loc = convert.getLoc(); 555092cee5fSValentin Clement auto convertFpToFp = [&](mlir::Value val, unsigned fromBits, 556092cee5fSValentin Clement unsigned toBits, mlir::Type toTy) -> mlir::Value { 557092cee5fSValentin Clement if (fromBits == toBits) { 558092cee5fSValentin Clement // TODO: Converting between two floating-point representations with the 559092cee5fSValentin Clement // same bitwidth is not allowed for now. 560092cee5fSValentin Clement mlir::emitError(loc, 561092cee5fSValentin Clement "cannot implicitly convert between two floating-point " 562092cee5fSValentin Clement "representations of the same bitwidth"); 563092cee5fSValentin Clement return {}; 564092cee5fSValentin Clement } 565092cee5fSValentin Clement if (fromBits > toBits) 566092cee5fSValentin Clement return rewriter.create<mlir::LLVM::FPTruncOp>(loc, toTy, val); 567092cee5fSValentin Clement return rewriter.create<mlir::LLVM::FPExtOp>(loc, toTy, val); 568092cee5fSValentin Clement }; 569092cee5fSValentin Clement // Complex to complex conversion. 570092cee5fSValentin Clement if (fir::isa_complex(convert.value().getType()) && 571092cee5fSValentin Clement fir::isa_complex(convert.res().getType())) { 572092cee5fSValentin Clement // Special case: handle the conversion of a complex such that both the 573092cee5fSValentin Clement // real and imaginary parts are converted together. 574092cee5fSValentin Clement auto zero = mlir::ArrayAttr::get(convert.getContext(), 575092cee5fSValentin Clement rewriter.getI32IntegerAttr(0)); 576092cee5fSValentin Clement auto one = mlir::ArrayAttr::get(convert.getContext(), 577092cee5fSValentin Clement rewriter.getI32IntegerAttr(1)); 578092cee5fSValentin Clement auto ty = convertType(getComplexEleTy(convert.value().getType())); 579092cee5fSValentin Clement auto rp = rewriter.create<mlir::LLVM::ExtractValueOp>(loc, ty, op0, zero); 580092cee5fSValentin Clement auto ip = rewriter.create<mlir::LLVM::ExtractValueOp>(loc, ty, op0, one); 581092cee5fSValentin Clement auto nt = convertType(getComplexEleTy(convert.res().getType())); 582092cee5fSValentin Clement auto fromBits = mlir::LLVM::getPrimitiveTypeSizeInBits(ty); 583092cee5fSValentin Clement auto toBits = mlir::LLVM::getPrimitiveTypeSizeInBits(nt); 584092cee5fSValentin Clement auto rc = convertFpToFp(rp, fromBits, toBits, nt); 585092cee5fSValentin Clement auto ic = convertFpToFp(ip, fromBits, toBits, nt); 586092cee5fSValentin Clement auto un = rewriter.create<mlir::LLVM::UndefOp>(loc, toTy); 587092cee5fSValentin Clement auto i1 = 588092cee5fSValentin Clement rewriter.create<mlir::LLVM::InsertValueOp>(loc, toTy, un, rc, zero); 589092cee5fSValentin Clement rewriter.replaceOpWithNewOp<mlir::LLVM::InsertValueOp>(convert, toTy, i1, 590092cee5fSValentin Clement ic, one); 591092cee5fSValentin Clement return mlir::success(); 592092cee5fSValentin Clement } 593092cee5fSValentin Clement // Floating point to floating point conversion. 594092cee5fSValentin Clement if (isFloatingPointTy(fromTy)) { 595092cee5fSValentin Clement if (isFloatingPointTy(toTy)) { 596092cee5fSValentin Clement auto fromBits = mlir::LLVM::getPrimitiveTypeSizeInBits(fromTy); 597092cee5fSValentin Clement auto toBits = mlir::LLVM::getPrimitiveTypeSizeInBits(toTy); 598092cee5fSValentin Clement auto v = convertFpToFp(op0, fromBits, toBits, toTy); 599092cee5fSValentin Clement rewriter.replaceOp(convert, v); 600092cee5fSValentin Clement return mlir::success(); 601092cee5fSValentin Clement } 602092cee5fSValentin Clement if (toTy.isa<mlir::IntegerType>()) { 603092cee5fSValentin Clement rewriter.replaceOpWithNewOp<mlir::LLVM::FPToSIOp>(convert, toTy, op0); 604092cee5fSValentin Clement return mlir::success(); 605092cee5fSValentin Clement } 606092cee5fSValentin Clement } else if (fromTy.isa<mlir::IntegerType>()) { 607092cee5fSValentin Clement // Integer to integer conversion. 608092cee5fSValentin Clement if (toTy.isa<mlir::IntegerType>()) { 609092cee5fSValentin Clement auto fromBits = mlir::LLVM::getPrimitiveTypeSizeInBits(fromTy); 610092cee5fSValentin Clement auto toBits = mlir::LLVM::getPrimitiveTypeSizeInBits(toTy); 611092cee5fSValentin Clement assert(fromBits != toBits); 612092cee5fSValentin Clement if (fromBits > toBits) { 613092cee5fSValentin Clement rewriter.replaceOpWithNewOp<mlir::LLVM::TruncOp>(convert, toTy, op0); 614092cee5fSValentin Clement return mlir::success(); 615092cee5fSValentin Clement } 616092cee5fSValentin Clement rewriter.replaceOpWithNewOp<mlir::LLVM::SExtOp>(convert, toTy, op0); 617092cee5fSValentin Clement return mlir::success(); 618092cee5fSValentin Clement } 619092cee5fSValentin Clement // Integer to floating point conversion. 620092cee5fSValentin Clement if (isFloatingPointTy(toTy)) { 621092cee5fSValentin Clement rewriter.replaceOpWithNewOp<mlir::LLVM::SIToFPOp>(convert, toTy, op0); 622092cee5fSValentin Clement return mlir::success(); 623092cee5fSValentin Clement } 624092cee5fSValentin Clement // Integer to pointer conversion. 625092cee5fSValentin Clement if (toTy.isa<mlir::LLVM::LLVMPointerType>()) { 626092cee5fSValentin Clement rewriter.replaceOpWithNewOp<mlir::LLVM::IntToPtrOp>(convert, toTy, op0); 627092cee5fSValentin Clement return mlir::success(); 628092cee5fSValentin Clement } 629092cee5fSValentin Clement } else if (fromTy.isa<mlir::LLVM::LLVMPointerType>()) { 630092cee5fSValentin Clement // Pointer to integer conversion. 631092cee5fSValentin Clement if (toTy.isa<mlir::IntegerType>()) { 632092cee5fSValentin Clement rewriter.replaceOpWithNewOp<mlir::LLVM::PtrToIntOp>(convert, toTy, op0); 633092cee5fSValentin Clement return mlir::success(); 634092cee5fSValentin Clement } 635092cee5fSValentin Clement // Pointer to pointer conversion. 636092cee5fSValentin Clement if (toTy.isa<mlir::LLVM::LLVMPointerType>()) { 637092cee5fSValentin Clement rewriter.replaceOpWithNewOp<mlir::LLVM::BitcastOp>(convert, toTy, op0); 638092cee5fSValentin Clement return mlir::success(); 639092cee5fSValentin Clement } 640092cee5fSValentin Clement } 641092cee5fSValentin Clement return emitError(loc) << "cannot convert " << fromTy << " to " << toTy; 642092cee5fSValentin Clement } 643092cee5fSValentin Clement }; 644092cee5fSValentin Clement 6459534e361SValentin Clement /// Lower `fir.dispatch` operation. A virtual call to a method in a dispatch 6469534e361SValentin Clement /// table. 6479534e361SValentin Clement struct DispatchOpConversion : public FIROpConversion<fir::DispatchOp> { 6489534e361SValentin Clement using FIROpConversion::FIROpConversion; 6499534e361SValentin Clement 6509534e361SValentin Clement mlir::LogicalResult 6519534e361SValentin Clement matchAndRewrite(fir::DispatchOp dispatch, OpAdaptor adaptor, 6529534e361SValentin Clement mlir::ConversionPatternRewriter &rewriter) const override { 6539534e361SValentin Clement return rewriter.notifyMatchFailure( 6549534e361SValentin Clement dispatch, "fir.dispatch codegen is not implemented yet"); 6559534e361SValentin Clement } 6569534e361SValentin Clement }; 6579534e361SValentin Clement 6589534e361SValentin Clement /// Lower `fir.dispatch_table` operation. The dispatch table for a Fortran 6599534e361SValentin Clement /// derived type. 6609534e361SValentin Clement struct DispatchTableOpConversion 6619534e361SValentin Clement : public FIROpConversion<fir::DispatchTableOp> { 6629534e361SValentin Clement using FIROpConversion::FIROpConversion; 6639534e361SValentin Clement 6649534e361SValentin Clement mlir::LogicalResult 6659534e361SValentin Clement matchAndRewrite(fir::DispatchTableOp dispTab, OpAdaptor adaptor, 6669534e361SValentin Clement mlir::ConversionPatternRewriter &rewriter) const override { 6679534e361SValentin Clement return rewriter.notifyMatchFailure( 6689534e361SValentin Clement dispTab, "fir.dispatch_table codegen is not implemented yet"); 6699534e361SValentin Clement } 6709534e361SValentin Clement }; 6719534e361SValentin Clement 6729534e361SValentin Clement /// Lower `fir.dt_entry` operation. An entry in a dispatch table; binds a 6739534e361SValentin Clement /// method-name to a function. 6749534e361SValentin Clement struct DTEntryOpConversion : public FIROpConversion<fir::DTEntryOp> { 6759534e361SValentin Clement using FIROpConversion::FIROpConversion; 6769534e361SValentin Clement 6779534e361SValentin Clement mlir::LogicalResult 6789534e361SValentin Clement matchAndRewrite(fir::DTEntryOp dtEnt, OpAdaptor adaptor, 6799534e361SValentin Clement mlir::ConversionPatternRewriter &rewriter) const override { 6809534e361SValentin Clement return rewriter.notifyMatchFailure( 6819534e361SValentin Clement dtEnt, "fir.dt_entry codegen is not implemented yet"); 6829534e361SValentin Clement } 6839534e361SValentin Clement }; 6849534e361SValentin Clement 685677df8c7SValentin Clement /// Lower `fir.global_len` operation. 686677df8c7SValentin Clement struct GlobalLenOpConversion : public FIROpConversion<fir::GlobalLenOp> { 687677df8c7SValentin Clement using FIROpConversion::FIROpConversion; 688677df8c7SValentin Clement 689677df8c7SValentin Clement mlir::LogicalResult 690677df8c7SValentin Clement matchAndRewrite(fir::GlobalLenOp globalLen, OpAdaptor adaptor, 691677df8c7SValentin Clement mlir::ConversionPatternRewriter &rewriter) const override { 692677df8c7SValentin Clement return rewriter.notifyMatchFailure( 693677df8c7SValentin Clement globalLen, "fir.global_len codegen is not implemented yet"); 694677df8c7SValentin Clement } 695677df8c7SValentin Clement }; 696677df8c7SValentin Clement 69731246187SValentin Clement /// Lower `fir.gentypedesc` to a global constant. 69831246187SValentin Clement struct GenTypeDescOpConversion : public FIROpConversion<fir::GenTypeDescOp> { 69931246187SValentin Clement using FIROpConversion::FIROpConversion; 70031246187SValentin Clement 70131246187SValentin Clement mlir::LogicalResult 70231246187SValentin Clement matchAndRewrite(fir::GenTypeDescOp gentypedesc, OpAdaptor adaptor, 70331246187SValentin Clement mlir::ConversionPatternRewriter &rewriter) const override { 70431246187SValentin Clement return rewriter.notifyMatchFailure( 70531246187SValentin Clement gentypedesc, "fir.fir.gentypedesc codegen is not implemented yet"); 70631246187SValentin Clement } 70731246187SValentin Clement }; 70831246187SValentin Clement 7090c4a7a52SValentin Clement /// Lower `fir.has_value` operation to `llvm.return` operation. 710044d5b5dSValentin Clement struct HasValueOpConversion : public FIROpConversion<fir::HasValueOp> { 711044d5b5dSValentin Clement using FIROpConversion::FIROpConversion; 712044d5b5dSValentin Clement 713044d5b5dSValentin Clement mlir::LogicalResult 714044d5b5dSValentin Clement matchAndRewrite(fir::HasValueOp op, OpAdaptor adaptor, 715044d5b5dSValentin Clement mlir::ConversionPatternRewriter &rewriter) const override { 716044d5b5dSValentin Clement rewriter.replaceOpWithNewOp<LLVM::ReturnOp>(op, adaptor.getOperands()); 717044d5b5dSValentin Clement return success(); 718044d5b5dSValentin Clement } 719044d5b5dSValentin Clement }; 720044d5b5dSValentin Clement 7210c4a7a52SValentin Clement /// Lower `fir.global` operation to `llvm.global` operation. 7220c4a7a52SValentin Clement /// `fir.insert_on_range` operations are replaced with constant dense attribute 7230c4a7a52SValentin Clement /// if they are applied on the full range. 724044d5b5dSValentin Clement struct GlobalOpConversion : public FIROpConversion<fir::GlobalOp> { 725044d5b5dSValentin Clement using FIROpConversion::FIROpConversion; 726044d5b5dSValentin Clement 727044d5b5dSValentin Clement mlir::LogicalResult 728044d5b5dSValentin Clement matchAndRewrite(fir::GlobalOp global, OpAdaptor adaptor, 729044d5b5dSValentin Clement mlir::ConversionPatternRewriter &rewriter) const override { 730044d5b5dSValentin Clement auto tyAttr = convertType(global.getType()); 731044d5b5dSValentin Clement if (global.getType().isa<fir::BoxType>()) 732044d5b5dSValentin Clement tyAttr = tyAttr.cast<mlir::LLVM::LLVMPointerType>().getElementType(); 733044d5b5dSValentin Clement auto loc = global.getLoc(); 734044d5b5dSValentin Clement mlir::Attribute initAttr{}; 735044d5b5dSValentin Clement if (global.initVal()) 736044d5b5dSValentin Clement initAttr = global.initVal().getValue(); 737044d5b5dSValentin Clement auto linkage = convertLinkage(global.linkName()); 738044d5b5dSValentin Clement auto isConst = global.constant().hasValue(); 739044d5b5dSValentin Clement auto g = rewriter.create<mlir::LLVM::GlobalOp>( 740044d5b5dSValentin Clement loc, tyAttr, isConst, linkage, global.sym_name(), initAttr); 741044d5b5dSValentin Clement auto &gr = g.getInitializerRegion(); 742044d5b5dSValentin Clement rewriter.inlineRegionBefore(global.region(), gr, gr.end()); 743044d5b5dSValentin Clement if (!gr.empty()) { 744044d5b5dSValentin Clement // Replace insert_on_range with a constant dense attribute if the 745044d5b5dSValentin Clement // initialization is on the full range. 746044d5b5dSValentin Clement auto insertOnRangeOps = gr.front().getOps<fir::InsertOnRangeOp>(); 747044d5b5dSValentin Clement for (auto insertOp : insertOnRangeOps) { 748044d5b5dSValentin Clement if (isFullRange(insertOp.coor(), insertOp.getType())) { 749044d5b5dSValentin Clement auto seqTyAttr = convertType(insertOp.getType()); 750044d5b5dSValentin Clement auto *op = insertOp.val().getDefiningOp(); 751044d5b5dSValentin Clement auto constant = mlir::dyn_cast<mlir::arith::ConstantOp>(op); 752044d5b5dSValentin Clement if (!constant) { 753044d5b5dSValentin Clement auto convertOp = mlir::dyn_cast<fir::ConvertOp>(op); 754044d5b5dSValentin Clement if (!convertOp) 755044d5b5dSValentin Clement continue; 756044d5b5dSValentin Clement constant = cast<mlir::arith::ConstantOp>( 757044d5b5dSValentin Clement convertOp.value().getDefiningOp()); 758044d5b5dSValentin Clement } 759044d5b5dSValentin Clement mlir::Type vecType = mlir::VectorType::get( 760044d5b5dSValentin Clement insertOp.getType().getShape(), constant.getType()); 761044d5b5dSValentin Clement auto denseAttr = mlir::DenseElementsAttr::get( 762044d5b5dSValentin Clement vecType.cast<ShapedType>(), constant.value()); 763044d5b5dSValentin Clement rewriter.setInsertionPointAfter(insertOp); 764044d5b5dSValentin Clement rewriter.replaceOpWithNewOp<mlir::arith::ConstantOp>( 765044d5b5dSValentin Clement insertOp, seqTyAttr, denseAttr); 766044d5b5dSValentin Clement } 767044d5b5dSValentin Clement } 768044d5b5dSValentin Clement } 769044d5b5dSValentin Clement rewriter.eraseOp(global); 770044d5b5dSValentin Clement return success(); 771044d5b5dSValentin Clement } 772044d5b5dSValentin Clement 773044d5b5dSValentin Clement bool isFullRange(mlir::ArrayAttr indexes, fir::SequenceType seqTy) const { 774044d5b5dSValentin Clement auto extents = seqTy.getShape(); 775044d5b5dSValentin Clement if (indexes.size() / 2 != extents.size()) 776044d5b5dSValentin Clement return false; 777044d5b5dSValentin Clement for (unsigned i = 0; i < indexes.size(); i += 2) { 778044d5b5dSValentin Clement if (indexes[i].cast<IntegerAttr>().getInt() != 0) 779044d5b5dSValentin Clement return false; 780044d5b5dSValentin Clement if (indexes[i + 1].cast<IntegerAttr>().getInt() != extents[i / 2] - 1) 781044d5b5dSValentin Clement return false; 782044d5b5dSValentin Clement } 783044d5b5dSValentin Clement return true; 784044d5b5dSValentin Clement } 785044d5b5dSValentin Clement 7860c4a7a52SValentin Clement // TODO: String comparaison should be avoided. Replace linkName with an 7870c4a7a52SValentin Clement // enumeration. 788044d5b5dSValentin Clement mlir::LLVM::Linkage convertLinkage(Optional<StringRef> optLinkage) const { 789044d5b5dSValentin Clement if (optLinkage.hasValue()) { 790044d5b5dSValentin Clement auto name = optLinkage.getValue(); 791044d5b5dSValentin Clement if (name == "internal") 792044d5b5dSValentin Clement return mlir::LLVM::Linkage::Internal; 793044d5b5dSValentin Clement if (name == "linkonce") 794044d5b5dSValentin Clement return mlir::LLVM::Linkage::Linkonce; 795044d5b5dSValentin Clement if (name == "common") 796044d5b5dSValentin Clement return mlir::LLVM::Linkage::Common; 797044d5b5dSValentin Clement if (name == "weak") 798044d5b5dSValentin Clement return mlir::LLVM::Linkage::Weak; 799044d5b5dSValentin Clement } 800044d5b5dSValentin Clement return mlir::LLVM::Linkage::External; 801044d5b5dSValentin Clement } 802044d5b5dSValentin Clement }; 803044d5b5dSValentin Clement 80439f4ef81SValentin Clement void genCondBrOp(mlir::Location loc, mlir::Value cmp, mlir::Block *dest, 80539f4ef81SValentin Clement Optional<mlir::ValueRange> destOps, 80639f4ef81SValentin Clement mlir::ConversionPatternRewriter &rewriter, 80739f4ef81SValentin Clement mlir::Block *newBlock) { 80839f4ef81SValentin Clement if (destOps.hasValue()) 80939f4ef81SValentin Clement rewriter.create<mlir::LLVM::CondBrOp>(loc, cmp, dest, destOps.getValue(), 81039f4ef81SValentin Clement newBlock, mlir::ValueRange()); 81139f4ef81SValentin Clement else 81239f4ef81SValentin Clement rewriter.create<mlir::LLVM::CondBrOp>(loc, cmp, dest, newBlock); 81339f4ef81SValentin Clement } 81439f4ef81SValentin Clement 81539f4ef81SValentin Clement template <typename A, typename B> 81639f4ef81SValentin Clement void genBrOp(A caseOp, mlir::Block *dest, Optional<B> destOps, 81739f4ef81SValentin Clement mlir::ConversionPatternRewriter &rewriter) { 81839f4ef81SValentin Clement if (destOps.hasValue()) 81939f4ef81SValentin Clement rewriter.replaceOpWithNewOp<mlir::LLVM::BrOp>(caseOp, destOps.getValue(), 82039f4ef81SValentin Clement dest); 82139f4ef81SValentin Clement else 82239f4ef81SValentin Clement rewriter.replaceOpWithNewOp<mlir::LLVM::BrOp>(caseOp, llvm::None, dest); 82339f4ef81SValentin Clement } 82439f4ef81SValentin Clement 82539f4ef81SValentin Clement void genCaseLadderStep(mlir::Location loc, mlir::Value cmp, mlir::Block *dest, 82639f4ef81SValentin Clement Optional<mlir::ValueRange> destOps, 82739f4ef81SValentin Clement mlir::ConversionPatternRewriter &rewriter) { 82839f4ef81SValentin Clement auto *thisBlock = rewriter.getInsertionBlock(); 82939f4ef81SValentin Clement auto *newBlock = createBlock(rewriter, dest); 83039f4ef81SValentin Clement rewriter.setInsertionPointToEnd(thisBlock); 83139f4ef81SValentin Clement genCondBrOp(loc, cmp, dest, destOps, rewriter, newBlock); 83239f4ef81SValentin Clement rewriter.setInsertionPointToEnd(newBlock); 83339f4ef81SValentin Clement } 83439f4ef81SValentin Clement 83539f4ef81SValentin Clement /// Conversion of `fir.select_case` 83639f4ef81SValentin Clement /// 83739f4ef81SValentin Clement /// The `fir.select_case` operation is converted to a if-then-else ladder. 83839f4ef81SValentin Clement /// Depending on the case condition type, one or several comparison and 83939f4ef81SValentin Clement /// conditional branching can be generated. 84039f4ef81SValentin Clement /// 84139f4ef81SValentin Clement /// A a point value case such as `case(4)`, a lower bound case such as 84239f4ef81SValentin Clement /// `case(5:)` or an upper bound case such as `case(:3)` are converted to a 84339f4ef81SValentin Clement /// simple comparison between the selector value and the constant value in the 84439f4ef81SValentin Clement /// case. The block associated with the case condition is then executed if 84539f4ef81SValentin Clement /// the comparison succeed otherwise it branch to the next block with the 84639f4ef81SValentin Clement /// comparison for the the next case conditon. 84739f4ef81SValentin Clement /// 84839f4ef81SValentin Clement /// A closed interval case condition such as `case(7:10)` is converted with a 84939f4ef81SValentin Clement /// first comparison and conditional branching for the lower bound. If 85039f4ef81SValentin Clement /// successful, it branch to a second block with the comparison for the 85139f4ef81SValentin Clement /// upper bound in the same case condition. 85239f4ef81SValentin Clement /// 85339f4ef81SValentin Clement /// TODO: lowering of CHARACTER type cases is not handled yet. 85439f4ef81SValentin Clement struct SelectCaseOpConversion : public FIROpConversion<fir::SelectCaseOp> { 85539f4ef81SValentin Clement using FIROpConversion::FIROpConversion; 85639f4ef81SValentin Clement 85739f4ef81SValentin Clement mlir::LogicalResult 85839f4ef81SValentin Clement matchAndRewrite(fir::SelectCaseOp caseOp, OpAdaptor adaptor, 85939f4ef81SValentin Clement mlir::ConversionPatternRewriter &rewriter) const override { 86039f4ef81SValentin Clement unsigned conds = caseOp.getNumConditions(); 86139f4ef81SValentin Clement llvm::ArrayRef<mlir::Attribute> cases = caseOp.getCases().getValue(); 86239f4ef81SValentin Clement // Type can be CHARACTER, INTEGER, or LOGICAL (C1145) 86339f4ef81SValentin Clement LLVM_ATTRIBUTE_UNUSED auto ty = caseOp.getSelector().getType(); 86439f4ef81SValentin Clement if (ty.isa<fir::CharacterType>()) 86539f4ef81SValentin Clement return rewriter.notifyMatchFailure(caseOp, 86639f4ef81SValentin Clement "conversion of fir.select_case with " 86739f4ef81SValentin Clement "character type not implemented yet"); 86839f4ef81SValentin Clement mlir::Value selector = caseOp.getSelector(adaptor.getOperands()); 86939f4ef81SValentin Clement auto loc = caseOp.getLoc(); 87039f4ef81SValentin Clement for (unsigned t = 0; t != conds; ++t) { 87139f4ef81SValentin Clement mlir::Block *dest = caseOp.getSuccessor(t); 87239f4ef81SValentin Clement llvm::Optional<mlir::ValueRange> destOps = 87339f4ef81SValentin Clement caseOp.getSuccessorOperands(adaptor.getOperands(), t); 87439f4ef81SValentin Clement llvm::Optional<mlir::ValueRange> cmpOps = 87539f4ef81SValentin Clement *caseOp.getCompareOperands(adaptor.getOperands(), t); 87639f4ef81SValentin Clement mlir::Value caseArg = *(cmpOps.getValue().begin()); 87739f4ef81SValentin Clement mlir::Attribute attr = cases[t]; 87839f4ef81SValentin Clement if (attr.isa<fir::PointIntervalAttr>()) { 87939f4ef81SValentin Clement auto cmp = rewriter.create<mlir::LLVM::ICmpOp>( 88039f4ef81SValentin Clement loc, mlir::LLVM::ICmpPredicate::eq, selector, caseArg); 88139f4ef81SValentin Clement genCaseLadderStep(loc, cmp, dest, destOps, rewriter); 88239f4ef81SValentin Clement continue; 88339f4ef81SValentin Clement } 88439f4ef81SValentin Clement if (attr.isa<fir::LowerBoundAttr>()) { 88539f4ef81SValentin Clement auto cmp = rewriter.create<mlir::LLVM::ICmpOp>( 88639f4ef81SValentin Clement loc, mlir::LLVM::ICmpPredicate::sle, caseArg, selector); 88739f4ef81SValentin Clement genCaseLadderStep(loc, cmp, dest, destOps, rewriter); 88839f4ef81SValentin Clement continue; 88939f4ef81SValentin Clement } 89039f4ef81SValentin Clement if (attr.isa<fir::UpperBoundAttr>()) { 89139f4ef81SValentin Clement auto cmp = rewriter.create<mlir::LLVM::ICmpOp>( 89239f4ef81SValentin Clement loc, mlir::LLVM::ICmpPredicate::sle, selector, caseArg); 89339f4ef81SValentin Clement genCaseLadderStep(loc, cmp, dest, destOps, rewriter); 89439f4ef81SValentin Clement continue; 89539f4ef81SValentin Clement } 89639f4ef81SValentin Clement if (attr.isa<fir::ClosedIntervalAttr>()) { 89739f4ef81SValentin Clement auto cmp = rewriter.create<mlir::LLVM::ICmpOp>( 89839f4ef81SValentin Clement loc, mlir::LLVM::ICmpPredicate::sle, caseArg, selector); 89939f4ef81SValentin Clement auto *thisBlock = rewriter.getInsertionBlock(); 90039f4ef81SValentin Clement auto *newBlock1 = createBlock(rewriter, dest); 90139f4ef81SValentin Clement auto *newBlock2 = createBlock(rewriter, dest); 90239f4ef81SValentin Clement rewriter.setInsertionPointToEnd(thisBlock); 90339f4ef81SValentin Clement rewriter.create<mlir::LLVM::CondBrOp>(loc, cmp, newBlock1, newBlock2); 90439f4ef81SValentin Clement rewriter.setInsertionPointToEnd(newBlock1); 90539f4ef81SValentin Clement mlir::Value caseArg0 = *(cmpOps.getValue().begin() + 1); 90639f4ef81SValentin Clement auto cmp0 = rewriter.create<mlir::LLVM::ICmpOp>( 90739f4ef81SValentin Clement loc, mlir::LLVM::ICmpPredicate::sle, selector, caseArg0); 90839f4ef81SValentin Clement genCondBrOp(loc, cmp0, dest, destOps, rewriter, newBlock2); 90939f4ef81SValentin Clement rewriter.setInsertionPointToEnd(newBlock2); 91039f4ef81SValentin Clement continue; 91139f4ef81SValentin Clement } 91239f4ef81SValentin Clement assert(attr.isa<mlir::UnitAttr>()); 91339f4ef81SValentin Clement assert((t + 1 == conds) && "unit must be last"); 91439f4ef81SValentin Clement genBrOp(caseOp, dest, destOps, rewriter); 91539f4ef81SValentin Clement } 91639f4ef81SValentin Clement return success(); 91739f4ef81SValentin Clement } 91839f4ef81SValentin Clement }; 91939f4ef81SValentin Clement 9208c239909SValentin Clement template <typename OP> 9218c239909SValentin Clement void selectMatchAndRewrite(fir::LLVMTypeConverter &lowering, OP select, 9228c239909SValentin Clement typename OP::Adaptor adaptor, 9238c239909SValentin Clement mlir::ConversionPatternRewriter &rewriter) { 9248c239909SValentin Clement unsigned conds = select.getNumConditions(); 9258c239909SValentin Clement auto cases = select.getCases().getValue(); 9268c239909SValentin Clement mlir::Value selector = adaptor.selector(); 9278c239909SValentin Clement auto loc = select.getLoc(); 9288c239909SValentin Clement assert(conds > 0 && "select must have cases"); 9298c239909SValentin Clement 9308c239909SValentin Clement llvm::SmallVector<mlir::Block *> destinations; 9318c239909SValentin Clement llvm::SmallVector<mlir::ValueRange> destinationsOperands; 9328c239909SValentin Clement mlir::Block *defaultDestination; 9338c239909SValentin Clement mlir::ValueRange defaultOperands; 9348c239909SValentin Clement llvm::SmallVector<int32_t> caseValues; 9358c239909SValentin Clement 9368c239909SValentin Clement for (unsigned t = 0; t != conds; ++t) { 9378c239909SValentin Clement mlir::Block *dest = select.getSuccessor(t); 9388c239909SValentin Clement auto destOps = select.getSuccessorOperands(adaptor.getOperands(), t); 9398c239909SValentin Clement const mlir::Attribute &attr = cases[t]; 9408c239909SValentin Clement if (auto intAttr = attr.template dyn_cast<mlir::IntegerAttr>()) { 9418c239909SValentin Clement destinations.push_back(dest); 9428c239909SValentin Clement destinationsOperands.push_back(destOps.hasValue() ? *destOps 9438c239909SValentin Clement : ValueRange()); 9448c239909SValentin Clement caseValues.push_back(intAttr.getInt()); 9458c239909SValentin Clement continue; 9468c239909SValentin Clement } 9478c239909SValentin Clement assert(attr.template dyn_cast_or_null<mlir::UnitAttr>()); 9488c239909SValentin Clement assert((t + 1 == conds) && "unit must be last"); 9498c239909SValentin Clement defaultDestination = dest; 9508c239909SValentin Clement defaultOperands = destOps.hasValue() ? *destOps : ValueRange(); 9518c239909SValentin Clement } 9528c239909SValentin Clement 9538c239909SValentin Clement // LLVM::SwitchOp takes a i32 type for the selector. 9548c239909SValentin Clement if (select.getSelector().getType() != rewriter.getI32Type()) 9558c239909SValentin Clement selector = 9568c239909SValentin Clement rewriter.create<LLVM::TruncOp>(loc, rewriter.getI32Type(), selector); 9578c239909SValentin Clement 9588c239909SValentin Clement rewriter.replaceOpWithNewOp<mlir::LLVM::SwitchOp>( 9598c239909SValentin Clement select, selector, 9608c239909SValentin Clement /*defaultDestination=*/defaultDestination, 9618c239909SValentin Clement /*defaultOperands=*/defaultOperands, 9628c239909SValentin Clement /*caseValues=*/caseValues, 9638c239909SValentin Clement /*caseDestinations=*/destinations, 9648c239909SValentin Clement /*caseOperands=*/destinationsOperands, 9658c239909SValentin Clement /*branchWeights=*/ArrayRef<int32_t>()); 9668c239909SValentin Clement } 9678c239909SValentin Clement 9688c239909SValentin Clement /// conversion of fir::SelectOp to an if-then-else ladder 9698c239909SValentin Clement struct SelectOpConversion : public FIROpConversion<fir::SelectOp> { 9708c239909SValentin Clement using FIROpConversion::FIROpConversion; 9718c239909SValentin Clement 9728c239909SValentin Clement mlir::LogicalResult 9738c239909SValentin Clement matchAndRewrite(fir::SelectOp op, OpAdaptor adaptor, 9748c239909SValentin Clement mlir::ConversionPatternRewriter &rewriter) const override { 9758c239909SValentin Clement selectMatchAndRewrite<fir::SelectOp>(lowerTy(), op, adaptor, rewriter); 9768c239909SValentin Clement return success(); 9778c239909SValentin Clement } 9788c239909SValentin Clement }; 9798c239909SValentin Clement 980e3349fa1SAndrzej Warzynski /// `fir.load` --> `llvm.load` 981e3349fa1SAndrzej Warzynski struct LoadOpConversion : public FIROpConversion<fir::LoadOp> { 982e3349fa1SAndrzej Warzynski using FIROpConversion::FIROpConversion; 983e3349fa1SAndrzej Warzynski 984e3349fa1SAndrzej Warzynski mlir::LogicalResult 985e3349fa1SAndrzej Warzynski matchAndRewrite(fir::LoadOp load, OpAdaptor adaptor, 986e3349fa1SAndrzej Warzynski mlir::ConversionPatternRewriter &rewriter) const override { 987e3349fa1SAndrzej Warzynski // fir.box is a special case because it is considered as an ssa values in 988e3349fa1SAndrzej Warzynski // fir, but it is lowered as a pointer to a descriptor. So fir.ref<fir.box> 989e3349fa1SAndrzej Warzynski // and fir.box end up being the same llvm types and loading a 990e3349fa1SAndrzej Warzynski // fir.ref<fir.box> is actually a no op in LLVM. 991e3349fa1SAndrzej Warzynski if (load.getType().isa<fir::BoxType>()) { 992e3349fa1SAndrzej Warzynski rewriter.replaceOp(load, adaptor.getOperands()[0]); 993e3349fa1SAndrzej Warzynski } else { 994e3349fa1SAndrzej Warzynski mlir::Type ty = convertType(load.getType()); 995e3349fa1SAndrzej Warzynski ArrayRef<NamedAttribute> at = load->getAttrs(); 996e3349fa1SAndrzej Warzynski rewriter.replaceOpWithNewOp<mlir::LLVM::LoadOp>( 997e3349fa1SAndrzej Warzynski load, ty, adaptor.getOperands(), at); 998e3349fa1SAndrzej Warzynski } 999e3349fa1SAndrzej Warzynski return success(); 1000e3349fa1SAndrzej Warzynski } 1001e3349fa1SAndrzej Warzynski }; 1002e3349fa1SAndrzej Warzynski 10032a299e4fSValentin Clement /// Lower `fir.select_type` to LLVM IR dialect. 10042a299e4fSValentin Clement struct SelectTypeOpConversion : public FIROpConversion<fir::SelectTypeOp> { 10052a299e4fSValentin Clement using FIROpConversion::FIROpConversion; 10062a299e4fSValentin Clement 10072a299e4fSValentin Clement mlir::LogicalResult 10082a299e4fSValentin Clement matchAndRewrite(fir::SelectTypeOp select, OpAdaptor adaptor, 10092a299e4fSValentin Clement mlir::ConversionPatternRewriter &rewriter) const override { 10102a299e4fSValentin Clement return rewriter.notifyMatchFailure( 10112a299e4fSValentin Clement select, "fir.select_type codegen is not implemented yet"); 10122a299e4fSValentin Clement } 10132a299e4fSValentin Clement }; 10142a299e4fSValentin Clement 10158c239909SValentin Clement /// conversion of fir::SelectRankOp to an if-then-else ladder 10168c239909SValentin Clement struct SelectRankOpConversion : public FIROpConversion<fir::SelectRankOp> { 10178c239909SValentin Clement using FIROpConversion::FIROpConversion; 10188c239909SValentin Clement 10198c239909SValentin Clement mlir::LogicalResult 10208c239909SValentin Clement matchAndRewrite(fir::SelectRankOp op, OpAdaptor adaptor, 10218c239909SValentin Clement mlir::ConversionPatternRewriter &rewriter) const override { 10228c239909SValentin Clement selectMatchAndRewrite<fir::SelectRankOp>(lowerTy(), op, adaptor, rewriter); 10238c239909SValentin Clement return success(); 10248c239909SValentin Clement } 10258c239909SValentin Clement }; 10268c239909SValentin Clement 1027e3349fa1SAndrzej Warzynski /// `fir.store` --> `llvm.store` 1028e3349fa1SAndrzej Warzynski struct StoreOpConversion : public FIROpConversion<fir::StoreOp> { 1029e3349fa1SAndrzej Warzynski using FIROpConversion::FIROpConversion; 1030e3349fa1SAndrzej Warzynski 1031e3349fa1SAndrzej Warzynski mlir::LogicalResult 1032e3349fa1SAndrzej Warzynski matchAndRewrite(fir::StoreOp store, OpAdaptor adaptor, 1033e3349fa1SAndrzej Warzynski mlir::ConversionPatternRewriter &rewriter) const override { 1034e3349fa1SAndrzej Warzynski if (store.value().getType().isa<fir::BoxType>()) { 1035e3349fa1SAndrzej Warzynski // fir.box value is actually in memory, load it first before storing it. 1036e3349fa1SAndrzej Warzynski mlir::Location loc = store.getLoc(); 1037e3349fa1SAndrzej Warzynski mlir::Type boxPtrTy = adaptor.getOperands()[0].getType(); 1038e3349fa1SAndrzej Warzynski auto val = rewriter.create<mlir::LLVM::LoadOp>( 1039e3349fa1SAndrzej Warzynski loc, boxPtrTy.cast<mlir::LLVM::LLVMPointerType>().getElementType(), 1040e3349fa1SAndrzej Warzynski adaptor.getOperands()[0]); 1041e3349fa1SAndrzej Warzynski rewriter.replaceOpWithNewOp<mlir::LLVM::StoreOp>( 1042e3349fa1SAndrzej Warzynski store, val, adaptor.getOperands()[1]); 1043e3349fa1SAndrzej Warzynski } else { 1044e3349fa1SAndrzej Warzynski rewriter.replaceOpWithNewOp<mlir::LLVM::StoreOp>( 1045e3349fa1SAndrzej Warzynski store, adaptor.getOperands()[0], adaptor.getOperands()[1]); 1046e3349fa1SAndrzej Warzynski } 1047e3349fa1SAndrzej Warzynski return success(); 1048e3349fa1SAndrzej Warzynski } 1049e3349fa1SAndrzej Warzynski }; 1050e3349fa1SAndrzej Warzynski 1051e3349fa1SAndrzej Warzynski /// convert to LLVM IR dialect `undef` 1052044d5b5dSValentin Clement struct UndefOpConversion : public FIROpConversion<fir::UndefOp> { 1053044d5b5dSValentin Clement using FIROpConversion::FIROpConversion; 1054044d5b5dSValentin Clement 1055044d5b5dSValentin Clement mlir::LogicalResult 1056044d5b5dSValentin Clement matchAndRewrite(fir::UndefOp undef, OpAdaptor, 1057044d5b5dSValentin Clement mlir::ConversionPatternRewriter &rewriter) const override { 1058044d5b5dSValentin Clement rewriter.replaceOpWithNewOp<mlir::LLVM::UndefOp>( 1059044d5b5dSValentin Clement undef, convertType(undef.getType())); 1060044d5b5dSValentin Clement return success(); 1061044d5b5dSValentin Clement } 1062044d5b5dSValentin Clement }; 1063a7a61359SValentin Clement 1064e3349fa1SAndrzej Warzynski /// `fir.unreachable` --> `llvm.unreachable` 106532e08248SAndrzej Warzynski struct UnreachableOpConversion : public FIROpConversion<fir::UnreachableOp> { 106632e08248SAndrzej Warzynski using FIROpConversion::FIROpConversion; 106732e08248SAndrzej Warzynski 106832e08248SAndrzej Warzynski mlir::LogicalResult 106932e08248SAndrzej Warzynski matchAndRewrite(fir::UnreachableOp unreach, OpAdaptor adaptor, 107032e08248SAndrzej Warzynski mlir::ConversionPatternRewriter &rewriter) const override { 107132e08248SAndrzej Warzynski rewriter.replaceOpWithNewOp<mlir::LLVM::UnreachableOp>(unreach); 107232e08248SAndrzej Warzynski return success(); 107332e08248SAndrzej Warzynski } 107432e08248SAndrzej Warzynski }; 107532e08248SAndrzej Warzynski 1076a7a61359SValentin Clement struct ZeroOpConversion : public FIROpConversion<fir::ZeroOp> { 1077a7a61359SValentin Clement using FIROpConversion::FIROpConversion; 1078a7a61359SValentin Clement 1079a7a61359SValentin Clement mlir::LogicalResult 1080a7a61359SValentin Clement matchAndRewrite(fir::ZeroOp zero, OpAdaptor, 1081a7a61359SValentin Clement mlir::ConversionPatternRewriter &rewriter) const override { 1082a7a61359SValentin Clement auto ty = convertType(zero.getType()); 1083a7a61359SValentin Clement if (ty.isa<mlir::LLVM::LLVMPointerType>()) { 1084a7a61359SValentin Clement rewriter.replaceOpWithNewOp<mlir::LLVM::NullOp>(zero, ty); 1085a7a61359SValentin Clement } else if (ty.isa<mlir::IntegerType>()) { 1086a7a61359SValentin Clement rewriter.replaceOpWithNewOp<mlir::LLVM::ConstantOp>( 1087a7a61359SValentin Clement zero, ty, mlir::IntegerAttr::get(zero.getType(), 0)); 1088a7a61359SValentin Clement } else if (mlir::LLVM::isCompatibleFloatingPointType(ty)) { 1089a7a61359SValentin Clement rewriter.replaceOpWithNewOp<mlir::LLVM::ConstantOp>( 1090a7a61359SValentin Clement zero, ty, mlir::FloatAttr::get(zero.getType(), 0.0)); 1091a7a61359SValentin Clement } else { 1092a7a61359SValentin Clement // TODO: create ConstantAggregateZero for FIR aggregate/array types. 109352d813edSValentin Clement return rewriter.notifyMatchFailure( 109452d813edSValentin Clement zero, 1095a7a61359SValentin Clement "conversion of fir.zero with aggregate type not implemented yet"); 1096a7a61359SValentin Clement } 1097a7a61359SValentin Clement return success(); 1098a7a61359SValentin Clement } 1099a7a61359SValentin Clement }; 110032e08248SAndrzej Warzynski 110154c56347SValentin Clement // Code shared between insert_value and extract_value Ops. 110254c56347SValentin Clement struct ValueOpCommon { 110354c56347SValentin Clement // Translate the arguments pertaining to any multidimensional array to 110454c56347SValentin Clement // row-major order for LLVM-IR. 110554c56347SValentin Clement static void toRowMajor(SmallVectorImpl<mlir::Attribute> &attrs, 110654c56347SValentin Clement mlir::Type ty) { 110754c56347SValentin Clement assert(ty && "type is null"); 110854c56347SValentin Clement const auto end = attrs.size(); 110954c56347SValentin Clement for (std::remove_const_t<decltype(end)> i = 0; i < end; ++i) { 111054c56347SValentin Clement if (auto seq = ty.dyn_cast<mlir::LLVM::LLVMArrayType>()) { 111154c56347SValentin Clement const auto dim = getDimension(seq); 111254c56347SValentin Clement if (dim > 1) { 111354c56347SValentin Clement auto ub = std::min(i + dim, end); 111454c56347SValentin Clement std::reverse(attrs.begin() + i, attrs.begin() + ub); 111554c56347SValentin Clement i += dim - 1; 111654c56347SValentin Clement } 111754c56347SValentin Clement ty = getArrayElementType(seq); 111854c56347SValentin Clement } else if (auto st = ty.dyn_cast<mlir::LLVM::LLVMStructType>()) { 111954c56347SValentin Clement ty = st.getBody()[attrs[i].cast<mlir::IntegerAttr>().getInt()]; 112054c56347SValentin Clement } else { 112154c56347SValentin Clement llvm_unreachable("index into invalid type"); 112254c56347SValentin Clement } 112354c56347SValentin Clement } 112454c56347SValentin Clement } 112554c56347SValentin Clement 112654c56347SValentin Clement static llvm::SmallVector<mlir::Attribute> 112754c56347SValentin Clement collectIndices(mlir::ConversionPatternRewriter &rewriter, 112854c56347SValentin Clement mlir::ArrayAttr arrAttr) { 112954c56347SValentin Clement llvm::SmallVector<mlir::Attribute> attrs; 113054c56347SValentin Clement for (auto i = arrAttr.begin(), e = arrAttr.end(); i != e; ++i) { 113154c56347SValentin Clement if (i->isa<mlir::IntegerAttr>()) { 113254c56347SValentin Clement attrs.push_back(*i); 113354c56347SValentin Clement } else { 113454c56347SValentin Clement auto fieldName = i->cast<mlir::StringAttr>().getValue(); 113554c56347SValentin Clement ++i; 113654c56347SValentin Clement auto ty = i->cast<mlir::TypeAttr>().getValue(); 113754c56347SValentin Clement auto index = ty.cast<fir::RecordType>().getFieldIndex(fieldName); 113854c56347SValentin Clement attrs.push_back(mlir::IntegerAttr::get(rewriter.getI32Type(), index)); 113954c56347SValentin Clement } 114054c56347SValentin Clement } 114154c56347SValentin Clement return attrs; 114254c56347SValentin Clement } 114354c56347SValentin Clement 114454c56347SValentin Clement private: 114554c56347SValentin Clement static unsigned getDimension(mlir::LLVM::LLVMArrayType ty) { 114654c56347SValentin Clement unsigned result = 1; 114754c56347SValentin Clement for (auto eleTy = ty.getElementType().dyn_cast<mlir::LLVM::LLVMArrayType>(); 114854c56347SValentin Clement eleTy; 114954c56347SValentin Clement eleTy = eleTy.getElementType().dyn_cast<mlir::LLVM::LLVMArrayType>()) 115054c56347SValentin Clement ++result; 115154c56347SValentin Clement return result; 115254c56347SValentin Clement } 115354c56347SValentin Clement 115454c56347SValentin Clement static mlir::Type getArrayElementType(mlir::LLVM::LLVMArrayType ty) { 115554c56347SValentin Clement auto eleTy = ty.getElementType(); 115654c56347SValentin Clement while (auto arrTy = eleTy.dyn_cast<mlir::LLVM::LLVMArrayType>()) 115754c56347SValentin Clement eleTy = arrTy.getElementType(); 115854c56347SValentin Clement return eleTy; 115954c56347SValentin Clement } 116054c56347SValentin Clement }; 116154c56347SValentin Clement 116254c56347SValentin Clement /// Extract a subobject value from an ssa-value of aggregate type 116354c56347SValentin Clement struct ExtractValueOpConversion 116454c56347SValentin Clement : public FIROpAndTypeConversion<fir::ExtractValueOp>, 116554c56347SValentin Clement public ValueOpCommon { 116654c56347SValentin Clement using FIROpAndTypeConversion::FIROpAndTypeConversion; 116754c56347SValentin Clement 116854c56347SValentin Clement mlir::LogicalResult 116954c56347SValentin Clement doRewrite(fir::ExtractValueOp extractVal, mlir::Type ty, OpAdaptor adaptor, 117054c56347SValentin Clement mlir::ConversionPatternRewriter &rewriter) const override { 117154c56347SValentin Clement auto attrs = collectIndices(rewriter, extractVal.coor()); 117254c56347SValentin Clement toRowMajor(attrs, adaptor.getOperands()[0].getType()); 117354c56347SValentin Clement auto position = mlir::ArrayAttr::get(extractVal.getContext(), attrs); 117454c56347SValentin Clement rewriter.replaceOpWithNewOp<mlir::LLVM::ExtractValueOp>( 117554c56347SValentin Clement extractVal, ty, adaptor.getOperands()[0], position); 117654c56347SValentin Clement return success(); 117754c56347SValentin Clement } 117854c56347SValentin Clement }; 117954c56347SValentin Clement 118054c56347SValentin Clement /// InsertValue is the generalized instruction for the composition of new 118154c56347SValentin Clement /// aggregate type values. 118254c56347SValentin Clement struct InsertValueOpConversion 118354c56347SValentin Clement : public FIROpAndTypeConversion<fir::InsertValueOp>, 118454c56347SValentin Clement public ValueOpCommon { 118554c56347SValentin Clement using FIROpAndTypeConversion::FIROpAndTypeConversion; 118654c56347SValentin Clement 118754c56347SValentin Clement mlir::LogicalResult 118854c56347SValentin Clement doRewrite(fir::InsertValueOp insertVal, mlir::Type ty, OpAdaptor adaptor, 118954c56347SValentin Clement mlir::ConversionPatternRewriter &rewriter) const override { 119054c56347SValentin Clement auto attrs = collectIndices(rewriter, insertVal.coor()); 119154c56347SValentin Clement toRowMajor(attrs, adaptor.getOperands()[0].getType()); 119254c56347SValentin Clement auto position = mlir::ArrayAttr::get(insertVal.getContext(), attrs); 119354c56347SValentin Clement rewriter.replaceOpWithNewOp<mlir::LLVM::InsertValueOp>( 119454c56347SValentin Clement insertVal, ty, adaptor.getOperands()[0], adaptor.getOperands()[1], 119554c56347SValentin Clement position); 119654c56347SValentin Clement return success(); 119754c56347SValentin Clement } 119854c56347SValentin Clement }; 119954c56347SValentin Clement 12003ae8e442SValentin Clement /// InsertOnRange inserts a value into a sequence over a range of offsets. 12013ae8e442SValentin Clement struct InsertOnRangeOpConversion 12023ae8e442SValentin Clement : public FIROpAndTypeConversion<fir::InsertOnRangeOp> { 12033ae8e442SValentin Clement using FIROpAndTypeConversion::FIROpAndTypeConversion; 12043ae8e442SValentin Clement 12053ae8e442SValentin Clement // Increments an array of subscripts in a row major fasion. 12063ae8e442SValentin Clement void incrementSubscripts(const SmallVector<uint64_t> &dims, 12073ae8e442SValentin Clement SmallVector<uint64_t> &subscripts) const { 12083ae8e442SValentin Clement for (size_t i = dims.size(); i > 0; --i) { 12093ae8e442SValentin Clement if (++subscripts[i - 1] < dims[i - 1]) { 12103ae8e442SValentin Clement return; 12113ae8e442SValentin Clement } 12123ae8e442SValentin Clement subscripts[i - 1] = 0; 12133ae8e442SValentin Clement } 12143ae8e442SValentin Clement } 12153ae8e442SValentin Clement 12163ae8e442SValentin Clement mlir::LogicalResult 12173ae8e442SValentin Clement doRewrite(fir::InsertOnRangeOp range, mlir::Type ty, OpAdaptor adaptor, 12183ae8e442SValentin Clement mlir::ConversionPatternRewriter &rewriter) const override { 12193ae8e442SValentin Clement 12203ae8e442SValentin Clement llvm::SmallVector<uint64_t> dims; 12213ae8e442SValentin Clement auto type = adaptor.getOperands()[0].getType(); 12223ae8e442SValentin Clement 12233ae8e442SValentin Clement // Iteratively extract the array dimensions from the type. 12243ae8e442SValentin Clement while (auto t = type.dyn_cast<mlir::LLVM::LLVMArrayType>()) { 12253ae8e442SValentin Clement dims.push_back(t.getNumElements()); 12263ae8e442SValentin Clement type = t.getElementType(); 12273ae8e442SValentin Clement } 12283ae8e442SValentin Clement 12293ae8e442SValentin Clement SmallVector<uint64_t> lBounds; 12303ae8e442SValentin Clement SmallVector<uint64_t> uBounds; 12313ae8e442SValentin Clement 12323ae8e442SValentin Clement // Extract integer value from the attribute 12333ae8e442SValentin Clement SmallVector<int64_t> coordinates = llvm::to_vector<4>( 12343ae8e442SValentin Clement llvm::map_range(range.coor(), [](Attribute a) -> int64_t { 12353ae8e442SValentin Clement return a.cast<IntegerAttr>().getInt(); 12363ae8e442SValentin Clement })); 12373ae8e442SValentin Clement 12383ae8e442SValentin Clement // Unzip the upper and lower bound and convert to a row major format. 12393ae8e442SValentin Clement for (auto i = coordinates.rbegin(), e = coordinates.rend(); i != e; ++i) { 12403ae8e442SValentin Clement uBounds.push_back(*i++); 12413ae8e442SValentin Clement lBounds.push_back(*i); 12423ae8e442SValentin Clement } 12433ae8e442SValentin Clement 12443ae8e442SValentin Clement auto &subscripts = lBounds; 12453ae8e442SValentin Clement auto loc = range.getLoc(); 12463ae8e442SValentin Clement mlir::Value lastOp = adaptor.getOperands()[0]; 12473ae8e442SValentin Clement mlir::Value insertVal = adaptor.getOperands()[1]; 12483ae8e442SValentin Clement 12493ae8e442SValentin Clement auto i64Ty = rewriter.getI64Type(); 12503ae8e442SValentin Clement while (subscripts != uBounds) { 12513ae8e442SValentin Clement // Convert uint64_t's to Attribute's. 12523ae8e442SValentin Clement SmallVector<mlir::Attribute> subscriptAttrs; 12533ae8e442SValentin Clement for (const auto &subscript : subscripts) 12543ae8e442SValentin Clement subscriptAttrs.push_back(IntegerAttr::get(i64Ty, subscript)); 12553ae8e442SValentin Clement lastOp = rewriter.create<mlir::LLVM::InsertValueOp>( 12563ae8e442SValentin Clement loc, ty, lastOp, insertVal, 12573ae8e442SValentin Clement ArrayAttr::get(range.getContext(), subscriptAttrs)); 12583ae8e442SValentin Clement 12593ae8e442SValentin Clement incrementSubscripts(dims, subscripts); 12603ae8e442SValentin Clement } 12613ae8e442SValentin Clement 12623ae8e442SValentin Clement // Convert uint64_t's to Attribute's. 12633ae8e442SValentin Clement SmallVector<mlir::Attribute> subscriptAttrs; 12643ae8e442SValentin Clement for (const auto &subscript : subscripts) 12653ae8e442SValentin Clement subscriptAttrs.push_back( 12663ae8e442SValentin Clement IntegerAttr::get(rewriter.getI64Type(), subscript)); 12673ae8e442SValentin Clement mlir::ArrayRef<mlir::Attribute> arrayRef(subscriptAttrs); 12683ae8e442SValentin Clement 12693ae8e442SValentin Clement rewriter.replaceOpWithNewOp<mlir::LLVM::InsertValueOp>( 12703ae8e442SValentin Clement range, ty, lastOp, insertVal, 12713ae8e442SValentin Clement ArrayAttr::get(range.getContext(), arrayRef)); 12723ae8e442SValentin Clement 12733ae8e442SValentin Clement return success(); 12743ae8e442SValentin Clement } 12753ae8e442SValentin Clement }; 12767b5132daSValentin Clement 12777b5132daSValentin Clement // 12787b5132daSValentin Clement // Primitive operations on Complex types 12797b5132daSValentin Clement // 12807b5132daSValentin Clement 12817b5132daSValentin Clement /// Generate inline code for complex addition/subtraction 12827b5132daSValentin Clement template <typename LLVMOP, typename OPTY> 12837b5132daSValentin Clement mlir::LLVM::InsertValueOp complexSum(OPTY sumop, mlir::ValueRange opnds, 12847b5132daSValentin Clement mlir::ConversionPatternRewriter &rewriter, 12857b5132daSValentin Clement fir::LLVMTypeConverter &lowering) { 12867b5132daSValentin Clement mlir::Value a = opnds[0]; 12877b5132daSValentin Clement mlir::Value b = opnds[1]; 12887b5132daSValentin Clement auto loc = sumop.getLoc(); 12897b5132daSValentin Clement auto ctx = sumop.getContext(); 12907b5132daSValentin Clement auto c0 = mlir::ArrayAttr::get(ctx, rewriter.getI32IntegerAttr(0)); 12917b5132daSValentin Clement auto c1 = mlir::ArrayAttr::get(ctx, rewriter.getI32IntegerAttr(1)); 12927b5132daSValentin Clement mlir::Type eleTy = lowering.convertType(getComplexEleTy(sumop.getType())); 12937b5132daSValentin Clement mlir::Type ty = lowering.convertType(sumop.getType()); 12947b5132daSValentin Clement auto x0 = rewriter.create<mlir::LLVM::ExtractValueOp>(loc, eleTy, a, c0); 12957b5132daSValentin Clement auto y0 = rewriter.create<mlir::LLVM::ExtractValueOp>(loc, eleTy, a, c1); 12967b5132daSValentin Clement auto x1 = rewriter.create<mlir::LLVM::ExtractValueOp>(loc, eleTy, b, c0); 12977b5132daSValentin Clement auto y1 = rewriter.create<mlir::LLVM::ExtractValueOp>(loc, eleTy, b, c1); 12987b5132daSValentin Clement auto rx = rewriter.create<LLVMOP>(loc, eleTy, x0, x1); 12997b5132daSValentin Clement auto ry = rewriter.create<LLVMOP>(loc, eleTy, y0, y1); 13007b5132daSValentin Clement auto r0 = rewriter.create<mlir::LLVM::UndefOp>(loc, ty); 13017b5132daSValentin Clement auto r1 = rewriter.create<mlir::LLVM::InsertValueOp>(loc, ty, r0, rx, c0); 13027b5132daSValentin Clement return rewriter.create<mlir::LLVM::InsertValueOp>(loc, ty, r1, ry, c1); 13037b5132daSValentin Clement } 13047b5132daSValentin Clement 13057b5132daSValentin Clement struct AddcOpConversion : public FIROpConversion<fir::AddcOp> { 13067b5132daSValentin Clement using FIROpConversion::FIROpConversion; 13077b5132daSValentin Clement 13087b5132daSValentin Clement mlir::LogicalResult 13097b5132daSValentin Clement matchAndRewrite(fir::AddcOp addc, OpAdaptor adaptor, 13107b5132daSValentin Clement mlir::ConversionPatternRewriter &rewriter) const override { 13117b5132daSValentin Clement // given: (x + iy) + (x' + iy') 13127b5132daSValentin Clement // result: (x + x') + i(y + y') 13137b5132daSValentin Clement auto r = complexSum<mlir::LLVM::FAddOp>(addc, adaptor.getOperands(), 13147b5132daSValentin Clement rewriter, lowerTy()); 13157b5132daSValentin Clement rewriter.replaceOp(addc, r.getResult()); 13167b5132daSValentin Clement return success(); 13177b5132daSValentin Clement } 13187b5132daSValentin Clement }; 13197b5132daSValentin Clement 13207b5132daSValentin Clement struct SubcOpConversion : public FIROpConversion<fir::SubcOp> { 13217b5132daSValentin Clement using FIROpConversion::FIROpConversion; 13227b5132daSValentin Clement 13237b5132daSValentin Clement mlir::LogicalResult 13247b5132daSValentin Clement matchAndRewrite(fir::SubcOp subc, OpAdaptor adaptor, 13257b5132daSValentin Clement mlir::ConversionPatternRewriter &rewriter) const override { 13267b5132daSValentin Clement // given: (x + iy) - (x' + iy') 13277b5132daSValentin Clement // result: (x - x') + i(y - y') 13287b5132daSValentin Clement auto r = complexSum<mlir::LLVM::FSubOp>(subc, adaptor.getOperands(), 13297b5132daSValentin Clement rewriter, lowerTy()); 13307b5132daSValentin Clement rewriter.replaceOp(subc, r.getResult()); 13317b5132daSValentin Clement return success(); 13327b5132daSValentin Clement } 13337b5132daSValentin Clement }; 13347b5132daSValentin Clement 13357b5132daSValentin Clement /// Inlined complex multiply 13367b5132daSValentin Clement struct MulcOpConversion : public FIROpConversion<fir::MulcOp> { 13377b5132daSValentin Clement using FIROpConversion::FIROpConversion; 13387b5132daSValentin Clement 13397b5132daSValentin Clement mlir::LogicalResult 13407b5132daSValentin Clement matchAndRewrite(fir::MulcOp mulc, OpAdaptor adaptor, 13417b5132daSValentin Clement mlir::ConversionPatternRewriter &rewriter) const override { 13427b5132daSValentin Clement // TODO: Can we use a call to __muldc3 ? 13437b5132daSValentin Clement // given: (x + iy) * (x' + iy') 13447b5132daSValentin Clement // result: (xx'-yy')+i(xy'+yx') 13457b5132daSValentin Clement mlir::Value a = adaptor.getOperands()[0]; 13467b5132daSValentin Clement mlir::Value b = adaptor.getOperands()[1]; 13477b5132daSValentin Clement auto loc = mulc.getLoc(); 13487b5132daSValentin Clement auto *ctx = mulc.getContext(); 13497b5132daSValentin Clement auto c0 = mlir::ArrayAttr::get(ctx, rewriter.getI32IntegerAttr(0)); 13507b5132daSValentin Clement auto c1 = mlir::ArrayAttr::get(ctx, rewriter.getI32IntegerAttr(1)); 13517b5132daSValentin Clement mlir::Type eleTy = convertType(getComplexEleTy(mulc.getType())); 13527b5132daSValentin Clement mlir::Type ty = convertType(mulc.getType()); 13537b5132daSValentin Clement auto x0 = rewriter.create<mlir::LLVM::ExtractValueOp>(loc, eleTy, a, c0); 13547b5132daSValentin Clement auto y0 = rewriter.create<mlir::LLVM::ExtractValueOp>(loc, eleTy, a, c1); 13557b5132daSValentin Clement auto x1 = rewriter.create<mlir::LLVM::ExtractValueOp>(loc, eleTy, b, c0); 13567b5132daSValentin Clement auto y1 = rewriter.create<mlir::LLVM::ExtractValueOp>(loc, eleTy, b, c1); 13577b5132daSValentin Clement auto xx = rewriter.create<mlir::LLVM::FMulOp>(loc, eleTy, x0, x1); 13587b5132daSValentin Clement auto yx = rewriter.create<mlir::LLVM::FMulOp>(loc, eleTy, y0, x1); 13597b5132daSValentin Clement auto xy = rewriter.create<mlir::LLVM::FMulOp>(loc, eleTy, x0, y1); 13607b5132daSValentin Clement auto ri = rewriter.create<mlir::LLVM::FAddOp>(loc, eleTy, xy, yx); 13617b5132daSValentin Clement auto yy = rewriter.create<mlir::LLVM::FMulOp>(loc, eleTy, y0, y1); 13627b5132daSValentin Clement auto rr = rewriter.create<mlir::LLVM::FSubOp>(loc, eleTy, xx, yy); 13637b5132daSValentin Clement auto ra = rewriter.create<mlir::LLVM::UndefOp>(loc, ty); 13647b5132daSValentin Clement auto r1 = rewriter.create<mlir::LLVM::InsertValueOp>(loc, ty, ra, rr, c0); 13657b5132daSValentin Clement auto r0 = rewriter.create<mlir::LLVM::InsertValueOp>(loc, ty, r1, ri, c1); 13667b5132daSValentin Clement rewriter.replaceOp(mulc, r0.getResult()); 13677b5132daSValentin Clement return success(); 13687b5132daSValentin Clement } 13697b5132daSValentin Clement }; 13707b5132daSValentin Clement 13717b5132daSValentin Clement /// Inlined complex division 13727b5132daSValentin Clement struct DivcOpConversion : public FIROpConversion<fir::DivcOp> { 13737b5132daSValentin Clement using FIROpConversion::FIROpConversion; 13747b5132daSValentin Clement 13757b5132daSValentin Clement mlir::LogicalResult 13767b5132daSValentin Clement matchAndRewrite(fir::DivcOp divc, OpAdaptor adaptor, 13777b5132daSValentin Clement mlir::ConversionPatternRewriter &rewriter) const override { 13787b5132daSValentin Clement // TODO: Can we use a call to __divdc3 instead? 13797b5132daSValentin Clement // Just generate inline code for now. 13807b5132daSValentin Clement // given: (x + iy) / (x' + iy') 13817b5132daSValentin Clement // result: ((xx'+yy')/d) + i((yx'-xy')/d) where d = x'x' + y'y' 13827b5132daSValentin Clement mlir::Value a = adaptor.getOperands()[0]; 13837b5132daSValentin Clement mlir::Value b = adaptor.getOperands()[1]; 13847b5132daSValentin Clement auto loc = divc.getLoc(); 13857b5132daSValentin Clement auto *ctx = divc.getContext(); 13867b5132daSValentin Clement auto c0 = mlir::ArrayAttr::get(ctx, rewriter.getI32IntegerAttr(0)); 13877b5132daSValentin Clement auto c1 = mlir::ArrayAttr::get(ctx, rewriter.getI32IntegerAttr(1)); 13887b5132daSValentin Clement mlir::Type eleTy = convertType(getComplexEleTy(divc.getType())); 13897b5132daSValentin Clement mlir::Type ty = convertType(divc.getType()); 13907b5132daSValentin Clement auto x0 = rewriter.create<mlir::LLVM::ExtractValueOp>(loc, eleTy, a, c0); 13917b5132daSValentin Clement auto y0 = rewriter.create<mlir::LLVM::ExtractValueOp>(loc, eleTy, a, c1); 13927b5132daSValentin Clement auto x1 = rewriter.create<mlir::LLVM::ExtractValueOp>(loc, eleTy, b, c0); 13937b5132daSValentin Clement auto y1 = rewriter.create<mlir::LLVM::ExtractValueOp>(loc, eleTy, b, c1); 13947b5132daSValentin Clement auto xx = rewriter.create<mlir::LLVM::FMulOp>(loc, eleTy, x0, x1); 13957b5132daSValentin Clement auto x1x1 = rewriter.create<mlir::LLVM::FMulOp>(loc, eleTy, x1, x1); 13967b5132daSValentin Clement auto yx = rewriter.create<mlir::LLVM::FMulOp>(loc, eleTy, y0, x1); 13977b5132daSValentin Clement auto xy = rewriter.create<mlir::LLVM::FMulOp>(loc, eleTy, x0, y1); 13987b5132daSValentin Clement auto yy = rewriter.create<mlir::LLVM::FMulOp>(loc, eleTy, y0, y1); 13997b5132daSValentin Clement auto y1y1 = rewriter.create<mlir::LLVM::FMulOp>(loc, eleTy, y1, y1); 14007b5132daSValentin Clement auto d = rewriter.create<mlir::LLVM::FAddOp>(loc, eleTy, x1x1, y1y1); 14017b5132daSValentin Clement auto rrn = rewriter.create<mlir::LLVM::FAddOp>(loc, eleTy, xx, yy); 14027b5132daSValentin Clement auto rin = rewriter.create<mlir::LLVM::FSubOp>(loc, eleTy, yx, xy); 14037b5132daSValentin Clement auto rr = rewriter.create<mlir::LLVM::FDivOp>(loc, eleTy, rrn, d); 14047b5132daSValentin Clement auto ri = rewriter.create<mlir::LLVM::FDivOp>(loc, eleTy, rin, d); 14057b5132daSValentin Clement auto ra = rewriter.create<mlir::LLVM::UndefOp>(loc, ty); 14067b5132daSValentin Clement auto r1 = rewriter.create<mlir::LLVM::InsertValueOp>(loc, ty, ra, rr, c0); 14077b5132daSValentin Clement auto r0 = rewriter.create<mlir::LLVM::InsertValueOp>(loc, ty, r1, ri, c1); 14087b5132daSValentin Clement rewriter.replaceOp(divc, r0.getResult()); 14097b5132daSValentin Clement return success(); 14107b5132daSValentin Clement } 14117b5132daSValentin Clement }; 14127b5132daSValentin Clement 14137b5132daSValentin Clement /// Inlined complex negation 14147b5132daSValentin Clement struct NegcOpConversion : public FIROpConversion<fir::NegcOp> { 14157b5132daSValentin Clement using FIROpConversion::FIROpConversion; 14167b5132daSValentin Clement 14177b5132daSValentin Clement mlir::LogicalResult 14187b5132daSValentin Clement matchAndRewrite(fir::NegcOp neg, OpAdaptor adaptor, 14197b5132daSValentin Clement mlir::ConversionPatternRewriter &rewriter) const override { 14207b5132daSValentin Clement // given: -(x + iy) 14217b5132daSValentin Clement // result: -x - iy 14227b5132daSValentin Clement auto *ctxt = neg.getContext(); 14237b5132daSValentin Clement auto eleTy = convertType(getComplexEleTy(neg.getType())); 14247b5132daSValentin Clement auto ty = convertType(neg.getType()); 14257b5132daSValentin Clement auto loc = neg.getLoc(); 14267b5132daSValentin Clement mlir::Value o0 = adaptor.getOperands()[0]; 14277b5132daSValentin Clement auto c0 = mlir::ArrayAttr::get(ctxt, rewriter.getI32IntegerAttr(0)); 14287b5132daSValentin Clement auto c1 = mlir::ArrayAttr::get(ctxt, rewriter.getI32IntegerAttr(1)); 14297b5132daSValentin Clement auto rp = rewriter.create<mlir::LLVM::ExtractValueOp>(loc, eleTy, o0, c0); 14307b5132daSValentin Clement auto ip = rewriter.create<mlir::LLVM::ExtractValueOp>(loc, eleTy, o0, c1); 14317b5132daSValentin Clement auto nrp = rewriter.create<mlir::LLVM::FNegOp>(loc, eleTy, rp); 14327b5132daSValentin Clement auto nip = rewriter.create<mlir::LLVM::FNegOp>(loc, eleTy, ip); 14337b5132daSValentin Clement auto r = rewriter.create<mlir::LLVM::InsertValueOp>(loc, ty, o0, nrp, c0); 14347b5132daSValentin Clement rewriter.replaceOpWithNewOp<mlir::LLVM::InsertValueOp>(neg, ty, r, nip, c1); 14357b5132daSValentin Clement return success(); 14367b5132daSValentin Clement } 14377b5132daSValentin Clement }; 14387b5132daSValentin Clement 1439420ad7ceSAndrzej Warzynski /// `fir.is_present` --> 1440420ad7ceSAndrzej Warzynski /// ``` 1441420ad7ceSAndrzej Warzynski /// %0 = llvm.mlir.constant(0 : i64) 1442420ad7ceSAndrzej Warzynski /// %1 = llvm.ptrtoint %0 1443420ad7ceSAndrzej Warzynski /// %2 = llvm.icmp "ne" %1, %0 : i64 1444420ad7ceSAndrzej Warzynski /// ``` 1445420ad7ceSAndrzej Warzynski struct IsPresentOpConversion : public FIROpConversion<fir::IsPresentOp> { 1446420ad7ceSAndrzej Warzynski using FIROpConversion::FIROpConversion; 1447420ad7ceSAndrzej Warzynski 1448420ad7ceSAndrzej Warzynski mlir::LogicalResult 1449420ad7ceSAndrzej Warzynski matchAndRewrite(fir::IsPresentOp isPresent, OpAdaptor adaptor, 1450420ad7ceSAndrzej Warzynski mlir::ConversionPatternRewriter &rewriter) const override { 1451420ad7ceSAndrzej Warzynski mlir::Type idxTy = lowerTy().indexType(); 1452420ad7ceSAndrzej Warzynski mlir::Location loc = isPresent.getLoc(); 1453420ad7ceSAndrzej Warzynski auto ptr = adaptor.getOperands()[0]; 1454420ad7ceSAndrzej Warzynski 1455420ad7ceSAndrzej Warzynski if (isPresent.val().getType().isa<fir::BoxCharType>()) { 1456420ad7ceSAndrzej Warzynski auto structTy = ptr.getType().cast<mlir::LLVM::LLVMStructType>(); 1457420ad7ceSAndrzej Warzynski assert(!structTy.isOpaque() && !structTy.getBody().empty()); 1458420ad7ceSAndrzej Warzynski 1459420ad7ceSAndrzej Warzynski mlir::Type ty = structTy.getBody()[0]; 1460420ad7ceSAndrzej Warzynski mlir::MLIRContext *ctx = isPresent.getContext(); 1461420ad7ceSAndrzej Warzynski auto c0 = mlir::ArrayAttr::get(ctx, rewriter.getI32IntegerAttr(0)); 1462420ad7ceSAndrzej Warzynski ptr = rewriter.create<mlir::LLVM::ExtractValueOp>(loc, ty, ptr, c0); 1463420ad7ceSAndrzej Warzynski } 1464420ad7ceSAndrzej Warzynski mlir::LLVM::ConstantOp c0 = 1465420ad7ceSAndrzej Warzynski genConstantIndex(isPresent.getLoc(), idxTy, rewriter, 0); 1466420ad7ceSAndrzej Warzynski auto addr = rewriter.create<mlir::LLVM::PtrToIntOp>(loc, idxTy, ptr); 1467420ad7ceSAndrzej Warzynski rewriter.replaceOpWithNewOp<mlir::LLVM::ICmpOp>( 1468420ad7ceSAndrzej Warzynski isPresent, mlir::LLVM::ICmpPredicate::ne, addr, c0); 1469420ad7ceSAndrzej Warzynski 1470420ad7ceSAndrzej Warzynski return success(); 1471420ad7ceSAndrzej Warzynski } 1472420ad7ceSAndrzej Warzynski }; 14731e77b095SAndrzej Warzynski 14741e77b095SAndrzej Warzynski /// Convert `!fir.emboxchar<!fir.char<KIND, ?>, #n>` into a sequence of 14751e77b095SAndrzej Warzynski /// instructions that generate `!llvm.struct<(ptr<ik>, i64)>`. The 1st element 14761e77b095SAndrzej Warzynski /// in this struct is a pointer. Its type is determined from `KIND`. The 2nd 14771e77b095SAndrzej Warzynski /// element is the length of the character buffer (`#n`). 14781e77b095SAndrzej Warzynski struct EmboxCharOpConversion : public FIROpConversion<fir::EmboxCharOp> { 14791e77b095SAndrzej Warzynski using FIROpConversion::FIROpConversion; 14801e77b095SAndrzej Warzynski 14811e77b095SAndrzej Warzynski mlir::LogicalResult 14821e77b095SAndrzej Warzynski matchAndRewrite(fir::EmboxCharOp emboxChar, OpAdaptor adaptor, 14831e77b095SAndrzej Warzynski mlir::ConversionPatternRewriter &rewriter) const override { 14841e77b095SAndrzej Warzynski mlir::ValueRange operands = adaptor.getOperands(); 14851e77b095SAndrzej Warzynski MLIRContext *ctx = emboxChar.getContext(); 14861e77b095SAndrzej Warzynski 14871e77b095SAndrzej Warzynski mlir::Value charBuffer = operands[0]; 14881e77b095SAndrzej Warzynski mlir::Value charBufferLen = operands[1]; 14891e77b095SAndrzej Warzynski 14901e77b095SAndrzej Warzynski mlir::Location loc = emboxChar.getLoc(); 14911e77b095SAndrzej Warzynski mlir::Type llvmStructTy = convertType(emboxChar.getType()); 14921e77b095SAndrzej Warzynski auto llvmStruct = rewriter.create<mlir::LLVM::UndefOp>(loc, llvmStructTy); 14931e77b095SAndrzej Warzynski 14941e77b095SAndrzej Warzynski mlir::Type lenTy = 14951e77b095SAndrzej Warzynski llvmStructTy.cast<mlir::LLVM::LLVMStructType>().getBody()[1]; 14961e77b095SAndrzej Warzynski mlir::Value lenAfterCast = integerCast(loc, rewriter, lenTy, charBufferLen); 14971e77b095SAndrzej Warzynski 14981e77b095SAndrzej Warzynski auto c0 = mlir::ArrayAttr::get(ctx, rewriter.getI32IntegerAttr(0)); 14991e77b095SAndrzej Warzynski auto c1 = mlir::ArrayAttr::get(ctx, rewriter.getI32IntegerAttr(1)); 15001e77b095SAndrzej Warzynski auto insertBufferOp = rewriter.create<mlir::LLVM::InsertValueOp>( 15011e77b095SAndrzej Warzynski loc, llvmStructTy, llvmStruct, charBuffer, c0); 15021e77b095SAndrzej Warzynski rewriter.replaceOpWithNewOp<mlir::LLVM::InsertValueOp>( 15031e77b095SAndrzej Warzynski emboxChar, llvmStructTy, insertBufferOp, lenAfterCast, c1); 15041e77b095SAndrzej Warzynski 15051e77b095SAndrzej Warzynski return success(); 15061e77b095SAndrzej Warzynski } 15071e77b095SAndrzej Warzynski }; 150814867ffcSAndrzej Warzynski 150914867ffcSAndrzej Warzynski /// Construct an `llvm.extractvalue` instruction. It will return value at 151014867ffcSAndrzej Warzynski /// element \p x from \p tuple. 151114867ffcSAndrzej Warzynski mlir::LLVM::ExtractValueOp 151214867ffcSAndrzej Warzynski genExtractValueWithIndex(mlir::Location loc, mlir::Value tuple, mlir::Type ty, 151314867ffcSAndrzej Warzynski mlir::ConversionPatternRewriter &rewriter, 151414867ffcSAndrzej Warzynski mlir::MLIRContext *ctx, int x) { 151514867ffcSAndrzej Warzynski auto cx = mlir::ArrayAttr::get(ctx, rewriter.getI32IntegerAttr(x)); 151614867ffcSAndrzej Warzynski auto xty = ty.cast<mlir::LLVM::LLVMStructType>().getBody()[x]; 151714867ffcSAndrzej Warzynski return rewriter.create<mlir::LLVM::ExtractValueOp>(loc, xty, tuple, cx); 151814867ffcSAndrzej Warzynski } 151914867ffcSAndrzej Warzynski 1520*6c3d7fd4SAndrzej Warzynski /// Convert `!fir.boxchar_len` to `!llvm.extractvalue` for the 2nd part of the 1521*6c3d7fd4SAndrzej Warzynski /// boxchar. 1522*6c3d7fd4SAndrzej Warzynski struct BoxCharLenOpConversion : public FIROpConversion<fir::BoxCharLenOp> { 1523*6c3d7fd4SAndrzej Warzynski using FIROpConversion::FIROpConversion; 1524*6c3d7fd4SAndrzej Warzynski 1525*6c3d7fd4SAndrzej Warzynski mlir::LogicalResult 1526*6c3d7fd4SAndrzej Warzynski matchAndRewrite(fir::BoxCharLenOp boxCharLen, OpAdaptor adaptor, 1527*6c3d7fd4SAndrzej Warzynski mlir::ConversionPatternRewriter &rewriter) const override { 1528*6c3d7fd4SAndrzej Warzynski mlir::Value boxChar = adaptor.getOperands()[0]; 1529*6c3d7fd4SAndrzej Warzynski mlir::Location loc = boxChar.getLoc(); 1530*6c3d7fd4SAndrzej Warzynski mlir::MLIRContext *ctx = boxChar.getContext(); 1531*6c3d7fd4SAndrzej Warzynski mlir::Type returnValTy = boxCharLen.getResult().getType(); 1532*6c3d7fd4SAndrzej Warzynski 1533*6c3d7fd4SAndrzej Warzynski constexpr int boxcharLenIdx = 1; 1534*6c3d7fd4SAndrzej Warzynski mlir::LLVM::ExtractValueOp len = genExtractValueWithIndex( 1535*6c3d7fd4SAndrzej Warzynski loc, boxChar, boxChar.getType(), rewriter, ctx, boxcharLenIdx); 1536*6c3d7fd4SAndrzej Warzynski mlir::Value lenAfterCast = integerCast(loc, rewriter, returnValTy, len); 1537*6c3d7fd4SAndrzej Warzynski rewriter.replaceOp(boxCharLen, lenAfterCast); 1538*6c3d7fd4SAndrzej Warzynski 1539*6c3d7fd4SAndrzej Warzynski return success(); 1540*6c3d7fd4SAndrzej Warzynski } 1541*6c3d7fd4SAndrzej Warzynski }; 1542*6c3d7fd4SAndrzej Warzynski 154314867ffcSAndrzej Warzynski /// Convert `fir.unboxchar` into two `llvm.extractvalue` instructions. One for 154414867ffcSAndrzej Warzynski /// the character buffer and one for the buffer length. 154514867ffcSAndrzej Warzynski struct UnboxCharOpConversion : public FIROpConversion<fir::UnboxCharOp> { 154614867ffcSAndrzej Warzynski using FIROpConversion::FIROpConversion; 154714867ffcSAndrzej Warzynski 154814867ffcSAndrzej Warzynski mlir::LogicalResult 154914867ffcSAndrzej Warzynski matchAndRewrite(fir::UnboxCharOp unboxchar, OpAdaptor adaptor, 155014867ffcSAndrzej Warzynski mlir::ConversionPatternRewriter &rewriter) const override { 155114867ffcSAndrzej Warzynski MLIRContext *ctx = unboxchar.getContext(); 155214867ffcSAndrzej Warzynski 155314867ffcSAndrzej Warzynski mlir::Type lenTy = convertType(unboxchar.getType(1)); 155414867ffcSAndrzej Warzynski mlir::Value tuple = adaptor.getOperands()[0]; 155514867ffcSAndrzej Warzynski mlir::Type tupleTy = tuple.getType(); 155614867ffcSAndrzej Warzynski 155714867ffcSAndrzej Warzynski mlir::Location loc = unboxchar.getLoc(); 155814867ffcSAndrzej Warzynski mlir::Value ptrToBuffer = 155914867ffcSAndrzej Warzynski genExtractValueWithIndex(loc, tuple, tupleTy, rewriter, ctx, 0); 156014867ffcSAndrzej Warzynski 156114867ffcSAndrzej Warzynski mlir::LLVM::ExtractValueOp len = 156214867ffcSAndrzej Warzynski genExtractValueWithIndex(loc, tuple, tupleTy, rewriter, ctx, 1); 156314867ffcSAndrzej Warzynski mlir::Value lenAfterCast = integerCast(loc, rewriter, lenTy, len); 156414867ffcSAndrzej Warzynski 156514867ffcSAndrzej Warzynski rewriter.replaceOp(unboxchar, 156614867ffcSAndrzej Warzynski ArrayRef<mlir::Value>{ptrToBuffer, lenAfterCast}); 156714867ffcSAndrzej Warzynski return success(); 156814867ffcSAndrzej Warzynski } 156914867ffcSAndrzej Warzynski }; 157014867ffcSAndrzej Warzynski 1571044d5b5dSValentin Clement } // namespace 1572044d5b5dSValentin Clement 1573044d5b5dSValentin Clement namespace { 1574044d5b5dSValentin Clement /// Convert FIR dialect to LLVM dialect 1575044d5b5dSValentin Clement /// 1576044d5b5dSValentin Clement /// This pass lowers all FIR dialect operations to LLVM IR dialect. An 1577044d5b5dSValentin Clement /// MLIR pass is used to lower residual Std dialect to LLVM IR dialect. 1578044d5b5dSValentin Clement /// 1579044d5b5dSValentin Clement /// This pass is not complete yet. We are upstreaming it in small patches. 1580044d5b5dSValentin Clement class FIRToLLVMLowering : public fir::FIRToLLVMLoweringBase<FIRToLLVMLowering> { 1581044d5b5dSValentin Clement public: 1582044d5b5dSValentin Clement mlir::ModuleOp getModule() { return getOperation(); } 1583044d5b5dSValentin Clement 1584044d5b5dSValentin Clement void runOnOperation() override final { 15857b5132daSValentin Clement auto mod = getModule(); 15867b5132daSValentin Clement if (!forcedTargetTriple.empty()) { 15877b5132daSValentin Clement fir::setTargetTriple(mod, forcedTargetTriple); 15887b5132daSValentin Clement } 15897b5132daSValentin Clement 1590044d5b5dSValentin Clement auto *context = getModule().getContext(); 1591044d5b5dSValentin Clement fir::LLVMTypeConverter typeConverter{getModule()}; 1592044d5b5dSValentin Clement mlir::OwningRewritePatternList pattern(context); 1593df3b9810SValentin Clement pattern.insert< 1594420ad7ceSAndrzej Warzynski AbsentOpConversion, AddcOpConversion, AddrOfOpConversion, 1595*6c3d7fd4SAndrzej Warzynski AllocaOpConversion, BoxAddrOpConversion, BoxCharLenOpConversion, BoxDimsOpConversion, 1596420ad7ceSAndrzej Warzynski BoxEleSizeOpConversion, BoxIsAllocOpConversion, BoxIsArrayOpConversion, 1597420ad7ceSAndrzej Warzynski BoxIsPtrOpConversion, BoxRankOpConversion, CallOpConversion, 1598f1dfc027SDiana Picus CmpcOpConversion, ConvertOpConversion, DispatchOpConversion, 1599f1dfc027SDiana Picus DispatchTableOpConversion, DTEntryOpConversion, DivcOpConversion, 1600f1dfc027SDiana Picus EmboxCharOpConversion, ExtractValueOpConversion, HasValueOpConversion, 160131246187SValentin Clement GenTypeDescOpConversion, GlobalLenOpConversion, GlobalOpConversion, 160231246187SValentin Clement InsertOnRangeOpConversion, InsertValueOpConversion, 160331246187SValentin Clement IsPresentOpConversion, LoadOpConversion, NegcOpConversion, 160431246187SValentin Clement MulcOpConversion, SelectCaseOpConversion, SelectOpConversion, 160531246187SValentin Clement SelectRankOpConversion, SelectTypeOpConversion, StoreOpConversion, 160631246187SValentin Clement SubcOpConversion, UnboxCharOpConversion, UndefOpConversion, 160731246187SValentin Clement UnreachableOpConversion, ZeroOpConversion>(typeConverter); 1608044d5b5dSValentin Clement mlir::populateStdToLLVMConversionPatterns(typeConverter, pattern); 1609044d5b5dSValentin Clement mlir::arith::populateArithmeticToLLVMConversionPatterns(typeConverter, 1610044d5b5dSValentin Clement pattern); 1611044d5b5dSValentin Clement mlir::ConversionTarget target{*context}; 1612044d5b5dSValentin Clement target.addLegalDialect<mlir::LLVM::LLVMDialect>(); 1613044d5b5dSValentin Clement 1614044d5b5dSValentin Clement // required NOPs for applying a full conversion 1615044d5b5dSValentin Clement target.addLegalOp<mlir::ModuleOp>(); 1616044d5b5dSValentin Clement 1617044d5b5dSValentin Clement // apply the patterns 1618044d5b5dSValentin Clement if (mlir::failed(mlir::applyFullConversion(getModule(), target, 1619044d5b5dSValentin Clement std::move(pattern)))) { 1620044d5b5dSValentin Clement signalPassFailure(); 1621044d5b5dSValentin Clement } 1622044d5b5dSValentin Clement } 1623044d5b5dSValentin Clement }; 1624044d5b5dSValentin Clement } // namespace 1625044d5b5dSValentin Clement 1626044d5b5dSValentin Clement std::unique_ptr<mlir::Pass> fir::createFIRToLLVMPass() { 1627044d5b5dSValentin Clement return std::make_unique<FIRToLLVMLowering>(); 1628044d5b5dSValentin Clement } 1629