xref: /src/contrib/llvm-project/llvm/lib/Target/AMDGPU/AMDGPULibFunc.cpp (revision 0fca6ea1d4eea4c934cfff25ac9ee8ad6fe95583)
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 &copyFrom) {
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