1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
|
//===-- 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 =
mlir::dyn_cast<fir::ClassType>(fir::dyn_cast_ptrEleTy(from.getType()));
mlir::Type derivedType = fir::unwrapInnerType(clTy.getEleTy());
declaredTypeDesc =
fir::TypeDescOp::create(builder, 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 fir::CallOp::create(builder, 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)};
fir::CallOp::create(builder, 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)};
fir::CallOp::create(builder, 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 asyncObject = builder.createNullConstant(loc);
mlir::Value sourceFile{fir::factory::locationToFilename(builder, loc)};
mlir::Value sourceLine{
fir::factory::locationToLineNo(builder, loc, fTy.getInput(5))};
if (!hasStat)
hasStat = builder.createBool(loc, false);
if (!errMsg) {
mlir::Type boxNoneTy = fir::BoxType::get(builder.getNoneType());
errMsg = fir::AbsentOp::create(builder, loc, boxNoneTy).getResult();
}
llvm::SmallVector<mlir::Value> args{
fir::runtime::createArguments(builder, loc, fTy, desc, asyncObject,
hasStat, errMsg, sourceFile, sourceLine)};
fir::CallOp::create(builder, loc, func, args);
}
|
