unittests/tools/llvm-exegesis/X86/RegisterAliasingTest.cpp

   1 //===-- RegisterAliasingTest.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
  10 #include "RegisterAliasing.h"
  11
  12 #include <cassert>
  13 #include <memory>
  14
  15 #include "TestBase.h"
  16 #include "X86InstrInfo.h"
  17 #include "llvm/Support/TargetRegistry.h"
  18 #include "llvm/Support/TargetSelect.h"
  19 #include "gmock/gmock.h"
  20 #include "gtest/gtest.h"
  21
  22 namespace llvm {
  23 namespace exegesis {
  24 namespace {
  25
  26 class RegisterAliasingTest : public X86TestBase {};
  27
  28 TEST_F(RegisterAliasingTest, TrackSimpleRegister) {
  29   const auto &RegInfo = State.getRegInfo();
  30   const RegisterAliasingTracker tracker(RegInfo, X86::EAX);
  31   std::set<MCPhysReg> ActualAliasedRegisters;
  32   for (unsigned I : tracker.aliasedBits().set_bits())
  33     ActualAliasedRegisters.insert(static_cast<MCPhysReg>(I));
  34   const std::set<MCPhysReg> ExpectedAliasedRegisters = {
  35       X86::AL, X86::AH, X86::AX, X86::EAX, X86::HAX, X86::RAX};
  36   ASSERT_THAT(ActualAliasedRegisters, ExpectedAliasedRegisters);
  37   for (MCPhysReg aliased : ExpectedAliasedRegisters) {
  38     ASSERT_THAT(tracker.getOrigin(aliased), X86::EAX);
  39   }
  40 }
  41
  42 TEST_F(RegisterAliasingTest, TrackRegisterClass) {
  43   // The alias bits for GR8_ABCD_LRegClassID are the union of the alias bits for
  44   // AL, BL, CL and DL.
  45   const auto &RegInfo = State.getRegInfo();
  46   const BitVector NoReservedReg(RegInfo.getNumRegs());
  47
  48   const RegisterAliasingTracker RegClassTracker(
  49       RegInfo, NoReservedReg, RegInfo.getRegClass(X86::GR8_ABCD_LRegClassID));
  50
  51   BitVector sum(RegInfo.getNumRegs());
  52   sum |= RegisterAliasingTracker(RegInfo, X86::AL).aliasedBits();
  53   sum |= RegisterAliasingTracker(RegInfo, X86::BL).aliasedBits();
  54   sum |= RegisterAliasingTracker(RegInfo, X86::CL).aliasedBits();
  55   sum |= RegisterAliasingTracker(RegInfo, X86::DL).aliasedBits();
  56
  57   ASSERT_THAT(RegClassTracker.aliasedBits(), sum);
  58 }
  59
  60 TEST_F(RegisterAliasingTest, TrackRegisterClassCache) {
  61   // Fetching twice the same tracker yields the same pointers.
  62   const auto &RegInfo = State.getRegInfo();
  63   const BitVector NoReservedReg(RegInfo.getNumRegs());
  64   RegisterAliasingTrackerCache Cache(RegInfo, NoReservedReg);
  65   ASSERT_THAT(&Cache.getRegister(X86::AX), &Cache.getRegister(X86::AX));
  66
  67   ASSERT_THAT(&Cache.getRegisterClass(X86::GR8_ABCD_LRegClassID),
  68               &Cache.getRegisterClass(X86::GR8_ABCD_LRegClassID));
  69 }
  70
  71 } // namespace
  72 } // namespace exegesis
  73 } // namespace llvm