rocm-systems/rocclr/compiler/lib/backends/common/optimizer.cpp

//
// Copyright (c) 2008 Advanced Micro Devices, Inc. All rights reserved.
//
#include "top.hpp"
#include "optimizer.hpp"
#include "opt_level.hpp"
#include "os/os.hpp"
#include "utils/bif_section_labels.hpp"
#include "utils/libUtils.h"
#include "utils/options.hpp"

#include "llvm/DataLayout.h"
#include "llvm/Analysis/Passes.h"
#include "llvm/Support/raw_ostream.h"
#include "llvm/LinkAllPasses.h"
#include <cassert>
#include <sstream>
using namespace amdcl;
using namespace llvm;

static OptLevel* getOptLevel(amd::option::Options* Options, bool isGPU) {
  switch(Options->oVariables->OptLevel) {
    case amd::option::OPT_O0:
      return (isGPU) ? new GPUO0OptLevel(Options) : new O0OptLevel(Options);
    case amd::option::OPT_O1:
      return new O1OptLevel(Options);
    default:
      assert(!"Found an invalid optimization level!");
    case amd::option::OPT_O2:
      return new O2OptLevel(Options);
    case amd::option::OPT_O3:
      return new O3OptLevel(Options);
    case amd::option::OPT_O4:
      return new O4OptLevel(Options);
    case amd::option::OPT_OS:
      return new OsOptLevel(Options);
  }
  assert(!"Unreachable!");
  return NULL;
}
int
CPUOptimizer::preOptimizer(llvm::Module* M)
{
    llvm::PassManager Passes;
    Passes.add(new llvm::DataLayout(M));

    Passes.add(createAMDExportKernelNaturePass());

    Passes.run(*M);

    return 0;
}

  int
CPUOptimizer::optimize(llvm::Module *input)
{
  if (!input) {
    return 1;
  }
  int ret = 0;
  uint64_t start_time = 0ULL, time_opt = 0ULL;
  llvmbinary_ = input;
  setWholeProgram(true);

  setGPU(false);
  if (Options()->oVariables->EnableBuildTiming) {
    start_time = amd::Os::timeNanos();
  }
  ret = preOptimizer(LLVMBinary());
  OptLevel* cpuOpt = getOptLevel(Options(), false);
  if (Options()->oVariables->EnableBuildTiming) {
    time_opt = amd::Os::timeNanos();
  }
  ret = cpuOpt->optimize(Elf(), LLVMBinary(), false);
  if (Options()->oVariables->EnableBuildTiming) {
    time_opt = amd::Os::timeNanos() - time_opt;
    std::stringstream tmp_ss;
    tmp_ss << "    LLVM Opt time: "
      << time_opt/1000ULL
      << "us\n";
    appendLogToCL(CL(), tmp_ss.str());
  }
  delete cpuOpt;

  if ( ret ) {
    BuildLog() += "Internal Error: optimizer failed!\n";
    return 1;
  }
  if (Options()->isDumpFlagSet(amd::option::DUMP_BC_OPTIMIZED)) {
    std::string MyErrorInfo;
    std::string fileName = Options()->getDumpFileName("_optimized.bc");
    raw_fd_ostream outs (fileName.c_str(), MyErrorInfo, raw_fd_ostream::F_Binary);
    // FIXME: Need to add this to the elf binary!
    if (MyErrorInfo.empty())
      WriteBitcodeToFile(LLVMBinary(), outs);
    else
      printf(MyErrorInfo.c_str());
  }
  return ret;
}

  int
GPUOptimizer::optimize(llvm::Module *input)
{
  if (!input) {
    return 1;
  }
  int ret = 0;
  uint64_t start_time = 0ULL, time_opt = 0ULL;
  llvmbinary_ = input;

  setGPU(true);
  setWholeProgram(true);
#ifdef WITH_TARGET_HSAIL
  if (isHSAILTarget(Elf()->target)) {
    if (Options()->NumAvailGPRs == -1)
      Options()->NumAvailGPRs = 128; // Default HSAIL number of GPRs
    if (hookup_.amdoptions.NumAvailGPRs == ~0u)
      hookup_.amdoptions.NumAvailGPRs = Options()->NumAvailGPRs;
  }
#endif
  OptLevel* gpuOpt = getOptLevel(Options(), true);
  if (Options()->oVariables->EnableBuildTiming) {
    time_opt = amd::Os::timeNanos();
  }
  ret = gpuOpt->optimize(Elf(), LLVMBinary(), true);
  if (Options()->oVariables->EnableBuildTiming) {
    time_opt = amd::Os::timeNanos() - time_opt;
    std::stringstream tmp_ss;
    tmp_ss << "    LLVM Opt time: "
      << time_opt/1000ULL
      << "us\n";
    appendLogToCL(CL(), tmp_ss.str());
  }
  delete gpuOpt;

  if ( ret ) {
    BuildLog() += "Internal Error: optimizer failed!\n";
    return 1;
  }
  if (Options()->isDumpFlagSet(amd::option::DUMP_BC_OPTIMIZED)) {
    std::string MyErrorInfo;
    std::string fileName = Options()->getDumpFileName("_optimized.bc");
    raw_fd_ostream outs (fileName.c_str(), MyErrorInfo, raw_fd_ostream::F_Binary);
    // FIXME: Need to add this to the elf binary!
    if (MyErrorInfo.empty())
      WriteBitcodeToFile(LLVMBinary(), outs);
    else
      printf(MyErrorInfo.c_str());
  }
  return ret;
}