91 lines
4.3 KiB
C++
91 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);
|
|
}
|