22544e2a54
The implemented logic matches the logic used for Clang in emitting these attributes. Although it's hoped that function attributes won't be needed in the future (vs using fast math flags in individual IR instructions), there are codegen differences currently with/without these attributes, as can be seen in issues like #79257 or by hacking Clang to avoid producing these attributes and observing codegen changes.
98 lines
3.5 KiB
C++
98 lines
3.5 KiB
C++
//===- FunctionAttr.cpp ---------------------------------------------------===//
|
|
//
|
|
// 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
/// \file
|
|
/// This is a generic pass for adding attributes to functions.
|
|
//===----------------------------------------------------------------------===//
|
|
#include "flang/Optimizer/Transforms/Passes.h"
|
|
#include "mlir/Dialect/LLVMIR/LLVMAttrs.h"
|
|
#include "mlir/Dialect/LLVMIR/LLVMDialect.h"
|
|
|
|
namespace fir {
|
|
#define GEN_PASS_DECL_FUNCTIONATTR
|
|
#define GEN_PASS_DEF_FUNCTIONATTR
|
|
#include "flang/Optimizer/Transforms/Passes.h.inc"
|
|
} // namespace fir
|
|
|
|
#define DEBUG_TYPE "func-attr"
|
|
|
|
namespace {
|
|
|
|
class FunctionAttrPass : public fir::impl::FunctionAttrBase<FunctionAttrPass> {
|
|
public:
|
|
FunctionAttrPass(const fir::FunctionAttrOptions &options) {
|
|
framePointerKind = options.framePointerKind;
|
|
noInfsFPMath = options.noInfsFPMath;
|
|
noNaNsFPMath = options.noNaNsFPMath;
|
|
approxFuncFPMath = options.approxFuncFPMath;
|
|
noSignedZerosFPMath = options.noSignedZerosFPMath;
|
|
unsafeFPMath = options.unsafeFPMath;
|
|
}
|
|
FunctionAttrPass() {}
|
|
void runOnOperation() override;
|
|
};
|
|
|
|
} // namespace
|
|
|
|
void FunctionAttrPass::runOnOperation() {
|
|
LLVM_DEBUG(llvm::dbgs() << "=== Begin " DEBUG_TYPE " ===\n");
|
|
mlir::func::FuncOp func = getOperation();
|
|
|
|
LLVM_DEBUG(llvm::dbgs() << "Func-name:" << func.getSymName() << "\n");
|
|
|
|
mlir::MLIRContext *context = &getContext();
|
|
if (framePointerKind != mlir::LLVM::framePointerKind::FramePointerKind::None)
|
|
func->setAttr("frame_pointer", mlir::LLVM::FramePointerKindAttr::get(
|
|
context, framePointerKind));
|
|
|
|
auto llvmFuncOpName =
|
|
mlir::OperationName(mlir::LLVM::LLVMFuncOp::getOperationName(), context);
|
|
if (noInfsFPMath)
|
|
func->setAttr(
|
|
mlir::LLVM::LLVMFuncOp::getNoInfsFpMathAttrName(llvmFuncOpName),
|
|
mlir::BoolAttr::get(context, true));
|
|
if (noNaNsFPMath)
|
|
func->setAttr(
|
|
mlir::LLVM::LLVMFuncOp::getNoNansFpMathAttrName(llvmFuncOpName),
|
|
mlir::BoolAttr::get(context, true));
|
|
if (approxFuncFPMath)
|
|
func->setAttr(
|
|
mlir::LLVM::LLVMFuncOp::getApproxFuncFpMathAttrName(llvmFuncOpName),
|
|
mlir::BoolAttr::get(context, true));
|
|
if (noSignedZerosFPMath)
|
|
func->setAttr(
|
|
mlir::LLVM::LLVMFuncOp::getNoSignedZerosFpMathAttrName(llvmFuncOpName),
|
|
mlir::BoolAttr::get(context, true));
|
|
if (unsafeFPMath)
|
|
func->setAttr(
|
|
mlir::LLVM::LLVMFuncOp::getUnsafeFpMathAttrName(llvmFuncOpName),
|
|
mlir::BoolAttr::get(context, true));
|
|
|
|
LLVM_DEBUG(llvm::dbgs() << "=== End " DEBUG_TYPE " ===\n");
|
|
}
|
|
|
|
std::unique_ptr<mlir::Pass> fir::createFunctionAttrPass(
|
|
fir::FunctionAttrTypes &functionAttr, bool noInfsFPMath, bool noNaNsFPMath,
|
|
bool approxFuncFPMath, bool noSignedZerosFPMath, bool unsafeFPMath) {
|
|
FunctionAttrOptions opts;
|
|
// Frame pointer
|
|
opts.framePointerKind = functionAttr.framePointerKind;
|
|
opts.noInfsFPMath = noInfsFPMath;
|
|
opts.noNaNsFPMath = noNaNsFPMath;
|
|
opts.approxFuncFPMath = approxFuncFPMath;
|
|
opts.noSignedZerosFPMath = noSignedZerosFPMath;
|
|
opts.unsafeFPMath = unsafeFPMath;
|
|
|
|
return std::make_unique<FunctionAttrPass>(opts);
|
|
}
|
|
|
|
std::unique_ptr<mlir::Pass> fir::createFunctionAttrPass() {
|
|
return std::make_unique<FunctionAttrPass>();
|
|
}
|