CrossToolHelpers.h

//===-- Tools/CrossToolHelpers.h --------------------------------- *-C++-*-===//
//
// 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
//
//===----------------------------------------------------------------------===//
// A header file for containing functionallity that is used across Flang tools,
// such as helper functions which apply or generate information needed accross
// tools like bbc and flang.
//===----------------------------------------------------------------------===//
 
#ifndef FORTRAN_TOOLS_CROSS_TOOL_HELPERS_H
#define FORTRAN_TOOLS_CROSS_TOOL_HELPERS_H
 
#include "flang/Frontend/CodeGenOptions.h"
#include "flang/Support/FPMaxminBehavior.h"
#include "flang/Support/LangOptions.h"
#include "flang/Support/MathOptionsBase.h"
#include <cstdint>
 
#include "mlir/Dialect/OpenMP/OpenMPDialect.h"
#include "mlir/Pass/PassRegistry.h"
#include "llvm/ADT/SmallVector.h"
#include "llvm/Frontend/Debug/Options.h"
#include "llvm/Passes/OptimizationLevel.h"
 
// Flang Extension Point Callbacks
class FlangEPCallBacks {
public:
  void registerFIROptEarlyEPCallbacks(
      const std::function<void(mlir::PassManager &, llvm::OptimizationLevel)>
          &C) {
    FIROptEarlyEPCallbacks.push_back(C);
  }
 
  void registerFIRInlinerCallback(
      const std::function<void(mlir::PassManager &, llvm::OptimizationLevel)>
          &C) {
    FIRInlinerCallback.push_back(C);
  }
 
  void registerFIROptLastEPCallbacks(
      const std::function<void(mlir::PassManager &, llvm::OptimizationLevel)>
          &C) {
    FIROptLastEPCallbacks.push_back(C);
  }
 
  void registerHLFIROptEarlyEPCallbacks(
      const std::function<void(mlir::PassManager &, llvm::OptimizationLevel)>
          &C) {
    HLFIROptEarlyEPCallbacks.push_back(C);
  }
 
  void registerHLFIROptLastEPCallbacks(
      const std::function<void(mlir::PassManager &, llvm::OptimizationLevel)>
          &C) {
    HLFIROptLastEPCallbacks.push_back(C);
  }
 
  void invokeFIROptEarlyEPCallbacks(
      mlir::PassManager &pm, llvm::OptimizationLevel optLevel) {
    for (auto &C : FIROptEarlyEPCallbacks)
      C(pm, optLevel);
  };
 
  void invokeFIRInlinerCallback(
      mlir::PassManager &pm, llvm::OptimizationLevel optLevel) {
    for (auto &C : FIRInlinerCallback)
      C(pm, optLevel);
  };
 
  void invokeFIROptLastEPCallbacks(
      mlir::PassManager &pm, llvm::OptimizationLevel optLevel) {
    for (auto &C : FIROptLastEPCallbacks)
      C(pm, optLevel);
  };
 
  void invokeHLFIROptEarlyEPCallbacks(
      mlir::PassManager &pm, llvm::OptimizationLevel optLevel) const {
    for (auto &C : HLFIROptEarlyEPCallbacks)
      C(pm, optLevel);
  };
 
  void invokeHLFIROptLastEPCallbacks(
      mlir::PassManager &pm, llvm::OptimizationLevel optLevel) const {
    for (auto &C : HLFIROptLastEPCallbacks)
      C(pm, optLevel);
  };
 
private:
  llvm::SmallVector<
      std::function<void(mlir::PassManager &, llvm::OptimizationLevel)>, 1>
      FIROptEarlyEPCallbacks;
 
  llvm::SmallVector<
      std::function<void(mlir::PassManager &, llvm::OptimizationLevel)>, 1>
      FIRInlinerCallback;
 
  llvm::SmallVector<
      std::function<void(mlir::PassManager &, llvm::OptimizationLevel)>, 1>
      FIROptLastEPCallbacks;
 
  llvm::SmallVector<
      std::function<void(mlir::PassManager &, llvm::OptimizationLevel)>, 1>
      HLFIROptEarlyEPCallbacks;
 
  llvm::SmallVector<
      std::function<void(mlir::PassManager &, llvm::OptimizationLevel)>, 1>
      HLFIROptLastEPCallbacks;
};

