b815a3942a
Move non-common files from FortranCommon to FortranSupport (analogous to
LLVMSupport) such that
* declarations and definitions that are only used by the Flang compiler,
but not by the runtime, are moved to FortranSupport
* declarations and definitions that are used by both ("common"), the
compiler and the runtime, remain in FortranCommon
* generic STL-like/ADT/utility classes and algorithms remain in
FortranCommon
This allows a for cleaner separation between compiler and runtime
components, which are compiled differently. For instance, runtime
sources must not use STL's `<optional>` which causes problems with CUDA
support. Instead, the surrogate header `flang/Common/optional.h` must be
used. This PR fixes this for `fast-int-sel.h`.
Declarations in include/Runtime are also used by both, but are
header-only. `ISO_Fortran_binding_wrapper.h`, a header used by compiler
and runtime, is also moved into FortranCommon.
52 lines
1.6 KiB
C++
52 lines
1.6 KiB
C++
//===- StackReclaim.cpp -- Insert stacksave/stackrestore in region --------===//
|
|
//
|
|
// 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/FIRBuilder.h"
|
|
#include "flang/Optimizer/Dialect/FIRDialect.h"
|
|
#include "flang/Optimizer/Dialect/FIROps.h"
|
|
#include "flang/Optimizer/Transforms/Passes.h"
|
|
#include "flang/Support/Fortran.h"
|
|
#include "mlir/Dialect/LLVMIR/LLVMDialect.h"
|
|
#include "mlir/IR/Matchers.h"
|
|
#include "mlir/Pass/Pass.h"
|
|
|
|
namespace fir {
|
|
#define GEN_PASS_DEF_STACKRECLAIM
|
|
#include "flang/Optimizer/Transforms/Passes.h.inc"
|
|
} // namespace fir
|
|
|
|
using namespace mlir;
|
|
|
|
namespace {
|
|
|
|
class StackReclaimPass : public fir::impl::StackReclaimBase<StackReclaimPass> {
|
|
public:
|
|
using StackReclaimBase<StackReclaimPass>::StackReclaimBase;
|
|
|
|
void runOnOperation() override;
|
|
};
|
|
} // namespace
|
|
|
|
void StackReclaimPass::runOnOperation() {
|
|
auto *op = getOperation();
|
|
fir::FirOpBuilder builder(op, fir::getKindMapping(op));
|
|
|
|
op->walk([&](fir::DoLoopOp loopOp) {
|
|
mlir::Location loc = loopOp.getLoc();
|
|
|
|
if (!loopOp.getRegion().getOps<fir::AllocaOp>().empty()) {
|
|
builder.setInsertionPointToStart(&loopOp.getRegion().front());
|
|
mlir::Value sp = builder.genStackSave(loc);
|
|
|
|
auto *terminator = loopOp.getRegion().back().getTerminator();
|
|
builder.setInsertionPoint(terminator);
|
|
builder.genStackRestore(loc, sp);
|
|
}
|
|
});
|
|
}
|