blob: 32ff7af32f078f919e9ae0a8b9c4c18183752133 [file] [log] [blame]
Chandler Carruth7132e002007-08-04 01:51:18 +00001//===-- AutoUpgrade.cpp - Implement auto-upgrade helper functions ---------===//
2//
3// The LLVM Compiler Infrastructure
4//
Chris Lattnerf3ebc3f2007-12-29 20:36:04 +00005// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
Chandler Carruth7132e002007-08-04 01:51:18 +00007//
8//===----------------------------------------------------------------------===//
9//
Sanjay Patel19792fb2015-03-10 16:08:36 +000010// This file implements the auto-upgrade helper functions.
11// This is where deprecated IR intrinsics and other IR features are updated to
12// current specifications.
Chandler Carruth7132e002007-08-04 01:51:18 +000013//
14//===----------------------------------------------------------------------===//
15
Chandler Carruth91065212014-03-05 10:34:14 +000016#include "llvm/IR/AutoUpgrade.h"
Justin Lebar46624a82017-01-21 01:00:32 +000017#include "llvm/ADT/StringSwitch.h"
Chandler Carruth9fb823b2013-01-02 11:36:10 +000018#include "llvm/IR/Constants.h"
Chandler Carruthd9903882015-01-14 11:23:27 +000019#include "llvm/IR/DIBuilder.h"
Chandler Carruth9a4c9e52014-03-06 00:46:21 +000020#include "llvm/IR/DebugInfo.h"
Manman Ren2ebfb422014-01-16 01:51:12 +000021#include "llvm/IR/DiagnosticInfo.h"
Chandler Carruth9fb823b2013-01-02 11:36:10 +000022#include "llvm/IR/Function.h"
23#include "llvm/IR/IRBuilder.h"
24#include "llvm/IR/Instruction.h"
Daniel Neilson1e687242018-01-19 17:13:12 +000025#include "llvm/IR/IntrinsicInst.h"
Chandler Carruth9fb823b2013-01-02 11:36:10 +000026#include "llvm/IR/LLVMContext.h"
27#include "llvm/IR/Module.h"
Adrian Prantla8b2ddb2017-10-02 18:31:29 +000028#include "llvm/IR/Verifier.h"
Torok Edwin56d06592009-07-11 20:10:48 +000029#include "llvm/Support/ErrorHandling.h"
Jeroen Ketemaab99b592015-09-30 10:56:37 +000030#include "llvm/Support/Regex.h"
Anton Korobeynikov579f0712008-02-20 11:08:44 +000031#include <cstring>
Chandler Carruth7132e002007-08-04 01:51:18 +000032using namespace llvm;
33
Rafael Espindolad7325ee2016-10-03 15:51:42 +000034static void rename(GlobalValue *GV) { GV->setName(GV->getName() + ".old"); }
35
Craig Topper905cc752017-02-17 07:07:21 +000036// Upgrade the declarations of the SSE4.1 ptest intrinsics whose arguments have
Nadav Rotem17ee58a2012-06-10 18:42:51 +000037// changed their type from v4f32 to v2i64.
Craig Topper905cc752017-02-17 07:07:21 +000038static bool UpgradePTESTIntrinsic(Function* F, Intrinsic::ID IID,
39 Function *&NewFn) {
Nadav Rotem17ee58a2012-06-10 18:42:51 +000040 // Check whether this is an old version of the function, which received
41 // v4f32 arguments.
42 Type *Arg0Type = F->getFunctionType()->getParamType(0);
43 if (Arg0Type != VectorType::get(Type::getFloatTy(F->getContext()), 4))
44 return false;
45
46 // Yes, it's old, replace it with new version.
Rafael Espindolad7325ee2016-10-03 15:51:42 +000047 rename(F);
Nadav Rotem17ee58a2012-06-10 18:42:51 +000048 NewFn = Intrinsic::getDeclaration(F->getParent(), IID);
49 return true;
50}
Chandler Carruth7132e002007-08-04 01:51:18 +000051
Chandler Carruth373b2b12014-09-06 10:00:01 +000052// Upgrade the declarations of intrinsic functions whose 8-bit immediate mask
53// arguments have changed their type from i32 to i8.
54static bool UpgradeX86IntrinsicsWith8BitMask(Function *F, Intrinsic::ID IID,
55 Function *&NewFn) {
56 // Check that the last argument is an i32.
57 Type *LastArgType = F->getFunctionType()->getParamType(
58 F->getFunctionType()->getNumParams() - 1);
59 if (!LastArgType->isIntegerTy(32))
60 return false;
61
62 // Move this function aside and map down.
Rafael Espindolad7325ee2016-10-03 15:51:42 +000063 rename(F);
Chandler Carruth373b2b12014-09-06 10:00:01 +000064 NewFn = Intrinsic::getDeclaration(F->getParent(), IID);
65 return true;
66}
67
Craig Topper4dccffc82018-02-10 23:33:55 +000068// Upgrade the declaration of fp compare intrinsics that change return type
69// from scalar to vXi1 mask.
70static bool UpgradeX86MaskedFPCompare(Function *F, Intrinsic::ID IID,
71 Function *&NewFn) {
72 // Check if the return type is a vector.
73 if (F->getReturnType()->isVectorTy())
74 return false;
75
76 rename(F);
77 NewFn = Intrinsic::getDeclaration(F->getParent(), IID);
78 return true;
79}
80
Craig Topperb9b9cb02017-02-17 07:07:19 +000081static bool ShouldUpgradeX86Intrinsic(Function *F, StringRef Name) {
82 // All of the intrinsics matches below should be marked with which llvm
83 // version started autoupgrading them. At some point in the future we would
84 // like to use this information to remove upgrade code for some older
85 // intrinsics. It is currently undecided how we will determine that future
86 // point.
Uriel Korach5d5da5f2017-09-13 09:02:36 +000087 if (Name=="ssse3.pabs.b.128" || // Added in 6.0
88 Name=="ssse3.pabs.w.128" || // Added in 6.0
89 Name=="ssse3.pabs.d.128" || // Added in 6.0
Jina Nahias9a7f9f12017-11-13 09:16:39 +000090 Name.startswith("avx512.mask.shuf.i") || // Added in 6.0
91 Name.startswith("avx512.mask.shuf.f") || // Added in 6.0
Jina Nahias51c1a622017-12-05 15:42:56 +000092 Name.startswith("avx512.kunpck") || //added in 6.0
Uriel Korach5d5da5f2017-09-13 09:02:36 +000093 Name.startswith("avx2.pabs.") || // Added in 6.0
94 Name.startswith("avx512.mask.pabs.") || // Added in 6.0
Jina Nahias7b705f12017-11-06 07:09:24 +000095 Name.startswith("avx512.broadcastm") || // Added in 6.0
Jina Nahiasccfb8d42017-09-19 11:03:06 +000096 Name.startswith("avx512.mask.pbroadcast") || // Added in 6.0
Uriel Korach5d5da5f2017-09-13 09:02:36 +000097 Name.startswith("sse2.pcmpeq.") || // Added in 3.1
Craig Topperb9b9cb02017-02-17 07:07:19 +000098 Name.startswith("sse2.pcmpgt.") || // Added in 3.1
99 Name.startswith("avx2.pcmpeq.") || // Added in 3.1
100 Name.startswith("avx2.pcmpgt.") || // Added in 3.1
101 Name.startswith("avx512.mask.pcmpeq.") || // Added in 3.9
102 Name.startswith("avx512.mask.pcmpgt.") || // Added in 3.9
Craig Topperf264fcc2017-09-16 07:36:14 +0000103 Name.startswith("avx.vperm2f128.") || // Added in 6.0
104 Name == "avx2.vperm2i128" || // Added in 6.0
Craig Topperb9b9cb02017-02-17 07:07:19 +0000105 Name == "sse.add.ss" || // Added in 4.0
106 Name == "sse2.add.sd" || // Added in 4.0
107 Name == "sse.sub.ss" || // Added in 4.0
108 Name == "sse2.sub.sd" || // Added in 4.0
109 Name == "sse.mul.ss" || // Added in 4.0
110 Name == "sse2.mul.sd" || // Added in 4.0
111 Name == "sse.div.ss" || // Added in 4.0
112 Name == "sse2.div.sd" || // Added in 4.0
113 Name == "sse41.pmaxsb" || // Added in 3.9
114 Name == "sse2.pmaxs.w" || // Added in 3.9
115 Name == "sse41.pmaxsd" || // Added in 3.9
116 Name == "sse2.pmaxu.b" || // Added in 3.9
117 Name == "sse41.pmaxuw" || // Added in 3.9
118 Name == "sse41.pmaxud" || // Added in 3.9
119 Name == "sse41.pminsb" || // Added in 3.9
120 Name == "sse2.pmins.w" || // Added in 3.9
121 Name == "sse41.pminsd" || // Added in 3.9
122 Name == "sse2.pminu.b" || // Added in 3.9
123 Name == "sse41.pminuw" || // Added in 3.9
124 Name == "sse41.pminud" || // Added in 3.9
Craig Topper071ad9c2018-02-03 20:18:25 +0000125 Name == "avx512.kand.w" || // Added in 7.0
126 Name == "avx512.kandn.w" || // Added in 7.0
127 Name == "avx512.knot.w" || // Added in 7.0
128 Name == "avx512.kor.w" || // Added in 7.0
129 Name == "avx512.kxor.w" || // Added in 7.0
130 Name == "avx512.kxnor.w" || // Added in 7.0
Craig Topperdccf72b2018-02-08 20:16:06 +0000131 Name == "avx512.kortestc.w" || // Added in 7.0
132 Name == "avx512.kortestz.w" || // Added in 7.0
Craig Topperb9b9cb02017-02-17 07:07:19 +0000133 Name.startswith("avx512.mask.pshuf.b.") || // Added in 4.0
134 Name.startswith("avx2.pmax") || // Added in 3.9
135 Name.startswith("avx2.pmin") || // Added in 3.9
136 Name.startswith("avx512.mask.pmax") || // Added in 4.0
137 Name.startswith("avx512.mask.pmin") || // Added in 4.0
138 Name.startswith("avx2.vbroadcast") || // Added in 3.8
139 Name.startswith("avx2.pbroadcast") || // Added in 3.8
140 Name.startswith("avx.vpermil.") || // Added in 3.1
141 Name.startswith("sse2.pshuf") || // Added in 3.9
142 Name.startswith("avx512.pbroadcast") || // Added in 3.9
143 Name.startswith("avx512.mask.broadcast.s") || // Added in 3.9
144 Name.startswith("avx512.mask.movddup") || // Added in 3.9
145 Name.startswith("avx512.mask.movshdup") || // Added in 3.9
146 Name.startswith("avx512.mask.movsldup") || // Added in 3.9
147 Name.startswith("avx512.mask.pshuf.d.") || // Added in 3.9
148 Name.startswith("avx512.mask.pshufl.w.") || // Added in 3.9
149 Name.startswith("avx512.mask.pshufh.w.") || // Added in 3.9
150 Name.startswith("avx512.mask.shuf.p") || // Added in 4.0
151 Name.startswith("avx512.mask.vpermil.p") || // Added in 3.9
152 Name.startswith("avx512.mask.perm.df.") || // Added in 3.9
153 Name.startswith("avx512.mask.perm.di.") || // Added in 3.9
154 Name.startswith("avx512.mask.punpckl") || // Added in 3.9
155 Name.startswith("avx512.mask.punpckh") || // Added in 3.9
156 Name.startswith("avx512.mask.unpckl.") || // Added in 3.9
157 Name.startswith("avx512.mask.unpckh.") || // Added in 3.9
158 Name.startswith("avx512.mask.pand.") || // Added in 3.9
159 Name.startswith("avx512.mask.pandn.") || // Added in 3.9
160 Name.startswith("avx512.mask.por.") || // Added in 3.9
161 Name.startswith("avx512.mask.pxor.") || // Added in 3.9
162 Name.startswith("avx512.mask.and.") || // Added in 3.9
163 Name.startswith("avx512.mask.andn.") || // Added in 3.9
164 Name.startswith("avx512.mask.or.") || // Added in 3.9
165 Name.startswith("avx512.mask.xor.") || // Added in 3.9
166 Name.startswith("avx512.mask.padd.") || // Added in 4.0
167 Name.startswith("avx512.mask.psub.") || // Added in 4.0
168 Name.startswith("avx512.mask.pmull.") || // Added in 4.0
169 Name.startswith("avx512.mask.cvtdq2pd.") || // Added in 4.0
170 Name.startswith("avx512.mask.cvtudq2pd.") || // Added in 4.0
171 Name.startswith("avx512.mask.pmul.dq.") || // Added in 4.0
172 Name.startswith("avx512.mask.pmulu.dq.") || // Added in 4.0
Craig Topperc43f3f32017-02-24 05:35:07 +0000173 Name.startswith("avx512.mask.packsswb.") || // Added in 5.0
174 Name.startswith("avx512.mask.packssdw.") || // Added in 5.0
175 Name.startswith("avx512.mask.packuswb.") || // Added in 5.0
176 Name.startswith("avx512.mask.packusdw.") || // Added in 5.0
Craig Topper792fc922017-06-22 20:11:01 +0000177 Name.startswith("avx512.mask.cmp.b") || // Added in 5.0
178 Name.startswith("avx512.mask.cmp.d") || // Added in 5.0
179 Name.startswith("avx512.mask.cmp.q") || // Added in 5.0
180 Name.startswith("avx512.mask.cmp.w") || // Added in 5.0
181 Name.startswith("avx512.mask.ucmp.") || // Added in 5.0
Craig Toppercc342d42018-01-09 00:50:47 +0000182 Name.startswith("avx512.cvtb2mask.") || // Added in 7.0
183 Name.startswith("avx512.cvtw2mask.") || // Added in 7.0
184 Name.startswith("avx512.cvtd2mask.") || // Added in 7.0
185 Name.startswith("avx512.cvtq2mask.") || // Added in 7.0
Craig Topperb9b9cb02017-02-17 07:07:19 +0000186 Name == "avx512.mask.add.pd.128" || // Added in 4.0
187 Name == "avx512.mask.add.pd.256" || // Added in 4.0
188 Name == "avx512.mask.add.ps.128" || // Added in 4.0
189 Name == "avx512.mask.add.ps.256" || // Added in 4.0
190 Name == "avx512.mask.div.pd.128" || // Added in 4.0
191 Name == "avx512.mask.div.pd.256" || // Added in 4.0
192 Name == "avx512.mask.div.ps.128" || // Added in 4.0
193 Name == "avx512.mask.div.ps.256" || // Added in 4.0
194 Name == "avx512.mask.mul.pd.128" || // Added in 4.0
195 Name == "avx512.mask.mul.pd.256" || // Added in 4.0
196 Name == "avx512.mask.mul.ps.128" || // Added in 4.0
197 Name == "avx512.mask.mul.ps.256" || // Added in 4.0
198 Name == "avx512.mask.sub.pd.128" || // Added in 4.0
199 Name == "avx512.mask.sub.pd.256" || // Added in 4.0
200 Name == "avx512.mask.sub.ps.128" || // Added in 4.0
201 Name == "avx512.mask.sub.ps.256" || // Added in 4.0
Craig Topperc43f3f32017-02-24 05:35:07 +0000202 Name == "avx512.mask.max.pd.128" || // Added in 5.0
203 Name == "avx512.mask.max.pd.256" || // Added in 5.0
204 Name == "avx512.mask.max.ps.128" || // Added in 5.0
205 Name == "avx512.mask.max.ps.256" || // Added in 5.0
206 Name == "avx512.mask.min.pd.128" || // Added in 5.0
207 Name == "avx512.mask.min.pd.256" || // Added in 5.0
208 Name == "avx512.mask.min.ps.128" || // Added in 5.0
209 Name == "avx512.mask.min.ps.256" || // Added in 5.0
Craig Topperb9b9cb02017-02-17 07:07:19 +0000210 Name.startswith("avx512.mask.vpermilvar.") || // Added in 4.0
211 Name.startswith("avx512.mask.psll.d") || // Added in 4.0
212 Name.startswith("avx512.mask.psll.q") || // Added in 4.0
213 Name.startswith("avx512.mask.psll.w") || // Added in 4.0
214 Name.startswith("avx512.mask.psra.d") || // Added in 4.0
215 Name.startswith("avx512.mask.psra.q") || // Added in 4.0
216 Name.startswith("avx512.mask.psra.w") || // Added in 4.0
217 Name.startswith("avx512.mask.psrl.d") || // Added in 4.0
218 Name.startswith("avx512.mask.psrl.q") || // Added in 4.0
219 Name.startswith("avx512.mask.psrl.w") || // Added in 4.0
220 Name.startswith("avx512.mask.pslli") || // Added in 4.0
221 Name.startswith("avx512.mask.psrai") || // Added in 4.0
222 Name.startswith("avx512.mask.psrli") || // Added in 4.0
223 Name.startswith("avx512.mask.psllv") || // Added in 4.0
224 Name.startswith("avx512.mask.psrav") || // Added in 4.0
225 Name.startswith("avx512.mask.psrlv") || // Added in 4.0
226 Name.startswith("sse41.pmovsx") || // Added in 3.8
227 Name.startswith("sse41.pmovzx") || // Added in 3.9
228 Name.startswith("avx2.pmovsx") || // Added in 3.9
229 Name.startswith("avx2.pmovzx") || // Added in 3.9
230 Name.startswith("avx512.mask.pmovsx") || // Added in 4.0
231 Name.startswith("avx512.mask.pmovzx") || // Added in 4.0
Craig Topperf2529c12017-02-24 05:35:04 +0000232 Name.startswith("avx512.mask.lzcnt.") || // Added in 5.0
Craig Topperb9b9cb02017-02-17 07:07:19 +0000233 Name == "sse2.cvtdq2pd" || // Added in 3.9
234 Name == "sse2.cvtps2pd" || // Added in 3.9
235 Name == "avx.cvtdq2.pd.256" || // Added in 3.9
236 Name == "avx.cvt.ps2.pd.256" || // Added in 3.9
237 Name.startswith("avx.vinsertf128.") || // Added in 3.7
238 Name == "avx2.vinserti128" || // Added in 3.7
239 Name.startswith("avx512.mask.insert") || // Added in 4.0
240 Name.startswith("avx.vextractf128.") || // Added in 3.7
241 Name == "avx2.vextracti128" || // Added in 3.7
242 Name.startswith("avx512.mask.vextract") || // Added in 4.0
243 Name.startswith("sse4a.movnt.") || // Added in 3.9
244 Name.startswith("avx.movnt.") || // Added in 3.2
245 Name.startswith("avx512.storent.") || // Added in 3.9
Simon Pilgrim5a22eaa2017-04-14 15:05:35 +0000246 Name == "sse41.movntdqa" || // Added in 5.0
247 Name == "avx2.movntdqa" || // Added in 5.0
248 Name == "avx512.movntdqa" || // Added in 5.0
Craig Topperb9b9cb02017-02-17 07:07:19 +0000249 Name == "sse2.storel.dq" || // Added in 3.9
250 Name.startswith("sse.storeu.") || // Added in 3.9
251 Name.startswith("sse2.storeu.") || // Added in 3.9
252 Name.startswith("avx.storeu.") || // Added in 3.9
253 Name.startswith("avx512.mask.storeu.") || // Added in 3.9
254 Name.startswith("avx512.mask.store.p") || // Added in 3.9
255 Name.startswith("avx512.mask.store.b.") || // Added in 3.9
256 Name.startswith("avx512.mask.store.w.") || // Added in 3.9
257 Name.startswith("avx512.mask.store.d.") || // Added in 3.9
258 Name.startswith("avx512.mask.store.q.") || // Added in 3.9
259 Name.startswith("avx512.mask.loadu.") || // Added in 3.9
260 Name.startswith("avx512.mask.load.") || // Added in 3.9
261 Name == "sse42.crc32.64.8" || // Added in 3.4
262 Name.startswith("avx.vbroadcast.s") || // Added in 3.5
263 Name.startswith("avx512.mask.palignr.") || // Added in 3.9
264 Name.startswith("avx512.mask.valign.") || // Added in 4.0
265 Name.startswith("sse2.psll.dq") || // Added in 3.7
266 Name.startswith("sse2.psrl.dq") || // Added in 3.7
267 Name.startswith("avx2.psll.dq") || // Added in 3.7
268 Name.startswith("avx2.psrl.dq") || // Added in 3.7
269 Name.startswith("avx512.psll.dq") || // Added in 3.9
270 Name.startswith("avx512.psrl.dq") || // Added in 3.9
271 Name == "sse41.pblendw" || // Added in 3.7
272 Name.startswith("sse41.blendp") || // Added in 3.7
273 Name.startswith("avx.blend.p") || // Added in 3.7
274 Name == "avx2.pblendw" || // Added in 3.7
275 Name.startswith("avx2.pblendd.") || // Added in 3.7
276 Name.startswith("avx.vbroadcastf128") || // Added in 4.0
277 Name == "avx2.vbroadcasti128" || // Added in 3.7
Uriel Korach0ecc9842017-09-26 07:39:39 +0000278 Name.startswith("avx512.mask.broadcastf") || // Added in 6.0
279 Name.startswith("avx512.mask.broadcasti") || // Added in 6.0
Craig Topper185ced82017-02-23 03:22:14 +0000280 Name == "xop.vpcmov" || // Added in 3.8
Craig Topperc43f3f32017-02-24 05:35:07 +0000281 Name == "xop.vpcmov.256" || // Added in 5.0
Craig Topperb9b9cb02017-02-17 07:07:19 +0000282 Name.startswith("avx512.mask.move.s") || // Added in 4.0
Michael Zuckerman88fb1712017-04-04 13:32:14 +0000283 Name.startswith("avx512.cvtmask2") || // Added in 5.0
Craig Topperb9b9cb02017-02-17 07:07:19 +0000284 (Name.startswith("xop.vpcom") && // Added in 3.2
Yael Tsafrir47668b52017-09-12 07:50:35 +0000285 F->arg_size() == 2) ||
Uriel Korach2aa707b2017-11-13 12:51:18 +0000286 Name.startswith("avx512.ptestm") || //Added in 6.0
287 Name.startswith("avx512.ptestnm") || //Added in 6.0
Yael Tsafrir47668b52017-09-12 07:50:35 +0000288 Name.startswith("sse2.pavg") || // Added in 6.0
289 Name.startswith("avx2.pavg") || // Added in 6.0
290 Name.startswith("avx512.mask.pavg")) // Added in 6.0
Craig Topperb9b9cb02017-02-17 07:07:19 +0000291 return true;
292
293 return false;
294}
295
296static bool UpgradeX86IntrinsicFunction(Function *F, StringRef Name,
297 Function *&NewFn) {
298 // Only handle intrinsics that start with "x86.".
299 if (!Name.startswith("x86."))
300 return false;
301 // Remove "x86." prefix.
302 Name = Name.substr(4);
303
304 if (ShouldUpgradeX86Intrinsic(F, Name)) {
305 NewFn = nullptr;
306 return true;
307 }
308
309 // SSE4.1 ptest functions may have an old signature.
310 if (Name.startswith("sse41.ptest")) { // Added in 3.2
311 if (Name.substr(11) == "c")
Craig Topper905cc752017-02-17 07:07:21 +0000312 return UpgradePTESTIntrinsic(F, Intrinsic::x86_sse41_ptestc, NewFn);
Craig Topperb9b9cb02017-02-17 07:07:19 +0000313 if (Name.substr(11) == "z")
Craig Topper905cc752017-02-17 07:07:21 +0000314 return UpgradePTESTIntrinsic(F, Intrinsic::x86_sse41_ptestz, NewFn);
Craig Topperb9b9cb02017-02-17 07:07:19 +0000315 if (Name.substr(11) == "nzc")
Craig Topper905cc752017-02-17 07:07:21 +0000316 return UpgradePTESTIntrinsic(F, Intrinsic::x86_sse41_ptestnzc, NewFn);
Craig Topperb9b9cb02017-02-17 07:07:19 +0000317 }
318 // Several blend and other instructions with masks used the wrong number of
319 // bits.
320 if (Name == "sse41.insertps") // Added in 3.6
321 return UpgradeX86IntrinsicsWith8BitMask(F, Intrinsic::x86_sse41_insertps,
322 NewFn);
323 if (Name == "sse41.dppd") // Added in 3.6
324 return UpgradeX86IntrinsicsWith8BitMask(F, Intrinsic::x86_sse41_dppd,
325 NewFn);
326 if (Name == "sse41.dpps") // Added in 3.6
327 return UpgradeX86IntrinsicsWith8BitMask(F, Intrinsic::x86_sse41_dpps,
328 NewFn);
329 if (Name == "sse41.mpsadbw") // Added in 3.6
330 return UpgradeX86IntrinsicsWith8BitMask(F, Intrinsic::x86_sse41_mpsadbw,
331 NewFn);
332 if (Name == "avx.dp.ps.256") // Added in 3.6
333 return UpgradeX86IntrinsicsWith8BitMask(F, Intrinsic::x86_avx_dp_ps_256,
334 NewFn);
335 if (Name == "avx2.mpsadbw") // Added in 3.6
336 return UpgradeX86IntrinsicsWith8BitMask(F, Intrinsic::x86_avx2_mpsadbw,
337 NewFn);
Craig Topper4dccffc82018-02-10 23:33:55 +0000338 if (Name == "avx512.mask.cmp.pd.128") // Added in 7.0
339 return UpgradeX86MaskedFPCompare(F, Intrinsic::x86_avx512_mask_cmp_pd_128,
340 NewFn);
341 if (Name == "avx512.mask.cmp.pd.256") // Added in 7.0
342 return UpgradeX86MaskedFPCompare(F, Intrinsic::x86_avx512_mask_cmp_pd_256,
343 NewFn);
344 if (Name == "avx512.mask.cmp.pd.512") // Added in 7.0
345 return UpgradeX86MaskedFPCompare(F, Intrinsic::x86_avx512_mask_cmp_pd_512,
346 NewFn);
347 if (Name == "avx512.mask.cmp.ps.128") // Added in 7.0
348 return UpgradeX86MaskedFPCompare(F, Intrinsic::x86_avx512_mask_cmp_ps_128,
349 NewFn);
350 if (Name == "avx512.mask.cmp.ps.256") // Added in 7.0
351 return UpgradeX86MaskedFPCompare(F, Intrinsic::x86_avx512_mask_cmp_ps_256,
352 NewFn);
353 if (Name == "avx512.mask.cmp.ps.512") // Added in 7.0
354 return UpgradeX86MaskedFPCompare(F, Intrinsic::x86_avx512_mask_cmp_ps_512,
355 NewFn);
Craig Topperb9b9cb02017-02-17 07:07:19 +0000356
357 // frcz.ss/sd may need to have an argument dropped. Added in 3.2
358 if (Name.startswith("xop.vfrcz.ss") && F->arg_size() == 2) {
359 rename(F);
360 NewFn = Intrinsic::getDeclaration(F->getParent(),
361 Intrinsic::x86_xop_vfrcz_ss);
362 return true;
363 }
364 if (Name.startswith("xop.vfrcz.sd") && F->arg_size() == 2) {
365 rename(F);
366 NewFn = Intrinsic::getDeclaration(F->getParent(),
367 Intrinsic::x86_xop_vfrcz_sd);
368 return true;
369 }
370 // Upgrade any XOP PERMIL2 index operand still using a float/double vector.
371 if (Name.startswith("xop.vpermil2")) { // Added in 3.9
Craig Toppercbd1b602017-02-17 07:07:24 +0000372 auto Idx = F->getFunctionType()->getParamType(2);
373 if (Idx->isFPOrFPVectorTy()) {
Craig Topperb9b9cb02017-02-17 07:07:19 +0000374 rename(F);
375 unsigned IdxSize = Idx->getPrimitiveSizeInBits();
376 unsigned EltSize = Idx->getScalarSizeInBits();
377 Intrinsic::ID Permil2ID;
378 if (EltSize == 64 && IdxSize == 128)
379 Permil2ID = Intrinsic::x86_xop_vpermil2pd;
380 else if (EltSize == 32 && IdxSize == 128)
381 Permil2ID = Intrinsic::x86_xop_vpermil2ps;
382 else if (EltSize == 64 && IdxSize == 256)
383 Permil2ID = Intrinsic::x86_xop_vpermil2pd_256;
384 else
385 Permil2ID = Intrinsic::x86_xop_vpermil2ps_256;
386 NewFn = Intrinsic::getDeclaration(F->getParent(), Permil2ID);
387 return true;
388 }
389 }
390
391 return false;
392}
393
Evan Cheng0e179d02007-12-17 22:33:23 +0000394static bool UpgradeIntrinsicFunction1(Function *F, Function *&NewFn) {
Chandler Carruth7132e002007-08-04 01:51:18 +0000395 assert(F && "Illegal to upgrade a non-existent Function.");
396
Chandler Carruth7132e002007-08-04 01:51:18 +0000397 // Quickly eliminate it, if it's not a candidate.
Eric Liuc9c68172016-07-08 16:09:51 +0000398 StringRef Name = F->getName();
Chris Lattnerb372f662011-06-18 18:56:39 +0000399 if (Name.size() <= 8 || !Name.startswith("llvm."))
Evan Cheng0e179d02007-12-17 22:33:23 +0000400 return false;
Chris Lattnerb372f662011-06-18 18:56:39 +0000401 Name = Name.substr(5); // Strip off "llvm."
Chris Lattner0bcbde42011-11-27 08:42:07 +0000402
Chris Lattnerb372f662011-06-18 18:56:39 +0000403 switch (Name[0]) {
Chandler Carruth7132e002007-08-04 01:51:18 +0000404 default: break;
Joel Jones43cb8782012-07-13 23:25:25 +0000405 case 'a': {
Chad Rosierd0114fc2017-01-10 19:23:51 +0000406 if (Name.startswith("arm.rbit") || Name.startswith("aarch64.rbit")) {
Chad Rosier3daffbf2017-01-10 17:20:33 +0000407 NewFn = Intrinsic::getDeclaration(F->getParent(), Intrinsic::bitreverse,
408 F->arg_begin()->getType());
409 return true;
410 }
Joel Jones43cb8782012-07-13 23:25:25 +0000411 if (Name.startswith("arm.neon.vclz")) {
412 Type* args[2] = {
Matt Arsenaultc4c92262013-07-20 17:46:00 +0000413 F->arg_begin()->getType(),
Joel Jones43cb8782012-07-13 23:25:25 +0000414 Type::getInt1Ty(F->getContext())
415 };
416 // Can't use Intrinsic::getDeclaration here as it adds a ".i1" to
417 // the end of the name. Change name from llvm.arm.neon.vclz.* to
418 // llvm.ctlz.*
419 FunctionType* fType = FunctionType::get(F->getReturnType(), args, false);
Matt Arsenaultc4c92262013-07-20 17:46:00 +0000420 NewFn = Function::Create(fType, F->getLinkage(),
Joel Jones43cb8782012-07-13 23:25:25 +0000421 "llvm.ctlz." + Name.substr(14), F->getParent());
422 return true;
423 }
Joel Jonesb84f7be2012-07-18 00:02:16 +0000424 if (Name.startswith("arm.neon.vcnt")) {
425 NewFn = Intrinsic::getDeclaration(F->getParent(), Intrinsic::ctpop,
426 F->arg_begin()->getType());
427 return true;
428 }
Jeroen Ketemaab99b592015-09-30 10:56:37 +0000429 Regex vldRegex("^arm\\.neon\\.vld([1234]|[234]lane)\\.v[a-z0-9]*$");
430 if (vldRegex.match(Name)) {
431 auto fArgs = F->getFunctionType()->params();
432 SmallVector<Type *, 4> Tys(fArgs.begin(), fArgs.end());
433 // Can't use Intrinsic::getDeclaration here as the return types might
434 // then only be structurally equal.
435 FunctionType* fType = FunctionType::get(F->getReturnType(), Tys, false);
436 NewFn = Function::Create(fType, F->getLinkage(),
437 "llvm." + Name + ".p0i8", F->getParent());
438 return true;
439 }
440 Regex vstRegex("^arm\\.neon\\.vst([1234]|[234]lane)\\.v[a-z0-9]*$");
441 if (vstRegex.match(Name)) {
Craig Topper26260942015-10-18 05:15:34 +0000442 static const Intrinsic::ID StoreInts[] = {Intrinsic::arm_neon_vst1,
443 Intrinsic::arm_neon_vst2,
444 Intrinsic::arm_neon_vst3,
445 Intrinsic::arm_neon_vst4};
Jeroen Ketemaab99b592015-09-30 10:56:37 +0000446
Craig Topper26260942015-10-18 05:15:34 +0000447 static const Intrinsic::ID StoreLaneInts[] = {
448 Intrinsic::arm_neon_vst2lane, Intrinsic::arm_neon_vst3lane,
449 Intrinsic::arm_neon_vst4lane
450 };
Jeroen Ketemaab99b592015-09-30 10:56:37 +0000451
452 auto fArgs = F->getFunctionType()->params();
453 Type *Tys[] = {fArgs[0], fArgs[1]};
454 if (Name.find("lane") == StringRef::npos)
455 NewFn = Intrinsic::getDeclaration(F->getParent(),
456 StoreInts[fArgs.size() - 3], Tys);
457 else
458 NewFn = Intrinsic::getDeclaration(F->getParent(),
459 StoreLaneInts[fArgs.size() - 5], Tys);
460 return true;
461 }
Marcin Koscielnicki3fdc2572016-04-19 20:51:05 +0000462 if (Name == "aarch64.thread.pointer" || Name == "arm.thread.pointer") {
463 NewFn = Intrinsic::getDeclaration(F->getParent(), Intrinsic::thread_pointer);
464 return true;
465 }
Joel Jones43cb8782012-07-13 23:25:25 +0000466 break;
467 }
Jeroen Ketemaab99b592015-09-30 10:56:37 +0000468
Chandler Carruth58a71ed2011-12-12 04:26:04 +0000469 case 'c': {
Chandler Carruth58a71ed2011-12-12 04:26:04 +0000470 if (Name.startswith("ctlz.") && F->arg_size() == 1) {
Rafael Espindolad7325ee2016-10-03 15:51:42 +0000471 rename(F);
Chandler Carruthd4a02402011-12-12 10:57:20 +0000472 NewFn = Intrinsic::getDeclaration(F->getParent(), Intrinsic::ctlz,
473 F->arg_begin()->getType());
Chandler Carruth58a71ed2011-12-12 04:26:04 +0000474 return true;
475 }
476 if (Name.startswith("cttz.") && F->arg_size() == 1) {
Rafael Espindolad7325ee2016-10-03 15:51:42 +0000477 rename(F);
Chandler Carruthd4a02402011-12-12 10:57:20 +0000478 NewFn = Intrinsic::getDeclaration(F->getParent(), Intrinsic::cttz,
479 F->arg_begin()->getType());
Chandler Carruth58a71ed2011-12-12 04:26:04 +0000480 return true;
481 }
482 break;
483 }
Adrian Prantlabe04752017-07-28 20:21:02 +0000484 case 'd': {
485 if (Name == "dbg.value" && F->arg_size() == 4) {
486 rename(F);
487 NewFn = Intrinsic::getDeclaration(F->getParent(), Intrinsic::dbg_value);
488 return true;
489 }
490 break;
491 }
Matt Arsenaultf10061e2017-04-10 20:18:21 +0000492 case 'i':
493 case 'l': {
494 bool IsLifetimeStart = Name.startswith("lifetime.start");
495 if (IsLifetimeStart || Name.startswith("invariant.start")) {
496 Intrinsic::ID ID = IsLifetimeStart ?
497 Intrinsic::lifetime_start : Intrinsic::invariant_start;
Mehdi Amini8c629ec2016-08-13 23:31:24 +0000498 auto Args = F->getFunctionType()->params();
499 Type* ObjectPtr[1] = {Args[1]};
Matt Arsenaultf10061e2017-04-10 20:18:21 +0000500 if (F->getName() != Intrinsic::getName(ID, ObjectPtr)) {
Rafael Espindolad7325ee2016-10-03 15:51:42 +0000501 rename(F);
Matt Arsenaultf10061e2017-04-10 20:18:21 +0000502 NewFn = Intrinsic::getDeclaration(F->getParent(), ID, ObjectPtr);
Mehdi Amini8c629ec2016-08-13 23:31:24 +0000503 return true;
504 }
505 }
Matt Arsenaultf10061e2017-04-10 20:18:21 +0000506
507 bool IsLifetimeEnd = Name.startswith("lifetime.end");
508 if (IsLifetimeEnd || Name.startswith("invariant.end")) {
509 Intrinsic::ID ID = IsLifetimeEnd ?
510 Intrinsic::lifetime_end : Intrinsic::invariant_end;
511
Mehdi Amini8c629ec2016-08-13 23:31:24 +0000512 auto Args = F->getFunctionType()->params();
Matt Arsenaultf10061e2017-04-10 20:18:21 +0000513 Type* ObjectPtr[1] = {Args[IsLifetimeEnd ? 1 : 2]};
514 if (F->getName() != Intrinsic::getName(ID, ObjectPtr)) {
Rafael Espindolad7325ee2016-10-03 15:51:42 +0000515 rename(F);
Matt Arsenaultf10061e2017-04-10 20:18:21 +0000516 NewFn = Intrinsic::getDeclaration(F->getParent(), ID, ObjectPtr);
Mehdi Amini8c629ec2016-08-13 23:31:24 +0000517 return true;
518 }
519 }
520 break;
521 }
Artur Pilipenko7ad95ec2016-06-28 18:27:25 +0000522 case 'm': {
523 if (Name.startswith("masked.load.")) {
524 Type *Tys[] = { F->getReturnType(), F->arg_begin()->getType() };
525 if (F->getName() != Intrinsic::getName(Intrinsic::masked_load, Tys)) {
Rafael Espindolad7325ee2016-10-03 15:51:42 +0000526 rename(F);
Artur Pilipenko7ad95ec2016-06-28 18:27:25 +0000527 NewFn = Intrinsic::getDeclaration(F->getParent(),
528 Intrinsic::masked_load,
529 Tys);
530 return true;
531 }
532 }
533 if (Name.startswith("masked.store.")) {
534 auto Args = F->getFunctionType()->params();
535 Type *Tys[] = { Args[0], Args[1] };
536 if (F->getName() != Intrinsic::getName(Intrinsic::masked_store, Tys)) {
Rafael Espindolad7325ee2016-10-03 15:51:42 +0000537 rename(F);
Artur Pilipenko7ad95ec2016-06-28 18:27:25 +0000538 NewFn = Intrinsic::getDeclaration(F->getParent(),
539 Intrinsic::masked_store,
540 Tys);
541 return true;
542 }
543 }
Elad Cohenef5798a2017-05-03 12:28:54 +0000544 // Renaming gather/scatter intrinsics with no address space overloading
545 // to the new overload which includes an address space
546 if (Name.startswith("masked.gather.")) {
547 Type *Tys[] = {F->getReturnType(), F->arg_begin()->getType()};
548 if (F->getName() != Intrinsic::getName(Intrinsic::masked_gather, Tys)) {
549 rename(F);
550 NewFn = Intrinsic::getDeclaration(F->getParent(),
551 Intrinsic::masked_gather, Tys);
552 return true;
553 }
554 }
555 if (Name.startswith("masked.scatter.")) {
556 auto Args = F->getFunctionType()->params();
557 Type *Tys[] = {Args[0], Args[1]};
558 if (F->getName() != Intrinsic::getName(Intrinsic::masked_scatter, Tys)) {
559 rename(F);
560 NewFn = Intrinsic::getDeclaration(F->getParent(),
561 Intrinsic::masked_scatter, Tys);
562 return true;
563 }
564 }
Daniel Neilson1e687242018-01-19 17:13:12 +0000565 // Updating the memory intrinsics (memcpy/memmove/memset) that have an
566 // alignment parameter to embedding the alignment as an attribute of
567 // the pointer args.
568 if (Name.startswith("memcpy.") && F->arg_size() == 5) {
569 rename(F);
570 // Get the types of dest, src, and len
571 ArrayRef<Type *> ParamTypes = F->getFunctionType()->params().slice(0, 3);
572 NewFn = Intrinsic::getDeclaration(F->getParent(), Intrinsic::memcpy,
573 ParamTypes);
574 return true;
575 }
576 if (Name.startswith("memmove.") && F->arg_size() == 5) {
577 rename(F);
578 // Get the types of dest, src, and len
579 ArrayRef<Type *> ParamTypes = F->getFunctionType()->params().slice(0, 3);
580 NewFn = Intrinsic::getDeclaration(F->getParent(), Intrinsic::memmove,
581 ParamTypes);
582 return true;
583 }
584 if (Name.startswith("memset.") && F->arg_size() == 5) {
585 rename(F);
586 // Get the types of dest, and len
587 const auto *FT = F->getFunctionType();
588 Type *ParamTypes[2] = {
589 FT->getParamType(0), // Dest
590 FT->getParamType(2) // len
591 };
592 NewFn = Intrinsic::getDeclaration(F->getParent(), Intrinsic::memset,
593 ParamTypes);
594 return true;
595 }
Artur Pilipenko7ad95ec2016-06-28 18:27:25 +0000596 break;
597 }
Justin Lebar46624a82017-01-21 01:00:32 +0000598 case 'n': {
599 if (Name.startswith("nvvm.")) {
600 Name = Name.substr(5);
Artur Pilipenko7ad95ec2016-06-28 18:27:25 +0000601
Justin Lebar46624a82017-01-21 01:00:32 +0000602 // The following nvvm intrinsics correspond exactly to an LLVM intrinsic.
603 Intrinsic::ID IID = StringSwitch<Intrinsic::ID>(Name)
604 .Cases("brev32", "brev64", Intrinsic::bitreverse)
605 .Case("clz.i", Intrinsic::ctlz)
606 .Case("popc.i", Intrinsic::ctpop)
607 .Default(Intrinsic::not_intrinsic);
608 if (IID != Intrinsic::not_intrinsic && F->arg_size() == 1) {
609 NewFn = Intrinsic::getDeclaration(F->getParent(), IID,
610 {F->getReturnType()});
611 return true;
612 }
613
614 // The following nvvm intrinsics correspond exactly to an LLVM idiom, but
615 // not to an intrinsic alone. We expand them in UpgradeIntrinsicCall.
616 //
617 // TODO: We could add lohi.i2d.
618 bool Expand = StringSwitch<bool>(Name)
619 .Cases("abs.i", "abs.ll", true)
620 .Cases("clz.ll", "popc.ll", "h2f", true)
621 .Cases("max.i", "max.ll", "max.ui", "max.ull", true)
622 .Cases("min.i", "min.ll", "min.ui", "min.ull", true)
623 .Default(false);
624 if (Expand) {
625 NewFn = nullptr;
626 return true;
627 }
628 }
Galina Kistanovaf525c762017-05-19 20:31:51 +0000629 break;
Justin Lebar46624a82017-01-21 01:00:32 +0000630 }
Matt Arsenaultfbcbce42013-10-07 18:06:48 +0000631 case 'o':
632 // We only need to change the name to match the mangling including the
633 // address space.
George Burgess IV56c7e882017-03-21 20:08:59 +0000634 if (Name.startswith("objectsize.")) {
Matt Arsenaultfbcbce42013-10-07 18:06:48 +0000635 Type *Tys[2] = { F->getReturnType(), F->arg_begin()->getType() };
George Burgess IV56c7e882017-03-21 20:08:59 +0000636 if (F->arg_size() == 2 ||
637 F->getName() != Intrinsic::getName(Intrinsic::objectsize, Tys)) {
Rafael Espindolad7325ee2016-10-03 15:51:42 +0000638 rename(F);
George Burgess IV56c7e882017-03-21 20:08:59 +0000639 NewFn = Intrinsic::getDeclaration(F->getParent(), Intrinsic::objectsize,
640 Tys);
Matt Arsenaultfbcbce42013-10-07 18:06:48 +0000641 return true;
642 }
643 }
644 break;
645
Tim Shen00127562016-04-08 21:26:31 +0000646 case 's':
647 if (Name == "stackprotectorcheck") {
648 NewFn = nullptr;
649 return true;
650 }
George Burgess IV63b06e12016-11-08 04:01:50 +0000651 break;
Tim Shen00127562016-04-08 21:26:31 +0000652
Craig Topperb9b9cb02017-02-17 07:07:19 +0000653 case 'x':
654 if (UpgradeX86IntrinsicFunction(F, Name, NewFn))
Craig Topper3b1817d2012-02-03 06:10:55 +0000655 return true;
Chris Lattnerb372f662011-06-18 18:56:39 +0000656 }
Daniel Berlin3c1432f2017-02-15 23:16:20 +0000657 // Remangle our intrinsic since we upgrade the mangling
658 auto Result = llvm::Intrinsic::remangleIntrinsicFunction(F);
659 if (Result != None) {
660 NewFn = Result.getValue();
661 return true;
662 }
Chandler Carruth7132e002007-08-04 01:51:18 +0000663
Nadav Rotem17ee58a2012-06-10 18:42:51 +0000664 // This may not belong here. This function is effectively being overloaded
665 // to both detect an intrinsic which needs upgrading, and to provide the
666 // upgraded form of the intrinsic. We should perhaps have two separate
Chandler Carruth7132e002007-08-04 01:51:18 +0000667 // functions for this.
Evan Cheng0e179d02007-12-17 22:33:23 +0000668 return false;
Chandler Carruth7132e002007-08-04 01:51:18 +0000669}
670
Evan Cheng0e179d02007-12-17 22:33:23 +0000671bool llvm::UpgradeIntrinsicFunction(Function *F, Function *&NewFn) {
Craig Topperc6207612014-04-09 06:08:46 +0000672 NewFn = nullptr;
Evan Cheng0e179d02007-12-17 22:33:23 +0000673 bool Upgraded = UpgradeIntrinsicFunction1(F, NewFn);
Filipe Cabecinhas0011c582015-07-03 20:12:01 +0000674 assert(F != NewFn && "Intrinsic function upgraded to the same function");
Duncan Sands38ef3a82007-12-03 20:06:50 +0000675
676 // Upgrade intrinsic attributes. This does not change the function.
Evan Cheng0e179d02007-12-17 22:33:23 +0000677 if (NewFn)
678 F = NewFn;
Pete Cooper9e1d3352015-05-20 17:16:39 +0000679 if (Intrinsic::ID id = F->getIntrinsicID())
680 F->setAttributes(Intrinsic::getAttributes(F->getContext(), id));
Duncan Sands38ef3a82007-12-03 20:06:50 +0000681 return Upgraded;
682}
683
Bill Wendlinge26fffc2010-09-10 18:51:56 +0000684bool llvm::UpgradeGlobalVariable(GlobalVariable *GV) {
Chris Lattner80ed9dc2011-06-18 06:05:24 +0000685 // Nothing to do yet.
Bill Wendlinge26fffc2010-09-10 18:51:56 +0000686 return false;
687}
688
Simon Pilgrimf7186822016-06-09 21:09:03 +0000689// Handles upgrading SSE2/AVX2/AVX512BW PSLLDQ intrinsics by converting them
Craig Topperb324e432015-02-18 06:24:44 +0000690// to byte shuffles.
Craig Topper46b34fe2016-07-12 01:42:33 +0000691static Value *UpgradeX86PSLLDQIntrinsics(IRBuilder<> &Builder,
Craig Topper7355ac32016-05-29 06:37:33 +0000692 Value *Op, unsigned Shift) {
693 Type *ResultTy = Op->getType();
694 unsigned NumElts = ResultTy->getVectorNumElements() * 8;
Craig Topperb324e432015-02-18 06:24:44 +0000695
696 // Bitcast from a 64-bit element type to a byte element type.
Craig Topper46b34fe2016-07-12 01:42:33 +0000697 Type *VecTy = VectorType::get(Builder.getInt8Ty(), NumElts);
Craig Topper7355ac32016-05-29 06:37:33 +0000698 Op = Builder.CreateBitCast(Op, VecTy, "cast");
699
Craig Topperb324e432015-02-18 06:24:44 +0000700 // We'll be shuffling in zeroes.
Craig Topper7355ac32016-05-29 06:37:33 +0000701 Value *Res = Constant::getNullValue(VecTy);
Craig Topperb324e432015-02-18 06:24:44 +0000702
703 // If shift is less than 16, emit a shuffle to move the bytes. Otherwise,
704 // we'll just return the zero vector.
705 if (Shift < 16) {
Craig Topper99d1eab2016-06-12 00:41:19 +0000706 uint32_t Idxs[64];
Simon Pilgrimf7186822016-06-09 21:09:03 +0000707 // 256/512-bit version is split into 2/4 16-byte lanes.
Craig Topperb324e432015-02-18 06:24:44 +0000708 for (unsigned l = 0; l != NumElts; l += 16)
709 for (unsigned i = 0; i != 16; ++i) {
710 unsigned Idx = NumElts + i - Shift;
711 if (Idx < NumElts)
712 Idx -= NumElts - 16; // end of lane, switch operand.
Craig Topper7355ac32016-05-29 06:37:33 +0000713 Idxs[l + i] = Idx + l;
Craig Topperb324e432015-02-18 06:24:44 +0000714 }
715
Craig Topper7355ac32016-05-29 06:37:33 +0000716 Res = Builder.CreateShuffleVector(Res, Op, makeArrayRef(Idxs, NumElts));
Craig Topperb324e432015-02-18 06:24:44 +0000717 }
718
719 // Bitcast back to a 64-bit element type.
Craig Topper7355ac32016-05-29 06:37:33 +0000720 return Builder.CreateBitCast(Res, ResultTy, "cast");
Craig Topperb324e432015-02-18 06:24:44 +0000721}
722
Craig Topperea703ae2016-06-13 02:36:42 +0000723// Handles upgrading SSE2/AVX2/AVX512BW PSRLDQ intrinsics by converting them
724// to byte shuffles.
Craig Topper46b34fe2016-07-12 01:42:33 +0000725static Value *UpgradeX86PSRLDQIntrinsics(IRBuilder<> &Builder, Value *Op,
Craig Topperea703ae2016-06-13 02:36:42 +0000726 unsigned Shift) {
727 Type *ResultTy = Op->getType();
728 unsigned NumElts = ResultTy->getVectorNumElements() * 8;
729
730 // Bitcast from a 64-bit element type to a byte element type.
Craig Topper46b34fe2016-07-12 01:42:33 +0000731 Type *VecTy = VectorType::get(Builder.getInt8Ty(), NumElts);
Craig Topperea703ae2016-06-13 02:36:42 +0000732 Op = Builder.CreateBitCast(Op, VecTy, "cast");
733
734 // We'll be shuffling in zeroes.
735 Value *Res = Constant::getNullValue(VecTy);
736
737 // If shift is less than 16, emit a shuffle to move the bytes. Otherwise,
738 // we'll just return the zero vector.
739 if (Shift < 16) {
740 uint32_t Idxs[64];
741 // 256/512-bit version is split into 2/4 16-byte lanes.
742 for (unsigned l = 0; l != NumElts; l += 16)
743 for (unsigned i = 0; i != 16; ++i) {
744 unsigned Idx = i + Shift;
745 if (Idx >= 16)
746 Idx += NumElts - 16; // end of lane, switch operand.
747 Idxs[l + i] = Idx + l;
748 }
749
750 Res = Builder.CreateShuffleVector(Op, Res, makeArrayRef(Idxs, NumElts));
751 }
752
753 // Bitcast back to a 64-bit element type.
754 return Builder.CreateBitCast(Res, ResultTy, "cast");
755}
756
757static Value *getX86MaskVec(IRBuilder<> &Builder, Value *Mask,
758 unsigned NumElts) {
759 llvm::VectorType *MaskTy = llvm::VectorType::get(Builder.getInt1Ty(),
760 cast<IntegerType>(Mask->getType())->getBitWidth());
761 Mask = Builder.CreateBitCast(Mask, MaskTy);
762
763 // If we have less than 8 elements, then the starting mask was an i8 and
764 // we need to extract down to the right number of elements.
765 if (NumElts < 8) {
766 uint32_t Indices[4];
767 for (unsigned i = 0; i != NumElts; ++i)
768 Indices[i] = i;
769 Mask = Builder.CreateShuffleVector(Mask, Mask,
770 makeArrayRef(Indices, NumElts),
771 "extract");
772 }
773
774 return Mask;
775}
776
777static Value *EmitX86Select(IRBuilder<> &Builder, Value *Mask,
778 Value *Op0, Value *Op1) {
779 // If the mask is all ones just emit the align operation.
780 if (const auto *C = dyn_cast<Constant>(Mask))
781 if (C->isAllOnesValue())
782 return Op0;
783
784 Mask = getX86MaskVec(Builder, Mask, Op0->getType()->getVectorNumElements());
785 return Builder.CreateSelect(Mask, Op0, Op1);
786}
787
Craig Topperf57e17d2016-11-23 06:54:55 +0000788// Handle autoupgrade for masked PALIGNR and VALIGND/Q intrinsics.
789// PALIGNR handles large immediates by shifting while VALIGN masks the immediate
790// so we need to handle both cases. VALIGN also doesn't have 128-bit lanes.
791static Value *UpgradeX86ALIGNIntrinsics(IRBuilder<> &Builder, Value *Op0,
792 Value *Op1, Value *Shift,
793 Value *Passthru, Value *Mask,
794 bool IsVALIGN) {
Craig Topper33350cc2016-06-06 06:12:54 +0000795 unsigned ShiftVal = cast<llvm::ConstantInt>(Shift)->getZExtValue();
796
797 unsigned NumElts = Op0->getType()->getVectorNumElements();
Craig Topperf57e17d2016-11-23 06:54:55 +0000798 assert((IsVALIGN || NumElts % 16 == 0) && "Illegal NumElts for PALIGNR!");
799 assert((!IsVALIGN || NumElts <= 16) && "NumElts too large for VALIGN!");
800 assert(isPowerOf2_32(NumElts) && "NumElts not a power of 2!");
801
802 // Mask the immediate for VALIGN.
803 if (IsVALIGN)
804 ShiftVal &= (NumElts - 1);
Craig Topper33350cc2016-06-06 06:12:54 +0000805
806 // If palignr is shifting the pair of vectors more than the size of two
807 // lanes, emit zero.
808 if (ShiftVal >= 32)
809 return llvm::Constant::getNullValue(Op0->getType());
810
811 // If palignr is shifting the pair of input vectors more than one lane,
812 // but less than two lanes, convert to shifting in zeroes.
813 if (ShiftVal > 16) {
814 ShiftVal -= 16;
815 Op1 = Op0;
816 Op0 = llvm::Constant::getNullValue(Op0->getType());
817 }
818
Craig Topper99d1eab2016-06-12 00:41:19 +0000819 uint32_t Indices[64];
Craig Topper33350cc2016-06-06 06:12:54 +0000820 // 256-bit palignr operates on 128-bit lanes so we need to handle that
Craig Topperf57e17d2016-11-23 06:54:55 +0000821 for (unsigned l = 0; l < NumElts; l += 16) {
Craig Topper33350cc2016-06-06 06:12:54 +0000822 for (unsigned i = 0; i != 16; ++i) {
823 unsigned Idx = ShiftVal + i;
Craig Topperf57e17d2016-11-23 06:54:55 +0000824 if (!IsVALIGN && Idx >= 16) // Disable wrap for VALIGN.
Craig Topper33350cc2016-06-06 06:12:54 +0000825 Idx += NumElts - 16; // End of lane, switch operand.
826 Indices[l + i] = Idx + l;
827 }
828 }
829
830 Value *Align = Builder.CreateShuffleVector(Op1, Op0,
831 makeArrayRef(Indices, NumElts),
832 "palignr");
833
Craig Topperea703ae2016-06-13 02:36:42 +0000834 return EmitX86Select(Builder, Mask, Align, Passthru);
Craig Topperb324e432015-02-18 06:24:44 +0000835}
836
Craig Topper46b34fe2016-07-12 01:42:33 +0000837static Value *UpgradeMaskedStore(IRBuilder<> &Builder,
Craig Topper50f85c22016-05-31 01:50:02 +0000838 Value *Ptr, Value *Data, Value *Mask,
839 bool Aligned) {
840 // Cast the pointer to the right type.
841 Ptr = Builder.CreateBitCast(Ptr,
842 llvm::PointerType::getUnqual(Data->getType()));
843 unsigned Align =
844 Aligned ? cast<VectorType>(Data->getType())->getBitWidth() / 8 : 1;
845
846 // If the mask is all ones just emit a regular store.
847 if (const auto *C = dyn_cast<Constant>(Mask))
848 if (C->isAllOnesValue())
849 return Builder.CreateAlignedStore(Data, Ptr, Align);
850
851 // Convert the mask from an integer type to a vector of i1.
852 unsigned NumElts = Data->getType()->getVectorNumElements();
Craig Topperea703ae2016-06-13 02:36:42 +0000853 Mask = getX86MaskVec(Builder, Mask, NumElts);
Craig Topper50f85c22016-05-31 01:50:02 +0000854 return Builder.CreateMaskedStore(Data, Ptr, Align, Mask);
855}
856
Craig Topper46b34fe2016-07-12 01:42:33 +0000857static Value *UpgradeMaskedLoad(IRBuilder<> &Builder,
Craig Topperf10fbfa2016-06-02 04:19:36 +0000858 Value *Ptr, Value *Passthru, Value *Mask,
859 bool Aligned) {
860 // Cast the pointer to the right type.
861 Ptr = Builder.CreateBitCast(Ptr,
862 llvm::PointerType::getUnqual(Passthru->getType()));
863 unsigned Align =
864 Aligned ? cast<VectorType>(Passthru->getType())->getBitWidth() / 8 : 1;
865
866 // If the mask is all ones just emit a regular store.
867 if (const auto *C = dyn_cast<Constant>(Mask))
868 if (C->isAllOnesValue())
869 return Builder.CreateAlignedLoad(Ptr, Align);
870
871 // Convert the mask from an integer type to a vector of i1.
872 unsigned NumElts = Passthru->getType()->getVectorNumElements();
Craig Topperea703ae2016-06-13 02:36:42 +0000873 Mask = getX86MaskVec(Builder, Mask, NumElts);
Craig Topperf10fbfa2016-06-02 04:19:36 +0000874 return Builder.CreateMaskedLoad(Ptr, Align, Mask, Passthru);
875}
876
Uriel Korach5d5da5f2017-09-13 09:02:36 +0000877static Value *upgradeAbs(IRBuilder<> &Builder, CallInst &CI) {
878 Value *Op0 = CI.getArgOperand(0);
879 llvm::Type *Ty = Op0->getType();
880 Value *Zero = llvm::Constant::getNullValue(Ty);
881 Value *Cmp = Builder.CreateICmp(ICmpInst::ICMP_SGT, Op0, Zero);
882 Value *Neg = Builder.CreateNeg(Op0);
883 Value *Res = Builder.CreateSelect(Cmp, Op0, Neg);
884
885 if (CI.getNumArgOperands() == 3)
886 Res = EmitX86Select(Builder,CI.getArgOperand(2), Res, CI.getArgOperand(1));
887
888 return Res;
889}
890
Sanjay Patel51ab7572016-06-16 15:48:30 +0000891static Value *upgradeIntMinMax(IRBuilder<> &Builder, CallInst &CI,
892 ICmpInst::Predicate Pred) {
893 Value *Op0 = CI.getArgOperand(0);
894 Value *Op1 = CI.getArgOperand(1);
895 Value *Cmp = Builder.CreateICmp(Pred, Op0, Op1);
Craig Topper8ec5c732016-10-24 04:04:16 +0000896 Value *Res = Builder.CreateSelect(Cmp, Op0, Op1);
897
898 if (CI.getNumArgOperands() == 4)
899 Res = EmitX86Select(Builder, CI.getArgOperand(3), Res, CI.getArgOperand(2));
900
901 return Res;
Sanjay Patel51ab7572016-06-16 15:48:30 +0000902}
903
Uriel Korach2aa707b2017-11-13 12:51:18 +0000904// Applying mask on vector of i1's and make sure result is at least 8 bits wide.
905static Value *ApplyX86MaskOn1BitsVec(IRBuilder<> &Builder,Value *Vec, Value *Mask,
906 unsigned NumElts) {
Craig Toppercc342d42018-01-09 00:50:47 +0000907 if (Mask) {
908 const auto *C = dyn_cast<Constant>(Mask);
909 if (!C || !C->isAllOnesValue())
910 Vec = Builder.CreateAnd(Vec, getX86MaskVec(Builder, Mask, NumElts));
911 }
Uriel Korach2aa707b2017-11-13 12:51:18 +0000912
913 if (NumElts < 8) {
914 uint32_t Indices[8];
915 for (unsigned i = 0; i != NumElts; ++i)
916 Indices[i] = i;
917 for (unsigned i = NumElts; i != 8; ++i)
918 Indices[i] = NumElts + i % NumElts;
919 Vec = Builder.CreateShuffleVector(Vec,
920 Constant::getNullValue(Vec->getType()),
921 Indices);
922 }
923 return Builder.CreateBitCast(Vec, Builder.getIntNTy(std::max(NumElts, 8U)));
924}
925
Craig Topper0a0fb0f2016-06-21 03:53:24 +0000926static Value *upgradeMaskedCompare(IRBuilder<> &Builder, CallInst &CI,
Craig Topper792fc922017-06-22 20:11:01 +0000927 unsigned CC, bool Signed) {
Craig Topper0a0fb0f2016-06-21 03:53:24 +0000928 Value *Op0 = CI.getArgOperand(0);
929 unsigned NumElts = Op0->getType()->getVectorNumElements();
Craig Topper0a0fb0f2016-06-21 03:53:24 +0000930
Craig Topper792fc922017-06-22 20:11:01 +0000931 Value *Cmp;
932 if (CC == 3) {
933 Cmp = Constant::getNullValue(llvm::VectorType::get(Builder.getInt1Ty(), NumElts));
934 } else if (CC == 7) {
935 Cmp = Constant::getAllOnesValue(llvm::VectorType::get(Builder.getInt1Ty(), NumElts));
936 } else {
937 ICmpInst::Predicate Pred;
938 switch (CC) {
939 default: llvm_unreachable("Unknown condition code");
940 case 0: Pred = ICmpInst::ICMP_EQ; break;
941 case 1: Pred = Signed ? ICmpInst::ICMP_SLT : ICmpInst::ICMP_ULT; break;
942 case 2: Pred = Signed ? ICmpInst::ICMP_SLE : ICmpInst::ICMP_ULE; break;
943 case 4: Pred = ICmpInst::ICMP_NE; break;
944 case 5: Pred = Signed ? ICmpInst::ICMP_SGE : ICmpInst::ICMP_UGE; break;
945 case 6: Pred = Signed ? ICmpInst::ICMP_SGT : ICmpInst::ICMP_UGT; break;
946 }
947 Cmp = Builder.CreateICmp(Pred, Op0, CI.getArgOperand(1));
948 }
949
950 Value *Mask = CI.getArgOperand(CI.getNumArgOperands() - 1);
Craig Topper0a0fb0f2016-06-21 03:53:24 +0000951
Uriel Korach2aa707b2017-11-13 12:51:18 +0000952 return ApplyX86MaskOn1BitsVec(Builder, Cmp, Mask, NumElts);
Craig Topper0a0fb0f2016-06-21 03:53:24 +0000953}
954
Craig Topperc9467ed2016-11-06 16:29:08 +0000955// Replace a masked intrinsic with an older unmasked intrinsic.
956static Value *UpgradeX86MaskedShift(IRBuilder<> &Builder, CallInst &CI,
957 Intrinsic::ID IID) {
958 Function *F = CI.getCalledFunction();
959 Function *Intrin = Intrinsic::getDeclaration(F->getParent(), IID);
960 Value *Rep = Builder.CreateCall(Intrin,
961 { CI.getArgOperand(0), CI.getArgOperand(1) });
962 return EmitX86Select(Builder, CI.getArgOperand(3), Rep, CI.getArgOperand(2));
963}
964
Ayman Musa4d602432016-11-16 09:00:28 +0000965static Value* upgradeMaskedMove(IRBuilder<> &Builder, CallInst &CI) {
966 Value* A = CI.getArgOperand(0);
967 Value* B = CI.getArgOperand(1);
968 Value* Src = CI.getArgOperand(2);
969 Value* Mask = CI.getArgOperand(3);
970
971 Value* AndNode = Builder.CreateAnd(Mask, APInt(8, 1));
972 Value* Cmp = Builder.CreateIsNotNull(AndNode);
973 Value* Extract1 = Builder.CreateExtractElement(B, (uint64_t)0);
974 Value* Extract2 = Builder.CreateExtractElement(Src, (uint64_t)0);
975 Value* Select = Builder.CreateSelect(Cmp, Extract1, Extract2);
976 return Builder.CreateInsertElement(A, Select, (uint64_t)0);
977}
Craig Topperc9467ed2016-11-06 16:29:08 +0000978
Michael Zuckerman88fb1712017-04-04 13:32:14 +0000979
980static Value* UpgradeMaskToInt(IRBuilder<> &Builder, CallInst &CI) {
981 Value* Op = CI.getArgOperand(0);
982 Type* ReturnOp = CI.getType();
983 unsigned NumElts = CI.getType()->getVectorNumElements();
984 Value *Mask = getX86MaskVec(Builder, Op, NumElts);
985 return Builder.CreateSExt(Mask, ReturnOp, "vpmovm2");
986}
987
Sanjay Patel595098f2016-06-15 22:01:28 +0000988/// Upgrade a call to an old intrinsic. All argument and return casting must be
989/// provided to seamlessly integrate with existing context.
Chandler Carruth7132e002007-08-04 01:51:18 +0000990void llvm::UpgradeIntrinsicCall(CallInst *CI, Function *NewFn) {
Craig Topper3b1817d2012-02-03 06:10:55 +0000991 Function *F = CI->getCalledFunction();
Nick Lewycky2eb3ade2011-12-12 22:59:34 +0000992 LLVMContext &C = CI->getContext();
Chandler Carruth58a71ed2011-12-12 04:26:04 +0000993 IRBuilder<> Builder(C);
Duncan P. N. Exon Smith52888a62015-10-08 23:49:46 +0000994 Builder.SetInsertPoint(CI->getParent(), CI->getIterator());
Chandler Carruth58a71ed2011-12-12 04:26:04 +0000995
Craig Topper3b1817d2012-02-03 06:10:55 +0000996 assert(F && "Intrinsic call is not direct?");
997
998 if (!NewFn) {
999 // Get the Function's name.
1000 StringRef Name = F->getName();
1001
Craig Topper5aebb862016-07-04 20:56:38 +00001002 assert(Name.startswith("llvm.") && "Intrinsic doesn't start with 'llvm.'");
1003 Name = Name.substr(5);
1004
1005 bool IsX86 = Name.startswith("x86.");
1006 if (IsX86)
1007 Name = Name.substr(4);
Justin Lebar46624a82017-01-21 01:00:32 +00001008 bool IsNVVM = Name.startswith("nvvm.");
1009 if (IsNVVM)
1010 Name = Name.substr(5);
Craig Topper5aebb862016-07-04 20:56:38 +00001011
Craig Toppera4744d12016-12-10 21:15:48 +00001012 if (IsX86 && Name.startswith("sse4a.movnt.")) {
1013 Module *M = F->getParent();
1014 SmallVector<Metadata *, 1> Elts;
1015 Elts.push_back(
1016 ConstantAsMetadata::get(ConstantInt::get(Type::getInt32Ty(C), 1)));
1017 MDNode *Node = MDNode::get(C, Elts);
1018
1019 Value *Arg0 = CI->getArgOperand(0);
1020 Value *Arg1 = CI->getArgOperand(1);
1021
1022 // Nontemporal (unaligned) store of the 0'th element of the float/double
1023 // vector.
1024 Type *SrcEltTy = cast<VectorType>(Arg1->getType())->getElementType();
1025 PointerType *EltPtrTy = PointerType::getUnqual(SrcEltTy);
1026 Value *Addr = Builder.CreateBitCast(Arg0, EltPtrTy, "cast");
1027 Value *Extract =
1028 Builder.CreateExtractElement(Arg1, (uint64_t)0, "extractelement");
1029
1030 StoreInst *SI = Builder.CreateAlignedStore(Extract, Addr, 1);
1031 SI->setMetadata(M->getMDKindID("nontemporal"), Node);
1032
1033 // Remove intrinsic.
1034 CI->eraseFromParent();
1035 return;
1036 }
1037
1038 if (IsX86 && (Name.startswith("avx.movnt.") ||
1039 Name.startswith("avx512.storent."))) {
1040 Module *M = F->getParent();
1041 SmallVector<Metadata *, 1> Elts;
1042 Elts.push_back(
1043 ConstantAsMetadata::get(ConstantInt::get(Type::getInt32Ty(C), 1)));
1044 MDNode *Node = MDNode::get(C, Elts);
1045
1046 Value *Arg0 = CI->getArgOperand(0);
1047 Value *Arg1 = CI->getArgOperand(1);
1048
1049 // Convert the type of the pointer to a pointer to the stored type.
1050 Value *BC = Builder.CreateBitCast(Arg0,
1051 PointerType::getUnqual(Arg1->getType()),
1052 "cast");
1053 VectorType *VTy = cast<VectorType>(Arg1->getType());
1054 StoreInst *SI = Builder.CreateAlignedStore(Arg1, BC,
1055 VTy->getBitWidth() / 8);
1056 SI->setMetadata(M->getMDKindID("nontemporal"), Node);
1057
1058 // Remove intrinsic.
1059 CI->eraseFromParent();
1060 return;
1061 }
1062
1063 if (IsX86 && Name == "sse2.storel.dq") {
1064 Value *Arg0 = CI->getArgOperand(0);
1065 Value *Arg1 = CI->getArgOperand(1);
1066
1067 Type *NewVecTy = VectorType::get(Type::getInt64Ty(C), 2);
1068 Value *BC0 = Builder.CreateBitCast(Arg1, NewVecTy, "cast");
1069 Value *Elt = Builder.CreateExtractElement(BC0, (uint64_t)0);
1070 Value *BC = Builder.CreateBitCast(Arg0,
1071 PointerType::getUnqual(Elt->getType()),
1072 "cast");
1073 Builder.CreateAlignedStore(Elt, BC, 1);
1074
1075 // Remove intrinsic.
1076 CI->eraseFromParent();
1077 return;
1078 }
1079
1080 if (IsX86 && (Name.startswith("sse.storeu.") ||
1081 Name.startswith("sse2.storeu.") ||
1082 Name.startswith("avx.storeu."))) {
1083 Value *Arg0 = CI->getArgOperand(0);
1084 Value *Arg1 = CI->getArgOperand(1);
1085
1086 Arg0 = Builder.CreateBitCast(Arg0,
1087 PointerType::getUnqual(Arg1->getType()),
1088 "cast");
1089 Builder.CreateAlignedStore(Arg1, Arg0, 1);
1090
1091 // Remove intrinsic.
1092 CI->eraseFromParent();
1093 return;
1094 }
1095
Craig Topperaa49f142017-02-18 19:51:14 +00001096 if (IsX86 && (Name.startswith("avx512.mask.store"))) {
1097 // "avx512.mask.storeu." or "avx512.mask.store."
1098 bool Aligned = Name[17] != 'u'; // "avx512.mask.storeu".
Craig Toppera4744d12016-12-10 21:15:48 +00001099 UpgradeMaskedStore(Builder, CI->getArgOperand(0), CI->getArgOperand(1),
Craig Topperaa49f142017-02-18 19:51:14 +00001100 CI->getArgOperand(2), Aligned);
Craig Toppera4744d12016-12-10 21:15:48 +00001101
1102 // Remove intrinsic.
1103 CI->eraseFromParent();
1104 return;
1105 }
1106
Craig Topper3b1817d2012-02-03 06:10:55 +00001107 Value *Rep;
Sanjay Patel595098f2016-06-15 22:01:28 +00001108 // Upgrade packed integer vector compare intrinsics to compare instructions.
Craig Topperaa49f142017-02-18 19:51:14 +00001109 if (IsX86 && (Name.startswith("sse2.pcmp") ||
1110 Name.startswith("avx2.pcmp"))) {
1111 // "sse2.pcpmpeq." "sse2.pcmpgt." "avx2.pcmpeq." or "avx2.pcmpgt."
1112 bool CmpEq = Name[9] == 'e';
1113 Rep = Builder.CreateICmp(CmpEq ? ICmpInst::ICMP_EQ : ICmpInst::ICMP_SGT,
1114 CI->getArgOperand(0), CI->getArgOperand(1));
Craig Topper3b1817d2012-02-03 06:10:55 +00001115 Rep = Builder.CreateSExt(Rep, CI->getType(), "");
Jina Nahias7b705f12017-11-06 07:09:24 +00001116 } else if (IsX86 && (Name.startswith("avx512.broadcastm"))) {
1117 Type *ExtTy = Type::getInt32Ty(C);
1118 if (CI->getOperand(0)->getType()->isIntegerTy(8))
1119 ExtTy = Type::getInt64Ty(C);
1120 unsigned NumElts = CI->getType()->getPrimitiveSizeInBits() /
1121 ExtTy->getPrimitiveSizeInBits();
1122 Rep = Builder.CreateZExt(CI->getArgOperand(0), ExtTy);
1123 Rep = Builder.CreateVectorSplat(NumElts, Rep);
Uriel Korach2aa707b2017-11-13 12:51:18 +00001124 } else if (IsX86 && (Name.startswith("avx512.ptestm") ||
1125 Name.startswith("avx512.ptestnm"))) {
1126 Value *Op0 = CI->getArgOperand(0);
1127 Value *Op1 = CI->getArgOperand(1);
1128 Value *Mask = CI->getArgOperand(2);
1129 Rep = Builder.CreateAnd(Op0, Op1);
1130 llvm::Type *Ty = Op0->getType();
1131 Value *Zero = llvm::Constant::getNullValue(Ty);
1132 ICmpInst::Predicate Pred =
1133 Name.startswith("avx512.ptestm") ? ICmpInst::ICMP_NE : ICmpInst::ICMP_EQ;
1134 Rep = Builder.CreateICmp(Pred, Rep, Zero);
1135 unsigned NumElts = Op0->getType()->getVectorNumElements();
1136 Rep = ApplyX86MaskOn1BitsVec(Builder, Rep, Mask, NumElts);
1137 } else if (IsX86 && (Name.startswith("avx512.mask.pbroadcast"))){
Jina Nahiasccfb8d42017-09-19 11:03:06 +00001138 unsigned NumElts =
1139 CI->getArgOperand(1)->getType()->getVectorNumElements();
1140 Rep = Builder.CreateVectorSplat(NumElts, CI->getArgOperand(0));
1141 Rep = EmitX86Select(Builder, CI->getArgOperand(2), Rep,
1142 CI->getArgOperand(1));
Jina Nahias51c1a622017-12-05 15:42:56 +00001143 } else if (IsX86 && (Name.startswith("avx512.kunpck"))) {
Craig Topper7197a452018-01-14 19:24:10 +00001144 unsigned NumElts = CI->getType()->getScalarSizeInBits();
1145 Value *LHS = getX86MaskVec(Builder, CI->getArgOperand(0), NumElts);
1146 Value *RHS = getX86MaskVec(Builder, CI->getArgOperand(1), NumElts);
1147 uint32_t Indices[64];
1148 for (unsigned i = 0; i != NumElts; ++i)
1149 Indices[i] = i;
1150
1151 // First extract half of each vector. This gives better codegen than
1152 // doing it in a single shuffle.
1153 LHS = Builder.CreateShuffleVector(LHS, LHS,
1154 makeArrayRef(Indices, NumElts / 2));
1155 RHS = Builder.CreateShuffleVector(RHS, RHS,
1156 makeArrayRef(Indices, NumElts / 2));
1157 // Concat the vectors.
1158 Rep = Builder.CreateShuffleVector(LHS, RHS,
1159 makeArrayRef(Indices, NumElts));
1160 Rep = Builder.CreateBitCast(Rep, CI->getType());
Craig Topper071ad9c2018-02-03 20:18:25 +00001161 } else if (IsX86 && Name == "avx512.kand.w") {
1162 Value *LHS = getX86MaskVec(Builder, CI->getArgOperand(0), 16);
1163 Value *RHS = getX86MaskVec(Builder, CI->getArgOperand(1), 16);
1164 Rep = Builder.CreateAnd(LHS, RHS);
1165 Rep = Builder.CreateBitCast(Rep, CI->getType());
1166 } else if (IsX86 && Name == "avx512.kandn.w") {
1167 Value *LHS = getX86MaskVec(Builder, CI->getArgOperand(0), 16);
1168 Value *RHS = getX86MaskVec(Builder, CI->getArgOperand(1), 16);
1169 LHS = Builder.CreateNot(LHS);
1170 Rep = Builder.CreateAnd(LHS, RHS);
1171 Rep = Builder.CreateBitCast(Rep, CI->getType());
1172 } else if (IsX86 && Name == "avx512.kor.w") {
1173 Value *LHS = getX86MaskVec(Builder, CI->getArgOperand(0), 16);
1174 Value *RHS = getX86MaskVec(Builder, CI->getArgOperand(1), 16);
1175 Rep = Builder.CreateOr(LHS, RHS);
1176 Rep = Builder.CreateBitCast(Rep, CI->getType());
1177 } else if (IsX86 && Name == "avx512.kxor.w") {
1178 Value *LHS = getX86MaskVec(Builder, CI->getArgOperand(0), 16);
1179 Value *RHS = getX86MaskVec(Builder, CI->getArgOperand(1), 16);
1180 Rep = Builder.CreateXor(LHS, RHS);
1181 Rep = Builder.CreateBitCast(Rep, CI->getType());
1182 } else if (IsX86 && Name == "avx512.kxnor.w") {
1183 Value *LHS = getX86MaskVec(Builder, CI->getArgOperand(0), 16);
1184 Value *RHS = getX86MaskVec(Builder, CI->getArgOperand(1), 16);
1185 LHS = Builder.CreateNot(LHS);
1186 Rep = Builder.CreateXor(LHS, RHS);
1187 Rep = Builder.CreateBitCast(Rep, CI->getType());
1188 } else if (IsX86 && Name == "avx512.knot.w") {
1189 Rep = getX86MaskVec(Builder, CI->getArgOperand(0), 16);
1190 Rep = Builder.CreateNot(Rep);
1191 Rep = Builder.CreateBitCast(Rep, CI->getType());
Craig Topperdccf72b2018-02-08 20:16:06 +00001192 } else if (IsX86 &&
1193 (Name == "avx512.kortestz.w" || Name == "avx512.kortestc.w")) {
1194 Value *LHS = getX86MaskVec(Builder, CI->getArgOperand(0), 16);
1195 Value *RHS = getX86MaskVec(Builder, CI->getArgOperand(1), 16);
1196 Rep = Builder.CreateOr(LHS, RHS);
1197 Rep = Builder.CreateBitCast(Rep, Builder.getInt16Ty());
1198 Value *C;
1199 if (Name[14] == 'c')
1200 C = ConstantInt::getAllOnesValue(Builder.getInt16Ty());
1201 else
1202 C = ConstantInt::getNullValue(Builder.getInt16Ty());
1203 Rep = Builder.CreateICmpEQ(Rep, C);
1204 Rep = Builder.CreateZExt(Rep, Builder.getInt32Ty());
Craig Topper6910fa02016-11-16 05:24:10 +00001205 } else if (IsX86 && (Name == "sse.add.ss" || Name == "sse2.add.sd")) {
1206 Type *I32Ty = Type::getInt32Ty(C);
1207 Value *Elt0 = Builder.CreateExtractElement(CI->getArgOperand(0),
1208 ConstantInt::get(I32Ty, 0));
1209 Value *Elt1 = Builder.CreateExtractElement(CI->getArgOperand(1),
1210 ConstantInt::get(I32Ty, 0));
1211 Rep = Builder.CreateInsertElement(CI->getArgOperand(0),
1212 Builder.CreateFAdd(Elt0, Elt1),
1213 ConstantInt::get(I32Ty, 0));
1214 } else if (IsX86 && (Name == "sse.sub.ss" || Name == "sse2.sub.sd")) {
1215 Type *I32Ty = Type::getInt32Ty(C);
1216 Value *Elt0 = Builder.CreateExtractElement(CI->getArgOperand(0),
1217 ConstantInt::get(I32Ty, 0));
1218 Value *Elt1 = Builder.CreateExtractElement(CI->getArgOperand(1),
1219 ConstantInt::get(I32Ty, 0));
1220 Rep = Builder.CreateInsertElement(CI->getArgOperand(0),
1221 Builder.CreateFSub(Elt0, Elt1),
1222 ConstantInt::get(I32Ty, 0));
1223 } else if (IsX86 && (Name == "sse.mul.ss" || Name == "sse2.mul.sd")) {
1224 Type *I32Ty = Type::getInt32Ty(C);
1225 Value *Elt0 = Builder.CreateExtractElement(CI->getArgOperand(0),
1226 ConstantInt::get(I32Ty, 0));
1227 Value *Elt1 = Builder.CreateExtractElement(CI->getArgOperand(1),
1228 ConstantInt::get(I32Ty, 0));
1229 Rep = Builder.CreateInsertElement(CI->getArgOperand(0),
1230 Builder.CreateFMul(Elt0, Elt1),
1231 ConstantInt::get(I32Ty, 0));
1232 } else if (IsX86 && (Name == "sse.div.ss" || Name == "sse2.div.sd")) {
1233 Type *I32Ty = Type::getInt32Ty(C);
1234 Value *Elt0 = Builder.CreateExtractElement(CI->getArgOperand(0),
1235 ConstantInt::get(I32Ty, 0));
1236 Value *Elt1 = Builder.CreateExtractElement(CI->getArgOperand(1),
1237 ConstantInt::get(I32Ty, 0));
1238 Rep = Builder.CreateInsertElement(CI->getArgOperand(0),
1239 Builder.CreateFDiv(Elt0, Elt1),
1240 ConstantInt::get(I32Ty, 0));
Craig Topperaa49f142017-02-18 19:51:14 +00001241 } else if (IsX86 && Name.startswith("avx512.mask.pcmp")) {
1242 // "avx512.mask.pcmpeq." or "avx512.mask.pcmpgt."
1243 bool CmpEq = Name[16] == 'e';
Craig Topper792fc922017-06-22 20:11:01 +00001244 Rep = upgradeMaskedCompare(Builder, *CI, CmpEq ? 0 : 6, true);
1245 } else if (IsX86 && Name.startswith("avx512.mask.cmp")) {
1246 unsigned Imm = cast<ConstantInt>(CI->getArgOperand(2))->getZExtValue();
1247 Rep = upgradeMaskedCompare(Builder, *CI, Imm, true);
1248 } else if (IsX86 && Name.startswith("avx512.mask.ucmp")) {
1249 unsigned Imm = cast<ConstantInt>(CI->getArgOperand(2))->getZExtValue();
1250 Rep = upgradeMaskedCompare(Builder, *CI, Imm, false);
Craig Toppercc342d42018-01-09 00:50:47 +00001251 } else if (IsX86 && (Name.startswith("avx512.cvtb2mask.") ||
1252 Name.startswith("avx512.cvtw2mask.") ||
1253 Name.startswith("avx512.cvtd2mask.") ||
1254 Name.startswith("avx512.cvtq2mask."))) {
1255 Value *Op = CI->getArgOperand(0);
1256 Value *Zero = llvm::Constant::getNullValue(Op->getType());
1257 Rep = Builder.CreateICmp(ICmpInst::ICMP_SLT, Op, Zero);
1258 Rep = ApplyX86MaskOn1BitsVec(Builder, Rep, nullptr,
1259 Op->getType()->getVectorNumElements());
Uriel Korach5d5da5f2017-09-13 09:02:36 +00001260 } else if(IsX86 && (Name == "ssse3.pabs.b.128" ||
1261 Name == "ssse3.pabs.w.128" ||
1262 Name == "ssse3.pabs.d.128" ||
1263 Name.startswith("avx2.pabs") ||
1264 Name.startswith("avx512.mask.pabs"))) {
1265 Rep = upgradeAbs(Builder, *CI);
Craig Topper5aebb862016-07-04 20:56:38 +00001266 } else if (IsX86 && (Name == "sse41.pmaxsb" ||
1267 Name == "sse2.pmaxs.w" ||
1268 Name == "sse41.pmaxsd" ||
Craig Topper8ec5c732016-10-24 04:04:16 +00001269 Name.startswith("avx2.pmaxs") ||
1270 Name.startswith("avx512.mask.pmaxs"))) {
Sanjay Patel51ab7572016-06-16 15:48:30 +00001271 Rep = upgradeIntMinMax(Builder, *CI, ICmpInst::ICMP_SGT);
Craig Topper5aebb862016-07-04 20:56:38 +00001272 } else if (IsX86 && (Name == "sse2.pmaxu.b" ||
1273 Name == "sse41.pmaxuw" ||
1274 Name == "sse41.pmaxud" ||
Craig Topper8ec5c732016-10-24 04:04:16 +00001275 Name.startswith("avx2.pmaxu") ||
1276 Name.startswith("avx512.mask.pmaxu"))) {
Sanjay Patel51ab7572016-06-16 15:48:30 +00001277 Rep = upgradeIntMinMax(Builder, *CI, ICmpInst::ICMP_UGT);
Craig Topper5aebb862016-07-04 20:56:38 +00001278 } else if (IsX86 && (Name == "sse41.pminsb" ||
1279 Name == "sse2.pmins.w" ||
1280 Name == "sse41.pminsd" ||
Craig Topper8ec5c732016-10-24 04:04:16 +00001281 Name.startswith("avx2.pmins") ||
1282 Name.startswith("avx512.mask.pmins"))) {
Sanjay Patel51ab7572016-06-16 15:48:30 +00001283 Rep = upgradeIntMinMax(Builder, *CI, ICmpInst::ICMP_SLT);
Craig Topper5aebb862016-07-04 20:56:38 +00001284 } else if (IsX86 && (Name == "sse2.pminu.b" ||
1285 Name == "sse41.pminuw" ||
1286 Name == "sse41.pminud" ||
Craig Topper8ec5c732016-10-24 04:04:16 +00001287 Name.startswith("avx2.pminu") ||
1288 Name.startswith("avx512.mask.pminu"))) {
Sanjay Patel51ab7572016-06-16 15:48:30 +00001289 Rep = upgradeIntMinMax(Builder, *CI, ICmpInst::ICMP_ULT);
Craig Topper5aebb862016-07-04 20:56:38 +00001290 } else if (IsX86 && (Name == "sse2.cvtdq2pd" ||
1291 Name == "sse2.cvtps2pd" ||
1292 Name == "avx.cvtdq2.pd.256" ||
Simon Pilgrimb57dd172016-11-16 14:48:32 +00001293 Name == "avx.cvt.ps2.pd.256" ||
1294 Name.startswith("avx512.mask.cvtdq2pd.") ||
1295 Name.startswith("avx512.mask.cvtudq2pd."))) {
Simon Pilgrim4298d062016-05-25 08:59:18 +00001296 // Lossless i32/float to double conversion.
1297 // Extract the bottom elements if necessary and convert to double vector.
1298 Value *Src = CI->getArgOperand(0);
1299 VectorType *SrcTy = cast<VectorType>(Src->getType());
1300 VectorType *DstTy = cast<VectorType>(CI->getType());
1301 Rep = CI->getArgOperand(0);
1302
1303 unsigned NumDstElts = DstTy->getNumElements();
1304 if (NumDstElts < SrcTy->getNumElements()) {
1305 assert(NumDstElts == 2 && "Unexpected vector size");
Craig Topper99d1eab2016-06-12 00:41:19 +00001306 uint32_t ShuffleMask[2] = { 0, 1 };
1307 Rep = Builder.CreateShuffleVector(Rep, UndefValue::get(SrcTy),
1308 ShuffleMask);
Simon Pilgrim4298d062016-05-25 08:59:18 +00001309 }
1310
Simon Pilgrimb57dd172016-11-16 14:48:32 +00001311 bool SInt2Double = (StringRef::npos != Name.find("cvtdq2"));
1312 bool UInt2Double = (StringRef::npos != Name.find("cvtudq2"));
1313 if (SInt2Double)
Simon Pilgrim4298d062016-05-25 08:59:18 +00001314 Rep = Builder.CreateSIToFP(Rep, DstTy, "cvtdq2pd");
Simon Pilgrimb57dd172016-11-16 14:48:32 +00001315 else if (UInt2Double)
1316 Rep = Builder.CreateUIToFP(Rep, DstTy, "cvtudq2pd");
Simon Pilgrim4298d062016-05-25 08:59:18 +00001317 else
1318 Rep = Builder.CreateFPExt(Rep, DstTy, "cvtps2pd");
Simon Pilgrimb57dd172016-11-16 14:48:32 +00001319
1320 if (CI->getNumArgOperands() == 3)
1321 Rep = EmitX86Select(Builder, CI->getArgOperand(2), Rep,
1322 CI->getArgOperand(1));
Craig Toppera57d2ca2016-09-03 23:55:13 +00001323 } else if (IsX86 && (Name.startswith("avx512.mask.loadu."))) {
Craig Topper46b34fe2016-07-12 01:42:33 +00001324 Rep = UpgradeMaskedLoad(Builder, CI->getArgOperand(0),
Craig Topperf10fbfa2016-06-02 04:19:36 +00001325 CI->getArgOperand(1), CI->getArgOperand(2),
1326 /*Aligned*/false);
Craig Toppera57d2ca2016-09-03 23:55:13 +00001327 } else if (IsX86 && (Name.startswith("avx512.mask.load."))) {
Craig Topper46b34fe2016-07-12 01:42:33 +00001328 Rep = UpgradeMaskedLoad(Builder, CI->getArgOperand(0),
Craig Topperf10fbfa2016-06-02 04:19:36 +00001329 CI->getArgOperand(1),CI->getArgOperand(2),
1330 /*Aligned*/true);
Craig Topper5aebb862016-07-04 20:56:38 +00001331 } else if (IsX86 && Name.startswith("xop.vpcom")) {
Craig Topper3352ba52012-06-09 16:46:13 +00001332 Intrinsic::ID intID;
1333 if (Name.endswith("ub"))
1334 intID = Intrinsic::x86_xop_vpcomub;
1335 else if (Name.endswith("uw"))
1336 intID = Intrinsic::x86_xop_vpcomuw;
1337 else if (Name.endswith("ud"))
1338 intID = Intrinsic::x86_xop_vpcomud;
1339 else if (Name.endswith("uq"))
1340 intID = Intrinsic::x86_xop_vpcomuq;
1341 else if (Name.endswith("b"))
1342 intID = Intrinsic::x86_xop_vpcomb;
1343 else if (Name.endswith("w"))
1344 intID = Intrinsic::x86_xop_vpcomw;
1345 else if (Name.endswith("d"))
1346 intID = Intrinsic::x86_xop_vpcomd;
1347 else if (Name.endswith("q"))
1348 intID = Intrinsic::x86_xop_vpcomq;
1349 else
1350 llvm_unreachable("Unknown suffix");
1351
Craig Topper5aebb862016-07-04 20:56:38 +00001352 Name = Name.substr(9); // strip off "xop.vpcom"
Craig Topper3352ba52012-06-09 16:46:13 +00001353 unsigned Imm;
1354 if (Name.startswith("lt"))
1355 Imm = 0;
1356 else if (Name.startswith("le"))
1357 Imm = 1;
1358 else if (Name.startswith("gt"))
1359 Imm = 2;
1360 else if (Name.startswith("ge"))
1361 Imm = 3;
1362 else if (Name.startswith("eq"))
1363 Imm = 4;
1364 else if (Name.startswith("ne"))
1365 Imm = 5;
Craig Topper3352ba52012-06-09 16:46:13 +00001366 else if (Name.startswith("false"))
Craig Toppere32546d2015-02-13 07:42:15 +00001367 Imm = 6;
1368 else if (Name.startswith("true"))
Craig Topper3352ba52012-06-09 16:46:13 +00001369 Imm = 7;
1370 else
1371 llvm_unreachable("Unknown condition");
1372
1373 Function *VPCOM = Intrinsic::getDeclaration(F->getParent(), intID);
David Blaikieff6409d2015-05-18 22:13:54 +00001374 Rep =
1375 Builder.CreateCall(VPCOM, {CI->getArgOperand(0), CI->getArgOperand(1),
1376 Builder.getInt8(Imm)});
Craig Topperde103122017-02-18 21:50:58 +00001377 } else if (IsX86 && Name.startswith("xop.vpcmov")) {
Simon Pilgrime88dc042015-11-03 20:27:01 +00001378 Value *Sel = CI->getArgOperand(2);
Craig Topper03a9adc2017-02-18 19:51:19 +00001379 Value *NotSel = Builder.CreateNot(Sel);
1380 Value *Sel0 = Builder.CreateAnd(CI->getArgOperand(0), Sel);
1381 Value *Sel1 = Builder.CreateAnd(CI->getArgOperand(1), NotSel);
Simon Pilgrime88dc042015-11-03 20:27:01 +00001382 Rep = Builder.CreateOr(Sel0, Sel1);
Craig Topper5aebb862016-07-04 20:56:38 +00001383 } else if (IsX86 && Name == "sse42.crc32.64.8") {
Craig Topperef9e9932013-10-15 05:20:47 +00001384 Function *CRC32 = Intrinsic::getDeclaration(F->getParent(),
1385 Intrinsic::x86_sse42_crc32_32_8);
1386 Value *Trunc0 = Builder.CreateTrunc(CI->getArgOperand(0), Type::getInt32Ty(C));
David Blaikieff6409d2015-05-18 22:13:54 +00001387 Rep = Builder.CreateCall(CRC32, {Trunc0, CI->getArgOperand(1)});
Craig Topperef9e9932013-10-15 05:20:47 +00001388 Rep = Builder.CreateZExt(Rep, CI->getType(), "");
Simon Pilgrimea0d4f92016-07-22 13:58:44 +00001389 } else if (IsX86 && Name.startswith("avx.vbroadcast.s")) {
Adam Nemet39066802014-05-29 23:35:33 +00001390 // Replace broadcasts with a series of insertelements.
1391 Type *VecTy = CI->getType();
1392 Type *EltTy = VecTy->getVectorElementType();
1393 unsigned EltNum = VecTy->getVectorNumElements();
1394 Value *Cast = Builder.CreateBitCast(CI->getArgOperand(0),
1395 EltTy->getPointerTo());
David Blaikie0c28fd72015-05-20 21:46:30 +00001396 Value *Load = Builder.CreateLoad(EltTy, Cast);
Adam Nemet39066802014-05-29 23:35:33 +00001397 Type *I32Ty = Type::getInt32Ty(C);
1398 Rep = UndefValue::get(VecTy);
1399 for (unsigned I = 0; I < EltNum; ++I)
1400 Rep = Builder.CreateInsertElement(Rep, Load,
1401 ConstantInt::get(I32Ty, I));
Craig Topper5aebb862016-07-04 20:56:38 +00001402 } else if (IsX86 && (Name.startswith("sse41.pmovsx") ||
1403 Name.startswith("sse41.pmovzx") ||
1404 Name.startswith("avx2.pmovsx") ||
Craig Topperb110e042016-11-07 02:12:57 +00001405 Name.startswith("avx2.pmovzx") ||
1406 Name.startswith("avx512.mask.pmovsx") ||
1407 Name.startswith("avx512.mask.pmovzx"))) {
Simon Pilgrim9cb018b2015-09-23 08:48:33 +00001408 VectorType *SrcTy = cast<VectorType>(CI->getArgOperand(0)->getType());
1409 VectorType *DstTy = cast<VectorType>(CI->getType());
1410 unsigned NumDstElts = DstTy->getNumElements();
1411
Simon Pilgrim9602d672016-05-28 18:03:41 +00001412 // Extract a subvector of the first NumDstElts lanes and sign/zero extend.
Craig Topperc0a5fa02016-06-12 04:48:00 +00001413 SmallVector<uint32_t, 8> ShuffleMask(NumDstElts);
Craig Topper99d1eab2016-06-12 00:41:19 +00001414 for (unsigned i = 0; i != NumDstElts; ++i)
Craig Topperc0a5fa02016-06-12 04:48:00 +00001415 ShuffleMask[i] = i;
Simon Pilgrim9cb018b2015-09-23 08:48:33 +00001416
1417 Value *SV = Builder.CreateShuffleVector(
1418 CI->getArgOperand(0), UndefValue::get(SrcTy), ShuffleMask);
Simon Pilgrim9602d672016-05-28 18:03:41 +00001419
1420 bool DoSext = (StringRef::npos != Name.find("pmovsx"));
1421 Rep = DoSext ? Builder.CreateSExt(SV, DstTy)
1422 : Builder.CreateZExt(SV, DstTy);
Craig Topperb110e042016-11-07 02:12:57 +00001423 // If there are 3 arguments, it's a masked intrinsic so we need a select.
1424 if (CI->getNumArgOperands() == 3)
1425 Rep = EmitX86Select(Builder, CI->getArgOperand(2), Rep,
1426 CI->getArgOperand(1));
Simon Pilgrimea0d4f92016-07-22 13:58:44 +00001427 } else if (IsX86 && (Name.startswith("avx.vbroadcastf128") ||
1428 Name == "avx2.vbroadcasti128")) {
1429 // Replace vbroadcastf128/vbroadcasti128 with a vector load+shuffle.
1430 Type *EltTy = CI->getType()->getVectorElementType();
1431 unsigned NumSrcElts = 128 / EltTy->getPrimitiveSizeInBits();
1432 Type *VT = VectorType::get(EltTy, NumSrcElts);
David Blaikie0c28fd72015-05-20 21:46:30 +00001433 Value *Op = Builder.CreatePointerCast(CI->getArgOperand(0),
1434 PointerType::getUnqual(VT));
Chandler Carruth0215e762016-08-10 07:41:26 +00001435 Value *Load = Builder.CreateAlignedLoad(Op, 1);
Simon Pilgrimea0d4f92016-07-22 13:58:44 +00001436 if (NumSrcElts == 2)
1437 Rep = Builder.CreateShuffleVector(Load, UndefValue::get(Load->getType()),
1438 { 0, 1, 0, 1 });
1439 else
1440 Rep = Builder.CreateShuffleVector(Load, UndefValue::get(Load->getType()),
1441 { 0, 1, 2, 3, 0, 1, 2, 3 });
Jina Nahias9a7f9f12017-11-13 09:16:39 +00001442 } else if (IsX86 && (Name.startswith("avx512.mask.shuf.i") ||
1443 Name.startswith("avx512.mask.shuf.f"))) {
1444 unsigned Imm = cast<ConstantInt>(CI->getArgOperand(2))->getZExtValue();
1445 Type *VT = CI->getType();
1446 unsigned NumLanes = VT->getPrimitiveSizeInBits() / 128;
1447 unsigned NumElementsInLane = 128 / VT->getScalarSizeInBits();
1448 unsigned ControlBitsMask = NumLanes - 1;
1449 unsigned NumControlBits = NumLanes / 2;
1450 SmallVector<uint32_t, 8> ShuffleMask(0);
1451
1452 for (unsigned l = 0; l != NumLanes; ++l) {
1453 unsigned LaneMask = (Imm >> (l * NumControlBits)) & ControlBitsMask;
1454 // We actually need the other source.
1455 if (l >= NumLanes / 2)
1456 LaneMask += NumLanes;
1457 for (unsigned i = 0; i != NumElementsInLane; ++i)
1458 ShuffleMask.push_back(LaneMask * NumElementsInLane + i);
1459 }
1460 Rep = Builder.CreateShuffleVector(CI->getArgOperand(0),
1461 CI->getArgOperand(1), ShuffleMask);
1462 Rep = EmitX86Select(Builder, CI->getArgOperand(4), Rep,
1463 CI->getArgOperand(3));
1464 }else if (IsX86 && (Name.startswith("avx512.mask.broadcastf") ||
Craig Topper561092f2017-08-11 16:22:45 +00001465 Name.startswith("avx512.mask.broadcasti"))) {
1466 unsigned NumSrcElts =
1467 CI->getArgOperand(0)->getType()->getVectorNumElements();
1468 unsigned NumDstElts = CI->getType()->getVectorNumElements();
1469
1470 SmallVector<uint32_t, 8> ShuffleMask(NumDstElts);
1471 for (unsigned i = 0; i != NumDstElts; ++i)
1472 ShuffleMask[i] = i % NumSrcElts;
1473
1474 Rep = Builder.CreateShuffleVector(CI->getArgOperand(0),
1475 CI->getArgOperand(0),
1476 ShuffleMask);
1477 Rep = EmitX86Select(Builder, CI->getArgOperand(2), Rep,
1478 CI->getArgOperand(1));
Craig Topper5aebb862016-07-04 20:56:38 +00001479 } else if (IsX86 && (Name.startswith("avx2.pbroadcast") ||
Simon Pilgrim4e96fbf2016-07-05 13:58:47 +00001480 Name.startswith("avx2.vbroadcast") ||
1481 Name.startswith("avx512.pbroadcast") ||
1482 Name.startswith("avx512.mask.broadcast.s"))) {
Ahmed Bougacha1a4987052015-08-20 20:36:19 +00001483 // Replace vp?broadcasts with a vector shuffle.
1484 Value *Op = CI->getArgOperand(0);
1485 unsigned NumElts = CI->getType()->getVectorNumElements();
1486 Type *MaskTy = VectorType::get(Type::getInt32Ty(C), NumElts);
1487 Rep = Builder.CreateShuffleVector(Op, UndefValue::get(Op->getType()),
1488 Constant::getNullValue(MaskTy));
Simon Pilgrim4e96fbf2016-07-05 13:58:47 +00001489
1490 if (CI->getNumArgOperands() == 3)
1491 Rep = EmitX86Select(Builder, CI->getArgOperand(2), Rep,
1492 CI->getArgOperand(1));
Craig Topper5aebb862016-07-04 20:56:38 +00001493 } else if (IsX86 && Name.startswith("avx512.mask.palignr.")) {
Craig Topperf57e17d2016-11-23 06:54:55 +00001494 Rep = UpgradeX86ALIGNIntrinsics(Builder, CI->getArgOperand(0),
1495 CI->getArgOperand(1),
1496 CI->getArgOperand(2),
1497 CI->getArgOperand(3),
1498 CI->getArgOperand(4),
1499 false);
1500 } else if (IsX86 && Name.startswith("avx512.mask.valign.")) {
1501 Rep = UpgradeX86ALIGNIntrinsics(Builder, CI->getArgOperand(0),
1502 CI->getArgOperand(1),
1503 CI->getArgOperand(2),
1504 CI->getArgOperand(3),
1505 CI->getArgOperand(4),
1506 true);
Craig Topper5aebb862016-07-04 20:56:38 +00001507 } else if (IsX86 && (Name == "sse2.psll.dq" ||
1508 Name == "avx2.psll.dq")) {
Craig Topper7355ac32016-05-29 06:37:33 +00001509 // 128/256-bit shift left specified in bits.
Craig Topperb324e432015-02-18 06:24:44 +00001510 unsigned Shift = cast<ConstantInt>(CI->getArgOperand(1))->getZExtValue();
Craig Topper46b34fe2016-07-12 01:42:33 +00001511 Rep = UpgradeX86PSLLDQIntrinsics(Builder, CI->getArgOperand(0),
Craig Topperb324e432015-02-18 06:24:44 +00001512 Shift / 8); // Shift is in bits.
Craig Topper5aebb862016-07-04 20:56:38 +00001513 } else if (IsX86 && (Name == "sse2.psrl.dq" ||
1514 Name == "avx2.psrl.dq")) {
Craig Topper7355ac32016-05-29 06:37:33 +00001515 // 128/256-bit shift right specified in bits.
Craig Topperb324e432015-02-18 06:24:44 +00001516 unsigned Shift = cast<ConstantInt>(CI->getArgOperand(1))->getZExtValue();
Craig Topper46b34fe2016-07-12 01:42:33 +00001517 Rep = UpgradeX86PSRLDQIntrinsics(Builder, CI->getArgOperand(0),
Craig Topperb324e432015-02-18 06:24:44 +00001518 Shift / 8); // Shift is in bits.
Craig Topper5aebb862016-07-04 20:56:38 +00001519 } else if (IsX86 && (Name == "sse2.psll.dq.bs" ||
1520 Name == "avx2.psll.dq.bs" ||
1521 Name == "avx512.psll.dq.512")) {
Simon Pilgrimf7186822016-06-09 21:09:03 +00001522 // 128/256/512-bit shift left specified in bytes.
Craig Topperb324e432015-02-18 06:24:44 +00001523 unsigned Shift = cast<ConstantInt>(CI->getArgOperand(1))->getZExtValue();
Craig Topper46b34fe2016-07-12 01:42:33 +00001524 Rep = UpgradeX86PSLLDQIntrinsics(Builder, CI->getArgOperand(0), Shift);
Craig Topper5aebb862016-07-04 20:56:38 +00001525 } else if (IsX86 && (Name == "sse2.psrl.dq.bs" ||
1526 Name == "avx2.psrl.dq.bs" ||
1527 Name == "avx512.psrl.dq.512")) {
Simon Pilgrimf7186822016-06-09 21:09:03 +00001528 // 128/256/512-bit shift right specified in bytes.
Craig Topperb324e432015-02-18 06:24:44 +00001529 unsigned Shift = cast<ConstantInt>(CI->getArgOperand(1))->getZExtValue();
Craig Topper46b34fe2016-07-12 01:42:33 +00001530 Rep = UpgradeX86PSRLDQIntrinsics(Builder, CI->getArgOperand(0), Shift);
Craig Topper5aebb862016-07-04 20:56:38 +00001531 } else if (IsX86 && (Name == "sse41.pblendw" ||
1532 Name.startswith("sse41.blendp") ||
1533 Name.startswith("avx.blend.p") ||
1534 Name == "avx2.pblendw" ||
1535 Name.startswith("avx2.pblendd."))) {
Craig Topper782d6202015-02-28 19:33:17 +00001536 Value *Op0 = CI->getArgOperand(0);
1537 Value *Op1 = CI->getArgOperand(1);
1538 unsigned Imm = cast <ConstantInt>(CI->getArgOperand(2))->getZExtValue();
1539 VectorType *VecTy = cast<VectorType>(CI->getType());
1540 unsigned NumElts = VecTy->getNumElements();
1541
Craig Topperc0a5fa02016-06-12 04:48:00 +00001542 SmallVector<uint32_t, 16> Idxs(NumElts);
1543 for (unsigned i = 0; i != NumElts; ++i)
1544 Idxs[i] = ((Imm >> (i%8)) & 1) ? i + NumElts : i;
Craig Topper782d6202015-02-28 19:33:17 +00001545
Craig Topper2f561822016-06-12 01:05:59 +00001546 Rep = Builder.CreateShuffleVector(Op0, Op1, Idxs);
Craig Topper5aebb862016-07-04 20:56:38 +00001547 } else if (IsX86 && (Name.startswith("avx.vinsertf128.") ||
Craig Topper0cda8bb2017-01-03 05:45:57 +00001548 Name == "avx2.vinserti128" ||
1549 Name.startswith("avx512.mask.insert"))) {
Sanjay Patel19792fb2015-03-10 16:08:36 +00001550 Value *Op0 = CI->getArgOperand(0);
1551 Value *Op1 = CI->getArgOperand(1);
1552 unsigned Imm = cast<ConstantInt>(CI->getArgOperand(2))->getZExtValue();
Craig Topper0cda8bb2017-01-03 05:45:57 +00001553 unsigned DstNumElts = CI->getType()->getVectorNumElements();
1554 unsigned SrcNumElts = Op1->getType()->getVectorNumElements();
1555 unsigned Scale = DstNumElts / SrcNumElts;
Simon Pilgrim9cb018b2015-09-23 08:48:33 +00001556
Sanjay Patel19792fb2015-03-10 16:08:36 +00001557 // Mask off the high bits of the immediate value; hardware ignores those.
Craig Topper0cda8bb2017-01-03 05:45:57 +00001558 Imm = Imm % Scale;
Simon Pilgrim9cb018b2015-09-23 08:48:33 +00001559
Craig Topper0cda8bb2017-01-03 05:45:57 +00001560 // Extend the second operand into a vector the size of the destination.
Sanjay Patel19792fb2015-03-10 16:08:36 +00001561 Value *UndefV = UndefValue::get(Op1->getType());
Craig Topper0cda8bb2017-01-03 05:45:57 +00001562 SmallVector<uint32_t, 8> Idxs(DstNumElts);
1563 for (unsigned i = 0; i != SrcNumElts; ++i)
Craig Topperc0a5fa02016-06-12 04:48:00 +00001564 Idxs[i] = i;
Craig Topper0cda8bb2017-01-03 05:45:57 +00001565 for (unsigned i = SrcNumElts; i != DstNumElts; ++i)
1566 Idxs[i] = SrcNumElts;
Craig Topper2f561822016-06-12 01:05:59 +00001567 Rep = Builder.CreateShuffleVector(Op1, UndefV, Idxs);
Sanjay Patel19792fb2015-03-10 16:08:36 +00001568
1569 // Insert the second operand into the first operand.
1570
1571 // Note that there is no guarantee that instruction lowering will actually
1572 // produce a vinsertf128 instruction for the created shuffles. In
1573 // particular, the 0 immediate case involves no lane changes, so it can
1574 // be handled as a blend.
1575
1576 // Example of shuffle mask for 32-bit elements:
1577 // Imm = 1 <i32 0, i32 1, i32 2, i32 3, i32 8, i32 9, i32 10, i32 11>
1578 // Imm = 0 <i32 8, i32 9, i32 10, i32 11, i32 4, i32 5, i32 6, i32 7 >
1579
Craig Topper0cda8bb2017-01-03 05:45:57 +00001580 // First fill with identify mask.
1581 for (unsigned i = 0; i != DstNumElts; ++i)
1582 Idxs[i] = i;
1583 // Then replace the elements where we need to insert.
1584 for (unsigned i = 0; i != SrcNumElts; ++i)
1585 Idxs[i + Imm * SrcNumElts] = i + DstNumElts;
Craig Topper2f561822016-06-12 01:05:59 +00001586 Rep = Builder.CreateShuffleVector(Op0, Rep, Idxs);
Craig Topper0cda8bb2017-01-03 05:45:57 +00001587
1588 // If the intrinsic has a mask operand, handle that.
1589 if (CI->getNumArgOperands() == 5)
1590 Rep = EmitX86Select(Builder, CI->getArgOperand(4), Rep,
1591 CI->getArgOperand(3));
Craig Topper5aebb862016-07-04 20:56:38 +00001592 } else if (IsX86 && (Name.startswith("avx.vextractf128.") ||
Craig Topper4d47c6a2017-01-03 05:45:46 +00001593 Name == "avx2.vextracti128" ||
1594 Name.startswith("avx512.mask.vextract"))) {
Sanjay Patelaf1846c2015-03-12 15:15:19 +00001595 Value *Op0 = CI->getArgOperand(0);
1596 unsigned Imm = cast<ConstantInt>(CI->getArgOperand(1))->getZExtValue();
Craig Topper4d47c6a2017-01-03 05:45:46 +00001597 unsigned DstNumElts = CI->getType()->getVectorNumElements();
1598 unsigned SrcNumElts = Op0->getType()->getVectorNumElements();
1599 unsigned Scale = SrcNumElts / DstNumElts;
Simon Pilgrim9cb018b2015-09-23 08:48:33 +00001600
Sanjay Patelaf1846c2015-03-12 15:15:19 +00001601 // Mask off the high bits of the immediate value; hardware ignores those.
Craig Topper4d47c6a2017-01-03 05:45:46 +00001602 Imm = Imm % Scale;
Sanjay Patelaf1846c2015-03-12 15:15:19 +00001603
Craig Topper4d47c6a2017-01-03 05:45:46 +00001604 // Get indexes for the subvector of the input vector.
1605 SmallVector<uint32_t, 8> Idxs(DstNumElts);
1606 for (unsigned i = 0; i != DstNumElts; ++i) {
1607 Idxs[i] = i + (Imm * DstNumElts);
Sanjay Patelaf1846c2015-03-12 15:15:19 +00001608 }
Craig Topper4d47c6a2017-01-03 05:45:46 +00001609 Rep = Builder.CreateShuffleVector(Op0, Op0, Idxs);
Sanjay Patelaf1846c2015-03-12 15:15:19 +00001610
Craig Topper4d47c6a2017-01-03 05:45:46 +00001611 // If the intrinsic has a mask operand, handle that.
1612 if (CI->getNumArgOperands() == 4)
1613 Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep,
1614 CI->getArgOperand(2));
Craig Topper5aebb862016-07-04 20:56:38 +00001615 } else if (!IsX86 && Name == "stackprotectorcheck") {
Tim Shen00127562016-04-08 21:26:31 +00001616 Rep = nullptr;
Craig Topper5aebb862016-07-04 20:56:38 +00001617 } else if (IsX86 && (Name.startswith("avx512.mask.perm.df.") ||
1618 Name.startswith("avx512.mask.perm.di."))) {
Simon Pilgrim02d435d2016-07-04 14:19:05 +00001619 Value *Op0 = CI->getArgOperand(0);
1620 unsigned Imm = cast<ConstantInt>(CI->getArgOperand(1))->getZExtValue();
1621 VectorType *VecTy = cast<VectorType>(CI->getType());
1622 unsigned NumElts = VecTy->getNumElements();
1623
1624 SmallVector<uint32_t, 8> Idxs(NumElts);
1625 for (unsigned i = 0; i != NumElts; ++i)
1626 Idxs[i] = (i & ~0x3) + ((Imm >> (2 * (i & 0x3))) & 3);
1627
1628 Rep = Builder.CreateShuffleVector(Op0, Op0, Idxs);
1629
1630 if (CI->getNumArgOperands() == 4)
1631 Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep,
1632 CI->getArgOperand(2));
Craig Topperf264fcc2017-09-16 07:36:14 +00001633 } else if (IsX86 && (Name.startswith("avx.vperm2f128.") ||
1634 Name == "avx2.vperm2i128")) {
1635 // The immediate permute control byte looks like this:
1636 // [1:0] - select 128 bits from sources for low half of destination
1637 // [2] - ignore
1638 // [3] - zero low half of destination
1639 // [5:4] - select 128 bits from sources for high half of destination
1640 // [6] - ignore
1641 // [7] - zero high half of destination
1642
1643 uint8_t Imm = cast<ConstantInt>(CI->getArgOperand(2))->getZExtValue();
1644
1645 unsigned NumElts = CI->getType()->getVectorNumElements();
1646 unsigned HalfSize = NumElts / 2;
1647 SmallVector<uint32_t, 8> ShuffleMask(NumElts);
1648
1649 // Determine which operand(s) are actually in use for this instruction.
1650 Value *V0 = (Imm & 0x02) ? CI->getArgOperand(1) : CI->getArgOperand(0);
1651 Value *V1 = (Imm & 0x20) ? CI->getArgOperand(1) : CI->getArgOperand(0);
1652
1653 // If needed, replace operands based on zero mask.
1654 V0 = (Imm & 0x08) ? ConstantAggregateZero::get(CI->getType()) : V0;
1655 V1 = (Imm & 0x80) ? ConstantAggregateZero::get(CI->getType()) : V1;
1656
1657 // Permute low half of result.
1658 unsigned StartIndex = (Imm & 0x01) ? HalfSize : 0;
1659 for (unsigned i = 0; i < HalfSize; ++i)
1660 ShuffleMask[i] = StartIndex + i;
1661
1662 // Permute high half of result.
1663 StartIndex = (Imm & 0x10) ? HalfSize : 0;
1664 for (unsigned i = 0; i < HalfSize; ++i)
1665 ShuffleMask[i + HalfSize] = NumElts + StartIndex + i;
1666
1667 Rep = Builder.CreateShuffleVector(V0, V1, ShuffleMask);
1668
Craig Topper5aebb862016-07-04 20:56:38 +00001669 } else if (IsX86 && (Name.startswith("avx.vpermil.") ||
1670 Name == "sse2.pshuf.d" ||
1671 Name.startswith("avx512.mask.vpermil.p") ||
1672 Name.startswith("avx512.mask.pshuf.d."))) {
Craig Topper8a105052016-06-12 03:10:47 +00001673 Value *Op0 = CI->getArgOperand(0);
1674 unsigned Imm = cast<ConstantInt>(CI->getArgOperand(1))->getZExtValue();
1675 VectorType *VecTy = cast<VectorType>(CI->getType());
1676 unsigned NumElts = VecTy->getNumElements();
Simon Pilgrim9fca3002016-07-04 12:40:54 +00001677 // Calculate the size of each index in the immediate.
Craig Topper8a105052016-06-12 03:10:47 +00001678 unsigned IdxSize = 64 / VecTy->getScalarSizeInBits();
1679 unsigned IdxMask = ((1 << IdxSize) - 1);
1680
1681 SmallVector<uint32_t, 8> Idxs(NumElts);
1682 // Lookup the bits for this element, wrapping around the immediate every
1683 // 8-bits. Elements are grouped into sets of 2 or 4 elements so we need
1684 // to offset by the first index of each group.
1685 for (unsigned i = 0; i != NumElts; ++i)
1686 Idxs[i] = ((Imm >> ((i * IdxSize) % 8)) & IdxMask) | (i & ~IdxMask);
1687
1688 Rep = Builder.CreateShuffleVector(Op0, Op0, Idxs);
Craig Topper13cf7ca2016-06-13 02:36:48 +00001689
1690 if (CI->getNumArgOperands() == 4)
1691 Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep,
1692 CI->getArgOperand(2));
Craig Topper5aebb862016-07-04 20:56:38 +00001693 } else if (IsX86 && (Name == "sse2.pshufl.w" ||
1694 Name.startswith("avx512.mask.pshufl.w."))) {
Craig Topper10679862016-06-12 14:11:32 +00001695 Value *Op0 = CI->getArgOperand(0);
1696 unsigned Imm = cast<ConstantInt>(CI->getArgOperand(1))->getZExtValue();
1697 unsigned NumElts = CI->getType()->getVectorNumElements();
1698
1699 SmallVector<uint32_t, 16> Idxs(NumElts);
1700 for (unsigned l = 0; l != NumElts; l += 8) {
1701 for (unsigned i = 0; i != 4; ++i)
1702 Idxs[i + l] = ((Imm >> (2 * i)) & 0x3) + l;
1703 for (unsigned i = 4; i != 8; ++i)
1704 Idxs[i + l] = i + l;
1705 }
1706
1707 Rep = Builder.CreateShuffleVector(Op0, Op0, Idxs);
Craig Topper13cf7ca2016-06-13 02:36:48 +00001708
1709 if (CI->getNumArgOperands() == 4)
1710 Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep,
1711 CI->getArgOperand(2));
Craig Topper5aebb862016-07-04 20:56:38 +00001712 } else if (IsX86 && (Name == "sse2.pshufh.w" ||
1713 Name.startswith("avx512.mask.pshufh.w."))) {
Craig Topper10679862016-06-12 14:11:32 +00001714 Value *Op0 = CI->getArgOperand(0);
1715 unsigned Imm = cast<ConstantInt>(CI->getArgOperand(1))->getZExtValue();
1716 unsigned NumElts = CI->getType()->getVectorNumElements();
1717
1718 SmallVector<uint32_t, 16> Idxs(NumElts);
1719 for (unsigned l = 0; l != NumElts; l += 8) {
1720 for (unsigned i = 0; i != 4; ++i)
1721 Idxs[i + l] = i + l;
1722 for (unsigned i = 0; i != 4; ++i)
1723 Idxs[i + l + 4] = ((Imm >> (2 * i)) & 0x3) + 4 + l;
1724 }
1725
1726 Rep = Builder.CreateShuffleVector(Op0, Op0, Idxs);
Craig Topper13cf7ca2016-06-13 02:36:48 +00001727
1728 if (CI->getNumArgOperands() == 4)
1729 Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep,
1730 CI->getArgOperand(2));
Craig Topper4619c9e2016-09-13 07:40:53 +00001731 } else if (IsX86 && Name.startswith("avx512.mask.shuf.p")) {
1732 Value *Op0 = CI->getArgOperand(0);
1733 Value *Op1 = CI->getArgOperand(1);
1734 unsigned Imm = cast<ConstantInt>(CI->getArgOperand(2))->getZExtValue();
1735 unsigned NumElts = CI->getType()->getVectorNumElements();
1736
1737 unsigned NumLaneElts = 128/CI->getType()->getScalarSizeInBits();
1738 unsigned HalfLaneElts = NumLaneElts / 2;
1739
1740 SmallVector<uint32_t, 16> Idxs(NumElts);
1741 for (unsigned i = 0; i != NumElts; ++i) {
1742 // Base index is the starting element of the lane.
1743 Idxs[i] = i - (i % NumLaneElts);
1744 // If we are half way through the lane switch to the other source.
1745 if ((i % NumLaneElts) >= HalfLaneElts)
1746 Idxs[i] += NumElts;
1747 // Now select the specific element. By adding HalfLaneElts bits from
1748 // the immediate. Wrapping around the immediate every 8-bits.
1749 Idxs[i] += (Imm >> ((i * HalfLaneElts) % 8)) & ((1 << HalfLaneElts) - 1);
1750 }
1751
1752 Rep = Builder.CreateShuffleVector(Op0, Op1, Idxs);
1753
1754 Rep = EmitX86Select(Builder, CI->getArgOperand(4), Rep,
1755 CI->getArgOperand(3));
Craig Topper5aebb862016-07-04 20:56:38 +00001756 } else if (IsX86 && (Name.startswith("avx512.mask.movddup") ||
1757 Name.startswith("avx512.mask.movshdup") ||
1758 Name.startswith("avx512.mask.movsldup"))) {
Simon Pilgrim19adee92016-07-02 14:42:35 +00001759 Value *Op0 = CI->getArgOperand(0);
1760 unsigned NumElts = CI->getType()->getVectorNumElements();
1761 unsigned NumLaneElts = 128/CI->getType()->getScalarSizeInBits();
1762
1763 unsigned Offset = 0;
Craig Topper5aebb862016-07-04 20:56:38 +00001764 if (Name.startswith("avx512.mask.movshdup."))
Simon Pilgrim19adee92016-07-02 14:42:35 +00001765 Offset = 1;
1766
1767 SmallVector<uint32_t, 16> Idxs(NumElts);
1768 for (unsigned l = 0; l != NumElts; l += NumLaneElts)
1769 for (unsigned i = 0; i != NumLaneElts; i += 2) {
1770 Idxs[i + l + 0] = i + l + Offset;
1771 Idxs[i + l + 1] = i + l + Offset;
1772 }
1773
1774 Rep = Builder.CreateShuffleVector(Op0, Op0, Idxs);
1775
1776 Rep = EmitX86Select(Builder, CI->getArgOperand(2), Rep,
1777 CI->getArgOperand(1));
Craig Topper5aebb862016-07-04 20:56:38 +00001778 } else if (IsX86 && (Name.startswith("avx512.mask.punpckl") ||
1779 Name.startswith("avx512.mask.unpckl."))) {
Craig Topper597aa422016-06-23 07:37:33 +00001780 Value *Op0 = CI->getArgOperand(0);
1781 Value *Op1 = CI->getArgOperand(1);
1782 int NumElts = CI->getType()->getVectorNumElements();
1783 int NumLaneElts = 128/CI->getType()->getScalarSizeInBits();
1784
1785 SmallVector<uint32_t, 64> Idxs(NumElts);
1786 for (int l = 0; l != NumElts; l += NumLaneElts)
1787 for (int i = 0; i != NumLaneElts; ++i)
1788 Idxs[i + l] = l + (i / 2) + NumElts * (i % 2);
1789
1790 Rep = Builder.CreateShuffleVector(Op0, Op1, Idxs);
1791
1792 Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep,
1793 CI->getArgOperand(2));
Craig Topper5aebb862016-07-04 20:56:38 +00001794 } else if (IsX86 && (Name.startswith("avx512.mask.punpckh") ||
1795 Name.startswith("avx512.mask.unpckh."))) {
Craig Topper597aa422016-06-23 07:37:33 +00001796 Value *Op0 = CI->getArgOperand(0);
1797 Value *Op1 = CI->getArgOperand(1);
1798 int NumElts = CI->getType()->getVectorNumElements();
1799 int NumLaneElts = 128/CI->getType()->getScalarSizeInBits();
1800
1801 SmallVector<uint32_t, 64> Idxs(NumElts);
1802 for (int l = 0; l != NumElts; l += NumLaneElts)
1803 for (int i = 0; i != NumLaneElts; ++i)
1804 Idxs[i + l] = (NumLaneElts / 2) + l + (i / 2) + NumElts * (i % 2);
1805
1806 Rep = Builder.CreateShuffleVector(Op0, Op1, Idxs);
1807
1808 Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep,
1809 CI->getArgOperand(2));
Craig Toppera6e6feb2016-07-12 05:27:53 +00001810 } else if (IsX86 && Name.startswith("avx512.mask.pand.")) {
1811 Rep = Builder.CreateAnd(CI->getArgOperand(0), CI->getArgOperand(1));
1812 Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep,
1813 CI->getArgOperand(2));
1814 } else if (IsX86 && Name.startswith("avx512.mask.pandn.")) {
1815 Rep = Builder.CreateAnd(Builder.CreateNot(CI->getArgOperand(0)),
1816 CI->getArgOperand(1));
1817 Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep,
1818 CI->getArgOperand(2));
1819 } else if (IsX86 && Name.startswith("avx512.mask.por.")) {
1820 Rep = Builder.CreateOr(CI->getArgOperand(0), CI->getArgOperand(1));
1821 Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep,
1822 CI->getArgOperand(2));
1823 } else if (IsX86 && Name.startswith("avx512.mask.pxor.")) {
1824 Rep = Builder.CreateXor(CI->getArgOperand(0), CI->getArgOperand(1));
1825 Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep,
1826 CI->getArgOperand(2));
Craig Toppere75c4952016-09-02 05:29:17 +00001827 } else if (IsX86 && Name.startswith("avx512.mask.and.")) {
1828 VectorType *FTy = cast<VectorType>(CI->getType());
1829 VectorType *ITy = VectorType::getInteger(FTy);
1830 Rep = Builder.CreateAnd(Builder.CreateBitCast(CI->getArgOperand(0), ITy),
1831 Builder.CreateBitCast(CI->getArgOperand(1), ITy));
1832 Rep = Builder.CreateBitCast(Rep, FTy);
1833 Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep,
1834 CI->getArgOperand(2));
1835 } else if (IsX86 && Name.startswith("avx512.mask.andn.")) {
1836 VectorType *FTy = cast<VectorType>(CI->getType());
1837 VectorType *ITy = VectorType::getInteger(FTy);
1838 Rep = Builder.CreateNot(Builder.CreateBitCast(CI->getArgOperand(0), ITy));
1839 Rep = Builder.CreateAnd(Rep,
1840 Builder.CreateBitCast(CI->getArgOperand(1), ITy));
1841 Rep = Builder.CreateBitCast(Rep, FTy);
1842 Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep,
1843 CI->getArgOperand(2));
1844 } else if (IsX86 && Name.startswith("avx512.mask.or.")) {
1845 VectorType *FTy = cast<VectorType>(CI->getType());
1846 VectorType *ITy = VectorType::getInteger(FTy);
1847 Rep = Builder.CreateOr(Builder.CreateBitCast(CI->getArgOperand(0), ITy),
1848 Builder.CreateBitCast(CI->getArgOperand(1), ITy));
1849 Rep = Builder.CreateBitCast(Rep, FTy);
1850 Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep,
1851 CI->getArgOperand(2));
1852 } else if (IsX86 && Name.startswith("avx512.mask.xor.")) {
1853 VectorType *FTy = cast<VectorType>(CI->getType());
1854 VectorType *ITy = VectorType::getInteger(FTy);
1855 Rep = Builder.CreateXor(Builder.CreateBitCast(CI->getArgOperand(0), ITy),
1856 Builder.CreateBitCast(CI->getArgOperand(1), ITy));
1857 Rep = Builder.CreateBitCast(Rep, FTy);
1858 Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep,
1859 CI->getArgOperand(2));
Craig Topperaf0d63d2016-09-04 02:09:53 +00001860 } else if (IsX86 && Name.startswith("avx512.mask.padd.")) {
1861 Rep = Builder.CreateAdd(CI->getArgOperand(0), CI->getArgOperand(1));
1862 Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep,
1863 CI->getArgOperand(2));
1864 } else if (IsX86 && Name.startswith("avx512.mask.psub.")) {
1865 Rep = Builder.CreateSub(CI->getArgOperand(0), CI->getArgOperand(1));
1866 Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep,
1867 CI->getArgOperand(2));
1868 } else if (IsX86 && Name.startswith("avx512.mask.pmull.")) {
1869 Rep = Builder.CreateMul(CI->getArgOperand(0), CI->getArgOperand(1));
1870 Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep,
1871 CI->getArgOperand(2));
Craig Topper8f354a72016-11-05 05:35:23 +00001872 } else if (IsX86 && (Name.startswith("avx512.mask.add.p"))) {
Craig Topper41773452016-09-04 18:13:33 +00001873 Rep = Builder.CreateFAdd(CI->getArgOperand(0), CI->getArgOperand(1));
1874 Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep,
1875 CI->getArgOperand(2));
Craig Topper96041c62016-11-07 00:13:42 +00001876 } else if (IsX86 && Name.startswith("avx512.mask.div.p")) {
Craig Topper41773452016-09-04 18:13:33 +00001877 Rep = Builder.CreateFDiv(CI->getArgOperand(0), CI->getArgOperand(1));
1878 Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep,
1879 CI->getArgOperand(2));
Craig Topper96041c62016-11-07 00:13:42 +00001880 } else if (IsX86 && Name.startswith("avx512.mask.mul.p")) {
Craig Topper41773452016-09-04 18:13:33 +00001881 Rep = Builder.CreateFMul(CI->getArgOperand(0), CI->getArgOperand(1));
1882 Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep,
1883 CI->getArgOperand(2));
Craig Topper96041c62016-11-07 00:13:42 +00001884 } else if (IsX86 && Name.startswith("avx512.mask.sub.p")) {
Craig Topper41773452016-09-04 18:13:33 +00001885 Rep = Builder.CreateFSub(CI->getArgOperand(0), CI->getArgOperand(1));
1886 Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep,
1887 CI->getArgOperand(2));
Craig Topperf2529c12017-02-24 05:35:04 +00001888 } else if (IsX86 && Name.startswith("avx512.mask.lzcnt.")) {
1889 Rep = Builder.CreateCall(Intrinsic::getDeclaration(F->getParent(),
1890 Intrinsic::ctlz,
1891 CI->getType()),
1892 { CI->getArgOperand(0), Builder.getInt1(false) });
1893 Rep = EmitX86Select(Builder, CI->getArgOperand(2), Rep,
1894 CI->getArgOperand(1));
Craig Toppera5051692017-02-18 07:07:50 +00001895 } else if (IsX86 && (Name.startswith("avx512.mask.max.p") ||
1896 Name.startswith("avx512.mask.min.p"))) {
1897 bool IsMin = Name[13] == 'i';
1898 VectorType *VecTy = cast<VectorType>(CI->getType());
1899 unsigned VecWidth = VecTy->getPrimitiveSizeInBits();
1900 unsigned EltWidth = VecTy->getScalarSizeInBits();
1901 Intrinsic::ID IID;
1902 if (!IsMin && VecWidth == 128 && EltWidth == 32)
1903 IID = Intrinsic::x86_sse_max_ps;
1904 else if (!IsMin && VecWidth == 128 && EltWidth == 64)
1905 IID = Intrinsic::x86_sse2_max_pd;
1906 else if (!IsMin && VecWidth == 256 && EltWidth == 32)
1907 IID = Intrinsic::x86_avx_max_ps_256;
1908 else if (!IsMin && VecWidth == 256 && EltWidth == 64)
1909 IID = Intrinsic::x86_avx_max_pd_256;
1910 else if (IsMin && VecWidth == 128 && EltWidth == 32)
1911 IID = Intrinsic::x86_sse_min_ps;
1912 else if (IsMin && VecWidth == 128 && EltWidth == 64)
1913 IID = Intrinsic::x86_sse2_min_pd;
1914 else if (IsMin && VecWidth == 256 && EltWidth == 32)
1915 IID = Intrinsic::x86_avx_min_ps_256;
1916 else if (IsMin && VecWidth == 256 && EltWidth == 64)
1917 IID = Intrinsic::x86_avx_min_pd_256;
1918 else
1919 llvm_unreachable("Unexpected intrinsic");
1920
1921 Rep = Builder.CreateCall(Intrinsic::getDeclaration(F->getParent(), IID),
1922 { CI->getArgOperand(0), CI->getArgOperand(1) });
1923 Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep,
1924 CI->getArgOperand(2));
Craig Topper7e545332016-11-07 00:13:39 +00001925 } else if (IsX86 && Name.startswith("avx512.mask.pshuf.b.")) {
1926 VectorType *VecTy = cast<VectorType>(CI->getType());
1927 Intrinsic::ID IID;
1928 if (VecTy->getPrimitiveSizeInBits() == 128)
1929 IID = Intrinsic::x86_ssse3_pshuf_b_128;
1930 else if (VecTy->getPrimitiveSizeInBits() == 256)
1931 IID = Intrinsic::x86_avx2_pshuf_b;
Craig Topperedab02b2016-12-10 23:09:43 +00001932 else if (VecTy->getPrimitiveSizeInBits() == 512)
1933 IID = Intrinsic::x86_avx512_pshuf_b_512;
Craig Topper7e545332016-11-07 00:13:39 +00001934 else
1935 llvm_unreachable("Unexpected intrinsic");
1936
1937 Rep = Builder.CreateCall(Intrinsic::getDeclaration(F->getParent(), IID),
1938 { CI->getArgOperand(0), CI->getArgOperand(1) });
1939 Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep,
1940 CI->getArgOperand(2));
Craig Topper2da265b2016-12-27 05:30:14 +00001941 } else if (IsX86 && (Name.startswith("avx512.mask.pmul.dq.") ||
1942 Name.startswith("avx512.mask.pmulu.dq."))) {
1943 bool IsUnsigned = Name[16] == 'u';
1944 VectorType *VecTy = cast<VectorType>(CI->getType());
1945 Intrinsic::ID IID;
1946 if (!IsUnsigned && VecTy->getPrimitiveSizeInBits() == 128)
1947 IID = Intrinsic::x86_sse41_pmuldq;
1948 else if (!IsUnsigned && VecTy->getPrimitiveSizeInBits() == 256)
1949 IID = Intrinsic::x86_avx2_pmul_dq;
1950 else if (!IsUnsigned && VecTy->getPrimitiveSizeInBits() == 512)
1951 IID = Intrinsic::x86_avx512_pmul_dq_512;
1952 else if (IsUnsigned && VecTy->getPrimitiveSizeInBits() == 128)
1953 IID = Intrinsic::x86_sse2_pmulu_dq;
1954 else if (IsUnsigned && VecTy->getPrimitiveSizeInBits() == 256)
1955 IID = Intrinsic::x86_avx2_pmulu_dq;
1956 else if (IsUnsigned && VecTy->getPrimitiveSizeInBits() == 512)
1957 IID = Intrinsic::x86_avx512_pmulu_dq_512;
1958 else
1959 llvm_unreachable("Unexpected intrinsic");
1960
1961 Rep = Builder.CreateCall(Intrinsic::getDeclaration(F->getParent(), IID),
1962 { CI->getArgOperand(0), CI->getArgOperand(1) });
1963 Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep,
1964 CI->getArgOperand(2));
Craig Topper715873e2017-02-16 06:31:54 +00001965 } else if (IsX86 && Name.startswith("avx512.mask.pack")) {
1966 bool IsUnsigned = Name[16] == 'u';
1967 bool IsDW = Name[18] == 'd';
1968 VectorType *VecTy = cast<VectorType>(CI->getType());
1969 Intrinsic::ID IID;
1970 if (!IsUnsigned && !IsDW && VecTy->getPrimitiveSizeInBits() == 128)
1971 IID = Intrinsic::x86_sse2_packsswb_128;
1972 else if (!IsUnsigned && !IsDW && VecTy->getPrimitiveSizeInBits() == 256)
1973 IID = Intrinsic::x86_avx2_packsswb;
1974 else if (!IsUnsigned && !IsDW && VecTy->getPrimitiveSizeInBits() == 512)
1975 IID = Intrinsic::x86_avx512_packsswb_512;
1976 else if (!IsUnsigned && IsDW && VecTy->getPrimitiveSizeInBits() == 128)
1977 IID = Intrinsic::x86_sse2_packssdw_128;
1978 else if (!IsUnsigned && IsDW && VecTy->getPrimitiveSizeInBits() == 256)
1979 IID = Intrinsic::x86_avx2_packssdw;
1980 else if (!IsUnsigned && IsDW && VecTy->getPrimitiveSizeInBits() == 512)
1981 IID = Intrinsic::x86_avx512_packssdw_512;
1982 else if (IsUnsigned && !IsDW && VecTy->getPrimitiveSizeInBits() == 128)
1983 IID = Intrinsic::x86_sse2_packuswb_128;
1984 else if (IsUnsigned && !IsDW && VecTy->getPrimitiveSizeInBits() == 256)
1985 IID = Intrinsic::x86_avx2_packuswb;
1986 else if (IsUnsigned && !IsDW && VecTy->getPrimitiveSizeInBits() == 512)
1987 IID = Intrinsic::x86_avx512_packuswb_512;
1988 else if (IsUnsigned && IsDW && VecTy->getPrimitiveSizeInBits() == 128)
1989 IID = Intrinsic::x86_sse41_packusdw;
1990 else if (IsUnsigned && IsDW && VecTy->getPrimitiveSizeInBits() == 256)
1991 IID = Intrinsic::x86_avx2_packusdw;
1992 else if (IsUnsigned && IsDW && VecTy->getPrimitiveSizeInBits() == 512)
1993 IID = Intrinsic::x86_avx512_packusdw_512;
1994 else
1995 llvm_unreachable("Unexpected intrinsic");
1996
1997 Rep = Builder.CreateCall(Intrinsic::getDeclaration(F->getParent(), IID),
1998 { CI->getArgOperand(0), CI->getArgOperand(1) });
1999 Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep,
2000 CI->getArgOperand(2));
Craig Topper987dad22016-11-13 19:09:56 +00002001 } else if (IsX86 && Name.startswith("avx512.mask.psll")) {
2002 bool IsImmediate = Name[16] == 'i' ||
2003 (Name.size() > 18 && Name[18] == 'i');
2004 bool IsVariable = Name[16] == 'v';
2005 char Size = Name[16] == '.' ? Name[17] :
2006 Name[17] == '.' ? Name[18] :
Craig Topper02b5a1b2016-11-18 05:04:44 +00002007 Name[18] == '.' ? Name[19] :
2008 Name[20];
Craig Topper987dad22016-11-13 19:09:56 +00002009
2010 Intrinsic::ID IID;
Craig Topper353e59b2016-11-14 01:53:22 +00002011 if (IsVariable && Name[17] != '.') {
Craig Topper987dad22016-11-13 19:09:56 +00002012 if (Size == 'd' && Name[17] == '2') // avx512.mask.psllv2.di
2013 IID = Intrinsic::x86_avx2_psllv_q;
2014 else if (Size == 'd' && Name[17] == '4') // avx512.mask.psllv4.di
2015 IID = Intrinsic::x86_avx2_psllv_q_256;
2016 else if (Size == 's' && Name[17] == '4') // avx512.mask.psllv4.si
2017 IID = Intrinsic::x86_avx2_psllv_d;
2018 else if (Size == 's' && Name[17] == '8') // avx512.mask.psllv8.si
2019 IID = Intrinsic::x86_avx2_psllv_d_256;
Craig Topper02b5a1b2016-11-18 05:04:44 +00002020 else if (Size == 'h' && Name[17] == '8') // avx512.mask.psllv8.hi
2021 IID = Intrinsic::x86_avx512_psllv_w_128;
2022 else if (Size == 'h' && Name[17] == '1') // avx512.mask.psllv16.hi
2023 IID = Intrinsic::x86_avx512_psllv_w_256;
2024 else if (Name[17] == '3' && Name[18] == '2') // avx512.mask.psllv32hi
2025 IID = Intrinsic::x86_avx512_psllv_w_512;
Craig Topper987dad22016-11-13 19:09:56 +00002026 else
2027 llvm_unreachable("Unexpected size");
2028 } else if (Name.endswith(".128")) {
2029 if (Size == 'd') // avx512.mask.psll.d.128, avx512.mask.psll.di.128
2030 IID = IsImmediate ? Intrinsic::x86_sse2_pslli_d
2031 : Intrinsic::x86_sse2_psll_d;
2032 else if (Size == 'q') // avx512.mask.psll.q.128, avx512.mask.psll.qi.128
2033 IID = IsImmediate ? Intrinsic::x86_sse2_pslli_q
2034 : Intrinsic::x86_sse2_psll_q;
2035 else if (Size == 'w') // avx512.mask.psll.w.128, avx512.mask.psll.wi.128
2036 IID = IsImmediate ? Intrinsic::x86_sse2_pslli_w
2037 : Intrinsic::x86_sse2_psll_w;
2038 else
2039 llvm_unreachable("Unexpected size");
2040 } else if (Name.endswith(".256")) {
2041 if (Size == 'd') // avx512.mask.psll.d.256, avx512.mask.psll.di.256
2042 IID = IsImmediate ? Intrinsic::x86_avx2_pslli_d
2043 : Intrinsic::x86_avx2_psll_d;
2044 else if (Size == 'q') // avx512.mask.psll.q.256, avx512.mask.psll.qi.256
2045 IID = IsImmediate ? Intrinsic::x86_avx2_pslli_q
2046 : Intrinsic::x86_avx2_psll_q;
2047 else if (Size == 'w') // avx512.mask.psll.w.256, avx512.mask.psll.wi.256
2048 IID = IsImmediate ? Intrinsic::x86_avx2_pslli_w
2049 : Intrinsic::x86_avx2_psll_w;
2050 else
2051 llvm_unreachable("Unexpected size");
2052 } else {
Craig Topper353e59b2016-11-14 01:53:22 +00002053 if (Size == 'd') // psll.di.512, pslli.d, psll.d, psllv.d.512
2054 IID = IsImmediate ? Intrinsic::x86_avx512_pslli_d_512 :
2055 IsVariable ? Intrinsic::x86_avx512_psllv_d_512 :
2056 Intrinsic::x86_avx512_psll_d_512;
2057 else if (Size == 'q') // psll.qi.512, pslli.q, psll.q, psllv.q.512
2058 IID = IsImmediate ? Intrinsic::x86_avx512_pslli_q_512 :
2059 IsVariable ? Intrinsic::x86_avx512_psllv_q_512 :
2060 Intrinsic::x86_avx512_psll_q_512;
Craig Topper987dad22016-11-13 19:09:56 +00002061 else if (Size == 'w') // psll.wi.512, pslli.w, psll.w
2062 IID = IsImmediate ? Intrinsic::x86_avx512_pslli_w_512
2063 : Intrinsic::x86_avx512_psll_w_512;
2064 else
2065 llvm_unreachable("Unexpected size");
2066 }
2067
2068 Rep = UpgradeX86MaskedShift(Builder, *CI, IID);
2069 } else if (IsX86 && Name.startswith("avx512.mask.psrl")) {
2070 bool IsImmediate = Name[16] == 'i' ||
2071 (Name.size() > 18 && Name[18] == 'i');
2072 bool IsVariable = Name[16] == 'v';
2073 char Size = Name[16] == '.' ? Name[17] :
2074 Name[17] == '.' ? Name[18] :
Craig Topper02b5a1b2016-11-18 05:04:44 +00002075 Name[18] == '.' ? Name[19] :
2076 Name[20];
Craig Topper987dad22016-11-13 19:09:56 +00002077
2078 Intrinsic::ID IID;
Craig Topper353e59b2016-11-14 01:53:22 +00002079 if (IsVariable && Name[17] != '.') {
Craig Topper987dad22016-11-13 19:09:56 +00002080 if (Size == 'd' && Name[17] == '2') // avx512.mask.psrlv2.di
2081 IID = Intrinsic::x86_avx2_psrlv_q;
2082 else if (Size == 'd' && Name[17] == '4') // avx512.mask.psrlv4.di
2083 IID = Intrinsic::x86_avx2_psrlv_q_256;
2084 else if (Size == 's' && Name[17] == '4') // avx512.mask.psrlv4.si
2085 IID = Intrinsic::x86_avx2_psrlv_d;
2086 else if (Size == 's' && Name[17] == '8') // avx512.mask.psrlv8.si
2087 IID = Intrinsic::x86_avx2_psrlv_d_256;
Craig Topper02b5a1b2016-11-18 05:04:44 +00002088 else if (Size == 'h' && Name[17] == '8') // avx512.mask.psrlv8.hi
2089 IID = Intrinsic::x86_avx512_psrlv_w_128;
2090 else if (Size == 'h' && Name[17] == '1') // avx512.mask.psrlv16.hi
2091 IID = Intrinsic::x86_avx512_psrlv_w_256;
2092 else if (Name[17] == '3' && Name[18] == '2') // avx512.mask.psrlv32hi
2093 IID = Intrinsic::x86_avx512_psrlv_w_512;
Craig Topper987dad22016-11-13 19:09:56 +00002094 else
2095 llvm_unreachable("Unexpected size");
2096 } else if (Name.endswith(".128")) {
2097 if (Size == 'd') // avx512.mask.psrl.d.128, avx512.mask.psrl.di.128
2098 IID = IsImmediate ? Intrinsic::x86_sse2_psrli_d
2099 : Intrinsic::x86_sse2_psrl_d;
2100 else if (Size == 'q') // avx512.mask.psrl.q.128, avx512.mask.psrl.qi.128
2101 IID = IsImmediate ? Intrinsic::x86_sse2_psrli_q
2102 : Intrinsic::x86_sse2_psrl_q;
2103 else if (Size == 'w') // avx512.mask.psrl.w.128, avx512.mask.psrl.wi.128
2104 IID = IsImmediate ? Intrinsic::x86_sse2_psrli_w
2105 : Intrinsic::x86_sse2_psrl_w;
2106 else
2107 llvm_unreachable("Unexpected size");
2108 } else if (Name.endswith(".256")) {
2109 if (Size == 'd') // avx512.mask.psrl.d.256, avx512.mask.psrl.di.256
2110 IID = IsImmediate ? Intrinsic::x86_avx2_psrli_d
2111 : Intrinsic::x86_avx2_psrl_d;
2112 else if (Size == 'q') // avx512.mask.psrl.q.256, avx512.mask.psrl.qi.256
2113 IID = IsImmediate ? Intrinsic::x86_avx2_psrli_q
2114 : Intrinsic::x86_avx2_psrl_q;
2115 else if (Size == 'w') // avx512.mask.psrl.w.256, avx512.mask.psrl.wi.256
2116 IID = IsImmediate ? Intrinsic::x86_avx2_psrli_w
2117 : Intrinsic::x86_avx2_psrl_w;
2118 else
2119 llvm_unreachable("Unexpected size");
2120 } else {
Craig Topper353e59b2016-11-14 01:53:22 +00002121 if (Size == 'd') // psrl.di.512, psrli.d, psrl.d, psrl.d.512
2122 IID = IsImmediate ? Intrinsic::x86_avx512_psrli_d_512 :
2123 IsVariable ? Intrinsic::x86_avx512_psrlv_d_512 :
2124 Intrinsic::x86_avx512_psrl_d_512;
2125 else if (Size == 'q') // psrl.qi.512, psrli.q, psrl.q, psrl.q.512
2126 IID = IsImmediate ? Intrinsic::x86_avx512_psrli_q_512 :
2127 IsVariable ? Intrinsic::x86_avx512_psrlv_q_512 :
2128 Intrinsic::x86_avx512_psrl_q_512;
Craig Topper987dad22016-11-13 19:09:56 +00002129 else if (Size == 'w') // psrl.wi.512, psrli.w, psrl.w)
2130 IID = IsImmediate ? Intrinsic::x86_avx512_psrli_w_512
2131 : Intrinsic::x86_avx512_psrl_w_512;
2132 else
2133 llvm_unreachable("Unexpected size");
2134 }
2135
2136 Rep = UpgradeX86MaskedShift(Builder, *CI, IID);
2137 } else if (IsX86 && Name.startswith("avx512.mask.psra")) {
2138 bool IsImmediate = Name[16] == 'i' ||
2139 (Name.size() > 18 && Name[18] == 'i');
2140 bool IsVariable = Name[16] == 'v';
2141 char Size = Name[16] == '.' ? Name[17] :
2142 Name[17] == '.' ? Name[18] :
Craig Topper02b5a1b2016-11-18 05:04:44 +00002143 Name[18] == '.' ? Name[19] :
2144 Name[20];
Craig Topper987dad22016-11-13 19:09:56 +00002145
2146 Intrinsic::ID IID;
Craig Topper353e59b2016-11-14 01:53:22 +00002147 if (IsVariable && Name[17] != '.') {
Craig Topper987dad22016-11-13 19:09:56 +00002148 if (Size == 's' && Name[17] == '4') // avx512.mask.psrav4.si
2149 IID = Intrinsic::x86_avx2_psrav_d;
2150 else if (Size == 's' && Name[17] == '8') // avx512.mask.psrav8.si
2151 IID = Intrinsic::x86_avx2_psrav_d_256;
Craig Topper02b5a1b2016-11-18 05:04:44 +00002152 else if (Size == 'h' && Name[17] == '8') // avx512.mask.psrav8.hi
2153 IID = Intrinsic::x86_avx512_psrav_w_128;
2154 else if (Size == 'h' && Name[17] == '1') // avx512.mask.psrav16.hi
2155 IID = Intrinsic::x86_avx512_psrav_w_256;
2156 else if (Name[17] == '3' && Name[18] == '2') // avx512.mask.psrav32hi
2157 IID = Intrinsic::x86_avx512_psrav_w_512;
Craig Topper987dad22016-11-13 19:09:56 +00002158 else
2159 llvm_unreachable("Unexpected size");
2160 } else if (Name.endswith(".128")) {
2161 if (Size == 'd') // avx512.mask.psra.d.128, avx512.mask.psra.di.128
2162 IID = IsImmediate ? Intrinsic::x86_sse2_psrai_d
2163 : Intrinsic::x86_sse2_psra_d;
2164 else if (Size == 'q') // avx512.mask.psra.q.128, avx512.mask.psra.qi.128
Craig Topper353e59b2016-11-14 01:53:22 +00002165 IID = IsImmediate ? Intrinsic::x86_avx512_psrai_q_128 :
2166 IsVariable ? Intrinsic::x86_avx512_psrav_q_128 :
2167 Intrinsic::x86_avx512_psra_q_128;
Craig Topper987dad22016-11-13 19:09:56 +00002168 else if (Size == 'w') // avx512.mask.psra.w.128, avx512.mask.psra.wi.128
2169 IID = IsImmediate ? Intrinsic::x86_sse2_psrai_w
2170 : Intrinsic::x86_sse2_psra_w;
2171 else
2172 llvm_unreachable("Unexpected size");
2173 } else if (Name.endswith(".256")) {
2174 if (Size == 'd') // avx512.mask.psra.d.256, avx512.mask.psra.di.256
2175 IID = IsImmediate ? Intrinsic::x86_avx2_psrai_d
2176 : Intrinsic::x86_avx2_psra_d;
2177 else if (Size == 'q') // avx512.mask.psra.q.256, avx512.mask.psra.qi.256
Craig Topper353e59b2016-11-14 01:53:22 +00002178 IID = IsImmediate ? Intrinsic::x86_avx512_psrai_q_256 :
2179 IsVariable ? Intrinsic::x86_avx512_psrav_q_256 :
2180 Intrinsic::x86_avx512_psra_q_256;
Craig Topper987dad22016-11-13 19:09:56 +00002181 else if (Size == 'w') // avx512.mask.psra.w.256, avx512.mask.psra.wi.256
2182 IID = IsImmediate ? Intrinsic::x86_avx2_psrai_w
2183 : Intrinsic::x86_avx2_psra_w;
2184 else
2185 llvm_unreachable("Unexpected size");
2186 } else {
Craig Topper353e59b2016-11-14 01:53:22 +00002187 if (Size == 'd') // psra.di.512, psrai.d, psra.d, psrav.d.512
2188 IID = IsImmediate ? Intrinsic::x86_avx512_psrai_d_512 :
2189 IsVariable ? Intrinsic::x86_avx512_psrav_d_512 :
2190 Intrinsic::x86_avx512_psra_d_512;
Craig Topper987dad22016-11-13 19:09:56 +00002191 else if (Size == 'q') // psra.qi.512, psrai.q, psra.q
Craig Topper353e59b2016-11-14 01:53:22 +00002192 IID = IsImmediate ? Intrinsic::x86_avx512_psrai_q_512 :
2193 IsVariable ? Intrinsic::x86_avx512_psrav_q_512 :
2194 Intrinsic::x86_avx512_psra_q_512;
Craig Topper987dad22016-11-13 19:09:56 +00002195 else if (Size == 'w') // psra.wi.512, psrai.w, psra.w
2196 IID = IsImmediate ? Intrinsic::x86_avx512_psrai_w_512
2197 : Intrinsic::x86_avx512_psra_w_512;
2198 else
2199 llvm_unreachable("Unexpected size");
2200 }
2201
2202 Rep = UpgradeX86MaskedShift(Builder, *CI, IID);
Ayman Musa4d602432016-11-16 09:00:28 +00002203 } else if (IsX86 && Name.startswith("avx512.mask.move.s")) {
2204 Rep = upgradeMaskedMove(Builder, *CI);
Michael Zuckerman88fb1712017-04-04 13:32:14 +00002205 } else if (IsX86 && Name.startswith("avx512.cvtmask2")) {
2206 Rep = UpgradeMaskToInt(Builder, *CI);
Craig Topperabe7c5b2016-12-10 21:15:52 +00002207 } else if (IsX86 && Name.startswith("avx512.mask.vpermilvar.")) {
2208 Intrinsic::ID IID;
2209 if (Name.endswith("ps.128"))
2210 IID = Intrinsic::x86_avx_vpermilvar_ps;
2211 else if (Name.endswith("pd.128"))
2212 IID = Intrinsic::x86_avx_vpermilvar_pd;
2213 else if (Name.endswith("ps.256"))
2214 IID = Intrinsic::x86_avx_vpermilvar_ps_256;
2215 else if (Name.endswith("pd.256"))
2216 IID = Intrinsic::x86_avx_vpermilvar_pd_256;
Craig Topper1f1b4412016-12-11 01:26:44 +00002217 else if (Name.endswith("ps.512"))
2218 IID = Intrinsic::x86_avx512_vpermilvar_ps_512;
2219 else if (Name.endswith("pd.512"))
2220 IID = Intrinsic::x86_avx512_vpermilvar_pd_512;
Craig Topperabe7c5b2016-12-10 21:15:52 +00002221 else
2222 llvm_unreachable("Unexpected vpermilvar intrinsic");
2223
2224 Function *Intrin = Intrinsic::getDeclaration(F->getParent(), IID);
2225 Rep = Builder.CreateCall(Intrin,
2226 { CI->getArgOperand(0), CI->getArgOperand(1) });
2227 Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep,
2228 CI->getArgOperand(2));
Simon Pilgrim5a22eaa2017-04-14 15:05:35 +00002229 } else if (IsX86 && Name.endswith(".movntdqa")) {
2230 Module *M = F->getParent();
2231 MDNode *Node = MDNode::get(
2232 C, ConstantAsMetadata::get(ConstantInt::get(Type::getInt32Ty(C), 1)));
2233
2234 Value *Ptr = CI->getArgOperand(0);
2235 VectorType *VTy = cast<VectorType>(CI->getType());
2236
2237 // Convert the type of the pointer to a pointer to the stored type.
2238 Value *BC =
2239 Builder.CreateBitCast(Ptr, PointerType::getUnqual(VTy), "cast");
2240 LoadInst *LI = Builder.CreateAlignedLoad(BC, VTy->getBitWidth() / 8);
2241 LI->setMetadata(M->getMDKindID("nontemporal"), Node);
2242 Rep = LI;
Yael Tsafrir47668b52017-09-12 07:50:35 +00002243 } else if (IsX86 &&
2244 (Name.startswith("sse2.pavg") || Name.startswith("avx2.pavg") ||
2245 Name.startswith("avx512.mask.pavg"))) {
2246 // llvm.x86.sse2.pavg.b/w, llvm.x86.avx2.pavg.b/w,
2247 // llvm.x86.avx512.mask.pavg.b/w
2248 Value *A = CI->getArgOperand(0);
2249 Value *B = CI->getArgOperand(1);
2250 VectorType *ZextType = VectorType::getExtendedElementVectorType(
2251 cast<VectorType>(A->getType()));
2252 Value *ExtendedA = Builder.CreateZExt(A, ZextType);
2253 Value *ExtendedB = Builder.CreateZExt(B, ZextType);
2254 Value *Sum = Builder.CreateAdd(ExtendedA, ExtendedB);
2255 Value *AddOne = Builder.CreateAdd(Sum, ConstantInt::get(ZextType, 1));
2256 Value *ShiftR = Builder.CreateLShr(AddOne, ConstantInt::get(ZextType, 1));
2257 Rep = Builder.CreateTrunc(ShiftR, A->getType());
2258 if (CI->getNumArgOperands() > 2) {
2259 Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep,
2260 CI->getArgOperand(2));
2261 }
Justin Lebar46624a82017-01-21 01:00:32 +00002262 } else if (IsNVVM && (Name == "abs.i" || Name == "abs.ll")) {
2263 Value *Arg = CI->getArgOperand(0);
2264 Value *Neg = Builder.CreateNeg(Arg, "neg");
2265 Value *Cmp = Builder.CreateICmpSGE(
2266 Arg, llvm::Constant::getNullValue(Arg->getType()), "abs.cond");
2267 Rep = Builder.CreateSelect(Cmp, Arg, Neg, "abs");
2268 } else if (IsNVVM && (Name == "max.i" || Name == "max.ll" ||
2269 Name == "max.ui" || Name == "max.ull")) {
2270 Value *Arg0 = CI->getArgOperand(0);
2271 Value *Arg1 = CI->getArgOperand(1);
2272 Value *Cmp = Name.endswith(".ui") || Name.endswith(".ull")
2273 ? Builder.CreateICmpUGE(Arg0, Arg1, "max.cond")
2274 : Builder.CreateICmpSGE(Arg0, Arg1, "max.cond");
2275 Rep = Builder.CreateSelect(Cmp, Arg0, Arg1, "max");
2276 } else if (IsNVVM && (Name == "min.i" || Name == "min.ll" ||
2277 Name == "min.ui" || Name == "min.ull")) {
2278 Value *Arg0 = CI->getArgOperand(0);
2279 Value *Arg1 = CI->getArgOperand(1);
2280 Value *Cmp = Name.endswith(".ui") || Name.endswith(".ull")
2281 ? Builder.CreateICmpULE(Arg0, Arg1, "min.cond")
2282 : Builder.CreateICmpSLE(Arg0, Arg1, "min.cond");
2283 Rep = Builder.CreateSelect(Cmp, Arg0, Arg1, "min");
2284 } else if (IsNVVM && Name == "clz.ll") {
2285 // llvm.nvvm.clz.ll returns an i32, but llvm.ctlz.i64 and returns an i64.
2286 Value *Arg = CI->getArgOperand(0);
2287 Value *Ctlz = Builder.CreateCall(
2288 Intrinsic::getDeclaration(F->getParent(), Intrinsic::ctlz,
2289 {Arg->getType()}),
2290 {Arg, Builder.getFalse()}, "ctlz");
2291 Rep = Builder.CreateTrunc(Ctlz, Builder.getInt32Ty(), "ctlz.trunc");
2292 } else if (IsNVVM && Name == "popc.ll") {
2293 // llvm.nvvm.popc.ll returns an i32, but llvm.ctpop.i64 and returns an
2294 // i64.
2295 Value *Arg = CI->getArgOperand(0);
2296 Value *Popc = Builder.CreateCall(
2297 Intrinsic::getDeclaration(F->getParent(), Intrinsic::ctpop,
2298 {Arg->getType()}),
2299 Arg, "ctpop");
2300 Rep = Builder.CreateTrunc(Popc, Builder.getInt32Ty(), "ctpop.trunc");
2301 } else if (IsNVVM && Name == "h2f") {
2302 Rep = Builder.CreateCall(Intrinsic::getDeclaration(
2303 F->getParent(), Intrinsic::convert_from_fp16,
2304 {Builder.getFloatTy()}),
2305 CI->getArgOperand(0), "h2f");
Craig Topper3b1817d2012-02-03 06:10:55 +00002306 } else {
Craig Topper8a105052016-06-12 03:10:47 +00002307 llvm_unreachable("Unknown function for CallInst upgrade.");
Craig Topper3b1817d2012-02-03 06:10:55 +00002308 }
2309
Tim Shen00127562016-04-08 21:26:31 +00002310 if (Rep)
2311 CI->replaceAllUsesWith(Rep);
Craig Topper3b1817d2012-02-03 06:10:55 +00002312 CI->eraseFromParent();
2313 return;
2314 }
2315
Daniel Neilson1e687242018-01-19 17:13:12 +00002316 const auto &DefaultCase = [&NewFn, &CI]() -> void {
Daniel Berlin3c1432f2017-02-15 23:16:20 +00002317 // Handle generic mangling change, but nothing else
2318 assert(
2319 (CI->getCalledFunction()->getName() != NewFn->getName()) &&
2320 "Unknown function for CallInst upgrade and isn't just a name change");
Daniel Berlin3f910042017-03-01 01:49:13 +00002321 CI->setCalledFunction(NewFn);
Daniel Neilson1e687242018-01-19 17:13:12 +00002322 };
2323 CallInst *NewCall = nullptr;
2324 switch (NewFn->getIntrinsicID()) {
2325 default: {
2326 DefaultCase();
Daniel Berlin3c1432f2017-02-15 23:16:20 +00002327 return;
2328 }
Chandler Carruth58a71ed2011-12-12 04:26:04 +00002329
Jeroen Ketemaab99b592015-09-30 10:56:37 +00002330 case Intrinsic::arm_neon_vld1:
2331 case Intrinsic::arm_neon_vld2:
2332 case Intrinsic::arm_neon_vld3:
2333 case Intrinsic::arm_neon_vld4:
2334 case Intrinsic::arm_neon_vld2lane:
2335 case Intrinsic::arm_neon_vld3lane:
2336 case Intrinsic::arm_neon_vld4lane:
2337 case Intrinsic::arm_neon_vst1:
2338 case Intrinsic::arm_neon_vst2:
2339 case Intrinsic::arm_neon_vst3:
2340 case Intrinsic::arm_neon_vst4:
2341 case Intrinsic::arm_neon_vst2lane:
2342 case Intrinsic::arm_neon_vst3lane:
2343 case Intrinsic::arm_neon_vst4lane: {
2344 SmallVector<Value *, 4> Args(CI->arg_operands().begin(),
2345 CI->arg_operands().end());
Daniel Berlin3f910042017-03-01 01:49:13 +00002346 NewCall = Builder.CreateCall(NewFn, Args);
2347 break;
Jeroen Ketemaab99b592015-09-30 10:56:37 +00002348 }
2349
Chad Rosier3daffbf2017-01-10 17:20:33 +00002350 case Intrinsic::bitreverse:
Daniel Berlin3f910042017-03-01 01:49:13 +00002351 NewCall = Builder.CreateCall(NewFn, {CI->getArgOperand(0)});
2352 break;
Chad Rosier3daffbf2017-01-10 17:20:33 +00002353
Chandler Carruth58a71ed2011-12-12 04:26:04 +00002354 case Intrinsic::ctlz:
Nuno Lopesad40c0a2012-05-22 15:25:31 +00002355 case Intrinsic::cttz:
Chandler Carruth58a71ed2011-12-12 04:26:04 +00002356 assert(CI->getNumArgOperands() == 1 &&
2357 "Mismatch between function args and call args");
Daniel Berlin3f910042017-03-01 01:49:13 +00002358 NewCall =
2359 Builder.CreateCall(NewFn, {CI->getArgOperand(0), Builder.getFalse()});
2360 break;
Nadav Rotem17ee58a2012-06-10 18:42:51 +00002361
George Burgess IV56c7e882017-03-21 20:08:59 +00002362 case Intrinsic::objectsize: {
2363 Value *NullIsUnknownSize = CI->getNumArgOperands() == 2
2364 ? Builder.getFalse()
2365 : CI->getArgOperand(2);
2366 NewCall = Builder.CreateCall(
2367 NewFn, {CI->getArgOperand(0), CI->getArgOperand(1), NullIsUnknownSize});
Daniel Berlin3f910042017-03-01 01:49:13 +00002368 break;
George Burgess IV56c7e882017-03-21 20:08:59 +00002369 }
Matt Arsenaultfbcbce42013-10-07 18:06:48 +00002370
Justin Lebar46624a82017-01-21 01:00:32 +00002371 case Intrinsic::ctpop:
Daniel Berlin3f910042017-03-01 01:49:13 +00002372 NewCall = Builder.CreateCall(NewFn, {CI->getArgOperand(0)});
2373 break;
Justin Lebar46624a82017-01-21 01:00:32 +00002374
2375 case Intrinsic::convert_from_fp16:
Daniel Berlin3f910042017-03-01 01:49:13 +00002376 NewCall = Builder.CreateCall(NewFn, {CI->getArgOperand(0)});
2377 break;
Joel Jones43cb8782012-07-13 23:25:25 +00002378
Adrian Prantlabe04752017-07-28 20:21:02 +00002379 case Intrinsic::dbg_value:
2380 // Upgrade from the old version that had an extra offset argument.
2381 assert(CI->getNumArgOperands() == 4);
2382 // Drop nonzero offsets instead of attempting to upgrade them.
2383 if (auto *Offset = dyn_cast_or_null<Constant>(CI->getArgOperand(1)))
2384 if (Offset->isZeroValue()) {
2385 NewCall = Builder.CreateCall(
2386 NewFn,
2387 {CI->getArgOperand(0), CI->getArgOperand(2), CI->getArgOperand(3)});
2388 break;
2389 }
2390 CI->eraseFromParent();
2391 return;
2392
Craig Topper71dc02d2012-06-13 07:18:53 +00002393 case Intrinsic::x86_xop_vfrcz_ss:
2394 case Intrinsic::x86_xop_vfrcz_sd:
Daniel Berlin3f910042017-03-01 01:49:13 +00002395 NewCall = Builder.CreateCall(NewFn, {CI->getArgOperand(1)});
2396 break;
Craig Topper71dc02d2012-06-13 07:18:53 +00002397
Simon Pilgrime85506b2016-06-03 08:06:03 +00002398 case Intrinsic::x86_xop_vpermil2pd:
2399 case Intrinsic::x86_xop_vpermil2ps:
2400 case Intrinsic::x86_xop_vpermil2pd_256:
2401 case Intrinsic::x86_xop_vpermil2ps_256: {
2402 SmallVector<Value *, 4> Args(CI->arg_operands().begin(),
2403 CI->arg_operands().end());
2404 VectorType *FltIdxTy = cast<VectorType>(Args[2]->getType());
2405 VectorType *IntIdxTy = VectorType::getInteger(FltIdxTy);
2406 Args[2] = Builder.CreateBitCast(Args[2], IntIdxTy);
Daniel Berlin3f910042017-03-01 01:49:13 +00002407 NewCall = Builder.CreateCall(NewFn, Args);
2408 break;
Simon Pilgrime85506b2016-06-03 08:06:03 +00002409 }
2410
Nadav Rotem17ee58a2012-06-10 18:42:51 +00002411 case Intrinsic::x86_sse41_ptestc:
2412 case Intrinsic::x86_sse41_ptestz:
Craig Topper71dc02d2012-06-13 07:18:53 +00002413 case Intrinsic::x86_sse41_ptestnzc: {
Nadav Rotem17ee58a2012-06-10 18:42:51 +00002414 // The arguments for these intrinsics used to be v4f32, and changed
2415 // to v2i64. This is purely a nop, since those are bitwise intrinsics.
2416 // So, the only thing required is a bitcast for both arguments.
2417 // First, check the arguments have the old type.
2418 Value *Arg0 = CI->getArgOperand(0);
2419 if (Arg0->getType() != VectorType::get(Type::getFloatTy(C), 4))
2420 return;
2421
2422 // Old intrinsic, add bitcasts
2423 Value *Arg1 = CI->getArgOperand(1);
2424
David Blaikie5bacf372015-04-24 21:16:07 +00002425 Type *NewVecTy = VectorType::get(Type::getInt64Ty(C), 2);
Nadav Rotem17ee58a2012-06-10 18:42:51 +00002426
David Blaikie5bacf372015-04-24 21:16:07 +00002427 Value *BC0 = Builder.CreateBitCast(Arg0, NewVecTy, "cast");
2428 Value *BC1 = Builder.CreateBitCast(Arg1, NewVecTy, "cast");
2429
Daniel Berlin3f910042017-03-01 01:49:13 +00002430 NewCall = Builder.CreateCall(NewFn, {BC0, BC1});
2431 break;
Evan Cheng0e179d02007-12-17 22:33:23 +00002432 }
Chandler Carruth373b2b12014-09-06 10:00:01 +00002433
Chandler Carruth373b2b12014-09-06 10:00:01 +00002434 case Intrinsic::x86_sse41_insertps:
2435 case Intrinsic::x86_sse41_dppd:
2436 case Intrinsic::x86_sse41_dpps:
2437 case Intrinsic::x86_sse41_mpsadbw:
Chandler Carruth373b2b12014-09-06 10:00:01 +00002438 case Intrinsic::x86_avx_dp_ps_256:
Chandler Carruth373b2b12014-09-06 10:00:01 +00002439 case Intrinsic::x86_avx2_mpsadbw: {
2440 // Need to truncate the last argument from i32 to i8 -- this argument models
2441 // an inherently 8-bit immediate operand to these x86 instructions.
2442 SmallVector<Value *, 4> Args(CI->arg_operands().begin(),
2443 CI->arg_operands().end());
2444
2445 // Replace the last argument with a trunc.
2446 Args.back() = Builder.CreateTrunc(Args.back(), Type::getInt8Ty(C), "trunc");
Daniel Berlin3f910042017-03-01 01:49:13 +00002447 NewCall = Builder.CreateCall(NewFn, Args);
2448 break;
Chandler Carruth373b2b12014-09-06 10:00:01 +00002449 }
Marcin Koscielnicki3fdc2572016-04-19 20:51:05 +00002450
Craig Topper4dccffc82018-02-10 23:33:55 +00002451 case Intrinsic::x86_avx512_mask_cmp_pd_128:
2452 case Intrinsic::x86_avx512_mask_cmp_pd_256:
2453 case Intrinsic::x86_avx512_mask_cmp_pd_512:
2454 case Intrinsic::x86_avx512_mask_cmp_ps_128:
2455 case Intrinsic::x86_avx512_mask_cmp_ps_256:
2456 case Intrinsic::x86_avx512_mask_cmp_ps_512: {
2457 SmallVector<Value *, 4> Args;
2458 Args.push_back(CI->getArgOperand(0));
2459 Args.push_back(CI->getArgOperand(1));
2460 Args.push_back(CI->getArgOperand(2));
2461 if (CI->getNumArgOperands() == 5)
2462 Args.push_back(CI->getArgOperand(4));
2463
2464 NewCall = Builder.CreateCall(NewFn, Args);
2465 unsigned NumElts = Args[0]->getType()->getVectorNumElements();
2466 Value *Res = ApplyX86MaskOn1BitsVec(Builder, NewCall, CI->getArgOperand(3),
2467 NumElts);
2468
2469 std::string Name = CI->getName();
2470 if (!Name.empty()) {
2471 CI->setName(Name + ".old");
2472 NewCall->setName(Name);
2473 }
2474 CI->replaceAllUsesWith(Res);
2475 CI->eraseFromParent();
2476 return;
2477 }
2478
Marcin Koscielnicki3fdc2572016-04-19 20:51:05 +00002479 case Intrinsic::thread_pointer: {
Daniel Berlin3f910042017-03-01 01:49:13 +00002480 NewCall = Builder.CreateCall(NewFn, {});
2481 break;
Marcin Koscielnicki3fdc2572016-04-19 20:51:05 +00002482 }
Artur Pilipenko7ad95ec2016-06-28 18:27:25 +00002483
Mehdi Amini8c629ec2016-08-13 23:31:24 +00002484 case Intrinsic::invariant_start:
2485 case Intrinsic::invariant_end:
Artur Pilipenko7ad95ec2016-06-28 18:27:25 +00002486 case Intrinsic::masked_load:
Elad Cohenef5798a2017-05-03 12:28:54 +00002487 case Intrinsic::masked_store:
2488 case Intrinsic::masked_gather:
2489 case Intrinsic::masked_scatter: {
Artur Pilipenko7ad95ec2016-06-28 18:27:25 +00002490 SmallVector<Value *, 4> Args(CI->arg_operands().begin(),
2491 CI->arg_operands().end());
Daniel Berlin3f910042017-03-01 01:49:13 +00002492 NewCall = Builder.CreateCall(NewFn, Args);
2493 break;
Artur Pilipenko7ad95ec2016-06-28 18:27:25 +00002494 }
Daniel Neilson1e687242018-01-19 17:13:12 +00002495
2496 case Intrinsic::memcpy:
2497 case Intrinsic::memmove:
2498 case Intrinsic::memset: {
2499 // We have to make sure that the call signature is what we're expecting.
2500 // We only want to change the old signatures by removing the alignment arg:
2501 // @llvm.mem[cpy|move]...(i8*, i8*, i[32|i64], i32, i1)
2502 // -> @llvm.mem[cpy|move]...(i8*, i8*, i[32|i64], i1)
2503 // @llvm.memset...(i8*, i8, i[32|64], i32, i1)
2504 // -> @llvm.memset...(i8*, i8, i[32|64], i1)
2505 // Note: i8*'s in the above can be any pointer type
2506 if (CI->getNumArgOperands() != 5) {
2507 DefaultCase();
2508 return;
2509 }
2510 // Remove alignment argument (3), and add alignment attributes to the
2511 // dest/src pointers.
2512 Value *Args[4] = {CI->getArgOperand(0), CI->getArgOperand(1),
2513 CI->getArgOperand(2), CI->getArgOperand(4)};
2514 NewCall = Builder.CreateCall(NewFn, Args);
2515 auto *MemCI = cast<MemIntrinsic>(NewCall);
2516 // All mem intrinsics support dest alignment.
2517 const ConstantInt *Align = cast<ConstantInt>(CI->getArgOperand(3));
2518 MemCI->setDestAlignment(Align->getZExtValue());
2519 // Memcpy/Memmove also support source alignment.
2520 if (auto *MTI = dyn_cast<MemTransferInst>(MemCI))
2521 MTI->setSourceAlignment(Align->getZExtValue());
2522 break;
2523 }
Craig Topper71dc02d2012-06-13 07:18:53 +00002524 }
Daniel Berlin3f910042017-03-01 01:49:13 +00002525 assert(NewCall && "Should have either set this variable or returned through "
2526 "the default case");
2527 std::string Name = CI->getName();
2528 if (!Name.empty()) {
2529 CI->setName(Name + ".old");
2530 NewCall->setName(Name);
2531 }
2532 CI->replaceAllUsesWith(NewCall);
2533 CI->eraseFromParent();
Chandler Carruth7132e002007-08-04 01:51:18 +00002534}
2535
Sanjay Patelfdf0d5f2016-04-18 19:11:57 +00002536void llvm::UpgradeCallsToIntrinsic(Function *F) {
Chandler Carruth7132e002007-08-04 01:51:18 +00002537 assert(F && "Illegal attempt to upgrade a non-existent intrinsic.");
2538
Sanjay Patelfdf0d5f2016-04-18 19:11:57 +00002539 // Check if this function should be upgraded and get the replacement function
2540 // if there is one.
Chris Lattner80ed9dc2011-06-18 06:05:24 +00002541 Function *NewFn;
Evan Cheng0e179d02007-12-17 22:33:23 +00002542 if (UpgradeIntrinsicFunction(F, NewFn)) {
Sanjay Patelfdf0d5f2016-04-18 19:11:57 +00002543 // Replace all users of the old function with the new function or new
2544 // instructions. This is not a range loop because the call is deleted.
2545 for (auto UI = F->user_begin(), UE = F->user_end(); UI != UE; )
Duncan P. N. Exon Smith93f53c42016-04-17 03:59:37 +00002546 if (CallInst *CI = dyn_cast<CallInst>(*UI++))
Filipe Cabecinhas0011c582015-07-03 20:12:01 +00002547 UpgradeIntrinsicCall(CI, NewFn);
Sanjay Patelfdf0d5f2016-04-18 19:11:57 +00002548
Filipe Cabecinhas0011c582015-07-03 20:12:01 +00002549 // Remove old function, no longer used, from the module.
2550 F->eraseFromParent();
Chandler Carruth7132e002007-08-04 01:51:18 +00002551 }
2552}
Devang Patel80ae3492009-08-28 23:24:31 +00002553
Mehdi Aminie4709272016-09-14 22:29:59 +00002554MDNode *llvm::UpgradeTBAANode(MDNode &MD) {
Manman Ren209b17c2013-09-28 00:22:27 +00002555 // Check if the tag uses struct-path aware TBAA format.
Mehdi Aminie4709272016-09-14 22:29:59 +00002556 if (isa<MDNode>(MD.getOperand(0)) && MD.getNumOperands() >= 3)
2557 return &MD;
Manman Ren209b17c2013-09-28 00:22:27 +00002558
Mehdi Aminie4709272016-09-14 22:29:59 +00002559 auto &Context = MD.getContext();
2560 if (MD.getNumOperands() == 3) {
2561 Metadata *Elts[] = {MD.getOperand(0), MD.getOperand(1)};
2562 MDNode *ScalarType = MDNode::get(Context, Elts);
Manman Ren209b17c2013-09-28 00:22:27 +00002563 // Create a MDNode <ScalarType, ScalarType, offset 0, const>
Duncan P. N. Exon Smith5bf8fef2014-12-09 18:38:53 +00002564 Metadata *Elts2[] = {ScalarType, ScalarType,
Mehdi Aminie4709272016-09-14 22:29:59 +00002565 ConstantAsMetadata::get(
2566 Constant::getNullValue(Type::getInt64Ty(Context))),
2567 MD.getOperand(2)};
2568 return MDNode::get(Context, Elts2);
Manman Ren209b17c2013-09-28 00:22:27 +00002569 }
Mehdi Aminie4709272016-09-14 22:29:59 +00002570 // Create a MDNode <MD, MD, offset 0>
2571 Metadata *Elts[] = {&MD, &MD, ConstantAsMetadata::get(Constant::getNullValue(
2572 Type::getInt64Ty(Context)))};
2573 return MDNode::get(Context, Elts);
Manman Ren209b17c2013-09-28 00:22:27 +00002574}
Matt Arsenaultb03bd4d2013-11-15 01:34:59 +00002575
2576Instruction *llvm::UpgradeBitCastInst(unsigned Opc, Value *V, Type *DestTy,
2577 Instruction *&Temp) {
2578 if (Opc != Instruction::BitCast)
Craig Topperc6207612014-04-09 06:08:46 +00002579 return nullptr;
Matt Arsenaultb03bd4d2013-11-15 01:34:59 +00002580
Craig Topperc6207612014-04-09 06:08:46 +00002581 Temp = nullptr;
Matt Arsenaultb03bd4d2013-11-15 01:34:59 +00002582 Type *SrcTy = V->getType();
2583 if (SrcTy->isPtrOrPtrVectorTy() && DestTy->isPtrOrPtrVectorTy() &&
2584 SrcTy->getPointerAddressSpace() != DestTy->getPointerAddressSpace()) {
2585 LLVMContext &Context = V->getContext();
2586
2587 // We have no information about target data layout, so we assume that
2588 // the maximum pointer size is 64bit.
2589 Type *MidTy = Type::getInt64Ty(Context);
2590 Temp = CastInst::Create(Instruction::PtrToInt, V, MidTy);
2591
2592 return CastInst::Create(Instruction::IntToPtr, Temp, DestTy);
2593 }
2594
Craig Topperc6207612014-04-09 06:08:46 +00002595 return nullptr;
Matt Arsenaultb03bd4d2013-11-15 01:34:59 +00002596}
2597
2598Value *llvm::UpgradeBitCastExpr(unsigned Opc, Constant *C, Type *DestTy) {
2599 if (Opc != Instruction::BitCast)
Craig Topperc6207612014-04-09 06:08:46 +00002600 return nullptr;
Matt Arsenaultb03bd4d2013-11-15 01:34:59 +00002601
2602 Type *SrcTy = C->getType();
2603 if (SrcTy->isPtrOrPtrVectorTy() && DestTy->isPtrOrPtrVectorTy() &&
2604 SrcTy->getPointerAddressSpace() != DestTy->getPointerAddressSpace()) {
2605 LLVMContext &Context = C->getContext();
2606
2607 // We have no information about target data layout, so we assume that
2608 // the maximum pointer size is 64bit.
2609 Type *MidTy = Type::getInt64Ty(Context);
2610
2611 return ConstantExpr::getIntToPtr(ConstantExpr::getPtrToInt(C, MidTy),
2612 DestTy);
2613 }
2614
Craig Topperc6207612014-04-09 06:08:46 +00002615 return nullptr;
Matt Arsenaultb03bd4d2013-11-15 01:34:59 +00002616}
Manman Ren8b4306c2013-12-02 21:29:56 +00002617
2618/// Check the debug info version number, if it is out-dated, drop the debug
2619/// info. Return true if module is modified.
2620bool llvm::UpgradeDebugInfo(Module &M) {
Manman Ren2ebfb422014-01-16 01:51:12 +00002621 unsigned Version = getDebugMetadataVersionFromModule(M);
Adrian Prantla8b2ddb2017-10-02 18:31:29 +00002622 if (Version == DEBUG_METADATA_VERSION) {
2623 bool BrokenDebugInfo = false;
2624 if (verifyModule(M, &llvm::errs(), &BrokenDebugInfo))
2625 report_fatal_error("Broken module found, compilation aborted!");
2626 if (!BrokenDebugInfo)
2627 // Everything is ok.
2628 return false;
2629 else {
2630 // Diagnose malformed debug info.
2631 DiagnosticInfoIgnoringInvalidDebugMetadata Diag(M);
2632 M.getContext().diagnose(Diag);
2633 }
2634 }
2635 bool Modified = StripDebugInfo(M);
2636 if (Modified && Version != DEBUG_METADATA_VERSION) {
2637 // Diagnose a version mismatch.
Manman Ren2ebfb422014-01-16 01:51:12 +00002638 DiagnosticInfoDebugMetadataVersion DiagVersion(M, Version);
2639 M.getContext().diagnose(DiagVersion);
2640 }
Adrian Prantla8b2ddb2017-10-02 18:31:29 +00002641 return Modified;
Manman Ren8b4306c2013-12-02 21:29:56 +00002642}
Eli Bendersky5d5e18d2014-06-25 15:41:00 +00002643
Manman Renb5d7ff42016-05-25 23:14:48 +00002644bool llvm::UpgradeModuleFlags(Module &M) {
Steven Wu010fc492017-08-21 21:49:13 +00002645 NamedMDNode *ModFlags = M.getModuleFlagsMetadata();
Manman Renb5d7ff42016-05-25 23:14:48 +00002646 if (!ModFlags)
2647 return false;
2648
Steven Wu010fc492017-08-21 21:49:13 +00002649 bool HasObjCFlag = false, HasClassProperties = false, Changed = false;
Manman Renb5d7ff42016-05-25 23:14:48 +00002650 for (unsigned I = 0, E = ModFlags->getNumOperands(); I != E; ++I) {
2651 MDNode *Op = ModFlags->getOperand(I);
Steven Wu010fc492017-08-21 21:49:13 +00002652 if (Op->getNumOperands() != 3)
Manman Renb5d7ff42016-05-25 23:14:48 +00002653 continue;
2654 MDString *ID = dyn_cast_or_null<MDString>(Op->getOperand(1));
2655 if (!ID)
2656 continue;
2657 if (ID->getString() == "Objective-C Image Info Version")
2658 HasObjCFlag = true;
2659 if (ID->getString() == "Objective-C Class Properties")
2660 HasClassProperties = true;
Steven Wu010fc492017-08-21 21:49:13 +00002661 // Upgrade PIC/PIE Module Flags. The module flag behavior for these two
2662 // field was Error and now they are Max.
2663 if (ID->getString() == "PIC Level" || ID->getString() == "PIE Level") {
2664 if (auto *Behavior =
2665 mdconst::dyn_extract_or_null<ConstantInt>(Op->getOperand(0))) {
2666 if (Behavior->getLimitedValue() == Module::Error) {
2667 Type *Int32Ty = Type::getInt32Ty(M.getContext());
2668 Metadata *Ops[3] = {
2669 ConstantAsMetadata::get(ConstantInt::get(Int32Ty, Module::Max)),
2670 MDString::get(M.getContext(), ID->getString()),
2671 Op->getOperand(2)};
2672 ModFlags->setOperand(I, MDNode::get(M.getContext(), Ops));
2673 Changed = true;
2674 }
2675 }
2676 }
Steven Wuab211df2017-09-15 21:12:14 +00002677 // Upgrade Objective-C Image Info Section. Removed the whitespce in the
2678 // section name so that llvm-lto will not complain about mismatching
2679 // module flags that is functionally the same.
2680 if (ID->getString() == "Objective-C Image Info Section") {
2681 if (auto *Value = dyn_cast_or_null<MDString>(Op->getOperand(2))) {
2682 SmallVector<StringRef, 4> ValueComp;
2683 Value->getString().split(ValueComp, " ");
2684 if (ValueComp.size() != 1) {
2685 std::string NewValue;
2686 for (auto &S : ValueComp)
2687 NewValue += S.str();
2688 Metadata *Ops[3] = {Op->getOperand(0), Op->getOperand(1),
2689 MDString::get(M.getContext(), NewValue)};
2690 ModFlags->setOperand(I, MDNode::get(M.getContext(), Ops));
2691 Changed = true;
2692 }
2693 }
2694 }
Manman Renb5d7ff42016-05-25 23:14:48 +00002695 }
Steven Wu010fc492017-08-21 21:49:13 +00002696
Manman Renb5d7ff42016-05-25 23:14:48 +00002697 // "Objective-C Class Properties" is recently added for Objective-C. We
2698 // upgrade ObjC bitcodes to contain a "Objective-C Class Properties" module
Mehdi Aminib53b62e2016-09-16 00:38:18 +00002699 // flag of value 0, so we can correclty downgrade this flag when trying to
2700 // link an ObjC bitcode without this module flag with an ObjC bitcode with
2701 // this module flag.
Manman Renb5d7ff42016-05-25 23:14:48 +00002702 if (HasObjCFlag && !HasClassProperties) {
Mehdi Aminib53b62e2016-09-16 00:38:18 +00002703 M.addModuleFlag(llvm::Module::Override, "Objective-C Class Properties",
Manman Renb5d7ff42016-05-25 23:14:48 +00002704 (uint32_t)0);
Steven Wu010fc492017-08-21 21:49:13 +00002705 Changed = true;
Manman Renb5d7ff42016-05-25 23:14:48 +00002706 }
Steven Wu010fc492017-08-21 21:49:13 +00002707
2708 return Changed;
Manman Renb5d7ff42016-05-25 23:14:48 +00002709}
2710
Saleem Abdulrasool46a59fd2017-10-06 18:06:59 +00002711void llvm::UpgradeSectionAttributes(Module &M) {
2712 auto TrimSpaces = [](StringRef Section) -> std::string {
2713 SmallVector<StringRef, 5> Components;
2714 Section.split(Components, ',');
2715
2716 SmallString<32> Buffer;
2717 raw_svector_ostream OS(Buffer);
2718
2719 for (auto Component : Components)
2720 OS << ',' << Component.trim();
2721
2722 return OS.str().substr(1);
2723 };
2724
2725 for (auto &GV : M.globals()) {
2726 if (!GV.hasSection())
2727 continue;
2728
2729 StringRef Section = GV.getSection();
2730
2731 if (!Section.startswith("__DATA, __objc_catlist"))
2732 continue;
2733
2734 // __DATA, __objc_catlist, regular, no_dead_strip
2735 // __DATA,__objc_catlist,regular,no_dead_strip
2736 GV.setSection(TrimSpaces(Section));
2737 }
2738}
2739
Duncan P. N. Exon Smithefe16c82016-03-25 00:56:13 +00002740static bool isOldLoopArgument(Metadata *MD) {
2741 auto *T = dyn_cast_or_null<MDTuple>(MD);
2742 if (!T)
2743 return false;
2744 if (T->getNumOperands() < 1)
2745 return false;
2746 auto *S = dyn_cast_or_null<MDString>(T->getOperand(0));
2747 if (!S)
2748 return false;
2749 return S->getString().startswith("llvm.vectorizer.");
2750}
2751
2752static MDString *upgradeLoopTag(LLVMContext &C, StringRef OldTag) {
2753 StringRef OldPrefix = "llvm.vectorizer.";
2754 assert(OldTag.startswith(OldPrefix) && "Expected old prefix");
2755
2756 if (OldTag == "llvm.vectorizer.unroll")
2757 return MDString::get(C, "llvm.loop.interleave.count");
2758
2759 return MDString::get(
2760 C, (Twine("llvm.loop.vectorize.") + OldTag.drop_front(OldPrefix.size()))
2761 .str());
2762}
2763
2764static Metadata *upgradeLoopArgument(Metadata *MD) {
2765 auto *T = dyn_cast_or_null<MDTuple>(MD);
2766 if (!T)
2767 return MD;
2768 if (T->getNumOperands() < 1)
2769 return MD;
2770 auto *OldTag = dyn_cast_or_null<MDString>(T->getOperand(0));
2771 if (!OldTag)
2772 return MD;
2773 if (!OldTag->getString().startswith("llvm.vectorizer."))
2774 return MD;
2775
2776 // This has an old tag. Upgrade it.
2777 SmallVector<Metadata *, 8> Ops;
2778 Ops.reserve(T->getNumOperands());
2779 Ops.push_back(upgradeLoopTag(T->getContext(), OldTag->getString()));
2780 for (unsigned I = 1, E = T->getNumOperands(); I != E; ++I)
2781 Ops.push_back(T->getOperand(I));
2782
2783 return MDTuple::get(T->getContext(), Ops);
2784}
2785
2786MDNode *llvm::upgradeInstructionLoopAttachment(MDNode &N) {
2787 auto *T = dyn_cast<MDTuple>(&N);
2788 if (!T)
2789 return &N;
2790
David Majnemer0a16c222016-08-11 21:15:00 +00002791 if (none_of(T->operands(), isOldLoopArgument))
Duncan P. N. Exon Smithefe16c82016-03-25 00:56:13 +00002792 return &N;
2793
2794 SmallVector<Metadata *, 8> Ops;
2795 Ops.reserve(T->getNumOperands());
2796 for (Metadata *MD : T->operands())
2797 Ops.push_back(upgradeLoopArgument(MD));
2798
2799 return MDTuple::get(T->getContext(), Ops);
Eli Bendersky5d5e18d2014-06-25 15:41:00 +00002800}