Details | Last modification | View Log | RSS feed
Rev | Author | Line No. | Line |
---|---|---|---|
14 | pmbaty | 1 | //===---- llvm/MDBuilder.h - Builder for LLVM metadata ----------*- 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 MDBuilder class, which is used as a convenient way to |
||
10 | // create LLVM metadata with a consistent and simplified interface. |
||
11 | // |
||
12 | //===----------------------------------------------------------------------===// |
||
13 | |||
14 | #ifndef LLVM_IR_MDBUILDER_H |
||
15 | #define LLVM_IR_MDBUILDER_H |
||
16 | |||
17 | #include "llvm/ADT/DenseSet.h" |
||
18 | #include "llvm/ADT/SmallVector.h" |
||
19 | #include "llvm/ADT/StringRef.h" |
||
20 | #include "llvm/IR/GlobalValue.h" |
||
21 | #include "llvm/Support/DataTypes.h" |
||
22 | #include <utility> |
||
23 | |||
24 | namespace llvm { |
||
25 | |||
26 | class APInt; |
||
27 | template <typename T> class ArrayRef; |
||
28 | class LLVMContext; |
||
29 | class Constant; |
||
30 | class ConstantAsMetadata; |
||
31 | class Function; |
||
32 | class MDNode; |
||
33 | class MDString; |
||
34 | class Metadata; |
||
35 | |||
36 | class MDBuilder { |
||
37 | LLVMContext &Context; |
||
38 | |||
39 | public: |
||
40 | MDBuilder(LLVMContext &context) : Context(context) {} |
||
41 | |||
42 | /// Return the given string as metadata. |
||
43 | MDString *createString(StringRef Str); |
||
44 | |||
45 | /// Return the given constant as metadata. |
||
46 | ConstantAsMetadata *createConstant(Constant *C); |
||
47 | |||
48 | //===------------------------------------------------------------------===// |
||
49 | // FPMath metadata. |
||
50 | //===------------------------------------------------------------------===// |
||
51 | |||
52 | /// Return metadata with the given settings. The special value 0.0 |
||
53 | /// for the Accuracy parameter indicates the default (maximal precision) |
||
54 | /// setting. |
||
55 | MDNode *createFPMath(float Accuracy); |
||
56 | |||
57 | //===------------------------------------------------------------------===// |
||
58 | // Prof metadata. |
||
59 | //===------------------------------------------------------------------===// |
||
60 | |||
61 | /// Return metadata containing two branch weights. |
||
62 | MDNode *createBranchWeights(uint32_t TrueWeight, uint32_t FalseWeight); |
||
63 | |||
64 | /// Return metadata containing a number of branch weights. |
||
65 | MDNode *createBranchWeights(ArrayRef<uint32_t> Weights); |
||
66 | |||
67 | /// Return metadata specifying that a branch or switch is unpredictable. |
||
68 | MDNode *createUnpredictable(); |
||
69 | |||
70 | /// Return metadata containing the entry \p Count for a function, a boolean |
||
71 | /// \Synthetic indicating whether the counts were synthetized, and the |
||
72 | /// GUIDs stored in \p Imports that need to be imported for sample PGO, to |
||
73 | /// enable the same inlines as the profiled optimized binary |
||
74 | MDNode *createFunctionEntryCount(uint64_t Count, bool Synthetic, |
||
75 | const DenseSet<GlobalValue::GUID> *Imports); |
||
76 | |||
77 | /// Return metadata containing the section prefix for a function. |
||
78 | MDNode *createFunctionSectionPrefix(StringRef Prefix); |
||
79 | |||
80 | /// Return metadata containing the pseudo probe descriptor for a function. |
||
81 | MDNode *createPseudoProbeDesc(uint64_t GUID, uint64_t Hash, Function *F); |
||
82 | |||
83 | /// Return metadata containing llvm statistics. |
||
84 | MDNode * |
||
85 | createLLVMStats(ArrayRef<std::pair<StringRef, uint64_t>> LLVMStatsVec); |
||
86 | |||
87 | //===------------------------------------------------------------------===// |
||
88 | // Range metadata. |
||
89 | //===------------------------------------------------------------------===// |
||
90 | |||
91 | /// Return metadata describing the range [Lo, Hi). |
||
92 | MDNode *createRange(const APInt &Lo, const APInt &Hi); |
||
93 | |||
94 | /// Return metadata describing the range [Lo, Hi). |
||
95 | MDNode *createRange(Constant *Lo, Constant *Hi); |
||
96 | |||
97 | //===------------------------------------------------------------------===// |
||
98 | // Callees metadata. |
||
99 | //===------------------------------------------------------------------===// |
||
100 | |||
101 | /// Return metadata indicating the possible callees of indirect |
||
102 | /// calls. |
||
103 | MDNode *createCallees(ArrayRef<Function *> Callees); |
||
104 | |||
105 | //===------------------------------------------------------------------===// |
||
106 | // Callback metadata. |
||
107 | //===------------------------------------------------------------------===// |
||
108 | |||
109 | /// Return metadata describing a callback (see llvm::AbstractCallSite). |
||
110 | MDNode *createCallbackEncoding(unsigned CalleeArgNo, ArrayRef<int> Arguments, |
||
111 | bool VarArgsArePassed); |
||
112 | |||
113 | /// Merge the new callback encoding \p NewCB into \p ExistingCallbacks. |
||
114 | MDNode *mergeCallbackEncodings(MDNode *ExistingCallbacks, MDNode *NewCB); |
||
115 | |||
116 | /// Return metadata feeding to the CodeGen about how to generate a function |
||
117 | /// prologue for the "function" santizier. |
||
118 | MDNode *createRTTIPointerPrologue(Constant *PrologueSig, Constant *RTTI); |
||
119 | |||
120 | //===------------------------------------------------------------------===// |
||
121 | // PC sections metadata. |
||
122 | //===------------------------------------------------------------------===// |
||
123 | |||
124 | /// A pair of PC section name with auxilliary constant data. |
||
125 | using PCSection = std::pair<StringRef, SmallVector<Constant *>>; |
||
126 | |||
127 | /// Return metadata for PC sections. |
||
128 | MDNode *createPCSections(ArrayRef<PCSection> Sections); |
||
129 | |||
130 | //===------------------------------------------------------------------===// |
||
131 | // AA metadata. |
||
132 | //===------------------------------------------------------------------===// |
||
133 | |||
134 | protected: |
||
135 | /// Return metadata appropriate for a AA root node (scope or TBAA). |
||
136 | /// Each returned node is distinct from all other metadata and will never |
||
137 | /// be identified (uniqued) with anything else. |
||
138 | MDNode *createAnonymousAARoot(StringRef Name = StringRef(), |
||
139 | MDNode *Extra = nullptr); |
||
140 | |||
141 | public: |
||
142 | /// Return metadata appropriate for a TBAA root node. Each returned |
||
143 | /// node is distinct from all other metadata and will never be identified |
||
144 | /// (uniqued) with anything else. |
||
145 | MDNode *createAnonymousTBAARoot() { |
||
146 | return createAnonymousAARoot(); |
||
147 | } |
||
148 | |||
149 | /// Return metadata appropriate for an alias scope domain node. |
||
150 | /// Each returned node is distinct from all other metadata and will never |
||
151 | /// be identified (uniqued) with anything else. |
||
152 | MDNode *createAnonymousAliasScopeDomain(StringRef Name = StringRef()) { |
||
153 | return createAnonymousAARoot(Name); |
||
154 | } |
||
155 | |||
156 | /// Return metadata appropriate for an alias scope root node. |
||
157 | /// Each returned node is distinct from all other metadata and will never |
||
158 | /// be identified (uniqued) with anything else. |
||
159 | MDNode *createAnonymousAliasScope(MDNode *Domain, |
||
160 | StringRef Name = StringRef()) { |
||
161 | return createAnonymousAARoot(Name, Domain); |
||
162 | } |
||
163 | |||
164 | /// Return metadata appropriate for a TBAA root node with the given |
||
165 | /// name. This may be identified (uniqued) with other roots with the same |
||
166 | /// name. |
||
167 | MDNode *createTBAARoot(StringRef Name); |
||
168 | |||
169 | /// Return metadata appropriate for an alias scope domain node with |
||
170 | /// the given name. This may be identified (uniqued) with other roots with |
||
171 | /// the same name. |
||
172 | MDNode *createAliasScopeDomain(StringRef Name); |
||
173 | |||
174 | /// Return metadata appropriate for an alias scope node with |
||
175 | /// the given name. This may be identified (uniqued) with other scopes with |
||
176 | /// the same name and domain. |
||
177 | MDNode *createAliasScope(StringRef Name, MDNode *Domain); |
||
178 | |||
179 | /// Return metadata for a non-root TBAA node with the given name, |
||
180 | /// parent in the TBAA tree, and value for 'pointsToConstantMemory'. |
||
181 | MDNode *createTBAANode(StringRef Name, MDNode *Parent, |
||
182 | bool isConstant = false); |
||
183 | |||
184 | struct TBAAStructField { |
||
185 | uint64_t Offset; |
||
186 | uint64_t Size; |
||
187 | MDNode *Type; |
||
188 | TBAAStructField(uint64_t Offset, uint64_t Size, MDNode *Type) : |
||
189 | Offset(Offset), Size(Size), Type(Type) {} |
||
190 | }; |
||
191 | |||
192 | /// Return metadata for a tbaa.struct node with the given |
||
193 | /// struct field descriptions. |
||
194 | MDNode *createTBAAStructNode(ArrayRef<TBAAStructField> Fields); |
||
195 | |||
196 | /// Return metadata for a TBAA struct node in the type DAG |
||
197 | /// with the given name, a list of pairs (offset, field type in the type DAG). |
||
198 | MDNode * |
||
199 | createTBAAStructTypeNode(StringRef Name, |
||
200 | ArrayRef<std::pair<MDNode *, uint64_t>> Fields); |
||
201 | |||
202 | /// Return metadata for a TBAA scalar type node with the |
||
203 | /// given name, an offset and a parent in the TBAA type DAG. |
||
204 | MDNode *createTBAAScalarTypeNode(StringRef Name, MDNode *Parent, |
||
205 | uint64_t Offset = 0); |
||
206 | |||
207 | /// Return metadata for a TBAA tag node with the given |
||
208 | /// base type, access type and offset relative to the base type. |
||
209 | MDNode *createTBAAStructTagNode(MDNode *BaseType, MDNode *AccessType, |
||
210 | uint64_t Offset, bool IsConstant = false); |
||
211 | |||
212 | /// Return metadata for a TBAA type node in the TBAA type DAG with the |
||
213 | /// given parent type, size in bytes, type identifier and a list of fields. |
||
214 | MDNode *createTBAATypeNode(MDNode *Parent, uint64_t Size, Metadata *Id, |
||
215 | ArrayRef<TBAAStructField> Fields = |
||
216 | ArrayRef<TBAAStructField>()); |
||
217 | |||
218 | /// Return metadata for a TBAA access tag with the given base type, |
||
219 | /// final access type, offset of the access relative to the base type, size of |
||
220 | /// the access and flag indicating whether the accessed object can be |
||
221 | /// considered immutable for the purposes of the TBAA analysis. |
||
222 | MDNode *createTBAAAccessTag(MDNode *BaseType, MDNode *AccessType, |
||
223 | uint64_t Offset, uint64_t Size, |
||
224 | bool IsImmutable = false); |
||
225 | |||
226 | /// Return mutable version of the given mutable or immutable TBAA |
||
227 | /// access tag. |
||
228 | MDNode *createMutableTBAAAccessTag(MDNode *Tag); |
||
229 | |||
230 | /// Return metadata containing an irreducible loop header weight. |
||
231 | MDNode *createIrrLoopHeaderWeight(uint64_t Weight); |
||
232 | }; |
||
233 | |||
234 | } // end namespace llvm |
||
235 | |||
236 | #endif |