a3d560342c
The polymorphic temporary array is created using the provided mold and the shape of the hlfir.elemental. The array is allocated right away, because it is going to be initialized element per element. Depends on D157315 Reviewed By: clementval, tblah Differential Revision: https://reviews.llvm.org/D157316
92 lines
4.3 KiB
C++
92 lines
4.3 KiB
C++
//===-- Allocatable.cpp -- generate allocatable runtime API calls----------===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "flang/Optimizer/Builder/Runtime/Allocatable.h"
|
|
#include "flang/Optimizer/Builder/FIRBuilder.h"
|
|
#include "flang/Optimizer/Builder/Runtime/RTBuilder.h"
|
|
#include "flang/Runtime/allocatable.h"
|
|
|
|
using namespace Fortran::runtime;
|
|
|
|
mlir::Value fir::runtime::genMoveAlloc(fir::FirOpBuilder &builder,
|
|
mlir::Location loc, mlir::Value to,
|
|
mlir::Value from, mlir::Value hasStat,
|
|
mlir::Value errMsg) {
|
|
mlir::func::FuncOp func{
|
|
fir::runtime::getRuntimeFunc<mkRTKey(MoveAlloc)>(loc, builder)};
|
|
mlir::FunctionType fTy{func.getFunctionType()};
|
|
mlir::Value sourceFile{fir::factory::locationToFilename(builder, loc)};
|
|
mlir::Value sourceLine{
|
|
fir::factory::locationToLineNo(builder, loc, fTy.getInput(6))};
|
|
mlir::Value declaredTypeDesc;
|
|
if (fir::isPolymorphicType(from.getType()) &&
|
|
!fir::isUnlimitedPolymorphicType(from.getType())) {
|
|
fir::ClassType clTy =
|
|
fir::dyn_cast_ptrEleTy(from.getType()).dyn_cast<fir::ClassType>();
|
|
mlir::Type derivedType = fir::unwrapInnerType(clTy.getEleTy());
|
|
declaredTypeDesc =
|
|
builder.create<fir::TypeDescOp>(loc, mlir::TypeAttr::get(derivedType));
|
|
} else {
|
|
declaredTypeDesc = builder.createNullConstant(loc);
|
|
}
|
|
llvm::SmallVector<mlir::Value> args{fir::runtime::createArguments(
|
|
builder, loc, fTy, to, from, declaredTypeDesc, hasStat, errMsg,
|
|
sourceFile, sourceLine)};
|
|
|
|
return builder.create<fir::CallOp>(loc, func, args).getResult(0);
|
|
}
|
|
|
|
void fir::runtime::genAllocatableApplyMold(fir::FirOpBuilder &builder,
|
|
mlir::Location loc, mlir::Value desc,
|
|
mlir::Value mold, int rank) {
|
|
mlir::func::FuncOp func{
|
|
fir::runtime::getRuntimeFunc<mkRTKey(AllocatableApplyMold)>(loc,
|
|
builder)};
|
|
mlir::FunctionType fTy = func.getFunctionType();
|
|
mlir::Value rankVal =
|
|
builder.createIntegerConstant(loc, fTy.getInput(2), rank);
|
|
llvm::SmallVector<mlir::Value> args{
|
|
fir::runtime::createArguments(builder, loc, fTy, desc, mold, rankVal)};
|
|
builder.create<fir::CallOp>(loc, func, args);
|
|
}
|
|
|
|
void fir::runtime::genAllocatableSetBounds(fir::FirOpBuilder &builder,
|
|
mlir::Location loc, mlir::Value desc,
|
|
mlir::Value dimIndex,
|
|
mlir::Value lowerBound,
|
|
mlir::Value upperBound) {
|
|
mlir::func::FuncOp func{
|
|
fir::runtime::getRuntimeFunc<mkRTKey(AllocatableSetBounds)>(loc,
|
|
builder)};
|
|
mlir::FunctionType fTy{func.getFunctionType()};
|
|
llvm::SmallVector<mlir::Value> args{fir::runtime::createArguments(
|
|
builder, loc, fTy, desc, dimIndex, lowerBound, upperBound)};
|
|
builder.create<fir::CallOp>(loc, func, args);
|
|
}
|
|
|
|
void fir::runtime::genAllocatableAllocate(fir::FirOpBuilder &builder,
|
|
mlir::Location loc, mlir::Value desc,
|
|
mlir::Value hasStat,
|
|
mlir::Value errMsg) {
|
|
mlir::func::FuncOp func{
|
|
fir::runtime::getRuntimeFunc<mkRTKey(AllocatableAllocate)>(loc, builder)};
|
|
mlir::FunctionType fTy{func.getFunctionType()};
|
|
mlir::Value sourceFile{fir::factory::locationToFilename(builder, loc)};
|
|
mlir::Value sourceLine{
|
|
fir::factory::locationToLineNo(builder, loc, fTy.getInput(4))};
|
|
if (!hasStat)
|
|
hasStat = builder.createBool(loc, false);
|
|
if (!errMsg) {
|
|
mlir::Type boxNoneTy = fir::BoxType::get(builder.getNoneType());
|
|
errMsg = builder.create<fir::AbsentOp>(loc, boxNoneTy).getResult();
|
|
}
|
|
llvm::SmallVector<mlir::Value> args{fir::runtime::createArguments(
|
|
builder, loc, fTy, desc, hasStat, errMsg, sourceFile, sourceLine)};
|
|
builder.create<fir::CallOp>(loc, func, args);
|
|
}
|