From e2a65fe5ec93ab4c9671c9655768275973ff3b30 Mon Sep 17 00:00:00 2001 From: Mehdi Amini Date: Wed, 27 Apr 2016 01:46:48 +0000 Subject: Support "preserving" the summary information when using setModule() API in LTOCodeGenerator From: Mehdi Amini llvm-svn: 267655 --- llvm/lib/LTO/LTOCodeGenerator.cpp | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) (limited to 'llvm/lib/LTO') diff --git a/llvm/lib/LTO/LTOCodeGenerator.cpp b/llvm/lib/LTO/LTOCodeGenerator.cpp index 01304641598..8497114f64f 100644 --- a/llvm/lib/LTO/LTOCodeGenerator.cpp +++ b/llvm/lib/LTO/LTOCodeGenerator.cpp @@ -18,6 +18,7 @@ #include "llvm/ADT/Statistic.h" #include "llvm/ADT/StringExtras.h" #include "llvm/Analysis/Passes.h" +#include "llvm/Analysis/ModuleSummaryAnalysis.h" #include "llvm/Analysis/TargetLibraryInfo.h" #include "llvm/Analysis/TargetTransformInfo.h" #include "llvm/Bitcode/ReaderWriter.h" @@ -145,6 +146,7 @@ void LTOCodeGenerator::setModule(std::unique_ptr Mod) { MergedModule = Mod->takeModule(); TheLinker = make_unique(*MergedModule); + MainModuleHasSummary = Mod->isThinLTO(); const std::vector &Undefs = Mod->getAsmUndefinedRefs(); for (int I = 0, E = Undefs.size(); I != E; ++I) @@ -209,8 +211,13 @@ bool LTOCodeGenerator::writeMergedModules(const char *Path) { return false; } + std::unique_ptr Index; + if (MainModuleHasSummary) + Index = ModuleSummaryIndexBuilder(MergedModule.get()).takeIndex(); + // write bitcode to it - WriteBitcodeToFile(MergedModule.get(), Out.os(), ShouldEmbedUselists); + WriteBitcodeToFile(MergedModule.get(), Out.os(), ShouldEmbedUselists, + Index.get()); Out.os().close(); if (Out.os().has_error()) { -- cgit v1.2.3