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 14 using namespace Fortran::runtime; 15 16 namespace fir::runtime { 17 template <> 18 constexpr TypeBuilderFunc 19 getModel<Fortran::runtime::ArrayConstructorVector &>() { 20 return getModel<void *>(); 21 } 22 } // namespace fir::runtime 23 24 mlir::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 57 void 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 69 void 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