unittests/tools/llvm-exegesis/PowerPC/AnalysisTest.cpp

   1 //===-- AnalysisTest.cpp ---------------------------------------*- C++ -*-===//
   2 //
   3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
   4 // See https://llvm.org/LICENSE.txt for license information.
   5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8
   9 #include "Analysis.h"
  10
  11 #include <cassert>
  12 #include <memory>
  13
  14 #include "llvm/Support/TargetRegistry.h"
  15 #include "llvm/Support/TargetSelect.h"
  16 #include "gmock/gmock.h"
  17 #include "gtest/gtest.h"
  18
  19 namespace llvm{
  20 namespace exegesis {
  21 namespace {
  22
  23 using testing::Pair;
  24 using testing::UnorderedElementsAre;
  25
  26 class AnalysisTest : public ::testing::Test {
  27 protected:
  28   AnalysisTest() {
  29     const std::string TT = "powerpc64le-unknown-linux";
  30     std::string error;
  31     const Target *const TheTarget = TargetRegistry::lookupTarget(TT, error);
  32     if (!TheTarget) {
  33       errs() << error << "\n";
  34       return;
  35     }
  36     STI.reset(TheTarget->createMCSubtargetInfo(TT, "pwr9", ""));
  37
  38     // Compute the ProxResIdx of ports uses in tests.
  39     const auto &SM = STI->getSchedModel();
  40     for (unsigned I = 0, E = SM.getNumProcResourceKinds(); I < E; ++I) {
  41       const std::string Name = SM.getProcResource(I)->Name;
  42       if (Name == "ALU") {
  43         ALUIdx = I;
  44       } else if (Name == "ALUE") {
  45         ALUEIdx = I;
  46       } else if (Name == "ALUO") {
  47         ALUOIdx = I;
  48       } else if (Name == "IP_AGEN") {
  49         IPAGENIdx = I;
  50       }
  51     }
  52     EXPECT_NE(ALUIdx, 0);
  53     EXPECT_NE(ALUEIdx, 0);
  54     EXPECT_NE(ALUOIdx, 0);
  55     EXPECT_NE(IPAGENIdx, 0);
  56   }
  57
  58   static void SetUpTestCase() {
  59     LLVMInitializePowerPCTargetInfo();
  60     LLVMInitializePowerPCTarget();
  61     LLVMInitializePowerPCTargetMC();
  62   }
  63
  64 protected:
  65   std::unique_ptr<const MCSubtargetInfo> STI;
  66   uint16_t ALUIdx = 0;
  67   uint16_t ALUEIdx = 0;
  68   uint16_t ALUOIdx = 0;
  69   uint16_t IPAGENIdx = 0;
  70 };
  71
  72 TEST_F(AnalysisTest, ComputeIdealizedProcResPressure_2ALU) {
  73   const auto Pressure =
  74       computeIdealizedProcResPressure(STI->getSchedModel(), {{ALUIdx, 2}});
  75   EXPECT_THAT(Pressure, UnorderedElementsAre(Pair(ALUIdx, 2.0)));
  76 }
  77
  78 TEST_F(AnalysisTest, ComputeIdealizedProcResPressure_1ALUE) {
  79   const auto Pressure =
  80       computeIdealizedProcResPressure(STI->getSchedModel(), {{ALUEIdx, 2}});
  81   EXPECT_THAT(Pressure, UnorderedElementsAre(Pair(ALUEIdx, 2.0)));
  82 }
  83
  84 TEST_F(AnalysisTest, ComputeIdealizedProcResPressure_1ALU1IPAGEN) {
  85   const auto Pressure =
  86       computeIdealizedProcResPressure(STI->getSchedModel(), {{ALUIdx, 1}, {IPAGENIdx, 1}});
  87   EXPECT_THAT(Pressure, UnorderedElementsAre(Pair(ALUIdx, 1.0),Pair(IPAGENIdx, 1)));
  88 }
  89 } // namespace
  90 } // namespace exegesis
  91 } // namespace llvm