1 //===-- ModuleSummaryIndex.cpp - Module Summary Index ---------------------===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 //
10 // This file implements the module index and summary classes for the
11 // IR library.
12 //
13 //===----------------------------------------------------------------------===//
14 
15 #include "llvm/IR/ModuleSummaryIndex.h"
16 #include "llvm/ADT/StringMap.h"
17 using namespace llvm;
18 
19 // Collect for the given module the list of function it defines
20 // (GUID -> Summary).
21 void ModuleSummaryIndex::collectDefinedFunctionsForModule(
22     StringRef ModulePath, GVSummaryMapTy &GVSummaryMap) const {
23   for (auto &GlobalList : *this) {
24     auto GUID = GlobalList.first;
25     for (auto &GlobSummary : GlobalList.second.SummaryList) {
26       auto *Summary = dyn_cast_or_null<FunctionSummary>(GlobSummary.get());
27       if (!Summary)
28         // Ignore global variable, focus on functions
29         continue;
30       // Ignore summaries from other modules.
31       if (Summary->modulePath() != ModulePath)
32         continue;
33       GVSummaryMap[GUID] = Summary;
34     }
35   }
36 }
37 
38 // Collect for each module the list of function it defines (GUID -> Summary).
39 void ModuleSummaryIndex::collectDefinedGVSummariesPerModule(
40     StringMap<GVSummaryMapTy> &ModuleToDefinedGVSummaries) const {
41   for (auto &GlobalList : *this) {
42     auto GUID = GlobalList.first;
43     for (auto &Summary : GlobalList.second.SummaryList) {
44       ModuleToDefinedGVSummaries[Summary->modulePath()][GUID] = Summary.get();
45     }
46   }
47 }
48 
49 GlobalValueSummary *
50 ModuleSummaryIndex::getGlobalValueSummary(uint64_t ValueGUID,
51                                           bool PerModuleIndex) const {
52   auto VI = getValueInfo(ValueGUID);
53   assert(VI && "GlobalValue not found in index");
54   assert((!PerModuleIndex || VI.getSummaryList().size() == 1) &&
55          "Expected a single entry per global value in per-module index");
56   auto &Summary = VI.getSummaryList()[0];
57   return Summary.get();
58 }
59 
60 bool ModuleSummaryIndex::isGUIDLive(GlobalValue::GUID GUID) const {
61   auto VI = getValueInfo(GUID);
62   if (!VI)
63     return true;
64   const auto &SummaryList = VI.getSummaryList();
65   if (SummaryList.empty())
66     return true;
67   for (auto &I : SummaryList)
68     if (isGlobalValueLive(I.get()))
69       return true;
70   return false;
71 }
72