blob: 027c0255bcece6feb7de3be2eaa00556bea6e755 [file] [log] [blame]
Chris Lattner17079fc2009-12-28 21:28:46 +00001//===---- IRBuilder.cpp - Builder for LLVM Instrs -------------------------===//
2//
3// The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9//
10// This file implements the IRBuilder class, which is used as a convenient way
11// to create LLVM instructions with a consistent and simplified interface.
12//
13//===----------------------------------------------------------------------===//
14
Chandler Carruth6bda14b2017-06-06 11:49:48 +000015#include "llvm/IR/IRBuilder.h"
Chandler Carruth9fb823b2013-01-02 11:36:10 +000016#include "llvm/IR/Function.h"
17#include "llvm/IR/GlobalVariable.h"
Chandler Carruth9fb823b2013-01-02 11:36:10 +000018#include "llvm/IR/Intrinsics.h"
19#include "llvm/IR/LLVMContext.h"
Pat Gavlincc0431d2015-05-08 18:07:42 +000020#include "llvm/IR/Statepoint.h"
Chris Lattner17079fc2009-12-28 21:28:46 +000021using namespace llvm;
22
23/// CreateGlobalString - Make a new global variable with an initializer that
Dan Gohman97c59022010-02-10 20:04:19 +000024/// has array of i8 type filled in with the nul terminated string value
Chris Lattner17079fc2009-12-28 21:28:46 +000025/// specified. If Name is specified, it is the name of the global variable
26/// created.
David Blaikieaa41cd52015-04-03 21:33:42 +000027GlobalVariable *IRBuilderBase::CreateGlobalString(StringRef Str,
Tobias Grossercdb89142015-06-19 02:12:07 +000028 const Twine &Name,
29 unsigned AddressSpace) {
Chris Lattnercf9e8f62012-02-05 02:29:43 +000030 Constant *StrConstant = ConstantDataArray::getString(Context, Str);
Chris Lattner17079fc2009-12-28 21:28:46 +000031 Module &M = *BB->getParent()->getParent();
32 GlobalVariable *GV = new GlobalVariable(M, StrConstant->getType(),
Benjamin Kramerf1fd6e32011-12-22 14:22:14 +000033 true, GlobalValue::PrivateLinkage,
Tobias Grossercdb89142015-06-19 02:12:07 +000034 StrConstant, Name, nullptr,
35 GlobalVariable::NotThreadLocal,
36 AddressSpace);
Peter Collingbourne96efdd62016-06-14 21:01:22 +000037 GV->setUnnamedAddr(GlobalValue::UnnamedAddr::Global);
Chris Lattner17079fc2009-12-28 21:28:46 +000038 return GV;
39}
Chris Lattner7ef1cac2009-12-28 21:45:40 +000040
Chris Lattnerb1907b22011-07-12 04:14:22 +000041Type *IRBuilderBase::getCurrentFunctionReturnType() const {
Chris Lattner49f9f762009-12-28 21:50:56 +000042 assert(BB && BB->getParent() && "No current function!");
43 return BB->getParent()->getReturnType();
44}
Chris Lattner143a07c2010-12-26 22:49:25 +000045
46Value *IRBuilderBase::getCastedInt8PtrValue(Value *Ptr) {
Chris Lattner229907c2011-07-18 04:54:35 +000047 PointerType *PT = cast<PointerType>(Ptr->getType());
Chris Lattner143a07c2010-12-26 22:49:25 +000048 if (PT->getElementType()->isIntegerTy(8))
49 return Ptr;
50
51 // Otherwise, we need to insert a bitcast.
52 PT = getInt8PtrTy(PT->getAddressSpace());
53 BitCastInst *BCI = new BitCastInst(Ptr, PT, "");
54 BB->getInstList().insert(InsertPt, BCI);
55 SetInstDebugLocation(BCI);
56 return BCI;
57}
58
Jay Foad5bd375a2011-07-15 08:37:34 +000059static CallInst *createCallHelper(Value *Callee, ArrayRef<Value *> Ops,
Philip Reames4750dd12014-12-30 05:55:58 +000060 IRBuilderBase *Builder,
61 const Twine& Name="") {
62 CallInst *CI = CallInst::Create(Callee, Ops, Name);
Chris Lattner143a07c2010-12-26 22:49:25 +000063 Builder->GetInsertBlock()->getInstList().insert(Builder->GetInsertPoint(),CI);
64 Builder->SetInstDebugLocation(CI);
65 return CI;
66}
67
Sanjoy Dasabe1c682015-05-06 23:53:09 +000068static InvokeInst *createInvokeHelper(Value *Invokee, BasicBlock *NormalDest,
69 BasicBlock *UnwindDest,
70 ArrayRef<Value *> Ops,
71 IRBuilderBase *Builder,
72 const Twine &Name = "") {
73 InvokeInst *II =
74 InvokeInst::Create(Invokee, NormalDest, UnwindDest, Ops, Name);
75 Builder->GetInsertBlock()->getInstList().insert(Builder->GetInsertPoint(),
76 II);
77 Builder->SetInstDebugLocation(II);
78 return II;
79}
80
Chris Lattner143a07c2010-12-26 22:49:25 +000081CallInst *IRBuilderBase::
Pete Cooper67cf9a72015-11-19 05:56:52 +000082CreateMemSet(Value *Ptr, Value *Val, Value *Size, unsigned Align,
Hal Finkel94146652014-07-24 14:25:39 +000083 bool isVolatile, MDNode *TBAATag, MDNode *ScopeTag,
84 MDNode *NoAliasTag) {
Chris Lattner143a07c2010-12-26 22:49:25 +000085 Ptr = getCastedInt8PtrValue(Ptr);
Pete Cooper67cf9a72015-11-19 05:56:52 +000086 Value *Ops[] = { Ptr, Val, Size, getInt32(Align), getInt1(isVolatile) };
Jay Foadb804a2b2011-07-12 14:06:48 +000087 Type *Tys[] = { Ptr->getType(), Size->getType() };
Chris Lattner143a07c2010-12-26 22:49:25 +000088 Module *M = BB->getParent()->getParent();
Benjamin Kramere6e19332011-07-14 17:45:39 +000089 Value *TheFn = Intrinsic::getDeclaration(M, Intrinsic::memset, Tys);
Chris Lattner143a07c2010-12-26 22:49:25 +000090
Jay Foad5bd375a2011-07-15 08:37:34 +000091 CallInst *CI = createCallHelper(TheFn, Ops, this);
Chris Lattner143a07c2010-12-26 22:49:25 +000092
93 // Set the TBAA info if present.
94 if (TBAATag)
95 CI->setMetadata(LLVMContext::MD_tbaa, TBAATag);
Hal Finkel94146652014-07-24 14:25:39 +000096
97 if (ScopeTag)
98 CI->setMetadata(LLVMContext::MD_alias_scope, ScopeTag);
99
100 if (NoAliasTag)
101 CI->setMetadata(LLVMContext::MD_noalias, NoAliasTag);
102
Chris Lattner143a07c2010-12-26 22:49:25 +0000103 return CI;
104}
105
106CallInst *IRBuilderBase::
Pete Cooper67cf9a72015-11-19 05:56:52 +0000107CreateMemCpy(Value *Dst, Value *Src, Value *Size, unsigned Align,
Hal Finkel94146652014-07-24 14:25:39 +0000108 bool isVolatile, MDNode *TBAATag, MDNode *TBAAStructTag,
109 MDNode *ScopeTag, MDNode *NoAliasTag) {
Chris Lattner143a07c2010-12-26 22:49:25 +0000110 Dst = getCastedInt8PtrValue(Dst);
111 Src = getCastedInt8PtrValue(Src);
112
Pete Cooper67cf9a72015-11-19 05:56:52 +0000113 Value *Ops[] = { Dst, Src, Size, getInt32(Align), getInt1(isVolatile) };
Jay Foadb804a2b2011-07-12 14:06:48 +0000114 Type *Tys[] = { Dst->getType(), Src->getType(), Size->getType() };
Chris Lattner143a07c2010-12-26 22:49:25 +0000115 Module *M = BB->getParent()->getParent();
Benjamin Kramere6e19332011-07-14 17:45:39 +0000116 Value *TheFn = Intrinsic::getDeclaration(M, Intrinsic::memcpy, Tys);
Chris Lattner143a07c2010-12-26 22:49:25 +0000117
Jay Foad5bd375a2011-07-15 08:37:34 +0000118 CallInst *CI = createCallHelper(TheFn, Ops, this);
Chris Lattner143a07c2010-12-26 22:49:25 +0000119
120 // Set the TBAA info if present.
121 if (TBAATag)
122 CI->setMetadata(LLVMContext::MD_tbaa, TBAATag);
Dan Gohman099727f2012-09-26 22:17:14 +0000123
124 // Set the TBAA Struct info if present.
125 if (TBAAStructTag)
126 CI->setMetadata(LLVMContext::MD_tbaa_struct, TBAAStructTag);
Hal Finkel94146652014-07-24 14:25:39 +0000127
128 if (ScopeTag)
129 CI->setMetadata(LLVMContext::MD_alias_scope, ScopeTag);
130
131 if (NoAliasTag)
132 CI->setMetadata(LLVMContext::MD_noalias, NoAliasTag);
133
Chris Lattner143a07c2010-12-26 22:49:25 +0000134 return CI;
135}
136
Daniel Neilson3faabbb2017-06-16 14:43:59 +0000137CallInst *IRBuilderBase::CreateElementUnorderedAtomicMemCpy(
Daniel Neilson6e4aa1e2017-11-10 19:38:12 +0000138 Value *Dst, unsigned DstAlign, Value *Src, unsigned SrcAlign, Value *Size,
139 uint32_t ElementSize, MDNode *TBAATag, MDNode *TBAAStructTag,
140 MDNode *ScopeTag, MDNode *NoAliasTag) {
141 assert(DstAlign >= ElementSize &&
142 "Pointer alignment must be at least element size");
143 assert(SrcAlign >= ElementSize &&
144 "Pointer alignment must be at least element size");
Anna Thomasb2a212c2017-06-06 16:45:25 +0000145 Dst = getCastedInt8PtrValue(Dst);
146 Src = getCastedInt8PtrValue(Src);
147
Daniel Neilson3faabbb2017-06-16 14:43:59 +0000148 Value *Ops[] = {Dst, Src, Size, getInt32(ElementSize)};
149 Type *Tys[] = {Dst->getType(), Src->getType(), Size->getType()};
Anna Thomasb2a212c2017-06-06 16:45:25 +0000150 Module *M = BB->getParent()->getParent();
Daniel Neilson3faabbb2017-06-16 14:43:59 +0000151 Value *TheFn = Intrinsic::getDeclaration(
152 M, Intrinsic::memcpy_element_unordered_atomic, Tys);
Anna Thomasb2a212c2017-06-06 16:45:25 +0000153
154 CallInst *CI = createCallHelper(TheFn, Ops, this);
155
Daniel Neilson6e4aa1e2017-11-10 19:38:12 +0000156 // Set the alignment of the pointer args.
157 CI->addParamAttr(0, Attribute::getWithAlignment(CI->getContext(), DstAlign));
158 CI->addParamAttr(1, Attribute::getWithAlignment(CI->getContext(), SrcAlign));
159
Anna Thomasb2a212c2017-06-06 16:45:25 +0000160 // Set the TBAA info if present.
161 if (TBAATag)
162 CI->setMetadata(LLVMContext::MD_tbaa, TBAATag);
163
164 // Set the TBAA Struct info if present.
165 if (TBAAStructTag)
166 CI->setMetadata(LLVMContext::MD_tbaa_struct, TBAAStructTag);
167
168 if (ScopeTag)
169 CI->setMetadata(LLVMContext::MD_alias_scope, ScopeTag);
170
171 if (NoAliasTag)
172 CI->setMetadata(LLVMContext::MD_noalias, NoAliasTag);
173
174 return CI;
175}
176
Chris Lattner143a07c2010-12-26 22:49:25 +0000177CallInst *IRBuilderBase::
Pete Cooper67cf9a72015-11-19 05:56:52 +0000178CreateMemMove(Value *Dst, Value *Src, Value *Size, unsigned Align,
Hal Finkel94146652014-07-24 14:25:39 +0000179 bool isVolatile, MDNode *TBAATag, MDNode *ScopeTag,
180 MDNode *NoAliasTag) {
Chris Lattner143a07c2010-12-26 22:49:25 +0000181 Dst = getCastedInt8PtrValue(Dst);
182 Src = getCastedInt8PtrValue(Src);
183
Pete Cooper67cf9a72015-11-19 05:56:52 +0000184 Value *Ops[] = { Dst, Src, Size, getInt32(Align), getInt1(isVolatile) };
Jay Foadb804a2b2011-07-12 14:06:48 +0000185 Type *Tys[] = { Dst->getType(), Src->getType(), Size->getType() };
Chris Lattner143a07c2010-12-26 22:49:25 +0000186 Module *M = BB->getParent()->getParent();
Benjamin Kramere6e19332011-07-14 17:45:39 +0000187 Value *TheFn = Intrinsic::getDeclaration(M, Intrinsic::memmove, Tys);
Chris Lattner143a07c2010-12-26 22:49:25 +0000188
Jay Foad5bd375a2011-07-15 08:37:34 +0000189 CallInst *CI = createCallHelper(TheFn, Ops, this);
Chris Lattner143a07c2010-12-26 22:49:25 +0000190
191 // Set the TBAA info if present.
192 if (TBAATag)
193 CI->setMetadata(LLVMContext::MD_tbaa, TBAATag);
Hal Finkel94146652014-07-24 14:25:39 +0000194
195 if (ScopeTag)
196 CI->setMetadata(LLVMContext::MD_alias_scope, ScopeTag);
197
198 if (NoAliasTag)
199 CI->setMetadata(LLVMContext::MD_noalias, NoAliasTag);
200
Chris Lattner143a07c2010-12-26 22:49:25 +0000201 return CI;
202}
Nick Lewyckybabca9a2011-05-21 23:14:36 +0000203
Amara Emersoncf9daa32017-05-09 10:43:25 +0000204static CallInst *getReductionIntrinsic(IRBuilderBase *Builder, Intrinsic::ID ID,
205 Value *Src) {
206 Module *M = Builder->GetInsertBlock()->getParent()->getParent();
207 Value *Ops[] = {Src};
208 Type *Tys[] = { Src->getType()->getVectorElementType(), Src->getType() };
209 auto Decl = Intrinsic::getDeclaration(M, ID, Tys);
210 return createCallHelper(Decl, Ops, Builder);
211}
212
213CallInst *IRBuilderBase::CreateFAddReduce(Value *Acc, Value *Src) {
214 Module *M = GetInsertBlock()->getParent()->getParent();
215 Value *Ops[] = {Acc, Src};
216 Type *Tys[] = {Src->getType()->getVectorElementType(), Acc->getType(),
217 Src->getType()};
218 auto Decl = Intrinsic::getDeclaration(
219 M, Intrinsic::experimental_vector_reduce_fadd, Tys);
220 return createCallHelper(Decl, Ops, this);
221}
222
223CallInst *IRBuilderBase::CreateFMulReduce(Value *Acc, Value *Src) {
224 Module *M = GetInsertBlock()->getParent()->getParent();
225 Value *Ops[] = {Acc, Src};
226 Type *Tys[] = {Src->getType()->getVectorElementType(), Acc->getType(),
227 Src->getType()};
228 auto Decl = Intrinsic::getDeclaration(
229 M, Intrinsic::experimental_vector_reduce_fmul, Tys);
230 return createCallHelper(Decl, Ops, this);
231}
232
233CallInst *IRBuilderBase::CreateAddReduce(Value *Src) {
234 return getReductionIntrinsic(this, Intrinsic::experimental_vector_reduce_add,
235 Src);
236}
237
238CallInst *IRBuilderBase::CreateMulReduce(Value *Src) {
239 return getReductionIntrinsic(this, Intrinsic::experimental_vector_reduce_mul,
240 Src);
241}
242
243CallInst *IRBuilderBase::CreateAndReduce(Value *Src) {
244 return getReductionIntrinsic(this, Intrinsic::experimental_vector_reduce_and,
245 Src);
246}
247
248CallInst *IRBuilderBase::CreateOrReduce(Value *Src) {
249 return getReductionIntrinsic(this, Intrinsic::experimental_vector_reduce_or,
250 Src);
251}
252
253CallInst *IRBuilderBase::CreateXorReduce(Value *Src) {
254 return getReductionIntrinsic(this, Intrinsic::experimental_vector_reduce_xor,
255 Src);
256}
257
258CallInst *IRBuilderBase::CreateIntMaxReduce(Value *Src, bool IsSigned) {
259 auto ID = IsSigned ? Intrinsic::experimental_vector_reduce_smax
260 : Intrinsic::experimental_vector_reduce_umax;
261 return getReductionIntrinsic(this, ID, Src);
262}
263
264CallInst *IRBuilderBase::CreateIntMinReduce(Value *Src, bool IsSigned) {
265 auto ID = IsSigned ? Intrinsic::experimental_vector_reduce_smin
266 : Intrinsic::experimental_vector_reduce_umin;
267 return getReductionIntrinsic(this, ID, Src);
268}
269
270CallInst *IRBuilderBase::CreateFPMaxReduce(Value *Src, bool NoNaN) {
271 auto Rdx = getReductionIntrinsic(
272 this, Intrinsic::experimental_vector_reduce_fmax, Src);
273 if (NoNaN) {
274 FastMathFlags FMF;
275 FMF.setNoNaNs();
276 Rdx->setFastMathFlags(FMF);
277 }
278 return Rdx;
279}
280
281CallInst *IRBuilderBase::CreateFPMinReduce(Value *Src, bool NoNaN) {
282 auto Rdx = getReductionIntrinsic(
283 this, Intrinsic::experimental_vector_reduce_fmin, Src);
284 if (NoNaN) {
285 FastMathFlags FMF;
286 FMF.setNoNaNs();
287 Rdx->setFastMathFlags(FMF);
288 }
289 return Rdx;
290}
291
Nick Lewyckybabca9a2011-05-21 23:14:36 +0000292CallInst *IRBuilderBase::CreateLifetimeStart(Value *Ptr, ConstantInt *Size) {
293 assert(isa<PointerType>(Ptr->getType()) &&
Bill Wendlingea6397f2012-07-19 00:11:40 +0000294 "lifetime.start only applies to pointers.");
Nick Lewyckybabca9a2011-05-21 23:14:36 +0000295 Ptr = getCastedInt8PtrValue(Ptr);
296 if (!Size)
297 Size = getInt64(-1);
298 else
299 assert(Size->getType() == getInt64Ty() &&
Bill Wendlingea6397f2012-07-19 00:11:40 +0000300 "lifetime.start requires the size to be an i64");
Nick Lewyckybabca9a2011-05-21 23:14:36 +0000301 Value *Ops[] = { Size, Ptr };
302 Module *M = BB->getParent()->getParent();
Matt Arsenaultf10061e2017-04-10 20:18:21 +0000303 Value *TheFn = Intrinsic::getDeclaration(M, Intrinsic::lifetime_start,
304 { Ptr->getType() });
Jay Foad5bd375a2011-07-15 08:37:34 +0000305 return createCallHelper(TheFn, Ops, this);
Nick Lewyckybabca9a2011-05-21 23:14:36 +0000306}
307
308CallInst *IRBuilderBase::CreateLifetimeEnd(Value *Ptr, ConstantInt *Size) {
309 assert(isa<PointerType>(Ptr->getType()) &&
Bill Wendlingea6397f2012-07-19 00:11:40 +0000310 "lifetime.end only applies to pointers.");
Nick Lewyckybabca9a2011-05-21 23:14:36 +0000311 Ptr = getCastedInt8PtrValue(Ptr);
312 if (!Size)
313 Size = getInt64(-1);
314 else
315 assert(Size->getType() == getInt64Ty() &&
Bill Wendlingea6397f2012-07-19 00:11:40 +0000316 "lifetime.end requires the size to be an i64");
Nick Lewyckybabca9a2011-05-21 23:14:36 +0000317 Value *Ops[] = { Size, Ptr };
318 Module *M = BB->getParent()->getParent();
Matt Arsenaultf10061e2017-04-10 20:18:21 +0000319 Value *TheFn = Intrinsic::getDeclaration(M, Intrinsic::lifetime_end,
320 { Ptr->getType() });
Jay Foad5bd375a2011-07-15 08:37:34 +0000321 return createCallHelper(TheFn, Ops, this);
Nick Lewyckybabca9a2011-05-21 23:14:36 +0000322}
Hal Finkel6f814db2014-10-15 23:44:22 +0000323
Anna Thomas58d11922016-07-22 20:57:23 +0000324CallInst *IRBuilderBase::CreateInvariantStart(Value *Ptr, ConstantInt *Size) {
325
326 assert(isa<PointerType>(Ptr->getType()) &&
327 "invariant.start only applies to pointers.");
328 Ptr = getCastedInt8PtrValue(Ptr);
329 if (!Size)
330 Size = getInt64(-1);
331 else
332 assert(Size->getType() == getInt64Ty() &&
333 "invariant.start requires the size to be an i64");
334
335 Value *Ops[] = {Size, Ptr};
336 // Fill in the single overloaded type: memory object type.
337 Type *ObjectPtr[1] = {Ptr->getType()};
338 Module *M = BB->getParent()->getParent();
339 Value *TheFn =
340 Intrinsic::getDeclaration(M, Intrinsic::invariant_start, ObjectPtr);
341 return createCallHelper(TheFn, Ops, this);
342}
343
Hal Finkel6f814db2014-10-15 23:44:22 +0000344CallInst *IRBuilderBase::CreateAssumption(Value *Cond) {
345 assert(Cond->getType() == getInt1Ty() &&
346 "an assumption condition must be of type i1");
347
348 Value *Ops[] = { Cond };
349 Module *M = BB->getParent()->getParent();
350 Value *FnAssume = Intrinsic::getDeclaration(M, Intrinsic::assume);
351 return createCallHelper(FnAssume, Ops, this);
352}
353
Elena Demikhovsky88e76ca2016-02-17 19:23:04 +0000354/// \brief Create a call to a Masked Load intrinsic.
355/// \p Ptr - base pointer for the load
356/// \p Align - alignment of the source location
357/// \p Mask - vector of booleans which indicates what vector lanes should
358/// be accessed in memory
359/// \p PassThru - pass-through value that is used to fill the masked-off lanes
360/// of the result
361/// \p Name - name of the result variable
Elena Demikhovsky84d19972014-12-30 14:28:14 +0000362CallInst *IRBuilderBase::CreateMaskedLoad(Value *Ptr, unsigned Align,
363 Value *Mask, Value *PassThru,
364 const Twine &Name) {
Artur Pilipenko7ad95ec2016-06-28 18:27:25 +0000365 PointerType *PtrTy = cast<PointerType>(Ptr->getType());
366 Type *DataTy = PtrTy->getElementType();
Elena Demikhovsky84d19972014-12-30 14:28:14 +0000367 assert(DataTy->isVectorTy() && "Ptr should point to a vector");
Ayal Zakse841b212017-07-31 13:21:42 +0000368 assert(Mask && "Mask should not be all-ones (null)");
Elena Demikhovsky84d19972014-12-30 14:28:14 +0000369 if (!PassThru)
370 PassThru = UndefValue::get(DataTy);
Artur Pilipenko7ad95ec2016-06-28 18:27:25 +0000371 Type *OverloadedTypes[] = { DataTy, PtrTy };
Elena Demikhovsky84d19972014-12-30 14:28:14 +0000372 Value *Ops[] = { Ptr, getInt32(Align), Mask, PassThru};
Artur Pilipenko7ad95ec2016-06-28 18:27:25 +0000373 return CreateMaskedIntrinsic(Intrinsic::masked_load, Ops,
374 OverloadedTypes, Name);
Elena Demikhovskyf1de34b2014-12-04 09:40:44 +0000375}
376
Elena Demikhovsky88e76ca2016-02-17 19:23:04 +0000377/// \brief Create a call to a Masked Store intrinsic.
378/// \p Val - data to be stored,
379/// \p Ptr - base pointer for the store
380/// \p Align - alignment of the destination location
381/// \p Mask - vector of booleans which indicates what vector lanes should
382/// be accessed in memory
Elena Demikhovsky84d19972014-12-30 14:28:14 +0000383CallInst *IRBuilderBase::CreateMaskedStore(Value *Val, Value *Ptr,
384 unsigned Align, Value *Mask) {
Artur Pilipenko7ad95ec2016-06-28 18:27:25 +0000385 PointerType *PtrTy = cast<PointerType>(Ptr->getType());
386 Type *DataTy = PtrTy->getElementType();
387 assert(DataTy->isVectorTy() && "Ptr should point to a vector");
Ayal Zakse841b212017-07-31 13:21:42 +0000388 assert(Mask && "Mask should not be all-ones (null)");
Artur Pilipenko7ad95ec2016-06-28 18:27:25 +0000389 Type *OverloadedTypes[] = { DataTy, PtrTy };
Elena Demikhovsky84d19972014-12-30 14:28:14 +0000390 Value *Ops[] = { Val, Ptr, getInt32(Align), Mask };
Artur Pilipenko7ad95ec2016-06-28 18:27:25 +0000391 return CreateMaskedIntrinsic(Intrinsic::masked_store, Ops, OverloadedTypes);
Elena Demikhovskyf1de34b2014-12-04 09:40:44 +0000392}
393
394/// Create a call to a Masked intrinsic, with given intrinsic Id,
Artur Pilipenko7ad95ec2016-06-28 18:27:25 +0000395/// an array of operands - Ops, and an array of overloaded types -
396/// OverloadedTypes.
Pete Cooper9e1d3352015-05-20 17:16:39 +0000397CallInst *IRBuilderBase::CreateMaskedIntrinsic(Intrinsic::ID Id,
Elena Demikhovskyf1de34b2014-12-04 09:40:44 +0000398 ArrayRef<Value *> Ops,
Artur Pilipenko7ad95ec2016-06-28 18:27:25 +0000399 ArrayRef<Type *> OverloadedTypes,
Elena Demikhovsky84d19972014-12-30 14:28:14 +0000400 const Twine &Name) {
Elena Demikhovskyf1de34b2014-12-04 09:40:44 +0000401 Module *M = BB->getParent()->getParent();
Pete Cooper9e1d3352015-05-20 17:16:39 +0000402 Value *TheFn = Intrinsic::getDeclaration(M, Id, OverloadedTypes);
Elena Demikhovsky84d19972014-12-30 14:28:14 +0000403 return createCallHelper(TheFn, Ops, this, Name);
Elena Demikhovskyf1de34b2014-12-04 09:40:44 +0000404}
Philip Reames4750dd12014-12-30 05:55:58 +0000405
Elena Demikhovsky88e76ca2016-02-17 19:23:04 +0000406/// \brief Create a call to a Masked Gather intrinsic.
407/// \p Ptrs - vector of pointers for loading
408/// \p Align - alignment for one element
409/// \p Mask - vector of booleans which indicates what vector lanes should
410/// be accessed in memory
411/// \p PassThru - pass-through value that is used to fill the masked-off lanes
412/// of the result
413/// \p Name - name of the result variable
414CallInst *IRBuilderBase::CreateMaskedGather(Value *Ptrs, unsigned Align,
415 Value *Mask, Value *PassThru,
416 const Twine& Name) {
417 auto PtrsTy = cast<VectorType>(Ptrs->getType());
418 auto PtrTy = cast<PointerType>(PtrsTy->getElementType());
419 unsigned NumElts = PtrsTy->getVectorNumElements();
420 Type *DataTy = VectorType::get(PtrTy->getElementType(), NumElts);
421
422 if (!Mask)
423 Mask = Constant::getAllOnesValue(VectorType::get(Type::getInt1Ty(Context),
424 NumElts));
425
Amara Emerson4d33c862017-05-19 10:40:18 +0000426 if (!PassThru)
427 PassThru = UndefValue::get(DataTy);
428
Elad Cohenef5798a2017-05-03 12:28:54 +0000429 Type *OverloadedTypes[] = {DataTy, PtrsTy};
Amara Emerson4d33c862017-05-19 10:40:18 +0000430 Value * Ops[] = {Ptrs, getInt32(Align), Mask, PassThru};
Elena Demikhovsky88e76ca2016-02-17 19:23:04 +0000431
432 // We specify only one type when we create this intrinsic. Types of other
433 // arguments are derived from this type.
Elad Cohenef5798a2017-05-03 12:28:54 +0000434 return CreateMaskedIntrinsic(Intrinsic::masked_gather, Ops, OverloadedTypes,
435 Name);
Elena Demikhovsky88e76ca2016-02-17 19:23:04 +0000436}
437
438/// \brief Create a call to a Masked Scatter intrinsic.
439/// \p Data - data to be stored,
440/// \p Ptrs - the vector of pointers, where the \p Data elements should be
441/// stored
442/// \p Align - alignment for one element
443/// \p Mask - vector of booleans which indicates what vector lanes should
444/// be accessed in memory
445CallInst *IRBuilderBase::CreateMaskedScatter(Value *Data, Value *Ptrs,
446 unsigned Align, Value *Mask) {
447 auto PtrsTy = cast<VectorType>(Ptrs->getType());
448 auto DataTy = cast<VectorType>(Data->getType());
Elena Demikhovsky88e76ca2016-02-17 19:23:04 +0000449 unsigned NumElts = PtrsTy->getVectorNumElements();
450
Tim Northover5a1a56c2016-02-17 21:16:59 +0000451#ifndef NDEBUG
452 auto PtrTy = cast<PointerType>(PtrsTy->getElementType());
Elena Demikhovsky88e76ca2016-02-17 19:23:04 +0000453 assert(NumElts == DataTy->getVectorNumElements() &&
Tim Northover5a1a56c2016-02-17 21:16:59 +0000454 PtrTy->getElementType() == DataTy->getElementType() &&
455 "Incompatible pointer and data types");
456#endif
Elena Demikhovsky88e76ca2016-02-17 19:23:04 +0000457
458 if (!Mask)
459 Mask = Constant::getAllOnesValue(VectorType::get(Type::getInt1Ty(Context),
460 NumElts));
Elad Cohenef5798a2017-05-03 12:28:54 +0000461
462 Type *OverloadedTypes[] = {DataTy, PtrsTy};
Elena Demikhovsky88e76ca2016-02-17 19:23:04 +0000463 Value * Ops[] = {Data, Ptrs, getInt32(Align), Mask};
464
465 // We specify only one type when we create this intrinsic. Types of other
466 // arguments are derived from this type.
Elad Cohenef5798a2017-05-03 12:28:54 +0000467 return CreateMaskedIntrinsic(Intrinsic::masked_scatter, Ops, OverloadedTypes);
Elena Demikhovsky88e76ca2016-02-17 19:23:04 +0000468}
469
Sanjoy Dasaf6980c2015-10-07 19:52:12 +0000470template <typename T0, typename T1, typename T2, typename T3>
Sanjoy Dasa1d39ba2015-05-12 23:52:24 +0000471static std::vector<Value *>
472getStatepointArgs(IRBuilderBase &B, uint64_t ID, uint32_t NumPatchBytes,
Sanjoy Das4fd3d402015-10-08 23:18:33 +0000473 Value *ActualCallee, uint32_t Flags, ArrayRef<T0> CallArgs,
474 ArrayRef<T1> TransitionArgs, ArrayRef<T2> DeoptArgs,
475 ArrayRef<T3> GCArgs) {
Sanjoy Dasabe1c682015-05-06 23:53:09 +0000476 std::vector<Value *> Args;
Sanjoy Dasa1d39ba2015-05-12 23:52:24 +0000477 Args.push_back(B.getInt64(ID));
478 Args.push_back(B.getInt32(NumPatchBytes));
Sanjoy Dasabe1c682015-05-06 23:53:09 +0000479 Args.push_back(ActualCallee);
480 Args.push_back(B.getInt32(CallArgs.size()));
Sanjoy Das4fd3d402015-10-08 23:18:33 +0000481 Args.push_back(B.getInt32(Flags));
Sanjoy Dasabe1c682015-05-06 23:53:09 +0000482 Args.insert(Args.end(), CallArgs.begin(), CallArgs.end());
Sanjoy Dasaf6980c2015-10-07 19:52:12 +0000483 Args.push_back(B.getInt32(TransitionArgs.size()));
484 Args.insert(Args.end(), TransitionArgs.begin(), TransitionArgs.end());
Sanjoy Dasabe1c682015-05-06 23:53:09 +0000485 Args.push_back(B.getInt32(DeoptArgs.size()));
486 Args.insert(Args.end(), DeoptArgs.begin(), DeoptArgs.end());
487 Args.insert(Args.end(), GCArgs.begin(), GCArgs.end());
488
489 return Args;
490}
491
Sanjoy Dasaf6980c2015-10-07 19:52:12 +0000492template <typename T0, typename T1, typename T2, typename T3>
493static CallInst *CreateGCStatepointCallCommon(
494 IRBuilderBase *Builder, uint64_t ID, uint32_t NumPatchBytes,
Sanjoy Das4fd3d402015-10-08 23:18:33 +0000495 Value *ActualCallee, uint32_t Flags, ArrayRef<T0> CallArgs,
Sanjoy Dasaf6980c2015-10-07 19:52:12 +0000496 ArrayRef<T1> TransitionArgs, ArrayRef<T2> DeoptArgs, ArrayRef<T3> GCArgs,
497 const Twine &Name) {
Sanjoy Das63245b52015-05-06 02:36:34 +0000498 // Extract out the type of the callee.
499 PointerType *FuncPtrType = cast<PointerType>(ActualCallee->getType());
500 assert(isa<FunctionType>(FuncPtrType->getElementType()) &&
501 "actual callee must be a callable value");
Philip Reames4750dd12014-12-30 05:55:58 +0000502
Sanjoy Dasaf6980c2015-10-07 19:52:12 +0000503 Module *M = Builder->GetInsertBlock()->getParent()->getParent();
Sanjoy Das63245b52015-05-06 02:36:34 +0000504 // Fill in the one generic type'd argument (the function is also vararg)
505 Type *ArgTypes[] = { FuncPtrType };
506 Function *FnStatepoint =
507 Intrinsic::getDeclaration(M, Intrinsic::experimental_gc_statepoint,
508 ArgTypes);
Philip Reames4750dd12014-12-30 05:55:58 +0000509
Sanjoy Dasaf6980c2015-10-07 19:52:12 +0000510 std::vector<llvm::Value *> Args =
511 getStatepointArgs(*Builder, ID, NumPatchBytes, ActualCallee, Flags,
512 CallArgs, TransitionArgs, DeoptArgs, GCArgs);
513 return createCallHelper(FnStatepoint, Args, Builder, Name);
514}
515
516CallInst *IRBuilderBase::CreateGCStatepointCall(
517 uint64_t ID, uint32_t NumPatchBytes, Value *ActualCallee,
518 ArrayRef<Value *> CallArgs, ArrayRef<Value *> DeoptArgs,
519 ArrayRef<Value *> GCArgs, const Twine &Name) {
520 return CreateGCStatepointCallCommon<Value *, Value *, Value *, Value *>(
Sanjoy Das4fd3d402015-10-08 23:18:33 +0000521 this, ID, NumPatchBytes, ActualCallee, uint32_t(StatepointFlags::None),
522 CallArgs, None /* No Transition Args */, DeoptArgs, GCArgs, Name);
Sanjoy Dasaf6980c2015-10-07 19:52:12 +0000523}
524
525CallInst *IRBuilderBase::CreateGCStatepointCall(
Sanjoy Das4fd3d402015-10-08 23:18:33 +0000526 uint64_t ID, uint32_t NumPatchBytes, Value *ActualCallee, uint32_t Flags,
527 ArrayRef<Use> CallArgs, ArrayRef<Use> TransitionArgs,
Sanjoy Dasaf6980c2015-10-07 19:52:12 +0000528 ArrayRef<Use> DeoptArgs, ArrayRef<Value *> GCArgs, const Twine &Name) {
529 return CreateGCStatepointCallCommon<Use, Use, Use, Value *>(
530 this, ID, NumPatchBytes, ActualCallee, Flags, CallArgs, TransitionArgs,
531 DeoptArgs, GCArgs, Name);
Philip Reames4750dd12014-12-30 05:55:58 +0000532}
533
Sanjoy Dasa1d39ba2015-05-12 23:52:24 +0000534CallInst *IRBuilderBase::CreateGCStatepointCall(
535 uint64_t ID, uint32_t NumPatchBytes, Value *ActualCallee,
536 ArrayRef<Use> CallArgs, ArrayRef<Value *> DeoptArgs,
537 ArrayRef<Value *> GCArgs, const Twine &Name) {
Sanjoy Dasaf6980c2015-10-07 19:52:12 +0000538 return CreateGCStatepointCallCommon<Use, Value *, Value *, Value *>(
Sanjoy Das4fd3d402015-10-08 23:18:33 +0000539 this, ID, NumPatchBytes, ActualCallee, uint32_t(StatepointFlags::None),
540 CallArgs, None, DeoptArgs, GCArgs, Name);
Sanjoy Dasaf6980c2015-10-07 19:52:12 +0000541}
542
543template <typename T0, typename T1, typename T2, typename T3>
544static InvokeInst *CreateGCStatepointInvokeCommon(
545 IRBuilderBase *Builder, uint64_t ID, uint32_t NumPatchBytes,
546 Value *ActualInvokee, BasicBlock *NormalDest, BasicBlock *UnwindDest,
Sanjoy Das4fd3d402015-10-08 23:18:33 +0000547 uint32_t Flags, ArrayRef<T0> InvokeArgs, ArrayRef<T1> TransitionArgs,
Sanjoy Dasaf6980c2015-10-07 19:52:12 +0000548 ArrayRef<T2> DeoptArgs, ArrayRef<T3> GCArgs, const Twine &Name) {
549 // Extract out the type of the callee.
550 PointerType *FuncPtrType = cast<PointerType>(ActualInvokee->getType());
551 assert(isa<FunctionType>(FuncPtrType->getElementType()) &&
552 "actual callee must be a callable value");
553
554 Module *M = Builder->GetInsertBlock()->getParent()->getParent();
555 // Fill in the one generic type'd argument (the function is also vararg)
556 Function *FnStatepoint = Intrinsic::getDeclaration(
557 M, Intrinsic::experimental_gc_statepoint, {FuncPtrType});
558
559 std::vector<llvm::Value *> Args =
560 getStatepointArgs(*Builder, ID, NumPatchBytes, ActualInvokee, Flags,
561 InvokeArgs, TransitionArgs, DeoptArgs, GCArgs);
562 return createInvokeHelper(FnStatepoint, NormalDest, UnwindDest, Args, Builder,
563 Name);
Sanjoy Dasabe1c682015-05-06 23:53:09 +0000564}
565
566InvokeInst *IRBuilderBase::CreateGCStatepointInvoke(
Sanjoy Dasa1d39ba2015-05-12 23:52:24 +0000567 uint64_t ID, uint32_t NumPatchBytes, Value *ActualInvokee,
568 BasicBlock *NormalDest, BasicBlock *UnwindDest,
Sanjoy Dasabe1c682015-05-06 23:53:09 +0000569 ArrayRef<Value *> InvokeArgs, ArrayRef<Value *> DeoptArgs,
570 ArrayRef<Value *> GCArgs, const Twine &Name) {
Sanjoy Dasaf6980c2015-10-07 19:52:12 +0000571 return CreateGCStatepointInvokeCommon<Value *, Value *, Value *, Value *>(
572 this, ID, NumPatchBytes, ActualInvokee, NormalDest, UnwindDest,
Sanjoy Das4fd3d402015-10-08 23:18:33 +0000573 uint32_t(StatepointFlags::None), InvokeArgs, None /* No Transition Args*/,
Sanjoy Dasaf6980c2015-10-07 19:52:12 +0000574 DeoptArgs, GCArgs, Name);
575}
Sanjoy Dasabe1c682015-05-06 23:53:09 +0000576
Sanjoy Dasaf6980c2015-10-07 19:52:12 +0000577InvokeInst *IRBuilderBase::CreateGCStatepointInvoke(
578 uint64_t ID, uint32_t NumPatchBytes, Value *ActualInvokee,
Sanjoy Das4fd3d402015-10-08 23:18:33 +0000579 BasicBlock *NormalDest, BasicBlock *UnwindDest, uint32_t Flags,
Sanjoy Dasaf6980c2015-10-07 19:52:12 +0000580 ArrayRef<Use> InvokeArgs, ArrayRef<Use> TransitionArgs,
581 ArrayRef<Use> DeoptArgs, ArrayRef<Value *> GCArgs, const Twine &Name) {
582 return CreateGCStatepointInvokeCommon<Use, Use, Use, Value *>(
583 this, ID, NumPatchBytes, ActualInvokee, NormalDest, UnwindDest, Flags,
584 InvokeArgs, TransitionArgs, DeoptArgs, GCArgs, Name);
Sanjoy Dasabe1c682015-05-06 23:53:09 +0000585}
586
587InvokeInst *IRBuilderBase::CreateGCStatepointInvoke(
Sanjoy Dasa1d39ba2015-05-12 23:52:24 +0000588 uint64_t ID, uint32_t NumPatchBytes, Value *ActualInvokee,
589 BasicBlock *NormalDest, BasicBlock *UnwindDest, ArrayRef<Use> InvokeArgs,
590 ArrayRef<Value *> DeoptArgs, ArrayRef<Value *> GCArgs, const Twine &Name) {
Sanjoy Dasaf6980c2015-10-07 19:52:12 +0000591 return CreateGCStatepointInvokeCommon<Use, Value *, Value *, Value *>(
592 this, ID, NumPatchBytes, ActualInvokee, NormalDest, UnwindDest,
Sanjoy Das4fd3d402015-10-08 23:18:33 +0000593 uint32_t(StatepointFlags::None), InvokeArgs, None, DeoptArgs, GCArgs,
594 Name);
Ramkumar Ramachandra3408f3e2015-02-26 00:35:56 +0000595}
596
Philip Reames4750dd12014-12-30 05:55:58 +0000597CallInst *IRBuilderBase::CreateGCResult(Instruction *Statepoint,
598 Type *ResultType,
599 const Twine &Name) {
Ramkumar Ramachandra75a4f352015-01-22 20:14:38 +0000600 Intrinsic::ID ID = Intrinsic::experimental_gc_result;
Philip Reames4750dd12014-12-30 05:55:58 +0000601 Module *M = BB->getParent()->getParent();
602 Type *Types[] = {ResultType};
603 Value *FnGCResult = Intrinsic::getDeclaration(M, ID, Types);
604
605 Value *Args[] = {Statepoint};
606 return createCallHelper(FnGCResult, Args, this, Name);
607}
608
609CallInst *IRBuilderBase::CreateGCRelocate(Instruction *Statepoint,
610 int BaseOffset,
611 int DerivedOffset,
612 Type *ResultType,
613 const Twine &Name) {
614 Module *M = BB->getParent()->getParent();
615 Type *Types[] = {ResultType};
616 Value *FnGCRelocate =
617 Intrinsic::getDeclaration(M, Intrinsic::experimental_gc_relocate, Types);
618
619 Value *Args[] = {Statepoint,
620 getInt32(BaseOffset),
621 getInt32(DerivedOffset)};
622 return createCallHelper(FnGCRelocate, Args, this, Name);
623}
Matt Arsenaultcdb468c2017-02-27 23:08:49 +0000624
625CallInst *IRBuilderBase::CreateBinaryIntrinsic(Intrinsic::ID ID,
626 Value *LHS, Value *RHS,
627 const Twine &Name) {
628 Module *M = BB->getParent()->getParent();
629 Function *Fn = Intrinsic::getDeclaration(M, ID, { LHS->getType() });
630 return createCallHelper(Fn, { LHS, RHS }, this, Name);
631}