wangtao
/
llvm-project
forked from OSchip/llvm-project
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[LTO] Call the optimizer before invoking codegen. 

This is the required plumbing needed to run the LTO passes.

Differential Revision:  http://reviews.llvm.org/D18235

llvm-svn: 263761
This commit is contained in:
Davide Italiano 2016-03-17 23:35:34 +00:00
parent 26cc0377bc
commit c20c061749
1 changed files with 33 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
lld/ELF/SymbolTable.cpp
View File

@ -18,12 +18,16 @@
#include "Config.h" #include "Config.h"
#include "Error.h" #include "Error.h"
#include "Symbols.h" #include "Symbols.h"
#include "llvm/Analysis/TargetLibraryInfo.h"
#include "llvm/Analysis/TargetTransformInfo.h"
#include "llvm/Bitcode/ReaderWriter.h" #include "llvm/Bitcode/ReaderWriter.h"
#include "llvm/IR/LegacyPassManager.h" #include "llvm/IR/LegacyPassManager.h"
#include "llvm/Linker/IRMover.h" #include "llvm/Linker/IRMover.h"
#include "llvm/Support/StringSaver.h" #include "llvm/Support/StringSaver.h"
#include "llvm/Support/TargetRegistry.h" #include "llvm/Support/TargetRegistry.h"
#include "llvm/Target/TargetMachine.h" #include "llvm/Target/TargetMachine.h"
#include "llvm/Transforms/IPO.h"
#include "llvm/Transforms/IPO/PassManagerBuilder.h"
using namespace llvm; using namespace llvm;
using namespace llvm::object; using namespace llvm::object;
@ -106,6 +110,15 @@ static void saveLtoObjectFile(StringRef Buffer) {
OS << Buffer; OS << Buffer;
} }
// This is for use when debugging LTO.
static void saveBCFile(Module &M, StringRef Suffix) {
std::error_code EC;
raw_fd_ostream OS(Config->OutputFile.str() + Suffix.str(), EC,
sys::fs::OpenFlags::F_None);
check(EC);
WriteBitcodeToFile(&M, OS, /* ShouldPreserveUseListOrder */ true);
}
// Codegen the module M and returns the resulting InputFile. // Codegen the module M and returns the resulting InputFile.
template <class ELFT> template <class ELFT>
std::unique_ptr<InputFile> SymbolTable<ELFT>::codegen(Module &M) { std::unique_ptr<InputFile> SymbolTable<ELFT>::codegen(Module &M) {
@ -126,6 +139,25 @@ std::unique_ptr<InputFile> SymbolTable<ELFT>::codegen(Module &M) {
std::unique_ptr<TargetMachine> TM( std::unique_ptr<TargetMachine> TM(
TheTarget->createTargetMachine(TripleStr, "", "", Options, R)); TheTarget->createTargetMachine(TripleStr, "", "", Options, R));
// Run LTO passes.
// FIXME: Reduce code duplication by sharing this code with the gold plugin.
legacy::PassManager LtoPasses;
LtoPasses.add(
createTargetTransformInfoWrapperPass(TM->getTargetIRAnalysis()));
PassManagerBuilder PMB;
PMB.LibraryInfo = new TargetLibraryInfoImpl(Triple(TM->getTargetTriple()));
PMB.Inliner = createFunctionInliningPass();
PMB.VerifyInput = true;
PMB.VerifyOutput = true;
PMB.LoopVectorize = true;
PMB.SLPVectorize = true;
PMB.OptLevel = 2; // FIXME: This should be an option.
PMB.populateLTOPassManager(LtoPasses);
LtoPasses.run(M);
if (Config->SaveTemps)
saveBCFile(M, ".lto.opt.bc");
raw_svector_ostream OS(OwningLTOData); raw_svector_ostream OS(OwningLTOData);
legacy::PassManager CodeGenPasses; legacy::PassManager CodeGenPasses;
if (TM->addPassesToEmitFile(CodeGenPasses, OS, if (TM->addPassesToEmitFile(CodeGenPasses, OS,
@ -169,15 +201,6 @@ static void addBitcodeFile(IRMover &Mover, BitcodeFile &F,
[](GlobalValue &, IRMover::ValueAdder) {}); [](GlobalValue &, IRMover::ValueAdder) {});
} }
// This is for use when debugging LTO.
static void saveBCFile(Module &M) {
std::error_code EC;
raw_fd_ostream OS(Config->OutputFile.str() + ".lto.bc", EC,
sys::fs::OpenFlags::F_None);
check(EC);
WriteBitcodeToFile(&M, OS, /* ShouldPreserveUseListOrder */ true);
}
// Merge all the bitcode files we have seen, codegen the result and return // Merge all the bitcode files we have seen, codegen the result and return
// the resulting ObjectFile. // the resulting ObjectFile.
template <class ELFT> template <class ELFT>
@ -188,7 +211,7 @@ elf::ObjectFile<ELFT> *SymbolTable<ELFT>::createCombinedLtoObject() {
for (const std::unique_ptr<BitcodeFile> &F : BitcodeFiles) for (const std::unique_ptr<BitcodeFile> &F : BitcodeFiles)
addBitcodeFile(Mover, *F, Context); addBitcodeFile(Mover, *F, Context);
if (Config->SaveTemps) if (Config->SaveTemps)
saveBCFile(Combined); saveBCFile(Combined, ".lto.bc");
std::unique_ptr<InputFile> F = codegen(Combined); std::unique_ptr<InputFile> F = codegen(Combined);
ObjectFiles.emplace_back(cast<ObjectFile<ELFT>>(F.release())); ObjectFiles.emplace_back(cast<ObjectFile<ELFT>>(F.release()));
return &*ObjectFiles.back(); return &*ObjectFiles.back();