78 lines · cpp
1#include "llvm/Target/TargetOptions.h"2#include "llvm/CodeGen/TargetPassConfig.h"3#include "llvm/IR/LLVMContext.h"4#include "llvm/IR/LegacyPassManager.h"5#include "llvm/InitializePasses.h"6#include "llvm/MC/TargetRegistry.h"7#include "llvm/Support/TargetSelect.h"8#include "llvm/Target/TargetMachine.h"9#include "gtest/gtest.h"10 11using namespace llvm;12 13namespace llvm {14 void initializeTestPassPass(PassRegistry &);15}16 17namespace {18 19void initLLVM() {20 InitializeAllTargets();21 InitializeAllTargetMCs();22 InitializeAllAsmPrinters();23 InitializeAllAsmParsers();24 25 PassRegistry *Registry = PassRegistry::getPassRegistry();26 initializeCore(*Registry);27 initializeCodeGen(*Registry);28}29 30/// Create a TargetMachine. We need a target that doesn't have IPRA enabled by31/// default. That turns out to be all targets at the moment, so just use X86.32std::unique_ptr<TargetMachine> createTargetMachine(bool EnableIPRA) {33 Triple TargetTriple("x86_64--");34 std::string Error;35 const Target *T = TargetRegistry::lookupTarget("", TargetTriple, Error);36 if (!T)37 return nullptr;38 39 TargetOptions Options;40 Options.EnableIPRA = EnableIPRA;41 return std::unique_ptr<TargetMachine>(42 T->createTargetMachine(TargetTriple, "", "", Options, std::nullopt,43 std::nullopt, CodeGenOptLevel::Aggressive));44}45 46typedef std::function<void(bool)> TargetOptionsTest;47 48static void targetOptionsTest(bool EnableIPRA) {49 std::unique_ptr<TargetMachine> TM = createTargetMachine(EnableIPRA);50 // This test is designed for the X86 backend; stop if it is not available.51 if (!TM)52 GTEST_SKIP();53 legacy::PassManager PM;54 55 TargetPassConfig *TPC = TM->createPassConfig(PM);56 (void)TPC;57 58 ASSERT_TRUE(TM->Options.EnableIPRA == EnableIPRA);59 60 delete TPC;61}62 63} // End of anonymous namespace.64 65TEST(TargetOptionsTest, IPRASetToOff) {66 targetOptionsTest(false);67}68 69TEST(TargetOptionsTest, IPRASetToOn) {70 targetOptionsTest(true);71}72 73int main(int argc, char **argv) {74 ::testing::InitGoogleTest(&argc, argv);75 initLLVM();76 return RUN_ALL_TESTS();77}78