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-exception
6//
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 TypeBuilderFunc
19getModel<Fortran::runtime::ArrayConstructorVector &>() {
20 return getModel<void *>();
21}
22} // namespace fir::runtime
23
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 all
29 // supported targets. Whether the actual ArrayConstructorVector object fits
30 // into the available MaxArrayConstructorVectorSizeInBytes is verified when
31 // 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 builder.create<fir::CallOp>(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 builder.create<fir::CallOp>(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 builder.create<fir::CallOp>(loc, func, args);
79}
80

source code of flang/lib/Optimizer/Builder/Runtime/ArrayConstructor.cpp