struct MLIRToLLVMPassPipelineConfig : public FlangEPCallBacks {
  explicit MLIRToLLVMPassPipelineConfig(llvm::OptimizationLevel level) {
    OptLevel = level;
  }
  explicit MLIRToLLVMPassPipelineConfig(llvm::OptimizationLevel level,
      const Fortran::frontend::CodeGenOptions &opts,
      const Fortran::common::MathOptionsBase &mathOpts) {
    OptLevel = level;
    StackArrays = opts.StackArrays;
    EnableSafeTrampoline = opts.EnableSafeTrampoline;
    Underscoring = opts.Underscoring;
    LoopVersioning = opts.LoopVersioning;
    DebugInfo = opts.getDebugInfo();
    AliasAnalysis = opts.AliasAnalysis;
    FramePointerKind = opts.getFramePointer();
    // The logic for setting these attributes is intended to match the logic
    // used in Clang.
    NoInfsFPMath = mathOpts.getNoHonorInfs();
    NoNaNsFPMath = mathOpts.getNoHonorNaNs();
    ApproxFuncFPMath = mathOpts.getApproxFunc();
    NoSignedZerosFPMath = mathOpts.getNoSignedZeros();
    UnsafeFPMath = mathOpts.getAssociativeMath() &&
        mathOpts.getReciprocalMath() && NoSignedZerosFPMath &&
        ApproxFuncFPMath && mathOpts.getFPContractEnabled();
    Reciprocals = opts.Reciprocals;
    PreferVectorWidth = opts.PreferVectorWidth;
    UseSampleProfile = !opts.SampleProfileFile.empty();
    DebugInfoForProfiling = opts.DebugInfoForProfiling;
    if (opts.InstrumentFunctions) {
      InstrumentFunctionEntry = "__cyg_profile_func_enter";
      InstrumentFunctionExit = "__cyg_profile_func_exit";
    }
    DwarfVersion = opts.DwarfVersion;
    SplitDwarfFile = opts.SplitDwarfFile;
    DwarfDebugFlags = opts.DwarfDebugFlags;
  }
 
  llvm::OptimizationLevel OptLevel; 
  bool StackArrays = false; 
  bool EnableSafeTrampoline{false}; 
  bool Underscoring = true; 
  bool LoopVersioning = false; 
  bool AliasAnalysis = false; 
  llvm::codegenoptions::DebugInfoKind DebugInfo =
      llvm::codegenoptions::NoDebugInfo; 
  llvm::FramePointerKind FramePointerKind =
      llvm::FramePointerKind::None; 
  unsigned VScaleMin = 0; 
  unsigned VScaleMax = 0; 
  bool NoInfsFPMath = false; 
  bool NoNaNsFPMath = false; 
  bool ApproxFuncFPMath = false; 
  bool NoSignedZerosFPMath =
      false; 
  bool UnsafeFPMath = false; 
  std::string Reciprocals = ""; 
  std::string PreferVectorWidth = ""; 
  bool NSWOnLoopVarInc = true; 
  bool EnableOpenACC = false; 
  bool EnableOpenMP = false; 
  bool EnableOpenMPIsTargetDevice =
      false; 
  bool UseSampleProfile = false; 
  bool DebugInfoForProfiling = false; 
  bool EnableOpenMPSimd = false; 
  bool SkipConvertComplexPow = false; 
  std::string InstrumentFunctionEntry =
      ""; 
  std::string InstrumentFunctionExit =
      ""; 
  Fortran::frontend::CodeGenOptions::ComplexRangeKind ComplexRange =
      Fortran::frontend::CodeGenOptions::ComplexRangeKind::
          CX_Full; 
  int32_t DwarfVersion = 0; 
  std::string SplitDwarfFile = ""; 
  std::string DwarfDebugFlags = ""; 
  Fortran::common::FPMaxminBehavior fpMaxminBehavior =
      Fortran::common::FPMaxminBehavior::Legacy;
};

[[maybe_unused]] static mlir::omp::OffloadModuleOpts makeOffloadModuleOpts(
    Fortran::common::LangOptions &Opts) {
  return mlir::omp::OffloadModuleOpts(Opts.OpenMPTargetDebug,
      Opts.OpenMPTeamSubscription, Opts.OpenMPThreadSubscription,
      Opts.OpenMPNoThreadState, Opts.OpenMPNoNestedParallelism,
      Opts.OpenMPIsTargetDevice, Opts.OpenMPIsGPU, Opts.OpenMPForceUSM,
      Opts.OpenMPVersion, Opts.OMPHostIRFile, Opts.OMPTargetTriples,
      Opts.NoGPULib);
}
 
[[maybe_unused]] static void setOpenMPIntegerWrapAround(
    mlir::ModuleOp module, bool value) {
  module.getOperation()->setAttr("omp.integer_wrap_around",
      mlir::omp::IntegerWrapAroundAttr::get(module.getContext(), value));
}
 
#endif // FORTRAN_TOOLS_CROSS_TOOL_HELPERS_H