9f44d5d9d0
The runtime API for copy-in copy-out currently only has an entry only for the copy-out. This entry has a "skipInit" boolean that is never set to false by lowering and it does not deal with the deallocation of the temporary. The generated code was a mix of inline code and runtime calls This is not a big deal, but this is unneeded compiler and generated code complexity. With assumed-rank, it is also more cumbersome to establish a temporary descriptor. Instead, this patch: - Adds a CopyInAssignment API that deals with establishing the temporary descriptor and does the copy. - Removes unused arg to CopyOutAssign, and pushes destruction/deallocation responsibility inside it. Note that this runtime API are still not responsible for deciding the need of copying-in and out. This is kept as a separate runtime call to IsContiguous, which is easier to inline/replace by inline code with the hope of removing the copy-in/out calls after user function inlining. @vzakhari has already shown that always inlining all the copy part increase Fortran compilation time due to loop optimization attempts for loops that are known to have little optimization profitability (the variable being copied from and to is not contiguous).
98 lines
4.7 KiB
C++
98 lines
4.7 KiB
C++
//===-- Assign.cpp -- generate assignment 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/Assign.h"
|
|
#include "flang/Optimizer/Builder/FIRBuilder.h"
|
|
#include "flang/Optimizer/Builder/Runtime/RTBuilder.h"
|
|
#include "flang/Runtime/assign.h"
|
|
|
|
using namespace Fortran::runtime;
|
|
|
|
void fir::runtime::genAssign(fir::FirOpBuilder &builder, mlir::Location loc,
|
|
mlir::Value destBox, mlir::Value sourceBox) {
|
|
auto func = fir::runtime::getRuntimeFunc<mkRTKey(Assign)>(loc, builder);
|
|
auto fTy = func.getFunctionType();
|
|
auto sourceFile = fir::factory::locationToFilename(builder, loc);
|
|
auto sourceLine =
|
|
fir::factory::locationToLineNo(builder, loc, fTy.getInput(3));
|
|
auto args = fir::runtime::createArguments(builder, loc, fTy, destBox,
|
|
sourceBox, sourceFile, sourceLine);
|
|
builder.create<fir::CallOp>(loc, func, args);
|
|
}
|
|
|
|
void fir::runtime::genAssignPolymorphic(fir::FirOpBuilder &builder,
|
|
mlir::Location loc, mlir::Value destBox,
|
|
mlir::Value sourceBox) {
|
|
auto func =
|
|
fir::runtime::getRuntimeFunc<mkRTKey(AssignPolymorphic)>(loc, builder);
|
|
auto fTy = func.getFunctionType();
|
|
auto sourceFile = fir::factory::locationToFilename(builder, loc);
|
|
auto sourceLine =
|
|
fir::factory::locationToLineNo(builder, loc, fTy.getInput(3));
|
|
auto args = fir::runtime::createArguments(builder, loc, fTy, destBox,
|
|
sourceBox, sourceFile, sourceLine);
|
|
builder.create<fir::CallOp>(loc, func, args);
|
|
}
|
|
|
|
void fir::runtime::genAssignExplicitLengthCharacter(fir::FirOpBuilder &builder,
|
|
mlir::Location loc,
|
|
mlir::Value destBox,
|
|
mlir::Value sourceBox) {
|
|
auto func =
|
|
fir::runtime::getRuntimeFunc<mkRTKey(AssignExplicitLengthCharacter)>(
|
|
loc, builder);
|
|
auto fTy = func.getFunctionType();
|
|
auto sourceFile = fir::factory::locationToFilename(builder, loc);
|
|
auto sourceLine =
|
|
fir::factory::locationToLineNo(builder, loc, fTy.getInput(3));
|
|
auto args = fir::runtime::createArguments(builder, loc, fTy, destBox,
|
|
sourceBox, sourceFile, sourceLine);
|
|
builder.create<fir::CallOp>(loc, func, args);
|
|
}
|
|
|
|
void fir::runtime::genAssignTemporary(fir::FirOpBuilder &builder,
|
|
mlir::Location loc, mlir::Value destBox,
|
|
mlir::Value sourceBox) {
|
|
auto func =
|
|
fir::runtime::getRuntimeFunc<mkRTKey(AssignTemporary)>(loc, builder);
|
|
auto fTy = func.getFunctionType();
|
|
auto sourceFile = fir::factory::locationToFilename(builder, loc);
|
|
auto sourceLine =
|
|
fir::factory::locationToLineNo(builder, loc, fTy.getInput(3));
|
|
auto args = fir::runtime::createArguments(builder, loc, fTy, destBox,
|
|
sourceBox, sourceFile, sourceLine);
|
|
builder.create<fir::CallOp>(loc, func, args);
|
|
}
|
|
|
|
void fir::runtime::genCopyInAssign(fir::FirOpBuilder &builder,
|
|
mlir::Location loc, mlir::Value destBox,
|
|
mlir::Value sourceBox) {
|
|
auto func = fir::runtime::getRuntimeFunc<mkRTKey(CopyInAssign)>(loc, builder);
|
|
auto fTy = func.getFunctionType();
|
|
auto sourceFile = fir::factory::locationToFilename(builder, loc);
|
|
auto sourceLine =
|
|
fir::factory::locationToLineNo(builder, loc, fTy.getInput(3));
|
|
auto args = fir::runtime::createArguments(builder, loc, fTy, destBox,
|
|
sourceBox, sourceFile, sourceLine);
|
|
builder.create<fir::CallOp>(loc, func, args);
|
|
}
|
|
|
|
void fir::runtime::genCopyOutAssign(fir::FirOpBuilder &builder,
|
|
mlir::Location loc, mlir::Value destBox,
|
|
mlir::Value sourceBox) {
|
|
auto func =
|
|
fir::runtime::getRuntimeFunc<mkRTKey(CopyOutAssign)>(loc, builder);
|
|
auto fTy = func.getFunctionType();
|
|
auto sourceFile = fir::factory::locationToFilename(builder, loc);
|
|
auto sourceLine =
|
|
fir::factory::locationToLineNo(builder, loc, fTy.getInput(3));
|
|
auto args = fir::runtime::createArguments(builder, loc, fTy, destBox,
|
|
sourceBox, sourceFile, sourceLine);
|
|
builder.create<fir::CallOp>(loc, func, args);
|
|
}
|