blob: 65cc2d4f8586fdad23c3633f2c99b990d7b5788a [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"
Chandler Carruth1305dc32014-03-04 11:45:46 +000017#include "llvm/IR/CFG.h"
Chandler Carruth219b89b2014-03-04 11:01:28 +000018#include "llvm/IR/CallSite.h"
Chandler Carruth9fb823b2013-01-02 11:36:10 +000019#include "llvm/IR/Constants.h"
Chandler Carruthd9903882015-01-14 11:23:27 +000020#include "llvm/IR/DIBuilder.h"
Chandler Carruth9a4c9e52014-03-06 00:46:21 +000021#include "llvm/IR/DebugInfo.h"
Manman Ren2ebfb422014-01-16 01:51:12 +000022#include "llvm/IR/DiagnosticInfo.h"
Chandler Carruth9fb823b2013-01-02 11:36:10 +000023#include "llvm/IR/Function.h"
24#include "llvm/IR/IRBuilder.h"
25#include "llvm/IR/Instruction.h"
26#include "llvm/IR/IntrinsicInst.h"
27#include "llvm/IR/LLVMContext.h"
28#include "llvm/IR/Module.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
Nadav Rotem17ee58a2012-06-10 18:42:51 +000034// Upgrade the declarations of the SSE4.1 functions whose arguments have
35// changed their type from v4f32 to v2i64.
36static bool UpgradeSSE41Function(Function* F, Intrinsic::ID IID,
37 Function *&NewFn) {
38 // Check whether this is an old version of the function, which received
39 // v4f32 arguments.
40 Type *Arg0Type = F->getFunctionType()->getParamType(0);
41 if (Arg0Type != VectorType::get(Type::getFloatTy(F->getContext()), 4))
42 return false;
43
44 // Yes, it's old, replace it with new version.
45 F->setName(F->getName() + ".old");
46 NewFn = Intrinsic::getDeclaration(F->getParent(), IID);
47 return true;
48}
Chandler Carruth7132e002007-08-04 01:51:18 +000049
Chandler Carruth373b2b12014-09-06 10:00:01 +000050// Upgrade the declarations of intrinsic functions whose 8-bit immediate mask
51// arguments have changed their type from i32 to i8.
52static bool UpgradeX86IntrinsicsWith8BitMask(Function *F, Intrinsic::ID IID,
53 Function *&NewFn) {
54 // Check that the last argument is an i32.
55 Type *LastArgType = F->getFunctionType()->getParamType(
56 F->getFunctionType()->getNumParams() - 1);
57 if (!LastArgType->isIntegerTy(32))
58 return false;
59
60 // Move this function aside and map down.
61 F->setName(F->getName() + ".old");
62 NewFn = Intrinsic::getDeclaration(F->getParent(), IID);
63 return true;
64}
65
Evan Cheng0e179d02007-12-17 22:33:23 +000066static bool UpgradeIntrinsicFunction1(Function *F, Function *&NewFn) {
Chandler Carruth7132e002007-08-04 01:51:18 +000067 assert(F && "Illegal to upgrade a non-existent Function.");
68
Chandler Carruth7132e002007-08-04 01:51:18 +000069 // Quickly eliminate it, if it's not a candidate.
Chris Lattnerb372f662011-06-18 18:56:39 +000070 StringRef Name = F->getName();
71 if (Name.size() <= 8 || !Name.startswith("llvm."))
Evan Cheng0e179d02007-12-17 22:33:23 +000072 return false;
Chris Lattnerb372f662011-06-18 18:56:39 +000073 Name = Name.substr(5); // Strip off "llvm."
Chris Lattner0bcbde42011-11-27 08:42:07 +000074
Chris Lattnerb372f662011-06-18 18:56:39 +000075 switch (Name[0]) {
Chandler Carruth7132e002007-08-04 01:51:18 +000076 default: break;
Joel Jones43cb8782012-07-13 23:25:25 +000077 case 'a': {
78 if (Name.startswith("arm.neon.vclz")) {
79 Type* args[2] = {
Matt Arsenaultc4c92262013-07-20 17:46:00 +000080 F->arg_begin()->getType(),
Joel Jones43cb8782012-07-13 23:25:25 +000081 Type::getInt1Ty(F->getContext())
82 };
83 // Can't use Intrinsic::getDeclaration here as it adds a ".i1" to
84 // the end of the name. Change name from llvm.arm.neon.vclz.* to
85 // llvm.ctlz.*
86 FunctionType* fType = FunctionType::get(F->getReturnType(), args, false);
Matt Arsenaultc4c92262013-07-20 17:46:00 +000087 NewFn = Function::Create(fType, F->getLinkage(),
Joel Jones43cb8782012-07-13 23:25:25 +000088 "llvm.ctlz." + Name.substr(14), F->getParent());
89 return true;
90 }
Joel Jonesb84f7be2012-07-18 00:02:16 +000091 if (Name.startswith("arm.neon.vcnt")) {
92 NewFn = Intrinsic::getDeclaration(F->getParent(), Intrinsic::ctpop,
93 F->arg_begin()->getType());
94 return true;
95 }
Jeroen Ketemaab99b592015-09-30 10:56:37 +000096 Regex vldRegex("^arm\\.neon\\.vld([1234]|[234]lane)\\.v[a-z0-9]*$");
97 if (vldRegex.match(Name)) {
98 auto fArgs = F->getFunctionType()->params();
99 SmallVector<Type *, 4> Tys(fArgs.begin(), fArgs.end());
100 // Can't use Intrinsic::getDeclaration here as the return types might
101 // then only be structurally equal.
102 FunctionType* fType = FunctionType::get(F->getReturnType(), Tys, false);
103 NewFn = Function::Create(fType, F->getLinkage(),
104 "llvm." + Name + ".p0i8", F->getParent());
105 return true;
106 }
107 Regex vstRegex("^arm\\.neon\\.vst([1234]|[234]lane)\\.v[a-z0-9]*$");
108 if (vstRegex.match(Name)) {
Craig Topper26260942015-10-18 05:15:34 +0000109 static const Intrinsic::ID StoreInts[] = {Intrinsic::arm_neon_vst1,
110 Intrinsic::arm_neon_vst2,
111 Intrinsic::arm_neon_vst3,
112 Intrinsic::arm_neon_vst4};
Jeroen Ketemaab99b592015-09-30 10:56:37 +0000113
Craig Topper26260942015-10-18 05:15:34 +0000114 static const Intrinsic::ID StoreLaneInts[] = {
115 Intrinsic::arm_neon_vst2lane, Intrinsic::arm_neon_vst3lane,
116 Intrinsic::arm_neon_vst4lane
117 };
Jeroen Ketemaab99b592015-09-30 10:56:37 +0000118
119 auto fArgs = F->getFunctionType()->params();
120 Type *Tys[] = {fArgs[0], fArgs[1]};
121 if (Name.find("lane") == StringRef::npos)
122 NewFn = Intrinsic::getDeclaration(F->getParent(),
123 StoreInts[fArgs.size() - 3], Tys);
124 else
125 NewFn = Intrinsic::getDeclaration(F->getParent(),
126 StoreLaneInts[fArgs.size() - 5], Tys);
127 return true;
128 }
Marcin Koscielnicki3fdc2572016-04-19 20:51:05 +0000129 if (Name == "aarch64.thread.pointer" || Name == "arm.thread.pointer") {
130 NewFn = Intrinsic::getDeclaration(F->getParent(), Intrinsic::thread_pointer);
131 return true;
132 }
Joel Jones43cb8782012-07-13 23:25:25 +0000133 break;
134 }
Jeroen Ketemaab99b592015-09-30 10:56:37 +0000135
Chandler Carruth58a71ed2011-12-12 04:26:04 +0000136 case 'c': {
Chandler Carruth58a71ed2011-12-12 04:26:04 +0000137 if (Name.startswith("ctlz.") && F->arg_size() == 1) {
138 F->setName(Name + ".old");
Chandler Carruthd4a02402011-12-12 10:57:20 +0000139 NewFn = Intrinsic::getDeclaration(F->getParent(), Intrinsic::ctlz,
140 F->arg_begin()->getType());
Chandler Carruth58a71ed2011-12-12 04:26:04 +0000141 return true;
142 }
143 if (Name.startswith("cttz.") && F->arg_size() == 1) {
144 F->setName(Name + ".old");
Chandler Carruthd4a02402011-12-12 10:57:20 +0000145 NewFn = Intrinsic::getDeclaration(F->getParent(), Intrinsic::cttz,
146 F->arg_begin()->getType());
Chandler Carruth58a71ed2011-12-12 04:26:04 +0000147 return true;
148 }
149 break;
150 }
Adrian Prantl87b7eb92014-10-01 18:55:02 +0000151
Artur Pilipenko7ad95ec2016-06-28 18:27:25 +0000152 case 'm': {
153 if (Name.startswith("masked.load.")) {
154 Type *Tys[] = { F->getReturnType(), F->arg_begin()->getType() };
155 if (F->getName() != Intrinsic::getName(Intrinsic::masked_load, Tys)) {
156 F->setName(Name + ".old");
157 NewFn = Intrinsic::getDeclaration(F->getParent(),
158 Intrinsic::masked_load,
159 Tys);
160 return true;
161 }
162 }
163 if (Name.startswith("masked.store.")) {
164 auto Args = F->getFunctionType()->params();
165 Type *Tys[] = { Args[0], Args[1] };
166 if (F->getName() != Intrinsic::getName(Intrinsic::masked_store, Tys)) {
167 F->setName(Name + ".old");
168 NewFn = Intrinsic::getDeclaration(F->getParent(),
169 Intrinsic::masked_store,
170 Tys);
171 return true;
172 }
173 }
174 break;
175 }
176
Matt Arsenaultfbcbce42013-10-07 18:06:48 +0000177 case 'o':
178 // We only need to change the name to match the mangling including the
179 // address space.
180 if (F->arg_size() == 2 && Name.startswith("objectsize.")) {
181 Type *Tys[2] = { F->getReturnType(), F->arg_begin()->getType() };
182 if (F->getName() != Intrinsic::getName(Intrinsic::objectsize, Tys)) {
183 F->setName(Name + ".old");
184 NewFn = Intrinsic::getDeclaration(F->getParent(),
185 Intrinsic::objectsize, Tys);
186 return true;
187 }
188 }
189 break;
190
Tim Shen00127562016-04-08 21:26:31 +0000191 case 's':
192 if (Name == "stackprotectorcheck") {
193 NewFn = nullptr;
194 return true;
195 }
196
Craig Topper3b1817d2012-02-03 06:10:55 +0000197 case 'x': {
198 if (Name.startswith("x86.sse2.pcmpeq.") ||
199 Name.startswith("x86.sse2.pcmpgt.") ||
200 Name.startswith("x86.avx2.pcmpeq.") ||
Craig Topperd3c9e402012-04-18 05:24:00 +0000201 Name.startswith("x86.avx2.pcmpgt.") ||
Craig Topper0a0fb0f2016-06-21 03:53:24 +0000202 Name.startswith("x86.avx512.mask.pcmpeq.") ||
203 Name.startswith("x86.avx512.mask.pcmpgt.") ||
Sanjay Patel51ab7572016-06-16 15:48:30 +0000204 Name == "x86.sse41.pmaxsb" ||
205 Name == "x86.sse2.pmaxs.w" ||
206 Name == "x86.sse41.pmaxsd" ||
207 Name == "x86.sse2.pmaxu.b" ||
208 Name == "x86.sse41.pmaxuw" ||
209 Name == "x86.sse41.pmaxud" ||
210 Name == "x86.sse41.pminsb" ||
211 Name == "x86.sse2.pmins.w" ||
212 Name == "x86.sse41.pminsd" ||
213 Name == "x86.sse2.pminu.b" ||
214 Name == "x86.sse41.pminuw" ||
215 Name == "x86.sse41.pminud" ||
Sanjay Patel0e9afea2016-06-16 18:44:20 +0000216 Name.startswith("x86.avx2.pmax") ||
217 Name.startswith("x86.avx2.pmin") ||
Ahmed Bougacha1a4987052015-08-20 20:36:19 +0000218 Name.startswith("x86.avx2.vbroadcast") ||
219 Name.startswith("x86.avx2.pbroadcast") ||
Craig Topper7daf8972012-05-08 06:58:15 +0000220 Name.startswith("x86.avx.vpermil.") ||
Craig Topper10679862016-06-12 14:11:32 +0000221 Name.startswith("x86.sse2.pshuf") ||
Simon Pilgrim19adee92016-07-02 14:42:35 +0000222 Name.startswith("x86.avx512.mask.movddup") ||
223 Name.startswith("x86.avx512.mask.movshdup") ||
224 Name.startswith("x86.avx512.mask.movsldup") ||
Craig Topper13cf7ca2016-06-13 02:36:48 +0000225 Name.startswith("x86.avx512.mask.pshuf.d.") ||
226 Name.startswith("x86.avx512.mask.pshufl.w.") ||
227 Name.startswith("x86.avx512.mask.pshufh.w.") ||
Simon Pilgrim9fca3002016-07-04 12:40:54 +0000228 Name.startswith("x86.avx512.mask.vpermil.p") ||
Craig Topper597aa422016-06-23 07:37:33 +0000229 Name.startswith("x86.avx512.mask.punpckl") ||
230 Name.startswith("x86.avx512.mask.punpckh") ||
231 Name.startswith("x86.avx512.mask.unpckl.") ||
232 Name.startswith("x86.avx512.mask.unpckh.") ||
Simon Pilgrim9cb018b2015-09-23 08:48:33 +0000233 Name.startswith("x86.sse41.pmovsx") ||
Simon Pilgrim9602d672016-05-28 18:03:41 +0000234 Name.startswith("x86.sse41.pmovzx") ||
235 Name.startswith("x86.avx2.pmovsx") ||
236 Name.startswith("x86.avx2.pmovzx") ||
Simon Pilgrim4298d062016-05-25 08:59:18 +0000237 Name == "x86.sse2.cvtdq2pd" ||
238 Name == "x86.sse2.cvtps2pd" ||
239 Name == "x86.avx.cvtdq2.pd.256" ||
240 Name == "x86.avx.cvt.ps2.pd.256" ||
Simon Pilgrim0afd5a42016-06-02 10:55:21 +0000241 Name == "x86.sse2.cvttps2dq" ||
242 Name.startswith("x86.avx.cvtt.") ||
Craig Topperf9f1ecc2016-05-30 22:54:05 +0000243 Name.startswith("x86.avx.vinsertf128.") ||
Sanjay Patel4339abe2015-03-12 23:16:18 +0000244 Name == "x86.avx2.vinserti128" ||
Craig Topperf9f1ecc2016-05-30 22:54:05 +0000245 Name.startswith("x86.avx.vextractf128.") ||
Sanjay Patel4339abe2015-03-12 23:16:18 +0000246 Name == "x86.avx2.vextracti128" ||
Simon Pilgrimf4b2af12016-06-18 02:38:26 +0000247 Name.startswith("x86.sse4a.movnt.") ||
Craig Topperf9f1ecc2016-05-30 22:54:05 +0000248 Name.startswith("x86.avx.movnt.") ||
Craig Topper12e322a2016-05-25 06:56:32 +0000249 Name == "x86.sse2.storel.dq" ||
Craig Topper8287fd82016-05-30 23:15:56 +0000250 Name.startswith("x86.sse.storeu.") ||
251 Name.startswith("x86.sse2.storeu.") ||
252 Name.startswith("x86.avx.storeu.") ||
Craig Topper50f85c22016-05-31 01:50:02 +0000253 Name.startswith("x86.avx512.mask.storeu.p") ||
254 Name.startswith("x86.avx512.mask.storeu.b.") ||
255 Name.startswith("x86.avx512.mask.storeu.w.") ||
256 Name.startswith("x86.avx512.mask.storeu.d.") ||
257 Name.startswith("x86.avx512.mask.storeu.q.") ||
258 Name.startswith("x86.avx512.mask.store.p") ||
259 Name.startswith("x86.avx512.mask.store.b.") ||
260 Name.startswith("x86.avx512.mask.store.w.") ||
261 Name.startswith("x86.avx512.mask.store.d.") ||
262 Name.startswith("x86.avx512.mask.store.q.") ||
Craig Topperf10fbfa2016-06-02 04:19:36 +0000263 Name.startswith("x86.avx512.mask.loadu.p") ||
264 Name.startswith("x86.avx512.mask.loadu.b.") ||
265 Name.startswith("x86.avx512.mask.loadu.w.") ||
266 Name.startswith("x86.avx512.mask.loadu.d.") ||
267 Name.startswith("x86.avx512.mask.loadu.q.") ||
268 Name.startswith("x86.avx512.mask.load.p") ||
269 Name.startswith("x86.avx512.mask.load.b.") ||
270 Name.startswith("x86.avx512.mask.load.w.") ||
271 Name.startswith("x86.avx512.mask.load.d.") ||
272 Name.startswith("x86.avx512.mask.load.q.") ||
Craig Topperef9e9932013-10-15 05:20:47 +0000273 Name == "x86.sse42.crc32.64.8" ||
Craig Topperf9f1ecc2016-05-30 22:54:05 +0000274 Name.startswith("x86.avx.vbroadcast.s") ||
Craig Topper33350cc2016-06-06 06:12:54 +0000275 Name.startswith("x86.avx512.mask.palignr.") ||
Craig Topperf9f1ecc2016-05-30 22:54:05 +0000276 Name.startswith("x86.sse2.psll.dq") ||
277 Name.startswith("x86.sse2.psrl.dq") ||
278 Name.startswith("x86.avx2.psll.dq") ||
279 Name.startswith("x86.avx2.psrl.dq") ||
Simon Pilgrimf7186822016-06-09 21:09:03 +0000280 Name.startswith("x86.avx512.psll.dq") ||
281 Name.startswith("x86.avx512.psrl.dq") ||
Craig Topper782d6202015-02-28 19:33:17 +0000282 Name == "x86.sse41.pblendw" ||
Craig Topperf9f1ecc2016-05-30 22:54:05 +0000283 Name.startswith("x86.sse41.blendp") ||
284 Name.startswith("x86.avx.blend.p") ||
Craig Topper782d6202015-02-28 19:33:17 +0000285 Name == "x86.avx2.pblendw" ||
Craig Topperf9f1ecc2016-05-30 22:54:05 +0000286 Name.startswith("x86.avx2.pblendd.") ||
Juergen Ributzka1f7a1762015-03-04 00:13:25 +0000287 Name == "x86.avx2.vbroadcasti128" ||
Simon Pilgrime88dc042015-11-03 20:27:01 +0000288 Name == "x86.xop.vpcmov" ||
Craig Topper3352ba52012-06-09 16:46:13 +0000289 (Name.startswith("x86.xop.vpcom") && F->arg_size() == 2)) {
Craig Topperc6207612014-04-09 06:08:46 +0000290 NewFn = nullptr;
Craig Topper3b1817d2012-02-03 06:10:55 +0000291 return true;
292 }
Nadav Rotem17ee58a2012-06-10 18:42:51 +0000293 // SSE4.1 ptest functions may have an old signature.
294 if (Name.startswith("x86.sse41.ptest")) {
295 if (Name == "x86.sse41.ptestc")
296 return UpgradeSSE41Function(F, Intrinsic::x86_sse41_ptestc, NewFn);
297 if (Name == "x86.sse41.ptestz")
298 return UpgradeSSE41Function(F, Intrinsic::x86_sse41_ptestz, NewFn);
299 if (Name == "x86.sse41.ptestnzc")
300 return UpgradeSSE41Function(F, Intrinsic::x86_sse41_ptestnzc, NewFn);
301 }
Sanjay Patel1c3eaec2015-02-28 22:25:06 +0000302 // Several blend and other instructions with masks used the wrong number of
Chandler Carruth373b2b12014-09-06 10:00:01 +0000303 // bits.
Chandler Carruth373b2b12014-09-06 10:00:01 +0000304 if (Name == "x86.sse41.insertps")
305 return UpgradeX86IntrinsicsWith8BitMask(F, Intrinsic::x86_sse41_insertps,
306 NewFn);
307 if (Name == "x86.sse41.dppd")
308 return UpgradeX86IntrinsicsWith8BitMask(F, Intrinsic::x86_sse41_dppd,
309 NewFn);
310 if (Name == "x86.sse41.dpps")
311 return UpgradeX86IntrinsicsWith8BitMask(F, Intrinsic::x86_sse41_dpps,
312 NewFn);
313 if (Name == "x86.sse41.mpsadbw")
314 return UpgradeX86IntrinsicsWith8BitMask(F, Intrinsic::x86_sse41_mpsadbw,
315 NewFn);
Chandler Carruth373b2b12014-09-06 10:00:01 +0000316 if (Name == "x86.avx.dp.ps.256")
317 return UpgradeX86IntrinsicsWith8BitMask(F, Intrinsic::x86_avx_dp_ps_256,
318 NewFn);
Chandler Carruth373b2b12014-09-06 10:00:01 +0000319 if (Name == "x86.avx2.mpsadbw")
320 return UpgradeX86IntrinsicsWith8BitMask(F, Intrinsic::x86_avx2_mpsadbw,
321 NewFn);
Craig Topper29f2e952015-01-25 23:26:02 +0000322
Craig Topper71dc02d2012-06-13 07:18:53 +0000323 // frcz.ss/sd may need to have an argument dropped
324 if (Name.startswith("x86.xop.vfrcz.ss") && F->arg_size() == 2) {
325 F->setName(Name + ".old");
326 NewFn = Intrinsic::getDeclaration(F->getParent(),
327 Intrinsic::x86_xop_vfrcz_ss);
328 return true;
329 }
330 if (Name.startswith("x86.xop.vfrcz.sd") && F->arg_size() == 2) {
331 F->setName(Name + ".old");
332 NewFn = Intrinsic::getDeclaration(F->getParent(),
333 Intrinsic::x86_xop_vfrcz_sd);
334 return true;
335 }
Craig Topper720c7bd2012-06-03 08:07:25 +0000336 // Fix the FMA4 intrinsics to remove the 4
337 if (Name.startswith("x86.fma4.")) {
Craig Topper2c5ccd82012-06-03 16:48:52 +0000338 F->setName("llvm.x86.fma" + Name.substr(8));
339 NewFn = F;
340 return true;
Craig Topper720c7bd2012-06-03 08:07:25 +0000341 }
Simon Pilgrime85506b2016-06-03 08:06:03 +0000342 // Upgrade any XOP PERMIL2 index operand still using a float/double vector.
343 if (Name.startswith("x86.xop.vpermil2")) {
344 auto Params = F->getFunctionType()->params();
345 auto Idx = Params[2];
346 if (Idx->getScalarType()->isFloatingPointTy()) {
347 F->setName(Name + ".old");
348 unsigned IdxSize = Idx->getPrimitiveSizeInBits();
349 unsigned EltSize = Idx->getScalarSizeInBits();
350 Intrinsic::ID Permil2ID;
351 if (EltSize == 64 && IdxSize == 128)
352 Permil2ID = Intrinsic::x86_xop_vpermil2pd;
353 else if (EltSize == 32 && IdxSize == 128)
354 Permil2ID = Intrinsic::x86_xop_vpermil2ps;
355 else if (EltSize == 64 && IdxSize == 256)
356 Permil2ID = Intrinsic::x86_xop_vpermil2pd_256;
357 else
358 Permil2ID = Intrinsic::x86_xop_vpermil2ps_256;
359 NewFn = Intrinsic::getDeclaration(F->getParent(), Permil2ID);
360 return true;
361 }
362 }
Craig Topper3b1817d2012-02-03 06:10:55 +0000363 break;
364 }
Chris Lattnerb372f662011-06-18 18:56:39 +0000365 }
Chandler Carruth7132e002007-08-04 01:51:18 +0000366
Nadav Rotem17ee58a2012-06-10 18:42:51 +0000367 // This may not belong here. This function is effectively being overloaded
368 // to both detect an intrinsic which needs upgrading, and to provide the
369 // upgraded form of the intrinsic. We should perhaps have two separate
Chandler Carruth7132e002007-08-04 01:51:18 +0000370 // functions for this.
Evan Cheng0e179d02007-12-17 22:33:23 +0000371 return false;
Chandler Carruth7132e002007-08-04 01:51:18 +0000372}
373
Evan Cheng0e179d02007-12-17 22:33:23 +0000374bool llvm::UpgradeIntrinsicFunction(Function *F, Function *&NewFn) {
Craig Topperc6207612014-04-09 06:08:46 +0000375 NewFn = nullptr;
Evan Cheng0e179d02007-12-17 22:33:23 +0000376 bool Upgraded = UpgradeIntrinsicFunction1(F, NewFn);
Filipe Cabecinhas0011c582015-07-03 20:12:01 +0000377 assert(F != NewFn && "Intrinsic function upgraded to the same function");
Duncan Sands38ef3a82007-12-03 20:06:50 +0000378
379 // Upgrade intrinsic attributes. This does not change the function.
Evan Cheng0e179d02007-12-17 22:33:23 +0000380 if (NewFn)
381 F = NewFn;
Pete Cooper9e1d3352015-05-20 17:16:39 +0000382 if (Intrinsic::ID id = F->getIntrinsicID())
383 F->setAttributes(Intrinsic::getAttributes(F->getContext(), id));
Duncan Sands38ef3a82007-12-03 20:06:50 +0000384 return Upgraded;
385}
386
Bill Wendlinge26fffc2010-09-10 18:51:56 +0000387bool llvm::UpgradeGlobalVariable(GlobalVariable *GV) {
Chris Lattner80ed9dc2011-06-18 06:05:24 +0000388 // Nothing to do yet.
Bill Wendlinge26fffc2010-09-10 18:51:56 +0000389 return false;
390}
391
Simon Pilgrimf7186822016-06-09 21:09:03 +0000392// Handles upgrading SSE2/AVX2/AVX512BW PSLLDQ intrinsics by converting them
Craig Topperb324e432015-02-18 06:24:44 +0000393// to byte shuffles.
394static Value *UpgradeX86PSLLDQIntrinsics(IRBuilder<> &Builder, LLVMContext &C,
Craig Topper7355ac32016-05-29 06:37:33 +0000395 Value *Op, unsigned Shift) {
396 Type *ResultTy = Op->getType();
397 unsigned NumElts = ResultTy->getVectorNumElements() * 8;
Craig Topperb324e432015-02-18 06:24:44 +0000398
399 // Bitcast from a 64-bit element type to a byte element type.
Craig Topper7355ac32016-05-29 06:37:33 +0000400 Type *VecTy = VectorType::get(Type::getInt8Ty(C), NumElts);
401 Op = Builder.CreateBitCast(Op, VecTy, "cast");
402
Craig Topperb324e432015-02-18 06:24:44 +0000403 // We'll be shuffling in zeroes.
Craig Topper7355ac32016-05-29 06:37:33 +0000404 Value *Res = Constant::getNullValue(VecTy);
Craig Topperb324e432015-02-18 06:24:44 +0000405
406 // If shift is less than 16, emit a shuffle to move the bytes. Otherwise,
407 // we'll just return the zero vector.
408 if (Shift < 16) {
Craig Topper99d1eab2016-06-12 00:41:19 +0000409 uint32_t Idxs[64];
Simon Pilgrimf7186822016-06-09 21:09:03 +0000410 // 256/512-bit version is split into 2/4 16-byte lanes.
Craig Topperb324e432015-02-18 06:24:44 +0000411 for (unsigned l = 0; l != NumElts; l += 16)
412 for (unsigned i = 0; i != 16; ++i) {
413 unsigned Idx = NumElts + i - Shift;
414 if (Idx < NumElts)
415 Idx -= NumElts - 16; // end of lane, switch operand.
Craig Topper7355ac32016-05-29 06:37:33 +0000416 Idxs[l + i] = Idx + l;
Craig Topperb324e432015-02-18 06:24:44 +0000417 }
418
Craig Topper7355ac32016-05-29 06:37:33 +0000419 Res = Builder.CreateShuffleVector(Res, Op, makeArrayRef(Idxs, NumElts));
Craig Topperb324e432015-02-18 06:24:44 +0000420 }
421
422 // Bitcast back to a 64-bit element type.
Craig Topper7355ac32016-05-29 06:37:33 +0000423 return Builder.CreateBitCast(Res, ResultTy, "cast");
Craig Topperb324e432015-02-18 06:24:44 +0000424}
425
Craig Topperea703ae2016-06-13 02:36:42 +0000426// Handles upgrading SSE2/AVX2/AVX512BW PSRLDQ intrinsics by converting them
427// to byte shuffles.
428static Value *UpgradeX86PSRLDQIntrinsics(IRBuilder<> &Builder, LLVMContext &C,
429 Value *Op,
430 unsigned Shift) {
431 Type *ResultTy = Op->getType();
432 unsigned NumElts = ResultTy->getVectorNumElements() * 8;
433
434 // Bitcast from a 64-bit element type to a byte element type.
435 Type *VecTy = VectorType::get(Type::getInt8Ty(C), NumElts);
436 Op = Builder.CreateBitCast(Op, VecTy, "cast");
437
438 // We'll be shuffling in zeroes.
439 Value *Res = Constant::getNullValue(VecTy);
440
441 // If shift is less than 16, emit a shuffle to move the bytes. Otherwise,
442 // we'll just return the zero vector.
443 if (Shift < 16) {
444 uint32_t Idxs[64];
445 // 256/512-bit version is split into 2/4 16-byte lanes.
446 for (unsigned l = 0; l != NumElts; l += 16)
447 for (unsigned i = 0; i != 16; ++i) {
448 unsigned Idx = i + Shift;
449 if (Idx >= 16)
450 Idx += NumElts - 16; // end of lane, switch operand.
451 Idxs[l + i] = Idx + l;
452 }
453
454 Res = Builder.CreateShuffleVector(Op, Res, makeArrayRef(Idxs, NumElts));
455 }
456
457 // Bitcast back to a 64-bit element type.
458 return Builder.CreateBitCast(Res, ResultTy, "cast");
459}
460
461static Value *getX86MaskVec(IRBuilder<> &Builder, Value *Mask,
462 unsigned NumElts) {
463 llvm::VectorType *MaskTy = llvm::VectorType::get(Builder.getInt1Ty(),
464 cast<IntegerType>(Mask->getType())->getBitWidth());
465 Mask = Builder.CreateBitCast(Mask, MaskTy);
466
467 // If we have less than 8 elements, then the starting mask was an i8 and
468 // we need to extract down to the right number of elements.
469 if (NumElts < 8) {
470 uint32_t Indices[4];
471 for (unsigned i = 0; i != NumElts; ++i)
472 Indices[i] = i;
473 Mask = Builder.CreateShuffleVector(Mask, Mask,
474 makeArrayRef(Indices, NumElts),
475 "extract");
476 }
477
478 return Mask;
479}
480
481static Value *EmitX86Select(IRBuilder<> &Builder, Value *Mask,
482 Value *Op0, Value *Op1) {
483 // If the mask is all ones just emit the align operation.
484 if (const auto *C = dyn_cast<Constant>(Mask))
485 if (C->isAllOnesValue())
486 return Op0;
487
488 Mask = getX86MaskVec(Builder, Mask, Op0->getType()->getVectorNumElements());
489 return Builder.CreateSelect(Mask, Op0, Op1);
490}
491
Craig Topper33350cc2016-06-06 06:12:54 +0000492static Value *UpgradeX86PALIGNRIntrinsics(IRBuilder<> &Builder, LLVMContext &C,
493 Value *Op0, Value *Op1, Value *Shift,
494 Value *Passthru, Value *Mask) {
495 unsigned ShiftVal = cast<llvm::ConstantInt>(Shift)->getZExtValue();
496
497 unsigned NumElts = Op0->getType()->getVectorNumElements();
498 assert(NumElts % 16 == 0);
499
500 // If palignr is shifting the pair of vectors more than the size of two
501 // lanes, emit zero.
502 if (ShiftVal >= 32)
503 return llvm::Constant::getNullValue(Op0->getType());
504
505 // If palignr is shifting the pair of input vectors more than one lane,
506 // but less than two lanes, convert to shifting in zeroes.
507 if (ShiftVal > 16) {
508 ShiftVal -= 16;
509 Op1 = Op0;
510 Op0 = llvm::Constant::getNullValue(Op0->getType());
511 }
512
Craig Topper99d1eab2016-06-12 00:41:19 +0000513 uint32_t Indices[64];
Craig Topper33350cc2016-06-06 06:12:54 +0000514 // 256-bit palignr operates on 128-bit lanes so we need to handle that
515 for (unsigned l = 0; l != NumElts; l += 16) {
516 for (unsigned i = 0; i != 16; ++i) {
517 unsigned Idx = ShiftVal + i;
518 if (Idx >= 16)
519 Idx += NumElts - 16; // End of lane, switch operand.
520 Indices[l + i] = Idx + l;
521 }
522 }
523
524 Value *Align = Builder.CreateShuffleVector(Op1, Op0,
525 makeArrayRef(Indices, NumElts),
526 "palignr");
527
Craig Topperea703ae2016-06-13 02:36:42 +0000528 return EmitX86Select(Builder, Mask, Align, Passthru);
Craig Topperb324e432015-02-18 06:24:44 +0000529}
530
Craig Topper50f85c22016-05-31 01:50:02 +0000531static Value *UpgradeMaskedStore(IRBuilder<> &Builder, LLVMContext &C,
532 Value *Ptr, Value *Data, Value *Mask,
533 bool Aligned) {
534 // Cast the pointer to the right type.
535 Ptr = Builder.CreateBitCast(Ptr,
536 llvm::PointerType::getUnqual(Data->getType()));
537 unsigned Align =
538 Aligned ? cast<VectorType>(Data->getType())->getBitWidth() / 8 : 1;
539
540 // If the mask is all ones just emit a regular store.
541 if (const auto *C = dyn_cast<Constant>(Mask))
542 if (C->isAllOnesValue())
543 return Builder.CreateAlignedStore(Data, Ptr, Align);
544
545 // Convert the mask from an integer type to a vector of i1.
546 unsigned NumElts = Data->getType()->getVectorNumElements();
Craig Topperea703ae2016-06-13 02:36:42 +0000547 Mask = getX86MaskVec(Builder, Mask, NumElts);
Craig Topper50f85c22016-05-31 01:50:02 +0000548 return Builder.CreateMaskedStore(Data, Ptr, Align, Mask);
549}
550
Craig Topperf10fbfa2016-06-02 04:19:36 +0000551static Value *UpgradeMaskedLoad(IRBuilder<> &Builder, LLVMContext &C,
552 Value *Ptr, Value *Passthru, Value *Mask,
553 bool Aligned) {
554 // Cast the pointer to the right type.
555 Ptr = Builder.CreateBitCast(Ptr,
556 llvm::PointerType::getUnqual(Passthru->getType()));
557 unsigned Align =
558 Aligned ? cast<VectorType>(Passthru->getType())->getBitWidth() / 8 : 1;
559
560 // If the mask is all ones just emit a regular store.
561 if (const auto *C = dyn_cast<Constant>(Mask))
562 if (C->isAllOnesValue())
563 return Builder.CreateAlignedLoad(Ptr, Align);
564
565 // Convert the mask from an integer type to a vector of i1.
566 unsigned NumElts = Passthru->getType()->getVectorNumElements();
Craig Topperea703ae2016-06-13 02:36:42 +0000567 Mask = getX86MaskVec(Builder, Mask, NumElts);
Craig Topperf10fbfa2016-06-02 04:19:36 +0000568 return Builder.CreateMaskedLoad(Ptr, Align, Mask, Passthru);
569}
570
Sanjay Patel51ab7572016-06-16 15:48:30 +0000571static Value *upgradeIntMinMax(IRBuilder<> &Builder, CallInst &CI,
572 ICmpInst::Predicate Pred) {
573 Value *Op0 = CI.getArgOperand(0);
574 Value *Op1 = CI.getArgOperand(1);
575 Value *Cmp = Builder.CreateICmp(Pred, Op0, Op1);
576 return Builder.CreateSelect(Cmp, Op0, Op1);
577}
578
Craig Topper0a0fb0f2016-06-21 03:53:24 +0000579static Value *upgradeMaskedCompare(IRBuilder<> &Builder, CallInst &CI,
580 ICmpInst::Predicate Pred) {
581 Value *Op0 = CI.getArgOperand(0);
582 unsigned NumElts = Op0->getType()->getVectorNumElements();
583 Value *Cmp = Builder.CreateICmp(Pred, Op0, CI.getArgOperand(1));
584
585 Value *Mask = CI.getArgOperand(2);
586 const auto *C = dyn_cast<Constant>(Mask);
587 if (!C || !C->isAllOnesValue())
588 Cmp = Builder.CreateAnd(Cmp, getX86MaskVec(Builder, Mask, NumElts));
589
590 if (NumElts < 8) {
591 uint32_t Indices[8];
592 for (unsigned i = 0; i != NumElts; ++i)
593 Indices[i] = i;
594 for (unsigned i = NumElts; i != 8; ++i)
595 Indices[i] = NumElts;
596 Cmp = Builder.CreateShuffleVector(Cmp, UndefValue::get(Cmp->getType()),
597 Indices);
598 }
599 return Builder.CreateBitCast(Cmp, IntegerType::get(CI.getContext(),
600 std::max(NumElts, 8U)));
601}
602
Sanjay Patel595098f2016-06-15 22:01:28 +0000603/// Upgrade a call to an old intrinsic. All argument and return casting must be
604/// provided to seamlessly integrate with existing context.
Chandler Carruth7132e002007-08-04 01:51:18 +0000605void llvm::UpgradeIntrinsicCall(CallInst *CI, Function *NewFn) {
Craig Topper3b1817d2012-02-03 06:10:55 +0000606 Function *F = CI->getCalledFunction();
Nick Lewycky2eb3ade2011-12-12 22:59:34 +0000607 LLVMContext &C = CI->getContext();
Chandler Carruth58a71ed2011-12-12 04:26:04 +0000608 IRBuilder<> Builder(C);
Duncan P. N. Exon Smith52888a62015-10-08 23:49:46 +0000609 Builder.SetInsertPoint(CI->getParent(), CI->getIterator());
Chandler Carruth58a71ed2011-12-12 04:26:04 +0000610
Craig Topper3b1817d2012-02-03 06:10:55 +0000611 assert(F && "Intrinsic call is not direct?");
612
613 if (!NewFn) {
614 // Get the Function's name.
615 StringRef Name = F->getName();
616
617 Value *Rep;
Sanjay Patel595098f2016-06-15 22:01:28 +0000618 // Upgrade packed integer vector compare intrinsics to compare instructions.
Craig Topper3b1817d2012-02-03 06:10:55 +0000619 if (Name.startswith("llvm.x86.sse2.pcmpeq.") ||
620 Name.startswith("llvm.x86.avx2.pcmpeq.")) {
621 Rep = Builder.CreateICmpEQ(CI->getArgOperand(0), CI->getArgOperand(1),
622 "pcmpeq");
Craig Topper3b1817d2012-02-03 06:10:55 +0000623 Rep = Builder.CreateSExt(Rep, CI->getType(), "");
624 } else if (Name.startswith("llvm.x86.sse2.pcmpgt.") ||
625 Name.startswith("llvm.x86.avx2.pcmpgt.")) {
626 Rep = Builder.CreateICmpSGT(CI->getArgOperand(0), CI->getArgOperand(1),
627 "pcmpgt");
Craig Topper3b1817d2012-02-03 06:10:55 +0000628 Rep = Builder.CreateSExt(Rep, CI->getType(), "");
Craig Topper0a0fb0f2016-06-21 03:53:24 +0000629 } else if (Name.startswith("llvm.x86.avx512.mask.pcmpeq.")) {
630 Rep = upgradeMaskedCompare(Builder, *CI, ICmpInst::ICMP_EQ);
631 } else if (Name.startswith("llvm.x86.avx512.mask.pcmpgt.")) {
632 Rep = upgradeMaskedCompare(Builder, *CI, ICmpInst::ICMP_SGT);
Sanjay Patel51ab7572016-06-16 15:48:30 +0000633 } else if (Name == "llvm.x86.sse41.pmaxsb" ||
634 Name == "llvm.x86.sse2.pmaxs.w" ||
Sanjay Patel0e9afea2016-06-16 18:44:20 +0000635 Name == "llvm.x86.sse41.pmaxsd" ||
636 Name.startswith("llvm.x86.avx2.pmaxs")) {
Sanjay Patel51ab7572016-06-16 15:48:30 +0000637 Rep = upgradeIntMinMax(Builder, *CI, ICmpInst::ICMP_SGT);
638 } else if (Name == "llvm.x86.sse2.pmaxu.b" ||
639 Name == "llvm.x86.sse41.pmaxuw" ||
Sanjay Patel0e9afea2016-06-16 18:44:20 +0000640 Name == "llvm.x86.sse41.pmaxud" ||
641 Name.startswith("llvm.x86.avx2.pmaxu")) {
Sanjay Patel51ab7572016-06-16 15:48:30 +0000642 Rep = upgradeIntMinMax(Builder, *CI, ICmpInst::ICMP_UGT);
643 } else if (Name == "llvm.x86.sse41.pminsb" ||
644 Name == "llvm.x86.sse2.pmins.w" ||
Sanjay Patel0e9afea2016-06-16 18:44:20 +0000645 Name == "llvm.x86.sse41.pminsd" ||
646 Name.startswith("llvm.x86.avx2.pmins")) {
Sanjay Patel51ab7572016-06-16 15:48:30 +0000647 Rep = upgradeIntMinMax(Builder, *CI, ICmpInst::ICMP_SLT);
648 } else if (Name == "llvm.x86.sse2.pminu.b" ||
649 Name == "llvm.x86.sse41.pminuw" ||
Sanjay Patel0e9afea2016-06-16 18:44:20 +0000650 Name == "llvm.x86.sse41.pminud" ||
651 Name.startswith("llvm.x86.avx2.pminu")) {
Sanjay Patel51ab7572016-06-16 15:48:30 +0000652 Rep = upgradeIntMinMax(Builder, *CI, ICmpInst::ICMP_ULT);
Simon Pilgrim4298d062016-05-25 08:59:18 +0000653 } else if (Name == "llvm.x86.sse2.cvtdq2pd" ||
654 Name == "llvm.x86.sse2.cvtps2pd" ||
655 Name == "llvm.x86.avx.cvtdq2.pd.256" ||
656 Name == "llvm.x86.avx.cvt.ps2.pd.256") {
657 // Lossless i32/float to double conversion.
658 // Extract the bottom elements if necessary and convert to double vector.
659 Value *Src = CI->getArgOperand(0);
660 VectorType *SrcTy = cast<VectorType>(Src->getType());
661 VectorType *DstTy = cast<VectorType>(CI->getType());
662 Rep = CI->getArgOperand(0);
663
664 unsigned NumDstElts = DstTy->getNumElements();
665 if (NumDstElts < SrcTy->getNumElements()) {
666 assert(NumDstElts == 2 && "Unexpected vector size");
Craig Topper99d1eab2016-06-12 00:41:19 +0000667 uint32_t ShuffleMask[2] = { 0, 1 };
668 Rep = Builder.CreateShuffleVector(Rep, UndefValue::get(SrcTy),
669 ShuffleMask);
Simon Pilgrim4298d062016-05-25 08:59:18 +0000670 }
671
672 bool Int2Double = (StringRef::npos != Name.find("cvtdq2"));
673 if (Int2Double)
674 Rep = Builder.CreateSIToFP(Rep, DstTy, "cvtdq2pd");
675 else
676 Rep = Builder.CreateFPExt(Rep, DstTy, "cvtps2pd");
Simon Pilgrim0afd5a42016-06-02 10:55:21 +0000677 } else if (Name == "llvm.x86.sse2.cvttps2dq" ||
678 Name.startswith("llvm.x86.avx.cvtt.")) {
679 // Truncation (round to zero) float/double to i32 vector conversion.
680 Value *Src = CI->getArgOperand(0);
681 VectorType *DstTy = cast<VectorType>(CI->getType());
682 Rep = Builder.CreateFPToSI(Src, DstTy, "cvtt");
Simon Pilgrimf4b2af12016-06-18 02:38:26 +0000683 } else if (Name.startswith("llvm.x86.sse4a.movnt.")) {
684 Module *M = F->getParent();
685 SmallVector<Metadata *, 1> Elts;
686 Elts.push_back(
687 ConstantAsMetadata::get(ConstantInt::get(Type::getInt32Ty(C), 1)));
688 MDNode *Node = MDNode::get(C, Elts);
689
690 Value *Arg0 = CI->getArgOperand(0);
691 Value *Arg1 = CI->getArgOperand(1);
692
693 // Nontemporal (unaligned) store of the 0'th element of the float/double
694 // vector.
695 Type *SrcEltTy = cast<VectorType>(Arg1->getType())->getElementType();
696 PointerType *EltPtrTy = PointerType::getUnqual(SrcEltTy);
697 Value *Addr = Builder.CreateBitCast(Arg0, EltPtrTy, "cast");
698 Value *Extract =
699 Builder.CreateExtractElement(Arg1, (uint64_t)0, "extractelement");
700
701 StoreInst *SI = Builder.CreateAlignedStore(Extract, Addr, 1);
702 SI->setMetadata(M->getMDKindID("nontemporal"), Node);
703
704 // Remove intrinsic.
705 CI->eraseFromParent();
706 return;
Craig Topperf9f1ecc2016-05-30 22:54:05 +0000707 } else if (Name.startswith("llvm.x86.avx.movnt.")) {
Craig Topper7daf8972012-05-08 06:58:15 +0000708 Module *M = F->getParent();
Duncan P. N. Exon Smith5bf8fef2014-12-09 18:38:53 +0000709 SmallVector<Metadata *, 1> Elts;
710 Elts.push_back(
711 ConstantAsMetadata::get(ConstantInt::get(Type::getInt32Ty(C), 1)));
Craig Topper7daf8972012-05-08 06:58:15 +0000712 MDNode *Node = MDNode::get(C, Elts);
713
714 Value *Arg0 = CI->getArgOperand(0);
715 Value *Arg1 = CI->getArgOperand(1);
716
717 // Convert the type of the pointer to a pointer to the stored type.
718 Value *BC = Builder.CreateBitCast(Arg0,
719 PointerType::getUnqual(Arg1->getType()),
720 "cast");
Craig Topper29ce55d2016-05-30 22:54:12 +0000721 StoreInst *SI = Builder.CreateAlignedStore(Arg1, BC, 32);
Craig Topper7daf8972012-05-08 06:58:15 +0000722 SI->setMetadata(M->getMDKindID("nontemporal"), Node);
Craig Topper7daf8972012-05-08 06:58:15 +0000723
724 // Remove intrinsic.
725 CI->eraseFromParent();
726 return;
Craig Topper12e322a2016-05-25 06:56:32 +0000727 } else if (Name == "llvm.x86.sse2.storel.dq") {
Craig Topper12e322a2016-05-25 06:56:32 +0000728 Value *Arg0 = CI->getArgOperand(0);
729 Value *Arg1 = CI->getArgOperand(1);
730
731 Type *NewVecTy = VectorType::get(Type::getInt64Ty(C), 2);
732 Value *BC0 = Builder.CreateBitCast(Arg1, NewVecTy, "cast");
733 Value *Elt = Builder.CreateExtractElement(BC0, (uint64_t)0);
734 Value *BC = Builder.CreateBitCast(Arg0,
735 PointerType::getUnqual(Elt->getType()),
736 "cast");
Craig Topper29ce55d2016-05-30 22:54:12 +0000737 Builder.CreateAlignedStore(Elt, BC, 1);
Craig Topper12e322a2016-05-25 06:56:32 +0000738
739 // Remove intrinsic.
740 CI->eraseFromParent();
741 return;
Craig Topper8287fd82016-05-30 23:15:56 +0000742 } else if (Name.startswith("llvm.x86.sse.storeu.") ||
743 Name.startswith("llvm.x86.sse2.storeu.") ||
744 Name.startswith("llvm.x86.avx.storeu.")) {
745 Value *Arg0 = CI->getArgOperand(0);
746 Value *Arg1 = CI->getArgOperand(1);
747
748 Arg0 = Builder.CreateBitCast(Arg0,
749 PointerType::getUnqual(Arg1->getType()),
750 "cast");
751 Builder.CreateAlignedStore(Arg1, Arg0, 1);
752
753 // Remove intrinsic.
754 CI->eraseFromParent();
755 return;
Craig Topper50f85c22016-05-31 01:50:02 +0000756 } else if (Name.startswith("llvm.x86.avx512.mask.storeu.p") ||
757 Name.startswith("llvm.x86.avx512.mask.storeu.b.") ||
758 Name.startswith("llvm.x86.avx512.mask.storeu.w.") ||
759 Name.startswith("llvm.x86.avx512.mask.storeu.d.") ||
760 Name.startswith("llvm.x86.avx512.mask.storeu.q.")) {
761 UpgradeMaskedStore(Builder, C, CI->getArgOperand(0), CI->getArgOperand(1),
762 CI->getArgOperand(2), /*Aligned*/false);
763
764 // Remove intrinsic.
765 CI->eraseFromParent();
766 return;
767 } else if (Name.startswith("llvm.x86.avx512.mask.store.p") ||
768 Name.startswith("llvm.x86.avx512.mask.store.b.") ||
769 Name.startswith("llvm.x86.avx512.mask.store.w.") ||
770 Name.startswith("llvm.x86.avx512.mask.store.d.") ||
771 Name.startswith("llvm.x86.avx512.mask.store.q.")) {
772 UpgradeMaskedStore(Builder, C, CI->getArgOperand(0), CI->getArgOperand(1),
773 CI->getArgOperand(2), /*Aligned*/true);
774
775 // Remove intrinsic.
776 CI->eraseFromParent();
777 return;
Craig Topperf10fbfa2016-06-02 04:19:36 +0000778 } else if (Name.startswith("llvm.x86.avx512.mask.loadu.p") ||
779 Name.startswith("llvm.x86.avx512.mask.loadu.b.") ||
780 Name.startswith("llvm.x86.avx512.mask.loadu.w.") ||
781 Name.startswith("llvm.x86.avx512.mask.loadu.d.") ||
782 Name.startswith("llvm.x86.avx512.mask.loadu.q.")) {
783 Rep = UpgradeMaskedLoad(Builder, C, CI->getArgOperand(0),
784 CI->getArgOperand(1), CI->getArgOperand(2),
785 /*Aligned*/false);
786 } else if (Name.startswith("llvm.x86.avx512.mask.load.p") ||
787 Name.startswith("llvm.x86.avx512.mask.load.b.") ||
788 Name.startswith("llvm.x86.avx512.mask.load.w.") ||
789 Name.startswith("llvm.x86.avx512.mask.load.d.") ||
790 Name.startswith("llvm.x86.avx512.mask.load.q.")) {
791 Rep = UpgradeMaskedLoad(Builder, C, CI->getArgOperand(0),
792 CI->getArgOperand(1),CI->getArgOperand(2),
793 /*Aligned*/true);
Craig Topper3352ba52012-06-09 16:46:13 +0000794 } else if (Name.startswith("llvm.x86.xop.vpcom")) {
795 Intrinsic::ID intID;
796 if (Name.endswith("ub"))
797 intID = Intrinsic::x86_xop_vpcomub;
798 else if (Name.endswith("uw"))
799 intID = Intrinsic::x86_xop_vpcomuw;
800 else if (Name.endswith("ud"))
801 intID = Intrinsic::x86_xop_vpcomud;
802 else if (Name.endswith("uq"))
803 intID = Intrinsic::x86_xop_vpcomuq;
804 else if (Name.endswith("b"))
805 intID = Intrinsic::x86_xop_vpcomb;
806 else if (Name.endswith("w"))
807 intID = Intrinsic::x86_xop_vpcomw;
808 else if (Name.endswith("d"))
809 intID = Intrinsic::x86_xop_vpcomd;
810 else if (Name.endswith("q"))
811 intID = Intrinsic::x86_xop_vpcomq;
812 else
813 llvm_unreachable("Unknown suffix");
814
815 Name = Name.substr(18); // strip off "llvm.x86.xop.vpcom"
816 unsigned Imm;
817 if (Name.startswith("lt"))
818 Imm = 0;
819 else if (Name.startswith("le"))
820 Imm = 1;
821 else if (Name.startswith("gt"))
822 Imm = 2;
823 else if (Name.startswith("ge"))
824 Imm = 3;
825 else if (Name.startswith("eq"))
826 Imm = 4;
827 else if (Name.startswith("ne"))
828 Imm = 5;
Craig Topper3352ba52012-06-09 16:46:13 +0000829 else if (Name.startswith("false"))
Craig Toppere32546d2015-02-13 07:42:15 +0000830 Imm = 6;
831 else if (Name.startswith("true"))
Craig Topper3352ba52012-06-09 16:46:13 +0000832 Imm = 7;
833 else
834 llvm_unreachable("Unknown condition");
835
836 Function *VPCOM = Intrinsic::getDeclaration(F->getParent(), intID);
David Blaikieff6409d2015-05-18 22:13:54 +0000837 Rep =
838 Builder.CreateCall(VPCOM, {CI->getArgOperand(0), CI->getArgOperand(1),
839 Builder.getInt8(Imm)});
Simon Pilgrime88dc042015-11-03 20:27:01 +0000840 } else if (Name == "llvm.x86.xop.vpcmov") {
841 Value *Arg0 = CI->getArgOperand(0);
842 Value *Arg1 = CI->getArgOperand(1);
843 Value *Sel = CI->getArgOperand(2);
844 unsigned NumElts = CI->getType()->getVectorNumElements();
845 Constant *MinusOne = ConstantVector::getSplat(NumElts, Builder.getInt64(-1));
846 Value *NotSel = Builder.CreateXor(Sel, MinusOne);
847 Value *Sel0 = Builder.CreateAnd(Arg0, Sel);
848 Value *Sel1 = Builder.CreateAnd(Arg1, NotSel);
849 Rep = Builder.CreateOr(Sel0, Sel1);
Craig Topperef9e9932013-10-15 05:20:47 +0000850 } else if (Name == "llvm.x86.sse42.crc32.64.8") {
851 Function *CRC32 = Intrinsic::getDeclaration(F->getParent(),
852 Intrinsic::x86_sse42_crc32_32_8);
853 Value *Trunc0 = Builder.CreateTrunc(CI->getArgOperand(0), Type::getInt32Ty(C));
David Blaikieff6409d2015-05-18 22:13:54 +0000854 Rep = Builder.CreateCall(CRC32, {Trunc0, CI->getArgOperand(1)});
Craig Topperef9e9932013-10-15 05:20:47 +0000855 Rep = Builder.CreateZExt(Rep, CI->getType(), "");
Adam Nemet39066802014-05-29 23:35:33 +0000856 } else if (Name.startswith("llvm.x86.avx.vbroadcast")) {
857 // Replace broadcasts with a series of insertelements.
858 Type *VecTy = CI->getType();
859 Type *EltTy = VecTy->getVectorElementType();
860 unsigned EltNum = VecTy->getVectorNumElements();
861 Value *Cast = Builder.CreateBitCast(CI->getArgOperand(0),
862 EltTy->getPointerTo());
David Blaikie0c28fd72015-05-20 21:46:30 +0000863 Value *Load = Builder.CreateLoad(EltTy, Cast);
Adam Nemet39066802014-05-29 23:35:33 +0000864 Type *I32Ty = Type::getInt32Ty(C);
865 Rep = UndefValue::get(VecTy);
866 for (unsigned I = 0; I < EltNum; ++I)
867 Rep = Builder.CreateInsertElement(Rep, Load,
868 ConstantInt::get(I32Ty, I));
Simon Pilgrim9602d672016-05-28 18:03:41 +0000869 } else if (Name.startswith("llvm.x86.sse41.pmovsx") ||
870 Name.startswith("llvm.x86.sse41.pmovzx") ||
871 Name.startswith("llvm.x86.avx2.pmovsx") ||
872 Name.startswith("llvm.x86.avx2.pmovzx")) {
Simon Pilgrim9cb018b2015-09-23 08:48:33 +0000873 VectorType *SrcTy = cast<VectorType>(CI->getArgOperand(0)->getType());
874 VectorType *DstTy = cast<VectorType>(CI->getType());
875 unsigned NumDstElts = DstTy->getNumElements();
876
Simon Pilgrim9602d672016-05-28 18:03:41 +0000877 // Extract a subvector of the first NumDstElts lanes and sign/zero extend.
Craig Topperc0a5fa02016-06-12 04:48:00 +0000878 SmallVector<uint32_t, 8> ShuffleMask(NumDstElts);
Craig Topper99d1eab2016-06-12 00:41:19 +0000879 for (unsigned i = 0; i != NumDstElts; ++i)
Craig Topperc0a5fa02016-06-12 04:48:00 +0000880 ShuffleMask[i] = i;
Simon Pilgrim9cb018b2015-09-23 08:48:33 +0000881
882 Value *SV = Builder.CreateShuffleVector(
883 CI->getArgOperand(0), UndefValue::get(SrcTy), ShuffleMask);
Simon Pilgrim9602d672016-05-28 18:03:41 +0000884
885 bool DoSext = (StringRef::npos != Name.find("pmovsx"));
886 Rep = DoSext ? Builder.CreateSExt(SV, DstTy)
887 : Builder.CreateZExt(SV, DstTy);
Juergen Ributzka1f7a1762015-03-04 00:13:25 +0000888 } else if (Name == "llvm.x86.avx2.vbroadcasti128") {
889 // Replace vbroadcasts with a vector shuffle.
David Blaikie0c28fd72015-05-20 21:46:30 +0000890 Type *VT = VectorType::get(Type::getInt64Ty(C), 2);
891 Value *Op = Builder.CreatePointerCast(CI->getArgOperand(0),
892 PointerType::getUnqual(VT));
893 Value *Load = Builder.CreateLoad(VT, Op);
Craig Topper99d1eab2016-06-12 00:41:19 +0000894 uint32_t Idxs[4] = { 0, 1, 0, 1 };
Juergen Ributzka1f7a1762015-03-04 00:13:25 +0000895 Rep = Builder.CreateShuffleVector(Load, UndefValue::get(Load->getType()),
Sanjay Patel2db6d382015-03-12 15:27:07 +0000896 Idxs);
Ahmed Bougacha1a4987052015-08-20 20:36:19 +0000897 } else if (Name.startswith("llvm.x86.avx2.pbroadcast") ||
898 Name.startswith("llvm.x86.avx2.vbroadcast")) {
899 // Replace vp?broadcasts with a vector shuffle.
900 Value *Op = CI->getArgOperand(0);
901 unsigned NumElts = CI->getType()->getVectorNumElements();
902 Type *MaskTy = VectorType::get(Type::getInt32Ty(C), NumElts);
903 Rep = Builder.CreateShuffleVector(Op, UndefValue::get(Op->getType()),
904 Constant::getNullValue(MaskTy));
Craig Topper33350cc2016-06-06 06:12:54 +0000905 } else if (Name.startswith("llvm.x86.avx512.mask.palignr.")) {
906 Rep = UpgradeX86PALIGNRIntrinsics(Builder, C, CI->getArgOperand(0),
907 CI->getArgOperand(1),
908 CI->getArgOperand(2),
909 CI->getArgOperand(3),
910 CI->getArgOperand(4));
Craig Topper7355ac32016-05-29 06:37:33 +0000911 } else if (Name == "llvm.x86.sse2.psll.dq" ||
912 Name == "llvm.x86.avx2.psll.dq") {
913 // 128/256-bit shift left specified in bits.
Craig Topperb324e432015-02-18 06:24:44 +0000914 unsigned Shift = cast<ConstantInt>(CI->getArgOperand(1))->getZExtValue();
Craig Topper7355ac32016-05-29 06:37:33 +0000915 Rep = UpgradeX86PSLLDQIntrinsics(Builder, C, CI->getArgOperand(0),
Craig Topperb324e432015-02-18 06:24:44 +0000916 Shift / 8); // Shift is in bits.
Craig Topper7355ac32016-05-29 06:37:33 +0000917 } else if (Name == "llvm.x86.sse2.psrl.dq" ||
918 Name == "llvm.x86.avx2.psrl.dq") {
919 // 128/256-bit shift right specified in bits.
Craig Topperb324e432015-02-18 06:24:44 +0000920 unsigned Shift = cast<ConstantInt>(CI->getArgOperand(1))->getZExtValue();
Craig Topper7355ac32016-05-29 06:37:33 +0000921 Rep = UpgradeX86PSRLDQIntrinsics(Builder, C, CI->getArgOperand(0),
Craig Topperb324e432015-02-18 06:24:44 +0000922 Shift / 8); // Shift is in bits.
Craig Topper7355ac32016-05-29 06:37:33 +0000923 } else if (Name == "llvm.x86.sse2.psll.dq.bs" ||
Simon Pilgrimf7186822016-06-09 21:09:03 +0000924 Name == "llvm.x86.avx2.psll.dq.bs" ||
925 Name == "llvm.x86.avx512.psll.dq.512") {
926 // 128/256/512-bit shift left specified in bytes.
Craig Topperb324e432015-02-18 06:24:44 +0000927 unsigned Shift = cast<ConstantInt>(CI->getArgOperand(1))->getZExtValue();
Craig Topper7355ac32016-05-29 06:37:33 +0000928 Rep = UpgradeX86PSLLDQIntrinsics(Builder, C, CI->getArgOperand(0), Shift);
929 } else if (Name == "llvm.x86.sse2.psrl.dq.bs" ||
Simon Pilgrimf7186822016-06-09 21:09:03 +0000930 Name == "llvm.x86.avx2.psrl.dq.bs" ||
931 Name == "llvm.x86.avx512.psrl.dq.512") {
932 // 128/256/512-bit shift right specified in bytes.
Craig Topperb324e432015-02-18 06:24:44 +0000933 unsigned Shift = cast<ConstantInt>(CI->getArgOperand(1))->getZExtValue();
Craig Topper7355ac32016-05-29 06:37:33 +0000934 Rep = UpgradeX86PSRLDQIntrinsics(Builder, C, CI->getArgOperand(0), Shift);
Craig Topper782d6202015-02-28 19:33:17 +0000935 } else if (Name == "llvm.x86.sse41.pblendw" ||
Craig Topperf9f1ecc2016-05-30 22:54:05 +0000936 Name.startswith("llvm.x86.sse41.blendp") ||
937 Name.startswith("llvm.x86.avx.blend.p") ||
Craig Topper782d6202015-02-28 19:33:17 +0000938 Name == "llvm.x86.avx2.pblendw" ||
Craig Topperf9f1ecc2016-05-30 22:54:05 +0000939 Name.startswith("llvm.x86.avx2.pblendd.")) {
Craig Topper782d6202015-02-28 19:33:17 +0000940 Value *Op0 = CI->getArgOperand(0);
941 Value *Op1 = CI->getArgOperand(1);
942 unsigned Imm = cast <ConstantInt>(CI->getArgOperand(2))->getZExtValue();
943 VectorType *VecTy = cast<VectorType>(CI->getType());
944 unsigned NumElts = VecTy->getNumElements();
945
Craig Topperc0a5fa02016-06-12 04:48:00 +0000946 SmallVector<uint32_t, 16> Idxs(NumElts);
947 for (unsigned i = 0; i != NumElts; ++i)
948 Idxs[i] = ((Imm >> (i%8)) & 1) ? i + NumElts : i;
Craig Topper782d6202015-02-28 19:33:17 +0000949
Craig Topper2f561822016-06-12 01:05:59 +0000950 Rep = Builder.CreateShuffleVector(Op0, Op1, Idxs);
Craig Topperf9f1ecc2016-05-30 22:54:05 +0000951 } else if (Name.startswith("llvm.x86.avx.vinsertf128.") ||
Sanjay Patel4339abe2015-03-12 23:16:18 +0000952 Name == "llvm.x86.avx2.vinserti128") {
Sanjay Patel19792fb2015-03-10 16:08:36 +0000953 Value *Op0 = CI->getArgOperand(0);
954 Value *Op1 = CI->getArgOperand(1);
955 unsigned Imm = cast<ConstantInt>(CI->getArgOperand(2))->getZExtValue();
956 VectorType *VecTy = cast<VectorType>(CI->getType());
957 unsigned NumElts = VecTy->getNumElements();
Simon Pilgrim9cb018b2015-09-23 08:48:33 +0000958
Sanjay Patel19792fb2015-03-10 16:08:36 +0000959 // Mask off the high bits of the immediate value; hardware ignores those.
960 Imm = Imm & 1;
Simon Pilgrim9cb018b2015-09-23 08:48:33 +0000961
Sanjay Patel19792fb2015-03-10 16:08:36 +0000962 // Extend the second operand into a vector that is twice as big.
963 Value *UndefV = UndefValue::get(Op1->getType());
Craig Topperc0a5fa02016-06-12 04:48:00 +0000964 SmallVector<uint32_t, 8> Idxs(NumElts);
965 for (unsigned i = 0; i != NumElts; ++i)
966 Idxs[i] = i;
Craig Topper2f561822016-06-12 01:05:59 +0000967 Rep = Builder.CreateShuffleVector(Op1, UndefV, Idxs);
Sanjay Patel19792fb2015-03-10 16:08:36 +0000968
969 // Insert the second operand into the first operand.
970
971 // Note that there is no guarantee that instruction lowering will actually
972 // produce a vinsertf128 instruction for the created shuffles. In
973 // particular, the 0 immediate case involves no lane changes, so it can
974 // be handled as a blend.
975
976 // Example of shuffle mask for 32-bit elements:
977 // Imm = 1 <i32 0, i32 1, i32 2, i32 3, i32 8, i32 9, i32 10, i32 11>
978 // Imm = 0 <i32 8, i32 9, i32 10, i32 11, i32 4, i32 5, i32 6, i32 7 >
979
Sanjay Patel19792fb2015-03-10 16:08:36 +0000980 // The low half of the result is either the low half of the 1st operand
981 // or the low half of the 2nd operand (the inserted vector).
Craig Topperc0a5fa02016-06-12 04:48:00 +0000982 for (unsigned i = 0; i != NumElts / 2; ++i)
983 Idxs[i] = Imm ? i : (i + NumElts);
Sanjay Patel19792fb2015-03-10 16:08:36 +0000984 // The high half of the result is either the low half of the 2nd operand
985 // (the inserted vector) or the high half of the 1st operand.
Craig Topperc0a5fa02016-06-12 04:48:00 +0000986 for (unsigned i = NumElts / 2; i != NumElts; ++i)
987 Idxs[i] = Imm ? (i + NumElts / 2) : i;
Craig Topper2f561822016-06-12 01:05:59 +0000988 Rep = Builder.CreateShuffleVector(Op0, Rep, Idxs);
Craig Topperf9f1ecc2016-05-30 22:54:05 +0000989 } else if (Name.startswith("llvm.x86.avx.vextractf128.") ||
Sanjay Patel4339abe2015-03-12 23:16:18 +0000990 Name == "llvm.x86.avx2.vextracti128") {
Sanjay Patelaf1846c2015-03-12 15:15:19 +0000991 Value *Op0 = CI->getArgOperand(0);
992 unsigned Imm = cast<ConstantInt>(CI->getArgOperand(1))->getZExtValue();
993 VectorType *VecTy = cast<VectorType>(CI->getType());
994 unsigned NumElts = VecTy->getNumElements();
Simon Pilgrim9cb018b2015-09-23 08:48:33 +0000995
Sanjay Patelaf1846c2015-03-12 15:15:19 +0000996 // Mask off the high bits of the immediate value; hardware ignores those.
997 Imm = Imm & 1;
998
999 // Get indexes for either the high half or low half of the input vector.
Craig Topper2f561822016-06-12 01:05:59 +00001000 SmallVector<uint32_t, 4> Idxs(NumElts);
Sanjay Patelaf1846c2015-03-12 15:15:19 +00001001 for (unsigned i = 0; i != NumElts; ++i) {
Craig Topper2f561822016-06-12 01:05:59 +00001002 Idxs[i] = Imm ? (i + NumElts) : i;
Sanjay Patelaf1846c2015-03-12 15:15:19 +00001003 }
1004
1005 Value *UndefV = UndefValue::get(Op0->getType());
Craig Topper2f561822016-06-12 01:05:59 +00001006 Rep = Builder.CreateShuffleVector(Op0, UndefV, Idxs);
Tim Shen00127562016-04-08 21:26:31 +00001007 } else if (Name == "llvm.stackprotectorcheck") {
1008 Rep = nullptr;
Craig Topper10679862016-06-12 14:11:32 +00001009 } else if (Name.startswith("llvm.x86.avx.vpermil.") ||
Craig Topper13cf7ca2016-06-13 02:36:48 +00001010 Name == "llvm.x86.sse2.pshuf.d" ||
Simon Pilgrim9fca3002016-07-04 12:40:54 +00001011 Name.startswith("llvm.x86.avx512.mask.vpermil.p") ||
Craig Topper13cf7ca2016-06-13 02:36:48 +00001012 Name.startswith("llvm.x86.avx512.mask.pshuf.d.")) {
Craig Topper8a105052016-06-12 03:10:47 +00001013 Value *Op0 = CI->getArgOperand(0);
1014 unsigned Imm = cast<ConstantInt>(CI->getArgOperand(1))->getZExtValue();
1015 VectorType *VecTy = cast<VectorType>(CI->getType());
1016 unsigned NumElts = VecTy->getNumElements();
Simon Pilgrim9fca3002016-07-04 12:40:54 +00001017 // Calculate the size of each index in the immediate.
Craig Topper8a105052016-06-12 03:10:47 +00001018 unsigned IdxSize = 64 / VecTy->getScalarSizeInBits();
1019 unsigned IdxMask = ((1 << IdxSize) - 1);
1020
1021 SmallVector<uint32_t, 8> Idxs(NumElts);
1022 // Lookup the bits for this element, wrapping around the immediate every
1023 // 8-bits. Elements are grouped into sets of 2 or 4 elements so we need
1024 // to offset by the first index of each group.
1025 for (unsigned i = 0; i != NumElts; ++i)
1026 Idxs[i] = ((Imm >> ((i * IdxSize) % 8)) & IdxMask) | (i & ~IdxMask);
1027
1028 Rep = Builder.CreateShuffleVector(Op0, Op0, Idxs);
Craig Topper13cf7ca2016-06-13 02:36:48 +00001029
1030 if (CI->getNumArgOperands() == 4)
1031 Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep,
1032 CI->getArgOperand(2));
1033 } else if (Name == "llvm.x86.sse2.pshufl.w" ||
1034 Name.startswith("llvm.x86.avx512.mask.pshufl.w.")) {
Craig Topper10679862016-06-12 14:11:32 +00001035 Value *Op0 = CI->getArgOperand(0);
1036 unsigned Imm = cast<ConstantInt>(CI->getArgOperand(1))->getZExtValue();
1037 unsigned NumElts = CI->getType()->getVectorNumElements();
1038
1039 SmallVector<uint32_t, 16> Idxs(NumElts);
1040 for (unsigned l = 0; l != NumElts; l += 8) {
1041 for (unsigned i = 0; i != 4; ++i)
1042 Idxs[i + l] = ((Imm >> (2 * i)) & 0x3) + l;
1043 for (unsigned i = 4; i != 8; ++i)
1044 Idxs[i + l] = i + l;
1045 }
1046
1047 Rep = Builder.CreateShuffleVector(Op0, Op0, Idxs);
Craig Topper13cf7ca2016-06-13 02:36:48 +00001048
1049 if (CI->getNumArgOperands() == 4)
1050 Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep,
1051 CI->getArgOperand(2));
1052 } else if (Name == "llvm.x86.sse2.pshufh.w" ||
1053 Name.startswith("llvm.x86.avx512.mask.pshufh.w.")) {
Craig Topper10679862016-06-12 14:11:32 +00001054 Value *Op0 = CI->getArgOperand(0);
1055 unsigned Imm = cast<ConstantInt>(CI->getArgOperand(1))->getZExtValue();
1056 unsigned NumElts = CI->getType()->getVectorNumElements();
1057
1058 SmallVector<uint32_t, 16> Idxs(NumElts);
1059 for (unsigned l = 0; l != NumElts; l += 8) {
1060 for (unsigned i = 0; i != 4; ++i)
1061 Idxs[i + l] = i + l;
1062 for (unsigned i = 0; i != 4; ++i)
1063 Idxs[i + l + 4] = ((Imm >> (2 * i)) & 0x3) + 4 + l;
1064 }
1065
1066 Rep = Builder.CreateShuffleVector(Op0, Op0, Idxs);
Craig Topper13cf7ca2016-06-13 02:36:48 +00001067
1068 if (CI->getNumArgOperands() == 4)
1069 Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep,
1070 CI->getArgOperand(2));
Simon Pilgrim19adee92016-07-02 14:42:35 +00001071 } else if (Name.startswith("llvm.x86.avx512.mask.movddup") ||
1072 Name.startswith("llvm.x86.avx512.mask.movshdup") ||
1073 Name.startswith("llvm.x86.avx512.mask.movsldup")) {
1074 Value *Op0 = CI->getArgOperand(0);
1075 unsigned NumElts = CI->getType()->getVectorNumElements();
1076 unsigned NumLaneElts = 128/CI->getType()->getScalarSizeInBits();
1077
1078 unsigned Offset = 0;
1079 if (Name.startswith("llvm.x86.avx512.mask.movshdup."))
1080 Offset = 1;
1081
1082 SmallVector<uint32_t, 16> Idxs(NumElts);
1083 for (unsigned l = 0; l != NumElts; l += NumLaneElts)
1084 for (unsigned i = 0; i != NumLaneElts; i += 2) {
1085 Idxs[i + l + 0] = i + l + Offset;
1086 Idxs[i + l + 1] = i + l + Offset;
1087 }
1088
1089 Rep = Builder.CreateShuffleVector(Op0, Op0, Idxs);
1090
1091 Rep = EmitX86Select(Builder, CI->getArgOperand(2), Rep,
1092 CI->getArgOperand(1));
Craig Topper597aa422016-06-23 07:37:33 +00001093 } else if (Name.startswith("llvm.x86.avx512.mask.punpckl") ||
1094 Name.startswith("llvm.x86.avx512.mask.unpckl.")) {
1095 Value *Op0 = CI->getArgOperand(0);
1096 Value *Op1 = CI->getArgOperand(1);
1097 int NumElts = CI->getType()->getVectorNumElements();
1098 int NumLaneElts = 128/CI->getType()->getScalarSizeInBits();
1099
1100 SmallVector<uint32_t, 64> Idxs(NumElts);
1101 for (int l = 0; l != NumElts; l += NumLaneElts)
1102 for (int i = 0; i != NumLaneElts; ++i)
1103 Idxs[i + l] = l + (i / 2) + NumElts * (i % 2);
1104
1105 Rep = Builder.CreateShuffleVector(Op0, Op1, Idxs);
1106
1107 Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep,
1108 CI->getArgOperand(2));
1109 } else if (Name.startswith("llvm.x86.avx512.mask.punpckh") ||
1110 Name.startswith("llvm.x86.avx512.mask.unpckh.")) {
1111 Value *Op0 = CI->getArgOperand(0);
1112 Value *Op1 = CI->getArgOperand(1);
1113 int NumElts = CI->getType()->getVectorNumElements();
1114 int NumLaneElts = 128/CI->getType()->getScalarSizeInBits();
1115
1116 SmallVector<uint32_t, 64> Idxs(NumElts);
1117 for (int l = 0; l != NumElts; l += NumLaneElts)
1118 for (int i = 0; i != NumLaneElts; ++i)
1119 Idxs[i + l] = (NumLaneElts / 2) + l + (i / 2) + NumElts * (i % 2);
1120
1121 Rep = Builder.CreateShuffleVector(Op0, Op1, Idxs);
1122
1123 Rep = EmitX86Select(Builder, CI->getArgOperand(3), Rep,
1124 CI->getArgOperand(2));
Craig Topper3b1817d2012-02-03 06:10:55 +00001125 } else {
Craig Topper8a105052016-06-12 03:10:47 +00001126 llvm_unreachable("Unknown function for CallInst upgrade.");
Craig Topper3b1817d2012-02-03 06:10:55 +00001127 }
1128
Tim Shen00127562016-04-08 21:26:31 +00001129 if (Rep)
1130 CI->replaceAllUsesWith(Rep);
Craig Topper3b1817d2012-02-03 06:10:55 +00001131 CI->eraseFromParent();
1132 return;
1133 }
1134
Yaron Kerend1fdbe72015-03-30 16:10:39 +00001135 std::string Name = CI->getName();
Adrian Prantl87b7eb92014-10-01 18:55:02 +00001136 if (!Name.empty())
1137 CI->setName(Name + ".old");
Nadav Rotem17ee58a2012-06-10 18:42:51 +00001138
Chandler Carruth58a71ed2011-12-12 04:26:04 +00001139 switch (NewFn->getIntrinsicID()) {
1140 default:
Chris Lattner0bcbde42011-11-27 08:42:07 +00001141 llvm_unreachable("Unknown function for CallInst upgrade.");
Chandler Carruth58a71ed2011-12-12 04:26:04 +00001142
Jeroen Ketemaab99b592015-09-30 10:56:37 +00001143 case Intrinsic::arm_neon_vld1:
1144 case Intrinsic::arm_neon_vld2:
1145 case Intrinsic::arm_neon_vld3:
1146 case Intrinsic::arm_neon_vld4:
1147 case Intrinsic::arm_neon_vld2lane:
1148 case Intrinsic::arm_neon_vld3lane:
1149 case Intrinsic::arm_neon_vld4lane:
1150 case Intrinsic::arm_neon_vst1:
1151 case Intrinsic::arm_neon_vst2:
1152 case Intrinsic::arm_neon_vst3:
1153 case Intrinsic::arm_neon_vst4:
1154 case Intrinsic::arm_neon_vst2lane:
1155 case Intrinsic::arm_neon_vst3lane:
1156 case Intrinsic::arm_neon_vst4lane: {
1157 SmallVector<Value *, 4> Args(CI->arg_operands().begin(),
1158 CI->arg_operands().end());
1159 CI->replaceAllUsesWith(Builder.CreateCall(NewFn, Args));
1160 CI->eraseFromParent();
1161 return;
1162 }
1163
Chandler Carruth58a71ed2011-12-12 04:26:04 +00001164 case Intrinsic::ctlz:
Nuno Lopesad40c0a2012-05-22 15:25:31 +00001165 case Intrinsic::cttz:
Chandler Carruth58a71ed2011-12-12 04:26:04 +00001166 assert(CI->getNumArgOperands() == 1 &&
1167 "Mismatch between function args and call args");
David Blaikieff6409d2015-05-18 22:13:54 +00001168 CI->replaceAllUsesWith(Builder.CreateCall(
1169 NewFn, {CI->getArgOperand(0), Builder.getFalse()}, Name));
Chandler Carruth58a71ed2011-12-12 04:26:04 +00001170 CI->eraseFromParent();
1171 return;
Nadav Rotem17ee58a2012-06-10 18:42:51 +00001172
Matt Arsenaultfbcbce42013-10-07 18:06:48 +00001173 case Intrinsic::objectsize:
David Blaikieff6409d2015-05-18 22:13:54 +00001174 CI->replaceAllUsesWith(Builder.CreateCall(
1175 NewFn, {CI->getArgOperand(0), CI->getArgOperand(1)}, Name));
Matt Arsenaultfbcbce42013-10-07 18:06:48 +00001176 CI->eraseFromParent();
1177 return;
1178
Joel Jonesb84f7be2012-07-18 00:02:16 +00001179 case Intrinsic::ctpop: {
David Blaikieff6409d2015-05-18 22:13:54 +00001180 CI->replaceAllUsesWith(Builder.CreateCall(NewFn, {CI->getArgOperand(0)}));
Joel Jonesb84f7be2012-07-18 00:02:16 +00001181 CI->eraseFromParent();
1182 return;
1183 }
Joel Jones43cb8782012-07-13 23:25:25 +00001184
Craig Topper71dc02d2012-06-13 07:18:53 +00001185 case Intrinsic::x86_xop_vfrcz_ss:
1186 case Intrinsic::x86_xop_vfrcz_sd:
David Blaikieff6409d2015-05-18 22:13:54 +00001187 CI->replaceAllUsesWith(
1188 Builder.CreateCall(NewFn, {CI->getArgOperand(1)}, Name));
Craig Topper71dc02d2012-06-13 07:18:53 +00001189 CI->eraseFromParent();
1190 return;
1191
Simon Pilgrime85506b2016-06-03 08:06:03 +00001192 case Intrinsic::x86_xop_vpermil2pd:
1193 case Intrinsic::x86_xop_vpermil2ps:
1194 case Intrinsic::x86_xop_vpermil2pd_256:
1195 case Intrinsic::x86_xop_vpermil2ps_256: {
1196 SmallVector<Value *, 4> Args(CI->arg_operands().begin(),
1197 CI->arg_operands().end());
1198 VectorType *FltIdxTy = cast<VectorType>(Args[2]->getType());
1199 VectorType *IntIdxTy = VectorType::getInteger(FltIdxTy);
1200 Args[2] = Builder.CreateBitCast(Args[2], IntIdxTy);
1201 CI->replaceAllUsesWith(Builder.CreateCall(NewFn, Args, Name));
1202 CI->eraseFromParent();
1203 return;
1204 }
1205
Nadav Rotem17ee58a2012-06-10 18:42:51 +00001206 case Intrinsic::x86_sse41_ptestc:
1207 case Intrinsic::x86_sse41_ptestz:
Craig Topper71dc02d2012-06-13 07:18:53 +00001208 case Intrinsic::x86_sse41_ptestnzc: {
Nadav Rotem17ee58a2012-06-10 18:42:51 +00001209 // The arguments for these intrinsics used to be v4f32, and changed
1210 // to v2i64. This is purely a nop, since those are bitwise intrinsics.
1211 // So, the only thing required is a bitcast for both arguments.
1212 // First, check the arguments have the old type.
1213 Value *Arg0 = CI->getArgOperand(0);
1214 if (Arg0->getType() != VectorType::get(Type::getFloatTy(C), 4))
1215 return;
1216
1217 // Old intrinsic, add bitcasts
1218 Value *Arg1 = CI->getArgOperand(1);
1219
David Blaikie5bacf372015-04-24 21:16:07 +00001220 Type *NewVecTy = VectorType::get(Type::getInt64Ty(C), 2);
Nadav Rotem17ee58a2012-06-10 18:42:51 +00001221
David Blaikie5bacf372015-04-24 21:16:07 +00001222 Value *BC0 = Builder.CreateBitCast(Arg0, NewVecTy, "cast");
1223 Value *BC1 = Builder.CreateBitCast(Arg1, NewVecTy, "cast");
1224
David Blaikieff6409d2015-05-18 22:13:54 +00001225 CallInst *NewCall = Builder.CreateCall(NewFn, {BC0, BC1}, Name);
Nadav Rotem17ee58a2012-06-10 18:42:51 +00001226 CI->replaceAllUsesWith(NewCall);
1227 CI->eraseFromParent();
1228 return;
Evan Cheng0e179d02007-12-17 22:33:23 +00001229 }
Chandler Carruth373b2b12014-09-06 10:00:01 +00001230
Chandler Carruth373b2b12014-09-06 10:00:01 +00001231 case Intrinsic::x86_sse41_insertps:
1232 case Intrinsic::x86_sse41_dppd:
1233 case Intrinsic::x86_sse41_dpps:
1234 case Intrinsic::x86_sse41_mpsadbw:
Chandler Carruth373b2b12014-09-06 10:00:01 +00001235 case Intrinsic::x86_avx_dp_ps_256:
Chandler Carruth373b2b12014-09-06 10:00:01 +00001236 case Intrinsic::x86_avx2_mpsadbw: {
1237 // Need to truncate the last argument from i32 to i8 -- this argument models
1238 // an inherently 8-bit immediate operand to these x86 instructions.
1239 SmallVector<Value *, 4> Args(CI->arg_operands().begin(),
1240 CI->arg_operands().end());
1241
1242 // Replace the last argument with a trunc.
1243 Args.back() = Builder.CreateTrunc(Args.back(), Type::getInt8Ty(C), "trunc");
1244
1245 CallInst *NewCall = Builder.CreateCall(NewFn, Args);
1246 CI->replaceAllUsesWith(NewCall);
1247 CI->eraseFromParent();
1248 return;
1249 }
Marcin Koscielnicki3fdc2572016-04-19 20:51:05 +00001250
1251 case Intrinsic::thread_pointer: {
1252 CI->replaceAllUsesWith(Builder.CreateCall(NewFn, {}));
1253 CI->eraseFromParent();
1254 return;
1255 }
Artur Pilipenko7ad95ec2016-06-28 18:27:25 +00001256
1257 case Intrinsic::masked_load:
1258 case Intrinsic::masked_store: {
1259 SmallVector<Value *, 4> Args(CI->arg_operands().begin(),
1260 CI->arg_operands().end());
1261 CI->replaceAllUsesWith(Builder.CreateCall(NewFn, Args));
1262 CI->eraseFromParent();
1263 return;
1264 }
Craig Topper71dc02d2012-06-13 07:18:53 +00001265 }
Chandler Carruth7132e002007-08-04 01:51:18 +00001266}
1267
Sanjay Patelfdf0d5f2016-04-18 19:11:57 +00001268void llvm::UpgradeCallsToIntrinsic(Function *F) {
Chandler Carruth7132e002007-08-04 01:51:18 +00001269 assert(F && "Illegal attempt to upgrade a non-existent intrinsic.");
1270
Sanjay Patelfdf0d5f2016-04-18 19:11:57 +00001271 // Check if this function should be upgraded and get the replacement function
1272 // if there is one.
Chris Lattner80ed9dc2011-06-18 06:05:24 +00001273 Function *NewFn;
Evan Cheng0e179d02007-12-17 22:33:23 +00001274 if (UpgradeIntrinsicFunction(F, NewFn)) {
Sanjay Patelfdf0d5f2016-04-18 19:11:57 +00001275 // Replace all users of the old function with the new function or new
1276 // instructions. This is not a range loop because the call is deleted.
1277 for (auto UI = F->user_begin(), UE = F->user_end(); UI != UE; )
Duncan P. N. Exon Smith93f53c42016-04-17 03:59:37 +00001278 if (CallInst *CI = dyn_cast<CallInst>(*UI++))
Filipe Cabecinhas0011c582015-07-03 20:12:01 +00001279 UpgradeIntrinsicCall(CI, NewFn);
Sanjay Patelfdf0d5f2016-04-18 19:11:57 +00001280
Filipe Cabecinhas0011c582015-07-03 20:12:01 +00001281 // Remove old function, no longer used, from the module.
1282 F->eraseFromParent();
Chandler Carruth7132e002007-08-04 01:51:18 +00001283 }
1284}
Devang Patel80ae3492009-08-28 23:24:31 +00001285
Manman Ren209b17c2013-09-28 00:22:27 +00001286void llvm::UpgradeInstWithTBAATag(Instruction *I) {
Duncan P. N. Exon Smithde36e802014-11-11 21:30:22 +00001287 MDNode *MD = I->getMetadata(LLVMContext::MD_tbaa);
Manman Ren209b17c2013-09-28 00:22:27 +00001288 assert(MD && "UpgradeInstWithTBAATag should have a TBAA tag");
1289 // Check if the tag uses struct-path aware TBAA format.
1290 if (isa<MDNode>(MD->getOperand(0)) && MD->getNumOperands() >= 3)
1291 return;
1292
1293 if (MD->getNumOperands() == 3) {
Duncan P. N. Exon Smith5bf8fef2014-12-09 18:38:53 +00001294 Metadata *Elts[] = {MD->getOperand(0), MD->getOperand(1)};
Manman Ren209b17c2013-09-28 00:22:27 +00001295 MDNode *ScalarType = MDNode::get(I->getContext(), Elts);
1296 // Create a MDNode <ScalarType, ScalarType, offset 0, const>
Duncan P. N. Exon Smith5bf8fef2014-12-09 18:38:53 +00001297 Metadata *Elts2[] = {ScalarType, ScalarType,
1298 ConstantAsMetadata::get(Constant::getNullValue(
1299 Type::getInt64Ty(I->getContext()))),
1300 MD->getOperand(2)};
Manman Ren209b17c2013-09-28 00:22:27 +00001301 I->setMetadata(LLVMContext::MD_tbaa, MDNode::get(I->getContext(), Elts2));
1302 } else {
1303 // Create a MDNode <MD, MD, offset 0>
Duncan P. N. Exon Smith5bf8fef2014-12-09 18:38:53 +00001304 Metadata *Elts[] = {MD, MD, ConstantAsMetadata::get(Constant::getNullValue(
1305 Type::getInt64Ty(I->getContext())))};
Manman Ren209b17c2013-09-28 00:22:27 +00001306 I->setMetadata(LLVMContext::MD_tbaa, MDNode::get(I->getContext(), Elts));
1307 }
1308}
Matt Arsenaultb03bd4d2013-11-15 01:34:59 +00001309
1310Instruction *llvm::UpgradeBitCastInst(unsigned Opc, Value *V, Type *DestTy,
1311 Instruction *&Temp) {
1312 if (Opc != Instruction::BitCast)
Craig Topperc6207612014-04-09 06:08:46 +00001313 return nullptr;
Matt Arsenaultb03bd4d2013-11-15 01:34:59 +00001314
Craig Topperc6207612014-04-09 06:08:46 +00001315 Temp = nullptr;
Matt Arsenaultb03bd4d2013-11-15 01:34:59 +00001316 Type *SrcTy = V->getType();
1317 if (SrcTy->isPtrOrPtrVectorTy() && DestTy->isPtrOrPtrVectorTy() &&
1318 SrcTy->getPointerAddressSpace() != DestTy->getPointerAddressSpace()) {
1319 LLVMContext &Context = V->getContext();
1320
1321 // We have no information about target data layout, so we assume that
1322 // the maximum pointer size is 64bit.
1323 Type *MidTy = Type::getInt64Ty(Context);
1324 Temp = CastInst::Create(Instruction::PtrToInt, V, MidTy);
1325
1326 return CastInst::Create(Instruction::IntToPtr, Temp, DestTy);
1327 }
1328
Craig Topperc6207612014-04-09 06:08:46 +00001329 return nullptr;
Matt Arsenaultb03bd4d2013-11-15 01:34:59 +00001330}
1331
1332Value *llvm::UpgradeBitCastExpr(unsigned Opc, Constant *C, Type *DestTy) {
1333 if (Opc != Instruction::BitCast)
Craig Topperc6207612014-04-09 06:08:46 +00001334 return nullptr;
Matt Arsenaultb03bd4d2013-11-15 01:34:59 +00001335
1336 Type *SrcTy = C->getType();
1337 if (SrcTy->isPtrOrPtrVectorTy() && DestTy->isPtrOrPtrVectorTy() &&
1338 SrcTy->getPointerAddressSpace() != DestTy->getPointerAddressSpace()) {
1339 LLVMContext &Context = C->getContext();
1340
1341 // We have no information about target data layout, so we assume that
1342 // the maximum pointer size is 64bit.
1343 Type *MidTy = Type::getInt64Ty(Context);
1344
1345 return ConstantExpr::getIntToPtr(ConstantExpr::getPtrToInt(C, MidTy),
1346 DestTy);
1347 }
1348
Craig Topperc6207612014-04-09 06:08:46 +00001349 return nullptr;
Matt Arsenaultb03bd4d2013-11-15 01:34:59 +00001350}
Manman Ren8b4306c2013-12-02 21:29:56 +00001351
1352/// Check the debug info version number, if it is out-dated, drop the debug
1353/// info. Return true if module is modified.
1354bool llvm::UpgradeDebugInfo(Module &M) {
Manman Ren2ebfb422014-01-16 01:51:12 +00001355 unsigned Version = getDebugMetadataVersionFromModule(M);
1356 if (Version == DEBUG_METADATA_VERSION)
Manman Ren8b4306c2013-12-02 21:29:56 +00001357 return false;
1358
Manman Ren2ebfb422014-01-16 01:51:12 +00001359 bool RetCode = StripDebugInfo(M);
1360 if (RetCode) {
1361 DiagnosticInfoDebugMetadataVersion DiagVersion(M, Version);
1362 M.getContext().diagnose(DiagVersion);
1363 }
1364 return RetCode;
Manman Ren8b4306c2013-12-02 21:29:56 +00001365}
Eli Bendersky5d5e18d2014-06-25 15:41:00 +00001366
Manman Renb5d7ff42016-05-25 23:14:48 +00001367bool llvm::UpgradeModuleFlags(Module &M) {
1368 const NamedMDNode *ModFlags = M.getModuleFlagsMetadata();
1369 if (!ModFlags)
1370 return false;
1371
1372 bool HasObjCFlag = false, HasClassProperties = false;
1373 for (unsigned I = 0, E = ModFlags->getNumOperands(); I != E; ++I) {
1374 MDNode *Op = ModFlags->getOperand(I);
1375 if (Op->getNumOperands() < 2)
1376 continue;
1377 MDString *ID = dyn_cast_or_null<MDString>(Op->getOperand(1));
1378 if (!ID)
1379 continue;
1380 if (ID->getString() == "Objective-C Image Info Version")
1381 HasObjCFlag = true;
1382 if (ID->getString() == "Objective-C Class Properties")
1383 HasClassProperties = true;
1384 }
1385 // "Objective-C Class Properties" is recently added for Objective-C. We
1386 // upgrade ObjC bitcodes to contain a "Objective-C Class Properties" module
1387 // flag of value 0, so we can correclty report error when trying to link
1388 // an ObjC bitcode without this module flag with an ObjC bitcode with this
1389 // module flag.
1390 if (HasObjCFlag && !HasClassProperties) {
1391 M.addModuleFlag(llvm::Module::Error, "Objective-C Class Properties",
1392 (uint32_t)0);
1393 return true;
1394 }
1395 return false;
1396}
1397
Duncan P. N. Exon Smithefe16c82016-03-25 00:56:13 +00001398static bool isOldLoopArgument(Metadata *MD) {
1399 auto *T = dyn_cast_or_null<MDTuple>(MD);
1400 if (!T)
1401 return false;
1402 if (T->getNumOperands() < 1)
1403 return false;
1404 auto *S = dyn_cast_or_null<MDString>(T->getOperand(0));
1405 if (!S)
1406 return false;
1407 return S->getString().startswith("llvm.vectorizer.");
1408}
1409
1410static MDString *upgradeLoopTag(LLVMContext &C, StringRef OldTag) {
1411 StringRef OldPrefix = "llvm.vectorizer.";
1412 assert(OldTag.startswith(OldPrefix) && "Expected old prefix");
1413
1414 if (OldTag == "llvm.vectorizer.unroll")
1415 return MDString::get(C, "llvm.loop.interleave.count");
1416
1417 return MDString::get(
1418 C, (Twine("llvm.loop.vectorize.") + OldTag.drop_front(OldPrefix.size()))
1419 .str());
1420}
1421
1422static Metadata *upgradeLoopArgument(Metadata *MD) {
1423 auto *T = dyn_cast_or_null<MDTuple>(MD);
1424 if (!T)
1425 return MD;
1426 if (T->getNumOperands() < 1)
1427 return MD;
1428 auto *OldTag = dyn_cast_or_null<MDString>(T->getOperand(0));
1429 if (!OldTag)
1430 return MD;
1431 if (!OldTag->getString().startswith("llvm.vectorizer."))
1432 return MD;
1433
1434 // This has an old tag. Upgrade it.
1435 SmallVector<Metadata *, 8> Ops;
1436 Ops.reserve(T->getNumOperands());
1437 Ops.push_back(upgradeLoopTag(T->getContext(), OldTag->getString()));
1438 for (unsigned I = 1, E = T->getNumOperands(); I != E; ++I)
1439 Ops.push_back(T->getOperand(I));
1440
1441 return MDTuple::get(T->getContext(), Ops);
1442}
1443
1444MDNode *llvm::upgradeInstructionLoopAttachment(MDNode &N) {
1445 auto *T = dyn_cast<MDTuple>(&N);
1446 if (!T)
1447 return &N;
1448
1449 if (!llvm::any_of(T->operands(), isOldLoopArgument))
1450 return &N;
1451
1452 SmallVector<Metadata *, 8> Ops;
1453 Ops.reserve(T->getNumOperands());
1454 for (Metadata *MD : T->operands())
1455 Ops.push_back(upgradeLoopArgument(MD));
1456
1457 return MDTuple::get(T->getContext(), Ops);
Eli Bendersky5d5e18d2014-06-25 15:41:00 +00001458}