Details | Last modification | View Log | RSS feed
Rev | Author | Line No. | Line |
---|---|---|---|
14 | pmbaty | 1 | //===- ProfileSummary.h - Profile summary data structure. -------*- C++ -*-===// |
2 | // |
||
3 | // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
||
4 | // See https://llvm.org/LICENSE.txt for license information. |
||
5 | // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
||
6 | // |
||
7 | //===----------------------------------------------------------------------===// |
||
8 | // |
||
9 | // This file defines the profile summary data structure. |
||
10 | // |
||
11 | //===----------------------------------------------------------------------===// |
||
12 | |||
13 | #ifndef LLVM_IR_PROFILESUMMARY_H |
||
14 | #define LLVM_IR_PROFILESUMMARY_H |
||
15 | |||
16 | #include <algorithm> |
||
17 | #include <cassert> |
||
18 | #include <cstdint> |
||
19 | #include <vector> |
||
20 | |||
21 | namespace llvm { |
||
22 | |||
23 | class LLVMContext; |
||
24 | class Metadata; |
||
25 | class raw_ostream; |
||
26 | |||
27 | // The profile summary is one or more (Cutoff, MinCount, NumCounts) triplets. |
||
28 | // The semantics of counts depend on the type of profile. For instrumentation |
||
29 | // profile, counts are block counts and for sample profile, counts are |
||
30 | // per-line samples. Given a target counts percentile, we compute the minimum |
||
31 | // number of counts needed to reach this target and the minimum among these |
||
32 | // counts. |
||
33 | struct ProfileSummaryEntry { |
||
34 | const uint32_t Cutoff; ///< The required percentile of counts. |
||
35 | const uint64_t MinCount; ///< The minimum count for this percentile. |
||
36 | const uint64_t NumCounts; ///< Number of counts >= the minimum count. |
||
37 | |||
38 | ProfileSummaryEntry(uint32_t TheCutoff, uint64_t TheMinCount, |
||
39 | uint64_t TheNumCounts) |
||
40 | : Cutoff(TheCutoff), MinCount(TheMinCount), NumCounts(TheNumCounts) {} |
||
41 | }; |
||
42 | |||
43 | using SummaryEntryVector = std::vector<ProfileSummaryEntry>; |
||
44 | |||
45 | class ProfileSummary { |
||
46 | public: |
||
47 | enum Kind { PSK_Instr, PSK_CSInstr, PSK_Sample }; |
||
48 | |||
49 | private: |
||
50 | const Kind PSK; |
||
51 | const SummaryEntryVector DetailedSummary; |
||
52 | const uint64_t TotalCount, MaxCount, MaxInternalCount, MaxFunctionCount; |
||
53 | const uint32_t NumCounts, NumFunctions; |
||
54 | /// If 'Partial' is false, it means the profile being used to optimize |
||
55 | /// a target is collected from the same target. |
||
56 | /// If 'Partial' is true, it means the profile is for common/shared |
||
57 | /// code. The common profile is usually merged from profiles collected |
||
58 | /// from running other targets. |
||
59 | bool Partial = false; |
||
60 | /// This approximately represents the ratio of the number of profile counters |
||
61 | /// of the program being built to the number of profile counters in the |
||
62 | /// partial sample profile. When 'Partial' is false, it is undefined. This is |
||
63 | /// currently only available under thin LTO mode. |
||
64 | double PartialProfileRatio = 0.0; |
||
65 | /// Return detailed summary as metadata. |
||
66 | Metadata *getDetailedSummaryMD(LLVMContext &Context); |
||
67 | |||
68 | public: |
||
69 | static const int Scale = 1000000; |
||
70 | |||
71 | ProfileSummary(Kind K, const SummaryEntryVector &DetailedSummary, |
||
72 | uint64_t TotalCount, uint64_t MaxCount, |
||
73 | uint64_t MaxInternalCount, uint64_t MaxFunctionCount, |
||
74 | uint32_t NumCounts, uint32_t NumFunctions, |
||
75 | bool Partial = false, double PartialProfileRatio = 0) |
||
76 | : PSK(K), DetailedSummary(std::move(DetailedSummary)), |
||
77 | TotalCount(TotalCount), MaxCount(MaxCount), |
||
78 | MaxInternalCount(MaxInternalCount), MaxFunctionCount(MaxFunctionCount), |
||
79 | NumCounts(NumCounts), NumFunctions(NumFunctions), Partial(Partial), |
||
80 | PartialProfileRatio(PartialProfileRatio) {} |
||
81 | |||
82 | Kind getKind() const { return PSK; } |
||
83 | /// Return summary information as metadata. |
||
84 | Metadata *getMD(LLVMContext &Context, bool AddPartialField = true, |
||
85 | bool AddPartialProfileRatioField = true); |
||
86 | /// Construct profile summary from metdata. |
||
87 | static ProfileSummary *getFromMD(Metadata *MD); |
||
88 | const SummaryEntryVector &getDetailedSummary() { return DetailedSummary; } |
||
89 | uint32_t getNumFunctions() const { return NumFunctions; } |
||
90 | uint64_t getMaxFunctionCount() const { return MaxFunctionCount; } |
||
91 | uint32_t getNumCounts() const { return NumCounts; } |
||
92 | uint64_t getTotalCount() const { return TotalCount; } |
||
93 | uint64_t getMaxCount() const { return MaxCount; } |
||
94 | uint64_t getMaxInternalCount() const { return MaxInternalCount; } |
||
95 | void setPartialProfile(bool PP) { Partial = PP; } |
||
96 | bool isPartialProfile() const { return Partial; } |
||
97 | double getPartialProfileRatio() const { return PartialProfileRatio; } |
||
98 | void setPartialProfileRatio(double R) { |
||
99 | assert(isPartialProfile() && "Unexpected when not partial profile"); |
||
100 | PartialProfileRatio = R; |
||
101 | } |
||
102 | void printSummary(raw_ostream &OS) const; |
||
103 | void printDetailedSummary(raw_ostream &OS) const; |
||
104 | }; |
||
105 | |||
106 | } // end namespace llvm |
||
107 | |||
108 | #endif // LLVM_IR_PROFILESUMMARY_H |