89b57061f7
This moves the registry higher in the LLVM library dependency stack. Every client of the target registry needs to link against MC anyway to actually use the target, so we might as well move this out of Support. This allows us to ensure that Support doesn't have includes from MC/*. Differential Revision: https://reviews.llvm.org/D111454
78 lines
2.0 KiB
C++
78 lines
2.0 KiB
C++
#include "llvm/Target/TargetOptions.h"
|
|
#include "llvm/CodeGen/TargetPassConfig.h"
|
|
#include "llvm/IR/LLVMContext.h"
|
|
#include "llvm/IR/LegacyPassManager.h"
|
|
#include "llvm/InitializePasses.h"
|
|
#include "llvm/MC/TargetRegistry.h"
|
|
#include "llvm/Support/TargetSelect.h"
|
|
#include "llvm/Target/TargetMachine.h"
|
|
#include "gtest/gtest.h"
|
|
|
|
using namespace llvm;
|
|
|
|
namespace llvm {
|
|
void initializeTestPassPass(PassRegistry &);
|
|
}
|
|
|
|
namespace {
|
|
|
|
void initLLVM() {
|
|
InitializeAllTargets();
|
|
InitializeAllTargetMCs();
|
|
InitializeAllAsmPrinters();
|
|
InitializeAllAsmParsers();
|
|
|
|
PassRegistry *Registry = PassRegistry::getPassRegistry();
|
|
initializeCore(*Registry);
|
|
initializeCodeGen(*Registry);
|
|
}
|
|
|
|
/// Create a TargetMachine. We need a target that doesn't have IPRA enabled by
|
|
/// default. That turns out to be all targets at the moment, so just use X86.
|
|
std::unique_ptr<TargetMachine> createTargetMachine(bool EnableIPRA) {
|
|
Triple TargetTriple("x86_64--");
|
|
std::string Error;
|
|
const Target *T = TargetRegistry::lookupTarget("", TargetTriple, Error);
|
|
if (!T)
|
|
return nullptr;
|
|
|
|
TargetOptions Options;
|
|
Options.EnableIPRA = EnableIPRA;
|
|
return std::unique_ptr<TargetMachine>(T->createTargetMachine(
|
|
"X86", "", "", Options, None, None, CodeGenOpt::Aggressive));
|
|
}
|
|
|
|
typedef std::function<void(bool)> TargetOptionsTest;
|
|
|
|
static void targetOptionsTest(bool EnableIPRA) {
|
|
std::unique_ptr<TargetMachine> TM = createTargetMachine(EnableIPRA);
|
|
// This test is designed for the X86 backend; stop if it is not available.
|
|
if (!TM)
|
|
GTEST_SKIP();
|
|
legacy::PassManager PM;
|
|
LLVMTargetMachine *LLVMTM = static_cast<LLVMTargetMachine *>(TM.get());
|
|
|
|
TargetPassConfig *TPC = LLVMTM->createPassConfig(PM);
|
|
(void)TPC;
|
|
|
|
ASSERT_TRUE(TM->Options.EnableIPRA == EnableIPRA);
|
|
|
|
delete TPC;
|
|
}
|
|
|
|
} // End of anonymous namespace.
|
|
|
|
TEST(TargetOptionsTest, IPRASetToOff) {
|
|
targetOptionsTest(false);
|
|
}
|
|
|
|
TEST(TargetOptionsTest, IPRASetToOn) {
|
|
targetOptionsTest(true);
|
|
}
|
|
|
|
int main(int argc, char **argv) {
|
|
::testing::InitGoogleTest(&argc, argv);
|
|
initLLVM();
|
|
return RUN_ALL_TESTS();
|
|
}
|