JlmOptCommandTests.cpp

Go to the documentation of this file.

/*
 * Copyright 2023 Nico Reißmann <nico.reissmann@gmail.com>
 * See COPYING for terms of redistribution.
 */
 
#include <gtest/gtest.h>
 
#include <jlm/llvm/ir/RvsdgModule.hpp>
#include <jlm/llvm/opt/RvsdgTreePrinter.hpp>
#include <jlm/tooling/Command.hpp>
#include <jlm/util/strfmt.hpp>
 
#include <fstream>
 
TEST(JlmOptCommandTests, TestStatistics)
{
  using namespace jlm::llvm;
  using namespace jlm::tooling;
  using namespace jlm::util;
 
  // Arrange
  FilePath expectedStatisticsDir("/tmp/myStatisticsDir/");
 
  jlm::util::StatisticsCollectorSettings statisticsCollectorSettings(
      { jlm::util::Statistics::Id::AndersenAnalysis },
      expectedStatisticsDir,
      "inputFile");
 
  JlmOptCommandLineOptions commandLineOptions(
      jlm::util::FilePath("inputFile.ll"),
      JlmOptCommandLineOptions::InputFormat::Llvm,
      jlm::util::FilePath("outputFile.ll"),
      JlmOptCommandLineOptions::OutputFormat::Llvm,
      statisticsCollectorSettings,
      RvsdgTreePrinter::Configuration({}),
      { JlmOptCommandLineOptions::OptimizationId::DeadNodeElimination,
        JlmOptCommandLineOptions::OptimizationId::LoopUnrolling },
      false);
 
  JlmOptCommand command("jlm-opt", commandLineOptions);
 
  // Act
  auto receivedCommandLine = command.ToString();
 
  // Assert
  std::string expectedCommandLine = jlm::util::strfmt(
      "jlm-opt ",
      "--input-format=llvm ",
      "--output-format=llvm ",
      "--DeadNodeElimination --LoopUnrolling ",
      "-s " + expectedStatisticsDir.to_str() + " ",
      "--print-andersen-analysis ",
      "-o outputFile.ll ",
      "inputFile.ll");
 
  EXPECT_EQ(receivedCommandLine, expectedCommandLine);
}
 
TEST(JlmOptCommandTests, OptimizationIdToOptimizationTranslation)
{
  using namespace jlm::llvm;
  using namespace jlm::tooling;
  using namespace jlm::util;
 
  // Arrange
  std::vector<JlmOptCommandLineOptions::OptimizationId> optimizationIds;
  for (size_t n =
           static_cast<std::size_t>(JlmOptCommandLineOptions::OptimizationId::FirstEnumValue) + 1;
       n != static_cast<std::size_t>(JlmOptCommandLineOptions::OptimizationId::LastEnumValue);
       n++)
  {
    auto optimizationId = static_cast<JlmOptCommandLineOptions::OptimizationId>(n);
    optimizationIds.emplace_back(optimizationId);
  }
 
  JlmOptCommandLineOptions options(
      FilePath(""),
      JlmOptCommandLineOptions::InputFormat::Llvm,
      FilePath(""),
      JlmOptCommandLineOptions::OutputFormat::Llvm,
      StatisticsCollectorSettings(),
      RvsdgTreePrinter::Configuration({}),
      optimizationIds,
      false);
 
  // Act & Assert
  // terminates on unhandled optimization id
  JlmOptCommand command("jlm-opt", options);
}
 
TEST(JlmOptCommandTests, PrintRvsdgTreeToFile)
{
  using namespace jlm;
 
  // Arrange
  util::FilePath outputFile("/tmp/RvsdgTree");
 
  jlm::llvm::LlvmRvsdgModule rvsdgModule(jlm::util::FilePath(""), "", "");
  util::StatisticsCollector statisticsCollector;
 
  // Act
  tooling::JlmOptCommand::PrintRvsdgModule(
      rvsdgModule,
      outputFile,
      tooling::JlmOptCommandLineOptions::OutputFormat::Tree,
      statisticsCollector);
 
  // Assert
  std::stringstream buffer;
  std::ifstream istream(outputFile.to_str());
  buffer << istream.rdbuf();
 
  EXPECT_EQ(buffer.str(), "RootRegion\n");
}