80 lines · cpp
1//===- ArrayConstructor.cpp - array constructor runtime API calls ---------===//2//3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.4// See https://llvm.org/LICENSE.txt for license information.5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception6//7//===----------------------------------------------------------------------===//8 9#include "flang/Optimizer/Builder/Runtime/ArrayConstructor.h"10#include "flang/Optimizer/Builder/FIRBuilder.h"11#include "flang/Optimizer/Builder/Runtime/RTBuilder.h"12#include "flang/Runtime/array-constructor-consts.h"13 14using namespace Fortran::runtime;15 16namespace fir::runtime {17template <>18constexpr TypeBuilderFunc19getModel<Fortran::runtime::ArrayConstructorVector &>() {20 return getModel<void *>();21}22} // namespace fir::runtime23 24mlir::Value fir::runtime::genInitArrayConstructorVector(25 mlir::Location loc, fir::FirOpBuilder &builder, mlir::Value toBox,26 mlir::Value useValueLengthParameters) {27 // Allocate storage for the runtime cookie for the array constructor vector.28 // Use pessimistic values for size and alignment that are valid for all29 // supported targets. Whether the actual ArrayConstructorVector object fits30 // into the available MaxArrayConstructorVectorSizeInBytes is verified when31 // building clang-rt.32 std::size_t arrayVectorStructBitSize =33 MaxArrayConstructorVectorSizeInBytes * 8;34 std::size_t alignLike = MaxArrayConstructorVectorAlignInBytes * 8;35 fir::SequenceType::Extent numElem =36 (arrayVectorStructBitSize + alignLike - 1) / alignLike;37 mlir::Type intType = builder.getIntegerType(alignLike);38 mlir::Type seqType = fir::SequenceType::get({numElem}, intType);39 mlir::Value cookie =40 builder.createTemporary(loc, seqType, ".rt.arrayctor.vector");41 42 mlir::func::FuncOp func =43 fir::runtime::getRuntimeFunc<mkRTKey(InitArrayConstructorVector)>(44 loc, builder);45 mlir::FunctionType funcType = func.getFunctionType();46 cookie = builder.createConvert(loc, funcType.getInput(0), cookie);47 mlir::Value sourceFile = fir::factory::locationToFilename(builder, loc);48 mlir::Value sourceLine =49 fir::factory::locationToLineNo(builder, loc, funcType.getInput(4));50 auto args = fir::runtime::createArguments(builder, loc, funcType, cookie,51 toBox, useValueLengthParameters,52 sourceFile, sourceLine);53 fir::CallOp::create(builder, loc, func, args);54 return cookie;55}56 57void fir::runtime::genPushArrayConstructorValue(58 mlir::Location loc, fir::FirOpBuilder &builder,59 mlir::Value arrayConstructorVector, mlir::Value fromBox) {60 mlir::func::FuncOp func =61 fir::runtime::getRuntimeFunc<mkRTKey(PushArrayConstructorValue)>(loc,62 builder);63 mlir::FunctionType funcType = func.getFunctionType();64 auto args = fir::runtime::createArguments(builder, loc, funcType,65 arrayConstructorVector, fromBox);66 fir::CallOp::create(builder, loc, func, args);67}68 69void fir::runtime::genPushArrayConstructorSimpleScalar(70 mlir::Location loc, fir::FirOpBuilder &builder,71 mlir::Value arrayConstructorVector, mlir::Value fromAddress) {72 mlir::func::FuncOp func =73 fir::runtime::getRuntimeFunc<mkRTKey(PushArrayConstructorSimpleScalar)>(74 loc, builder);75 mlir::FunctionType funcType = func.getFunctionType();76 auto args = fir::runtime::createArguments(77 builder, loc, funcType, arrayConstructorVector, fromAddress);78 fir::CallOp::create(builder, loc, func, args);79}80