1044eb2f6SDimitry Andric //===-- AMDGPULibFunc.cpp -------------------------------------------------===//
2044eb2f6SDimitry Andric //
3e6d15924SDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4e6d15924SDimitry Andric // See https://llvm.org/LICENSE.txt for license information.
5e6d15924SDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6044eb2f6SDimitry Andric //
7044eb2f6SDimitry Andric //===----------------------------------------------------------------------===//
8044eb2f6SDimitry Andric //
9044eb2f6SDimitry Andric // This file contains utility functions to work with Itanium mangled names
10044eb2f6SDimitry Andric //
11044eb2f6SDimitry Andric //===----------------------------------------------------------------------===//
12044eb2f6SDimitry Andric
13044eb2f6SDimitry Andric #include "AMDGPULibFunc.h"
14cfca06d7SDimitry Andric #include "AMDGPU.h"
15cfca06d7SDimitry Andric #include "llvm/ADT/StringExtras.h"
16b60736ecSDimitry Andric #include "llvm/ADT/StringMap.h"
17cfca06d7SDimitry Andric #include "llvm/ADT/StringSwitch.h"
18044eb2f6SDimitry Andric #include "llvm/IR/DerivedTypes.h"
19044eb2f6SDimitry Andric #include "llvm/IR/Function.h"
20044eb2f6SDimitry Andric #include "llvm/IR/Module.h"
21044eb2f6SDimitry Andric #include "llvm/IR/ValueSymbolTable.h"
22344a3780SDimitry Andric #include "llvm/Support/CommandLine.h"
23e3b55780SDimitry Andric #include "llvm/Support/ModRef.h"
24cfca06d7SDimitry Andric #include "llvm/Support/raw_ostream.h"
25044eb2f6SDimitry Andric
26044eb2f6SDimitry Andric using namespace llvm;
27044eb2f6SDimitry Andric
28344a3780SDimitry Andric static cl::opt<bool> EnableOCLManglingMismatchWA(
29344a3780SDimitry Andric "amdgpu-enable-ocl-mangling-mismatch-workaround", cl::init(true),
30344a3780SDimitry Andric cl::ReallyHidden,
31344a3780SDimitry Andric cl::desc("Enable the workaround for OCL name mangling mismatch."));
32344a3780SDimitry Andric
33044eb2f6SDimitry Andric namespace {
34044eb2f6SDimitry Andric
35044eb2f6SDimitry Andric enum EManglingParam {
36044eb2f6SDimitry Andric E_NONE,
37044eb2f6SDimitry Andric EX_EVENT,
38044eb2f6SDimitry Andric EX_FLOAT4,
39044eb2f6SDimitry Andric EX_INTV4,
40044eb2f6SDimitry Andric EX_RESERVEDID,
41044eb2f6SDimitry Andric EX_SAMPLER,
42044eb2f6SDimitry Andric EX_SIZET,
43044eb2f6SDimitry Andric EX_UINT,
44044eb2f6SDimitry Andric EX_UINTV4,
45044eb2f6SDimitry Andric E_ANY,
46044eb2f6SDimitry Andric E_CONSTPTR_ANY,
47044eb2f6SDimitry Andric E_CONSTPTR_SWAPGL,
48044eb2f6SDimitry Andric E_COPY,
49044eb2f6SDimitry Andric E_IMAGECOORDS,
50044eb2f6SDimitry Andric E_POINTEE,
51044eb2f6SDimitry Andric E_SETBASE_I32,
52044eb2f6SDimitry Andric E_SETBASE_U32,
53044eb2f6SDimitry Andric E_MAKEBASE_UNS,
54044eb2f6SDimitry Andric E_V16_OF_POINTEE,
55044eb2f6SDimitry Andric E_V2_OF_POINTEE,
56044eb2f6SDimitry Andric E_V3_OF_POINTEE,
57044eb2f6SDimitry Andric E_V4_OF_POINTEE,
58044eb2f6SDimitry Andric E_V8_OF_POINTEE,
59044eb2f6SDimitry Andric E_VLTLPTR_ANY,
60044eb2f6SDimitry Andric };
61044eb2f6SDimitry Andric
62044eb2f6SDimitry Andric struct ManglingRule {
631d5ae102SDimitry Andric const char *Name;
64044eb2f6SDimitry Andric unsigned char Lead[2];
65044eb2f6SDimitry Andric unsigned char Param[5];
66044eb2f6SDimitry Andric
maxLeadIndex__anon7d00f1a90111::ManglingRule67044eb2f6SDimitry Andric int maxLeadIndex() const { return (std::max)(Lead[0], Lead[1]); }
getNumLeads__anon7d00f1a90111::ManglingRule68044eb2f6SDimitry Andric int getNumLeads() const { return (Lead[0] ? 1 : 0) + (Lead[1] ? 1 : 0); }
69044eb2f6SDimitry Andric
70044eb2f6SDimitry Andric unsigned getNumArgs() const;
71e6d15924SDimitry Andric
72e6d15924SDimitry Andric static StringMap<int> buildManglingRulesMap();
73044eb2f6SDimitry Andric };
74044eb2f6SDimitry Andric
75044eb2f6SDimitry Andric // Information about library functions with unmangled names.
76044eb2f6SDimitry Andric class UnmangledFuncInfo {
771d5ae102SDimitry Andric const char *Name;
78044eb2f6SDimitry Andric unsigned NumArgs;
79044eb2f6SDimitry Andric
80044eb2f6SDimitry Andric // Table for all lib functions with unmangled names.
81044eb2f6SDimitry Andric static const UnmangledFuncInfo Table[];
82044eb2f6SDimitry Andric
83044eb2f6SDimitry Andric // Number of entries in Table.
84044eb2f6SDimitry Andric static const unsigned TableSize;
85044eb2f6SDimitry Andric
86e6d15924SDimitry Andric static StringMap<unsigned> buildNameMap();
87044eb2f6SDimitry Andric
88044eb2f6SDimitry Andric public:
89044eb2f6SDimitry Andric using ID = AMDGPULibFunc::EFuncId;
UnmangledFuncInfo(const char * _Name,unsigned _NumArgs)901d5ae102SDimitry Andric constexpr UnmangledFuncInfo(const char *_Name, unsigned _NumArgs)
91044eb2f6SDimitry Andric : Name(_Name), NumArgs(_NumArgs) {}
92044eb2f6SDimitry Andric // Get index to Table by function name.
93044eb2f6SDimitry Andric static bool lookup(StringRef Name, ID &Id);
toIndex(ID Id)94044eb2f6SDimitry Andric static unsigned toIndex(ID Id) {
95044eb2f6SDimitry Andric assert(static_cast<unsigned>(Id) >
96044eb2f6SDimitry Andric static_cast<unsigned>(AMDGPULibFunc::EI_LAST_MANGLED) &&
97044eb2f6SDimitry Andric "Invalid unmangled library function");
98044eb2f6SDimitry Andric return static_cast<unsigned>(Id) - 1 -
99044eb2f6SDimitry Andric static_cast<unsigned>(AMDGPULibFunc::EI_LAST_MANGLED);
100044eb2f6SDimitry Andric }
toFuncId(unsigned Index)101044eb2f6SDimitry Andric static ID toFuncId(unsigned Index) {
102e6d15924SDimitry Andric assert(Index < TableSize &&
103e6d15924SDimitry Andric "Invalid unmangled library function");
104044eb2f6SDimitry Andric return static_cast<ID>(
105044eb2f6SDimitry Andric Index + 1 + static_cast<unsigned>(AMDGPULibFunc::EI_LAST_MANGLED));
106044eb2f6SDimitry Andric }
getNumArgs(ID Id)107044eb2f6SDimitry Andric static unsigned getNumArgs(ID Id) { return Table[toIndex(Id)].NumArgs; }
getName(ID Id)108044eb2f6SDimitry Andric static StringRef getName(ID Id) { return Table[toIndex(Id)].Name; }
109044eb2f6SDimitry Andric };
110044eb2f6SDimitry Andric
getNumArgs() const111044eb2f6SDimitry Andric unsigned ManglingRule::getNumArgs() const {
112044eb2f6SDimitry Andric unsigned I=0;
113044eb2f6SDimitry Andric while (I < (sizeof Param/sizeof Param[0]) && Param[I]) ++I;
114044eb2f6SDimitry Andric return I;
115044eb2f6SDimitry Andric }
116044eb2f6SDimitry Andric
117044eb2f6SDimitry Andric // This table describes function formal argument type rules. The order of rules
118044eb2f6SDimitry Andric // corresponds to the EFuncId enum at AMDGPULibFunc.h
119044eb2f6SDimitry Andric //
120044eb2f6SDimitry Andric // "<func name>", { <leads> }, { <param rules> }
121044eb2f6SDimitry Andric // where:
122044eb2f6SDimitry Andric // <leads> - list of integers that are one-based indexes of formal argument
123044eb2f6SDimitry Andric // used to mangle a function name. Other argument types are derived from types
124044eb2f6SDimitry Andric // of these 'leads'. The order of integers in this list correspond to the
125044eb2f6SDimitry Andric // order in which these arguments are mangled in the EDG mangling scheme. The
126044eb2f6SDimitry Andric // same order should be preserved for arguments in the AMDGPULibFunc structure
127044eb2f6SDimitry Andric // when it is used for mangling. For example:
128044eb2f6SDimitry Andric // { "vstorea_half", {3,1}, {E_ANY,EX_SIZET,E_ANY}},
129044eb2f6SDimitry Andric // will be mangled in EDG scheme as vstorea_half_<3dparam>_<1stparam>
130044eb2f6SDimitry Andric // When mangling from code use:
131044eb2f6SDimitry Andric // AMDGPULibFunc insc;
132044eb2f6SDimitry Andric // insc.param[0] = ... // describe 3rd parameter
133044eb2f6SDimitry Andric // insc.param[1] = ... // describe 1rd parameter
134044eb2f6SDimitry Andric //
135044eb2f6SDimitry Andric // <param rules> - list of rules used to derive all of the function formal
136044eb2f6SDimitry Andric // argument types. EX_ prefixed are simple types, other derived from the
137044eb2f6SDimitry Andric // latest 'lead' argument type in the order of encoding from first to last.
138044eb2f6SDimitry Andric // E_ANY - use prev lead type, E_CONSTPTR_ANY - make const pointer out of
139044eb2f6SDimitry Andric // prev lead type, etc. see ParamIterator::getNextParam() for details.
140044eb2f6SDimitry Andric
1411d5ae102SDimitry Andric static constexpr ManglingRule manglingRules[] = {
1421d5ae102SDimitry Andric { "", {0}, {0} },
143044eb2f6SDimitry Andric { "abs" , {1}, {E_ANY}},
144044eb2f6SDimitry Andric { "abs_diff" , {1}, {E_ANY,E_COPY}},
145044eb2f6SDimitry Andric { "acos" , {1}, {E_ANY}},
146044eb2f6SDimitry Andric { "acosh" , {1}, {E_ANY}},
147044eb2f6SDimitry Andric { "acospi" , {1}, {E_ANY}},
148044eb2f6SDimitry Andric { "add_sat" , {1}, {E_ANY,E_COPY}},
149044eb2f6SDimitry Andric { "all" , {1}, {E_ANY}},
150044eb2f6SDimitry Andric { "any" , {1}, {E_ANY}},
151044eb2f6SDimitry Andric { "asin" , {1}, {E_ANY}},
152044eb2f6SDimitry Andric { "asinh" , {1}, {E_ANY}},
153044eb2f6SDimitry Andric { "asinpi" , {1}, {E_ANY}},
154044eb2f6SDimitry Andric { "async_work_group_copy" , {1}, {E_ANY,E_CONSTPTR_SWAPGL,EX_SIZET,EX_EVENT}},
155044eb2f6SDimitry Andric { "async_work_group_strided_copy" , {1}, {E_ANY,E_CONSTPTR_SWAPGL,EX_SIZET,EX_SIZET,EX_EVENT}},
156044eb2f6SDimitry Andric { "atan" , {1}, {E_ANY}},
157044eb2f6SDimitry Andric { "atan2" , {1}, {E_ANY,E_COPY}},
158044eb2f6SDimitry Andric { "atan2pi" , {1}, {E_ANY,E_COPY}},
159044eb2f6SDimitry Andric { "atanh" , {1}, {E_ANY}},
160044eb2f6SDimitry Andric { "atanpi" , {1}, {E_ANY}},
161044eb2f6SDimitry Andric { "atomic_add" , {1}, {E_VLTLPTR_ANY,E_POINTEE}},
162044eb2f6SDimitry Andric { "atomic_and" , {1}, {E_VLTLPTR_ANY,E_POINTEE}},
163044eb2f6SDimitry Andric { "atomic_cmpxchg" , {1}, {E_VLTLPTR_ANY,E_POINTEE,E_POINTEE}},
164044eb2f6SDimitry Andric { "atomic_dec" , {1}, {E_VLTLPTR_ANY}},
165044eb2f6SDimitry Andric { "atomic_inc" , {1}, {E_VLTLPTR_ANY}},
166044eb2f6SDimitry Andric { "atomic_max" , {1}, {E_VLTLPTR_ANY,E_POINTEE}},
167044eb2f6SDimitry Andric { "atomic_min" , {1}, {E_VLTLPTR_ANY,E_POINTEE}},
168044eb2f6SDimitry Andric { "atomic_or" , {1}, {E_VLTLPTR_ANY,E_POINTEE}},
169044eb2f6SDimitry Andric { "atomic_sub" , {1}, {E_VLTLPTR_ANY,E_POINTEE}},
170044eb2f6SDimitry Andric { "atomic_xchg" , {1}, {E_VLTLPTR_ANY,E_POINTEE}},
171044eb2f6SDimitry Andric { "atomic_xor" , {1}, {E_VLTLPTR_ANY,E_POINTEE}},
172044eb2f6SDimitry Andric { "bitselect" , {1}, {E_ANY,E_COPY,E_COPY}},
173044eb2f6SDimitry Andric { "cbrt" , {1}, {E_ANY}},
174044eb2f6SDimitry Andric { "ceil" , {1}, {E_ANY}},
175044eb2f6SDimitry Andric { "clamp" , {1}, {E_ANY,E_COPY,E_COPY}},
176044eb2f6SDimitry Andric { "clz" , {1}, {E_ANY}},
177044eb2f6SDimitry Andric { "commit_read_pipe" , {1}, {E_ANY,EX_RESERVEDID}},
178044eb2f6SDimitry Andric { "commit_write_pipe" , {1}, {E_ANY,EX_RESERVEDID}},
179044eb2f6SDimitry Andric { "copysign" , {1}, {E_ANY,E_COPY}},
180044eb2f6SDimitry Andric { "cos" , {1}, {E_ANY}},
181044eb2f6SDimitry Andric { "cosh" , {1}, {E_ANY}},
182044eb2f6SDimitry Andric { "cospi" , {1}, {E_ANY}},
183044eb2f6SDimitry Andric { "cross" , {1}, {E_ANY,E_COPY}},
184044eb2f6SDimitry Andric { "ctz" , {1}, {E_ANY}},
185044eb2f6SDimitry Andric { "degrees" , {1}, {E_ANY}},
186044eb2f6SDimitry Andric { "distance" , {1}, {E_ANY,E_COPY}},
187044eb2f6SDimitry Andric { "divide" , {1}, {E_ANY,E_COPY}},
188044eb2f6SDimitry Andric { "dot" , {1}, {E_ANY,E_COPY}},
189044eb2f6SDimitry Andric { "erf" , {1}, {E_ANY}},
190044eb2f6SDimitry Andric { "erfc" , {1}, {E_ANY}},
191044eb2f6SDimitry Andric { "exp" , {1}, {E_ANY}},
192044eb2f6SDimitry Andric { "exp10" , {1}, {E_ANY}},
193044eb2f6SDimitry Andric { "exp2" , {1}, {E_ANY}},
194044eb2f6SDimitry Andric { "expm1" , {1}, {E_ANY}},
195044eb2f6SDimitry Andric { "fabs" , {1}, {E_ANY}},
196044eb2f6SDimitry Andric { "fast_distance" , {1}, {E_ANY,E_COPY}},
197044eb2f6SDimitry Andric { "fast_length" , {1}, {E_ANY}},
198044eb2f6SDimitry Andric { "fast_normalize" , {1}, {E_ANY}},
199044eb2f6SDimitry Andric { "fdim" , {1}, {E_ANY,E_COPY}},
200044eb2f6SDimitry Andric { "floor" , {1}, {E_ANY}},
201044eb2f6SDimitry Andric { "fma" , {1}, {E_ANY,E_COPY,E_COPY}},
202044eb2f6SDimitry Andric { "fmax" , {1}, {E_ANY,E_COPY}},
203044eb2f6SDimitry Andric { "fmin" , {1}, {E_ANY,E_COPY}},
204044eb2f6SDimitry Andric { "fmod" , {1}, {E_ANY,E_COPY}},
205044eb2f6SDimitry Andric { "fract" , {2}, {E_POINTEE,E_ANY}},
206044eb2f6SDimitry Andric { "frexp" , {1,2}, {E_ANY,E_ANY}},
207044eb2f6SDimitry Andric { "get_image_array_size" , {1}, {E_ANY}},
208044eb2f6SDimitry Andric { "get_image_channel_data_type" , {1}, {E_ANY}},
209044eb2f6SDimitry Andric { "get_image_channel_order" , {1}, {E_ANY}},
210044eb2f6SDimitry Andric { "get_image_dim" , {1}, {E_ANY}},
211044eb2f6SDimitry Andric { "get_image_height" , {1}, {E_ANY}},
212044eb2f6SDimitry Andric { "get_image_width" , {1}, {E_ANY}},
213044eb2f6SDimitry Andric { "get_pipe_max_packets" , {1}, {E_ANY}},
214044eb2f6SDimitry Andric { "get_pipe_num_packets" , {1}, {E_ANY}},
215044eb2f6SDimitry Andric { "hadd" , {1}, {E_ANY,E_COPY}},
216044eb2f6SDimitry Andric { "hypot" , {1}, {E_ANY,E_COPY}},
217044eb2f6SDimitry Andric { "ilogb" , {1}, {E_ANY}},
218044eb2f6SDimitry Andric { "isequal" , {1}, {E_ANY,E_COPY}},
219044eb2f6SDimitry Andric { "isfinite" , {1}, {E_ANY}},
220044eb2f6SDimitry Andric { "isgreater" , {1}, {E_ANY,E_COPY}},
221044eb2f6SDimitry Andric { "isgreaterequal" , {1}, {E_ANY,E_COPY}},
222044eb2f6SDimitry Andric { "isinf" , {1}, {E_ANY}},
223044eb2f6SDimitry Andric { "isless" , {1}, {E_ANY,E_COPY}},
224044eb2f6SDimitry Andric { "islessequal" , {1}, {E_ANY,E_COPY}},
225044eb2f6SDimitry Andric { "islessgreater" , {1}, {E_ANY,E_COPY}},
226044eb2f6SDimitry Andric { "isnan" , {1}, {E_ANY}},
227044eb2f6SDimitry Andric { "isnormal" , {1}, {E_ANY}},
228044eb2f6SDimitry Andric { "isnotequal" , {1}, {E_ANY,E_COPY}},
229044eb2f6SDimitry Andric { "isordered" , {1}, {E_ANY,E_COPY}},
230044eb2f6SDimitry Andric { "isunordered" , {1}, {E_ANY,E_COPY}},
231044eb2f6SDimitry Andric { "ldexp" , {1}, {E_ANY,E_SETBASE_I32}},
232044eb2f6SDimitry Andric { "length" , {1}, {E_ANY}},
233044eb2f6SDimitry Andric { "lgamma" , {1}, {E_ANY}},
234044eb2f6SDimitry Andric { "lgamma_r" , {1,2}, {E_ANY,E_ANY}},
235044eb2f6SDimitry Andric { "log" , {1}, {E_ANY}},
236044eb2f6SDimitry Andric { "log10" , {1}, {E_ANY}},
237044eb2f6SDimitry Andric { "log1p" , {1}, {E_ANY}},
238044eb2f6SDimitry Andric { "log2" , {1}, {E_ANY}},
239044eb2f6SDimitry Andric { "logb" , {1}, {E_ANY}},
240044eb2f6SDimitry Andric { "mad" , {1}, {E_ANY,E_COPY,E_COPY}},
241044eb2f6SDimitry Andric { "mad24" , {1}, {E_ANY,E_COPY,E_COPY}},
242044eb2f6SDimitry Andric { "mad_hi" , {1}, {E_ANY,E_COPY,E_COPY}},
243044eb2f6SDimitry Andric { "mad_sat" , {1}, {E_ANY,E_COPY,E_COPY}},
244044eb2f6SDimitry Andric { "max" , {1}, {E_ANY,E_COPY}},
245044eb2f6SDimitry Andric { "maxmag" , {1}, {E_ANY,E_COPY}},
246044eb2f6SDimitry Andric { "min" , {1}, {E_ANY,E_COPY}},
247044eb2f6SDimitry Andric { "minmag" , {1}, {E_ANY,E_COPY}},
248044eb2f6SDimitry Andric { "mix" , {1}, {E_ANY,E_COPY,E_COPY}},
249044eb2f6SDimitry Andric { "modf" , {2}, {E_POINTEE,E_ANY}},
250044eb2f6SDimitry Andric { "mul24" , {1}, {E_ANY,E_COPY}},
251044eb2f6SDimitry Andric { "mul_hi" , {1}, {E_ANY,E_COPY}},
252044eb2f6SDimitry Andric { "nan" , {1}, {E_ANY}},
253044eb2f6SDimitry Andric { "nextafter" , {1}, {E_ANY,E_COPY}},
254044eb2f6SDimitry Andric { "normalize" , {1}, {E_ANY}},
255044eb2f6SDimitry Andric { "popcount" , {1}, {E_ANY}},
256044eb2f6SDimitry Andric { "pow" , {1}, {E_ANY,E_COPY}},
257044eb2f6SDimitry Andric { "pown" , {1}, {E_ANY,E_SETBASE_I32}},
258044eb2f6SDimitry Andric { "powr" , {1}, {E_ANY,E_COPY}},
259044eb2f6SDimitry Andric { "prefetch" , {1}, {E_CONSTPTR_ANY,EX_SIZET}},
260044eb2f6SDimitry Andric { "radians" , {1}, {E_ANY}},
261044eb2f6SDimitry Andric { "recip" , {1}, {E_ANY}},
262044eb2f6SDimitry Andric { "remainder" , {1}, {E_ANY,E_COPY}},
263044eb2f6SDimitry Andric { "remquo" , {1,3}, {E_ANY,E_COPY,E_ANY}},
264044eb2f6SDimitry Andric { "reserve_read_pipe" , {1}, {E_ANY,EX_UINT}},
265044eb2f6SDimitry Andric { "reserve_write_pipe" , {1}, {E_ANY,EX_UINT}},
266044eb2f6SDimitry Andric { "rhadd" , {1}, {E_ANY,E_COPY}},
267044eb2f6SDimitry Andric { "rint" , {1}, {E_ANY}},
268044eb2f6SDimitry Andric { "rootn" , {1}, {E_ANY,E_SETBASE_I32}},
269044eb2f6SDimitry Andric { "rotate" , {1}, {E_ANY,E_COPY}},
270044eb2f6SDimitry Andric { "round" , {1}, {E_ANY}},
271044eb2f6SDimitry Andric { "rsqrt" , {1}, {E_ANY}},
272044eb2f6SDimitry Andric { "select" , {1,3}, {E_ANY,E_COPY,E_ANY}},
273044eb2f6SDimitry Andric { "shuffle" , {1,2}, {E_ANY,E_ANY}},
274044eb2f6SDimitry Andric { "shuffle2" , {1,3}, {E_ANY,E_COPY,E_ANY}},
275044eb2f6SDimitry Andric { "sign" , {1}, {E_ANY}},
276044eb2f6SDimitry Andric { "signbit" , {1}, {E_ANY}},
277044eb2f6SDimitry Andric { "sin" , {1}, {E_ANY}},
278044eb2f6SDimitry Andric { "sincos" , {2}, {E_POINTEE,E_ANY}},
279044eb2f6SDimitry Andric { "sinh" , {1}, {E_ANY}},
280044eb2f6SDimitry Andric { "sinpi" , {1}, {E_ANY}},
281044eb2f6SDimitry Andric { "smoothstep" , {1}, {E_ANY,E_COPY,E_COPY}},
282044eb2f6SDimitry Andric { "sqrt" , {1}, {E_ANY}},
283044eb2f6SDimitry Andric { "step" , {1}, {E_ANY,E_COPY}},
284044eb2f6SDimitry Andric { "sub_group_broadcast" , {1}, {E_ANY,EX_UINT}},
285044eb2f6SDimitry Andric { "sub_group_commit_read_pipe" , {1}, {E_ANY,EX_RESERVEDID}},
286044eb2f6SDimitry Andric { "sub_group_commit_write_pipe" , {1}, {E_ANY,EX_RESERVEDID}},
287044eb2f6SDimitry Andric { "sub_group_reduce_add" , {1}, {E_ANY}},
288044eb2f6SDimitry Andric { "sub_group_reduce_max" , {1}, {E_ANY}},
289044eb2f6SDimitry Andric { "sub_group_reduce_min" , {1}, {E_ANY}},
290044eb2f6SDimitry Andric { "sub_group_reserve_read_pipe" , {1}, {E_ANY,EX_UINT}},
291044eb2f6SDimitry Andric { "sub_group_reserve_write_pipe" , {1}, {E_ANY,EX_UINT}},
292044eb2f6SDimitry Andric { "sub_group_scan_exclusive_add" , {1}, {E_ANY}},
293044eb2f6SDimitry Andric { "sub_group_scan_exclusive_max" , {1}, {E_ANY}},
294044eb2f6SDimitry Andric { "sub_group_scan_exclusive_min" , {1}, {E_ANY}},
295044eb2f6SDimitry Andric { "sub_group_scan_inclusive_add" , {1}, {E_ANY}},
296044eb2f6SDimitry Andric { "sub_group_scan_inclusive_max" , {1}, {E_ANY}},
297044eb2f6SDimitry Andric { "sub_group_scan_inclusive_min" , {1}, {E_ANY}},
298044eb2f6SDimitry Andric { "sub_sat" , {1}, {E_ANY,E_COPY}},
299044eb2f6SDimitry Andric { "tan" , {1}, {E_ANY}},
300044eb2f6SDimitry Andric { "tanh" , {1}, {E_ANY}},
301044eb2f6SDimitry Andric { "tanpi" , {1}, {E_ANY}},
302044eb2f6SDimitry Andric { "tgamma" , {1}, {E_ANY}},
303044eb2f6SDimitry Andric { "trunc" , {1}, {E_ANY}},
304044eb2f6SDimitry Andric { "upsample" , {1}, {E_ANY,E_MAKEBASE_UNS}},
305044eb2f6SDimitry Andric { "vec_step" , {1}, {E_ANY}},
306044eb2f6SDimitry Andric { "vstore" , {3}, {E_POINTEE,EX_SIZET,E_ANY}},
307044eb2f6SDimitry Andric { "vstore16" , {3}, {E_V16_OF_POINTEE,EX_SIZET,E_ANY}},
308044eb2f6SDimitry Andric { "vstore2" , {3}, {E_V2_OF_POINTEE,EX_SIZET,E_ANY}},
309044eb2f6SDimitry Andric { "vstore3" , {3}, {E_V3_OF_POINTEE,EX_SIZET,E_ANY}},
310044eb2f6SDimitry Andric { "vstore4" , {3}, {E_V4_OF_POINTEE,EX_SIZET,E_ANY}},
311044eb2f6SDimitry Andric { "vstore8" , {3}, {E_V8_OF_POINTEE,EX_SIZET,E_ANY}},
312044eb2f6SDimitry Andric { "work_group_commit_read_pipe" , {1}, {E_ANY,EX_RESERVEDID}},
313044eb2f6SDimitry Andric { "work_group_commit_write_pipe" , {1}, {E_ANY,EX_RESERVEDID}},
314044eb2f6SDimitry Andric { "work_group_reduce_add" , {1}, {E_ANY}},
315044eb2f6SDimitry Andric { "work_group_reduce_max" , {1}, {E_ANY}},
316044eb2f6SDimitry Andric { "work_group_reduce_min" , {1}, {E_ANY}},
317044eb2f6SDimitry Andric { "work_group_reserve_read_pipe" , {1}, {E_ANY,EX_UINT}},
318044eb2f6SDimitry Andric { "work_group_reserve_write_pipe" , {1}, {E_ANY,EX_UINT}},
319044eb2f6SDimitry Andric { "work_group_scan_exclusive_add" , {1}, {E_ANY}},
320044eb2f6SDimitry Andric { "work_group_scan_exclusive_max" , {1}, {E_ANY}},
321044eb2f6SDimitry Andric { "work_group_scan_exclusive_min" , {1}, {E_ANY}},
322044eb2f6SDimitry Andric { "work_group_scan_inclusive_add" , {1}, {E_ANY}},
323044eb2f6SDimitry Andric { "work_group_scan_inclusive_max" , {1}, {E_ANY}},
324044eb2f6SDimitry Andric { "work_group_scan_inclusive_min" , {1}, {E_ANY}},
325044eb2f6SDimitry Andric { "write_imagef" , {1}, {E_ANY,E_IMAGECOORDS,EX_FLOAT4}},
326044eb2f6SDimitry Andric { "write_imagei" , {1}, {E_ANY,E_IMAGECOORDS,EX_INTV4}},
327044eb2f6SDimitry Andric { "write_imageui" , {1}, {E_ANY,E_IMAGECOORDS,EX_UINTV4}},
328044eb2f6SDimitry Andric { "ncos" , {1}, {E_ANY} },
329044eb2f6SDimitry Andric { "nexp2" , {1}, {E_ANY} },
330044eb2f6SDimitry Andric { "nfma" , {1}, {E_ANY, E_COPY, E_COPY} },
331044eb2f6SDimitry Andric { "nlog2" , {1}, {E_ANY} },
332044eb2f6SDimitry Andric { "nrcp" , {1}, {E_ANY} },
333044eb2f6SDimitry Andric { "nrsqrt" , {1}, {E_ANY} },
334044eb2f6SDimitry Andric { "nsin" , {1}, {E_ANY} },
335044eb2f6SDimitry Andric { "nsqrt" , {1}, {E_ANY} },
336044eb2f6SDimitry Andric { "ftz" , {1}, {E_ANY} },
337044eb2f6SDimitry Andric { "fldexp" , {1}, {E_ANY, EX_UINT} },
338044eb2f6SDimitry Andric { "class" , {1}, {E_ANY, EX_UINT} },
339044eb2f6SDimitry Andric { "rcbrt" , {1}, {E_ANY} },
340044eb2f6SDimitry Andric };
341044eb2f6SDimitry Andric
342044eb2f6SDimitry Andric // Library functions with unmangled name.
343044eb2f6SDimitry Andric const UnmangledFuncInfo UnmangledFuncInfo::Table[] = {
344044eb2f6SDimitry Andric {"__read_pipe_2", 4},
345044eb2f6SDimitry Andric {"__read_pipe_4", 6},
346044eb2f6SDimitry Andric {"__write_pipe_2", 4},
347044eb2f6SDimitry Andric {"__write_pipe_4", 6},
348044eb2f6SDimitry Andric };
349044eb2f6SDimitry Andric
350044eb2f6SDimitry Andric const unsigned UnmangledFuncInfo::TableSize =
351e3b55780SDimitry Andric std::size(UnmangledFuncInfo::Table);
352044eb2f6SDimitry Andric
getRetType(AMDGPULibFunc::EFuncId id,const AMDGPULibFunc::Param (& Leads)[2])353044eb2f6SDimitry Andric static AMDGPULibFunc::Param getRetType(AMDGPULibFunc::EFuncId id,
354044eb2f6SDimitry Andric const AMDGPULibFunc::Param (&Leads)[2]) {
355044eb2f6SDimitry Andric AMDGPULibFunc::Param Res = Leads[0];
356c0981da4SDimitry Andric // TBD - This switch may require to be extended for other intrinsics
357044eb2f6SDimitry Andric switch (id) {
358044eb2f6SDimitry Andric case AMDGPULibFunc::EI_SINCOS:
359044eb2f6SDimitry Andric Res.PtrKind = AMDGPULibFunc::BYVALUE;
360044eb2f6SDimitry Andric break;
361044eb2f6SDimitry Andric default:
362044eb2f6SDimitry Andric break;
363044eb2f6SDimitry Andric }
364044eb2f6SDimitry Andric return Res;
365044eb2f6SDimitry Andric }
366044eb2f6SDimitry Andric
367044eb2f6SDimitry Andric class ParamIterator {
368044eb2f6SDimitry Andric const AMDGPULibFunc::Param (&Leads)[2];
369044eb2f6SDimitry Andric const ManglingRule& Rule;
370ac9a064cSDimitry Andric int Index = 0;
371044eb2f6SDimitry Andric public:
ParamIterator(const AMDGPULibFunc::Param (& leads)[2],const ManglingRule & rule)372044eb2f6SDimitry Andric ParamIterator(const AMDGPULibFunc::Param (&leads)[2],
373044eb2f6SDimitry Andric const ManglingRule& rule)
374ac9a064cSDimitry Andric : Leads(leads), Rule(rule) {}
375044eb2f6SDimitry Andric
376044eb2f6SDimitry Andric AMDGPULibFunc::Param getNextParam();
377044eb2f6SDimitry Andric };
378044eb2f6SDimitry Andric
getNextParam()379044eb2f6SDimitry Andric AMDGPULibFunc::Param ParamIterator::getNextParam() {
380044eb2f6SDimitry Andric AMDGPULibFunc::Param P;
381044eb2f6SDimitry Andric if (Index >= int(sizeof Rule.Param/sizeof Rule.Param[0])) return P;
382044eb2f6SDimitry Andric
383044eb2f6SDimitry Andric const char R = Rule.Param[Index];
384044eb2f6SDimitry Andric switch (R) {
385044eb2f6SDimitry Andric case E_NONE: break;
386044eb2f6SDimitry Andric case EX_UINT:
387044eb2f6SDimitry Andric P.ArgType = AMDGPULibFunc::U32; break;
388044eb2f6SDimitry Andric case EX_INTV4:
389044eb2f6SDimitry Andric P.ArgType = AMDGPULibFunc::I32; P.VectorSize = 4; break;
390044eb2f6SDimitry Andric case EX_UINTV4:
391044eb2f6SDimitry Andric P.ArgType = AMDGPULibFunc::U32; P.VectorSize = 4; break;
392044eb2f6SDimitry Andric case EX_FLOAT4:
393044eb2f6SDimitry Andric P.ArgType = AMDGPULibFunc::F32; P.VectorSize = 4; break;
394044eb2f6SDimitry Andric case EX_SIZET:
395044eb2f6SDimitry Andric P.ArgType = AMDGPULibFunc::U64; break;
396044eb2f6SDimitry Andric case EX_EVENT:
397044eb2f6SDimitry Andric P.ArgType = AMDGPULibFunc::EVENT; break;
398044eb2f6SDimitry Andric case EX_SAMPLER:
399044eb2f6SDimitry Andric P.ArgType = AMDGPULibFunc::SAMPLER; break;
400044eb2f6SDimitry Andric case EX_RESERVEDID: break; // TBD
401044eb2f6SDimitry Andric default:
402044eb2f6SDimitry Andric if (Index == (Rule.Lead[1] - 1)) P = Leads[1];
403044eb2f6SDimitry Andric else P = Leads[0];
404044eb2f6SDimitry Andric
405044eb2f6SDimitry Andric switch (R) {
406044eb2f6SDimitry Andric case E_ANY:
407044eb2f6SDimitry Andric case E_COPY: break;
408044eb2f6SDimitry Andric
409044eb2f6SDimitry Andric case E_POINTEE:
410044eb2f6SDimitry Andric P.PtrKind = AMDGPULibFunc::BYVALUE; break;
411044eb2f6SDimitry Andric case E_V2_OF_POINTEE:
412044eb2f6SDimitry Andric P.VectorSize = 2; P.PtrKind = AMDGPULibFunc::BYVALUE; break;
413044eb2f6SDimitry Andric case E_V3_OF_POINTEE:
414044eb2f6SDimitry Andric P.VectorSize = 3; P.PtrKind = AMDGPULibFunc::BYVALUE; break;
415044eb2f6SDimitry Andric case E_V4_OF_POINTEE:
416044eb2f6SDimitry Andric P.VectorSize = 4; P.PtrKind = AMDGPULibFunc::BYVALUE; break;
417044eb2f6SDimitry Andric case E_V8_OF_POINTEE:
418044eb2f6SDimitry Andric P.VectorSize = 8; P.PtrKind = AMDGPULibFunc::BYVALUE; break;
419044eb2f6SDimitry Andric case E_V16_OF_POINTEE:
420044eb2f6SDimitry Andric P.VectorSize = 16; P.PtrKind = AMDGPULibFunc::BYVALUE; break;
421044eb2f6SDimitry Andric case E_CONSTPTR_ANY:
422044eb2f6SDimitry Andric P.PtrKind |= AMDGPULibFunc::CONST; break;
423044eb2f6SDimitry Andric case E_VLTLPTR_ANY:
424044eb2f6SDimitry Andric P.PtrKind |= AMDGPULibFunc::VOLATILE; break;
425044eb2f6SDimitry Andric case E_SETBASE_I32:
426044eb2f6SDimitry Andric P.ArgType = AMDGPULibFunc::I32; break;
427044eb2f6SDimitry Andric case E_SETBASE_U32:
428044eb2f6SDimitry Andric P.ArgType = AMDGPULibFunc::U32; break;
429044eb2f6SDimitry Andric
430044eb2f6SDimitry Andric case E_MAKEBASE_UNS:
431044eb2f6SDimitry Andric P.ArgType &= ~AMDGPULibFunc::BASE_TYPE_MASK;
432044eb2f6SDimitry Andric P.ArgType |= AMDGPULibFunc::UINT;
433044eb2f6SDimitry Andric break;
434044eb2f6SDimitry Andric
435044eb2f6SDimitry Andric case E_IMAGECOORDS:
436044eb2f6SDimitry Andric switch (P.ArgType) {
437044eb2f6SDimitry Andric case AMDGPULibFunc::IMG1DA: P.VectorSize = 2; break;
438044eb2f6SDimitry Andric case AMDGPULibFunc::IMG1DB: P.VectorSize = 1; break;
439044eb2f6SDimitry Andric case AMDGPULibFunc::IMG2DA: P.VectorSize = 4; break;
440044eb2f6SDimitry Andric case AMDGPULibFunc::IMG1D: P.VectorSize = 1; break;
441044eb2f6SDimitry Andric case AMDGPULibFunc::IMG2D: P.VectorSize = 2; break;
442044eb2f6SDimitry Andric case AMDGPULibFunc::IMG3D: P.VectorSize = 4; break;
443044eb2f6SDimitry Andric }
444044eb2f6SDimitry Andric P.PtrKind = AMDGPULibFunc::BYVALUE;
445044eb2f6SDimitry Andric P.ArgType = AMDGPULibFunc::I32;
446044eb2f6SDimitry Andric break;
447044eb2f6SDimitry Andric
448044eb2f6SDimitry Andric case E_CONSTPTR_SWAPGL: {
449044eb2f6SDimitry Andric unsigned AS = AMDGPULibFunc::getAddrSpaceFromEPtrKind(P.PtrKind);
450044eb2f6SDimitry Andric switch (AS) {
451044eb2f6SDimitry Andric case AMDGPUAS::GLOBAL_ADDRESS: AS = AMDGPUAS::LOCAL_ADDRESS; break;
452044eb2f6SDimitry Andric case AMDGPUAS::LOCAL_ADDRESS: AS = AMDGPUAS::GLOBAL_ADDRESS; break;
453044eb2f6SDimitry Andric }
454044eb2f6SDimitry Andric P.PtrKind = AMDGPULibFunc::getEPtrKindFromAddrSpace(AS);
455044eb2f6SDimitry Andric P.PtrKind |= AMDGPULibFunc::CONST;
456044eb2f6SDimitry Andric break;
457044eb2f6SDimitry Andric }
458044eb2f6SDimitry Andric
459c0981da4SDimitry Andric default:
460c0981da4SDimitry Andric llvm_unreachable("Unhandled param rule");
461044eb2f6SDimitry Andric }
462044eb2f6SDimitry Andric }
463044eb2f6SDimitry Andric ++Index;
464044eb2f6SDimitry Andric return P;
465044eb2f6SDimitry Andric }
466044eb2f6SDimitry Andric
drop_front(StringRef & str,size_t n=1)467044eb2f6SDimitry Andric inline static void drop_front(StringRef& str, size_t n = 1) {
468044eb2f6SDimitry Andric str = str.drop_front(n);
469044eb2f6SDimitry Andric }
470044eb2f6SDimitry Andric
eatTerm(StringRef & mangledName,const char c)471044eb2f6SDimitry Andric static bool eatTerm(StringRef& mangledName, const char c) {
472044eb2f6SDimitry Andric if (mangledName.front() == c) {
473044eb2f6SDimitry Andric drop_front(mangledName);
474044eb2f6SDimitry Andric return true;
475044eb2f6SDimitry Andric }
476044eb2f6SDimitry Andric return false;
477044eb2f6SDimitry Andric }
478044eb2f6SDimitry Andric
479044eb2f6SDimitry Andric template <size_t N>
eatTerm(StringRef & mangledName,const char (& str)[N])480044eb2f6SDimitry Andric static bool eatTerm(StringRef& mangledName, const char (&str)[N]) {
481312c0ed1SDimitry Andric if (mangledName.starts_with(StringRef(str, N - 1))) {
482044eb2f6SDimitry Andric drop_front(mangledName, N-1);
483044eb2f6SDimitry Andric return true;
484044eb2f6SDimitry Andric }
485044eb2f6SDimitry Andric return false;
486044eb2f6SDimitry Andric }
487044eb2f6SDimitry Andric
eatNumber(StringRef & s)488044eb2f6SDimitry Andric static int eatNumber(StringRef& s) {
489044eb2f6SDimitry Andric size_t const savedSize = s.size();
490044eb2f6SDimitry Andric int n = 0;
491044eb2f6SDimitry Andric while (!s.empty() && isDigit(s.front())) {
492044eb2f6SDimitry Andric n = n*10 + s.front() - '0';
493044eb2f6SDimitry Andric drop_front(s);
494044eb2f6SDimitry Andric }
495044eb2f6SDimitry Andric return s.size() < savedSize ? n : -1;
496044eb2f6SDimitry Andric }
497044eb2f6SDimitry Andric
eatLengthPrefixedName(StringRef & mangledName)498044eb2f6SDimitry Andric static StringRef eatLengthPrefixedName(StringRef& mangledName) {
499044eb2f6SDimitry Andric int const Len = eatNumber(mangledName);
500044eb2f6SDimitry Andric if (Len <= 0 || static_cast<size_t>(Len) > mangledName.size())
501044eb2f6SDimitry Andric return StringRef();
502044eb2f6SDimitry Andric StringRef Res = mangledName.substr(0, Len);
503044eb2f6SDimitry Andric drop_front(mangledName, Len);
504044eb2f6SDimitry Andric return Res;
505044eb2f6SDimitry Andric }
506044eb2f6SDimitry Andric
507044eb2f6SDimitry Andric } // end anonymous namespace
508044eb2f6SDimitry Andric
AMDGPUMangledLibFunc()509044eb2f6SDimitry Andric AMDGPUMangledLibFunc::AMDGPUMangledLibFunc() {
510044eb2f6SDimitry Andric FuncId = EI_NONE;
511044eb2f6SDimitry Andric FKind = NOPFX;
512044eb2f6SDimitry Andric Leads[0].reset();
513044eb2f6SDimitry Andric Leads[1].reset();
514044eb2f6SDimitry Andric Name.clear();
515044eb2f6SDimitry Andric }
516044eb2f6SDimitry Andric
AMDGPUUnmangledLibFunc()517044eb2f6SDimitry Andric AMDGPUUnmangledLibFunc::AMDGPUUnmangledLibFunc() {
518044eb2f6SDimitry Andric FuncId = EI_NONE;
519044eb2f6SDimitry Andric FuncTy = nullptr;
520044eb2f6SDimitry Andric }
521044eb2f6SDimitry Andric
AMDGPUMangledLibFunc(EFuncId id,const AMDGPUMangledLibFunc & copyFrom)522044eb2f6SDimitry Andric AMDGPUMangledLibFunc::AMDGPUMangledLibFunc(
523044eb2f6SDimitry Andric EFuncId id, const AMDGPUMangledLibFunc ©From) {
524044eb2f6SDimitry Andric FuncId = id;
525044eb2f6SDimitry Andric FKind = copyFrom.FKind;
526044eb2f6SDimitry Andric Leads[0] = copyFrom.Leads[0];
527044eb2f6SDimitry Andric Leads[1] = copyFrom.Leads[1];
528044eb2f6SDimitry Andric }
529044eb2f6SDimitry Andric
AMDGPUMangledLibFunc(EFuncId id,FunctionType * FT,bool SignedInts)530b1c73532SDimitry Andric AMDGPUMangledLibFunc::AMDGPUMangledLibFunc(EFuncId id, FunctionType *FT,
531b1c73532SDimitry Andric bool SignedInts) {
532b1c73532SDimitry Andric FuncId = id;
533b1c73532SDimitry Andric unsigned NumArgs = FT->getNumParams();
534b1c73532SDimitry Andric if (NumArgs >= 1)
535b1c73532SDimitry Andric Leads[0] = Param::getFromTy(FT->getParamType(0), SignedInts);
536b1c73532SDimitry Andric if (NumArgs >= 2)
537b1c73532SDimitry Andric Leads[1] = Param::getFromTy(FT->getParamType(1), SignedInts);
538b1c73532SDimitry Andric }
539b1c73532SDimitry Andric
540044eb2f6SDimitry Andric ///////////////////////////////////////////////////////////////////////////////
541044eb2f6SDimitry Andric // Demangling
542044eb2f6SDimitry Andric
parseVecSize(StringRef & mangledName)543044eb2f6SDimitry Andric static int parseVecSize(StringRef& mangledName) {
544044eb2f6SDimitry Andric size_t const Len = eatNumber(mangledName);
545044eb2f6SDimitry Andric switch (Len) {
546044eb2f6SDimitry Andric case 2: case 3: case 4: case 8: case 16:
547044eb2f6SDimitry Andric return Len;
548044eb2f6SDimitry Andric default:
549044eb2f6SDimitry Andric break;
550044eb2f6SDimitry Andric }
551044eb2f6SDimitry Andric return 1;
552044eb2f6SDimitry Andric }
553044eb2f6SDimitry Andric
parseNamePrefix(StringRef & mangledName)554044eb2f6SDimitry Andric static AMDGPULibFunc::ENamePrefix parseNamePrefix(StringRef& mangledName) {
555044eb2f6SDimitry Andric std::pair<StringRef, StringRef> const P = mangledName.split('_');
556044eb2f6SDimitry Andric AMDGPULibFunc::ENamePrefix Pfx =
557044eb2f6SDimitry Andric StringSwitch<AMDGPULibFunc::ENamePrefix>(P.first)
558044eb2f6SDimitry Andric .Case("native", AMDGPULibFunc::NATIVE)
559044eb2f6SDimitry Andric .Case("half" , AMDGPULibFunc::HALF)
560044eb2f6SDimitry Andric .Default(AMDGPULibFunc::NOPFX);
561044eb2f6SDimitry Andric
562044eb2f6SDimitry Andric if (Pfx != AMDGPULibFunc::NOPFX)
563044eb2f6SDimitry Andric mangledName = P.second;
564044eb2f6SDimitry Andric
565044eb2f6SDimitry Andric return Pfx;
566044eb2f6SDimitry Andric }
567044eb2f6SDimitry Andric
buildManglingRulesMap()568e6d15924SDimitry Andric StringMap<int> ManglingRule::buildManglingRulesMap() {
569e3b55780SDimitry Andric StringMap<int> Map(std::size(manglingRules));
570e6d15924SDimitry Andric int Id = 0;
571e6d15924SDimitry Andric for (auto Rule : manglingRules)
572e6d15924SDimitry Andric Map.insert({Rule.Name, Id++});
573e6d15924SDimitry Andric return Map;
574e6d15924SDimitry Andric }
575e6d15924SDimitry Andric
parseUnmangledName(StringRef FullName)576044eb2f6SDimitry Andric bool AMDGPUMangledLibFunc::parseUnmangledName(StringRef FullName) {
577e6d15924SDimitry Andric static const StringMap<int> manglingRulesMap =
578e6d15924SDimitry Andric ManglingRule::buildManglingRulesMap();
579044eb2f6SDimitry Andric FuncId = static_cast<EFuncId>(manglingRulesMap.lookup(FullName));
580044eb2f6SDimitry Andric return FuncId != EI_NONE;
581044eb2f6SDimitry Andric }
582044eb2f6SDimitry Andric
583044eb2f6SDimitry Andric ///////////////////////////////////////////////////////////////////////////////
584044eb2f6SDimitry Andric // Itanium Demangling
585044eb2f6SDimitry Andric
586044eb2f6SDimitry Andric namespace {
587044eb2f6SDimitry Andric struct ItaniumParamParser {
588044eb2f6SDimitry Andric AMDGPULibFunc::Param Prev;
589044eb2f6SDimitry Andric bool parseItaniumParam(StringRef& param, AMDGPULibFunc::Param &res);
590044eb2f6SDimitry Andric };
591044eb2f6SDimitry Andric } // namespace
592044eb2f6SDimitry Andric
parseItaniumParam(StringRef & param,AMDGPULibFunc::Param & res)593044eb2f6SDimitry Andric bool ItaniumParamParser::parseItaniumParam(StringRef& param,
594044eb2f6SDimitry Andric AMDGPULibFunc::Param &res) {
595044eb2f6SDimitry Andric res.reset();
596044eb2f6SDimitry Andric if (param.empty()) return false;
597044eb2f6SDimitry Andric
598044eb2f6SDimitry Andric // parse pointer prefix
599044eb2f6SDimitry Andric if (eatTerm(param, 'P')) {
600044eb2f6SDimitry Andric if (eatTerm(param, 'K')) res.PtrKind |= AMDGPULibFunc::CONST;
601044eb2f6SDimitry Andric if (eatTerm(param, 'V')) res.PtrKind |= AMDGPULibFunc::VOLATILE;
602044eb2f6SDimitry Andric unsigned AS;
603044eb2f6SDimitry Andric if (!eatTerm(param, "U3AS")) {
604044eb2f6SDimitry Andric AS = 0;
605044eb2f6SDimitry Andric } else {
606044eb2f6SDimitry Andric AS = param.front() - '0';
607044eb2f6SDimitry Andric drop_front(param, 1);
608044eb2f6SDimitry Andric }
609044eb2f6SDimitry Andric res.PtrKind |= AMDGPULibFuncBase::getEPtrKindFromAddrSpace(AS);
610044eb2f6SDimitry Andric } else {
611044eb2f6SDimitry Andric res.PtrKind = AMDGPULibFunc::BYVALUE;
612044eb2f6SDimitry Andric }
613044eb2f6SDimitry Andric
614044eb2f6SDimitry Andric // parse vector size
615044eb2f6SDimitry Andric if (eatTerm(param,"Dv")) {
616044eb2f6SDimitry Andric res.VectorSize = parseVecSize(param);
617044eb2f6SDimitry Andric if (res.VectorSize==1 || !eatTerm(param, '_')) return false;
618044eb2f6SDimitry Andric }
619044eb2f6SDimitry Andric
620044eb2f6SDimitry Andric // parse type
621044eb2f6SDimitry Andric char const TC = param.front();
622cfca06d7SDimitry Andric if (isDigit(TC)) {
623044eb2f6SDimitry Andric res.ArgType = StringSwitch<AMDGPULibFunc::EType>
624044eb2f6SDimitry Andric (eatLengthPrefixedName(param))
625044eb2f6SDimitry Andric .Case("ocl_image1darray" , AMDGPULibFunc::IMG1DA)
626044eb2f6SDimitry Andric .Case("ocl_image1dbuffer", AMDGPULibFunc::IMG1DB)
627044eb2f6SDimitry Andric .Case("ocl_image2darray" , AMDGPULibFunc::IMG2DA)
628044eb2f6SDimitry Andric .Case("ocl_image1d" , AMDGPULibFunc::IMG1D)
629044eb2f6SDimitry Andric .Case("ocl_image2d" , AMDGPULibFunc::IMG2D)
630044eb2f6SDimitry Andric .Case("ocl_image3d" , AMDGPULibFunc::IMG3D)
631044eb2f6SDimitry Andric .Case("ocl_event" , AMDGPULibFunc::DUMMY)
632044eb2f6SDimitry Andric .Case("ocl_sampler" , AMDGPULibFunc::DUMMY)
633044eb2f6SDimitry Andric .Default(AMDGPULibFunc::DUMMY);
634044eb2f6SDimitry Andric } else {
635044eb2f6SDimitry Andric drop_front(param);
636044eb2f6SDimitry Andric switch (TC) {
637044eb2f6SDimitry Andric case 'h': res.ArgType = AMDGPULibFunc::U8; break;
638044eb2f6SDimitry Andric case 't': res.ArgType = AMDGPULibFunc::U16; break;
639044eb2f6SDimitry Andric case 'j': res.ArgType = AMDGPULibFunc::U32; break;
640044eb2f6SDimitry Andric case 'm': res.ArgType = AMDGPULibFunc::U64; break;
641044eb2f6SDimitry Andric case 'c': res.ArgType = AMDGPULibFunc::I8; break;
642044eb2f6SDimitry Andric case 's': res.ArgType = AMDGPULibFunc::I16; break;
643044eb2f6SDimitry Andric case 'i': res.ArgType = AMDGPULibFunc::I32; break;
644044eb2f6SDimitry Andric case 'l': res.ArgType = AMDGPULibFunc::I64; break;
645044eb2f6SDimitry Andric case 'f': res.ArgType = AMDGPULibFunc::F32; break;
646044eb2f6SDimitry Andric case 'd': res.ArgType = AMDGPULibFunc::F64; break;
647044eb2f6SDimitry Andric case 'D': if (!eatTerm(param, 'h')) return false;
648044eb2f6SDimitry Andric res.ArgType = AMDGPULibFunc::F16; break;
649044eb2f6SDimitry Andric case 'S':
650044eb2f6SDimitry Andric if (!eatTerm(param, '_')) {
651044eb2f6SDimitry Andric eatNumber(param);
652044eb2f6SDimitry Andric if (!eatTerm(param, '_')) return false;
653044eb2f6SDimitry Andric }
654044eb2f6SDimitry Andric res.VectorSize = Prev.VectorSize;
655044eb2f6SDimitry Andric res.ArgType = Prev.ArgType;
656044eb2f6SDimitry Andric break;
657044eb2f6SDimitry Andric default:;
658044eb2f6SDimitry Andric }
659044eb2f6SDimitry Andric }
660044eb2f6SDimitry Andric if (res.ArgType == 0) return false;
661044eb2f6SDimitry Andric Prev.VectorSize = res.VectorSize;
662044eb2f6SDimitry Andric Prev.ArgType = res.ArgType;
663044eb2f6SDimitry Andric return true;
664044eb2f6SDimitry Andric }
665044eb2f6SDimitry Andric
parseFuncName(StringRef & mangledName)666044eb2f6SDimitry Andric bool AMDGPUMangledLibFunc::parseFuncName(StringRef &mangledName) {
667044eb2f6SDimitry Andric StringRef Name = eatLengthPrefixedName(mangledName);
668044eb2f6SDimitry Andric FKind = parseNamePrefix(Name);
669044eb2f6SDimitry Andric if (!parseUnmangledName(Name))
670044eb2f6SDimitry Andric return false;
671044eb2f6SDimitry Andric
672044eb2f6SDimitry Andric const ManglingRule& Rule = manglingRules[FuncId];
673044eb2f6SDimitry Andric ItaniumParamParser Parser;
674044eb2f6SDimitry Andric for (int I=0; I < Rule.maxLeadIndex(); ++I) {
675044eb2f6SDimitry Andric Param P;
676044eb2f6SDimitry Andric if (!Parser.parseItaniumParam(mangledName, P))
677044eb2f6SDimitry Andric return false;
678044eb2f6SDimitry Andric
679044eb2f6SDimitry Andric if ((I + 1) == Rule.Lead[0]) Leads[0] = P;
680044eb2f6SDimitry Andric if ((I + 1) == Rule.Lead[1]) Leads[1] = P;
681044eb2f6SDimitry Andric }
682044eb2f6SDimitry Andric return true;
683044eb2f6SDimitry Andric }
684044eb2f6SDimitry Andric
parseFuncName(StringRef & Name)685044eb2f6SDimitry Andric bool AMDGPUUnmangledLibFunc::parseFuncName(StringRef &Name) {
686044eb2f6SDimitry Andric if (!UnmangledFuncInfo::lookup(Name, FuncId))
687044eb2f6SDimitry Andric return false;
688044eb2f6SDimitry Andric setName(Name);
689044eb2f6SDimitry Andric return true;
690044eb2f6SDimitry Andric }
691044eb2f6SDimitry Andric
parse(StringRef FuncName,AMDGPULibFunc & F)692044eb2f6SDimitry Andric bool AMDGPULibFunc::parse(StringRef FuncName, AMDGPULibFunc &F) {
693044eb2f6SDimitry Andric if (FuncName.empty()) {
694044eb2f6SDimitry Andric F.Impl = std::unique_ptr<AMDGPULibFuncImpl>();
695044eb2f6SDimitry Andric return false;
696044eb2f6SDimitry Andric }
697044eb2f6SDimitry Andric
698044eb2f6SDimitry Andric if (eatTerm(FuncName, "_Z"))
6991d5ae102SDimitry Andric F.Impl = std::make_unique<AMDGPUMangledLibFunc>();
700044eb2f6SDimitry Andric else
7011d5ae102SDimitry Andric F.Impl = std::make_unique<AMDGPUUnmangledLibFunc>();
702044eb2f6SDimitry Andric if (F.Impl->parseFuncName(FuncName))
703044eb2f6SDimitry Andric return true;
704044eb2f6SDimitry Andric
705044eb2f6SDimitry Andric F.Impl = std::unique_ptr<AMDGPULibFuncImpl>();
706044eb2f6SDimitry Andric return false;
707044eb2f6SDimitry Andric }
708044eb2f6SDimitry Andric
getUnmangledName(StringRef mangledName)709044eb2f6SDimitry Andric StringRef AMDGPUMangledLibFunc::getUnmangledName(StringRef mangledName) {
710044eb2f6SDimitry Andric StringRef S = mangledName;
711044eb2f6SDimitry Andric if (eatTerm(S, "_Z"))
712044eb2f6SDimitry Andric return eatLengthPrefixedName(S);
713044eb2f6SDimitry Andric return StringRef();
714044eb2f6SDimitry Andric }
715044eb2f6SDimitry Andric
716044eb2f6SDimitry Andric ///////////////////////////////////////////////////////////////////////////////
717044eb2f6SDimitry Andric // Mangling
718044eb2f6SDimitry Andric
719044eb2f6SDimitry Andric template <typename Stream>
writeName(Stream & OS) const720044eb2f6SDimitry Andric void AMDGPUMangledLibFunc::writeName(Stream &OS) const {
721044eb2f6SDimitry Andric const char *Pfx = "";
722044eb2f6SDimitry Andric switch (FKind) {
723044eb2f6SDimitry Andric case NATIVE: Pfx = "native_"; break;
724044eb2f6SDimitry Andric case HALF: Pfx = "half_"; break;
725044eb2f6SDimitry Andric default: break;
726044eb2f6SDimitry Andric }
727044eb2f6SDimitry Andric if (!Name.empty()) {
728044eb2f6SDimitry Andric OS << Pfx << Name;
729044eb2f6SDimitry Andric } else if (FuncId != EI_NONE) {
730044eb2f6SDimitry Andric OS << Pfx;
731044eb2f6SDimitry Andric const StringRef& S = manglingRules[FuncId].Name;
732044eb2f6SDimitry Andric OS.write(S.data(), S.size());
733044eb2f6SDimitry Andric }
734044eb2f6SDimitry Andric }
735044eb2f6SDimitry Andric
mangle() const736044eb2f6SDimitry Andric std::string AMDGPUMangledLibFunc::mangle() const { return mangleNameItanium(); }
737044eb2f6SDimitry Andric
738044eb2f6SDimitry Andric ///////////////////////////////////////////////////////////////////////////////
739044eb2f6SDimitry Andric // Itanium Mangling
740044eb2f6SDimitry Andric
getItaniumTypeName(AMDGPULibFunc::EType T)741044eb2f6SDimitry Andric static const char *getItaniumTypeName(AMDGPULibFunc::EType T) {
742044eb2f6SDimitry Andric switch (T) {
743044eb2f6SDimitry Andric case AMDGPULibFunc::U8: return "h";
744044eb2f6SDimitry Andric case AMDGPULibFunc::U16: return "t";
745044eb2f6SDimitry Andric case AMDGPULibFunc::U32: return "j";
746044eb2f6SDimitry Andric case AMDGPULibFunc::U64: return "m";
747044eb2f6SDimitry Andric case AMDGPULibFunc::I8: return "c";
748044eb2f6SDimitry Andric case AMDGPULibFunc::I16: return "s";
749044eb2f6SDimitry Andric case AMDGPULibFunc::I32: return "i";
750044eb2f6SDimitry Andric case AMDGPULibFunc::I64: return "l";
751044eb2f6SDimitry Andric case AMDGPULibFunc::F16: return "Dh";
752044eb2f6SDimitry Andric case AMDGPULibFunc::F32: return "f";
753044eb2f6SDimitry Andric case AMDGPULibFunc::F64: return "d";
754044eb2f6SDimitry Andric case AMDGPULibFunc::IMG1DA: return "16ocl_image1darray";
755044eb2f6SDimitry Andric case AMDGPULibFunc::IMG1DB: return "17ocl_image1dbuffer";
756044eb2f6SDimitry Andric case AMDGPULibFunc::IMG2DA: return "16ocl_image2darray";
757044eb2f6SDimitry Andric case AMDGPULibFunc::IMG1D: return "11ocl_image1d";
758044eb2f6SDimitry Andric case AMDGPULibFunc::IMG2D: return "11ocl_image2d";
759044eb2f6SDimitry Andric case AMDGPULibFunc::IMG3D: return "11ocl_image3d";
760044eb2f6SDimitry Andric case AMDGPULibFunc::SAMPLER: return "11ocl_sampler";
761044eb2f6SDimitry Andric case AMDGPULibFunc::EVENT: return "9ocl_event";
762c0981da4SDimitry Andric default:
763c0981da4SDimitry Andric llvm_unreachable("Unhandled param type");
764044eb2f6SDimitry Andric }
765044eb2f6SDimitry Andric return nullptr;
766044eb2f6SDimitry Andric }
767044eb2f6SDimitry Andric
768044eb2f6SDimitry Andric namespace {
769044eb2f6SDimitry Andric // Itanium mangling ABI says:
770044eb2f6SDimitry Andric // "5.1.8. Compression
771044eb2f6SDimitry Andric // ... Each non-terminal in the grammar for which <substitution> appears on the
772044eb2f6SDimitry Andric // right-hand side is both a source of future substitutions and a candidate
773044eb2f6SDimitry Andric // for being substituted. There are two exceptions that appear to be
774044eb2f6SDimitry Andric // substitution candidates from the grammar, but are explicitly excluded:
775044eb2f6SDimitry Andric // 1. <builtin-type> other than vendor extended types ..."
776044eb2f6SDimitry Andric
777c0981da4SDimitry Andric // For the purpose of functions the following productions make sense for the
778044eb2f6SDimitry Andric // substitution:
779044eb2f6SDimitry Andric // <type> ::= <builtin-type>
780044eb2f6SDimitry Andric // ::= <class-enum-type>
781044eb2f6SDimitry Andric // ::= <array-type>
782044eb2f6SDimitry Andric // ::=<CV-qualifiers> <type>
783044eb2f6SDimitry Andric // ::= P <type> # pointer-to
784044eb2f6SDimitry Andric // ::= <substitution>
785044eb2f6SDimitry Andric //
786044eb2f6SDimitry Andric // Note that while types like images, samplers and events are by the ABI encoded
787044eb2f6SDimitry Andric // using <class-enum-type> production rule they're not used for substitution
788044eb2f6SDimitry Andric // because clang consider them as builtin types.
789044eb2f6SDimitry Andric //
790c0981da4SDimitry Andric // DvNN_ type is GCC extension for vectors and is a subject for the
791c0981da4SDimitry Andric // substitution.
792044eb2f6SDimitry Andric
793044eb2f6SDimitry Andric class ItaniumMangler {
794c0981da4SDimitry Andric SmallVector<AMDGPULibFunc::Param, 10> Str; // list of accumulated substitutions
795044eb2f6SDimitry Andric bool UseAddrSpace;
796044eb2f6SDimitry Andric
findSubst(const AMDGPULibFunc::Param & P) const797044eb2f6SDimitry Andric int findSubst(const AMDGPULibFunc::Param& P) const {
798044eb2f6SDimitry Andric for(unsigned I = 0; I < Str.size(); ++I) {
799044eb2f6SDimitry Andric const AMDGPULibFunc::Param& T = Str[I];
800044eb2f6SDimitry Andric if (P.PtrKind == T.PtrKind &&
801044eb2f6SDimitry Andric P.VectorSize == T.VectorSize &&
802044eb2f6SDimitry Andric P.ArgType == T.ArgType) {
803044eb2f6SDimitry Andric return I;
804044eb2f6SDimitry Andric }
805044eb2f6SDimitry Andric }
806044eb2f6SDimitry Andric return -1;
807044eb2f6SDimitry Andric }
808044eb2f6SDimitry Andric
809044eb2f6SDimitry Andric template <typename Stream>
trySubst(Stream & os,const AMDGPULibFunc::Param & p)810044eb2f6SDimitry Andric bool trySubst(Stream& os, const AMDGPULibFunc::Param& p) {
811044eb2f6SDimitry Andric int const subst = findSubst(p);
812044eb2f6SDimitry Andric if (subst < 0) return false;
813044eb2f6SDimitry Andric // Substitutions are mangled as S(XX)?_ where XX is a hexadecimal number
814044eb2f6SDimitry Andric // 0 1 2
815044eb2f6SDimitry Andric // S_ S0_ S1_
816044eb2f6SDimitry Andric if (subst == 0) os << "S_";
817044eb2f6SDimitry Andric else os << 'S' << (subst-1) << '_';
818044eb2f6SDimitry Andric return true;
819044eb2f6SDimitry Andric }
820044eb2f6SDimitry Andric
821044eb2f6SDimitry Andric public:
ItaniumMangler(bool useAddrSpace)822044eb2f6SDimitry Andric ItaniumMangler(bool useAddrSpace)
823044eb2f6SDimitry Andric : UseAddrSpace(useAddrSpace) {}
824044eb2f6SDimitry Andric
825044eb2f6SDimitry Andric template <typename Stream>
operator ()(Stream & os,AMDGPULibFunc::Param p)826044eb2f6SDimitry Andric void operator()(Stream& os, AMDGPULibFunc::Param p) {
827044eb2f6SDimitry Andric
828044eb2f6SDimitry Andric // Itanium mangling ABI 5.1.8. Compression:
829044eb2f6SDimitry Andric // Logically, the substitutable components of a mangled name are considered
830044eb2f6SDimitry Andric // left-to-right, components before the composite structure of which they
831044eb2f6SDimitry Andric // are a part. If a component has been encountered before, it is substituted
832044eb2f6SDimitry Andric // as described below. This decision is independent of whether its components
833044eb2f6SDimitry Andric // have been substituted, so an implementation may optimize by considering
834044eb2f6SDimitry Andric // large structures for substitution before their components. If a component
835044eb2f6SDimitry Andric // has not been encountered before, its mangling is identified, and it is
836044eb2f6SDimitry Andric // added to a dictionary of substitution candidates. No entity is added to
837044eb2f6SDimitry Andric // the dictionary twice.
838044eb2f6SDimitry Andric AMDGPULibFunc::Param Ptr;
839044eb2f6SDimitry Andric
840044eb2f6SDimitry Andric if (p.PtrKind) {
841044eb2f6SDimitry Andric if (trySubst(os, p)) return;
842044eb2f6SDimitry Andric os << 'P';
843044eb2f6SDimitry Andric if (p.PtrKind & AMDGPULibFunc::CONST) os << 'K';
844044eb2f6SDimitry Andric if (p.PtrKind & AMDGPULibFunc::VOLATILE) os << 'V';
845044eb2f6SDimitry Andric unsigned AS = UseAddrSpace
846044eb2f6SDimitry Andric ? AMDGPULibFuncBase::getAddrSpaceFromEPtrKind(p.PtrKind)
847044eb2f6SDimitry Andric : 0;
848344a3780SDimitry Andric if (EnableOCLManglingMismatchWA || AS != 0)
849344a3780SDimitry Andric os << "U3AS" << AS;
850044eb2f6SDimitry Andric Ptr = p;
851044eb2f6SDimitry Andric p.PtrKind = 0;
852044eb2f6SDimitry Andric }
853044eb2f6SDimitry Andric
854044eb2f6SDimitry Andric if (p.VectorSize > 1) {
855044eb2f6SDimitry Andric if (trySubst(os, p)) goto exit;
856044eb2f6SDimitry Andric Str.push_back(p);
857044eb2f6SDimitry Andric os << "Dv" << static_cast<unsigned>(p.VectorSize) << '_';
858044eb2f6SDimitry Andric }
859044eb2f6SDimitry Andric
860044eb2f6SDimitry Andric os << getItaniumTypeName((AMDGPULibFunc::EType)p.ArgType);
861044eb2f6SDimitry Andric
862044eb2f6SDimitry Andric exit:
863044eb2f6SDimitry Andric if (Ptr.ArgType) Str.push_back(Ptr);
864044eb2f6SDimitry Andric }
865044eb2f6SDimitry Andric };
866044eb2f6SDimitry Andric } // namespace
867044eb2f6SDimitry Andric
mangleNameItanium() const868044eb2f6SDimitry Andric std::string AMDGPUMangledLibFunc::mangleNameItanium() const {
869044eb2f6SDimitry Andric SmallString<128> Buf;
870044eb2f6SDimitry Andric raw_svector_ostream S(Buf);
871044eb2f6SDimitry Andric SmallString<128> NameBuf;
872044eb2f6SDimitry Andric raw_svector_ostream Name(NameBuf);
873044eb2f6SDimitry Andric writeName(Name);
874044eb2f6SDimitry Andric const StringRef& NameStr = Name.str();
875044eb2f6SDimitry Andric S << "_Z" << static_cast<int>(NameStr.size()) << NameStr;
876044eb2f6SDimitry Andric
877044eb2f6SDimitry Andric ItaniumMangler Mangler(true);
878044eb2f6SDimitry Andric ParamIterator I(Leads, manglingRules[FuncId]);
879044eb2f6SDimitry Andric Param P;
880044eb2f6SDimitry Andric while ((P = I.getNextParam()).ArgType != 0)
881044eb2f6SDimitry Andric Mangler(S, P);
882cfca06d7SDimitry Andric return std::string(S.str());
883044eb2f6SDimitry Andric }
884044eb2f6SDimitry Andric
885044eb2f6SDimitry Andric ///////////////////////////////////////////////////////////////////////////////
886044eb2f6SDimitry Andric // Misc
887044eb2f6SDimitry Andric
getFromTy(Type * Ty,bool Signed)888b1c73532SDimitry Andric AMDGPULibFuncBase::Param AMDGPULibFuncBase::Param::getFromTy(Type *Ty,
889b1c73532SDimitry Andric bool Signed) {
890b1c73532SDimitry Andric Param P;
891b1c73532SDimitry Andric if (FixedVectorType *VT = dyn_cast<FixedVectorType>(Ty)) {
892b1c73532SDimitry Andric P.VectorSize = VT->getNumElements();
893b1c73532SDimitry Andric Ty = VT->getElementType();
894b1c73532SDimitry Andric }
895b1c73532SDimitry Andric
896b1c73532SDimitry Andric switch (Ty->getTypeID()) {
897b1c73532SDimitry Andric case Type::FloatTyID:
898b1c73532SDimitry Andric P.ArgType = AMDGPULibFunc::F32;
899b1c73532SDimitry Andric break;
900b1c73532SDimitry Andric case Type::DoubleTyID:
901b1c73532SDimitry Andric P.ArgType = AMDGPULibFunc::F64;
902b1c73532SDimitry Andric break;
903b1c73532SDimitry Andric case Type::HalfTyID:
904b1c73532SDimitry Andric P.ArgType = AMDGPULibFunc::F16;
905b1c73532SDimitry Andric break;
906b1c73532SDimitry Andric case Type::IntegerTyID:
907b1c73532SDimitry Andric switch (cast<IntegerType>(Ty)->getBitWidth()) {
908b1c73532SDimitry Andric case 8:
909b1c73532SDimitry Andric P.ArgType = Signed ? AMDGPULibFunc::I8 : AMDGPULibFunc::U8;
910b1c73532SDimitry Andric break;
911b1c73532SDimitry Andric case 16:
912b1c73532SDimitry Andric P.ArgType = Signed ? AMDGPULibFunc::I16 : AMDGPULibFunc::U16;
913b1c73532SDimitry Andric break;
914b1c73532SDimitry Andric case 32:
915b1c73532SDimitry Andric P.ArgType = Signed ? AMDGPULibFunc::I32 : AMDGPULibFunc::U32;
916b1c73532SDimitry Andric break;
917b1c73532SDimitry Andric case 64:
918b1c73532SDimitry Andric P.ArgType = Signed ? AMDGPULibFunc::I64 : AMDGPULibFunc::U64;
919b1c73532SDimitry Andric break;
920b1c73532SDimitry Andric default:
921b1c73532SDimitry Andric llvm_unreachable("unhandled libcall argument type");
922b1c73532SDimitry Andric }
923b1c73532SDimitry Andric
924b1c73532SDimitry Andric break;
925b1c73532SDimitry Andric default:
926b1c73532SDimitry Andric llvm_unreachable("unhandled libcall argument type");
927b1c73532SDimitry Andric }
928b1c73532SDimitry Andric
929b1c73532SDimitry Andric return P;
930b1c73532SDimitry Andric }
931b1c73532SDimitry Andric
getIntrinsicParamType(LLVMContext & C,const AMDGPULibFunc::Param & P,bool useAddrSpace)932044eb2f6SDimitry Andric static Type* getIntrinsicParamType(
933044eb2f6SDimitry Andric LLVMContext& C,
934044eb2f6SDimitry Andric const AMDGPULibFunc::Param& P,
935044eb2f6SDimitry Andric bool useAddrSpace) {
936044eb2f6SDimitry Andric Type* T = nullptr;
937044eb2f6SDimitry Andric switch (P.ArgType) {
938044eb2f6SDimitry Andric case AMDGPULibFunc::U8:
939044eb2f6SDimitry Andric case AMDGPULibFunc::I8: T = Type::getInt8Ty(C); break;
940044eb2f6SDimitry Andric case AMDGPULibFunc::U16:
941044eb2f6SDimitry Andric case AMDGPULibFunc::I16: T = Type::getInt16Ty(C); break;
942044eb2f6SDimitry Andric case AMDGPULibFunc::U32:
943044eb2f6SDimitry Andric case AMDGPULibFunc::I32: T = Type::getInt32Ty(C); break;
944044eb2f6SDimitry Andric case AMDGPULibFunc::U64:
945044eb2f6SDimitry Andric case AMDGPULibFunc::I64: T = Type::getInt64Ty(C); break;
946044eb2f6SDimitry Andric case AMDGPULibFunc::F16: T = Type::getHalfTy(C); break;
947044eb2f6SDimitry Andric case AMDGPULibFunc::F32: T = Type::getFloatTy(C); break;
948044eb2f6SDimitry Andric case AMDGPULibFunc::F64: T = Type::getDoubleTy(C); break;
949044eb2f6SDimitry Andric
950044eb2f6SDimitry Andric case AMDGPULibFunc::IMG1DA:
951044eb2f6SDimitry Andric case AMDGPULibFunc::IMG1DB:
952044eb2f6SDimitry Andric case AMDGPULibFunc::IMG2DA:
953044eb2f6SDimitry Andric case AMDGPULibFunc::IMG1D:
954044eb2f6SDimitry Andric case AMDGPULibFunc::IMG2D:
955044eb2f6SDimitry Andric case AMDGPULibFunc::IMG3D:
956044eb2f6SDimitry Andric T = StructType::create(C,"ocl_image")->getPointerTo(); break;
957044eb2f6SDimitry Andric case AMDGPULibFunc::SAMPLER:
958044eb2f6SDimitry Andric T = StructType::create(C,"ocl_sampler")->getPointerTo(); break;
959044eb2f6SDimitry Andric case AMDGPULibFunc::EVENT:
960044eb2f6SDimitry Andric T = StructType::create(C,"ocl_event")->getPointerTo(); break;
961044eb2f6SDimitry Andric default:
962c0981da4SDimitry Andric llvm_unreachable("Unhandled param type");
963044eb2f6SDimitry Andric return nullptr;
964044eb2f6SDimitry Andric }
965044eb2f6SDimitry Andric if (P.VectorSize > 1)
966cfca06d7SDimitry Andric T = FixedVectorType::get(T, P.VectorSize);
967044eb2f6SDimitry Andric if (P.PtrKind != AMDGPULibFunc::BYVALUE)
968044eb2f6SDimitry Andric T = useAddrSpace ? T->getPointerTo((P.PtrKind & AMDGPULibFunc::ADDR_SPACE)
969044eb2f6SDimitry Andric - 1)
970044eb2f6SDimitry Andric : T->getPointerTo();
971044eb2f6SDimitry Andric return T;
972044eb2f6SDimitry Andric }
973044eb2f6SDimitry Andric
getFunctionType(Module & M) const974044eb2f6SDimitry Andric FunctionType *AMDGPUMangledLibFunc::getFunctionType(Module &M) const {
975044eb2f6SDimitry Andric LLVMContext& C = M.getContext();
976044eb2f6SDimitry Andric std::vector<Type*> Args;
977044eb2f6SDimitry Andric ParamIterator I(Leads, manglingRules[FuncId]);
978044eb2f6SDimitry Andric Param P;
979044eb2f6SDimitry Andric while ((P=I.getNextParam()).ArgType != 0)
980044eb2f6SDimitry Andric Args.push_back(getIntrinsicParamType(C, P, true));
981044eb2f6SDimitry Andric
982044eb2f6SDimitry Andric return FunctionType::get(
983044eb2f6SDimitry Andric getIntrinsicParamType(C, getRetType(FuncId, Leads), true),
984044eb2f6SDimitry Andric Args, false);
985044eb2f6SDimitry Andric }
986044eb2f6SDimitry Andric
getNumArgs() const987044eb2f6SDimitry Andric unsigned AMDGPUMangledLibFunc::getNumArgs() const {
988044eb2f6SDimitry Andric return manglingRules[FuncId].getNumArgs();
989044eb2f6SDimitry Andric }
990044eb2f6SDimitry Andric
getNumArgs() const991044eb2f6SDimitry Andric unsigned AMDGPUUnmangledLibFunc::getNumArgs() const {
992044eb2f6SDimitry Andric return UnmangledFuncInfo::getNumArgs(FuncId);
993044eb2f6SDimitry Andric }
994044eb2f6SDimitry Andric
getName() const995044eb2f6SDimitry Andric std::string AMDGPUMangledLibFunc::getName() const {
996044eb2f6SDimitry Andric SmallString<128> Buf;
997044eb2f6SDimitry Andric raw_svector_ostream OS(Buf);
998044eb2f6SDimitry Andric writeName(OS);
999cfca06d7SDimitry Andric return std::string(OS.str());
1000044eb2f6SDimitry Andric }
1001044eb2f6SDimitry Andric
isCompatibleSignature(const FunctionType * FuncTy) const1002b1c73532SDimitry Andric bool AMDGPULibFunc::isCompatibleSignature(const FunctionType *FuncTy) const {
1003b1c73532SDimitry Andric // TODO: Validate types make sense
1004b1c73532SDimitry Andric return !FuncTy->isVarArg() && FuncTy->getNumParams() == getNumArgs();
1005b1c73532SDimitry Andric }
1006b1c73532SDimitry Andric
getFunction(Module * M,const AMDGPULibFunc & fInfo)1007044eb2f6SDimitry Andric Function *AMDGPULibFunc::getFunction(Module *M, const AMDGPULibFunc &fInfo) {
1008044eb2f6SDimitry Andric std::string FuncName = fInfo.mangle();
1009044eb2f6SDimitry Andric Function *F = dyn_cast_or_null<Function>(
1010044eb2f6SDimitry Andric M->getValueSymbolTable().lookup(FuncName));
1011b1c73532SDimitry Andric if (!F || F->isDeclaration())
1012044eb2f6SDimitry Andric return nullptr;
1013b1c73532SDimitry Andric
1014b1c73532SDimitry Andric if (F->hasFnAttribute(Attribute::NoBuiltin))
1015b1c73532SDimitry Andric return nullptr;
1016b1c73532SDimitry Andric
1017b1c73532SDimitry Andric if (!fInfo.isCompatibleSignature(F->getFunctionType()))
1018b1c73532SDimitry Andric return nullptr;
1019b1c73532SDimitry Andric
1020b1c73532SDimitry Andric return F;
1021044eb2f6SDimitry Andric }
1022044eb2f6SDimitry Andric
getOrInsertFunction(Module * M,const AMDGPULibFunc & fInfo)1023e6d15924SDimitry Andric FunctionCallee AMDGPULibFunc::getOrInsertFunction(Module *M,
1024044eb2f6SDimitry Andric const AMDGPULibFunc &fInfo) {
1025044eb2f6SDimitry Andric std::string const FuncName = fInfo.mangle();
1026044eb2f6SDimitry Andric Function *F = dyn_cast_or_null<Function>(
1027044eb2f6SDimitry Andric M->getValueSymbolTable().lookup(FuncName));
1028044eb2f6SDimitry Andric
1029b1c73532SDimitry Andric if (F) {
1030b1c73532SDimitry Andric if (F->hasFnAttribute(Attribute::NoBuiltin))
1031b1c73532SDimitry Andric return nullptr;
1032b1c73532SDimitry Andric if (!F->isDeclaration() &&
1033b1c73532SDimitry Andric fInfo.isCompatibleSignature(F->getFunctionType()))
1034044eb2f6SDimitry Andric return F;
1035044eb2f6SDimitry Andric }
1036044eb2f6SDimitry Andric
1037044eb2f6SDimitry Andric FunctionType *FuncTy = fInfo.getFunctionType(*M);
1038044eb2f6SDimitry Andric
1039044eb2f6SDimitry Andric bool hasPtr = false;
1040044eb2f6SDimitry Andric for (FunctionType::param_iterator
1041044eb2f6SDimitry Andric PI = FuncTy->param_begin(),
1042044eb2f6SDimitry Andric PE = FuncTy->param_end();
1043044eb2f6SDimitry Andric PI != PE; ++PI) {
1044044eb2f6SDimitry Andric const Type* argTy = static_cast<const Type*>(*PI);
1045044eb2f6SDimitry Andric if (argTy->isPointerTy()) {
1046044eb2f6SDimitry Andric hasPtr = true;
1047044eb2f6SDimitry Andric break;
1048044eb2f6SDimitry Andric }
1049044eb2f6SDimitry Andric }
1050044eb2f6SDimitry Andric
1051e6d15924SDimitry Andric FunctionCallee C;
1052044eb2f6SDimitry Andric if (hasPtr) {
1053044eb2f6SDimitry Andric // Do not set extra attributes for functions with pointer arguments.
1054044eb2f6SDimitry Andric C = M->getOrInsertFunction(FuncName, FuncTy);
1055044eb2f6SDimitry Andric } else {
1056044eb2f6SDimitry Andric AttributeList Attr;
1057044eb2f6SDimitry Andric LLVMContext &Ctx = M->getContext();
1058e3b55780SDimitry Andric Attr = Attr.addFnAttribute(
1059e3b55780SDimitry Andric Ctx, Attribute::getWithMemoryEffects(Ctx, MemoryEffects::readOnly()));
1060c0981da4SDimitry Andric Attr = Attr.addFnAttribute(Ctx, Attribute::NoUnwind);
1061044eb2f6SDimitry Andric C = M->getOrInsertFunction(FuncName, FuncTy, Attr);
1062044eb2f6SDimitry Andric }
1063044eb2f6SDimitry Andric
1064e6d15924SDimitry Andric return C;
1065e6d15924SDimitry Andric }
1066e6d15924SDimitry Andric
buildNameMap()1067e6d15924SDimitry Andric StringMap<unsigned> UnmangledFuncInfo::buildNameMap() {
1068e6d15924SDimitry Andric StringMap<unsigned> Map;
1069e6d15924SDimitry Andric for (unsigned I = 0; I != TableSize; ++I)
1070e6d15924SDimitry Andric Map[Table[I].Name] = I;
1071e6d15924SDimitry Andric return Map;
1072044eb2f6SDimitry Andric }
1073044eb2f6SDimitry Andric
lookup(StringRef Name,ID & Id)1074044eb2f6SDimitry Andric bool UnmangledFuncInfo::lookup(StringRef Name, ID &Id) {
1075e6d15924SDimitry Andric static const StringMap<unsigned> Map = buildNameMap();
1076044eb2f6SDimitry Andric auto Loc = Map.find(Name);
1077044eb2f6SDimitry Andric if (Loc != Map.end()) {
1078044eb2f6SDimitry Andric Id = toFuncId(Loc->second);
1079044eb2f6SDimitry Andric return true;
1080044eb2f6SDimitry Andric }
1081044eb2f6SDimitry Andric Id = AMDGPULibFunc::EI_NONE;
1082044eb2f6SDimitry Andric return false;
1083044eb2f6SDimitry Andric }
1084044eb2f6SDimitry Andric
AMDGPULibFunc(const AMDGPULibFunc & F)1085044eb2f6SDimitry Andric AMDGPULibFunc::AMDGPULibFunc(const AMDGPULibFunc &F) {
1086044eb2f6SDimitry Andric if (auto *MF = dyn_cast<AMDGPUMangledLibFunc>(F.Impl.get()))
1087ac9a064cSDimitry Andric Impl = std::make_unique<AMDGPUMangledLibFunc>(*MF);
1088044eb2f6SDimitry Andric else if (auto *UMF = dyn_cast<AMDGPUUnmangledLibFunc>(F.Impl.get()))
1089ac9a064cSDimitry Andric Impl = std::make_unique<AMDGPUUnmangledLibFunc>(*UMF);
1090044eb2f6SDimitry Andric else
1091044eb2f6SDimitry Andric Impl = std::unique_ptr<AMDGPULibFuncImpl>();
1092044eb2f6SDimitry Andric }
1093044eb2f6SDimitry Andric
operator =(const AMDGPULibFunc & F)1094044eb2f6SDimitry Andric AMDGPULibFunc &AMDGPULibFunc::operator=(const AMDGPULibFunc &F) {
1095044eb2f6SDimitry Andric if (this == &F)
1096044eb2f6SDimitry Andric return *this;
1097044eb2f6SDimitry Andric new (this) AMDGPULibFunc(F);
1098044eb2f6SDimitry Andric return *this;
1099044eb2f6SDimitry Andric }
1100044eb2f6SDimitry Andric
AMDGPULibFunc(EFuncId Id,const AMDGPULibFunc & CopyFrom)1101044eb2f6SDimitry Andric AMDGPULibFunc::AMDGPULibFunc(EFuncId Id, const AMDGPULibFunc &CopyFrom) {
1102044eb2f6SDimitry Andric assert(AMDGPULibFuncBase::isMangled(Id) && CopyFrom.isMangled() &&
1103044eb2f6SDimitry Andric "not supported");
1104ac9a064cSDimitry Andric Impl = std::make_unique<AMDGPUMangledLibFunc>(
1105ac9a064cSDimitry Andric Id, *cast<AMDGPUMangledLibFunc>(CopyFrom.Impl.get()));
1106044eb2f6SDimitry Andric }
1107044eb2f6SDimitry Andric
AMDGPULibFunc(EFuncId Id,FunctionType * FT,bool SignedInts)1108b1c73532SDimitry Andric AMDGPULibFunc::AMDGPULibFunc(EFuncId Id, FunctionType *FT, bool SignedInts) {
1109ac9a064cSDimitry Andric Impl = std::make_unique<AMDGPUMangledLibFunc>(Id, FT, SignedInts);
1110b1c73532SDimitry Andric }
1111b1c73532SDimitry Andric
AMDGPULibFunc(StringRef Name,FunctionType * FT)1112044eb2f6SDimitry Andric AMDGPULibFunc::AMDGPULibFunc(StringRef Name, FunctionType *FT) {
1113ac9a064cSDimitry Andric Impl = std::make_unique<AMDGPUUnmangledLibFunc>(Name, FT);
1114044eb2f6SDimitry Andric }
1115044eb2f6SDimitry Andric
initMangled()1116ac9a064cSDimitry Andric void AMDGPULibFunc::initMangled() {
1117ac9a064cSDimitry Andric Impl = std::make_unique<AMDGPUMangledLibFunc>();
1118ac9a064cSDimitry Andric }
1119044eb2f6SDimitry Andric
getLeads()1120044eb2f6SDimitry Andric AMDGPULibFunc::Param *AMDGPULibFunc::getLeads() {
1121044eb2f6SDimitry Andric if (!Impl)
1122044eb2f6SDimitry Andric initMangled();
1123044eb2f6SDimitry Andric return cast<AMDGPUMangledLibFunc>(Impl.get())->Leads;
1124044eb2f6SDimitry Andric }
1125044eb2f6SDimitry Andric
getLeads() const1126044eb2f6SDimitry Andric const AMDGPULibFunc::Param *AMDGPULibFunc::getLeads() const {
1127044eb2f6SDimitry Andric return cast<const AMDGPUMangledLibFunc>(Impl.get())->Leads;
1128044eb2f6SDimitry Andric }
1129