xref: /netbsd-src/external/apache2/llvm/dist/llvm/lib/IR/AutoUpgrade.cpp (revision 82d56013d7b633d116a93943de88e08335357a7c)
1 //===-- AutoUpgrade.cpp - Implement auto-upgrade helper functions ---------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 //
9 // This file implements the auto-upgrade helper functions.
10 // This is where deprecated IR intrinsics and other IR features are updated to
11 // current specifications.
12 //
13 //===----------------------------------------------------------------------===//
14 
15 #include "llvm/IR/AutoUpgrade.h"
16 #include "llvm/ADT/StringSwitch.h"
17 #include "llvm/IR/Constants.h"
18 #include "llvm/IR/DIBuilder.h"
19 #include "llvm/IR/DebugInfo.h"
20 #include "llvm/IR/DiagnosticInfo.h"
21 #include "llvm/IR/Function.h"
22 #include "llvm/IR/IRBuilder.h"
23 #include "llvm/IR/InstVisitor.h"
24 #include "llvm/IR/Instruction.h"
25 #include "llvm/IR/IntrinsicInst.h"
26 #include "llvm/IR/Intrinsics.h"
27 #include "llvm/IR/IntrinsicsAArch64.h"
28 #include "llvm/IR/IntrinsicsARM.h"
29 #include "llvm/IR/IntrinsicsX86.h"
30 #include "llvm/IR/LLVMContext.h"
31 #include "llvm/IR/Module.h"
32 #include "llvm/IR/Verifier.h"
33 #include "llvm/Support/ErrorHandling.h"
34 #include "llvm/Support/Regex.h"
35 #include <cstring>
36 using namespace llvm;
37 
rename(GlobalValue * GV)38 static void rename(GlobalValue *GV) { GV->setName(GV->getName() + ".old"); }
39 
40 // Upgrade the declarations of the SSE4.1 ptest intrinsics whose arguments have
41 // changed their type from v4f32 to v2i64.
UpgradePTESTIntrinsic(Function * F,Intrinsic::ID IID,Function * & NewFn)42 static bool UpgradePTESTIntrinsic(Function* F, Intrinsic::ID IID,
43                                   Function *&NewFn) {
44   // Check whether this is an old version of the function, which received
45   // v4f32 arguments.
46   Type *Arg0Type = F->getFunctionType()->getParamType(0);
47   if (Arg0Type != FixedVectorType::get(Type::getFloatTy(F->getContext()), 4))
48     return false;
49 
50   // Yes, it's old, replace it with new version.
51   rename(F);
52   NewFn = Intrinsic::getDeclaration(F->getParent(), IID);
53   return true;
54 }
55 
56 // Upgrade the declarations of intrinsic functions whose 8-bit immediate mask
57 // arguments have changed their type from i32 to i8.
UpgradeX86IntrinsicsWith8BitMask(Function * F,Intrinsic::ID IID,Function * & NewFn)58 static bool UpgradeX86IntrinsicsWith8BitMask(Function *F, Intrinsic::ID IID,
59                                              Function *&NewFn) {
60   // Check that the last argument is an i32.
61   Type *LastArgType = F->getFunctionType()->getParamType(
62      F->getFunctionType()->getNumParams() - 1);
63   if (!LastArgType->isIntegerTy(32))
64     return false;
65 
66   // Move this function aside and map down.
67   rename(F);
68   NewFn = Intrinsic::getDeclaration(F->getParent(), IID);
69   return true;
70 }
71 
72 // Upgrade the declaration of fp compare intrinsics that change return type
73 // from scalar to vXi1 mask.
UpgradeX86MaskedFPCompare(Function * F,Intrinsic::ID IID,Function * & NewFn)74 static bool UpgradeX86MaskedFPCompare(Function *F, Intrinsic::ID IID,
75                                       Function *&NewFn) {
76   // Check if the return type is a vector.
77   if (F->getReturnType()->isVectorTy())
78     return false;
79 
80   rename(F);
81   NewFn = Intrinsic::getDeclaration(F->getParent(), IID);
82   return true;
83 }
84 
ShouldUpgradeX86Intrinsic(Function * F,StringRef Name)85 static bool ShouldUpgradeX86Intrinsic(Function *F, StringRef Name) {
86   // All of the intrinsics matches below should be marked with which llvm
87   // version started autoupgrading them. At some point in the future we would
88   // like to use this information to remove upgrade code for some older
89   // intrinsics. It is currently undecided how we will determine that future
90   // point.
91   if (Name == "addcarryx.u32" || // Added in 8.0
92       Name == "addcarryx.u64" || // Added in 8.0
93       Name == "addcarry.u32" || // Added in 8.0
94       Name == "addcarry.u64" || // Added in 8.0
95       Name == "subborrow.u32" || // Added in 8.0
96       Name == "subborrow.u64" || // Added in 8.0
97       Name.startswith("sse2.padds.") || // Added in 8.0
98       Name.startswith("sse2.psubs.") || // Added in 8.0
99       Name.startswith("sse2.paddus.") || // Added in 8.0
100       Name.startswith("sse2.psubus.") || // Added in 8.0
101       Name.startswith("avx2.padds.") || // Added in 8.0
102       Name.startswith("avx2.psubs.") || // Added in 8.0
103       Name.startswith("avx2.paddus.") || // Added in 8.0
104       Name.startswith("avx2.psubus.") || // Added in 8.0
105       Name.startswith("avx512.padds.") || // Added in 8.0
106       Name.startswith("avx512.psubs.") || // Added in 8.0
107       Name.startswith("avx512.mask.padds.") || // Added in 8.0
108       Name.startswith("avx512.mask.psubs.") || // Added in 8.0
109       Name.startswith("avx512.mask.paddus.") || // Added in 8.0
110       Name.startswith("avx512.mask.psubus.") || // Added in 8.0
111       Name=="ssse3.pabs.b.128" || // Added in 6.0
112       Name=="ssse3.pabs.w.128" || // Added in 6.0
113       Name=="ssse3.pabs.d.128" || // Added in 6.0
114       Name.startswith("fma4.vfmadd.s") || // Added in 7.0
115       Name.startswith("fma.vfmadd.") || // Added in 7.0
116       Name.startswith("fma.vfmsub.") || // Added in 7.0
117       Name.startswith("fma.vfmsubadd.") || // Added in 7.0
118       Name.startswith("fma.vfnmadd.") || // Added in 7.0
119       Name.startswith("fma.vfnmsub.") || // Added in 7.0
120       Name.startswith("avx512.mask.vfmadd.") || // Added in 7.0
121       Name.startswith("avx512.mask.vfnmadd.") || // Added in 7.0
122       Name.startswith("avx512.mask.vfnmsub.") || // Added in 7.0
123       Name.startswith("avx512.mask3.vfmadd.") || // Added in 7.0
124       Name.startswith("avx512.maskz.vfmadd.") || // Added in 7.0
125       Name.startswith("avx512.mask3.vfmsub.") || // Added in 7.0
126       Name.startswith("avx512.mask3.vfnmsub.") || // Added in 7.0
127       Name.startswith("avx512.mask.vfmaddsub.") || // Added in 7.0
128       Name.startswith("avx512.maskz.vfmaddsub.") || // Added in 7.0
129       Name.startswith("avx512.mask3.vfmaddsub.") || // Added in 7.0
130       Name.startswith("avx512.mask3.vfmsubadd.") || // Added in 7.0
131       Name.startswith("avx512.mask.shuf.i") || // Added in 6.0
132       Name.startswith("avx512.mask.shuf.f") || // Added in 6.0
133       Name.startswith("avx512.kunpck") || //added in 6.0
134       Name.startswith("avx2.pabs.") || // Added in 6.0
135       Name.startswith("avx512.mask.pabs.") || // Added in 6.0
136       Name.startswith("avx512.broadcastm") || // Added in 6.0
137       Name == "sse.sqrt.ss" || // Added in 7.0
138       Name == "sse2.sqrt.sd" || // Added in 7.0
139       Name.startswith("avx512.mask.sqrt.p") || // Added in 7.0
140       Name.startswith("avx.sqrt.p") || // Added in 7.0
141       Name.startswith("sse2.sqrt.p") || // Added in 7.0
142       Name.startswith("sse.sqrt.p") || // Added in 7.0
143       Name.startswith("avx512.mask.pbroadcast") || // Added in 6.0
144       Name.startswith("sse2.pcmpeq.") || // Added in 3.1
145       Name.startswith("sse2.pcmpgt.") || // Added in 3.1
146       Name.startswith("avx2.pcmpeq.") || // Added in 3.1
147       Name.startswith("avx2.pcmpgt.") || // Added in 3.1
148       Name.startswith("avx512.mask.pcmpeq.") || // Added in 3.9
149       Name.startswith("avx512.mask.pcmpgt.") || // Added in 3.9
150       Name.startswith("avx.vperm2f128.") || // Added in 6.0
151       Name == "avx2.vperm2i128" || // Added in 6.0
152       Name == "sse.add.ss" || // Added in 4.0
153       Name == "sse2.add.sd" || // Added in 4.0
154       Name == "sse.sub.ss" || // Added in 4.0
155       Name == "sse2.sub.sd" || // Added in 4.0
156       Name == "sse.mul.ss" || // Added in 4.0
157       Name == "sse2.mul.sd" || // Added in 4.0
158       Name == "sse.div.ss" || // Added in 4.0
159       Name == "sse2.div.sd" || // Added in 4.0
160       Name == "sse41.pmaxsb" || // Added in 3.9
161       Name == "sse2.pmaxs.w" || // Added in 3.9
162       Name == "sse41.pmaxsd" || // Added in 3.9
163       Name == "sse2.pmaxu.b" || // Added in 3.9
164       Name == "sse41.pmaxuw" || // Added in 3.9
165       Name == "sse41.pmaxud" || // Added in 3.9
166       Name == "sse41.pminsb" || // Added in 3.9
167       Name == "sse2.pmins.w" || // Added in 3.9
168       Name == "sse41.pminsd" || // Added in 3.9
169       Name == "sse2.pminu.b" || // Added in 3.9
170       Name == "sse41.pminuw" || // Added in 3.9
171       Name == "sse41.pminud" || // Added in 3.9
172       Name == "avx512.kand.w" || // Added in 7.0
173       Name == "avx512.kandn.w" || // Added in 7.0
174       Name == "avx512.knot.w" || // Added in 7.0
175       Name == "avx512.kor.w" || // Added in 7.0
176       Name == "avx512.kxor.w" || // Added in 7.0
177       Name == "avx512.kxnor.w" || // Added in 7.0
178       Name == "avx512.kortestc.w" || // Added in 7.0
179       Name == "avx512.kortestz.w" || // Added in 7.0
180       Name.startswith("avx512.mask.pshuf.b.") || // Added in 4.0
181       Name.startswith("avx2.pmax") || // Added in 3.9
182       Name.startswith("avx2.pmin") || // Added in 3.9
183       Name.startswith("avx512.mask.pmax") || // Added in 4.0
184       Name.startswith("avx512.mask.pmin") || // Added in 4.0
185       Name.startswith("avx2.vbroadcast") || // Added in 3.8
186       Name.startswith("avx2.pbroadcast") || // Added in 3.8
187       Name.startswith("avx.vpermil.") || // Added in 3.1
188       Name.startswith("sse2.pshuf") || // Added in 3.9
189       Name.startswith("avx512.pbroadcast") || // Added in 3.9
190       Name.startswith("avx512.mask.broadcast.s") || // Added in 3.9
191       Name.startswith("avx512.mask.movddup") || // Added in 3.9
192       Name.startswith("avx512.mask.movshdup") || // Added in 3.9
193       Name.startswith("avx512.mask.movsldup") || // Added in 3.9
194       Name.startswith("avx512.mask.pshuf.d.") || // Added in 3.9
195       Name.startswith("avx512.mask.pshufl.w.") || // Added in 3.9
196       Name.startswith("avx512.mask.pshufh.w.") || // Added in 3.9
197       Name.startswith("avx512.mask.shuf.p") || // Added in 4.0
198       Name.startswith("avx512.mask.vpermil.p") || // Added in 3.9
199       Name.startswith("avx512.mask.perm.df.") || // Added in 3.9
200       Name.startswith("avx512.mask.perm.di.") || // Added in 3.9
201       Name.startswith("avx512.mask.punpckl") || // Added in 3.9
202       Name.startswith("avx512.mask.punpckh") || // Added in 3.9
203       Name.startswith("avx512.mask.unpckl.") || // Added in 3.9
204       Name.startswith("avx512.mask.unpckh.") || // Added in 3.9
205       Name.startswith("avx512.mask.pand.") || // Added in 3.9
206       Name.startswith("avx512.mask.pandn.") || // Added in 3.9
207       Name.startswith("avx512.mask.por.") || // Added in 3.9
208       Name.startswith("avx512.mask.pxor.") || // Added in 3.9
209       Name.startswith("avx512.mask.and.") || // Added in 3.9
210       Name.startswith("avx512.mask.andn.") || // Added in 3.9
211       Name.startswith("avx512.mask.or.") || // Added in 3.9
212       Name.startswith("avx512.mask.xor.") || // Added in 3.9
213       Name.startswith("avx512.mask.padd.") || // Added in 4.0
214       Name.startswith("avx512.mask.psub.") || // Added in 4.0
215       Name.startswith("avx512.mask.pmull.") || // Added in 4.0
216       Name.startswith("avx512.mask.cvtdq2pd.") || // Added in 4.0
217       Name.startswith("avx512.mask.cvtudq2pd.") || // Added in 4.0
218       Name.startswith("avx512.mask.cvtudq2ps.") || // Added in 7.0 updated 9.0
219       Name.startswith("avx512.mask.cvtqq2pd.") || // Added in 7.0 updated 9.0
220       Name.startswith("avx512.mask.cvtuqq2pd.") || // Added in 7.0 updated 9.0
221       Name.startswith("avx512.mask.cvtdq2ps.") || // Added in 7.0 updated 9.0
222       Name == "avx512.mask.vcvtph2ps.128" || // Added in 11.0
223       Name == "avx512.mask.vcvtph2ps.256" || // Added in 11.0
224       Name == "avx512.mask.cvtqq2ps.256" || // Added in 9.0
225       Name == "avx512.mask.cvtqq2ps.512" || // Added in 9.0
226       Name == "avx512.mask.cvtuqq2ps.256" || // Added in 9.0
227       Name == "avx512.mask.cvtuqq2ps.512" || // Added in 9.0
228       Name == "avx512.mask.cvtpd2dq.256" || // Added in 7.0
229       Name == "avx512.mask.cvtpd2ps.256" || // Added in 7.0
230       Name == "avx512.mask.cvttpd2dq.256" || // Added in 7.0
231       Name == "avx512.mask.cvttps2dq.128" || // Added in 7.0
232       Name == "avx512.mask.cvttps2dq.256" || // Added in 7.0
233       Name == "avx512.mask.cvtps2pd.128" || // Added in 7.0
234       Name == "avx512.mask.cvtps2pd.256" || // Added in 7.0
235       Name == "avx512.cvtusi2sd" || // Added in 7.0
236       Name.startswith("avx512.mask.permvar.") || // Added in 7.0
237       Name == "sse2.pmulu.dq" || // Added in 7.0
238       Name == "sse41.pmuldq" || // Added in 7.0
239       Name == "avx2.pmulu.dq" || // Added in 7.0
240       Name == "avx2.pmul.dq" || // Added in 7.0
241       Name == "avx512.pmulu.dq.512" || // Added in 7.0
242       Name == "avx512.pmul.dq.512" || // Added in 7.0
243       Name.startswith("avx512.mask.pmul.dq.") || // Added in 4.0
244       Name.startswith("avx512.mask.pmulu.dq.") || // Added in 4.0
245       Name.startswith("avx512.mask.pmul.hr.sw.") || // Added in 7.0
246       Name.startswith("avx512.mask.pmulh.w.") || // Added in 7.0
247       Name.startswith("avx512.mask.pmulhu.w.") || // Added in 7.0
248       Name.startswith("avx512.mask.pmaddw.d.") || // Added in 7.0
249       Name.startswith("avx512.mask.pmaddubs.w.") || // Added in 7.0
250       Name.startswith("avx512.mask.packsswb.") || // Added in 5.0
251       Name.startswith("avx512.mask.packssdw.") || // Added in 5.0
252       Name.startswith("avx512.mask.packuswb.") || // Added in 5.0
253       Name.startswith("avx512.mask.packusdw.") || // Added in 5.0
254       Name.startswith("avx512.mask.cmp.b") || // Added in 5.0
255       Name.startswith("avx512.mask.cmp.d") || // Added in 5.0
256       Name.startswith("avx512.mask.cmp.q") || // Added in 5.0
257       Name.startswith("avx512.mask.cmp.w") || // Added in 5.0
258       Name.startswith("avx512.cmp.p") || // Added in 12.0
259       Name.startswith("avx512.mask.ucmp.") || // Added in 5.0
260       Name.startswith("avx512.cvtb2mask.") || // Added in 7.0
261       Name.startswith("avx512.cvtw2mask.") || // Added in 7.0
262       Name.startswith("avx512.cvtd2mask.") || // Added in 7.0
263       Name.startswith("avx512.cvtq2mask.") || // Added in 7.0
264       Name.startswith("avx512.mask.vpermilvar.") || // Added in 4.0
265       Name.startswith("avx512.mask.psll.d") || // Added in 4.0
266       Name.startswith("avx512.mask.psll.q") || // Added in 4.0
267       Name.startswith("avx512.mask.psll.w") || // Added in 4.0
268       Name.startswith("avx512.mask.psra.d") || // Added in 4.0
269       Name.startswith("avx512.mask.psra.q") || // Added in 4.0
270       Name.startswith("avx512.mask.psra.w") || // Added in 4.0
271       Name.startswith("avx512.mask.psrl.d") || // Added in 4.0
272       Name.startswith("avx512.mask.psrl.q") || // Added in 4.0
273       Name.startswith("avx512.mask.psrl.w") || // Added in 4.0
274       Name.startswith("avx512.mask.pslli") || // Added in 4.0
275       Name.startswith("avx512.mask.psrai") || // Added in 4.0
276       Name.startswith("avx512.mask.psrli") || // Added in 4.0
277       Name.startswith("avx512.mask.psllv") || // Added in 4.0
278       Name.startswith("avx512.mask.psrav") || // Added in 4.0
279       Name.startswith("avx512.mask.psrlv") || // Added in 4.0
280       Name.startswith("sse41.pmovsx") || // Added in 3.8
281       Name.startswith("sse41.pmovzx") || // Added in 3.9
282       Name.startswith("avx2.pmovsx") || // Added in 3.9
283       Name.startswith("avx2.pmovzx") || // Added in 3.9
284       Name.startswith("avx512.mask.pmovsx") || // Added in 4.0
285       Name.startswith("avx512.mask.pmovzx") || // Added in 4.0
286       Name.startswith("avx512.mask.lzcnt.") || // Added in 5.0
287       Name.startswith("avx512.mask.pternlog.") || // Added in 7.0
288       Name.startswith("avx512.maskz.pternlog.") || // Added in 7.0
289       Name.startswith("avx512.mask.vpmadd52") || // Added in 7.0
290       Name.startswith("avx512.maskz.vpmadd52") || // Added in 7.0
291       Name.startswith("avx512.mask.vpermi2var.") || // Added in 7.0
292       Name.startswith("avx512.mask.vpermt2var.") || // Added in 7.0
293       Name.startswith("avx512.maskz.vpermt2var.") || // Added in 7.0
294       Name.startswith("avx512.mask.vpdpbusd.") || // Added in 7.0
295       Name.startswith("avx512.maskz.vpdpbusd.") || // Added in 7.0
296       Name.startswith("avx512.mask.vpdpbusds.") || // Added in 7.0
297       Name.startswith("avx512.maskz.vpdpbusds.") || // Added in 7.0
298       Name.startswith("avx512.mask.vpdpwssd.") || // Added in 7.0
299       Name.startswith("avx512.maskz.vpdpwssd.") || // Added in 7.0
300       Name.startswith("avx512.mask.vpdpwssds.") || // Added in 7.0
301       Name.startswith("avx512.maskz.vpdpwssds.") || // Added in 7.0
302       Name.startswith("avx512.mask.dbpsadbw.") || // Added in 7.0
303       Name.startswith("avx512.mask.vpshld.") || // Added in 7.0
304       Name.startswith("avx512.mask.vpshrd.") || // Added in 7.0
305       Name.startswith("avx512.mask.vpshldv.") || // Added in 8.0
306       Name.startswith("avx512.mask.vpshrdv.") || // Added in 8.0
307       Name.startswith("avx512.maskz.vpshldv.") || // Added in 8.0
308       Name.startswith("avx512.maskz.vpshrdv.") || // Added in 8.0
309       Name.startswith("avx512.vpshld.") || // Added in 8.0
310       Name.startswith("avx512.vpshrd.") || // Added in 8.0
311       Name.startswith("avx512.mask.add.p") || // Added in 7.0. 128/256 in 4.0
312       Name.startswith("avx512.mask.sub.p") || // Added in 7.0. 128/256 in 4.0
313       Name.startswith("avx512.mask.mul.p") || // Added in 7.0. 128/256 in 4.0
314       Name.startswith("avx512.mask.div.p") || // Added in 7.0. 128/256 in 4.0
315       Name.startswith("avx512.mask.max.p") || // Added in 7.0. 128/256 in 5.0
316       Name.startswith("avx512.mask.min.p") || // Added in 7.0. 128/256 in 5.0
317       Name.startswith("avx512.mask.fpclass.p") || // Added in 7.0
318       Name.startswith("avx512.mask.vpshufbitqmb.") || // Added in 8.0
319       Name.startswith("avx512.mask.pmultishift.qb.") || // Added in 8.0
320       Name.startswith("avx512.mask.conflict.") || // Added in 9.0
321       Name == "avx512.mask.pmov.qd.256" || // Added in 9.0
322       Name == "avx512.mask.pmov.qd.512" || // Added in 9.0
323       Name == "avx512.mask.pmov.wb.256" || // Added in 9.0
324       Name == "avx512.mask.pmov.wb.512" || // Added in 9.0
325       Name == "sse.cvtsi2ss" || // Added in 7.0
326       Name == "sse.cvtsi642ss" || // Added in 7.0
327       Name == "sse2.cvtsi2sd" || // Added in 7.0
328       Name == "sse2.cvtsi642sd" || // Added in 7.0
329       Name == "sse2.cvtss2sd" || // Added in 7.0
330       Name == "sse2.cvtdq2pd" || // Added in 3.9
331       Name == "sse2.cvtdq2ps" || // Added in 7.0
332       Name == "sse2.cvtps2pd" || // Added in 3.9
333       Name == "avx.cvtdq2.pd.256" || // Added in 3.9
334       Name == "avx.cvtdq2.ps.256" || // Added in 7.0
335       Name == "avx.cvt.ps2.pd.256" || // Added in 3.9
336       Name.startswith("vcvtph2ps.") || // Added in 11.0
337       Name.startswith("avx.vinsertf128.") || // Added in 3.7
338       Name == "avx2.vinserti128" || // Added in 3.7
339       Name.startswith("avx512.mask.insert") || // Added in 4.0
340       Name.startswith("avx.vextractf128.") || // Added in 3.7
341       Name == "avx2.vextracti128" || // Added in 3.7
342       Name.startswith("avx512.mask.vextract") || // Added in 4.0
343       Name.startswith("sse4a.movnt.") || // Added in 3.9
344       Name.startswith("avx.movnt.") || // Added in 3.2
345       Name.startswith("avx512.storent.") || // Added in 3.9
346       Name == "sse41.movntdqa" || // Added in 5.0
347       Name == "avx2.movntdqa" || // Added in 5.0
348       Name == "avx512.movntdqa" || // Added in 5.0
349       Name == "sse2.storel.dq" || // Added in 3.9
350       Name.startswith("sse.storeu.") || // Added in 3.9
351       Name.startswith("sse2.storeu.") || // Added in 3.9
352       Name.startswith("avx.storeu.") || // Added in 3.9
353       Name.startswith("avx512.mask.storeu.") || // Added in 3.9
354       Name.startswith("avx512.mask.store.p") || // Added in 3.9
355       Name.startswith("avx512.mask.store.b.") || // Added in 3.9
356       Name.startswith("avx512.mask.store.w.") || // Added in 3.9
357       Name.startswith("avx512.mask.store.d.") || // Added in 3.9
358       Name.startswith("avx512.mask.store.q.") || // Added in 3.9
359       Name == "avx512.mask.store.ss" || // Added in 7.0
360       Name.startswith("avx512.mask.loadu.") || // Added in 3.9
361       Name.startswith("avx512.mask.load.") || // Added in 3.9
362       Name.startswith("avx512.mask.expand.load.") || // Added in 7.0
363       Name.startswith("avx512.mask.compress.store.") || // Added in 7.0
364       Name.startswith("avx512.mask.expand.b") || // Added in 9.0
365       Name.startswith("avx512.mask.expand.w") || // Added in 9.0
366       Name.startswith("avx512.mask.expand.d") || // Added in 9.0
367       Name.startswith("avx512.mask.expand.q") || // Added in 9.0
368       Name.startswith("avx512.mask.expand.p") || // Added in 9.0
369       Name.startswith("avx512.mask.compress.b") || // Added in 9.0
370       Name.startswith("avx512.mask.compress.w") || // Added in 9.0
371       Name.startswith("avx512.mask.compress.d") || // Added in 9.0
372       Name.startswith("avx512.mask.compress.q") || // Added in 9.0
373       Name.startswith("avx512.mask.compress.p") || // Added in 9.0
374       Name == "sse42.crc32.64.8" || // Added in 3.4
375       Name.startswith("avx.vbroadcast.s") || // Added in 3.5
376       Name.startswith("avx512.vbroadcast.s") || // Added in 7.0
377       Name.startswith("avx512.mask.palignr.") || // Added in 3.9
378       Name.startswith("avx512.mask.valign.") || // Added in 4.0
379       Name.startswith("sse2.psll.dq") || // Added in 3.7
380       Name.startswith("sse2.psrl.dq") || // Added in 3.7
381       Name.startswith("avx2.psll.dq") || // Added in 3.7
382       Name.startswith("avx2.psrl.dq") || // Added in 3.7
383       Name.startswith("avx512.psll.dq") || // Added in 3.9
384       Name.startswith("avx512.psrl.dq") || // Added in 3.9
385       Name == "sse41.pblendw" || // Added in 3.7
386       Name.startswith("sse41.blendp") || // Added in 3.7
387       Name.startswith("avx.blend.p") || // Added in 3.7
388       Name == "avx2.pblendw" || // Added in 3.7
389       Name.startswith("avx2.pblendd.") || // Added in 3.7
390       Name.startswith("avx.vbroadcastf128") || // Added in 4.0
391       Name == "avx2.vbroadcasti128" || // Added in 3.7
392       Name.startswith("avx512.mask.broadcastf32x4.") || // Added in 6.0
393       Name.startswith("avx512.mask.broadcastf64x2.") || // Added in 6.0
394       Name.startswith("avx512.mask.broadcastf32x8.") || // Added in 6.0
395       Name.startswith("avx512.mask.broadcastf64x4.") || // Added in 6.0
396       Name.startswith("avx512.mask.broadcasti32x4.") || // Added in 6.0
397       Name.startswith("avx512.mask.broadcasti64x2.") || // Added in 6.0
398       Name.startswith("avx512.mask.broadcasti32x8.") || // Added in 6.0
399       Name.startswith("avx512.mask.broadcasti64x4.") || // Added in 6.0
400       Name == "xop.vpcmov" || // Added in 3.8
401       Name == "xop.vpcmov.256" || // Added in 5.0
402       Name.startswith("avx512.mask.move.s") || // Added in 4.0
403       Name.startswith("avx512.cvtmask2") || // Added in 5.0
404       Name.startswith("xop.vpcom") || // Added in 3.2, Updated in 9.0
405       Name.startswith("xop.vprot") || // Added in 8.0
406       Name.startswith("avx512.prol") || // Added in 8.0
407       Name.startswith("avx512.pror") || // Added in 8.0
408       Name.startswith("avx512.mask.prorv.") || // Added in 8.0
409       Name.startswith("avx512.mask.pror.") ||  // Added in 8.0
410       Name.startswith("avx512.mask.prolv.") || // Added in 8.0
411       Name.startswith("avx512.mask.prol.") ||  // Added in 8.0
412       Name.startswith("avx512.ptestm") || //Added in 6.0
413       Name.startswith("avx512.ptestnm") || //Added in 6.0
414       Name.startswith("avx512.mask.pavg")) // Added in 6.0
415     return true;
416 
417   return false;
418 }
419 
UpgradeX86IntrinsicFunction(Function * F,StringRef Name,Function * & NewFn)420 static bool UpgradeX86IntrinsicFunction(Function *F, StringRef Name,
421                                         Function *&NewFn) {
422   // Only handle intrinsics that start with "x86.".
423   if (!Name.startswith("x86."))
424     return false;
425   // Remove "x86." prefix.
426   Name = Name.substr(4);
427 
428   if (ShouldUpgradeX86Intrinsic(F, Name)) {
429     NewFn = nullptr;
430     return true;
431   }
432 
433   if (Name == "rdtscp") { // Added in 8.0
434     // If this intrinsic has 0 operands, it's the new version.
435     if (F->getFunctionType()->getNumParams() == 0)
436       return false;
437 
438     rename(F);
439     NewFn = Intrinsic::getDeclaration(F->getParent(),
440                                       Intrinsic::x86_rdtscp);
441     return true;
442   }
443 
444   // SSE4.1 ptest functions may have an old signature.
445   if (Name.startswith("sse41.ptest")) { // Added in 3.2
446     if (Name.substr(11) == "c")
447       return UpgradePTESTIntrinsic(F, Intrinsic::x86_sse41_ptestc, NewFn);
448     if (Name.substr(11) == "z")
449       return UpgradePTESTIntrinsic(F, Intrinsic::x86_sse41_ptestz, NewFn);
450     if (Name.substr(11) == "nzc")
451       return UpgradePTESTIntrinsic(F, Intrinsic::x86_sse41_ptestnzc, NewFn);
452   }
453   // Several blend and other instructions with masks used the wrong number of
454   // bits.
455   if (Name == "sse41.insertps") // Added in 3.6
456     return UpgradeX86IntrinsicsWith8BitMask(F, Intrinsic::x86_sse41_insertps,
457                                             NewFn);
458   if (Name == "sse41.dppd") // Added in 3.6
459     return UpgradeX86IntrinsicsWith8BitMask(F, Intrinsic::x86_sse41_dppd,
460                                             NewFn);
461   if (Name == "sse41.dpps") // Added in 3.6
462     return UpgradeX86IntrinsicsWith8BitMask(F, Intrinsic::x86_sse41_dpps,
463                                             NewFn);
464   if (Name == "sse41.mpsadbw") // Added in 3.6
465     return UpgradeX86IntrinsicsWith8BitMask(F, Intrinsic::x86_sse41_mpsadbw,
466                                             NewFn);
467   if (Name == "avx.dp.ps.256") // Added in 3.6
468     return UpgradeX86IntrinsicsWith8BitMask(F, Intrinsic::x86_avx_dp_ps_256,
469                                             NewFn);
470   if (Name == "avx2.mpsadbw") // Added in 3.6
471     return UpgradeX86IntrinsicsWith8BitMask(F, Intrinsic::x86_avx2_mpsadbw,
472                                             NewFn);
473   if (Name == "avx512.mask.cmp.pd.128") // Added in 7.0
474     return UpgradeX86MaskedFPCompare(F, Intrinsic::x86_avx512_mask_cmp_pd_128,
475                                      NewFn);
476   if (Name == "avx512.mask.cmp.pd.256") // Added in 7.0
477     return UpgradeX86MaskedFPCompare(F, Intrinsic::x86_avx512_mask_cmp_pd_256,
478                                      NewFn);
479   if (Name == "avx512.mask.cmp.pd.512") // Added in 7.0
480     return UpgradeX86MaskedFPCompare(F, Intrinsic::x86_avx512_mask_cmp_pd_512,
481                                      NewFn);
482   if (Name == "avx512.mask.cmp.ps.128") // Added in 7.0
483     return UpgradeX86MaskedFPCompare(F, Intrinsic::x86_avx512_mask_cmp_ps_128,
484                                      NewFn);
485   if (Name == "avx512.mask.cmp.ps.256") // Added in 7.0
486     return UpgradeX86MaskedFPCompare(F, Intrinsic::x86_avx512_mask_cmp_ps_256,
487                                      NewFn);
488   if (Name == "avx512.mask.cmp.ps.512") // Added in 7.0
489     return UpgradeX86MaskedFPCompare(F, Intrinsic::x86_avx512_mask_cmp_ps_512,
490                                      NewFn);
491 
492   // frcz.ss/sd may need to have an argument dropped. Added in 3.2
493   if (Name.startswith("xop.vfrcz.ss") && F->arg_size() == 2) {
494     rename(F);
495     NewFn = Intrinsic::getDeclaration(F->getParent(),
496                                       Intrinsic::x86_xop_vfrcz_ss);
497     return true;
498   }
499   if (Name.startswith("xop.vfrcz.sd") && F->arg_size() == 2) {
500     rename(F);
501     NewFn = Intrinsic::getDeclaration(F->getParent(),
502                                       Intrinsic::x86_xop_vfrcz_sd);
503     return true;
504   }
505   // Upgrade any XOP PERMIL2 index operand still using a float/double vector.
506   if (Name.startswith("xop.vpermil2")) { // Added in 3.9
507     auto Idx = F->getFunctionType()->getParamType(2);
508     if (Idx->isFPOrFPVectorTy()) {
509       rename(F);
510       unsigned IdxSize = Idx->getPrimitiveSizeInBits();
511       unsigned EltSize = Idx->getScalarSizeInBits();
512       Intrinsic::ID Permil2ID;
513       if (EltSize == 64 && IdxSize == 128)
514         Permil2ID = Intrinsic::x86_xop_vpermil2pd;
515       else if (EltSize == 32 && IdxSize == 128)
516         Permil2ID = Intrinsic::x86_xop_vpermil2ps;
517       else if (EltSize == 64 && IdxSize == 256)
518         Permil2ID = Intrinsic::x86_xop_vpermil2pd_256;
519       else
520         Permil2ID = Intrinsic::x86_xop_vpermil2ps_256;
521       NewFn = Intrinsic::getDeclaration(F->getParent(), Permil2ID);
522       return true;
523     }
524   }
525 
526   if (Name == "seh.recoverfp") {
527     NewFn = Intrinsic::getDeclaration(F->getParent(), Intrinsic::eh_recoverfp);
528     return true;
529   }
530 
531   return false;
532 }
533 
UpgradeIntrinsicFunction1(Function * F,Function * & NewFn)534 static bool UpgradeIntrinsicFunction1(Function *F, Function *&NewFn) {
535   assert(F && "Illegal to upgrade a non-existent Function.");
536 
537   // Quickly eliminate it, if it's not a candidate.
538   StringRef Name = F->getName();
539   if (Name.size() <= 8 || !Name.startswith("llvm."))
540     return false;
541   Name = Name.substr(5); // Strip off "llvm."
542 
543   switch (Name[0]) {
544   default: break;
545   case 'a': {
546     if (Name.startswith("arm.rbit") || Name.startswith("aarch64.rbit")) {
547       NewFn = Intrinsic::getDeclaration(F->getParent(), Intrinsic::bitreverse,
548                                         F->arg_begin()->getType());
549       return true;
550     }
551     if (Name.startswith("aarch64.neon.frintn")) {
552       NewFn = Intrinsic::getDeclaration(F->getParent(), Intrinsic::roundeven,
553                                         F->arg_begin()->getType());
554       return true;
555     }
556     if (Name.startswith("aarch64.neon.rbit")) {
557       NewFn = Intrinsic::getDeclaration(F->getParent(), Intrinsic::bitreverse,
558                                         F->arg_begin()->getType());
559       return true;
560     }
561     if (Name.startswith("arm.neon.vclz")) {
562       Type* args[2] = {
563         F->arg_begin()->getType(),
564         Type::getInt1Ty(F->getContext())
565       };
566       // Can't use Intrinsic::getDeclaration here as it adds a ".i1" to
567       // the end of the name. Change name from llvm.arm.neon.vclz.* to
568       //  llvm.ctlz.*
569       FunctionType* fType = FunctionType::get(F->getReturnType(), args, false);
570       NewFn = Function::Create(fType, F->getLinkage(), F->getAddressSpace(),
571                                "llvm.ctlz." + Name.substr(14), F->getParent());
572       return true;
573     }
574     if (Name.startswith("arm.neon.vcnt")) {
575       NewFn = Intrinsic::getDeclaration(F->getParent(), Intrinsic::ctpop,
576                                         F->arg_begin()->getType());
577       return true;
578     }
579     static const Regex vldRegex("^arm\\.neon\\.vld([1234]|[234]lane)\\.v[a-z0-9]*$");
580     if (vldRegex.match(Name)) {
581       auto fArgs = F->getFunctionType()->params();
582       SmallVector<Type *, 4> Tys(fArgs.begin(), fArgs.end());
583       // Can't use Intrinsic::getDeclaration here as the return types might
584       // then only be structurally equal.
585       FunctionType* fType = FunctionType::get(F->getReturnType(), Tys, false);
586       NewFn = Function::Create(fType, F->getLinkage(), F->getAddressSpace(),
587                                "llvm." + Name + ".p0i8", F->getParent());
588       return true;
589     }
590     static const Regex vstRegex("^arm\\.neon\\.vst([1234]|[234]lane)\\.v[a-z0-9]*$");
591     if (vstRegex.match(Name)) {
592       static const Intrinsic::ID StoreInts[] = {Intrinsic::arm_neon_vst1,
593                                                 Intrinsic::arm_neon_vst2,
594                                                 Intrinsic::arm_neon_vst3,
595                                                 Intrinsic::arm_neon_vst4};
596 
597       static const Intrinsic::ID StoreLaneInts[] = {
598         Intrinsic::arm_neon_vst2lane, Intrinsic::arm_neon_vst3lane,
599         Intrinsic::arm_neon_vst4lane
600       };
601 
602       auto fArgs = F->getFunctionType()->params();
603       Type *Tys[] = {fArgs[0], fArgs[1]};
604       if (Name.find("lane") == StringRef::npos)
605         NewFn = Intrinsic::getDeclaration(F->getParent(),
606                                           StoreInts[fArgs.size() - 3], Tys);
607       else
608         NewFn = Intrinsic::getDeclaration(F->getParent(),
609                                           StoreLaneInts[fArgs.size() - 5], Tys);
610       return true;
611     }
612     if (Name == "aarch64.thread.pointer" || Name == "arm.thread.pointer") {
613       NewFn = Intrinsic::getDeclaration(F->getParent(), Intrinsic::thread_pointer);
614       return true;
615     }
616     if (Name.startswith("arm.neon.vqadds.")) {
617       NewFn = Intrinsic::getDeclaration(F->getParent(), Intrinsic::sadd_sat,
618                                         F->arg_begin()->getType());
619       return true;
620     }
621     if (Name.startswith("arm.neon.vqaddu.")) {
622       NewFn = Intrinsic::getDeclaration(F->getParent(), Intrinsic::uadd_sat,
623                                         F->arg_begin()->getType());
624       return true;
625     }
626     if (Name.startswith("arm.neon.vqsubs.")) {
627       NewFn = Intrinsic::getDeclaration(F->getParent(), Intrinsic::ssub_sat,
628                                         F->arg_begin()->getType());
629       return true;
630     }
631     if (Name.startswith("arm.neon.vqsubu.")) {
632       NewFn = Intrinsic::getDeclaration(F->getParent(), Intrinsic::usub_sat,
633                                         F->arg_begin()->getType());
634       return true;
635     }
636     if (Name.startswith("aarch64.neon.addp")) {
637       if (F->arg_size() != 2)
638         break; // Invalid IR.
639       VectorType *Ty = dyn_cast<VectorType>(F->getReturnType());
640       if (Ty && Ty->getElementType()->isFloatingPointTy()) {
641         NewFn = Intrinsic::getDeclaration(F->getParent(),
642                                           Intrinsic::aarch64_neon_faddp, Ty);
643         return true;
644       }
645     }
646 
647     // Changed in 12.0: bfdot accept v4bf16 and v8bf16 instead of v8i8 and v16i8
648     // respectively
649     if ((Name.startswith("arm.neon.bfdot.") ||
650          Name.startswith("aarch64.neon.bfdot.")) &&
651         Name.endswith("i8")) {
652       Intrinsic::ID IID =
653           StringSwitch<Intrinsic::ID>(Name)
654               .Cases("arm.neon.bfdot.v2f32.v8i8",
655                      "arm.neon.bfdot.v4f32.v16i8",
656                      Intrinsic::arm_neon_bfdot)
657               .Cases("aarch64.neon.bfdot.v2f32.v8i8",
658                      "aarch64.neon.bfdot.v4f32.v16i8",
659                      Intrinsic::aarch64_neon_bfdot)
660               .Default(Intrinsic::not_intrinsic);
661       if (IID == Intrinsic::not_intrinsic)
662         break;
663 
664       size_t OperandWidth = F->getReturnType()->getPrimitiveSizeInBits();
665       assert((OperandWidth == 64 || OperandWidth == 128) &&
666              "Unexpected operand width");
667       LLVMContext &Ctx = F->getParent()->getContext();
668       std::array<Type *, 2> Tys {{
669         F->getReturnType(),
670         FixedVectorType::get(Type::getBFloatTy(Ctx), OperandWidth / 16)
671       }};
672       NewFn = Intrinsic::getDeclaration(F->getParent(), IID, Tys);
673       return true;
674     }
675 
676     // Changed in 12.0: bfmmla, bfmlalb and bfmlalt are not polymorphic anymore
677     // and accept v8bf16 instead of v16i8
678     if ((Name.startswith("arm.neon.bfm") ||
679          Name.startswith("aarch64.neon.bfm")) &&
680         Name.endswith(".v4f32.v16i8")) {
681       Intrinsic::ID IID =
682           StringSwitch<Intrinsic::ID>(Name)
683               .Case("arm.neon.bfmmla.v4f32.v16i8",
684                     Intrinsic::arm_neon_bfmmla)
685               .Case("arm.neon.bfmlalb.v4f32.v16i8",
686                     Intrinsic::arm_neon_bfmlalb)
687               .Case("arm.neon.bfmlalt.v4f32.v16i8",
688                     Intrinsic::arm_neon_bfmlalt)
689               .Case("aarch64.neon.bfmmla.v4f32.v16i8",
690                     Intrinsic::aarch64_neon_bfmmla)
691               .Case("aarch64.neon.bfmlalb.v4f32.v16i8",
692                     Intrinsic::aarch64_neon_bfmlalb)
693               .Case("aarch64.neon.bfmlalt.v4f32.v16i8",
694                     Intrinsic::aarch64_neon_bfmlalt)
695               .Default(Intrinsic::not_intrinsic);
696       if (IID == Intrinsic::not_intrinsic)
697         break;
698 
699       std::array<Type *, 0> Tys;
700       NewFn = Intrinsic::getDeclaration(F->getParent(), IID, Tys);
701       return true;
702     }
703     break;
704   }
705 
706   case 'c': {
707     if (Name.startswith("ctlz.") && F->arg_size() == 1) {
708       rename(F);
709       NewFn = Intrinsic::getDeclaration(F->getParent(), Intrinsic::ctlz,
710                                         F->arg_begin()->getType());
711       return true;
712     }
713     if (Name.startswith("cttz.") && F->arg_size() == 1) {
714       rename(F);
715       NewFn = Intrinsic::getDeclaration(F->getParent(), Intrinsic::cttz,
716                                         F->arg_begin()->getType());
717       return true;
718     }
719     break;
720   }
721   case 'd': {
722     if (Name == "dbg.value" && F->arg_size() == 4) {
723       rename(F);
724       NewFn = Intrinsic::getDeclaration(F->getParent(), Intrinsic::dbg_value);
725       return true;
726     }
727     break;
728   }
729   case 'e': {
730     SmallVector<StringRef, 2> Groups;
731     static const Regex R("^experimental.vector.reduce.([a-z]+)\\.[a-z][0-9]+");
732     if (R.match(Name, &Groups)) {
733       Intrinsic::ID ID;
734       ID = StringSwitch<Intrinsic::ID>(Groups[1])
735                .Case("add", Intrinsic::vector_reduce_add)
736                .Case("mul", Intrinsic::vector_reduce_mul)
737                .Case("and", Intrinsic::vector_reduce_and)
738                .Case("or", Intrinsic::vector_reduce_or)
739                .Case("xor", Intrinsic::vector_reduce_xor)
740                .Case("smax", Intrinsic::vector_reduce_smax)
741                .Case("smin", Intrinsic::vector_reduce_smin)
742                .Case("umax", Intrinsic::vector_reduce_umax)
743                .Case("umin", Intrinsic::vector_reduce_umin)
744                .Case("fmax", Intrinsic::vector_reduce_fmax)
745                .Case("fmin", Intrinsic::vector_reduce_fmin)
746                .Default(Intrinsic::not_intrinsic);
747       if (ID != Intrinsic::not_intrinsic) {
748         rename(F);
749         auto Args = F->getFunctionType()->params();
750         NewFn = Intrinsic::getDeclaration(F->getParent(), ID, {Args[0]});
751         return true;
752       }
753     }
754     static const Regex R2(
755         "^experimental.vector.reduce.v2.([a-z]+)\\.[fi][0-9]+");
756     Groups.clear();
757     if (R2.match(Name, &Groups)) {
758       Intrinsic::ID ID = Intrinsic::not_intrinsic;
759       if (Groups[1] == "fadd")
760         ID = Intrinsic::vector_reduce_fadd;
761       if (Groups[1] == "fmul")
762         ID = Intrinsic::vector_reduce_fmul;
763       if (ID != Intrinsic::not_intrinsic) {
764         rename(F);
765         auto Args = F->getFunctionType()->params();
766         Type *Tys[] = {Args[1]};
767         NewFn = Intrinsic::getDeclaration(F->getParent(), ID, Tys);
768         return true;
769       }
770     }
771     break;
772   }
773   case 'i':
774   case 'l': {
775     bool IsLifetimeStart = Name.startswith("lifetime.start");
776     if (IsLifetimeStart || Name.startswith("invariant.start")) {
777       Intrinsic::ID ID = IsLifetimeStart ?
778         Intrinsic::lifetime_start : Intrinsic::invariant_start;
779       auto Args = F->getFunctionType()->params();
780       Type* ObjectPtr[1] = {Args[1]};
781       if (F->getName() != Intrinsic::getName(ID, ObjectPtr)) {
782         rename(F);
783         NewFn = Intrinsic::getDeclaration(F->getParent(), ID, ObjectPtr);
784         return true;
785       }
786     }
787 
788     bool IsLifetimeEnd = Name.startswith("lifetime.end");
789     if (IsLifetimeEnd || Name.startswith("invariant.end")) {
790       Intrinsic::ID ID = IsLifetimeEnd ?
791         Intrinsic::lifetime_end : Intrinsic::invariant_end;
792 
793       auto Args = F->getFunctionType()->params();
794       Type* ObjectPtr[1] = {Args[IsLifetimeEnd ? 1 : 2]};
795       if (F->getName() != Intrinsic::getName(ID, ObjectPtr)) {
796         rename(F);
797         NewFn = Intrinsic::getDeclaration(F->getParent(), ID, ObjectPtr);
798         return true;
799       }
800     }
801     if (Name.startswith("invariant.group.barrier")) {
802       // Rename invariant.group.barrier to launder.invariant.group
803       auto Args = F->getFunctionType()->params();
804       Type* ObjectPtr[1] = {Args[0]};
805       rename(F);
806       NewFn = Intrinsic::getDeclaration(F->getParent(),
807           Intrinsic::launder_invariant_group, ObjectPtr);
808       return true;
809 
810     }
811 
812     break;
813   }
814   case 'm': {
815     if (Name.startswith("masked.load.")) {
816       Type *Tys[] = { F->getReturnType(), F->arg_begin()->getType() };
817       if (F->getName() != Intrinsic::getName(Intrinsic::masked_load, Tys)) {
818         rename(F);
819         NewFn = Intrinsic::getDeclaration(F->getParent(),
820                                           Intrinsic::masked_load,
821                                           Tys);
822         return true;
823       }
824     }
825     if (Name.startswith("masked.store.")) {
826       auto Args = F->getFunctionType()->params();
827       Type *Tys[] = { Args[0], Args[1] };
828       if (F->getName() != Intrinsic::getName(Intrinsic::masked_store, Tys)) {
829         rename(F);
830         NewFn = Intrinsic::getDeclaration(F->getParent(),
831                                           Intrinsic::masked_store,
832                                           Tys);
833         return true;
834       }
835     }
836     // Renaming gather/scatter intrinsics with no address space overloading
837     // to the new overload which includes an address space
838     if (Name.startswith("masked.gather.")) {
839       Type *Tys[] = {F->getReturnType(), F->arg_begin()->getType()};
840       if (F->getName() != Intrinsic::getName(Intrinsic::masked_gather, Tys)) {
841         rename(F);
842         NewFn = Intrinsic::getDeclaration(F->getParent(),
843                                           Intrinsic::masked_gather, Tys);
844         return true;
845       }
846     }
847     if (Name.startswith("masked.scatter.")) {
848       auto Args = F->getFunctionType()->params();
849       Type *Tys[] = {Args[0], Args[1]};
850       if (F->getName() != Intrinsic::getName(Intrinsic::masked_scatter, Tys)) {
851         rename(F);
852         NewFn = Intrinsic::getDeclaration(F->getParent(),
853                                           Intrinsic::masked_scatter, Tys);
854         return true;
855       }
856     }
857     // Updating the memory intrinsics (memcpy/memmove/memset) that have an
858     // alignment parameter to embedding the alignment as an attribute of
859     // the pointer args.
860     if (Name.startswith("memcpy.") && F->arg_size() == 5) {
861       rename(F);
862       // Get the types of dest, src, and len
863       ArrayRef<Type *> ParamTypes = F->getFunctionType()->params().slice(0, 3);
864       NewFn = Intrinsic::getDeclaration(F->getParent(), Intrinsic::memcpy,
865                                         ParamTypes);
866       return true;
867     }
868     if (Name.startswith("memmove.") && F->arg_size() == 5) {
869       rename(F);
870       // Get the types of dest, src, and len
871       ArrayRef<Type *> ParamTypes = F->getFunctionType()->params().slice(0, 3);
872       NewFn = Intrinsic::getDeclaration(F->getParent(), Intrinsic::memmove,
873                                         ParamTypes);
874       return true;
875     }
876     if (Name.startswith("memset.") && F->arg_size() == 5) {
877       rename(F);
878       // Get the types of dest, and len
879       const auto *FT = F->getFunctionType();
880       Type *ParamTypes[2] = {
881           FT->getParamType(0), // Dest
882           FT->getParamType(2)  // len
883       };
884       NewFn = Intrinsic::getDeclaration(F->getParent(), Intrinsic::memset,
885                                         ParamTypes);
886       return true;
887     }
888     break;
889   }
890   case 'n': {
891     if (Name.startswith("nvvm.")) {
892       Name = Name.substr(5);
893 
894       // The following nvvm intrinsics correspond exactly to an LLVM intrinsic.
895       Intrinsic::ID IID = StringSwitch<Intrinsic::ID>(Name)
896                               .Cases("brev32", "brev64", Intrinsic::bitreverse)
897                               .Case("clz.i", Intrinsic::ctlz)
898                               .Case("popc.i", Intrinsic::ctpop)
899                               .Default(Intrinsic::not_intrinsic);
900       if (IID != Intrinsic::not_intrinsic && F->arg_size() == 1) {
901         NewFn = Intrinsic::getDeclaration(F->getParent(), IID,
902                                           {F->getReturnType()});
903         return true;
904       }
905 
906       // The following nvvm intrinsics correspond exactly to an LLVM idiom, but
907       // not to an intrinsic alone.  We expand them in UpgradeIntrinsicCall.
908       //
909       // TODO: We could add lohi.i2d.
910       bool Expand = StringSwitch<bool>(Name)
911                         .Cases("abs.i", "abs.ll", true)
912                         .Cases("clz.ll", "popc.ll", "h2f", true)
913                         .Cases("max.i", "max.ll", "max.ui", "max.ull", true)
914                         .Cases("min.i", "min.ll", "min.ui", "min.ull", true)
915                         .StartsWith("atomic.load.add.f32.p", true)
916                         .StartsWith("atomic.load.add.f64.p", true)
917                         .Default(false);
918       if (Expand) {
919         NewFn = nullptr;
920         return true;
921       }
922     }
923     break;
924   }
925   case 'o':
926     // We only need to change the name to match the mangling including the
927     // address space.
928     if (Name.startswith("objectsize.")) {
929       Type *Tys[2] = { F->getReturnType(), F->arg_begin()->getType() };
930       if (F->arg_size() == 2 || F->arg_size() == 3 ||
931           F->getName() != Intrinsic::getName(Intrinsic::objectsize, Tys)) {
932         rename(F);
933         NewFn = Intrinsic::getDeclaration(F->getParent(), Intrinsic::objectsize,
934                                           Tys);
935         return true;
936       }
937     }
938     break;
939 
940   case 'p':
941     if (Name == "prefetch") {
942       // Handle address space overloading.
943       Type *Tys[] = {F->arg_begin()->getType()};
944       if (F->getName() != Intrinsic::getName(Intrinsic::prefetch, Tys)) {
945         rename(F);
946         NewFn =
947             Intrinsic::getDeclaration(F->getParent(), Intrinsic::prefetch, Tys);
948         return true;
949       }
950     } else if (Name.startswith("ptr.annotation.") && F->arg_size() == 4) {
951       rename(F);
952       NewFn = Intrinsic::getDeclaration(F->getParent(),
953                                         Intrinsic::ptr_annotation,
954                                         F->arg_begin()->getType());
955       return true;
956     }
957     break;
958 
959   case 's':
960     if (Name == "stackprotectorcheck") {
961       NewFn = nullptr;
962       return true;
963     }
964     break;
965 
966   case 'v': {
967     if (Name == "var.annotation" && F->arg_size() == 4) {
968       rename(F);
969       NewFn = Intrinsic::getDeclaration(F->getParent(),
970                                         Intrinsic::var_annotation);
971       return true;
972     }
973     break;
974   }
975 
976   case 'x':
977     if (UpgradeX86IntrinsicFunction(F, Name, NewFn))
978       return true;
979   }
980   // Remangle our intrinsic since we upgrade the mangling
981   auto Result = llvm::Intrinsic::remangleIntrinsicFunction(F);
982   if (Result != None) {
983     NewFn = Result.getValue();
984     return true;
985   }
986 
987   //  This may not belong here. This function is effectively being overloaded
988   //  to both detect an intrinsic which needs upgrading, and to provide the
989   //  upgraded form of the intrinsic. We should perhaps have two separate
990   //  functions for this.
991   return false;
992 }
993 
UpgradeIntrinsicFunction(Function * F,Function * & NewFn)994 bool llvm::UpgradeIntrinsicFunction(Function *F, Function *&NewFn) {
995   NewFn = nullptr;
996   bool Upgraded = UpgradeIntrinsicFunction1(F, NewFn);
997   assert(F != NewFn && "Intrinsic function upgraded to the same function");
998 
999   // Upgrade intrinsic attributes.  This does not change the function.
1000   if (NewFn)
1001     F = NewFn;
1002   if (Intrinsic::ID id = F->getIntrinsicID())
1003     F->setAttributes(Intrinsic::getAttributes(F->getContext(), id));
1004   return Upgraded;
1005 }
1006 
UpgradeGlobalVariable(GlobalVariable * GV)1007 GlobalVariable *llvm::UpgradeGlobalVariable(GlobalVariable *GV) {
1008   if (!(GV->hasName() && (GV->getName() == "llvm.global_ctors" ||
1009                           GV->getName() == "llvm.global_dtors")) ||
1010       !GV->hasInitializer())
1011     return nullptr;
1012   ArrayType *ATy = dyn_cast<ArrayType>(GV->getValueType());
1013   if (!ATy)
1014     return nullptr;
1015   StructType *STy = dyn_cast<StructType>(ATy->getElementType());
1016   if (!STy || STy->getNumElements() != 2)
1017     return nullptr;
1018 
1019   LLVMContext &C = GV->getContext();
1020   IRBuilder<> IRB(C);
1021   auto EltTy = StructType::get(STy->getElementType(0), STy->getElementType(1),
1022                                IRB.getInt8PtrTy());
1023   Constant *Init = GV->getInitializer();
1024   unsigned N = Init->getNumOperands();
1025   std::vector<Constant *> NewCtors(N);
1026   for (unsigned i = 0; i != N; ++i) {
1027     auto Ctor = cast<Constant>(Init->getOperand(i));
1028     NewCtors[i] = ConstantStruct::get(
1029         EltTy, Ctor->getAggregateElement(0u), Ctor->getAggregateElement(1),
1030         Constant::getNullValue(IRB.getInt8PtrTy()));
1031   }
1032   Constant *NewInit = ConstantArray::get(ArrayType::get(EltTy, N), NewCtors);
1033 
1034   return new GlobalVariable(NewInit->getType(), false, GV->getLinkage(),
1035                             NewInit, GV->getName());
1036 }
1037 
1038 // Handles upgrading SSE2/AVX2/AVX512BW PSLLDQ intrinsics by converting them
1039 // to byte shuffles.
UpgradeX86PSLLDQIntrinsics(IRBuilder<> & Builder,Value * Op,unsigned Shift)1040 static Value *UpgradeX86PSLLDQIntrinsics(IRBuilder<> &Builder,
1041                                          Value *Op, unsigned Shift) {
1042   auto *ResultTy = cast<FixedVectorType>(Op->getType());
1043   unsigned NumElts = ResultTy->getNumElements() * 8;
1044 
1045   // Bitcast from a 64-bit element type to a byte element type.
1046   Type *VecTy = FixedVectorType::get(Builder.getInt8Ty(), NumElts);
1047   Op = Builder.CreateBitCast(Op, VecTy, "cast");
1048 
1049   // We'll be shuffling in zeroes.
1050   Value *Res = Constant::getNullValue(VecTy);
1051 
1052   // If shift is less than 16, emit a shuffle to move the bytes. Otherwise,
1053   // we'll just return the zero vector.
1054   if (Shift < 16) {
1055     int Idxs[64];
1056     // 256/512-bit version is split into 2/4 16-byte lanes.
1057     for (unsigned l = 0; l != NumElts; l += 16)
1058       for (unsigned i = 0; i != 16; ++i) {
1059         unsigned Idx = NumElts + i - Shift;
1060         if (Idx < NumElts)
1061           Idx -= NumElts - 16; // end of lane, switch operand.
1062         Idxs[l + i] = Idx + l;
1063       }
1064 
1065     Res = Builder.CreateShuffleVector(Res, Op, makeArrayRef(Idxs, NumElts));
1066   }
1067 
1068   // Bitcast back to a 64-bit element type.
1069   return Builder.CreateBitCast(Res, ResultTy, "cast");
1070 }
1071 
1072 // Handles upgrading SSE2/AVX2/AVX512BW PSRLDQ intrinsics by converting them
1073 // to byte shuffles.
UpgradeX86PSRLDQIntrinsics(IRBuilder<> & Builder,Value * Op,unsigned Shift)1074 static Value *UpgradeX86PSRLDQIntrinsics(IRBuilder<> &Builder, Value *Op,
1075                                          unsigned Shift) {
1076   auto *ResultTy = cast<FixedVectorType>(Op->getType());
1077   unsigned NumElts = ResultTy->getNumElements() * 8;
1078 
1079   // Bitcast from a 64-bit element type to a byte element type.
1080   Type *VecTy = FixedVectorType::get(Builder.getInt8Ty(), NumElts);
1081   Op = Builder.CreateBitCast(Op, VecTy, "cast");
1082 
1083   // We'll be shuffling in zeroes.
1084   Value *Res = Constant::getNullValue(VecTy);
1085 
1086   // If shift is less than 16, emit a shuffle to move the bytes. Otherwise,
1087   // we'll just return the zero vector.
1088   if (Shift < 16) {
1089     int Idxs[64];
1090     // 256/512-bit version is split into 2/4 16-byte lanes.
1091     for (unsigned l = 0; l != NumElts; l += 16)
1092       for (unsigned i = 0; i != 16; ++i) {
1093         unsigned Idx = i + Shift;
1094         if (Idx >= 16)
1095           Idx += NumElts - 16; // end of lane, switch operand.
1096         Idxs[l + i] = Idx + l;
1097       }
1098 
1099     Res = Builder.CreateShuffleVector(Op, Res, makeArrayRef(Idxs, NumElts));
1100   }
1101 
1102   // Bitcast back to a 64-bit element type.
1103   return Builder.CreateBitCast(Res, ResultTy, "cast");
1104 }
1105 
getX86MaskVec(IRBuilder<> & Builder,Value * Mask,unsigned NumElts)1106 static Value *getX86MaskVec(IRBuilder<> &Builder, Value *Mask,
1107                             unsigned NumElts) {
1108   assert(isPowerOf2_32(NumElts) && "Expected power-of-2 mask elements");
1109   llvm::VectorType *MaskTy = FixedVectorType::get(
1110       Builder.getInt1Ty(), cast<IntegerType>(Mask->getType())->getBitWidth());
1111   Mask = Builder.CreateBitCast(Mask, MaskTy);
1112 
1113   // If we have less than 8 elements (1, 2 or 4), then the starting mask was an
1114   // i8 and we need to extract down to the right number of elements.
1115   if (NumElts <= 4) {
1116     int Indices[4];
1117     for (unsigned i = 0; i != NumElts; ++i)
1118       Indices[i] = i;
1119     Mask = Builder.CreateShuffleVector(
1120         Mask, Mask, makeArrayRef(Indices, NumElts), "extract");
1121   }
1122 
1123   return Mask;
1124 }
1125 
EmitX86Select(IRBuilder<> & Builder,Value * Mask,Value * Op0,Value * Op1)1126 static Value *EmitX86Select(IRBuilder<> &Builder, Value *Mask,
1127                             Value *Op0, Value *Op1) {
1128   // If the mask is all ones just emit the first operation.
1129   if (const auto *C = dyn_cast<Constant>(Mask))
1130     if (C->isAllOnesValue())
1131       return Op0;
1132 
1133   Mask = getX86MaskVec(Builder, Mask,
1134                        cast<FixedVectorType>(Op0->getType())->getNumElements());
1135   return Builder.CreateSelect(Mask, Op0, Op1);
1136 }
1137 
EmitX86ScalarSelect(IRBuilder<> & Builder,Value * Mask,Value * Op0,Value * Op1)1138 static Value *EmitX86ScalarSelect(IRBuilder<> &Builder, Value *Mask,
1139                                   Value *Op0, Value *Op1) {
1140   // If the mask is all ones just emit the first operation.
1141   if (const auto *C = dyn_cast<Constant>(Mask))
1142     if (C->isAllOnesValue())
1143       return Op0;
1144 
1145   auto *MaskTy = FixedVectorType::get(Builder.getInt1Ty(),
1146                                       Mask->getType()->getIntegerBitWidth());
1147   Mask = Builder.CreateBitCast(Mask, MaskTy);
1148   Mask = Builder.CreateExtractElement(Mask, (uint64_t)0);
1149   return Builder.CreateSelect(Mask, Op0, Op1);
1150 }
1151 
1152 // Handle autoupgrade for masked PALIGNR and VALIGND/Q intrinsics.
1153 // PALIGNR handles large immediates by shifting while VALIGN masks the immediate
1154 // so we need to handle both cases. VALIGN also doesn't have 128-bit lanes.
UpgradeX86ALIGNIntrinsics(IRBuilder<> & Builder,Value * Op0,Value * Op1,Value * Shift,Value * Passthru,Value * Mask,bool IsVALIGN)1155 static Value *UpgradeX86ALIGNIntrinsics(IRBuilder<> &Builder, Value *Op0,
1156                                         Value *Op1, Value *Shift,
1157                                         Value *Passthru, Value *Mask,
1158                                         bool IsVALIGN) {
1159   unsigned ShiftVal = cast<llvm::ConstantInt>(Shift)->getZExtValue();
1160 
1161   unsigned NumElts = cast<FixedVectorType>(Op0->getType())->getNumElements();
1162   assert((IsVALIGN || NumElts % 16 == 0) && "Illegal NumElts for PALIGNR!");
1163   assert((!IsVALIGN || NumElts <= 16) && "NumElts too large for VALIGN!");
1164   assert(isPowerOf2_32(NumElts) && "NumElts not a power of 2!");
1165 
1166   // Mask the immediate for VALIGN.
1167   if (IsVALIGN)
1168     ShiftVal &= (NumElts - 1);
1169 
1170   // If palignr is shifting the pair of vectors more than the size of two
1171   // lanes, emit zero.
1172   if (ShiftVal >= 32)
1173     return llvm::Constant::getNullValue(Op0->getType());
1174 
1175   // If palignr is shifting the pair of input vectors more than one lane,
1176   // but less than two lanes, convert to shifting in zeroes.
1177   if (ShiftVal > 16) {
1178     ShiftVal -= 16;
1179     Op1 = Op0;
1180     Op0 = llvm::Constant::getNullValue(Op0->getType());
1181   }
1182 
1183   int Indices[64];
1184   // 256-bit palignr operates on 128-bit lanes so we need to handle that
1185   for (unsigned l = 0; l < NumElts; l += 16) {
1186     for (unsigned i = 0; i != 16; ++i) {
1187       unsigned Idx = ShiftVal + i;
1188       if (!IsVALIGN && Idx >= 16) // Disable wrap for VALIGN.
1189         Idx += NumElts - 16; // End of lane, switch operand.
1190       Indices[l + i] = Idx + l;
1191     }
1192   }
1193 
1194   Value *Align = Builder.CreateShuffleVector(Op1, Op0,
1195                                              makeArrayRef(Indices, NumElts),
1196                                              "palignr");
1197 
1198   return EmitX86Select(Builder, Mask, Align, Passthru);
1199 }
1200 
UpgradeX86VPERMT2Intrinsics(IRBuilder<> & Builder,CallInst & CI,bool ZeroMask,bool IndexForm)1201 static Value *UpgradeX86VPERMT2Intrinsics(IRBuilder<> &Builder, CallInst &CI,
1202                                           bool ZeroMask, bool IndexForm) {
1203   Type *Ty = CI.getType();
1204   unsigned VecWidth = Ty->getPrimitiveSizeInBits();
1205   unsigned EltWidth = Ty->getScalarSizeInBits();
1206   bool IsFloat = Ty->isFPOrFPVectorTy();
1207   Intrinsic::ID IID;
1208   if (VecWidth == 128 && EltWidth == 32 && IsFloat)
1209     IID = Intrinsic::x86_avx512_vpermi2var_ps_128;
1210   else if (VecWidth == 128 && EltWidth == 32 && !IsFloat)
1211     IID = Intrinsic::x86_avx512_vpermi2var_d_128;
1212   else if (VecWidth == 128 && EltWidth == 64 && IsFloat)
1213     IID = Intrinsic::x86_avx512_vpermi2var_pd_128;
1214   else if (VecWidth == 128 && EltWidth == 64 && !IsFloat)
1215     IID = Intrinsic::x86_avx512_vpermi2var_q_128;
1216   else if (VecWidth == 256 && EltWidth == 32 && IsFloat)
1217     IID = Intrinsic::x86_avx512_vpermi2var_ps_256;
1218   else if (VecWidth == 256 && EltWidth == 32 && !IsFloat)
1219     IID = Intrinsic::x86_avx512_vpermi2var_d_256;
1220   else if (VecWidth == 256 && EltWidth == 64 && IsFloat)
1221     IID = Intrinsic::x86_avx512_vpermi2var_pd_256;
1222   else if (VecWidth == 256 && EltWidth == 64 && !IsFloat)
1223     IID = Intrinsic::x86_avx512_vpermi2var_q_256;
1224   else if (VecWidth == 512 && EltWidth == 32 && IsFloat)
1225     IID = Intrinsic::x86_avx512_vpermi2var_ps_512;
1226   else if (VecWidth == 512 && EltWidth == 32 && !IsFloat)
1227     IID = Intrinsic::x86_avx512_vpermi2var_d_512;
1228   else if (VecWidth == 512 && EltWidth == 64 && IsFloat)
1229     IID = Intrinsic::x86_avx512_vpermi2var_pd_512;
1230   else if (VecWidth == 512 && EltWidth == 64 && !IsFloat)
1231     IID = Intrinsic::x86_avx512_vpermi2var_q_512;
1232   else if (VecWidth == 128 && EltWidth == 16)
1233     IID = Intrinsic::x86_avx512_vpermi2var_hi_128;
1234   else if (VecWidth == 256 && EltWidth == 16)
1235     IID = Intrinsic::x86_avx512_vpermi2var_hi_256;
1236   else if (VecWidth == 512 && EltWidth == 16)
1237     IID = Intrinsic::x86_avx512_vpermi2var_hi_512;
1238   else if (VecWidth == 128 && EltWidth == 8)
1239     IID = Intrinsic::x86_avx512_vpermi2var_qi_128;
1240   else if (VecWidth == 256 && EltWidth == 8)
1241     IID = Intrinsic::x86_avx512_vpermi2var_qi_256;
1242   else if (VecWidth == 512 && EltWidth == 8)
1243     IID = Intrinsic::x86_avx512_vpermi2var_qi_512;
1244   else
1245     llvm_unreachable("Unexpected intrinsic");
1246 
1247   Value *Args[] = { CI.getArgOperand(0) , CI.getArgOperand(1),
1248                     CI.getArgOperand(2) };
1249 
1250   // If this isn't index form we need to swap operand 0 and 1.
1251   if (!IndexForm)
1252     std::swap(Args[0], Args[1]);
1253 
1254   Value *V = Builder.CreateCall(Intrinsic::getDeclaration(CI.getModule(), IID),
1255                                 Args);
1256   Value *PassThru = ZeroMask ? ConstantAggregateZero::get(Ty)
1257                              : Builder.CreateBitCast(CI.getArgOperand(1),
1258                                                      Ty);
1259   return EmitX86Select(Builder, CI.getArgOperand(3), V, PassThru);
1260 }
1261 
UpgradeX86BinaryIntrinsics(IRBuilder<> & Builder,CallInst & CI,Intrinsic::ID IID)1262 static Value *UpgradeX86BinaryIntrinsics(IRBuilder<> &Builder, CallInst &CI,
1263                                          Intrinsic::ID IID) {
1264   Type *Ty = CI.getType();
1265   Value *Op0 = CI.getOperand(0);
1266   Value *Op1 = CI.getOperand(1);
1267   Function *Intrin = Intrinsic::getDeclaration(CI.getModule(), IID, Ty);
1268   Value *Res = Builder.CreateCall(Intrin, {Op0, Op1});
1269 
1270   if (CI.getNumArgOperands() == 4) { // For masked intrinsics.
1271     Value *VecSrc = CI.getOperand(2);
1272     Value *Mask = CI.getOperand(3);
1273     Res = EmitX86Select(Builder, Mask, Res, VecSrc);
1274   }
1275   return Res;
1276 }
1277 
upgradeX86Rotate(IRBuilder<> & Builder,CallInst & CI,bool IsRotateRight)1278 static Value *upgradeX86Rotate(IRBuilder<> &Builder, CallInst &CI,
1279                                bool IsRotateRight) {
1280   Type *Ty = CI.getType();
1281   Value *Src = CI.getArgOperand(0);
1282   Value *Amt = CI.getArgOperand(1);
1283 
1284   // Amount may be scalar immediate, in which case create a splat vector.
1285   // Funnel shifts amounts are treated as modulo and types are all power-of-2 so
1286   // we only care about the lowest log2 bits anyway.
1287   if (Amt->getType() != Ty) {
1288     unsigned NumElts = cast<FixedVectorType>(Ty)->getNumElements();
1289     Amt = Builder.CreateIntCast(Amt, Ty->getScalarType(), false);
1290     Amt = Builder.CreateVectorSplat(NumElts, Amt);
1291   }
1292 
1293   Intrinsic::ID IID = IsRotateRight ? Intrinsic::fshr : Intrinsic::fshl;
1294   Function *Intrin = Intrinsic::getDeclaration(CI.getModule(), IID, Ty);
1295   Value *Res = Builder.CreateCall(Intrin, {Src, Src, Amt});
1296 
1297   if (CI.getNumArgOperands() == 4) { // For masked intrinsics.
1298     Value *VecSrc = CI.getOperand(2);
1299     Value *Mask = CI.getOperand(3);
1300     Res = EmitX86Select(Builder, Mask, Res, VecSrc);
1301   }
1302   return Res;
1303 }
1304 
upgradeX86vpcom(IRBuilder<> & Builder,CallInst & CI,unsigned Imm,bool IsSigned)1305 static Value *upgradeX86vpcom(IRBuilder<> &Builder, CallInst &CI, unsigned Imm,
1306                               bool IsSigned) {
1307   Type *Ty = CI.getType();
1308   Value *LHS = CI.getArgOperand(0);
1309   Value *RHS = CI.getArgOperand(1);
1310 
1311   CmpInst::Predicate Pred;
1312   switch (Imm) {
1313   case 0x0:
1314     Pred = IsSigned ? ICmpInst::ICMP_SLT : ICmpInst::ICMP_ULT;
1315     break;
1316   case 0x1:
1317     Pred = IsSigned ? ICmpInst::ICMP_SLE : ICmpInst::ICMP_ULE;
1318     break;
1319   case 0x2:
1320     Pred = IsSigned ? ICmpInst::ICMP_SGT : ICmpInst::ICMP_UGT;
1321     break;
1322   case 0x3:
1323     Pred = IsSigned ? ICmpInst::ICMP_SGE : ICmpInst::ICMP_UGE;
1324     break;
1325   case 0x4:
1326     Pred = ICmpInst::ICMP_EQ;
1327     break;
1328   case 0x5:
1329     Pred = ICmpInst::ICMP_NE;
1330     break;
1331   case 0x6:
1332     return Constant::getNullValue(Ty); // FALSE
1333   case 0x7:
1334     return Constant::getAllOnesValue(Ty); // TRUE
1335   default:
1336     llvm_unreachable("Unknown XOP vpcom/vpcomu predicate");
1337   }
1338 
1339   Value *Cmp = Builder.CreateICmp(Pred, LHS, RHS);
1340   Value *Ext = Builder.CreateSExt(Cmp, Ty);
1341   return Ext;
1342 }
1343 
upgradeX86ConcatShift(IRBuilder<> & Builder,CallInst & CI,bool IsShiftRight,bool ZeroMask)1344 static Value *upgradeX86ConcatShift(IRBuilder<> &Builder, CallInst &CI,
1345                                     bool IsShiftRight, bool ZeroMask) {
1346   Type *Ty = CI.getType();
1347   Value *Op0 = CI.getArgOperand(0);
1348   Value *Op1 = CI.getArgOperand(1);
1349   Value *Amt = CI.getArgOperand(2);
1350 
1351   if (IsShiftRight)
1352     std::swap(Op0, Op1);
1353 
1354   // Amount may be scalar immediate, in which case create a splat vector.
1355   // Funnel shifts amounts are treated as modulo and types are all power-of-2 so
1356   // we only care about the lowest log2 bits anyway.
1357   if (Amt->getType() != Ty) {
1358     unsigned NumElts = cast<FixedVectorType>(Ty)->getNumElements();
1359     Amt = Builder.CreateIntCast(Amt, Ty->getScalarType(), false);
1360     Amt = Builder.CreateVectorSplat(NumElts, Amt);
1361   }
1362 
1363   Intrinsic::ID IID = IsShiftRight ? Intrinsic::fshr : Intrinsic::fshl;
1364   Function *Intrin = Intrinsic::getDeclaration(CI.getModule(), IID, Ty);
1365   Value *Res = Builder.CreateCall(Intrin, {Op0, Op1, Amt});
1366 
1367   unsigned NumArgs = CI.getNumArgOperands();
1368   if (NumArgs >= 4) { // For masked intrinsics.
1369     Value *VecSrc = NumArgs == 5 ? CI.getArgOperand(3) :
1370                     ZeroMask     ? ConstantAggregateZero::get(CI.getType()) :
1371                                    CI.getArgOperand(0);
1372     Value *Mask = CI.getOperand(NumArgs - 1);
1373     Res = EmitX86Select(Builder, Mask, Res, VecSrc);
1374   }
1375   return Res;
1376 }
1377 
UpgradeMaskedStore(IRBuilder<> & Builder,Value * Ptr,Value * Data,Value * Mask,bool Aligned)1378 static Value *UpgradeMaskedStore(IRBuilder<> &Builder,
1379                                  Value *Ptr, Value *Data, Value *Mask,
1380                                  bool Aligned) {
1381   // Cast the pointer to the right type.
1382   Ptr = Builder.CreateBitCast(Ptr,
1383                               llvm::PointerType::getUnqual(Data->getType()));
1384   const Align Alignment =
1385       Aligned
1386           ? Align(Data->getType()->getPrimitiveSizeInBits().getFixedSize() / 8)
1387           : Align(1);
1388 
1389   // If the mask is all ones just emit a regular store.
1390   if (const auto *C = dyn_cast<Constant>(Mask))
1391     if (C->isAllOnesValue())
1392       return Builder.CreateAlignedStore(Data, Ptr, Alignment);
1393 
1394   // Convert the mask from an integer type to a vector of i1.
1395   unsigned NumElts = cast<FixedVectorType>(Data->getType())->getNumElements();
1396   Mask = getX86MaskVec(Builder, Mask, NumElts);
1397   return Builder.CreateMaskedStore(Data, Ptr, Alignment, Mask);
1398 }
1399 
UpgradeMaskedLoad(IRBuilder<> & Builder,Value * Ptr,Value * Passthru,Value * Mask,bool Aligned)1400 static Value *UpgradeMaskedLoad(IRBuilder<> &Builder,
1401                                 Value *Ptr, Value *Passthru, Value *Mask,
1402                                 bool Aligned) {
1403   Type *ValTy = Passthru->getType();
1404   // Cast the pointer to the right type.
1405   Ptr = Builder.CreateBitCast(Ptr, llvm::PointerType::getUnqual(ValTy));
1406   const Align Alignment =
1407       Aligned
1408           ? Align(Passthru->getType()->getPrimitiveSizeInBits().getFixedSize() /
1409                   8)
1410           : Align(1);
1411 
1412   // If the mask is all ones just emit a regular store.
1413   if (const auto *C = dyn_cast<Constant>(Mask))
1414     if (C->isAllOnesValue())
1415       return Builder.CreateAlignedLoad(ValTy, Ptr, Alignment);
1416 
1417   // Convert the mask from an integer type to a vector of i1.
1418   unsigned NumElts =
1419       cast<FixedVectorType>(Passthru->getType())->getNumElements();
1420   Mask = getX86MaskVec(Builder, Mask, NumElts);
1421   return Builder.CreateMaskedLoad(Ptr, Alignment, Mask, Passthru);
1422 }
1423 
upgradeAbs(IRBuilder<> & Builder,CallInst & CI)1424 static Value *upgradeAbs(IRBuilder<> &Builder, CallInst &CI) {
1425   Type *Ty = CI.getType();
1426   Value *Op0 = CI.getArgOperand(0);
1427   Function *F = Intrinsic::getDeclaration(CI.getModule(), Intrinsic::abs, Ty);
1428   Value *Res = Builder.CreateCall(F, {Op0, Builder.getInt1(false)});
1429   if (CI.getNumArgOperands() == 3)
1430     Res = EmitX86Select(Builder, CI.getArgOperand(2), Res, CI.getArgOperand(1));
1431   return Res;
1432 }
1433 
upgradePMULDQ(IRBuilder<> & Builder,CallInst & CI,bool IsSigned)1434 static Value *upgradePMULDQ(IRBuilder<> &Builder, CallInst &CI, bool IsSigned) {
1435   Type *Ty = CI.getType();
1436 
1437   // Arguments have a vXi32 type so cast to vXi64.
1438   Value *LHS = Builder.CreateBitCast(CI.getArgOperand(0), Ty);
1439   Value *RHS = Builder.CreateBitCast(CI.getArgOperand(1), Ty);
1440 
1441   if (IsSigned) {
1442     // Shift left then arithmetic shift right.
1443     Constant *ShiftAmt = ConstantInt::get(Ty, 32);
1444     LHS = Builder.CreateShl(LHS, ShiftAmt);
1445     LHS = Builder.CreateAShr(LHS, ShiftAmt);
1446     RHS = Builder.CreateShl(RHS, ShiftAmt);
1447     RHS = Builder.CreateAShr(RHS, ShiftAmt);
1448   } else {
1449     // Clear the upper bits.
1450     Constant *Mask = ConstantInt::get(Ty, 0xffffffff);
1451     LHS = Builder.CreateAnd(LHS, Mask);
1452     RHS = Builder.CreateAnd(RHS, Mask);
1453   }
1454 
1455   Value *Res = Builder.CreateMul(LHS, RHS);
1456 
1457   if (CI.getNumArgOperands() == 4)
1458     Res = EmitX86Select(Builder, CI.getArgOperand(3), Res, CI.getArgOperand(2));
1459 
1460   return Res;
1461 }
1462 
1463 // Applying mask on vector of i1's and make sure result is at least 8 bits wide.
ApplyX86MaskOn1BitsVec(IRBuilder<> & Builder,Value * Vec,Value * Mask)1464 static Value *ApplyX86MaskOn1BitsVec(IRBuilder<> &Builder, Value *Vec,
1465                                      Value *Mask) {
1466   unsigned NumElts = cast<FixedVectorType>(Vec->getType())->getNumElements();
1467   if (Mask) {
1468     const auto *C = dyn_cast<Constant>(Mask);
1469     if (!C || !C->isAllOnesValue())
1470       Vec = Builder.CreateAnd(Vec, getX86MaskVec(Builder, Mask, NumElts));
1471   }
1472 
1473   if (NumElts < 8) {
1474     int Indices[8];
1475     for (unsigned i = 0; i != NumElts; ++i)
1476       Indices[i] = i;
1477     for (unsigned i = NumElts; i != 8; ++i)
1478       Indices[i] = NumElts + i % NumElts;
1479     Vec = Builder.CreateShuffleVector(Vec,
1480                                       Constant::getNullValue(Vec->getType()),
1481                                       Indices);
1482   }
1483   return Builder.CreateBitCast(Vec, Builder.getIntNTy(std::max(NumElts, 8U)));
1484 }
1485 
upgradeMaskedCompare(IRBuilder<> & Builder,CallInst & CI,unsigned CC,bool Signed)1486 static Value *upgradeMaskedCompare(IRBuilder<> &Builder, CallInst &CI,
1487                                    unsigned CC, bool Signed) {
1488   Value *Op0 = CI.getArgOperand(0);
1489   unsigned NumElts = cast<FixedVectorType>(Op0->getType())->getNumElements();
1490 
1491   Value *Cmp;
1492   if (CC == 3) {
1493     Cmp = Constant::getNullValue(
1494         FixedVectorType::get(Builder.getInt1Ty(), NumElts));
1495   } else if (CC == 7) {
1496     Cmp = Constant::getAllOnesValue(
1497         FixedVectorType::get(Builder.getInt1Ty(), NumElts));
1498   } else {
1499     ICmpInst::Predicate Pred;
1500     switch (CC) {
1501     default: llvm_unreachable("Unknown condition code");
1502     case 0: Pred = ICmpInst::ICMP_EQ;  break;
1503     case 1: Pred = Signed ? ICmpInst::ICMP_SLT : ICmpInst::ICMP_ULT; break;
1504     case 2: Pred = Signed ? ICmpInst::ICMP_SLE : ICmpInst::ICMP_ULE; break;
1505     case 4: Pred = ICmpInst::ICMP_NE;  break;
1506     case 5: Pred = Signed ? ICmpInst::ICMP_SGE : ICmpInst::ICMP_UGE; break;
1507     case 6: Pred = Signed ? ICmpInst::ICMP_SGT : ICmpInst::ICMP_UGT; break;
1508     }
1509     Cmp = Builder.CreateICmp(Pred, Op0, CI.getArgOperand(1));
1510   }
1511 
1512   Value *Mask = CI.getArgOperand(CI.getNumArgOperands() - 1);
1513 
1514   return ApplyX86MaskOn1BitsVec(Builder, Cmp, Mask);
1515 }
1516 
1517 // Replace a masked intrinsic with an older unmasked intrinsic.
UpgradeX86MaskedShift(IRBuilder<> & Builder,CallInst & CI,Intrinsic::ID IID)1518 static Value *UpgradeX86MaskedShift(IRBuilder<> &Builder, CallInst &CI,
1519                                     Intrinsic::ID IID) {
1520   Function *Intrin = Intrinsic::getDeclaration(CI.getModule(), IID);
1521   Value *Rep = Builder.CreateCall(Intrin,
1522                                  { CI.getArgOperand(0), CI.getArgOperand(1) });
1523   return EmitX86Select(Builder, CI.getArgOperand(3), Rep, CI.getArgOperand(2));
1524 }
1525 
upgradeMaskedMove(IRBuilder<> & Builder,CallInst & CI)1526 static Value* upgradeMaskedMove(IRBuilder<> &Builder, CallInst &CI) {
1527   Value* A = CI.getArgOperand(0);
1528   Value* B = CI.getArgOperand(1);
1529   Value* Src = CI.getArgOperand(2);
1530   Value* Mask = CI.getArgOperand(3);
1531 
1532   Value* AndNode = Builder.CreateAnd(Mask, APInt(8, 1));
1533   Value* Cmp = Builder.CreateIsNotNull(AndNode);
1534   Value* Extract1 = Builder.CreateExtractElement(B, (uint64_t)0);
1535   Value* Extract2 = Builder.CreateExtractElement(Src, (uint64_t)0);
1536   Value* Select = Builder.CreateSelect(Cmp, Extract1, Extract2);
1537   return Builder.CreateInsertElement(A, Select, (uint64_t)0);
1538 }
1539 
1540 
UpgradeMaskToInt(IRBuilder<> & Builder,CallInst & CI)1541 static Value* UpgradeMaskToInt(IRBuilder<> &Builder, CallInst &CI) {
1542   Value* Op = CI.getArgOperand(0);
1543   Type* ReturnOp = CI.getType();
1544   unsigned NumElts = cast<FixedVectorType>(CI.getType())->getNumElements();
1545   Value *Mask = getX86MaskVec(Builder, Op, NumElts);
1546   return Builder.CreateSExt(Mask, ReturnOp, "vpmovm2");
1547 }
1548 
1549 // Replace intrinsic with unmasked version and a select.
upgradeAVX512MaskToSelect(StringRef Name,IRBuilder<> & Builder,CallInst & CI,Value * & Rep)1550 static bool upgradeAVX512MaskToSelect(StringRef Name, IRBuilder<> &Builder,
1551                                       CallInst &CI, Value *&Rep) {
1552   Name = Name.substr(12); // Remove avx512.mask.
1553 
1554   unsigned VecWidth = CI.getType()->getPrimitiveSizeInBits();
1555   unsigned EltWidth = CI.getType()->getScalarSizeInBits();
1556   Intrinsic::ID IID;
1557   if (Name.startswith("max.p")) {
1558     if (VecWidth == 128 && EltWidth == 32)
1559       IID = Intrinsic::x86_sse_max_ps;
1560     else if (VecWidth == 128 && EltWidth == 64)
1561       IID = Intrinsic::x86_sse2_max_pd;
1562     else if (VecWidth == 256 && EltWidth == 32)
1563       IID = Intrinsic::x86_avx_max_ps_256;
1564     else if (VecWidth == 256 && EltWidth == 64)
1565       IID = Intrinsic::x86_avx_max_pd_256;
1566     else
1567       llvm_unreachable("Unexpected intrinsic");
1568   } else if (Name.startswith("min.p")) {
1569     if (VecWidth == 128 && EltWidth == 32)
1570       IID = Intrinsic::x86_sse_min_ps;
1571     else if (VecWidth == 128 && EltWidth == 64)
1572       IID = Intrinsic::x86_sse2_min_pd;
1573     else if (VecWidth == 256 && EltWidth == 32)
1574       IID = Intrinsic::x86_avx_min_ps_256;
1575     else if (VecWidth == 256 && EltWidth == 64)
1576       IID = Intrinsic::x86_avx_min_pd_256;
1577     else
1578       llvm_unreachable("Unexpected intrinsic");
1579   } else if (Name.startswith("pshuf.b.")) {
1580     if (VecWidth == 128)
1581       IID = Intrinsic::x86_ssse3_pshuf_b_128;
1582     else if (VecWidth == 256)
1583       IID = Intrinsic::x86_avx2_pshuf_b;
1584     else if (VecWidth == 512)
1585       IID = Intrinsic::x86_avx512_pshuf_b_512;
1586     else
1587       llvm_unreachable("Unexpected intrinsic");
1588   } else if (Name.startswith("pmul.hr.sw.")) {
1589     if (VecWidth == 128)
1590       IID = Intrinsic::x86_ssse3_pmul_hr_sw_128;
1591     else if (VecWidth == 256)
1592       IID = Intrinsic::x86_avx2_pmul_hr_sw;
1593     else if (VecWidth == 512)
1594       IID = Intrinsic::x86_avx512_pmul_hr_sw_512;
1595     else
1596       llvm_unreachable("Unexpected intrinsic");
1597   } else if (Name.startswith("pmulh.w.")) {
1598     if (VecWidth == 128)
1599       IID = Intrinsic::x86_sse2_pmulh_w;
1600     else if (VecWidth == 256)
1601       IID = Intrinsic::x86_avx2_pmulh_w;
1602     else if (VecWidth == 512)
1603       IID = Intrinsic::x86_avx512_pmulh_w_512;
1604     else
1605       llvm_unreachable("Unexpected intrinsic");
1606   } else if (Name.startswith("pmulhu.w.")) {
1607     if (VecWidth == 128)
1608       IID = Intrinsic::x86_sse2_pmulhu_w;
1609     else if (VecWidth == 256)
1610       IID = Intrinsic::x86_avx2_pmulhu_w;
1611     else if (VecWidth == 512)
1612       IID = Intrinsic::x86_avx512_pmulhu_w_512;
1613     else
1614       llvm_unreachable("Unexpected intrinsic");
1615   } else if (Name.startswith("pmaddw.d.")) {
1616     if (VecWidth == 128)
1617       IID = Intrinsic::x86_sse2_pmadd_wd;
1618     else if (VecWidth == 256)
1619       IID = Intrinsic::x86_avx2_pmadd_wd;
1620     else if (VecWidth == 512)
1621       IID = Intrinsic::x86_avx512_pmaddw_d_512;
1622     else
1623       llvm_unreachable("Unexpected intrinsic");
1624   } else if (Name.startswith("pmaddubs.w.")) {
1625     if (VecWidth == 128)
1626       IID = Intrinsic::x86_ssse3_pmadd_ub_sw_128;
1627     else if (VecWidth == 256)
1628       IID = Intrinsic::x86_avx2_pmadd_ub_sw;
1629     else if (VecWidth == 512)
1630       IID = Intrinsic::x86_avx512_pmaddubs_w_512;
1631     else
1632       llvm_unreachable("Unexpected intrinsic");
1633   } else if (Name.startswith("packsswb.")) {
1634     if (VecWidth == 128)
1635       IID = Intrinsic::x86_sse2_packsswb_128;
1636     else if (VecWidth == 256)
1637       IID = Intrinsic::x86_avx2_packsswb;
1638     else if (VecWidth == 512)
1639       IID = Intrinsic::x86_avx512_packsswb_512;
1640     else
1641       llvm_unreachable("Unexpected intrinsic");
1642   } else if (Name.startswith("packssdw.")) {
1643     if (VecWidth == 128)
1644       IID = Intrinsic::x86_sse2_packssdw_128;
1645     else if (VecWidth == 256)
1646       IID = Intrinsic::x86_avx2_packssdw;
1647     else if (VecWidth == 512)
1648       IID = Intrinsic::x86_avx512_packssdw_512;
1649     else
1650       llvm_unreachable("Unexpected intrinsic");
1651   } else if (Name.startswith("packuswb.")) {
1652     if (VecWidth == 128)
1653       IID = Intrinsic::x86_sse2_packuswb_128;
1654     else if (VecWidth == 256)
1655       IID = Intrinsic::x86_avx2_packuswb;
1656     else if (VecWidth == 512)
1657       IID = Intrinsic::x86_avx512_packuswb_512;
1658     else
1659       llvm_unreachable("Unexpected intrinsic");
1660   } else if (Name.startswith("packusdw.")) {
1661     if (VecWidth == 128)
1662       IID = Intrinsic::x86_sse41_packusdw;
1663     else if (VecWidth == 256)
1664       IID = Intrinsic::x86_avx2_packusdw;
1665     else if (VecWidth == 512)
1666       IID = Intrinsic::x86_avx512_packusdw_512;
1667     else
1668       llvm_unreachable("Unexpected intrinsic");
1669   } else if (Name.startswith("vpermilvar.")) {
1670     if (VecWidth == 128 && EltWidth == 32)
1671       IID = Intrinsic::x86_avx_vpermilvar_ps;
1672     else if (VecWidth == 128 && EltWidth == 64)
1673       IID = Intrinsic::x86_avx_vpermilvar_pd;
1674     else if (VecWidth == 256 && EltWidth == 32)
1675       IID = Intrinsic::x86_avx_vpermilvar_ps_256;
1676     else if (VecWidth == 256 && EltWidth == 64)
1677       IID = Intrinsic::x86_avx_vpermilvar_pd_256;
1678     else if (VecWidth == 512 && EltWidth == 32)
1679       IID = Intrinsic::x86_avx512_vpermilvar_ps_512;
1680     else if (VecWidth == 512 && EltWidth == 64)
1681       IID = Intrinsic::x86_avx512_vpermilvar_pd_512;
1682     else
1683       llvm_unreachable("Unexpected intrinsic");
1684   } else if (Name == "cvtpd2dq.256") {
1685     IID = Intrinsic::x86_avx_cvt_pd2dq_256;
1686   } else if (Name == "cvtpd2ps.256") {
1687     IID = Intrinsic::x86_avx_cvt_pd2_ps_256;
1688   } else if (Name == "cvttpd2dq.256") {
1689     IID = Intrinsic::x86_avx_cvtt_pd2dq_256;
1690   } else if (Name == "cvttps2dq.128") {
1691     IID = Intrinsic::x86_sse2_cvttps2dq;
1692   } else if (Name == "cvttps2dq.256") {
1693     IID = Intrinsic::x86_avx_cvtt_ps2dq_256;
1694   } else if (Name.startswith("permvar.")) {
1695     bool IsFloat = CI.getType()->isFPOrFPVectorTy();
1696     if (VecWidth == 256 && EltWidth == 32 && IsFloat)
1697       IID = Intrinsic::x86_avx2_permps;
1698     else if (VecWidth == 256 && EltWidth == 32 && !IsFloat)
1699       IID = Intrinsic::x86_avx2_permd;
1700     else if (VecWidth == 256 && EltWidth == 64 && IsFloat)
1701       IID = Intrinsic::x86_avx512_permvar_df_256;
1702     else if (VecWidth == 256 && EltWidth == 64 && !IsFloat)
1703       IID = Intrinsic::x86_avx512_permvar_di_256;
1704     else if (VecWidth == 512 && EltWidth == 32 && IsFloat)
1705       IID = Intrinsic::x86_avx512_permvar_sf_512;
1706     else if (VecWidth == 512 && EltWidth == 32 && !IsFloat)
1707       IID = Intrinsic::x86_avx512_permvar_si_512;
1708     else if (VecWidth == 512 && EltWidth == 64 && IsFloat)
1709       IID = Intrinsic::x86_avx512_permvar_df_512;
1710     else if (VecWidth == 512 && EltWidth == 64 && !IsFloat)
1711       IID = Intrinsic::x86_avx512_permvar_di_512;
1712     else if (VecWidth == 128 && EltWidth == 16)
1713       IID = Intrinsic::x86_avx512_permvar_hi_128;
1714     else if (VecWidth == 256 && EltWidth == 16)
1715       IID = Intrinsic::x86_avx512_permvar_hi_256;
1716     else if (VecWidth == 512 && EltWidth == 16)
1717       IID = Intrinsic::x86_avx512_permvar_hi_512;
1718     else if (VecWidth == 128 && EltWidth == 8)
1719       IID = Intrinsic::x86_avx512_permvar_qi_128;
1720     else if (VecWidth == 256 && EltWidth == 8)
1721       IID = Intrinsic::x86_avx512_permvar_qi_256;
1722     else if (VecWidth == 512 && EltWidth == 8)
1723       IID = Intrinsic::x86_avx512_permvar_qi_512;
1724     else
1725       llvm_unreachable("Unexpected intrinsic");
1726   } else if (Name.startswith("dbpsadbw.")) {
1727     if (VecWidth == 128)
1728       IID = Intrinsic::x86_avx512_dbpsadbw_128;
1729     else if (VecWidth == 256)
1730       IID = Intrinsic::x86_avx512_dbpsadbw_256;
1731     else if (VecWidth == 512)
1732       IID = Intrinsic::x86_avx512_dbpsadbw_512;
1733     else
1734       llvm_unreachable("Unexpected intrinsic");
1735   } else if (Name.startswith("pmultishift.qb.")) {
1736     if (VecWidth == 128)
1737       IID = Intrinsic::x86_avx512_pmultishift_qb_128;
1738     else if (VecWidth == 256)
1739       IID = Intrinsic::x86_avx512_pmultishift_qb_256;
1740     else if (VecWidth == 512)
1741       IID = Intrinsic::x86_avx512_pmultishift_qb_512;
1742     else
1743       llvm_unreachable("Unexpected intrinsic");
1744   } else if (Name.startswith("conflict.")) {
1745     if (Name[9] == 'd' && VecWidth == 128)
1746       IID = Intrinsic::x86_avx512_conflict_d_128;
1747     else if (Name[9] == 'd' && VecWidth == 256)
1748       IID = Intrinsic::x86_avx512_conflict_d_256;
1749     else if (Name[9] == 'd' && VecWidth == 512)
1750       IID = Intrinsic::x86_avx512_conflict_d_512;
1751     else if (Name[9] == 'q' && VecWidth == 128)
1752       IID = Intrinsic::x86_avx512_conflict_q_128;
1753     else if (Name[9] == 'q' && VecWidth == 256)
1754       IID = Intrinsic::x86_avx512_conflict_q_256;
1755     else if (Name[9] == 'q' && VecWidth == 512)
1756       IID = Intrinsic::x86_avx512_conflict_q_512;
1757     else
1758       llvm_unreachable("Unexpected intrinsic");
1759   } else if (Name.startswith("pavg.")) {
1760     if (Name[5] == 'b' && VecWidth == 128)
1761       IID = Intrinsic::x86_sse2_pavg_b;
1762     else if (Name[5] == 'b' && VecWidth == 256)
1763       IID = Intrinsic::x86_avx2_pavg_b;
1764     else if (Name[5] == 'b' && VecWidth == 512)
1765       IID = Intrinsic::x86_avx512_pavg_b_512;
1766     else if (Name[5] == 'w' && VecWidth == 128)
1767       IID = Intrinsic::x86_sse2_pavg_w;
1768     else if (Name[5] == 'w' && VecWidth == 256)
1769       IID = Intrinsic::x86_avx2_pavg_w;
1770     else if (Name[5] == 'w' && VecWidth == 512)
1771       IID = Intrinsic::x86_avx512_pavg_w_512;
1772     else
1773       llvm_unreachable("Unexpected intrinsic");
1774   } else
1775     return false;
1776 
1777   SmallVector<Value *, 4> Args(CI.arg_operands().begin(),
1778                                CI.arg_operands().end());
1779   Args.pop_back();
1780   Args.pop_back();
1781   Rep = Builder.CreateCall(Intrinsic::getDeclaration(CI.getModule(), IID),
1782                            Args);
1783   unsigned NumArgs = CI.getNumArgOperands();
1784   Rep = EmitX86Select(Builder, CI.getArgOperand(NumArgs - 1), Rep,
1785                       CI.getArgOperand(NumArgs - 2));
1786   return true;
1787 }
1788 
1789 /// Upgrade comment in call to inline asm that represents an objc retain release
1790 /// marker.
UpgradeInlineAsmString(std::string * AsmStr)1791 void llvm::UpgradeInlineAsmString(std::string *AsmStr) {
1792   size_t Pos;
1793   if (AsmStr->find("mov\tfp") == 0 &&
1794       AsmStr->find("objc_retainAutoreleaseReturnValue") != std::string::npos &&
1795       (Pos = AsmStr->find("# marker")) != std::string::npos) {
1796     AsmStr->replace(Pos, 1, ";");
1797   }
1798 }
1799 
1800 /// Upgrade a call to an old intrinsic. All argument and return casting must be
1801 /// provided to seamlessly integrate with existing context.
UpgradeIntrinsicCall(CallInst * CI,Function * NewFn)1802 void llvm::UpgradeIntrinsicCall(CallInst *CI, Function *NewFn) {
1803   Function *F = CI->getCalledFunction();
1804   LLVMContext &C = CI->getContext();
1805   IRBuilder<> Builder(C);
1806   Builder.SetInsertPoint(CI->getParent(), CI->getIterator());
1807 
1808   assert(F && "Intrinsic call is not direct?");
1809 
1810   if (!NewFn) {
1811     // Get the Function's name.
1812     StringRef Name = F->getName();
1813 
1814     assert(Name.startswith("llvm.") && "Intrinsic doesn't start with 'llvm.'");
1815     Name = Name.substr(5);
1816 
1817     bool IsX86 = Name.startswith("x86.");
1818     if (IsX86)
1819       Name = Name.substr(4);
1820     bool IsNVVM = Name.startswith("nvvm.");
1821     if (IsNVVM)
1822       Name = Name.substr(5);
1823 
1824     if (IsX86 && Name.startswith("sse4a.movnt.")) {
1825       Module *M = F->getParent();
1826       SmallVector<Metadata *, 1> Elts;
1827       Elts.push_back(
1828           ConstantAsMetadata::get(ConstantInt::get(Type::getInt32Ty(C), 1)));
1829       MDNode *Node = MDNode::get(C, Elts);
1830 
1831       Value *Arg0 = CI->getArgOperand(0);
1832       Value *Arg1 = CI->getArgOperand(1);
1833 
1834       // Nontemporal (unaligned) store of the 0'th element of the float/double
1835       // vector.
1836       Type *SrcEltTy = cast<VectorType>(Arg1->getType())->getElementType();
1837       PointerType *EltPtrTy = PointerType::getUnqual(SrcEltTy);
1838       Value *Addr = Builder.CreateBitCast(Arg0, EltPtrTy, "cast");
1839       Value *Extract =
1840           Builder.CreateExtractElement(Arg1, (uint64_t)0, "extractelement");
1841 
1842       StoreInst *SI = Builder.CreateAlignedStore(Extract, Addr, Align(1));
1843       SI->setMetadata(M->getMDKindID("nontemporal"), Node);
1844 
1845       // Remove intrinsic.
1846       CI->eraseFromParent();
1847       return;
1848     }
1849 
1850     if (IsX86 && (Name.startswith("avx.movnt.") ||
1851                   Name.startswith("avx512.storent."))) {
1852       Module *M = F->getParent();
1853       SmallVector<Metadata *, 1> Elts;
1854       Elts.push_back(
1855           ConstantAsMetadata::get(ConstantInt::get(Type::getInt32Ty(C), 1)));
1856       MDNode *Node = MDNode::get(C, Elts);
1857 
1858       Value *Arg0 = CI->getArgOperand(0);
1859       Value *Arg1 = CI->getArgOperand(1);
1860 
1861       // Convert the type of the pointer to a pointer to the stored type.
1862       Value *BC = Builder.CreateBitCast(Arg0,
1863                                         PointerType::getUnqual(Arg1->getType()),
1864                                         "cast");
1865       StoreInst *SI = Builder.CreateAlignedStore(
1866           Arg1, BC,
1867           Align(Arg1->getType()->getPrimitiveSizeInBits().getFixedSize() / 8));
1868       SI->setMetadata(M->getMDKindID("nontemporal"), Node);
1869 
1870       // Remove intrinsic.
1871       CI->eraseFromParent();
1872       return;
1873     }
1874 
1875     if (IsX86 && Name == "sse2.storel.dq") {
1876       Value *Arg0 = CI->getArgOperand(0);
1877       Value *Arg1 = CI->getArgOperand(1);
1878 
1879       auto *NewVecTy = FixedVectorType::get(Type::getInt64Ty(C), 2);
1880       Value *BC0 = Builder.CreateBitCast(Arg1, NewVecTy, "cast");
1881       Value *Elt = Builder.CreateExtractElement(BC0, (uint64_t)0);
1882       Value *BC = Builder.CreateBitCast(Arg0,
1883                                         PointerType::getUnqual(Elt->getType()),
1884                                         "cast");
1885       Builder.CreateAlignedStore(Elt, BC, Align(1));
1886 
1887       // Remove intrinsic.
1888       CI->eraseFromParent();
1889       return;
1890     }
1891 
1892     if (IsX86 && (Name.startswith("sse.storeu.") ||
1893                   Name.startswith("sse2.storeu.") ||
1894                   Name.startswith("avx.storeu."))) {
1895       Value *Arg0 = CI->getArgOperand(0);
1896       Value *Arg1 = CI->getArgOperand(1);
1897 
1898       Arg0 = Builder.CreateBitCast(Arg0,
1899                                    PointerType::getUnqual(Arg1->getType()),
1900                                    "cast");
1901       Builder.CreateAlignedStore(Arg1, Arg0, Align(1));
1902 
1903       // Remove intrinsic.
1904       CI->eraseFromParent();
1905       return;
1906     }
1907 
1908     if (IsX86 && Name == "avx512.mask.store.ss") {
1909       Value *Mask = Builder.CreateAnd(CI->getArgOperand(2), Builder.getInt8(1));
1910       UpgradeMaskedStore(Builder, CI->getArgOperand(0), CI->getArgOperand(1),
1911                          Mask, false);
1912 
1913       // Remove intrinsic.
1914       CI->eraseFromParent();
1915       return;
1916     }
1917 
1918     if (IsX86 && (Name.startswith("avx512.mask.store"))) {
1919       // "avx512.mask.storeu." or "avx512.mask.store."
1920       bool Aligned = Name[17] != 'u'; // "avx512.mask.storeu".
1921       UpgradeMaskedStore(Builder, CI->getArgOperand(0), CI->getArgOperand(1),
1922                          CI->getArgOperand(2), Aligned);
1923 
1924       // Remove intrinsic.
1925       CI->eraseFromParent();
1926       return;
1927     }
1928 
1929     Value *Rep;
1930     // Upgrade packed integer vector compare intrinsics to compare instructions.
1931     if (IsX86 && (Name.startswith("sse2.pcmp") ||
1932                   Name.startswith("avx2.pcmp"))) {
1933       // "sse2.pcpmpeq." "sse2.pcmpgt." "avx2.pcmpeq." or "avx2.pcmpgt."
1934       bool CmpEq = Name[9] == 'e';
1935       Rep = Builder.CreateICmp(CmpEq ? ICmpInst::ICMP_EQ : ICmpInst::ICMP_SGT,
1936                                CI->getArgOperand(0), CI->getArgOperand(1));
1937       Rep = Builder.CreateSExt(Rep, CI->getType(), "");
1938     } else if (IsX86 && (Name.startswith("avx512.broadcastm"))) {
1939       Type *ExtTy = Type::getInt32Ty(C);
1940       if (CI->getOperand(0)->getType()->isIntegerTy(8))
1941         ExtTy = Type::getInt64Ty(C);
1942       unsigned NumElts = CI->getType()->getPrimitiveSizeInBits() /
1943                          ExtTy->getPrimitiveSizeInBits();
1944       Rep = Builder.CreateZExt(CI->getArgOperand(0), ExtTy);
1945       Rep = Builder.CreateVectorSplat(NumElts, Rep);
1946     } else if (IsX86 && (Name == "sse.sqrt.ss" ||
1947                          Name == "sse2.sqrt.sd")) {
1948       Value *Vec = CI->getArgOperand(0);
1949       Value *Elt0 = Builder.CreateExtractElement(Vec, (uint64_t)0);
1950       Function *Intr = Intrinsic::getDeclaration(F->getParent(),
1951                                                  Intrinsic::sqrt, Elt0->getType());
1952       Elt0 = Builder.CreateCall(Intr, Elt0);
1953       Rep = Builder.CreateInsertElement(Vec, Elt0, (uint64_t)0);
1954     } else if (IsX86 && (Name.startswith("avx.sqrt.p") ||
1955                          Name.startswith("sse2.sqrt.p") ||
1956                          Name.startswith("sse.sqrt.p"))) {
1957       Rep = Builder.CreateCall(Intrinsic::getDeclaration(F->getParent(),
1958                                                          Intrinsic::sqrt,
1959                                                          CI->getType()),
1960                                {CI->getArgOperand(0)});
1961     } else if (IsX86 && (Name.startswith("avx512.mask.sqrt.p"))) {
1962       if (CI->getNumArgOperands() == 4 &&
1963           (!isa<ConstantInt>(CI->getArgOperand(3)) ||
1964            cast<ConstantInt>(CI->getArgOperand(3))->getZExtValue() != 4)) {
1965         Intrinsic::ID IID = Name[18] == 's' ? Intrinsic::x86_avx512_sqrt_ps_512
1966                                             : Intrinsic::x86_avx512_sqrt_pd_512;
1967 
1968         Value *Args[] = { CI->getArgOperand(0), CI->getArgOperand(3) };
1969         Rep = Builder.CreateCall(Intrinsic::getDeclaration(CI->getModule(),
1970                                                            IID), Args);
1971       } else {
1972         Rep = Builder.CreateCall(Intrinsic::getDeclaration(F->getParent(),
1973                                                            Intrinsic::sqrt,
1974                                                            CI->getType()),
1975                                  {CI->getArgOperand(0)});
1976       }
1977       Rep = EmitX86Select(Builder, CI->getArgOperand(2), Rep,
1978                           CI->getArgOperand(1));
1979     } else if (IsX86 && (Name.startswith("avx512.ptestm") ||
1980                          Name.startswith("avx512.ptestnm"))) {
1981       Value *Op0 = CI->getArgOperand(0);
1982       Value *Op1 = CI->getArgOperand(1);
1983       Value *Mask = CI->getArgOperand(2);
1984       Rep = Builder.CreateAnd(Op0, Op1);
1985       llvm::Type *Ty = Op0->getType();
1986       Value *Zero = llvm::Constant::getNullValue(Ty);
1987       ICmpInst::Predicate Pred =
1988         Name.startswith("avx512.ptestm") ? ICmpInst::ICMP_NE : ICmpInst::ICMP_EQ;
1989       Rep = Builder.CreateICmp(Pred, Rep, Zero);
1990       Rep = ApplyX86MaskOn1BitsVec(Builder, Rep, Mask);
1991     } else if (IsX86 && (Name.startswith("avx512.mask.pbroadcast"))){
1992       unsigned NumElts = cast<FixedVectorType>(CI->getArgOperand(1)->getType())
1993                              ->getNumElements();
1994       Rep = Builder.CreateVectorSplat(NumElts, CI->getArgOperand(0));
1995       Rep = EmitX86Select(Builder, CI->getArgOperand(2), Rep,
1996                           CI->getArgOperand(1));
1997     } else if (IsX86 && (Name.startswith("avx512.kunpck"))) {
1998       unsigned NumElts = CI->getType()->getScalarSizeInBits();
1999       Value *LHS = getX86MaskVec(Builder, CI->getArgOperand(0), NumElts);
2000       Value *RHS = getX86MaskVec(Builder, CI->getArgOperand(1), NumElts);
2001       int Indices[64];
2002       for (unsigned i = 0; i != NumElts; ++i)
2003         Indices[i] = i;
2004 
2005       // First extract half of each vector. This gives better codegen than
2006       // doing it in a single shuffle.
2007       LHS = Builder.CreateShuffleVector(LHS, LHS,
2008                                         makeArrayRef(Indices, NumElts / 2));
2009       RHS = Builder.CreateShuffleVector(RHS, RHS,
2010                                         makeArrayRef(Indices, NumElts / 2));
2011       // Concat the vectors.
2012       // NOTE: Operands have to be swapped to match intrinsic definition.
2013       Rep = Builder.CreateShuffleVector(RHS, LHS,
2014                                         makeArrayRef(Indices, NumElts));
2015       Rep = Builder.CreateBitCast(Rep, CI->getType());
2016     } else if (IsX86 && Name == "avx512.kand.w") {
2017       Value *LHS = getX86MaskVec(Builder, CI->getArgOperand(0), 16);
2018       Value *RHS = getX86MaskVec(Builder, CI->getArgOperand(1), 16);
2019       Rep = Builder.CreateAnd(LHS, RHS);
2020       Rep = Builder.CreateBitCast(Rep, CI->getType());
2021     } else if (IsX86 && Name == "avx512.kandn.w") {
2022       Value *LHS = getX86MaskVec(Builder, CI->getArgOperand(0), 16);
2023       Value *RHS = getX86MaskVec(Builder, CI->getArgOperand(1), 16);
2024       LHS = Builder.CreateNot(LHS);
2025       Rep = Builder.CreateAnd(LHS, RHS);
2026       Rep = Builder.CreateBitCast(Rep, CI->getType());
2027     } else if (IsX86 && Name == "avx512.kor.w") {
2028       Value *LHS = getX86MaskVec(Builder, CI->getArgOperand(0), 16);
2029       Value *RHS = getX86MaskVec(Builder, CI->getArgOperand(1), 16);
2030       Rep = Builder.CreateOr(LHS, RHS);
2031       Rep = Builder.CreateBitCast(Rep, CI->getType());
2032     } else if (IsX86 && Name == "avx512.kxor.w") {
2033       Value *LHS = getX86MaskVec(Builder, CI->getArgOperand(0), 16);
2034       Value *RHS = getX86MaskVec(Builder, CI->getArgOperand(1), 16);
2035       Rep = Builder.CreateXor(LHS, RHS);
2036       Rep = Builder.CreateBitCast(Rep, CI->getType());
2037     } else if (IsX86 && Name == "avx512.kxnor.w") {
2038       Value *LHS = getX86MaskVec(Builder, CI->getArgOperand(0), 16);
2039       Value *RHS = getX86MaskVec(Builder, CI->getArgOperand(1), 16);
2040       LHS = Builder.CreateNot(LHS);
2041       Rep = Builder.CreateXor(LHS, RHS);
2042       Rep = Builder.CreateBitCast(Rep, CI->getType());
2043     } else if (IsX86 && Name == "avx512.knot.w") {
2044       Rep = getX86MaskVec(Builder, CI->getArgOperand(0), 16);
2045       Rep = Builder.CreateNot(Rep);
2046       Rep = Builder.CreateBitCast(Rep, CI->getType());
2047     } else if (IsX86 &&
2048                (Name == "avx512.kortestz.w" || Name == "avx512.kortestc.w")) {
2049       Value *LHS = getX86MaskVec(Builder, CI->getArgOperand(0), 16);
2050       Value *RHS = getX86MaskVec(Builder, CI->getArgOperand(1), 16);
2051       Rep = Builder.CreateOr(LHS, RHS);
2052       Rep = Builder.CreateBitCast(Rep, Builder.getInt16Ty());
2053       Value *C;
2054       if (Name[14] == 'c')
2055         C = ConstantInt::getAllOnesValue(Builder.getInt16Ty());
2056       else
2057         C = ConstantInt::getNullValue(Builder.getInt16Ty());
2058       Rep = Builder.CreateICmpEQ(Rep, C);
2059       Rep = Builder.CreateZExt(Rep, Builder.getInt32Ty());
2060     } else if (IsX86 && (Name == "sse.add.ss" || Name == "sse2.add.sd" ||
2061                          Name == "sse.sub.ss" || Name == "sse2.sub.sd" ||
2062                          Name == "sse.mul.ss" || Name == "sse2.mul.sd" ||
2063                          Name == "sse.div.ss" || Name == "sse2.div.sd")) {
2064       Type *I32Ty = Type::getInt32Ty(C);
2065       Value *Elt0 = Builder.CreateExtractElement(CI->getArgOperand(0),
2066                                                  ConstantInt::get(I32Ty, 0));
2067       Value *Elt1 = Builder.CreateExtractElement(CI->getArgOperand(1),
2068                                                  ConstantInt::get(I32Ty, 0));
2069       Value *EltOp;
2070       if (Name.contains(".add."))
2071         EltOp = Builder.CreateFAdd(Elt0, Elt1);
2072       else if (Name.contains(".sub."))
2073         EltOp = Builder.CreateFSub(Elt0, Elt1);
2074       else if (Name.contains(".mul."))
2075         EltOp = Builder.CreateFMul(Elt0, Elt1);
2076       else
2077         EltOp = Builder.CreateFDiv(Elt0, Elt1);
2078       Rep = Builder.CreateInsertElement(CI->getArgOperand(0), EltOp,
2079                                         ConstantInt::get(I32Ty, 0));
2080     } else if (IsX86 && Name.startswith("avx512.mask.pcmp")) {
2081       // "avx512.mask.pcmpeq." or "avx512.mask.pcmpgt."
2082       bool CmpEq = Name[16] == 'e';
2083       Rep = upgradeMaskedCompare(Builder, *CI, CmpEq ? 0 : 6, true);
2084     } else if (IsX86 && Name.startswith("avx512.mask.vpshufbitqmb.")) {
2085       Type *OpTy = CI->getArgOperand(0)->getType();
2086       unsigned VecWidth = OpTy->getPrimitiveSizeInBits();
2087       Intrinsic::ID IID;
2088       switch (VecWidth) {
2089       default: llvm_unreachable("Unexpected intrinsic");
2090       case 128: IID = Intrinsic::x86_avx512_vpshufbitqmb_128; break;
2091       case 256: IID = Intrinsic::x86_avx512_vpshufbitqmb_256; break;
2092       case 512: IID = Intrinsic::x86_avx512_vpshufbitqmb_512; break;
2093       }
2094 
2095       Rep = Builder.CreateCall(Intrinsic::getDeclaration(F->getParent(), IID),
2096                                { CI->getOperand(0), CI->getArgOperand(1) });
2097       Rep = ApplyX86MaskOn1BitsVec(Builder, Rep, CI->getArgOperand(2));
2098     } else if (IsX86 && Name.startswith("avx512.mask.fpclass.p")) {
2099       Type *OpTy = CI->getArgOperand(0)->getType();
2100       unsigned VecWidth = OpTy->getPrimitiveSizeInBits();
2101       unsigned EltWidth = OpTy->getScalarSizeInBits();
2102       Intrinsic::ID IID;
2103       if (VecWidth == 128 && EltWidth == 32)
2104         IID = Intrinsic::x86_avx512_fpclass_ps_128;
2105       else if (VecWidth == 256 && EltWidth == 32)
2106         IID = Intrinsic::x86_avx512_fpclass_ps_256;
2107       else if (VecWidth == 512 && EltWidth == 32)
2108         IID = Intrinsic::x86_avx512_fpclass_ps_512;
2109       else if (VecWidth == 128 && EltWidth == 64)
2110         IID = Intrinsic::x86_avx512_fpclass_pd_128;
2111       else if (VecWidth == 256 && EltWidth == 64)
2112         IID = Intrinsic::x86_avx512_fpclass_pd_256;
2113       else if (VecWidth == 512 && EltWidth == 64)
2114         IID = Intrinsic::x86_avx512_fpclass_pd_512;
2115       else
2116         llvm_unreachable("Unexpected intrinsic");
2117 
2118       Rep = Builder.CreateCall(Intrinsic::getDeclaration(F->getParent(), IID),
2119                                { CI->getOperand(0), CI->getArgOperand(1) });
2120       Rep = ApplyX86MaskOn1BitsVec(Builder, Rep, CI->getArgOperand(2));
2121     } else if (IsX86 && Name.startswith("avx512.cmp.p")) {
2122       SmallVector<Value *, 4> Args(CI->arg_operands().begin(),
2123                                    CI->arg_operands().end());
2124       Type *OpTy = Args[0]->getType();
2125       unsigned VecWidth = OpTy->getPrimitiveSizeInBits();
2126       unsigned EltWidth = OpTy->getScalarSizeInBits();
2127       Intrinsic::ID IID;
2128       if (VecWidth == 128 && EltWidth == 32)
2129         IID = Intrinsic::x86_avx512_mask_cmp_ps_128;
2130       else if (VecWidth == 256 && EltWidth == 32)
2131         IID = Intrinsic::x86_avx512_mask_cmp_ps_256;
2132       else if (VecWidth == 512 && EltWidth == 32)
2133         IID = Intrinsic::x86_avx512_mask_cmp_ps_512;
2134       else if (VecWidth == 128 && EltWidth == 64)
2135         IID = Intrinsic::x86_avx512_mask_cmp_pd_128;
2136       else if (VecWidth == 256 && EltWidth == 64)
2137         IID = Intrinsic::x86_avx512_mask_cmp_pd_256;
2138       else if (VecWidth == 512 && EltWidth == 64)
2139         IID = Intrinsic::x86_avx512_mask_cmp_pd_512;
2140       else
2141         llvm_unreachable("Unexpected intrinsic");
2142 
2143       Value *Mask = Constant::getAllOnesValue(CI->getType());
2144       if (VecWidth == 512)
2145         std::swap(Mask, Args.back());
2146       Args.push_back(Mask);
2147 
2148       Rep = Builder.CreateCall(Intrinsic::getDeclaration(F->getParent(), IID),
2149                                Args);
2150     } else if (IsX86 && Name.startswith("avx512.mask.cmp.")) {
2151       // Integer compare intrinsics.
2152       unsigned Imm = cast<ConstantInt>(CI->getArgOperand(2))->getZExtValue();
2153       Rep = upgradeMaskedCompare(Builder, *CI, Imm, true);
2154     } else if (IsX86 && Name.startswith("avx512.mask.ucmp.")) {
2155       unsigned Imm = cast<ConstantInt>(CI->getArgOperand(2))->getZExtValue();
2156       Rep = upgradeMaskedCompare(Builder, *CI, Imm, false);
2157     } else if (IsX86 && (Name.startswith("avx512.cvtb2mask.") ||
2158                          Name.startswith("avx512.cvtw2mask.") ||
2159                          Name.startswith("avx512.cvtd2mask.") ||
2160                          Name.startswith("avx512.cvtq2mask."))) {
2161       Value *Op = CI->getArgOperand(0);
2162       Value *Zero = llvm::Constant::getNullValue(Op->getType());
2163       Rep = Builder.CreateICmp(ICmpInst::ICMP_SLT, Op, Zero);
2164       Rep = ApplyX86MaskOn1BitsVec(Builder, Rep, nullptr);
2165     } else if(IsX86 && (Name == "ssse3.pabs.b.128" ||
2166                         Name == "ssse3.pabs.w.128" ||
2167                         Name == "ssse3.pabs.d.128" ||
2168                         Name.startswith("avx2.pabs") ||
2169                         Name.startswith("avx512.mask.pabs"))) {
2170       Rep = upgradeAbs(Builder, *CI);
2171     } else if (IsX86 && (Name == "sse41.pmaxsb" ||
2172                          Name == "sse2.pmaxs.w" ||
2173                          Name == "sse41.pmaxsd" ||
2174                          Name.startswith("avx2.pmaxs") ||
2175                          Name.startswith("avx512.mask.pmaxs"))) {
2176       Rep = UpgradeX86BinaryIntrinsics(Builder, *CI, Intrinsic::smax);
2177     } else if (IsX86 && (Name == "sse2.pmaxu.b" ||
2178                          Name == "sse41.pmaxuw" ||
2179                          Name == "sse41.pmaxud" ||
2180                          Name.startswith("avx2.pmaxu") ||
2181                          Name.startswith("avx512.mask.pmaxu"))) {
2182       Rep = UpgradeX86BinaryIntrinsics(Builder, *CI, Intrinsic::umax);
2183     } else if (IsX86 && (Name == "sse41.pminsb" ||
2184                          Name == "sse2.pmins.w" ||
2185                          Name == "sse41.pminsd" ||
2186                          Name.startswith("avx2.pmins") ||
2187                          Name.startswith("avx512.mask.pmins"))) {
2188       Rep = UpgradeX86BinaryIntrinsics(Builder, *CI, Intrinsic::smin);
2189     } else if (IsX86 && (Name == "sse2.pminu.b" ||
2190                          Name == "sse41.pminuw" ||
2191                          Name == "sse41.pminud" ||
2192                          Name.startswith("avx2.pminu") ||
2193                          Name.startswith("avx512.mask.pminu"))) {
2194       Rep = UpgradeX86BinaryIntrinsics(Builder, *CI, Intrinsic::umin);
2195     } else if (IsX86 && (Name == "sse2.pmulu.dq" ||
2196                          Name == "avx2.pmulu.dq" ||
2197                          Name == "avx512.pmulu.dq.512" ||
2198                          Name.startswith("avx512.mask.pmulu.dq."))) {
2199       Rep = upgradePMULDQ(Builder, *CI, /*Signed*/false);
2200     } else if (IsX86 && (Name == "sse41.pmuldq" ||
2201                          Name == "avx2.pmul.dq" ||
2202                          Name == "avx512.pmul.dq.512" ||
2203                          Name.startswith("avx512.mask.pmul.dq."))) {
2204       Rep = upgradePMULDQ(Builder, *CI, /*Signed*/true);
2205     } else if (IsX86 && (Name == "sse.cvtsi2ss" ||
2206                          Name == "sse2.cvtsi2sd" ||
2207                          Name == "sse.cvtsi642ss" ||
2208                          Name == "sse2.cvtsi642sd")) {
2209       Rep = Builder.CreateSIToFP(
2210           CI->getArgOperand(1),
2211           cast<VectorType>(CI->getType())->getElementType());
2212       Rep = Builder.CreateInsertElement(CI->getArgOperand(0), Rep, (uint64_t)0);
2213     } else if (IsX86 && Name == "avx512.cvtusi2sd") {
2214       Rep = Builder.CreateUIToFP(
2215           CI->getArgOperand(1),
2216           cast<VectorType>(CI->getType())->getElementType());
2217       Rep = Builder.CreateInsertElement(CI->getArgOperand(0), Rep, (uint64_t)0);
2218     } else if (IsX86 && Name == "sse2.cvtss2sd") {
2219       Rep = Builder.CreateExtractElement(CI->getArgOperand(1), (uint64_t)0);
2220       Rep = Builder.CreateFPExt(
2221           Rep, cast<VectorType>(CI->getType())->getElementType());
2222       Rep = Builder.CreateInsertElement(CI->getArgOperand(0), Rep, (uint64_t)0);
2223     } else if (IsX86 && (Name == "sse2.cvtdq2pd" ||
2224                          Name == "sse2.cvtdq2ps" ||
2225                          Name == "avx.cvtdq2.pd.256" ||
2226                          Name == "avx.cvtdq2.ps.256" ||
2227                          Name.startswith("avx512.mask.cvtdq2pd.") ||
2228                          Name.startswith("avx512.mask.cvtudq2pd.") ||
2229                          Name.startswith("avx512.mask.cvtdq2ps.") ||
2230                          Name.startswith("avx512.mask.cvtudq2ps.") ||
2231                          Name.startswith("avx512.mask.cvtqq2pd.") ||
2232                          Name.startswith("avx512.mask.cvtuqq2pd.") ||
2233                          Name == "avx512.mask.cvtqq2ps.256" ||
2234                          Name == "avx512.mask.cvtqq2ps.512" ||
2235                          Name == "avx512.mask.cvtuqq2ps.256" ||
2236                          Name == "avx512.mask.cvtuqq2ps.512" ||
2237                          Name == "sse2.cvtps2pd" ||
2238                          Name == "avx.cvt.ps2.pd.256" ||
2239                          Name == "avx512.mask.cvtps2pd.128" ||
2240                          Name == "avx512.mask.cvtps2pd.256")) {
2241       auto *DstTy = cast<FixedVectorType>(CI->getType());
2242       Rep = CI->getArgOperand(0);
2243       auto *SrcTy = cast<FixedVectorType>(Rep->getType());
2244 
2245       unsigned NumDstElts = DstTy->getNumElements();
2246       if (NumDstElts < SrcTy->getNumElements()) {
2247         assert(NumDstElts == 2 && "Unexpected vector size");
2248         Rep = Builder.CreateShuffleVector(Rep, Rep, ArrayRef<int>{0, 1});
2249       }
2250 
2251       bool IsPS2PD = SrcTy->getElementType()->isFloatTy();
2252       bool IsUnsigned = (StringRef::npos != Name.find("cvtu"));
2253       if (IsPS2PD)
2254         Rep = Builder.CreateFPExt(Rep, DstTy, "cvtps2pd");
2255       else if (CI->getNumArgOperands() == 4 &&
2256                (!isa<ConstantInt>(CI->getArgOperand(3)) ||
2257                 cast<ConstantInt>(CI->getArgOperand(3))->getZExtValue() != 4)) {
2258         Intrinsic::ID IID = IsUnsigned ? Intrinsic::x86_avx512_uitofp_round
2259                                        : Intrinsic::x86_avx512_sitofp_round;
2260         Function *F = Intrinsic::getDeclaration(CI->getModule(), IID,
2261                                                 { DstTy, SrcTy });
2262         Rep = Builder.CreateCall(F, { Rep, CI->getArgOperand(3) });
2263       } else {
2264         Rep = IsUnsigned ? Builder.CreateUIToFP(Rep, DstTy, "cvt")
2265                          : Builder.CreateSIToFP(Rep, DstTy, "cvt");
2266       }
2267 
2268       if (CI->getNumArgOperands() >= 3)
2269         Rep = EmitX86Select(Builder, CI->getArgOperand(2), Rep,
2270                             CI->getArgOperand(1));
2271     } else if (IsX86 && (Name.startswith("avx512.mask.vcvtph2ps.") ||
2272                          Name.startswith("vcvtph2ps."))) {
2273       auto *DstTy = cast<FixedVectorType>(CI->getType());
2274       Rep = CI->getArgOperand(0);
2275       auto *SrcTy = cast<FixedVectorType>(Rep->getType());
2276       unsigned NumDstElts = DstTy->getNumElements();
2277       if (NumDstElts != SrcTy->getNumElements()) {
2278         assert(NumDstElts == 4 && "Unexpected vector size");
2279         Rep = Builder.CreateShuffleVector(Rep, Rep, ArrayRef<int>{0, 1, 2, 3});
2280       }
2281       Rep = Builder.CreateBitCast(
2282           Rep, FixedVectorType::get(Type::getHalfTy(C), NumDstElts));
2283       Rep = Builder.CreateFPExt(Rep, DstTy, "cvtph2ps");
2284       if (CI->getNumArgOperands() >= 3)
2285         Rep = EmitX86Select(Builder, CI->getArgOperand(2), Rep,
2286                             CI->getArgOperand(1));
2287     } else if (IsX86 && (Name.startswith("avx512.mask.loadu."))) {
2288       Rep = UpgradeMaskedLoad(Builder, CI->getArgOperand(0),
2289                               CI->getArgOperand(1), CI->getArgOperand(2),
2290                               /*Aligned*/false);
2291     } else if (IsX86 && (Name.startswith("avx512.mask.load."))) {
2292       Rep = UpgradeMaskedLoad(Builder, CI->getArgOperand(0),
2293                               CI->getArgOperand(1),CI->getArgOperand(2),
2294                               /*Aligned*/true);
2295     } else if (IsX86 && Name.startswith("avx512.mask.expand.load.")) {
2296       auto *ResultTy = cast<FixedVectorType>(CI->getType());
2297       Type *PtrTy = ResultTy->getElementType();
2298 
2299       // Cast the pointer to element type.
2300       Value *Ptr = Builder.CreateBitCast(CI->getOperand(0),
2301                                          llvm::PointerType::getUnqual(PtrTy));
2302 
2303       Value *MaskVec = getX86MaskVec(Builder, CI->getArgOperand(2),
2304                                      ResultTy->getNumElements());
2305 
2306       Function *ELd = Intrinsic::getDeclaration(F->getParent(),
2307                                                 Intrinsic::masked_expandload,
2308                                                 ResultTy);
2309       Rep = Builder.CreateCall(ELd, { Ptr, MaskVec, CI->getOperand(1) });
2310     } else if (IsX86 && Name.startswith("avx512.mask.compress.store.")) {
2311       auto *ResultTy = cast<VectorType>(CI->getArgOperand(1)->getType());
2312       Type *PtrTy = ResultTy->getElementType();
2313 
2314       // Cast the pointer to element type.
2315       Value *Ptr = Builder.CreateBitCast(CI->getOperand(0),
2316                                          llvm::PointerType::getUnqual(PtrTy));
2317 
2318       Value *MaskVec =
2319           getX86MaskVec(Builder, CI->getArgOperand(2),
2320                         cast<FixedVectorType>(ResultTy)->getNumElements());
2321 
2322       Function *CSt = Intrinsic::getDeclaration(F->getParent(),
2323                                                 Intrinsic::masked_compressstore,
2324                                                 ResultTy);
2325       Rep = Builder.CreateCall(CSt, { CI->getArgOperand(1), Ptr, MaskVec });
2326     } else if (IsX86 && (Name.startswith("avx512.mask.compress.") ||
2327                          Name.startswith("avx512.mask.expand."))) {
2328       auto *ResultTy = cast<FixedVectorType>(CI->getType());
2329 
2330       Value *MaskVec = getX86MaskVec(Builder, CI->getArgOperand(2),
2331                                      ResultTy->getNumElements());
2332 
2333       bool IsCompress = Name[12] == 'c';
2334       Intrinsic::ID IID = IsCompress ? Intrinsic::x86_avx512_mask_compress
2335                                      : Intrinsic::x86_avx512_mask_expand;
2336       Function *Intr = Intrinsic::getDeclaration(F->getParent(), IID, ResultTy);
2337       Rep = Builder.CreateCall(Intr, { CI->getOperand(0), CI->getOperand(1),
2338                                        MaskVec });
2339     } else if (IsX86 && Name.startswith("xop.vpcom")) {
2340       bool IsSigned;
2341       if (Name.endswith("ub") || Name.endswith("uw") || Name.endswith("ud") ||
2342           Name.endswith("uq"))
2343         IsSigned = false;
2344       else if (Name.endswith("b") || Name.endswith("w") || Name.endswith("d") ||
2345                Name.endswith("q"))
2346         IsSigned = true;
2347       else
2348         llvm_unreachable("Unknown suffix");
2349 
2350       unsigned Imm;
2351       if (CI->getNumArgOperands() == 3) {
2352         Imm = cast<ConstantInt>(CI->getArgOperand(2))->getZExtValue();
2353       } else {
2354         Name = Name.substr(9); // strip off "xop.vpcom"
2355         if (Name.startswith("lt"))
2356           Imm = 0;
2357         else if (Name.startswith("le"))
2358           Imm = 1;
2359         else if (Name.startswith("gt"))
2360           Imm = 2;
2361         else if (Name.startswith("ge"))
2362           Imm = 3;
2363         else if (Name.startswith("eq"))
2364           Imm = 4;
2365         else if (Name.startswith("ne"))
2366           Imm = 5;
2367         else if (Name.startswith("false"))
2368           Imm = 6;
2369         else if (Name.startswith("true"))
2370           Imm = 7;
2371         else
2372           llvm_unreachable("Unknown condition");
2373       }
2374 
2375       Rep = upgradeX86vpcom(Builder, *CI, Imm, IsSigned);
2376     } else if (IsX86 && Name.startswith("xop.vpcmov")) {
2377       Value *Sel = CI->getArgOperand(2);
2378       Value *NotSel = Builder.CreateNot(Sel);
2379       Value *Sel0 = Builder.CreateAnd(CI->getArgOperand(0), Sel);
2380       Value *Sel1 = Builder.CreateAnd(CI->getArgOperand(1), NotSel);
2381       Rep = Builder.CreateOr(Sel0, Sel1);
2382     } else if (IsX86 && (Name.startswith("xop.vprot") ||
2383                          Name.startswith("avx512.prol") ||
2384                          Name.startswith("avx512.mask.prol"))) {
2385       Rep = upgradeX86Rotate(Builder, *CI, false);
2386     } else if (IsX86 && (Name.startswith("avx512.pror") ||
2387                          Name.startswith("avx512.mask.pror"))) {
2388       Rep = upgradeX86Rotate(Builder, *CI, true);
2389     } else if (IsX86 && (Name.startswith("avx512.vpshld.") ||
2390                          Name.startswith("avx512.mask.vpshld") ||
2391                          Name.startswith("avx512.maskz.vpshld"))) {
2392       bool ZeroMask = Name[11] == 'z';
2393       Rep = upgradeX86ConcatShift(Builder, *CI, false, ZeroMask);
2394     } else if (IsX86 && (Name.startswith("avx512.vpshrd.") ||
2395                          Name.startswith("avx512.mask.vpshrd") ||
2396                          Name.startswith("avx512.maskz.vpshrd"))) {
2397       bool ZeroMask = Name[11] == 'z';
2398       Rep = upgradeX86ConcatShift(Builder, *CI, true, ZeroMask);
2399     } else if (IsX86 && Name == "sse42.crc32.64.8") {
2400       Function *CRC32 = Intrinsic::getDeclaration(F->getParent(),
2401                                                Intrinsic::x86_sse42_crc32_32_8);
2402       Value *Trunc0 = Builder.CreateTrunc(CI->getArgOperand(0), Type::getInt32Ty(C));
2403       Rep = Builder.CreateCall(CRC32, {Trunc0, CI->getArgOperand(1)});
2404       Rep = Builder.CreateZExt(Rep, CI->getType(), "");
2405     } else if (IsX86 && (Name.startswith("avx.vbroadcast.s") ||
2406                          Name.startswith("avx512.vbroadcast.s"))) {
2407       // Replace broadcasts with a series of insertelements.
2408       auto *VecTy = cast<FixedVectorType>(CI->getType());
2409       Type *EltTy = VecTy->getElementType();
2410       unsigned EltNum = VecTy->getNumElements();
2411       Value *Cast = Builder.CreateBitCast(CI->getArgOperand(0),
2412                                           EltTy->getPointerTo());
2413       Value *Load = Builder.CreateLoad(EltTy, Cast);
2414       Type *I32Ty = Type::getInt32Ty(C);
2415       Rep = UndefValue::get(VecTy);
2416       for (unsigned I = 0; I < EltNum; ++I)
2417         Rep = Builder.CreateInsertElement(Rep, Load,
2418                                           ConstantInt::get(I32Ty, I));
2419     } else if (IsX86 && (Name.startswith("sse41.pmovsx") ||
2420                          Name.startswith("sse41.pmovzx") ||
2421                          Name.startswith("avx2.pmovsx") ||
2422                          Name.startswith("avx2.pmovzx") ||
2423                          Name.startswith("avx512.mask.pmovsx") ||
2424                          Name.startswith("avx512.mask.pmovzx"))) {
2425       auto *DstTy = cast<FixedVectorType>(CI->getType());
2426       unsigned NumDstElts = DstTy->getNumElements();
2427 
2428       // Extract a subvector of the first NumDstElts lanes and sign/zero extend.
2429       SmallVector<int, 8> ShuffleMask(NumDstElts);
2430       for (unsigned i = 0; i != NumDstElts; ++i)
2431         ShuffleMask[i] = i;
2432 
2433       Value *SV =
2434           Builder.CreateShuffleVector(CI->getArgOperand(0), ShuffleMask);
2435 
2436       bool DoSext = (StringRef::npos != Name.find("pmovsx"));
2437       Rep = DoSext ? Builder.CreateSExt(SV, DstTy)
2438                    : Builder.CreateZExt(SV, DstTy);
2439       // If there are 3 arguments, it's a masked intrinsic so we need a select.
2440       if (CI->getNumArgOperands() == 3)
2441         Rep = EmitX86Select(Builder, CI->getArgOperand(2), Rep,
2442                             CI->getArgOperand(1));
2443     } else if (Name == "avx512.mask.pmov.qd.256" ||
2444                Name == "avx512.mask.pmov.qd.512" ||
2445                Name == "avx512.mask.pmov.wb.256" ||
2446                Name == "avx512.mask.pmov.wb.512") {
2447       Type *Ty = CI->getArgOperand(1)->getType();
2448       Rep = Builder.CreateTrunc(CI->getArgOperand(0), Ty);
2449       Rep = EmitX86Select(Builder, CI->getArgOperand(2), Rep,
2450                           CI->getArgOperand(1));
2451     } else if (IsX86 && (Name.startswith("avx.vbroadcastf128") ||
2452                          Name == "avx2.vbroadcasti128")) {
2453       // Replace vbroadcastf128/vbroadcasti128 with a vector load+shuffle.
2454       Type *EltTy = cast<VectorType>(CI->getType())->getElementType();
2455       unsigned NumSrcElts = 128 / EltTy->getPrimitiveSizeInBits();
2456       auto *VT = FixedVectorType::get(EltTy, NumSrcElts);
2457       Value *Op = Builder.CreatePointerCast(CI->getArgOperand(0),
2458                                             PointerType::getUnqual(VT));
2459       Value *Load = Builder.CreateAlignedLoad(VT, Op, Align(1));
2460       if (NumSrcElts == 2)
2461         Rep = Builder.CreateShuffleVector(Load, ArrayRef<int>{0, 1, 0, 1});
2462       else
2463         Rep = Builder.CreateShuffleVector(
2464             Load, ArrayRef<int>{0, 1, 2, 3, 0, 1, 2, 3});
2465     } else if (IsX86 && (Name.startswith("avx512.mask.shuf.i") ||
2466                          Name.startswith("avx512.mask.shuf.f"))) {
2467       unsigned Imm = cast<ConstantInt>(CI->getArgOperand(2))->getZExtValue();
2468       Type *VT = CI->getType();
2469       unsigned NumLanes = VT->getPrimitiveSizeInBits() / 128;
2470       unsigned NumElementsInLane = 128 / VT->getScalarSizeInBits();
2471       unsigned ControlBitsMask = NumLanes - 1;
2472       unsigned NumControlBits = NumLanes / 2;
2473       SmallVector<int, 8> ShuffleMask(0);
2474 
2475       for (unsigned l = 0; l != NumLanes; ++l) {
2476         unsigned LaneMask = (Imm >> (l * NumControlBits)) & ControlBitsMask;
2477         // We actually need the other source.
2478         if (l >= NumLanes / 2)
2479           LaneMask += NumLanes;
2480         for (unsigned i = 0; i != NumElementsInLane; ++i)
2481           ShuffleMask.push_back(LaneMask * NumElementsInLane + i);
2482       }
2483       Rep = Builder.CreateShuffleVector(CI->getArgOperand(0),
2484                                         CI->getArgOperand(1), ShuffleMask);
2485       Rep = EmitX86Select(Builder, CI->getArgOperand(4), Rep,
2486                           CI->getArgOperand(3));
2487     }else if (IsX86 && (Name.startswith("avx512.mask.broadcastf") ||
2488                          Name.startswith("avx512.mask.broadcasti"))) {
2489       unsigned NumSrcElts =
2490           cast<FixedVectorType>(CI->getArgOperand(0)->getType())
2491               ->getNumElements();
2492       unsigned NumDstElts =
2493           cast<FixedVectorType>(CI->getType())->getNumElements();
2494 
2495       SmallVector<int, 8> ShuffleMask(NumDstElts);
2496       for (unsigned i = 0; i != NumDstElts; ++i)
2497         ShuffleMask[i] = i % NumSrcElts;
2498 
2499       Rep = Builder.CreateShuffleVector(CI->getArgOperand(0),
2500                                         CI->getArgOperand(0),
2501                                         ShuffleMask);
2502       Rep = EmitX86Select(Builder, CI->getArgOperand(2), Rep,
2503                           CI->getArgOperand(1));
2504     } else if (IsX86 && (Name.startswith("avx2.pbroadcast") ||
2505                          Name.startswith("avx2.vbroadcast") ||
2506                          Name.startswith("avx512.pbroadcast") ||
2507                          Name.startswith("avx512.mask.broadcast.s"))) {
2508       // Replace vp?broadcasts with a vector shuffle.
2509       Value *Op = CI->getArgOperand(0);
2510       ElementCount EC = cast<VectorType>(CI->getType())->getElementCount();
2511       Type *MaskTy = VectorType::get(Type::getInt32Ty(C), EC);
2512       SmallVector<int, 8> M;
2513       ShuffleVectorInst::getShuffleMask(Constant::getNullValue(MaskTy), M);
2514       Rep = Builder.CreateShuffleVector(Op, M);
2515 
2516       if (CI->getNumArgOperands() == 3)
2517         Rep = EmitX86Select(Builder, CI->getArgOperand(2), Rep,
2518                             CI->getArgOperand(1));
2519     } else if (IsX86 && (Name.startswith("sse2.padds.") ||
2520                          Name.startswith("avx2.padds.") ||
2521                          Name.startswith("avx512.padds.") ||
2522                          Name.startswith("avx512.mask.padds."))) {
2523       Rep = UpgradeX86BinaryIntrinsics(Builder, *CI, Intrinsic::sadd_sat);
2524     } else if (IsX86 && (Name.startswith("sse2.psubs.") ||
2525                          Name.startswith("avx2.psubs.") ||
2526                          Name.startswith("avx512.psubs.") ||
2527                          Name.startswith("avx512.mask.psubs."))) {
2528       Rep = UpgradeX86BinaryIntrinsics(Builder, *CI, Intrinsic::ssub_sat);
2529     } else if (IsX86 && (Name.startswith("sse2.paddus.") ||
2530                          Name.startswith("avx2.paddus.") ||
2531                          Name.startswith("avx512.mask.paddus."))) {
2532       Rep = UpgradeX86BinaryIntrinsics(Builder, *CI, Intrinsic::uadd_sat);
2533     } else if (IsX86 && (Name.startswith("sse2.psubus.") ||
2534                          Name.startswith("avx2.psubus.") ||
2535                          Name.startswith("avx512.mask.psubus."))) {
2536       Rep = UpgradeX86BinaryIntrinsics(Builder, *CI, Intrinsic::usub_sat);
2537     } else if (IsX86 && Name.startswith("avx512.mask.palignr.")) {
2538       Rep = UpgradeX86ALIGNIntrinsics(Builder, CI->getArgOperand(0),
2539                                       CI->getArgOperand(1),
2540                                       CI->getArgOperand(2),
2541                                       CI->getArgOperand(3),
2542                                       CI->getArgOperand(4),
2543                                       false);
2544     } else if (IsX86 && Name.startswith("avx512.mask.valign.")) {
2545       Rep = UpgradeX86ALIGNIntrinsics(Builder, CI->getArgOperand(0),
2546                                       CI->getArgOperand(1),
2547                                       CI->getArgOperand(2),
2548                                       CI->getArgOperand(3),
2549                                       CI->getArgOperand(4),
2550                                       true);
2551     } else if (IsX86 && (Name == "sse2.psll.dq" ||
2552                          Name == "avx2.psll.dq")) {
2553       // 128/256-bit shift left specified in bits.
2554       unsigned Shift = cast<ConstantInt>(CI->getArgOperand(1))->getZExtValue();
2555       Rep = UpgradeX86PSLLDQIntrinsics(Builder, CI->getArgOperand(0),
2556                                        Shift / 8); // Shift is in bits.
2557     } else if (IsX86 && (Name == "sse2.psrl.dq" ||
2558                          Name == "avx2.psrl.dq")) {
2559       // 128/256-bit shift right specified in bits.
2560       unsigned Shift = cast<ConstantInt>(CI->getArgOperand(1))->getZExtValue();
2561       Rep = UpgradeX86PSRLDQIntrinsics(Builder, CI->getArgOperand(0),
2562                                        Shift / 8); // Shift is in bits.
2563     } else if (IsX86 && (Name == "sse2.psll.dq.bs" ||
2564                          Name == "avx2.psll.dq.bs" ||
2565                          Name == "avx512.psll.dq.512")) {
2566       // 128/256/512-bit shift left specified in bytes.
2567       unsigned Shift = cast<ConstantInt>(CI->getArgOperand(1))->getZExtValue();
2568       Rep = UpgradeX86PSLLDQIntrinsics(Builder, CI->getArgOperand(0), Shift);
2569     } else if (IsX86 && (Name == "sse2.psrl.dq.bs" ||
2570                          Name == "avx2.psrl.dq.bs" ||
2571                          Name == "avx512.psrl.dq.512")) {
2572       // 128/256/512-bit shift right specified in bytes.
2573       unsigned Shift = cast<ConstantInt>(CI->getArgOperand(1))->getZExtValue();
2574       Rep = UpgradeX86PSRLDQIntrinsics(Builder, CI->getArgOperand(0), Shift);
2575     } else if (IsX86 && (Name == "sse41.pblendw" ||
2576                          Name.startswith("sse41.blendp") ||
2577                          Name.startswith("avx.blend.p") ||
2578                          Name == "avx2.pblendw" ||
2579                          Name.startswith("avx2.pblendd."))) {
2580       Value *Op0 = CI->getArgOperand(0);
2581       Value *Op1 = CI->getArgOperand(1);
2582       unsigned Imm = cast <ConstantInt>(CI->getArgOperand(2))->getZExtValue();
2583       auto *VecTy = cast<FixedVectorType>(CI->getType());
2584       unsigned NumElts = VecTy->getNumElements();
2585 
2586       SmallVector<int, 16> Idxs(NumElts);
2587       for (unsigned i = 0; i != NumElts; ++i)
2588         Idxs[i] = ((Imm >> (i%8)) & 1) ? i + NumElts : i;
2589 
2590       Rep = Builder.CreateShuffleVector(Op0, Op1, Idxs);
2591     } else if (IsX86 && (Name.startswith("avx.vinsertf128.") ||
2592                          Name == "avx2.vinserti128" ||
2593                          Name.startswith("avx512.mask.insert"))) {
2594       Value *Op0 = CI->getArgOperand(0);
2595       Value *Op1 = CI->getArgOperand(1);
2596       unsigned Imm = cast<ConstantInt>(CI->getArgOperand(2))->getZExtValue();
2597       unsigned DstNumElts =
2598           cast<FixedVectorType>(CI->getType())->getNumElements();
2599       unsigned SrcNumElts =
2600           cast<FixedVectorType>(Op1->getType())->getNumElements();
2601       unsigned Scale = DstNumElts / SrcNumElts;
2602 
2603       // Mask off the high bits of the immediate value; hardware ignores those.
2604       Imm = Imm % Scale;
2605 
2606       // Extend the second operand into a vector the size of the destination.
2607       SmallVector<int, 8> Idxs(DstNumElts);
2608       for (unsigned i = 0; i != SrcNumElts; ++i)
2609         Idxs[i] = i;
2610       for (unsigned i = SrcNumElts; i != DstNumElts; ++i)
2611         Idxs[i] = SrcNumElts;
2612       Rep = Builder.CreateShuffleVector(Op1, Idxs);
2613 
2614       // Insert the second operand into the first operand.
2615 
2616       // Note that there is no guarantee that instruction lowering will actually
2617       // produce a vinsertf128 instruction for the created shuffles. In
2618       // particular, the 0 immediate case involves no lane changes, so it can
2619       // be handled as a blend.
2620 
2621       // Example of shuffle mask for 32-bit elements:
2622       // Imm = 1  <i32 0, i32 1, i32 2,  i32 3,  i32 8, i32 9, i32 10, i32 11>
2623       // Imm = 0  <i32 8, i32 9, i32 10, i32 11, i32 4, i32 5, i32 6,  i32 7 >
2624 
2625       // First fill with identify mask.
2626       for (unsigned i = 0; i != DstNumElts; ++i)
2627         Idxs[i] = i;
2628       // Then replace the elements where we need to insert.
2629       for (unsigned i = 0; i != SrcNumElts; ++i)
2630         Idxs[i + Imm * SrcNumElts] = i + DstNumElts;
2631       Rep = Builder.CreateShuffleVector(Op0, Rep, Idxs);
2632 
2633       // If the intrinsic has a mask operand, handle that.
2634       if (CI->getNumArgOperands() == 5)
2635         Rep = EmitX86Select(Builder, CI->getArgOperand(4), Rep,
2636                             CI->getArgOperand(3));
2637     } else if (IsX86 && (Name.startswith("avx.vextractf128.") ||
2638                          Name == "avx2.vextracti128" ||
2639                          Name.startswith("avx512.mask.vextract"))) {
2640       Value *Op0 = CI->getArgOperand(0);
2641       unsigned Imm = cast<ConstantInt>(CI->getArgOperand(1))->getZExtValue();
2642       unsigned DstNumElts =
2643           cast<FixedVectorType>(CI->getType())->getNumElements();
2644       unsigned SrcNumElts =
2645           cast<FixedVectorType>(Op0->getType())->getNumElements();
2646       unsigned Scale = SrcNumElts / DstNumElts;
2647 
2648       // Mask off the high bits of the immediate value; hardware ignores those.
2649       Imm = Imm % Scale;
2650 
2651       // Get indexes for the subvector of the input vector.
2652       SmallVector<int, 8> Idxs(DstNumElts);
2653       for (unsigned i = 0; i != DstNumElts; ++i) {
2654         Idxs[i] = i + (Imm * DstNumElts);
2655       }
2656       Rep = Builder.CreateShuffleVector(Op0, Op0, Idxs);
2657 
2658       // If the intrinsic has a mask operand, handle that.
2659       if (CI->getNumArgOperands() == 4)
2660         Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep,
2661                             CI->getArgOperand(2));
2662     } else if (!IsX86 && Name == "stackprotectorcheck") {
2663       Rep = nullptr;
2664     } else if (IsX86 && (Name.startswith("avx512.mask.perm.df.") ||
2665                          Name.startswith("avx512.mask.perm.di."))) {
2666       Value *Op0 = CI->getArgOperand(0);
2667       unsigned Imm = cast<ConstantInt>(CI->getArgOperand(1))->getZExtValue();
2668       auto *VecTy = cast<FixedVectorType>(CI->getType());
2669       unsigned NumElts = VecTy->getNumElements();
2670 
2671       SmallVector<int, 8> Idxs(NumElts);
2672       for (unsigned i = 0; i != NumElts; ++i)
2673         Idxs[i] = (i & ~0x3) + ((Imm >> (2 * (i & 0x3))) & 3);
2674 
2675       Rep = Builder.CreateShuffleVector(Op0, Op0, Idxs);
2676 
2677       if (CI->getNumArgOperands() == 4)
2678         Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep,
2679                             CI->getArgOperand(2));
2680     } else if (IsX86 && (Name.startswith("avx.vperm2f128.") ||
2681                          Name == "avx2.vperm2i128")) {
2682       // The immediate permute control byte looks like this:
2683       //    [1:0] - select 128 bits from sources for low half of destination
2684       //    [2]   - ignore
2685       //    [3]   - zero low half of destination
2686       //    [5:4] - select 128 bits from sources for high half of destination
2687       //    [6]   - ignore
2688       //    [7]   - zero high half of destination
2689 
2690       uint8_t Imm = cast<ConstantInt>(CI->getArgOperand(2))->getZExtValue();
2691 
2692       unsigned NumElts = cast<FixedVectorType>(CI->getType())->getNumElements();
2693       unsigned HalfSize = NumElts / 2;
2694       SmallVector<int, 8> ShuffleMask(NumElts);
2695 
2696       // Determine which operand(s) are actually in use for this instruction.
2697       Value *V0 = (Imm & 0x02) ? CI->getArgOperand(1) : CI->getArgOperand(0);
2698       Value *V1 = (Imm & 0x20) ? CI->getArgOperand(1) : CI->getArgOperand(0);
2699 
2700       // If needed, replace operands based on zero mask.
2701       V0 = (Imm & 0x08) ? ConstantAggregateZero::get(CI->getType()) : V0;
2702       V1 = (Imm & 0x80) ? ConstantAggregateZero::get(CI->getType()) : V1;
2703 
2704       // Permute low half of result.
2705       unsigned StartIndex = (Imm & 0x01) ? HalfSize : 0;
2706       for (unsigned i = 0; i < HalfSize; ++i)
2707         ShuffleMask[i] = StartIndex + i;
2708 
2709       // Permute high half of result.
2710       StartIndex = (Imm & 0x10) ? HalfSize : 0;
2711       for (unsigned i = 0; i < HalfSize; ++i)
2712         ShuffleMask[i + HalfSize] = NumElts + StartIndex + i;
2713 
2714       Rep = Builder.CreateShuffleVector(V0, V1, ShuffleMask);
2715 
2716     } else if (IsX86 && (Name.startswith("avx.vpermil.") ||
2717                          Name == "sse2.pshuf.d" ||
2718                          Name.startswith("avx512.mask.vpermil.p") ||
2719                          Name.startswith("avx512.mask.pshuf.d."))) {
2720       Value *Op0 = CI->getArgOperand(0);
2721       unsigned Imm = cast<ConstantInt>(CI->getArgOperand(1))->getZExtValue();
2722       auto *VecTy = cast<FixedVectorType>(CI->getType());
2723       unsigned NumElts = VecTy->getNumElements();
2724       // Calculate the size of each index in the immediate.
2725       unsigned IdxSize = 64 / VecTy->getScalarSizeInBits();
2726       unsigned IdxMask = ((1 << IdxSize) - 1);
2727 
2728       SmallVector<int, 8> Idxs(NumElts);
2729       // Lookup the bits for this element, wrapping around the immediate every
2730       // 8-bits. Elements are grouped into sets of 2 or 4 elements so we need
2731       // to offset by the first index of each group.
2732       for (unsigned i = 0; i != NumElts; ++i)
2733         Idxs[i] = ((Imm >> ((i * IdxSize) % 8)) & IdxMask) | (i & ~IdxMask);
2734 
2735       Rep = Builder.CreateShuffleVector(Op0, Op0, Idxs);
2736 
2737       if (CI->getNumArgOperands() == 4)
2738         Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep,
2739                             CI->getArgOperand(2));
2740     } else if (IsX86 && (Name == "sse2.pshufl.w" ||
2741                          Name.startswith("avx512.mask.pshufl.w."))) {
2742       Value *Op0 = CI->getArgOperand(0);
2743       unsigned Imm = cast<ConstantInt>(CI->getArgOperand(1))->getZExtValue();
2744       unsigned NumElts = cast<FixedVectorType>(CI->getType())->getNumElements();
2745 
2746       SmallVector<int, 16> Idxs(NumElts);
2747       for (unsigned l = 0; l != NumElts; l += 8) {
2748         for (unsigned i = 0; i != 4; ++i)
2749           Idxs[i + l] = ((Imm >> (2 * i)) & 0x3) + l;
2750         for (unsigned i = 4; i != 8; ++i)
2751           Idxs[i + l] = i + l;
2752       }
2753 
2754       Rep = Builder.CreateShuffleVector(Op0, Op0, Idxs);
2755 
2756       if (CI->getNumArgOperands() == 4)
2757         Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep,
2758                             CI->getArgOperand(2));
2759     } else if (IsX86 && (Name == "sse2.pshufh.w" ||
2760                          Name.startswith("avx512.mask.pshufh.w."))) {
2761       Value *Op0 = CI->getArgOperand(0);
2762       unsigned Imm = cast<ConstantInt>(CI->getArgOperand(1))->getZExtValue();
2763       unsigned NumElts = cast<FixedVectorType>(CI->getType())->getNumElements();
2764 
2765       SmallVector<int, 16> Idxs(NumElts);
2766       for (unsigned l = 0; l != NumElts; l += 8) {
2767         for (unsigned i = 0; i != 4; ++i)
2768           Idxs[i + l] = i + l;
2769         for (unsigned i = 0; i != 4; ++i)
2770           Idxs[i + l + 4] = ((Imm >> (2 * i)) & 0x3) + 4 + l;
2771       }
2772 
2773       Rep = Builder.CreateShuffleVector(Op0, Op0, Idxs);
2774 
2775       if (CI->getNumArgOperands() == 4)
2776         Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep,
2777                             CI->getArgOperand(2));
2778     } else if (IsX86 && Name.startswith("avx512.mask.shuf.p")) {
2779       Value *Op0 = CI->getArgOperand(0);
2780       Value *Op1 = CI->getArgOperand(1);
2781       unsigned Imm = cast<ConstantInt>(CI->getArgOperand(2))->getZExtValue();
2782       unsigned NumElts = cast<FixedVectorType>(CI->getType())->getNumElements();
2783 
2784       unsigned NumLaneElts = 128/CI->getType()->getScalarSizeInBits();
2785       unsigned HalfLaneElts = NumLaneElts / 2;
2786 
2787       SmallVector<int, 16> Idxs(NumElts);
2788       for (unsigned i = 0; i != NumElts; ++i) {
2789         // Base index is the starting element of the lane.
2790         Idxs[i] = i - (i % NumLaneElts);
2791         // If we are half way through the lane switch to the other source.
2792         if ((i % NumLaneElts) >= HalfLaneElts)
2793           Idxs[i] += NumElts;
2794         // Now select the specific element. By adding HalfLaneElts bits from
2795         // the immediate. Wrapping around the immediate every 8-bits.
2796         Idxs[i] += (Imm >> ((i * HalfLaneElts) % 8)) & ((1 << HalfLaneElts) - 1);
2797       }
2798 
2799       Rep = Builder.CreateShuffleVector(Op0, Op1, Idxs);
2800 
2801       Rep = EmitX86Select(Builder, CI->getArgOperand(4), Rep,
2802                           CI->getArgOperand(3));
2803     } else if (IsX86 && (Name.startswith("avx512.mask.movddup") ||
2804                          Name.startswith("avx512.mask.movshdup") ||
2805                          Name.startswith("avx512.mask.movsldup"))) {
2806       Value *Op0 = CI->getArgOperand(0);
2807       unsigned NumElts = cast<FixedVectorType>(CI->getType())->getNumElements();
2808       unsigned NumLaneElts = 128/CI->getType()->getScalarSizeInBits();
2809 
2810       unsigned Offset = 0;
2811       if (Name.startswith("avx512.mask.movshdup."))
2812         Offset = 1;
2813 
2814       SmallVector<int, 16> Idxs(NumElts);
2815       for (unsigned l = 0; l != NumElts; l += NumLaneElts)
2816         for (unsigned i = 0; i != NumLaneElts; i += 2) {
2817           Idxs[i + l + 0] = i + l + Offset;
2818           Idxs[i + l + 1] = i + l + Offset;
2819         }
2820 
2821       Rep = Builder.CreateShuffleVector(Op0, Op0, Idxs);
2822 
2823       Rep = EmitX86Select(Builder, CI->getArgOperand(2), Rep,
2824                           CI->getArgOperand(1));
2825     } else if (IsX86 && (Name.startswith("avx512.mask.punpckl") ||
2826                          Name.startswith("avx512.mask.unpckl."))) {
2827       Value *Op0 = CI->getArgOperand(0);
2828       Value *Op1 = CI->getArgOperand(1);
2829       int NumElts = cast<FixedVectorType>(CI->getType())->getNumElements();
2830       int NumLaneElts = 128/CI->getType()->getScalarSizeInBits();
2831 
2832       SmallVector<int, 64> Idxs(NumElts);
2833       for (int l = 0; l != NumElts; l += NumLaneElts)
2834         for (int i = 0; i != NumLaneElts; ++i)
2835           Idxs[i + l] = l + (i / 2) + NumElts * (i % 2);
2836 
2837       Rep = Builder.CreateShuffleVector(Op0, Op1, Idxs);
2838 
2839       Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep,
2840                           CI->getArgOperand(2));
2841     } else if (IsX86 && (Name.startswith("avx512.mask.punpckh") ||
2842                          Name.startswith("avx512.mask.unpckh."))) {
2843       Value *Op0 = CI->getArgOperand(0);
2844       Value *Op1 = CI->getArgOperand(1);
2845       int NumElts = cast<FixedVectorType>(CI->getType())->getNumElements();
2846       int NumLaneElts = 128/CI->getType()->getScalarSizeInBits();
2847 
2848       SmallVector<int, 64> Idxs(NumElts);
2849       for (int l = 0; l != NumElts; l += NumLaneElts)
2850         for (int i = 0; i != NumLaneElts; ++i)
2851           Idxs[i + l] = (NumLaneElts / 2) + l + (i / 2) + NumElts * (i % 2);
2852 
2853       Rep = Builder.CreateShuffleVector(Op0, Op1, Idxs);
2854 
2855       Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep,
2856                           CI->getArgOperand(2));
2857     } else if (IsX86 && (Name.startswith("avx512.mask.and.") ||
2858                          Name.startswith("avx512.mask.pand."))) {
2859       VectorType *FTy = cast<VectorType>(CI->getType());
2860       VectorType *ITy = VectorType::getInteger(FTy);
2861       Rep = Builder.CreateAnd(Builder.CreateBitCast(CI->getArgOperand(0), ITy),
2862                               Builder.CreateBitCast(CI->getArgOperand(1), ITy));
2863       Rep = Builder.CreateBitCast(Rep, FTy);
2864       Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep,
2865                           CI->getArgOperand(2));
2866     } else if (IsX86 && (Name.startswith("avx512.mask.andn.") ||
2867                          Name.startswith("avx512.mask.pandn."))) {
2868       VectorType *FTy = cast<VectorType>(CI->getType());
2869       VectorType *ITy = VectorType::getInteger(FTy);
2870       Rep = Builder.CreateNot(Builder.CreateBitCast(CI->getArgOperand(0), ITy));
2871       Rep = Builder.CreateAnd(Rep,
2872                               Builder.CreateBitCast(CI->getArgOperand(1), ITy));
2873       Rep = Builder.CreateBitCast(Rep, FTy);
2874       Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep,
2875                           CI->getArgOperand(2));
2876     } else if (IsX86 && (Name.startswith("avx512.mask.or.") ||
2877                          Name.startswith("avx512.mask.por."))) {
2878       VectorType *FTy = cast<VectorType>(CI->getType());
2879       VectorType *ITy = VectorType::getInteger(FTy);
2880       Rep = Builder.CreateOr(Builder.CreateBitCast(CI->getArgOperand(0), ITy),
2881                              Builder.CreateBitCast(CI->getArgOperand(1), ITy));
2882       Rep = Builder.CreateBitCast(Rep, FTy);
2883       Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep,
2884                           CI->getArgOperand(2));
2885     } else if (IsX86 && (Name.startswith("avx512.mask.xor.") ||
2886                          Name.startswith("avx512.mask.pxor."))) {
2887       VectorType *FTy = cast<VectorType>(CI->getType());
2888       VectorType *ITy = VectorType::getInteger(FTy);
2889       Rep = Builder.CreateXor(Builder.CreateBitCast(CI->getArgOperand(0), ITy),
2890                               Builder.CreateBitCast(CI->getArgOperand(1), ITy));
2891       Rep = Builder.CreateBitCast(Rep, FTy);
2892       Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep,
2893                           CI->getArgOperand(2));
2894     } else if (IsX86 && Name.startswith("avx512.mask.padd.")) {
2895       Rep = Builder.CreateAdd(CI->getArgOperand(0), CI->getArgOperand(1));
2896       Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep,
2897                           CI->getArgOperand(2));
2898     } else if (IsX86 && Name.startswith("avx512.mask.psub.")) {
2899       Rep = Builder.CreateSub(CI->getArgOperand(0), CI->getArgOperand(1));
2900       Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep,
2901                           CI->getArgOperand(2));
2902     } else if (IsX86 && Name.startswith("avx512.mask.pmull.")) {
2903       Rep = Builder.CreateMul(CI->getArgOperand(0), CI->getArgOperand(1));
2904       Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep,
2905                           CI->getArgOperand(2));
2906     } else if (IsX86 && Name.startswith("avx512.mask.add.p")) {
2907       if (Name.endswith(".512")) {
2908         Intrinsic::ID IID;
2909         if (Name[17] == 's')
2910           IID = Intrinsic::x86_avx512_add_ps_512;
2911         else
2912           IID = Intrinsic::x86_avx512_add_pd_512;
2913 
2914         Rep = Builder.CreateCall(Intrinsic::getDeclaration(F->getParent(), IID),
2915                                  { CI->getArgOperand(0), CI->getArgOperand(1),
2916                                    CI->getArgOperand(4) });
2917       } else {
2918         Rep = Builder.CreateFAdd(CI->getArgOperand(0), CI->getArgOperand(1));
2919       }
2920       Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep,
2921                           CI->getArgOperand(2));
2922     } else if (IsX86 && Name.startswith("avx512.mask.div.p")) {
2923       if (Name.endswith(".512")) {
2924         Intrinsic::ID IID;
2925         if (Name[17] == 's')
2926           IID = Intrinsic::x86_avx512_div_ps_512;
2927         else
2928           IID = Intrinsic::x86_avx512_div_pd_512;
2929 
2930         Rep = Builder.CreateCall(Intrinsic::getDeclaration(F->getParent(), IID),
2931                                  { CI->getArgOperand(0), CI->getArgOperand(1),
2932                                    CI->getArgOperand(4) });
2933       } else {
2934         Rep = Builder.CreateFDiv(CI->getArgOperand(0), CI->getArgOperand(1));
2935       }
2936       Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep,
2937                           CI->getArgOperand(2));
2938     } else if (IsX86 && Name.startswith("avx512.mask.mul.p")) {
2939       if (Name.endswith(".512")) {
2940         Intrinsic::ID IID;
2941         if (Name[17] == 's')
2942           IID = Intrinsic::x86_avx512_mul_ps_512;
2943         else
2944           IID = Intrinsic::x86_avx512_mul_pd_512;
2945 
2946         Rep = Builder.CreateCall(Intrinsic::getDeclaration(F->getParent(), IID),
2947                                  { CI->getArgOperand(0), CI->getArgOperand(1),
2948                                    CI->getArgOperand(4) });
2949       } else {
2950         Rep = Builder.CreateFMul(CI->getArgOperand(0), CI->getArgOperand(1));
2951       }
2952       Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep,
2953                           CI->getArgOperand(2));
2954     } else if (IsX86 && Name.startswith("avx512.mask.sub.p")) {
2955       if (Name.endswith(".512")) {
2956         Intrinsic::ID IID;
2957         if (Name[17] == 's')
2958           IID = Intrinsic::x86_avx512_sub_ps_512;
2959         else
2960           IID = Intrinsic::x86_avx512_sub_pd_512;
2961 
2962         Rep = Builder.CreateCall(Intrinsic::getDeclaration(F->getParent(), IID),
2963                                  { CI->getArgOperand(0), CI->getArgOperand(1),
2964                                    CI->getArgOperand(4) });
2965       } else {
2966         Rep = Builder.CreateFSub(CI->getArgOperand(0), CI->getArgOperand(1));
2967       }
2968       Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep,
2969                           CI->getArgOperand(2));
2970     } else if (IsX86 && (Name.startswith("avx512.mask.max.p") ||
2971                          Name.startswith("avx512.mask.min.p")) &&
2972                Name.drop_front(18) == ".512") {
2973       bool IsDouble = Name[17] == 'd';
2974       bool IsMin = Name[13] == 'i';
2975       static const Intrinsic::ID MinMaxTbl[2][2] = {
2976         { Intrinsic::x86_avx512_max_ps_512, Intrinsic::x86_avx512_max_pd_512 },
2977         { Intrinsic::x86_avx512_min_ps_512, Intrinsic::x86_avx512_min_pd_512 }
2978       };
2979       Intrinsic::ID IID = MinMaxTbl[IsMin][IsDouble];
2980 
2981       Rep = Builder.CreateCall(Intrinsic::getDeclaration(F->getParent(), IID),
2982                                { CI->getArgOperand(0), CI->getArgOperand(1),
2983                                  CI->getArgOperand(4) });
2984       Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep,
2985                           CI->getArgOperand(2));
2986     } else if (IsX86 && Name.startswith("avx512.mask.lzcnt.")) {
2987       Rep = Builder.CreateCall(Intrinsic::getDeclaration(F->getParent(),
2988                                                          Intrinsic::ctlz,
2989                                                          CI->getType()),
2990                                { CI->getArgOperand(0), Builder.getInt1(false) });
2991       Rep = EmitX86Select(Builder, CI->getArgOperand(2), Rep,
2992                           CI->getArgOperand(1));
2993     } else if (IsX86 && Name.startswith("avx512.mask.psll")) {
2994       bool IsImmediate = Name[16] == 'i' ||
2995                          (Name.size() > 18 && Name[18] == 'i');
2996       bool IsVariable = Name[16] == 'v';
2997       char Size = Name[16] == '.' ? Name[17] :
2998                   Name[17] == '.' ? Name[18] :
2999                   Name[18] == '.' ? Name[19] :
3000                                     Name[20];
3001 
3002       Intrinsic::ID IID;
3003       if (IsVariable && Name[17] != '.') {
3004         if (Size == 'd' && Name[17] == '2') // avx512.mask.psllv2.di
3005           IID = Intrinsic::x86_avx2_psllv_q;
3006         else if (Size == 'd' && Name[17] == '4') // avx512.mask.psllv4.di
3007           IID = Intrinsic::x86_avx2_psllv_q_256;
3008         else if (Size == 's' && Name[17] == '4') // avx512.mask.psllv4.si
3009           IID = Intrinsic::x86_avx2_psllv_d;
3010         else if (Size == 's' && Name[17] == '8') // avx512.mask.psllv8.si
3011           IID = Intrinsic::x86_avx2_psllv_d_256;
3012         else if (Size == 'h' && Name[17] == '8') // avx512.mask.psllv8.hi
3013           IID = Intrinsic::x86_avx512_psllv_w_128;
3014         else if (Size == 'h' && Name[17] == '1') // avx512.mask.psllv16.hi
3015           IID = Intrinsic::x86_avx512_psllv_w_256;
3016         else if (Name[17] == '3' && Name[18] == '2') // avx512.mask.psllv32hi
3017           IID = Intrinsic::x86_avx512_psllv_w_512;
3018         else
3019           llvm_unreachable("Unexpected size");
3020       } else if (Name.endswith(".128")) {
3021         if (Size == 'd') // avx512.mask.psll.d.128, avx512.mask.psll.di.128
3022           IID = IsImmediate ? Intrinsic::x86_sse2_pslli_d
3023                             : Intrinsic::x86_sse2_psll_d;
3024         else if (Size == 'q') // avx512.mask.psll.q.128, avx512.mask.psll.qi.128
3025           IID = IsImmediate ? Intrinsic::x86_sse2_pslli_q
3026                             : Intrinsic::x86_sse2_psll_q;
3027         else if (Size == 'w') // avx512.mask.psll.w.128, avx512.mask.psll.wi.128
3028           IID = IsImmediate ? Intrinsic::x86_sse2_pslli_w
3029                             : Intrinsic::x86_sse2_psll_w;
3030         else
3031           llvm_unreachable("Unexpected size");
3032       } else if (Name.endswith(".256")) {
3033         if (Size == 'd') // avx512.mask.psll.d.256, avx512.mask.psll.di.256
3034           IID = IsImmediate ? Intrinsic::x86_avx2_pslli_d
3035                             : Intrinsic::x86_avx2_psll_d;
3036         else if (Size == 'q') // avx512.mask.psll.q.256, avx512.mask.psll.qi.256
3037           IID = IsImmediate ? Intrinsic::x86_avx2_pslli_q
3038                             : Intrinsic::x86_avx2_psll_q;
3039         else if (Size == 'w') // avx512.mask.psll.w.256, avx512.mask.psll.wi.256
3040           IID = IsImmediate ? Intrinsic::x86_avx2_pslli_w
3041                             : Intrinsic::x86_avx2_psll_w;
3042         else
3043           llvm_unreachable("Unexpected size");
3044       } else {
3045         if (Size == 'd') // psll.di.512, pslli.d, psll.d, psllv.d.512
3046           IID = IsImmediate ? Intrinsic::x86_avx512_pslli_d_512 :
3047                 IsVariable  ? Intrinsic::x86_avx512_psllv_d_512 :
3048                               Intrinsic::x86_avx512_psll_d_512;
3049         else if (Size == 'q') // psll.qi.512, pslli.q, psll.q, psllv.q.512
3050           IID = IsImmediate ? Intrinsic::x86_avx512_pslli_q_512 :
3051                 IsVariable  ? Intrinsic::x86_avx512_psllv_q_512 :
3052                               Intrinsic::x86_avx512_psll_q_512;
3053         else if (Size == 'w') // psll.wi.512, pslli.w, psll.w
3054           IID = IsImmediate ? Intrinsic::x86_avx512_pslli_w_512
3055                             : Intrinsic::x86_avx512_psll_w_512;
3056         else
3057           llvm_unreachable("Unexpected size");
3058       }
3059 
3060       Rep = UpgradeX86MaskedShift(Builder, *CI, IID);
3061     } else if (IsX86 && Name.startswith("avx512.mask.psrl")) {
3062       bool IsImmediate = Name[16] == 'i' ||
3063                          (Name.size() > 18 && Name[18] == 'i');
3064       bool IsVariable = Name[16] == 'v';
3065       char Size = Name[16] == '.' ? Name[17] :
3066                   Name[17] == '.' ? Name[18] :
3067                   Name[18] == '.' ? Name[19] :
3068                                     Name[20];
3069 
3070       Intrinsic::ID IID;
3071       if (IsVariable && Name[17] != '.') {
3072         if (Size == 'd' && Name[17] == '2') // avx512.mask.psrlv2.di
3073           IID = Intrinsic::x86_avx2_psrlv_q;
3074         else if (Size == 'd' && Name[17] == '4') // avx512.mask.psrlv4.di
3075           IID = Intrinsic::x86_avx2_psrlv_q_256;
3076         else if (Size == 's' && Name[17] == '4') // avx512.mask.psrlv4.si
3077           IID = Intrinsic::x86_avx2_psrlv_d;
3078         else if (Size == 's' && Name[17] == '8') // avx512.mask.psrlv8.si
3079           IID = Intrinsic::x86_avx2_psrlv_d_256;
3080         else if (Size == 'h' && Name[17] == '8') // avx512.mask.psrlv8.hi
3081           IID = Intrinsic::x86_avx512_psrlv_w_128;
3082         else if (Size == 'h' && Name[17] == '1') // avx512.mask.psrlv16.hi
3083           IID = Intrinsic::x86_avx512_psrlv_w_256;
3084         else if (Name[17] == '3' && Name[18] == '2') // avx512.mask.psrlv32hi
3085           IID = Intrinsic::x86_avx512_psrlv_w_512;
3086         else
3087           llvm_unreachable("Unexpected size");
3088       } else if (Name.endswith(".128")) {
3089         if (Size == 'd') // avx512.mask.psrl.d.128, avx512.mask.psrl.di.128
3090           IID = IsImmediate ? Intrinsic::x86_sse2_psrli_d
3091                             : Intrinsic::x86_sse2_psrl_d;
3092         else if (Size == 'q') // avx512.mask.psrl.q.128, avx512.mask.psrl.qi.128
3093           IID = IsImmediate ? Intrinsic::x86_sse2_psrli_q
3094                             : Intrinsic::x86_sse2_psrl_q;
3095         else if (Size == 'w') // avx512.mask.psrl.w.128, avx512.mask.psrl.wi.128
3096           IID = IsImmediate ? Intrinsic::x86_sse2_psrli_w
3097                             : Intrinsic::x86_sse2_psrl_w;
3098         else
3099           llvm_unreachable("Unexpected size");
3100       } else if (Name.endswith(".256")) {
3101         if (Size == 'd') // avx512.mask.psrl.d.256, avx512.mask.psrl.di.256
3102           IID = IsImmediate ? Intrinsic::x86_avx2_psrli_d
3103                             : Intrinsic::x86_avx2_psrl_d;
3104         else if (Size == 'q') // avx512.mask.psrl.q.256, avx512.mask.psrl.qi.256
3105           IID = IsImmediate ? Intrinsic::x86_avx2_psrli_q
3106                             : Intrinsic::x86_avx2_psrl_q;
3107         else if (Size == 'w') // avx512.mask.psrl.w.256, avx512.mask.psrl.wi.256
3108           IID = IsImmediate ? Intrinsic::x86_avx2_psrli_w
3109                             : Intrinsic::x86_avx2_psrl_w;
3110         else
3111           llvm_unreachable("Unexpected size");
3112       } else {
3113         if (Size == 'd') // psrl.di.512, psrli.d, psrl.d, psrl.d.512
3114           IID = IsImmediate ? Intrinsic::x86_avx512_psrli_d_512 :
3115                 IsVariable  ? Intrinsic::x86_avx512_psrlv_d_512 :
3116                               Intrinsic::x86_avx512_psrl_d_512;
3117         else if (Size == 'q') // psrl.qi.512, psrli.q, psrl.q, psrl.q.512
3118           IID = IsImmediate ? Intrinsic::x86_avx512_psrli_q_512 :
3119                 IsVariable  ? Intrinsic::x86_avx512_psrlv_q_512 :
3120                               Intrinsic::x86_avx512_psrl_q_512;
3121         else if (Size == 'w') // psrl.wi.512, psrli.w, psrl.w)
3122           IID = IsImmediate ? Intrinsic::x86_avx512_psrli_w_512
3123                             : Intrinsic::x86_avx512_psrl_w_512;
3124         else
3125           llvm_unreachable("Unexpected size");
3126       }
3127 
3128       Rep = UpgradeX86MaskedShift(Builder, *CI, IID);
3129     } else if (IsX86 && Name.startswith("avx512.mask.psra")) {
3130       bool IsImmediate = Name[16] == 'i' ||
3131                          (Name.size() > 18 && Name[18] == 'i');
3132       bool IsVariable = Name[16] == 'v';
3133       char Size = Name[16] == '.' ? Name[17] :
3134                   Name[17] == '.' ? Name[18] :
3135                   Name[18] == '.' ? Name[19] :
3136                                     Name[20];
3137 
3138       Intrinsic::ID IID;
3139       if (IsVariable && Name[17] != '.') {
3140         if (Size == 's' && Name[17] == '4') // avx512.mask.psrav4.si
3141           IID = Intrinsic::x86_avx2_psrav_d;
3142         else if (Size == 's' && Name[17] == '8') // avx512.mask.psrav8.si
3143           IID = Intrinsic::x86_avx2_psrav_d_256;
3144         else if (Size == 'h' && Name[17] == '8') // avx512.mask.psrav8.hi
3145           IID = Intrinsic::x86_avx512_psrav_w_128;
3146         else if (Size == 'h' && Name[17] == '1') // avx512.mask.psrav16.hi
3147           IID = Intrinsic::x86_avx512_psrav_w_256;
3148         else if (Name[17] == '3' && Name[18] == '2') // avx512.mask.psrav32hi
3149           IID = Intrinsic::x86_avx512_psrav_w_512;
3150         else
3151           llvm_unreachable("Unexpected size");
3152       } else if (Name.endswith(".128")) {
3153         if (Size == 'd') // avx512.mask.psra.d.128, avx512.mask.psra.di.128
3154           IID = IsImmediate ? Intrinsic::x86_sse2_psrai_d
3155                             : Intrinsic::x86_sse2_psra_d;
3156         else if (Size == 'q') // avx512.mask.psra.q.128, avx512.mask.psra.qi.128
3157           IID = IsImmediate ? Intrinsic::x86_avx512_psrai_q_128 :
3158                 IsVariable  ? Intrinsic::x86_avx512_psrav_q_128 :
3159                               Intrinsic::x86_avx512_psra_q_128;
3160         else if (Size == 'w') // avx512.mask.psra.w.128, avx512.mask.psra.wi.128
3161           IID = IsImmediate ? Intrinsic::x86_sse2_psrai_w
3162                             : Intrinsic::x86_sse2_psra_w;
3163         else
3164           llvm_unreachable("Unexpected size");
3165       } else if (Name.endswith(".256")) {
3166         if (Size == 'd') // avx512.mask.psra.d.256, avx512.mask.psra.di.256
3167           IID = IsImmediate ? Intrinsic::x86_avx2_psrai_d
3168                             : Intrinsic::x86_avx2_psra_d;
3169         else if (Size == 'q') // avx512.mask.psra.q.256, avx512.mask.psra.qi.256
3170           IID = IsImmediate ? Intrinsic::x86_avx512_psrai_q_256 :
3171                 IsVariable  ? Intrinsic::x86_avx512_psrav_q_256 :
3172                               Intrinsic::x86_avx512_psra_q_256;
3173         else if (Size == 'w') // avx512.mask.psra.w.256, avx512.mask.psra.wi.256
3174           IID = IsImmediate ? Intrinsic::x86_avx2_psrai_w
3175                             : Intrinsic::x86_avx2_psra_w;
3176         else
3177           llvm_unreachable("Unexpected size");
3178       } else {
3179         if (Size == 'd') // psra.di.512, psrai.d, psra.d, psrav.d.512
3180           IID = IsImmediate ? Intrinsic::x86_avx512_psrai_d_512 :
3181                 IsVariable  ? Intrinsic::x86_avx512_psrav_d_512 :
3182                               Intrinsic::x86_avx512_psra_d_512;
3183         else if (Size == 'q') // psra.qi.512, psrai.q, psra.q
3184           IID = IsImmediate ? Intrinsic::x86_avx512_psrai_q_512 :
3185                 IsVariable  ? Intrinsic::x86_avx512_psrav_q_512 :
3186                               Intrinsic::x86_avx512_psra_q_512;
3187         else if (Size == 'w') // psra.wi.512, psrai.w, psra.w
3188           IID = IsImmediate ? Intrinsic::x86_avx512_psrai_w_512
3189                             : Intrinsic::x86_avx512_psra_w_512;
3190         else
3191           llvm_unreachable("Unexpected size");
3192       }
3193 
3194       Rep = UpgradeX86MaskedShift(Builder, *CI, IID);
3195     } else if (IsX86 && Name.startswith("avx512.mask.move.s")) {
3196       Rep = upgradeMaskedMove(Builder, *CI);
3197     } else if (IsX86 && Name.startswith("avx512.cvtmask2")) {
3198       Rep = UpgradeMaskToInt(Builder, *CI);
3199     } else if (IsX86 && Name.endswith(".movntdqa")) {
3200       Module *M = F->getParent();
3201       MDNode *Node = MDNode::get(
3202           C, ConstantAsMetadata::get(ConstantInt::get(Type::getInt32Ty(C), 1)));
3203 
3204       Value *Ptr = CI->getArgOperand(0);
3205 
3206       // Convert the type of the pointer to a pointer to the stored type.
3207       Value *BC = Builder.CreateBitCast(
3208           Ptr, PointerType::getUnqual(CI->getType()), "cast");
3209       LoadInst *LI = Builder.CreateAlignedLoad(
3210           CI->getType(), BC,
3211           Align(CI->getType()->getPrimitiveSizeInBits().getFixedSize() / 8));
3212       LI->setMetadata(M->getMDKindID("nontemporal"), Node);
3213       Rep = LI;
3214     } else if (IsX86 && (Name.startswith("fma.vfmadd.") ||
3215                          Name.startswith("fma.vfmsub.") ||
3216                          Name.startswith("fma.vfnmadd.") ||
3217                          Name.startswith("fma.vfnmsub."))) {
3218       bool NegMul = Name[6] == 'n';
3219       bool NegAcc = NegMul ? Name[8] == 's' : Name[7] == 's';
3220       bool IsScalar = NegMul ? Name[12] == 's' : Name[11] == 's';
3221 
3222       Value *Ops[] = { CI->getArgOperand(0), CI->getArgOperand(1),
3223                        CI->getArgOperand(2) };
3224 
3225       if (IsScalar) {
3226         Ops[0] = Builder.CreateExtractElement(Ops[0], (uint64_t)0);
3227         Ops[1] = Builder.CreateExtractElement(Ops[1], (uint64_t)0);
3228         Ops[2] = Builder.CreateExtractElement(Ops[2], (uint64_t)0);
3229       }
3230 
3231       if (NegMul && !IsScalar)
3232         Ops[0] = Builder.CreateFNeg(Ops[0]);
3233       if (NegMul && IsScalar)
3234         Ops[1] = Builder.CreateFNeg(Ops[1]);
3235       if (NegAcc)
3236         Ops[2] = Builder.CreateFNeg(Ops[2]);
3237 
3238       Rep = Builder.CreateCall(Intrinsic::getDeclaration(CI->getModule(),
3239                                                          Intrinsic::fma,
3240                                                          Ops[0]->getType()),
3241                                Ops);
3242 
3243       if (IsScalar)
3244         Rep = Builder.CreateInsertElement(CI->getArgOperand(0), Rep,
3245                                           (uint64_t)0);
3246     } else if (IsX86 && Name.startswith("fma4.vfmadd.s")) {
3247       Value *Ops[] = { CI->getArgOperand(0), CI->getArgOperand(1),
3248                        CI->getArgOperand(2) };
3249 
3250       Ops[0] = Builder.CreateExtractElement(Ops[0], (uint64_t)0);
3251       Ops[1] = Builder.CreateExtractElement(Ops[1], (uint64_t)0);
3252       Ops[2] = Builder.CreateExtractElement(Ops[2], (uint64_t)0);
3253 
3254       Rep = Builder.CreateCall(Intrinsic::getDeclaration(CI->getModule(),
3255                                                          Intrinsic::fma,
3256                                                          Ops[0]->getType()),
3257                                Ops);
3258 
3259       Rep = Builder.CreateInsertElement(Constant::getNullValue(CI->getType()),
3260                                         Rep, (uint64_t)0);
3261     } else if (IsX86 && (Name.startswith("avx512.mask.vfmadd.s") ||
3262                          Name.startswith("avx512.maskz.vfmadd.s") ||
3263                          Name.startswith("avx512.mask3.vfmadd.s") ||
3264                          Name.startswith("avx512.mask3.vfmsub.s") ||
3265                          Name.startswith("avx512.mask3.vfnmsub.s"))) {
3266       bool IsMask3 = Name[11] == '3';
3267       bool IsMaskZ = Name[11] == 'z';
3268       // Drop the "avx512.mask." to make it easier.
3269       Name = Name.drop_front(IsMask3 || IsMaskZ ? 13 : 12);
3270       bool NegMul = Name[2] == 'n';
3271       bool NegAcc = NegMul ? Name[4] == 's' : Name[3] == 's';
3272 
3273       Value *A = CI->getArgOperand(0);
3274       Value *B = CI->getArgOperand(1);
3275       Value *C = CI->getArgOperand(2);
3276 
3277       if (NegMul && (IsMask3 || IsMaskZ))
3278         A = Builder.CreateFNeg(A);
3279       if (NegMul && !(IsMask3 || IsMaskZ))
3280         B = Builder.CreateFNeg(B);
3281       if (NegAcc)
3282         C = Builder.CreateFNeg(C);
3283 
3284       A = Builder.CreateExtractElement(A, (uint64_t)0);
3285       B = Builder.CreateExtractElement(B, (uint64_t)0);
3286       C = Builder.CreateExtractElement(C, (uint64_t)0);
3287 
3288       if (!isa<ConstantInt>(CI->getArgOperand(4)) ||
3289           cast<ConstantInt>(CI->getArgOperand(4))->getZExtValue() != 4) {
3290         Value *Ops[] = { A, B, C, CI->getArgOperand(4) };
3291 
3292         Intrinsic::ID IID;
3293         if (Name.back() == 'd')
3294           IID = Intrinsic::x86_avx512_vfmadd_f64;
3295         else
3296           IID = Intrinsic::x86_avx512_vfmadd_f32;
3297         Function *FMA = Intrinsic::getDeclaration(CI->getModule(), IID);
3298         Rep = Builder.CreateCall(FMA, Ops);
3299       } else {
3300         Function *FMA = Intrinsic::getDeclaration(CI->getModule(),
3301                                                   Intrinsic::fma,
3302                                                   A->getType());
3303         Rep = Builder.CreateCall(FMA, { A, B, C });
3304       }
3305 
3306       Value *PassThru = IsMaskZ ? Constant::getNullValue(Rep->getType()) :
3307                         IsMask3 ? C : A;
3308 
3309       // For Mask3 with NegAcc, we need to create a new extractelement that
3310       // avoids the negation above.
3311       if (NegAcc && IsMask3)
3312         PassThru = Builder.CreateExtractElement(CI->getArgOperand(2),
3313                                                 (uint64_t)0);
3314 
3315       Rep = EmitX86ScalarSelect(Builder, CI->getArgOperand(3),
3316                                 Rep, PassThru);
3317       Rep = Builder.CreateInsertElement(CI->getArgOperand(IsMask3 ? 2 : 0),
3318                                         Rep, (uint64_t)0);
3319     } else if (IsX86 && (Name.startswith("avx512.mask.vfmadd.p") ||
3320                          Name.startswith("avx512.mask.vfnmadd.p") ||
3321                          Name.startswith("avx512.mask.vfnmsub.p") ||
3322                          Name.startswith("avx512.mask3.vfmadd.p") ||
3323                          Name.startswith("avx512.mask3.vfmsub.p") ||
3324                          Name.startswith("avx512.mask3.vfnmsub.p") ||
3325                          Name.startswith("avx512.maskz.vfmadd.p"))) {
3326       bool IsMask3 = Name[11] == '3';
3327       bool IsMaskZ = Name[11] == 'z';
3328       // Drop the "avx512.mask." to make it easier.
3329       Name = Name.drop_front(IsMask3 || IsMaskZ ? 13 : 12);
3330       bool NegMul = Name[2] == 'n';
3331       bool NegAcc = NegMul ? Name[4] == 's' : Name[3] == 's';
3332 
3333       Value *A = CI->getArgOperand(0);
3334       Value *B = CI->getArgOperand(1);
3335       Value *C = CI->getArgOperand(2);
3336 
3337       if (NegMul && (IsMask3 || IsMaskZ))
3338         A = Builder.CreateFNeg(A);
3339       if (NegMul && !(IsMask3 || IsMaskZ))
3340         B = Builder.CreateFNeg(B);
3341       if (NegAcc)
3342         C = Builder.CreateFNeg(C);
3343 
3344       if (CI->getNumArgOperands() == 5 &&
3345           (!isa<ConstantInt>(CI->getArgOperand(4)) ||
3346            cast<ConstantInt>(CI->getArgOperand(4))->getZExtValue() != 4)) {
3347         Intrinsic::ID IID;
3348         // Check the character before ".512" in string.
3349         if (Name[Name.size()-5] == 's')
3350           IID = Intrinsic::x86_avx512_vfmadd_ps_512;
3351         else
3352           IID = Intrinsic::x86_avx512_vfmadd_pd_512;
3353 
3354         Rep = Builder.CreateCall(Intrinsic::getDeclaration(F->getParent(), IID),
3355                                  { A, B, C, CI->getArgOperand(4) });
3356       } else {
3357         Function *FMA = Intrinsic::getDeclaration(CI->getModule(),
3358                                                   Intrinsic::fma,
3359                                                   A->getType());
3360         Rep = Builder.CreateCall(FMA, { A, B, C });
3361       }
3362 
3363       Value *PassThru = IsMaskZ ? llvm::Constant::getNullValue(CI->getType()) :
3364                         IsMask3 ? CI->getArgOperand(2) :
3365                                   CI->getArgOperand(0);
3366 
3367       Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep, PassThru);
3368     } else if (IsX86 &&  Name.startswith("fma.vfmsubadd.p")) {
3369       unsigned VecWidth = CI->getType()->getPrimitiveSizeInBits();
3370       unsigned EltWidth = CI->getType()->getScalarSizeInBits();
3371       Intrinsic::ID IID;
3372       if (VecWidth == 128 && EltWidth == 32)
3373         IID = Intrinsic::x86_fma_vfmaddsub_ps;
3374       else if (VecWidth == 256 && EltWidth == 32)
3375         IID = Intrinsic::x86_fma_vfmaddsub_ps_256;
3376       else if (VecWidth == 128 && EltWidth == 64)
3377         IID = Intrinsic::x86_fma_vfmaddsub_pd;
3378       else if (VecWidth == 256 && EltWidth == 64)
3379         IID = Intrinsic::x86_fma_vfmaddsub_pd_256;
3380       else
3381         llvm_unreachable("Unexpected intrinsic");
3382 
3383       Value *Ops[] = { CI->getArgOperand(0), CI->getArgOperand(1),
3384                        CI->getArgOperand(2) };
3385       Ops[2] = Builder.CreateFNeg(Ops[2]);
3386       Rep = Builder.CreateCall(Intrinsic::getDeclaration(F->getParent(), IID),
3387                                Ops);
3388     } else if (IsX86 && (Name.startswith("avx512.mask.vfmaddsub.p") ||
3389                          Name.startswith("avx512.mask3.vfmaddsub.p") ||
3390                          Name.startswith("avx512.maskz.vfmaddsub.p") ||
3391                          Name.startswith("avx512.mask3.vfmsubadd.p"))) {
3392       bool IsMask3 = Name[11] == '3';
3393       bool IsMaskZ = Name[11] == 'z';
3394       // Drop the "avx512.mask." to make it easier.
3395       Name = Name.drop_front(IsMask3 || IsMaskZ ? 13 : 12);
3396       bool IsSubAdd = Name[3] == 's';
3397       if (CI->getNumArgOperands() == 5) {
3398         Intrinsic::ID IID;
3399         // Check the character before ".512" in string.
3400         if (Name[Name.size()-5] == 's')
3401           IID = Intrinsic::x86_avx512_vfmaddsub_ps_512;
3402         else
3403           IID = Intrinsic::x86_avx512_vfmaddsub_pd_512;
3404 
3405         Value *Ops[] = { CI->getArgOperand(0), CI->getArgOperand(1),
3406                          CI->getArgOperand(2), CI->getArgOperand(4) };
3407         if (IsSubAdd)
3408           Ops[2] = Builder.CreateFNeg(Ops[2]);
3409 
3410         Rep = Builder.CreateCall(Intrinsic::getDeclaration(F->getParent(), IID),
3411                                  Ops);
3412       } else {
3413         int NumElts = cast<FixedVectorType>(CI->getType())->getNumElements();
3414 
3415         Value *Ops[] = { CI->getArgOperand(0), CI->getArgOperand(1),
3416                          CI->getArgOperand(2) };
3417 
3418         Function *FMA = Intrinsic::getDeclaration(CI->getModule(), Intrinsic::fma,
3419                                                   Ops[0]->getType());
3420         Value *Odd = Builder.CreateCall(FMA, Ops);
3421         Ops[2] = Builder.CreateFNeg(Ops[2]);
3422         Value *Even = Builder.CreateCall(FMA, Ops);
3423 
3424         if (IsSubAdd)
3425           std::swap(Even, Odd);
3426 
3427         SmallVector<int, 32> Idxs(NumElts);
3428         for (int i = 0; i != NumElts; ++i)
3429           Idxs[i] = i + (i % 2) * NumElts;
3430 
3431         Rep = Builder.CreateShuffleVector(Even, Odd, Idxs);
3432       }
3433 
3434       Value *PassThru = IsMaskZ ? llvm::Constant::getNullValue(CI->getType()) :
3435                         IsMask3 ? CI->getArgOperand(2) :
3436                                   CI->getArgOperand(0);
3437 
3438       Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep, PassThru);
3439     } else if (IsX86 && (Name.startswith("avx512.mask.pternlog.") ||
3440                          Name.startswith("avx512.maskz.pternlog."))) {
3441       bool ZeroMask = Name[11] == 'z';
3442       unsigned VecWidth = CI->getType()->getPrimitiveSizeInBits();
3443       unsigned EltWidth = CI->getType()->getScalarSizeInBits();
3444       Intrinsic::ID IID;
3445       if (VecWidth == 128 && EltWidth == 32)
3446         IID = Intrinsic::x86_avx512_pternlog_d_128;
3447       else if (VecWidth == 256 && EltWidth == 32)
3448         IID = Intrinsic::x86_avx512_pternlog_d_256;
3449       else if (VecWidth == 512 && EltWidth == 32)
3450         IID = Intrinsic::x86_avx512_pternlog_d_512;
3451       else if (VecWidth == 128 && EltWidth == 64)
3452         IID = Intrinsic::x86_avx512_pternlog_q_128;
3453       else if (VecWidth == 256 && EltWidth == 64)
3454         IID = Intrinsic::x86_avx512_pternlog_q_256;
3455       else if (VecWidth == 512 && EltWidth == 64)
3456         IID = Intrinsic::x86_avx512_pternlog_q_512;
3457       else
3458         llvm_unreachable("Unexpected intrinsic");
3459 
3460       Value *Args[] = { CI->getArgOperand(0) , CI->getArgOperand(1),
3461                         CI->getArgOperand(2), CI->getArgOperand(3) };
3462       Rep = Builder.CreateCall(Intrinsic::getDeclaration(CI->getModule(), IID),
3463                                Args);
3464       Value *PassThru = ZeroMask ? ConstantAggregateZero::get(CI->getType())
3465                                  : CI->getArgOperand(0);
3466       Rep = EmitX86Select(Builder, CI->getArgOperand(4), Rep, PassThru);
3467     } else if (IsX86 && (Name.startswith("avx512.mask.vpmadd52") ||
3468                          Name.startswith("avx512.maskz.vpmadd52"))) {
3469       bool ZeroMask = Name[11] == 'z';
3470       bool High = Name[20] == 'h' || Name[21] == 'h';
3471       unsigned VecWidth = CI->getType()->getPrimitiveSizeInBits();
3472       Intrinsic::ID IID;
3473       if (VecWidth == 128 && !High)
3474         IID = Intrinsic::x86_avx512_vpmadd52l_uq_128;
3475       else if (VecWidth == 256 && !High)
3476         IID = Intrinsic::x86_avx512_vpmadd52l_uq_256;
3477       else if (VecWidth == 512 && !High)
3478         IID = Intrinsic::x86_avx512_vpmadd52l_uq_512;
3479       else if (VecWidth == 128 && High)
3480         IID = Intrinsic::x86_avx512_vpmadd52h_uq_128;
3481       else if (VecWidth == 256 && High)
3482         IID = Intrinsic::x86_avx512_vpmadd52h_uq_256;
3483       else if (VecWidth == 512 && High)
3484         IID = Intrinsic::x86_avx512_vpmadd52h_uq_512;
3485       else
3486         llvm_unreachable("Unexpected intrinsic");
3487 
3488       Value *Args[] = { CI->getArgOperand(0) , CI->getArgOperand(1),
3489                         CI->getArgOperand(2) };
3490       Rep = Builder.CreateCall(Intrinsic::getDeclaration(CI->getModule(), IID),
3491                                Args);
3492       Value *PassThru = ZeroMask ? ConstantAggregateZero::get(CI->getType())
3493                                  : CI->getArgOperand(0);
3494       Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep, PassThru);
3495     } else if (IsX86 && (Name.startswith("avx512.mask.vpermi2var.") ||
3496                          Name.startswith("avx512.mask.vpermt2var.") ||
3497                          Name.startswith("avx512.maskz.vpermt2var."))) {
3498       bool ZeroMask = Name[11] == 'z';
3499       bool IndexForm = Name[17] == 'i';
3500       Rep = UpgradeX86VPERMT2Intrinsics(Builder, *CI, ZeroMask, IndexForm);
3501     } else if (IsX86 && (Name.startswith("avx512.mask.vpdpbusd.") ||
3502                          Name.startswith("avx512.maskz.vpdpbusd.") ||
3503                          Name.startswith("avx512.mask.vpdpbusds.") ||
3504                          Name.startswith("avx512.maskz.vpdpbusds."))) {
3505       bool ZeroMask = Name[11] == 'z';
3506       bool IsSaturating = Name[ZeroMask ? 21 : 20] == 's';
3507       unsigned VecWidth = CI->getType()->getPrimitiveSizeInBits();
3508       Intrinsic::ID IID;
3509       if (VecWidth == 128 && !IsSaturating)
3510         IID = Intrinsic::x86_avx512_vpdpbusd_128;
3511       else if (VecWidth == 256 && !IsSaturating)
3512         IID = Intrinsic::x86_avx512_vpdpbusd_256;
3513       else if (VecWidth == 512 && !IsSaturating)
3514         IID = Intrinsic::x86_avx512_vpdpbusd_512;
3515       else if (VecWidth == 128 && IsSaturating)
3516         IID = Intrinsic::x86_avx512_vpdpbusds_128;
3517       else if (VecWidth == 256 && IsSaturating)
3518         IID = Intrinsic::x86_avx512_vpdpbusds_256;
3519       else if (VecWidth == 512 && IsSaturating)
3520         IID = Intrinsic::x86_avx512_vpdpbusds_512;
3521       else
3522         llvm_unreachable("Unexpected intrinsic");
3523 
3524       Value *Args[] = { CI->getArgOperand(0), CI->getArgOperand(1),
3525                         CI->getArgOperand(2)  };
3526       Rep = Builder.CreateCall(Intrinsic::getDeclaration(CI->getModule(), IID),
3527                                Args);
3528       Value *PassThru = ZeroMask ? ConstantAggregateZero::get(CI->getType())
3529                                  : CI->getArgOperand(0);
3530       Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep, PassThru);
3531     } else if (IsX86 && (Name.startswith("avx512.mask.vpdpwssd.") ||
3532                          Name.startswith("avx512.maskz.vpdpwssd.") ||
3533                          Name.startswith("avx512.mask.vpdpwssds.") ||
3534                          Name.startswith("avx512.maskz.vpdpwssds."))) {
3535       bool ZeroMask = Name[11] == 'z';
3536       bool IsSaturating = Name[ZeroMask ? 21 : 20] == 's';
3537       unsigned VecWidth = CI->getType()->getPrimitiveSizeInBits();
3538       Intrinsic::ID IID;
3539       if (VecWidth == 128 && !IsSaturating)
3540         IID = Intrinsic::x86_avx512_vpdpwssd_128;
3541       else if (VecWidth == 256 && !IsSaturating)
3542         IID = Intrinsic::x86_avx512_vpdpwssd_256;
3543       else if (VecWidth == 512 && !IsSaturating)
3544         IID = Intrinsic::x86_avx512_vpdpwssd_512;
3545       else if (VecWidth == 128 && IsSaturating)
3546         IID = Intrinsic::x86_avx512_vpdpwssds_128;
3547       else if (VecWidth == 256 && IsSaturating)
3548         IID = Intrinsic::x86_avx512_vpdpwssds_256;
3549       else if (VecWidth == 512 && IsSaturating)
3550         IID = Intrinsic::x86_avx512_vpdpwssds_512;
3551       else
3552         llvm_unreachable("Unexpected intrinsic");
3553 
3554       Value *Args[] = { CI->getArgOperand(0), CI->getArgOperand(1),
3555                         CI->getArgOperand(2)  };
3556       Rep = Builder.CreateCall(Intrinsic::getDeclaration(CI->getModule(), IID),
3557                                Args);
3558       Value *PassThru = ZeroMask ? ConstantAggregateZero::get(CI->getType())
3559                                  : CI->getArgOperand(0);
3560       Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep, PassThru);
3561     } else if (IsX86 && (Name == "addcarryx.u32" || Name == "addcarryx.u64" ||
3562                          Name == "addcarry.u32" || Name == "addcarry.u64" ||
3563                          Name == "subborrow.u32" || Name == "subborrow.u64")) {
3564       Intrinsic::ID IID;
3565       if (Name[0] == 'a' && Name.back() == '2')
3566         IID = Intrinsic::x86_addcarry_32;
3567       else if (Name[0] == 'a' && Name.back() == '4')
3568         IID = Intrinsic::x86_addcarry_64;
3569       else if (Name[0] == 's' && Name.back() == '2')
3570         IID = Intrinsic::x86_subborrow_32;
3571       else if (Name[0] == 's' && Name.back() == '4')
3572         IID = Intrinsic::x86_subborrow_64;
3573       else
3574         llvm_unreachable("Unexpected intrinsic");
3575 
3576       // Make a call with 3 operands.
3577       Value *Args[] = { CI->getArgOperand(0), CI->getArgOperand(1),
3578                         CI->getArgOperand(2)};
3579       Value *NewCall = Builder.CreateCall(
3580                                 Intrinsic::getDeclaration(CI->getModule(), IID),
3581                                 Args);
3582 
3583       // Extract the second result and store it.
3584       Value *Data = Builder.CreateExtractValue(NewCall, 1);
3585       // Cast the pointer to the right type.
3586       Value *Ptr = Builder.CreateBitCast(CI->getArgOperand(3),
3587                                  llvm::PointerType::getUnqual(Data->getType()));
3588       Builder.CreateAlignedStore(Data, Ptr, Align(1));
3589       // Replace the original call result with the first result of the new call.
3590       Value *CF = Builder.CreateExtractValue(NewCall, 0);
3591 
3592       CI->replaceAllUsesWith(CF);
3593       Rep = nullptr;
3594     } else if (IsX86 && Name.startswith("avx512.mask.") &&
3595                upgradeAVX512MaskToSelect(Name, Builder, *CI, Rep)) {
3596       // Rep will be updated by the call in the condition.
3597     } else if (IsNVVM && (Name == "abs.i" || Name == "abs.ll")) {
3598       Value *Arg = CI->getArgOperand(0);
3599       Value *Neg = Builder.CreateNeg(Arg, "neg");
3600       Value *Cmp = Builder.CreateICmpSGE(
3601           Arg, llvm::Constant::getNullValue(Arg->getType()), "abs.cond");
3602       Rep = Builder.CreateSelect(Cmp, Arg, Neg, "abs");
3603     } else if (IsNVVM && (Name.startswith("atomic.load.add.f32.p") ||
3604                           Name.startswith("atomic.load.add.f64.p"))) {
3605       Value *Ptr = CI->getArgOperand(0);
3606       Value *Val = CI->getArgOperand(1);
3607       Rep = Builder.CreateAtomicRMW(AtomicRMWInst::FAdd, Ptr, Val, MaybeAlign(),
3608                                     AtomicOrdering::SequentiallyConsistent);
3609     } else if (IsNVVM && (Name == "max.i" || Name == "max.ll" ||
3610                           Name == "max.ui" || Name == "max.ull")) {
3611       Value *Arg0 = CI->getArgOperand(0);
3612       Value *Arg1 = CI->getArgOperand(1);
3613       Value *Cmp = Name.endswith(".ui") || Name.endswith(".ull")
3614                        ? Builder.CreateICmpUGE(Arg0, Arg1, "max.cond")
3615                        : Builder.CreateICmpSGE(Arg0, Arg1, "max.cond");
3616       Rep = Builder.CreateSelect(Cmp, Arg0, Arg1, "max");
3617     } else if (IsNVVM && (Name == "min.i" || Name == "min.ll" ||
3618                           Name == "min.ui" || Name == "min.ull")) {
3619       Value *Arg0 = CI->getArgOperand(0);
3620       Value *Arg1 = CI->getArgOperand(1);
3621       Value *Cmp = Name.endswith(".ui") || Name.endswith(".ull")
3622                        ? Builder.CreateICmpULE(Arg0, Arg1, "min.cond")
3623                        : Builder.CreateICmpSLE(Arg0, Arg1, "min.cond");
3624       Rep = Builder.CreateSelect(Cmp, Arg0, Arg1, "min");
3625     } else if (IsNVVM && Name == "clz.ll") {
3626       // llvm.nvvm.clz.ll returns an i32, but llvm.ctlz.i64 and returns an i64.
3627       Value *Arg = CI->getArgOperand(0);
3628       Value *Ctlz = Builder.CreateCall(
3629           Intrinsic::getDeclaration(F->getParent(), Intrinsic::ctlz,
3630                                     {Arg->getType()}),
3631           {Arg, Builder.getFalse()}, "ctlz");
3632       Rep = Builder.CreateTrunc(Ctlz, Builder.getInt32Ty(), "ctlz.trunc");
3633     } else if (IsNVVM && Name == "popc.ll") {
3634       // llvm.nvvm.popc.ll returns an i32, but llvm.ctpop.i64 and returns an
3635       // i64.
3636       Value *Arg = CI->getArgOperand(0);
3637       Value *Popc = Builder.CreateCall(
3638           Intrinsic::getDeclaration(F->getParent(), Intrinsic::ctpop,
3639                                     {Arg->getType()}),
3640           Arg, "ctpop");
3641       Rep = Builder.CreateTrunc(Popc, Builder.getInt32Ty(), "ctpop.trunc");
3642     } else if (IsNVVM && Name == "h2f") {
3643       Rep = Builder.CreateCall(Intrinsic::getDeclaration(
3644                                    F->getParent(), Intrinsic::convert_from_fp16,
3645                                    {Builder.getFloatTy()}),
3646                                CI->getArgOperand(0), "h2f");
3647     } else {
3648       llvm_unreachable("Unknown function for CallInst upgrade.");
3649     }
3650 
3651     if (Rep)
3652       CI->replaceAllUsesWith(Rep);
3653     CI->eraseFromParent();
3654     return;
3655   }
3656 
3657   const auto &DefaultCase = [&NewFn, &CI]() -> void {
3658     // Handle generic mangling change, but nothing else
3659     assert(
3660         (CI->getCalledFunction()->getName() != NewFn->getName()) &&
3661         "Unknown function for CallInst upgrade and isn't just a name change");
3662     CI->setCalledFunction(NewFn);
3663   };
3664   CallInst *NewCall = nullptr;
3665   switch (NewFn->getIntrinsicID()) {
3666   default: {
3667     DefaultCase();
3668     return;
3669   }
3670   case Intrinsic::arm_neon_vld1:
3671   case Intrinsic::arm_neon_vld2:
3672   case Intrinsic::arm_neon_vld3:
3673   case Intrinsic::arm_neon_vld4:
3674   case Intrinsic::arm_neon_vld2lane:
3675   case Intrinsic::arm_neon_vld3lane:
3676   case Intrinsic::arm_neon_vld4lane:
3677   case Intrinsic::arm_neon_vst1:
3678   case Intrinsic::arm_neon_vst2:
3679   case Intrinsic::arm_neon_vst3:
3680   case Intrinsic::arm_neon_vst4:
3681   case Intrinsic::arm_neon_vst2lane:
3682   case Intrinsic::arm_neon_vst3lane:
3683   case Intrinsic::arm_neon_vst4lane: {
3684     SmallVector<Value *, 4> Args(CI->arg_operands().begin(),
3685                                  CI->arg_operands().end());
3686     NewCall = Builder.CreateCall(NewFn, Args);
3687     break;
3688   }
3689 
3690   case Intrinsic::arm_neon_bfdot:
3691   case Intrinsic::arm_neon_bfmmla:
3692   case Intrinsic::arm_neon_bfmlalb:
3693   case Intrinsic::arm_neon_bfmlalt:
3694   case Intrinsic::aarch64_neon_bfdot:
3695   case Intrinsic::aarch64_neon_bfmmla:
3696   case Intrinsic::aarch64_neon_bfmlalb:
3697   case Intrinsic::aarch64_neon_bfmlalt: {
3698     SmallVector<Value *, 3> Args;
3699     assert(CI->getNumArgOperands() == 3 &&
3700            "Mismatch between function args and call args");
3701     size_t OperandWidth =
3702         CI->getArgOperand(1)->getType()->getPrimitiveSizeInBits();
3703     assert((OperandWidth == 64 || OperandWidth == 128) &&
3704            "Unexpected operand width");
3705     Type *NewTy = FixedVectorType::get(Type::getBFloatTy(C), OperandWidth / 16);
3706     auto Iter = CI->arg_operands().begin();
3707     Args.push_back(*Iter++);
3708     Args.push_back(Builder.CreateBitCast(*Iter++, NewTy));
3709     Args.push_back(Builder.CreateBitCast(*Iter++, NewTy));
3710     NewCall = Builder.CreateCall(NewFn, Args);
3711     break;
3712   }
3713 
3714   case Intrinsic::bitreverse:
3715     NewCall = Builder.CreateCall(NewFn, {CI->getArgOperand(0)});
3716     break;
3717 
3718   case Intrinsic::ctlz:
3719   case Intrinsic::cttz:
3720     assert(CI->getNumArgOperands() == 1 &&
3721            "Mismatch between function args and call args");
3722     NewCall =
3723         Builder.CreateCall(NewFn, {CI->getArgOperand(0), Builder.getFalse()});
3724     break;
3725 
3726   case Intrinsic::objectsize: {
3727     Value *NullIsUnknownSize = CI->getNumArgOperands() == 2
3728                                    ? Builder.getFalse()
3729                                    : CI->getArgOperand(2);
3730     Value *Dynamic =
3731         CI->getNumArgOperands() < 4 ? Builder.getFalse() : CI->getArgOperand(3);
3732     NewCall = Builder.CreateCall(
3733         NewFn, {CI->getArgOperand(0), CI->getArgOperand(1), NullIsUnknownSize, Dynamic});
3734     break;
3735   }
3736 
3737   case Intrinsic::ctpop:
3738     NewCall = Builder.CreateCall(NewFn, {CI->getArgOperand(0)});
3739     break;
3740 
3741   case Intrinsic::convert_from_fp16:
3742     NewCall = Builder.CreateCall(NewFn, {CI->getArgOperand(0)});
3743     break;
3744 
3745   case Intrinsic::dbg_value:
3746     // Upgrade from the old version that had an extra offset argument.
3747     assert(CI->getNumArgOperands() == 4);
3748     // Drop nonzero offsets instead of attempting to upgrade them.
3749     if (auto *Offset = dyn_cast_or_null<Constant>(CI->getArgOperand(1)))
3750       if (Offset->isZeroValue()) {
3751         NewCall = Builder.CreateCall(
3752             NewFn,
3753             {CI->getArgOperand(0), CI->getArgOperand(2), CI->getArgOperand(3)});
3754         break;
3755       }
3756     CI->eraseFromParent();
3757     return;
3758 
3759   case Intrinsic::ptr_annotation:
3760     // Upgrade from versions that lacked the annotation attribute argument.
3761     assert(CI->getNumArgOperands() == 4 &&
3762            "Before LLVM 12.0 this intrinsic took four arguments");
3763     // Create a new call with an added null annotation attribute argument.
3764     NewCall = Builder.CreateCall(
3765         NewFn,
3766         {CI->getArgOperand(0), CI->getArgOperand(1), CI->getArgOperand(2),
3767          CI->getArgOperand(3), Constant::getNullValue(Builder.getInt8PtrTy())});
3768     NewCall->takeName(CI);
3769     CI->replaceAllUsesWith(NewCall);
3770     CI->eraseFromParent();
3771     return;
3772 
3773   case Intrinsic::var_annotation:
3774     // Upgrade from versions that lacked the annotation attribute argument.
3775     assert(CI->getNumArgOperands() == 4 &&
3776            "Before LLVM 12.0 this intrinsic took four arguments");
3777     // Create a new call with an added null annotation attribute argument.
3778     NewCall = Builder.CreateCall(
3779         NewFn,
3780         {CI->getArgOperand(0), CI->getArgOperand(1), CI->getArgOperand(2),
3781          CI->getArgOperand(3), Constant::getNullValue(Builder.getInt8PtrTy())});
3782     CI->eraseFromParent();
3783     return;
3784 
3785   case Intrinsic::x86_xop_vfrcz_ss:
3786   case Intrinsic::x86_xop_vfrcz_sd:
3787     NewCall = Builder.CreateCall(NewFn, {CI->getArgOperand(1)});
3788     break;
3789 
3790   case Intrinsic::x86_xop_vpermil2pd:
3791   case Intrinsic::x86_xop_vpermil2ps:
3792   case Intrinsic::x86_xop_vpermil2pd_256:
3793   case Intrinsic::x86_xop_vpermil2ps_256: {
3794     SmallVector<Value *, 4> Args(CI->arg_operands().begin(),
3795                                  CI->arg_operands().end());
3796     VectorType *FltIdxTy = cast<VectorType>(Args[2]->getType());
3797     VectorType *IntIdxTy = VectorType::getInteger(FltIdxTy);
3798     Args[2] = Builder.CreateBitCast(Args[2], IntIdxTy);
3799     NewCall = Builder.CreateCall(NewFn, Args);
3800     break;
3801   }
3802 
3803   case Intrinsic::x86_sse41_ptestc:
3804   case Intrinsic::x86_sse41_ptestz:
3805   case Intrinsic::x86_sse41_ptestnzc: {
3806     // The arguments for these intrinsics used to be v4f32, and changed
3807     // to v2i64. This is purely a nop, since those are bitwise intrinsics.
3808     // So, the only thing required is a bitcast for both arguments.
3809     // First, check the arguments have the old type.
3810     Value *Arg0 = CI->getArgOperand(0);
3811     if (Arg0->getType() != FixedVectorType::get(Type::getFloatTy(C), 4))
3812       return;
3813 
3814     // Old intrinsic, add bitcasts
3815     Value *Arg1 = CI->getArgOperand(1);
3816 
3817     auto *NewVecTy = FixedVectorType::get(Type::getInt64Ty(C), 2);
3818 
3819     Value *BC0 = Builder.CreateBitCast(Arg0, NewVecTy, "cast");
3820     Value *BC1 = Builder.CreateBitCast(Arg1, NewVecTy, "cast");
3821 
3822     NewCall = Builder.CreateCall(NewFn, {BC0, BC1});
3823     break;
3824   }
3825 
3826   case Intrinsic::x86_rdtscp: {
3827     // This used to take 1 arguments. If we have no arguments, it is already
3828     // upgraded.
3829     if (CI->getNumOperands() == 0)
3830       return;
3831 
3832     NewCall = Builder.CreateCall(NewFn);
3833     // Extract the second result and store it.
3834     Value *Data = Builder.CreateExtractValue(NewCall, 1);
3835     // Cast the pointer to the right type.
3836     Value *Ptr = Builder.CreateBitCast(CI->getArgOperand(0),
3837                                  llvm::PointerType::getUnqual(Data->getType()));
3838     Builder.CreateAlignedStore(Data, Ptr, Align(1));
3839     // Replace the original call result with the first result of the new call.
3840     Value *TSC = Builder.CreateExtractValue(NewCall, 0);
3841 
3842     NewCall->takeName(CI);
3843     CI->replaceAllUsesWith(TSC);
3844     CI->eraseFromParent();
3845     return;
3846   }
3847 
3848   case Intrinsic::x86_sse41_insertps:
3849   case Intrinsic::x86_sse41_dppd:
3850   case Intrinsic::x86_sse41_dpps:
3851   case Intrinsic::x86_sse41_mpsadbw:
3852   case Intrinsic::x86_avx_dp_ps_256:
3853   case Intrinsic::x86_avx2_mpsadbw: {
3854     // Need to truncate the last argument from i32 to i8 -- this argument models
3855     // an inherently 8-bit immediate operand to these x86 instructions.
3856     SmallVector<Value *, 4> Args(CI->arg_operands().begin(),
3857                                  CI->arg_operands().end());
3858 
3859     // Replace the last argument with a trunc.
3860     Args.back() = Builder.CreateTrunc(Args.back(), Type::getInt8Ty(C), "trunc");
3861     NewCall = Builder.CreateCall(NewFn, Args);
3862     break;
3863   }
3864 
3865   case Intrinsic::x86_avx512_mask_cmp_pd_128:
3866   case Intrinsic::x86_avx512_mask_cmp_pd_256:
3867   case Intrinsic::x86_avx512_mask_cmp_pd_512:
3868   case Intrinsic::x86_avx512_mask_cmp_ps_128:
3869   case Intrinsic::x86_avx512_mask_cmp_ps_256:
3870   case Intrinsic::x86_avx512_mask_cmp_ps_512: {
3871     SmallVector<Value *, 4> Args(CI->arg_operands().begin(),
3872                                  CI->arg_operands().end());
3873     unsigned NumElts =
3874         cast<FixedVectorType>(Args[0]->getType())->getNumElements();
3875     Args[3] = getX86MaskVec(Builder, Args[3], NumElts);
3876 
3877     NewCall = Builder.CreateCall(NewFn, Args);
3878     Value *Res = ApplyX86MaskOn1BitsVec(Builder, NewCall, nullptr);
3879 
3880     NewCall->takeName(CI);
3881     CI->replaceAllUsesWith(Res);
3882     CI->eraseFromParent();
3883     return;
3884   }
3885 
3886   case Intrinsic::thread_pointer: {
3887     NewCall = Builder.CreateCall(NewFn, {});
3888     break;
3889   }
3890 
3891   case Intrinsic::invariant_start:
3892   case Intrinsic::invariant_end: {
3893     SmallVector<Value *, 4> Args(CI->arg_operands().begin(),
3894                                  CI->arg_operands().end());
3895     NewCall = Builder.CreateCall(NewFn, Args);
3896     break;
3897   }
3898   case Intrinsic::masked_load:
3899   case Intrinsic::masked_store:
3900   case Intrinsic::masked_gather:
3901   case Intrinsic::masked_scatter: {
3902     SmallVector<Value *, 4> Args(CI->arg_operands().begin(),
3903                                  CI->arg_operands().end());
3904     NewCall = Builder.CreateCall(NewFn, Args);
3905     NewCall->copyMetadata(*CI);
3906     break;
3907   }
3908 
3909   case Intrinsic::memcpy:
3910   case Intrinsic::memmove:
3911   case Intrinsic::memset: {
3912     // We have to make sure that the call signature is what we're expecting.
3913     // We only want to change the old signatures by removing the alignment arg:
3914     //  @llvm.mem[cpy|move]...(i8*, i8*, i[32|i64], i32, i1)
3915     //    -> @llvm.mem[cpy|move]...(i8*, i8*, i[32|i64], i1)
3916     //  @llvm.memset...(i8*, i8, i[32|64], i32, i1)
3917     //    -> @llvm.memset...(i8*, i8, i[32|64], i1)
3918     // Note: i8*'s in the above can be any pointer type
3919     if (CI->getNumArgOperands() != 5) {
3920       DefaultCase();
3921       return;
3922     }
3923     // Remove alignment argument (3), and add alignment attributes to the
3924     // dest/src pointers.
3925     Value *Args[4] = {CI->getArgOperand(0), CI->getArgOperand(1),
3926                       CI->getArgOperand(2), CI->getArgOperand(4)};
3927     NewCall = Builder.CreateCall(NewFn, Args);
3928     auto *MemCI = cast<MemIntrinsic>(NewCall);
3929     // All mem intrinsics support dest alignment.
3930     const ConstantInt *Align = cast<ConstantInt>(CI->getArgOperand(3));
3931     MemCI->setDestAlignment(Align->getMaybeAlignValue());
3932     // Memcpy/Memmove also support source alignment.
3933     if (auto *MTI = dyn_cast<MemTransferInst>(MemCI))
3934       MTI->setSourceAlignment(Align->getMaybeAlignValue());
3935     break;
3936   }
3937   }
3938   assert(NewCall && "Should have either set this variable or returned through "
3939                     "the default case");
3940   NewCall->takeName(CI);
3941   CI->replaceAllUsesWith(NewCall);
3942   CI->eraseFromParent();
3943 }
3944 
UpgradeCallsToIntrinsic(Function * F)3945 void llvm::UpgradeCallsToIntrinsic(Function *F) {
3946   assert(F && "Illegal attempt to upgrade a non-existent intrinsic.");
3947 
3948   // Check if this function should be upgraded and get the replacement function
3949   // if there is one.
3950   Function *NewFn;
3951   if (UpgradeIntrinsicFunction(F, NewFn)) {
3952     // Replace all users of the old function with the new function or new
3953     // instructions. This is not a range loop because the call is deleted.
3954     for (User *U : make_early_inc_range(F->users()))
3955       if (CallInst *CI = dyn_cast<CallInst>(U))
3956         UpgradeIntrinsicCall(CI, NewFn);
3957 
3958     // Remove old function, no longer used, from the module.
3959     F->eraseFromParent();
3960   }
3961 }
3962 
UpgradeTBAANode(MDNode & MD)3963 MDNode *llvm::UpgradeTBAANode(MDNode &MD) {
3964   // Check if the tag uses struct-path aware TBAA format.
3965   if (isa<MDNode>(MD.getOperand(0)) && MD.getNumOperands() >= 3)
3966     return &MD;
3967 
3968   auto &Context = MD.getContext();
3969   if (MD.getNumOperands() == 3) {
3970     Metadata *Elts[] = {MD.getOperand(0), MD.getOperand(1)};
3971     MDNode *ScalarType = MDNode::get(Context, Elts);
3972     // Create a MDNode <ScalarType, ScalarType, offset 0, const>
3973     Metadata *Elts2[] = {ScalarType, ScalarType,
3974                          ConstantAsMetadata::get(
3975                              Constant::getNullValue(Type::getInt64Ty(Context))),
3976                          MD.getOperand(2)};
3977     return MDNode::get(Context, Elts2);
3978   }
3979   // Create a MDNode <MD, MD, offset 0>
3980   Metadata *Elts[] = {&MD, &MD, ConstantAsMetadata::get(Constant::getNullValue(
3981                                     Type::getInt64Ty(Context)))};
3982   return MDNode::get(Context, Elts);
3983 }
3984 
UpgradeBitCastInst(unsigned Opc,Value * V,Type * DestTy,Instruction * & Temp)3985 Instruction *llvm::UpgradeBitCastInst(unsigned Opc, Value *V, Type *DestTy,
3986                                       Instruction *&Temp) {
3987   if (Opc != Instruction::BitCast)
3988     return nullptr;
3989 
3990   Temp = nullptr;
3991   Type *SrcTy = V->getType();
3992   if (SrcTy->isPtrOrPtrVectorTy() && DestTy->isPtrOrPtrVectorTy() &&
3993       SrcTy->getPointerAddressSpace() != DestTy->getPointerAddressSpace()) {
3994     LLVMContext &Context = V->getContext();
3995 
3996     // We have no information about target data layout, so we assume that
3997     // the maximum pointer size is 64bit.
3998     Type *MidTy = Type::getInt64Ty(Context);
3999     Temp = CastInst::Create(Instruction::PtrToInt, V, MidTy);
4000 
4001     return CastInst::Create(Instruction::IntToPtr, Temp, DestTy);
4002   }
4003 
4004   return nullptr;
4005 }
4006 
UpgradeBitCastExpr(unsigned Opc,Constant * C,Type * DestTy)4007 Value *llvm::UpgradeBitCastExpr(unsigned Opc, Constant *C, Type *DestTy) {
4008   if (Opc != Instruction::BitCast)
4009     return nullptr;
4010 
4011   Type *SrcTy = C->getType();
4012   if (SrcTy->isPtrOrPtrVectorTy() && DestTy->isPtrOrPtrVectorTy() &&
4013       SrcTy->getPointerAddressSpace() != DestTy->getPointerAddressSpace()) {
4014     LLVMContext &Context = C->getContext();
4015 
4016     // We have no information about target data layout, so we assume that
4017     // the maximum pointer size is 64bit.
4018     Type *MidTy = Type::getInt64Ty(Context);
4019 
4020     return ConstantExpr::getIntToPtr(ConstantExpr::getPtrToInt(C, MidTy),
4021                                      DestTy);
4022   }
4023 
4024   return nullptr;
4025 }
4026 
4027 /// Check the debug info version number, if it is out-dated, drop the debug
4028 /// info. Return true if module is modified.
UpgradeDebugInfo(Module & M)4029 bool llvm::UpgradeDebugInfo(Module &M) {
4030   unsigned Version = getDebugMetadataVersionFromModule(M);
4031   if (Version == DEBUG_METADATA_VERSION) {
4032     bool BrokenDebugInfo = false;
4033     if (verifyModule(M, &llvm::errs(), &BrokenDebugInfo))
4034       report_fatal_error("Broken module found, compilation aborted!");
4035     if (!BrokenDebugInfo)
4036       // Everything is ok.
4037       return false;
4038     else {
4039       // Diagnose malformed debug info.
4040       DiagnosticInfoIgnoringInvalidDebugMetadata Diag(M);
4041       M.getContext().diagnose(Diag);
4042     }
4043   }
4044   bool Modified = StripDebugInfo(M);
4045   if (Modified && Version != DEBUG_METADATA_VERSION) {
4046     // Diagnose a version mismatch.
4047     DiagnosticInfoDebugMetadataVersion DiagVersion(M, Version);
4048     M.getContext().diagnose(DiagVersion);
4049   }
4050   return Modified;
4051 }
4052 
4053 /// This checks for objc retain release marker which should be upgraded. It
4054 /// returns true if module is modified.
UpgradeRetainReleaseMarker(Module & M)4055 static bool UpgradeRetainReleaseMarker(Module &M) {
4056   bool Changed = false;
4057   const char *MarkerKey = "clang.arc.retainAutoreleasedReturnValueMarker";
4058   NamedMDNode *ModRetainReleaseMarker = M.getNamedMetadata(MarkerKey);
4059   if (ModRetainReleaseMarker) {
4060     MDNode *Op = ModRetainReleaseMarker->getOperand(0);
4061     if (Op) {
4062       MDString *ID = dyn_cast_or_null<MDString>(Op->getOperand(0));
4063       if (ID) {
4064         SmallVector<StringRef, 4> ValueComp;
4065         ID->getString().split(ValueComp, "#");
4066         if (ValueComp.size() == 2) {
4067           std::string NewValue = ValueComp[0].str() + ";" + ValueComp[1].str();
4068           ID = MDString::get(M.getContext(), NewValue);
4069         }
4070         M.addModuleFlag(Module::Error, MarkerKey, ID);
4071         M.eraseNamedMetadata(ModRetainReleaseMarker);
4072         Changed = true;
4073       }
4074     }
4075   }
4076   return Changed;
4077 }
4078 
UpgradeARCRuntime(Module & M)4079 void llvm::UpgradeARCRuntime(Module &M) {
4080   // This lambda converts normal function calls to ARC runtime functions to
4081   // intrinsic calls.
4082   auto UpgradeToIntrinsic = [&](const char *OldFunc,
4083                                 llvm::Intrinsic::ID IntrinsicFunc) {
4084     Function *Fn = M.getFunction(OldFunc);
4085 
4086     if (!Fn)
4087       return;
4088 
4089     Function *NewFn = llvm::Intrinsic::getDeclaration(&M, IntrinsicFunc);
4090 
4091     for (User *U : make_early_inc_range(Fn->users())) {
4092       CallInst *CI = dyn_cast<CallInst>(U);
4093       if (!CI || CI->getCalledFunction() != Fn)
4094         continue;
4095 
4096       IRBuilder<> Builder(CI->getParent(), CI->getIterator());
4097       FunctionType *NewFuncTy = NewFn->getFunctionType();
4098       SmallVector<Value *, 2> Args;
4099 
4100       // Don't upgrade the intrinsic if it's not valid to bitcast the return
4101       // value to the return type of the old function.
4102       if (NewFuncTy->getReturnType() != CI->getType() &&
4103           !CastInst::castIsValid(Instruction::BitCast, CI,
4104                                  NewFuncTy->getReturnType()))
4105         continue;
4106 
4107       bool InvalidCast = false;
4108 
4109       for (unsigned I = 0, E = CI->getNumArgOperands(); I != E; ++I) {
4110         Value *Arg = CI->getArgOperand(I);
4111 
4112         // Bitcast argument to the parameter type of the new function if it's
4113         // not a variadic argument.
4114         if (I < NewFuncTy->getNumParams()) {
4115           // Don't upgrade the intrinsic if it's not valid to bitcast the argument
4116           // to the parameter type of the new function.
4117           if (!CastInst::castIsValid(Instruction::BitCast, Arg,
4118                                      NewFuncTy->getParamType(I))) {
4119             InvalidCast = true;
4120             break;
4121           }
4122           Arg = Builder.CreateBitCast(Arg, NewFuncTy->getParamType(I));
4123         }
4124         Args.push_back(Arg);
4125       }
4126 
4127       if (InvalidCast)
4128         continue;
4129 
4130       // Create a call instruction that calls the new function.
4131       CallInst *NewCall = Builder.CreateCall(NewFuncTy, NewFn, Args);
4132       NewCall->setTailCallKind(cast<CallInst>(CI)->getTailCallKind());
4133       NewCall->takeName(CI);
4134 
4135       // Bitcast the return value back to the type of the old call.
4136       Value *NewRetVal = Builder.CreateBitCast(NewCall, CI->getType());
4137 
4138       if (!CI->use_empty())
4139         CI->replaceAllUsesWith(NewRetVal);
4140       CI->eraseFromParent();
4141     }
4142 
4143     if (Fn->use_empty())
4144       Fn->eraseFromParent();
4145   };
4146 
4147   // Unconditionally convert a call to "clang.arc.use" to a call to
4148   // "llvm.objc.clang.arc.use".
4149   UpgradeToIntrinsic("clang.arc.use", llvm::Intrinsic::objc_clang_arc_use);
4150 
4151   // Upgrade the retain release marker. If there is no need to upgrade
4152   // the marker, that means either the module is already new enough to contain
4153   // new intrinsics or it is not ARC. There is no need to upgrade runtime call.
4154   if (!UpgradeRetainReleaseMarker(M))
4155     return;
4156 
4157   std::pair<const char *, llvm::Intrinsic::ID> RuntimeFuncs[] = {
4158       {"objc_autorelease", llvm::Intrinsic::objc_autorelease},
4159       {"objc_autoreleasePoolPop", llvm::Intrinsic::objc_autoreleasePoolPop},
4160       {"objc_autoreleasePoolPush", llvm::Intrinsic::objc_autoreleasePoolPush},
4161       {"objc_autoreleaseReturnValue",
4162        llvm::Intrinsic::objc_autoreleaseReturnValue},
4163       {"objc_copyWeak", llvm::Intrinsic::objc_copyWeak},
4164       {"objc_destroyWeak", llvm::Intrinsic::objc_destroyWeak},
4165       {"objc_initWeak", llvm::Intrinsic::objc_initWeak},
4166       {"objc_loadWeak", llvm::Intrinsic::objc_loadWeak},
4167       {"objc_loadWeakRetained", llvm::Intrinsic::objc_loadWeakRetained},
4168       {"objc_moveWeak", llvm::Intrinsic::objc_moveWeak},
4169       {"objc_release", llvm::Intrinsic::objc_release},
4170       {"objc_retain", llvm::Intrinsic::objc_retain},
4171       {"objc_retainAutorelease", llvm::Intrinsic::objc_retainAutorelease},
4172       {"objc_retainAutoreleaseReturnValue",
4173        llvm::Intrinsic::objc_retainAutoreleaseReturnValue},
4174       {"objc_retainAutoreleasedReturnValue",
4175        llvm::Intrinsic::objc_retainAutoreleasedReturnValue},
4176       {"objc_retainBlock", llvm::Intrinsic::objc_retainBlock},
4177       {"objc_storeStrong", llvm::Intrinsic::objc_storeStrong},
4178       {"objc_storeWeak", llvm::Intrinsic::objc_storeWeak},
4179       {"objc_unsafeClaimAutoreleasedReturnValue",
4180        llvm::Intrinsic::objc_unsafeClaimAutoreleasedReturnValue},
4181       {"objc_retainedObject", llvm::Intrinsic::objc_retainedObject},
4182       {"objc_unretainedObject", llvm::Intrinsic::objc_unretainedObject},
4183       {"objc_unretainedPointer", llvm::Intrinsic::objc_unretainedPointer},
4184       {"objc_retain_autorelease", llvm::Intrinsic::objc_retain_autorelease},
4185       {"objc_sync_enter", llvm::Intrinsic::objc_sync_enter},
4186       {"objc_sync_exit", llvm::Intrinsic::objc_sync_exit},
4187       {"objc_arc_annotation_topdown_bbstart",
4188        llvm::Intrinsic::objc_arc_annotation_topdown_bbstart},
4189       {"objc_arc_annotation_topdown_bbend",
4190        llvm::Intrinsic::objc_arc_annotation_topdown_bbend},
4191       {"objc_arc_annotation_bottomup_bbstart",
4192        llvm::Intrinsic::objc_arc_annotation_bottomup_bbstart},
4193       {"objc_arc_annotation_bottomup_bbend",
4194        llvm::Intrinsic::objc_arc_annotation_bottomup_bbend}};
4195 
4196   for (auto &I : RuntimeFuncs)
4197     UpgradeToIntrinsic(I.first, I.second);
4198 }
4199 
UpgradeModuleFlags(Module & M)4200 bool llvm::UpgradeModuleFlags(Module &M) {
4201   NamedMDNode *ModFlags = M.getModuleFlagsMetadata();
4202   if (!ModFlags)
4203     return false;
4204 
4205   bool HasObjCFlag = false, HasClassProperties = false, Changed = false;
4206   bool HasSwiftVersionFlag = false;
4207   uint8_t SwiftMajorVersion, SwiftMinorVersion;
4208   uint32_t SwiftABIVersion;
4209   auto Int8Ty = Type::getInt8Ty(M.getContext());
4210   auto Int32Ty = Type::getInt32Ty(M.getContext());
4211 
4212   for (unsigned I = 0, E = ModFlags->getNumOperands(); I != E; ++I) {
4213     MDNode *Op = ModFlags->getOperand(I);
4214     if (Op->getNumOperands() != 3)
4215       continue;
4216     MDString *ID = dyn_cast_or_null<MDString>(Op->getOperand(1));
4217     if (!ID)
4218       continue;
4219     if (ID->getString() == "Objective-C Image Info Version")
4220       HasObjCFlag = true;
4221     if (ID->getString() == "Objective-C Class Properties")
4222       HasClassProperties = true;
4223     // Upgrade PIC/PIE Module Flags. The module flag behavior for these two
4224     // field was Error and now they are Max.
4225     if (ID->getString() == "PIC Level" || ID->getString() == "PIE Level") {
4226       if (auto *Behavior =
4227               mdconst::dyn_extract_or_null<ConstantInt>(Op->getOperand(0))) {
4228         if (Behavior->getLimitedValue() == Module::Error) {
4229           Type *Int32Ty = Type::getInt32Ty(M.getContext());
4230           Metadata *Ops[3] = {
4231               ConstantAsMetadata::get(ConstantInt::get(Int32Ty, Module::Max)),
4232               MDString::get(M.getContext(), ID->getString()),
4233               Op->getOperand(2)};
4234           ModFlags->setOperand(I, MDNode::get(M.getContext(), Ops));
4235           Changed = true;
4236         }
4237       }
4238     }
4239     // Upgrade Objective-C Image Info Section. Removed the whitespce in the
4240     // section name so that llvm-lto will not complain about mismatching
4241     // module flags that is functionally the same.
4242     if (ID->getString() == "Objective-C Image Info Section") {
4243       if (auto *Value = dyn_cast_or_null<MDString>(Op->getOperand(2))) {
4244         SmallVector<StringRef, 4> ValueComp;
4245         Value->getString().split(ValueComp, " ");
4246         if (ValueComp.size() != 1) {
4247           std::string NewValue;
4248           for (auto &S : ValueComp)
4249             NewValue += S.str();
4250           Metadata *Ops[3] = {Op->getOperand(0), Op->getOperand(1),
4251                               MDString::get(M.getContext(), NewValue)};
4252           ModFlags->setOperand(I, MDNode::get(M.getContext(), Ops));
4253           Changed = true;
4254         }
4255       }
4256     }
4257 
4258     // IRUpgrader turns a i32 type "Objective-C Garbage Collection" into i8 value.
4259     // If the higher bits are set, it adds new module flag for swift info.
4260     if (ID->getString() == "Objective-C Garbage Collection") {
4261       auto Md = dyn_cast<ConstantAsMetadata>(Op->getOperand(2));
4262       if (Md) {
4263         assert(Md->getValue() && "Expected non-empty metadata");
4264         auto Type = Md->getValue()->getType();
4265         if (Type == Int8Ty)
4266           continue;
4267         unsigned Val = Md->getValue()->getUniqueInteger().getZExtValue();
4268         if ((Val & 0xff) != Val) {
4269           HasSwiftVersionFlag = true;
4270           SwiftABIVersion = (Val & 0xff00) >> 8;
4271           SwiftMajorVersion = (Val & 0xff000000) >> 24;
4272           SwiftMinorVersion = (Val & 0xff0000) >> 16;
4273         }
4274         Metadata *Ops[3] = {
4275           ConstantAsMetadata::get(ConstantInt::get(Int32Ty,Module::Error)),
4276           Op->getOperand(1),
4277           ConstantAsMetadata::get(ConstantInt::get(Int8Ty,Val & 0xff))};
4278         ModFlags->setOperand(I, MDNode::get(M.getContext(), Ops));
4279         Changed = true;
4280       }
4281     }
4282   }
4283 
4284   // "Objective-C Class Properties" is recently added for Objective-C. We
4285   // upgrade ObjC bitcodes to contain a "Objective-C Class Properties" module
4286   // flag of value 0, so we can correclty downgrade this flag when trying to
4287   // link an ObjC bitcode without this module flag with an ObjC bitcode with
4288   // this module flag.
4289   if (HasObjCFlag && !HasClassProperties) {
4290     M.addModuleFlag(llvm::Module::Override, "Objective-C Class Properties",
4291                     (uint32_t)0);
4292     Changed = true;
4293   }
4294 
4295   if (HasSwiftVersionFlag) {
4296     M.addModuleFlag(Module::Error, "Swift ABI Version",
4297                     SwiftABIVersion);
4298     M.addModuleFlag(Module::Error, "Swift Major Version",
4299                     ConstantInt::get(Int8Ty, SwiftMajorVersion));
4300     M.addModuleFlag(Module::Error, "Swift Minor Version",
4301                     ConstantInt::get(Int8Ty, SwiftMinorVersion));
4302     Changed = true;
4303   }
4304 
4305   return Changed;
4306 }
4307 
UpgradeSectionAttributes(Module & M)4308 void llvm::UpgradeSectionAttributes(Module &M) {
4309   auto TrimSpaces = [](StringRef Section) -> std::string {
4310     SmallVector<StringRef, 5> Components;
4311     Section.split(Components, ',');
4312 
4313     SmallString<32> Buffer;
4314     raw_svector_ostream OS(Buffer);
4315 
4316     for (auto Component : Components)
4317       OS << ',' << Component.trim();
4318 
4319     return std::string(OS.str().substr(1));
4320   };
4321 
4322   for (auto &GV : M.globals()) {
4323     if (!GV.hasSection())
4324       continue;
4325 
4326     StringRef Section = GV.getSection();
4327 
4328     if (!Section.startswith("__DATA, __objc_catlist"))
4329       continue;
4330 
4331     // __DATA, __objc_catlist, regular, no_dead_strip
4332     // __DATA,__objc_catlist,regular,no_dead_strip
4333     GV.setSection(TrimSpaces(Section));
4334   }
4335 }
4336 
4337 namespace {
4338 // Prior to LLVM 10.0, the strictfp attribute could be used on individual
4339 // callsites within a function that did not also have the strictfp attribute.
4340 // Since 10.0, if strict FP semantics are needed within a function, the
4341 // function must have the strictfp attribute and all calls within the function
4342 // must also have the strictfp attribute. This latter restriction is
4343 // necessary to prevent unwanted libcall simplification when a function is
4344 // being cloned (such as for inlining).
4345 //
4346 // The "dangling" strictfp attribute usage was only used to prevent constant
4347 // folding and other libcall simplification. The nobuiltin attribute on the
4348 // callsite has the same effect.
4349 struct StrictFPUpgradeVisitor : public InstVisitor<StrictFPUpgradeVisitor> {
StrictFPUpgradeVisitor__anone4012b660411::StrictFPUpgradeVisitor4350   StrictFPUpgradeVisitor() {}
4351 
visitCallBase__anone4012b660411::StrictFPUpgradeVisitor4352   void visitCallBase(CallBase &Call) {
4353     if (!Call.isStrictFP())
4354       return;
4355     if (isa<ConstrainedFPIntrinsic>(&Call))
4356       return;
4357     // If we get here, the caller doesn't have the strictfp attribute
4358     // but this callsite does. Replace the strictfp attribute with nobuiltin.
4359     Call.removeAttribute(AttributeList::FunctionIndex, Attribute::StrictFP);
4360     Call.addAttribute(AttributeList::FunctionIndex, Attribute::NoBuiltin);
4361   }
4362 };
4363 } // namespace
4364 
UpgradeFunctionAttributes(Function & F)4365 void llvm::UpgradeFunctionAttributes(Function &F) {
4366   // If a function definition doesn't have the strictfp attribute,
4367   // convert any callsite strictfp attributes to nobuiltin.
4368   if (!F.isDeclaration() && !F.hasFnAttribute(Attribute::StrictFP)) {
4369     StrictFPUpgradeVisitor SFPV;
4370     SFPV.visit(F);
4371   }
4372 
4373   if (F.getCallingConv() == CallingConv::X86_INTR &&
4374       !F.arg_empty() && !F.hasParamAttribute(0, Attribute::ByVal)) {
4375     Type *ByValTy = cast<PointerType>(F.getArg(0)->getType())->getElementType();
4376     Attribute NewAttr = Attribute::getWithByValType(F.getContext(), ByValTy);
4377     F.addParamAttr(0, NewAttr);
4378   }
4379 
4380   // Remove all incompatibile attributes from function.
4381   F.removeAttributes(AttributeList::ReturnIndex,
4382                      AttributeFuncs::typeIncompatible(F.getReturnType()));
4383   for (auto &Arg : F.args())
4384     Arg.removeAttrs(AttributeFuncs::typeIncompatible(Arg.getType()));
4385 }
4386 
isOldLoopArgument(Metadata * MD)4387 static bool isOldLoopArgument(Metadata *MD) {
4388   auto *T = dyn_cast_or_null<MDTuple>(MD);
4389   if (!T)
4390     return false;
4391   if (T->getNumOperands() < 1)
4392     return false;
4393   auto *S = dyn_cast_or_null<MDString>(T->getOperand(0));
4394   if (!S)
4395     return false;
4396   return S->getString().startswith("llvm.vectorizer.");
4397 }
4398 
upgradeLoopTag(LLVMContext & C,StringRef OldTag)4399 static MDString *upgradeLoopTag(LLVMContext &C, StringRef OldTag) {
4400   StringRef OldPrefix = "llvm.vectorizer.";
4401   assert(OldTag.startswith(OldPrefix) && "Expected old prefix");
4402 
4403   if (OldTag == "llvm.vectorizer.unroll")
4404     return MDString::get(C, "llvm.loop.interleave.count");
4405 
4406   return MDString::get(
4407       C, (Twine("llvm.loop.vectorize.") + OldTag.drop_front(OldPrefix.size()))
4408              .str());
4409 }
4410 
upgradeLoopArgument(Metadata * MD)4411 static Metadata *upgradeLoopArgument(Metadata *MD) {
4412   auto *T = dyn_cast_or_null<MDTuple>(MD);
4413   if (!T)
4414     return MD;
4415   if (T->getNumOperands() < 1)
4416     return MD;
4417   auto *OldTag = dyn_cast_or_null<MDString>(T->getOperand(0));
4418   if (!OldTag)
4419     return MD;
4420   if (!OldTag->getString().startswith("llvm.vectorizer."))
4421     return MD;
4422 
4423   // This has an old tag.  Upgrade it.
4424   SmallVector<Metadata *, 8> Ops;
4425   Ops.reserve(T->getNumOperands());
4426   Ops.push_back(upgradeLoopTag(T->getContext(), OldTag->getString()));
4427   for (unsigned I = 1, E = T->getNumOperands(); I != E; ++I)
4428     Ops.push_back(T->getOperand(I));
4429 
4430   return MDTuple::get(T->getContext(), Ops);
4431 }
4432 
upgradeInstructionLoopAttachment(MDNode & N)4433 MDNode *llvm::upgradeInstructionLoopAttachment(MDNode &N) {
4434   auto *T = dyn_cast<MDTuple>(&N);
4435   if (!T)
4436     return &N;
4437 
4438   if (none_of(T->operands(), isOldLoopArgument))
4439     return &N;
4440 
4441   SmallVector<Metadata *, 8> Ops;
4442   Ops.reserve(T->getNumOperands());
4443   for (Metadata *MD : T->operands())
4444     Ops.push_back(upgradeLoopArgument(MD));
4445 
4446   return MDTuple::get(T->getContext(), Ops);
4447 }
4448 
UpgradeDataLayoutString(StringRef DL,StringRef TT)4449 std::string llvm::UpgradeDataLayoutString(StringRef DL, StringRef TT) {
4450   Triple T(TT);
4451   // For AMDGPU we uprgrade older DataLayouts to include the default globals
4452   // address space of 1.
4453   if (T.isAMDGPU() && !DL.contains("-G") && !DL.startswith("G")) {
4454     return DL.empty() ? std::string("G1") : (DL + "-G1").str();
4455   }
4456 
4457   std::string AddrSpaces = "-p270:32:32-p271:32:32-p272:64:64";
4458   // If X86, and the datalayout matches the expected format, add pointer size
4459   // address spaces to the datalayout.
4460   if (!T.isX86() || DL.contains(AddrSpaces))
4461     return std::string(DL);
4462 
4463   SmallVector<StringRef, 4> Groups;
4464   Regex R("(e-m:[a-z](-p:32:32)?)(-[if]64:.*$)");
4465   if (!R.match(DL, &Groups))
4466     return std::string(DL);
4467 
4468   return (Groups[1] + AddrSpaces + Groups[3]).str();
4469 }
4470 
UpgradeAttributes(AttrBuilder & B)4471 void llvm::UpgradeAttributes(AttrBuilder &B) {
4472   StringRef FramePointer;
4473   if (B.contains("no-frame-pointer-elim")) {
4474     // The value can be "true" or "false".
4475     for (const auto &I : B.td_attrs())
4476       if (I.first == "no-frame-pointer-elim")
4477         FramePointer = I.second == "true" ? "all" : "none";
4478     B.removeAttribute("no-frame-pointer-elim");
4479   }
4480   if (B.contains("no-frame-pointer-elim-non-leaf")) {
4481     // The value is ignored. "no-frame-pointer-elim"="true" takes priority.
4482     if (FramePointer != "all")
4483       FramePointer = "non-leaf";
4484     B.removeAttribute("no-frame-pointer-elim-non-leaf");
4485   }
4486   if (!FramePointer.empty())
4487     B.addAttribute("frame-pointer", FramePointer);
4488 
4489   if (B.contains("null-pointer-is-valid")) {
4490     // The value can be "true" or "false".
4491     bool NullPointerIsValid = false;
4492     for (const auto &I : B.td_attrs())
4493       if (I.first == "null-pointer-is-valid")
4494         NullPointerIsValid = I.second == "true";
4495     B.removeAttribute("null-pointer-is-valid");
4496     if (NullPointerIsValid)
4497       B.addAttribute(Attribute::NullPointerIsValid);
4498   }
4499 }
4500