| #include "Target.h" |
| |
| #include <cassert> |
| #include <memory> |
| |
| #include "MCTargetDesc/X86MCTargetDesc.h" |
| #include "llvm/Support/TargetRegistry.h" |
| #include "llvm/Support/TargetSelect.h" |
| #include "gmock/gmock.h" |
| #include "gtest/gtest.h" |
| |
| namespace exegesis { |
| |
| void InitializeX86ExegesisTarget(); |
| |
| namespace { |
| |
| using testing::Gt; |
| using testing::NotNull; |
| using testing::SizeIs; |
| |
| constexpr const char kTriple[] = "x86_64-unknown-linux"; |
| |
| class X86TargetTest : public ::testing::Test { |
| protected: |
| X86TargetTest() |
| : ExegesisTarget_(ExegesisTarget::lookup(llvm::Triple(kTriple))) { |
| EXPECT_THAT(ExegesisTarget_, NotNull()); |
| std::string error; |
| Target_ = llvm::TargetRegistry::lookupTarget(kTriple, error); |
| EXPECT_THAT(Target_, NotNull()); |
| } |
| static void SetUpTestCase() { |
| LLVMInitializeX86TargetInfo(); |
| LLVMInitializeX86Target(); |
| LLVMInitializeX86TargetMC(); |
| InitializeX86ExegesisTarget(); |
| } |
| |
| const llvm::Target *Target_; |
| const ExegesisTarget *const ExegesisTarget_; |
| }; |
| |
| TEST_F(X86TargetTest, SetRegToConstantGPR) { |
| const std::unique_ptr<llvm::MCSubtargetInfo> STI( |
| Target_->createMCSubtargetInfo(kTriple, "core2", "")); |
| const auto Insts = ExegesisTarget_->setRegToConstant(*STI, llvm::X86::EAX); |
| EXPECT_THAT(Insts, SizeIs(1)); |
| EXPECT_EQ(Insts[0].getOpcode(), llvm::X86::MOV32ri); |
| EXPECT_EQ(Insts[0].getOperand(0).getReg(), llvm::X86::EAX); |
| } |
| |
| TEST_F(X86TargetTest, SetRegToConstantXMM_SSE2) { |
| const std::unique_ptr<llvm::MCSubtargetInfo> STI( |
| Target_->createMCSubtargetInfo(kTriple, "core2", "")); |
| const auto Insts = ExegesisTarget_->setRegToConstant(*STI, llvm::X86::XMM1); |
| EXPECT_THAT(Insts, SizeIs(7U)); |
| EXPECT_EQ(Insts[0].getOpcode(), llvm::X86::SUB64ri8); |
| EXPECT_EQ(Insts[1].getOpcode(), llvm::X86::MOV32mi); |
| EXPECT_EQ(Insts[2].getOpcode(), llvm::X86::MOV32mi); |
| EXPECT_EQ(Insts[3].getOpcode(), llvm::X86::MOV32mi); |
| EXPECT_EQ(Insts[4].getOpcode(), llvm::X86::MOV32mi); |
| EXPECT_EQ(Insts[5].getOpcode(), llvm::X86::MOVDQUrm); |
| EXPECT_EQ(Insts[6].getOpcode(), llvm::X86::ADD64ri8); |
| } |
| |
| TEST_F(X86TargetTest, SetRegToConstantXMM_AVX) { |
| const std::unique_ptr<llvm::MCSubtargetInfo> STI( |
| Target_->createMCSubtargetInfo(kTriple, "core2", "+avx")); |
| const auto Insts = ExegesisTarget_->setRegToConstant(*STI, llvm::X86::XMM1); |
| EXPECT_THAT(Insts, SizeIs(7U)); |
| EXPECT_EQ(Insts[0].getOpcode(), llvm::X86::SUB64ri8); |
| EXPECT_EQ(Insts[1].getOpcode(), llvm::X86::MOV32mi); |
| EXPECT_EQ(Insts[2].getOpcode(), llvm::X86::MOV32mi); |
| EXPECT_EQ(Insts[3].getOpcode(), llvm::X86::MOV32mi); |
| EXPECT_EQ(Insts[4].getOpcode(), llvm::X86::MOV32mi); |
| EXPECT_EQ(Insts[5].getOpcode(), llvm::X86::VMOVDQUrm); |
| EXPECT_EQ(Insts[6].getOpcode(), llvm::X86::ADD64ri8); |
| } |
| |
| TEST_F(X86TargetTest, SetRegToConstantXMM_AVX512) { |
| const std::unique_ptr<llvm::MCSubtargetInfo> STI( |
| Target_->createMCSubtargetInfo(kTriple, "core2", "+avx512vl")); |
| const auto Insts = ExegesisTarget_->setRegToConstant(*STI, llvm::X86::XMM1); |
| EXPECT_THAT(Insts, SizeIs(7U)); |
| EXPECT_EQ(Insts[0].getOpcode(), llvm::X86::SUB64ri8); |
| EXPECT_EQ(Insts[1].getOpcode(), llvm::X86::MOV32mi); |
| EXPECT_EQ(Insts[2].getOpcode(), llvm::X86::MOV32mi); |
| EXPECT_EQ(Insts[3].getOpcode(), llvm::X86::MOV32mi); |
| EXPECT_EQ(Insts[4].getOpcode(), llvm::X86::MOV32mi); |
| EXPECT_EQ(Insts[5].getOpcode(), llvm::X86::VMOVDQU32Z128rm); |
| EXPECT_EQ(Insts[6].getOpcode(), llvm::X86::ADD64ri8); |
| } |
| |
| TEST_F(X86TargetTest, SetRegToConstantMMX) { |
| const std::unique_ptr<llvm::MCSubtargetInfo> STI( |
| Target_->createMCSubtargetInfo(kTriple, "core2", "")); |
| const auto Insts = ExegesisTarget_->setRegToConstant(*STI, llvm::X86::MM1); |
| EXPECT_THAT(Insts, SizeIs(5U)); |
| EXPECT_EQ(Insts[0].getOpcode(), llvm::X86::SUB64ri8); |
| EXPECT_EQ(Insts[1].getOpcode(), llvm::X86::MOV32mi); |
| EXPECT_EQ(Insts[2].getOpcode(), llvm::X86::MOV32mi); |
| EXPECT_EQ(Insts[3].getOpcode(), llvm::X86::MMX_MOVQ64rm); |
| EXPECT_EQ(Insts[4].getOpcode(), llvm::X86::ADD64ri8); |
| } |
| |
| TEST_F(X86TargetTest, SetRegToConstantYMM_AVX) { |
| const std::unique_ptr<llvm::MCSubtargetInfo> STI( |
| Target_->createMCSubtargetInfo(kTriple, "core2", "+avx")); |
| const auto Insts = ExegesisTarget_->setRegToConstant(*STI, llvm::X86::YMM1); |
| EXPECT_THAT(Insts, SizeIs(11U)); |
| EXPECT_EQ(Insts[0].getOpcode(), llvm::X86::SUB64ri8); |
| EXPECT_EQ(Insts[1].getOpcode(), llvm::X86::MOV32mi); |
| EXPECT_EQ(Insts[2].getOpcode(), llvm::X86::MOV32mi); |
| EXPECT_EQ(Insts[3].getOpcode(), llvm::X86::MOV32mi); |
| EXPECT_EQ(Insts[4].getOpcode(), llvm::X86::MOV32mi); |
| EXPECT_EQ(Insts[5].getOpcode(), llvm::X86::MOV32mi); |
| EXPECT_EQ(Insts[6].getOpcode(), llvm::X86::MOV32mi); |
| EXPECT_EQ(Insts[7].getOpcode(), llvm::X86::MOV32mi); |
| EXPECT_EQ(Insts[8].getOpcode(), llvm::X86::MOV32mi); |
| EXPECT_EQ(Insts[9].getOpcode(), llvm::X86::VMOVDQUYrm); |
| EXPECT_EQ(Insts[10].getOpcode(), llvm::X86::ADD64ri8); |
| } |
| |
| TEST_F(X86TargetTest, SetRegToConstantYMM_AVX512) { |
| const std::unique_ptr<llvm::MCSubtargetInfo> STI( |
| Target_->createMCSubtargetInfo(kTriple, "core2", "+avx512vl")); |
| const auto Insts = ExegesisTarget_->setRegToConstant(*STI, llvm::X86::YMM1); |
| EXPECT_THAT(Insts, SizeIs(11U)); |
| EXPECT_EQ(Insts[0].getOpcode(), llvm::X86::SUB64ri8); |
| EXPECT_EQ(Insts[1].getOpcode(), llvm::X86::MOV32mi); |
| EXPECT_EQ(Insts[2].getOpcode(), llvm::X86::MOV32mi); |
| EXPECT_EQ(Insts[3].getOpcode(), llvm::X86::MOV32mi); |
| EXPECT_EQ(Insts[4].getOpcode(), llvm::X86::MOV32mi); |
| EXPECT_EQ(Insts[5].getOpcode(), llvm::X86::MOV32mi); |
| EXPECT_EQ(Insts[6].getOpcode(), llvm::X86::MOV32mi); |
| EXPECT_EQ(Insts[7].getOpcode(), llvm::X86::MOV32mi); |
| EXPECT_EQ(Insts[8].getOpcode(), llvm::X86::MOV32mi); |
| EXPECT_EQ(Insts[9].getOpcode(), llvm::X86::VMOVDQU32Z256rm); |
| EXPECT_EQ(Insts[10].getOpcode(), llvm::X86::ADD64ri8); |
| } |
| |
| TEST_F(X86TargetTest, SetRegToConstantZMM_AVX512) { |
| const std::unique_ptr<llvm::MCSubtargetInfo> STI( |
| Target_->createMCSubtargetInfo(kTriple, "core2", "+avx512vl")); |
| const auto Insts = ExegesisTarget_->setRegToConstant(*STI, llvm::X86::ZMM1); |
| EXPECT_THAT(Insts, SizeIs(19U)); |
| EXPECT_EQ(Insts[0].getOpcode(), llvm::X86::SUB64ri8); |
| EXPECT_EQ(Insts[1].getOpcode(), llvm::X86::MOV32mi); |
| EXPECT_EQ(Insts[2].getOpcode(), llvm::X86::MOV32mi); |
| EXPECT_EQ(Insts[3].getOpcode(), llvm::X86::MOV32mi); |
| EXPECT_EQ(Insts[4].getOpcode(), llvm::X86::MOV32mi); |
| EXPECT_EQ(Insts[5].getOpcode(), llvm::X86::MOV32mi); |
| EXPECT_EQ(Insts[6].getOpcode(), llvm::X86::MOV32mi); |
| EXPECT_EQ(Insts[7].getOpcode(), llvm::X86::MOV32mi); |
| EXPECT_EQ(Insts[8].getOpcode(), llvm::X86::MOV32mi); |
| EXPECT_EQ(Insts[9].getOpcode(), llvm::X86::MOV32mi); |
| EXPECT_EQ(Insts[10].getOpcode(), llvm::X86::MOV32mi); |
| EXPECT_EQ(Insts[11].getOpcode(), llvm::X86::MOV32mi); |
| EXPECT_EQ(Insts[12].getOpcode(), llvm::X86::MOV32mi); |
| EXPECT_EQ(Insts[13].getOpcode(), llvm::X86::MOV32mi); |
| EXPECT_EQ(Insts[14].getOpcode(), llvm::X86::MOV32mi); |
| EXPECT_EQ(Insts[15].getOpcode(), llvm::X86::MOV32mi); |
| EXPECT_EQ(Insts[16].getOpcode(), llvm::X86::MOV32mi); |
| EXPECT_EQ(Insts[17].getOpcode(), llvm::X86::VMOVDQU32Zrm); |
| EXPECT_EQ(Insts[18].getOpcode(), llvm::X86::ADD64ri8); |
| } |
| |
| } // namespace |
| } // namespace exegesis |