blob: d361107cc0d2062a73dc9f8bd75f14384f155625 [file] [log] [blame]
Chandler Carruth74b6a772013-01-07 15:35:46 +00001//===- llvm/unittest/IR/IRBuilderTest.cpp - IRBuilder tests ---------------===//
Nick Lewyckybabca9a2011-05-21 23:14:36 +00002//
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
Chandler Carruth9fb823b2013-01-02 11:36:10 +000010#include "llvm/IR/IRBuilder.h"
Chandler Carruth9fb823b2013-01-02 11:36:10 +000011#include "llvm/IR/BasicBlock.h"
Adrian Prantl1292e242015-01-21 18:32:56 +000012#include "llvm/IR/DIBuilder.h"
Chandler Carruth9a67b072017-06-06 11:06:56 +000013#include "llvm/IR/DataLayout.h"
Chandler Carruth9fb823b2013-01-02 11:36:10 +000014#include "llvm/IR/Function.h"
15#include "llvm/IR/IntrinsicInst.h"
16#include "llvm/IR/LLVMContext.h"
17#include "llvm/IR/MDBuilder.h"
18#include "llvm/IR/Module.h"
Chandler Carruth64396b02014-03-04 12:05:47 +000019#include "llvm/IR/NoFolder.h"
Duncan P. N. Exon Smith3ee34e12015-03-31 02:09:55 +000020#include "llvm/IR/Verifier.h"
Nick Lewyckybabca9a2011-05-21 23:14:36 +000021#include "gtest/gtest.h"
22
23using namespace llvm;
24
David Blaikiea379b1812011-12-20 02:50:00 +000025namespace {
Chandler Carruth35e67062012-06-20 08:39:27 +000026
Nick Lewyckybabca9a2011-05-21 23:14:36 +000027class IRBuilderTest : public testing::Test {
28protected:
Alexander Kornienkof817c1c2015-04-11 02:11:45 +000029 void SetUp() override {
Benjamin Kramer63d39da2013-11-15 09:34:33 +000030 M.reset(new Module("MyModule", Ctx));
31 FunctionType *FTy = FunctionType::get(Type::getVoidTy(Ctx),
Nick Lewyckybabca9a2011-05-21 23:14:36 +000032 /*isVarArg=*/false);
Chandler Carruthb39c55f2012-07-16 07:44:51 +000033 F = Function::Create(FTy, Function::ExternalLinkage, "", M.get());
Benjamin Kramer63d39da2013-11-15 09:34:33 +000034 BB = BasicBlock::Create(Ctx, "", F);
35 GV = new GlobalVariable(*M, Type::getFloatTy(Ctx), true,
Craig Topper66f09ad2014-06-08 22:29:17 +000036 GlobalValue::ExternalLinkage, nullptr);
Nick Lewyckybabca9a2011-05-21 23:14:36 +000037 }
38
Alexander Kornienkof817c1c2015-04-11 02:11:45 +000039 void TearDown() override {
Craig Topper66f09ad2014-06-08 22:29:17 +000040 BB = nullptr;
Nick Lewyckybabca9a2011-05-21 23:14:36 +000041 M.reset();
42 }
43
Benjamin Kramer63d39da2013-11-15 09:34:33 +000044 LLVMContext Ctx;
Ahmed Charles56440fd2014-03-06 05:51:42 +000045 std::unique_ptr<Module> M;
Chandler Carruthb39c55f2012-07-16 07:44:51 +000046 Function *F;
Nick Lewyckybabca9a2011-05-21 23:14:36 +000047 BasicBlock *BB;
Michael Ilseman01853022012-11-28 21:17:34 +000048 GlobalVariable *GV;
Nick Lewyckybabca9a2011-05-21 23:14:36 +000049};
Nick Lewyckybabca9a2011-05-21 23:14:36 +000050
51TEST_F(IRBuilderTest, Lifetime) {
52 IRBuilder<> Builder(BB);
53 AllocaInst *Var1 = Builder.CreateAlloca(Builder.getInt8Ty());
54 AllocaInst *Var2 = Builder.CreateAlloca(Builder.getInt32Ty());
55 AllocaInst *Var3 = Builder.CreateAlloca(Builder.getInt8Ty(),
56 Builder.getInt32(123));
57
58 CallInst *Start1 = Builder.CreateLifetimeStart(Var1);
59 CallInst *Start2 = Builder.CreateLifetimeStart(Var2);
60 CallInst *Start3 = Builder.CreateLifetimeStart(Var3, Builder.getInt64(100));
61
62 EXPECT_EQ(Start1->getArgOperand(0), Builder.getInt64(-1));
63 EXPECT_EQ(Start2->getArgOperand(0), Builder.getInt64(-1));
64 EXPECT_EQ(Start3->getArgOperand(0), Builder.getInt64(100));
65
66 EXPECT_EQ(Start1->getArgOperand(1), Var1);
67 EXPECT_NE(Start2->getArgOperand(1), Var2);
68 EXPECT_EQ(Start3->getArgOperand(1), Var3);
69
70 Value *End1 = Builder.CreateLifetimeEnd(Var1);
71 Builder.CreateLifetimeEnd(Var2);
72 Builder.CreateLifetimeEnd(Var3);
73
74 IntrinsicInst *II_Start1 = dyn_cast<IntrinsicInst>(Start1);
75 IntrinsicInst *II_End1 = dyn_cast<IntrinsicInst>(End1);
Craig Topper66f09ad2014-06-08 22:29:17 +000076 ASSERT_TRUE(II_Start1 != nullptr);
Nick Lewyckybabca9a2011-05-21 23:14:36 +000077 EXPECT_EQ(II_Start1->getIntrinsicID(), Intrinsic::lifetime_start);
Craig Topper66f09ad2014-06-08 22:29:17 +000078 ASSERT_TRUE(II_End1 != nullptr);
Nick Lewyckybabca9a2011-05-21 23:14:36 +000079 EXPECT_EQ(II_End1->getIntrinsicID(), Intrinsic::lifetime_end);
80}
Chandler Carruth35e67062012-06-20 08:39:27 +000081
Chandler Carruthb39c55f2012-07-16 07:44:51 +000082TEST_F(IRBuilderTest, CreateCondBr) {
83 IRBuilder<> Builder(BB);
Benjamin Kramer63d39da2013-11-15 09:34:33 +000084 BasicBlock *TBB = BasicBlock::Create(Ctx, "", F);
85 BasicBlock *FBB = BasicBlock::Create(Ctx, "", F);
Chandler Carruthb39c55f2012-07-16 07:44:51 +000086
87 BranchInst *BI = Builder.CreateCondBr(Builder.getTrue(), TBB, FBB);
88 TerminatorInst *TI = BB->getTerminator();
89 EXPECT_EQ(BI, TI);
90 EXPECT_EQ(2u, TI->getNumSuccessors());
91 EXPECT_EQ(TBB, TI->getSuccessor(0));
92 EXPECT_EQ(FBB, TI->getSuccessor(1));
Chandler Carruthf5fe5562012-07-16 07:45:06 +000093
94 BI->eraseFromParent();
Benjamin Kramer63d39da2013-11-15 09:34:33 +000095 MDNode *Weights = MDBuilder(Ctx).createBranchWeights(42, 13);
Chandler Carruthf5fe5562012-07-16 07:45:06 +000096 BI = Builder.CreateCondBr(Builder.getTrue(), TBB, FBB, Weights);
97 TI = BB->getTerminator();
98 EXPECT_EQ(BI, TI);
99 EXPECT_EQ(2u, TI->getNumSuccessors());
100 EXPECT_EQ(TBB, TI->getSuccessor(0));
101 EXPECT_EQ(FBB, TI->getSuccessor(1));
102 EXPECT_EQ(Weights, TI->getMetadata(LLVMContext::MD_prof));
Chandler Carruthb39c55f2012-07-16 07:44:51 +0000103}
104
Duncan Sands0ac84732012-12-21 12:03:03 +0000105TEST_F(IRBuilderTest, LandingPadName) {
106 IRBuilder<> Builder(BB);
David Majnemer7fddecc2015-06-17 20:52:32 +0000107 LandingPadInst *LP = Builder.CreateLandingPad(Builder.getInt32Ty(), 0, "LP");
Duncan Sands0ac84732012-12-21 12:03:03 +0000108 EXPECT_EQ(LP->getName(), "LP");
109}
110
Rafael Espindola248ac132014-02-25 22:23:04 +0000111TEST_F(IRBuilderTest, DataLayout) {
Ahmed Charles56440fd2014-03-06 05:51:42 +0000112 std::unique_ptr<Module> M(new Module("test", Ctx));
Rafael Espindola248ac132014-02-25 22:23:04 +0000113 M->setDataLayout("e-n32");
Mehdi Amini46a43552015-03-04 18:43:29 +0000114 EXPECT_TRUE(M->getDataLayout().isLegalInteger(32));
Rafael Espindola248ac132014-02-25 22:23:04 +0000115 M->setDataLayout("e");
Mehdi Amini46a43552015-03-04 18:43:29 +0000116 EXPECT_FALSE(M->getDataLayout().isLegalInteger(32));
Rafael Espindola248ac132014-02-25 22:23:04 +0000117}
118
Evgeniy Stepanovef941692012-10-31 09:50:01 +0000119TEST_F(IRBuilderTest, GetIntTy) {
120 IRBuilder<> Builder(BB);
121 IntegerType *Ty1 = Builder.getInt1Ty();
Benjamin Kramer63d39da2013-11-15 09:34:33 +0000122 EXPECT_EQ(Ty1, IntegerType::get(Ctx, 1));
Evgeniy Stepanovef941692012-10-31 09:50:01 +0000123
124 DataLayout* DL = new DataLayout(M.get());
Mehdi Aminia28d91d2015-03-10 02:37:25 +0000125 IntegerType *IntPtrTy = Builder.getIntPtrTy(*DL);
Evgeniy Stepanovef941692012-10-31 09:50:01 +0000126 unsigned IntPtrBitSize = DL->getPointerSizeInBits(0);
Benjamin Kramer63d39da2013-11-15 09:34:33 +0000127 EXPECT_EQ(IntPtrTy, IntegerType::get(Ctx, IntPtrBitSize));
NAKAMURA Takumid86cd782013-01-23 08:31:28 +0000128 delete DL;
Evgeniy Stepanovef941692012-10-31 09:50:01 +0000129}
130
Michael Ilseman01853022012-11-28 21:17:34 +0000131TEST_F(IRBuilderTest, FastMathFlags) {
132 IRBuilder<> Builder(BB);
James Molloy88eb5352015-07-10 12:52:00 +0000133 Value *F, *FC;
Sanjay Patelfa54ace2015-12-14 21:59:03 +0000134 Instruction *FDiv, *FAdd, *FCmp, *FCall;
Michael Ilseman01853022012-11-28 21:17:34 +0000135
136 F = Builder.CreateLoad(GV);
137 F = Builder.CreateFAdd(F, F);
138
Michael Ilseman1833e002012-11-28 21:19:52 +0000139 EXPECT_FALSE(Builder.getFastMathFlags().any());
140 ASSERT_TRUE(isa<Instruction>(F));
141 FAdd = cast<Instruction>(F);
142 EXPECT_FALSE(FAdd->hasNoNaNs());
143
144 FastMathFlags FMF;
Sanjay Patela2528152016-01-12 18:03:37 +0000145 Builder.setFastMathFlags(FMF);
Michael Ilseman1833e002012-11-28 21:19:52 +0000146
147 F = Builder.CreateFAdd(F, F);
148 EXPECT_FALSE(Builder.getFastMathFlags().any());
149
Michael Ilseman65f14352012-12-09 21:12:04 +0000150 FMF.setUnsafeAlgebra();
Sanjay Patela2528152016-01-12 18:03:37 +0000151 Builder.setFastMathFlags(FMF);
Michael Ilseman1833e002012-11-28 21:19:52 +0000152
153 F = Builder.CreateFAdd(F, F);
154 EXPECT_TRUE(Builder.getFastMathFlags().any());
155 ASSERT_TRUE(isa<Instruction>(F));
156 FAdd = cast<Instruction>(F);
157 EXPECT_TRUE(FAdd->hasNoNaNs());
158
Michael Ilsemanbe92bcb2013-12-05 00:32:09 +0000159 // Now, try it with CreateBinOp
160 F = Builder.CreateBinOp(Instruction::FAdd, F, F);
161 EXPECT_TRUE(Builder.getFastMathFlags().any());
162 ASSERT_TRUE(isa<Instruction>(F));
163 FAdd = cast<Instruction>(F);
164 EXPECT_TRUE(FAdd->hasNoNaNs());
165
Michael Ilseman1833e002012-11-28 21:19:52 +0000166 F = Builder.CreateFDiv(F, F);
167 EXPECT_TRUE(Builder.getFastMathFlags().any());
168 EXPECT_TRUE(Builder.getFastMathFlags().UnsafeAlgebra);
169 ASSERT_TRUE(isa<Instruction>(F));
170 FDiv = cast<Instruction>(F);
171 EXPECT_TRUE(FDiv->hasAllowReciprocal());
172
173 Builder.clearFastMathFlags();
174
175 F = Builder.CreateFDiv(F, F);
176 ASSERT_TRUE(isa<Instruction>(F));
177 FDiv = cast<Instruction>(F);
178 EXPECT_FALSE(FDiv->hasAllowReciprocal());
179
180 FMF.clear();
Michael Ilseman65f14352012-12-09 21:12:04 +0000181 FMF.setAllowReciprocal();
Sanjay Patela2528152016-01-12 18:03:37 +0000182 Builder.setFastMathFlags(FMF);
Michael Ilseman1833e002012-11-28 21:19:52 +0000183
184 F = Builder.CreateFDiv(F, F);
185 EXPECT_TRUE(Builder.getFastMathFlags().any());
186 EXPECT_TRUE(Builder.getFastMathFlags().AllowReciprocal);
187 ASSERT_TRUE(isa<Instruction>(F));
188 FDiv = cast<Instruction>(F);
189 EXPECT_TRUE(FDiv->hasAllowReciprocal());
190
Michael Ilseman05d3bf77a2012-11-29 21:25:12 +0000191 Builder.clearFastMathFlags();
192
James Molloy88eb5352015-07-10 12:52:00 +0000193 FC = Builder.CreateFCmpOEQ(F, F);
194 ASSERT_TRUE(isa<Instruction>(FC));
195 FCmp = cast<Instruction>(FC);
196 EXPECT_FALSE(FCmp->hasAllowReciprocal());
197
198 FMF.clear();
199 FMF.setAllowReciprocal();
Sanjay Patela2528152016-01-12 18:03:37 +0000200 Builder.setFastMathFlags(FMF);
James Molloy88eb5352015-07-10 12:52:00 +0000201
202 FC = Builder.CreateFCmpOEQ(F, F);
203 EXPECT_TRUE(Builder.getFastMathFlags().any());
204 EXPECT_TRUE(Builder.getFastMathFlags().AllowReciprocal);
205 ASSERT_TRUE(isa<Instruction>(FC));
206 FCmp = cast<Instruction>(FC);
207 EXPECT_TRUE(FCmp->hasAllowReciprocal());
208
209 Builder.clearFastMathFlags();
Adam Nemetcd847a82017-03-28 20:11:52 +0000210
211 // Test FP-contract
212 FC = Builder.CreateFAdd(F, F);
213 ASSERT_TRUE(isa<Instruction>(FC));
214 FAdd = cast<Instruction>(FC);
215 EXPECT_FALSE(FAdd->hasAllowContract());
216
217 FMF.clear();
218 FMF.setAllowContract(true);
219 Builder.setFastMathFlags(FMF);
220
221 FC = Builder.CreateFAdd(F, F);
222 EXPECT_TRUE(Builder.getFastMathFlags().any());
223 EXPECT_TRUE(Builder.getFastMathFlags().AllowContract);
224 ASSERT_TRUE(isa<Instruction>(FC));
225 FAdd = cast<Instruction>(FC);
226 EXPECT_TRUE(FAdd->hasAllowContract());
227
228 Builder.clearFastMathFlags();
229
Sanjay Patelfa54ace2015-12-14 21:59:03 +0000230 // Test a call with FMF.
231 auto CalleeTy = FunctionType::get(Type::getFloatTy(Ctx),
232 /*isVarArg=*/false);
233 auto Callee =
234 Function::Create(CalleeTy, Function::ExternalLinkage, "", M.get());
235
236 FCall = Builder.CreateCall(Callee, None);
237 EXPECT_FALSE(FCall->hasNoNaNs());
238
Sanjay Patel947cccb2015-12-31 15:39:34 +0000239 Value *V =
240 Function::Create(CalleeTy, Function::ExternalLinkage, "", M.get());
241 FCall = Builder.CreateCall(V, None);
242 EXPECT_FALSE(FCall->hasNoNaNs());
243
Sanjay Patelfa54ace2015-12-14 21:59:03 +0000244 FMF.clear();
245 FMF.setNoNaNs();
Sanjay Patela2528152016-01-12 18:03:37 +0000246 Builder.setFastMathFlags(FMF);
Sanjay Patelfa54ace2015-12-14 21:59:03 +0000247
248 FCall = Builder.CreateCall(Callee, None);
249 EXPECT_TRUE(Builder.getFastMathFlags().any());
250 EXPECT_TRUE(Builder.getFastMathFlags().NoNaNs);
251 EXPECT_TRUE(FCall->hasNoNaNs());
252
Sanjay Patel947cccb2015-12-31 15:39:34 +0000253 FCall = Builder.CreateCall(V, None);
254 EXPECT_TRUE(Builder.getFastMathFlags().any());
255 EXPECT_TRUE(Builder.getFastMathFlags().NoNaNs);
256 EXPECT_TRUE(FCall->hasNoNaNs());
257
Sanjay Patelfa54ace2015-12-14 21:59:03 +0000258 Builder.clearFastMathFlags();
James Molloy88eb5352015-07-10 12:52:00 +0000259
Sanjay Patelb2325b92014-09-02 20:03:00 +0000260 // To test a copy, make sure that a '0' and a '1' change state.
Michael Ilseman05d3bf77a2012-11-29 21:25:12 +0000261 F = Builder.CreateFDiv(F, F);
262 ASSERT_TRUE(isa<Instruction>(F));
263 FDiv = cast<Instruction>(F);
264 EXPECT_FALSE(FDiv->getFastMathFlags().any());
Sanjay Patelb2325b92014-09-02 20:03:00 +0000265 FDiv->setHasAllowReciprocal(true);
266 FAdd->setHasAllowReciprocal(false);
Adam Nemet4226a9f2017-03-21 23:50:52 +0000267 FAdd->setHasNoNaNs(true);
Michael Ilseman05d3bf77a2012-11-29 21:25:12 +0000268 FDiv->copyFastMathFlags(FAdd);
269 EXPECT_TRUE(FDiv->hasNoNaNs());
Sanjay Patelb2325b92014-09-02 20:03:00 +0000270 EXPECT_FALSE(FDiv->hasAllowReciprocal());
Michael Ilseman05d3bf77a2012-11-29 21:25:12 +0000271
Michael Ilseman1833e002012-11-28 21:19:52 +0000272}
273
Chandler Carruth0db10c42014-01-05 03:22:33 +0000274TEST_F(IRBuilderTest, WrapFlags) {
Mehdi Aminiba9fba82016-03-13 21:05:13 +0000275 IRBuilder<NoFolder> Builder(BB);
Chandler Carruth0db10c42014-01-05 03:22:33 +0000276
277 // Test instructions.
278 GlobalVariable *G = new GlobalVariable(*M, Builder.getInt32Ty(), true,
Craig Topper66f09ad2014-06-08 22:29:17 +0000279 GlobalValue::ExternalLinkage, nullptr);
Chandler Carruth0db10c42014-01-05 03:22:33 +0000280 Value *V = Builder.CreateLoad(G);
281 EXPECT_TRUE(
282 cast<BinaryOperator>(Builder.CreateNSWAdd(V, V))->hasNoSignedWrap());
283 EXPECT_TRUE(
284 cast<BinaryOperator>(Builder.CreateNSWMul(V, V))->hasNoSignedWrap());
285 EXPECT_TRUE(
286 cast<BinaryOperator>(Builder.CreateNSWSub(V, V))->hasNoSignedWrap());
287 EXPECT_TRUE(cast<BinaryOperator>(
288 Builder.CreateShl(V, V, "", /* NUW */ false, /* NSW */ true))
289 ->hasNoSignedWrap());
290
291 EXPECT_TRUE(
292 cast<BinaryOperator>(Builder.CreateNUWAdd(V, V))->hasNoUnsignedWrap());
293 EXPECT_TRUE(
294 cast<BinaryOperator>(Builder.CreateNUWMul(V, V))->hasNoUnsignedWrap());
295 EXPECT_TRUE(
296 cast<BinaryOperator>(Builder.CreateNUWSub(V, V))->hasNoUnsignedWrap());
297 EXPECT_TRUE(cast<BinaryOperator>(
298 Builder.CreateShl(V, V, "", /* NUW */ true, /* NSW */ false))
299 ->hasNoUnsignedWrap());
300
301 // Test operators created with constants.
302 Constant *C = Builder.getInt32(42);
303 EXPECT_TRUE(cast<OverflowingBinaryOperator>(Builder.CreateNSWAdd(C, C))
304 ->hasNoSignedWrap());
305 EXPECT_TRUE(cast<OverflowingBinaryOperator>(Builder.CreateNSWSub(C, C))
306 ->hasNoSignedWrap());
307 EXPECT_TRUE(cast<OverflowingBinaryOperator>(Builder.CreateNSWMul(C, C))
308 ->hasNoSignedWrap());
309 EXPECT_TRUE(cast<OverflowingBinaryOperator>(
310 Builder.CreateShl(C, C, "", /* NUW */ false, /* NSW */ true))
311 ->hasNoSignedWrap());
312
313 EXPECT_TRUE(cast<OverflowingBinaryOperator>(Builder.CreateNUWAdd(C, C))
314 ->hasNoUnsignedWrap());
315 EXPECT_TRUE(cast<OverflowingBinaryOperator>(Builder.CreateNUWSub(C, C))
316 ->hasNoUnsignedWrap());
317 EXPECT_TRUE(cast<OverflowingBinaryOperator>(Builder.CreateNUWMul(C, C))
318 ->hasNoUnsignedWrap());
319 EXPECT_TRUE(cast<OverflowingBinaryOperator>(
320 Builder.CreateShl(C, C, "", /* NUW */ true, /* NSW */ false))
321 ->hasNoUnsignedWrap());
322}
323
Benjamin Kramerd36f1ab2013-09-30 15:39:48 +0000324TEST_F(IRBuilderTest, RAIIHelpersTest) {
325 IRBuilder<> Builder(BB);
326 EXPECT_FALSE(Builder.getFastMathFlags().allowReciprocal());
327 MDBuilder MDB(M->getContext());
328
Aaron Ballman3cab3742013-10-05 19:41:41 +0000329 MDNode *FPMathA = MDB.createFPMath(0.01f);
330 MDNode *FPMathB = MDB.createFPMath(0.1f);
Benjamin Kramerd36f1ab2013-09-30 15:39:48 +0000331
Sanjay Patela2528152016-01-12 18:03:37 +0000332 Builder.setDefaultFPMathTag(FPMathA);
Benjamin Kramerd36f1ab2013-09-30 15:39:48 +0000333
334 {
335 IRBuilder<>::FastMathFlagGuard Guard(Builder);
336 FastMathFlags FMF;
337 FMF.setAllowReciprocal();
Sanjay Patela2528152016-01-12 18:03:37 +0000338 Builder.setFastMathFlags(FMF);
339 Builder.setDefaultFPMathTag(FPMathB);
Benjamin Kramerd36f1ab2013-09-30 15:39:48 +0000340 EXPECT_TRUE(Builder.getFastMathFlags().allowReciprocal());
341 EXPECT_EQ(FPMathB, Builder.getDefaultFPMathTag());
342 }
343
344 EXPECT_FALSE(Builder.getFastMathFlags().allowReciprocal());
345 EXPECT_EQ(FPMathA, Builder.getDefaultFPMathTag());
346
347 Value *F = Builder.CreateLoad(GV);
348
349 {
350 IRBuilder<>::InsertPointGuard Guard(Builder);
351 Builder.SetInsertPoint(cast<Instruction>(F));
Duncan P. N. Exon Smithc8925b12015-10-20 18:30:20 +0000352 EXPECT_EQ(F, &*Builder.GetInsertPoint());
Benjamin Kramerd36f1ab2013-09-30 15:39:48 +0000353 }
354
355 EXPECT_EQ(BB->end(), Builder.GetInsertPoint());
356 EXPECT_EQ(BB, Builder.GetInsertBlock());
357}
358
Adrian Prantl1d12b882017-04-26 22:56:44 +0000359TEST_F(IRBuilderTest, createFunction) {
360 IRBuilder<> Builder(BB);
361 DIBuilder DIB(*M);
362 auto File = DIB.createFile("error.swift", "/");
363 auto CU =
364 DIB.createCompileUnit(dwarf::DW_LANG_Swift, File, "swiftc", true, "", 0);
365 auto Type = DIB.createSubroutineType(DIB.getOrCreateTypeArray(None));
366 auto NoErr = DIB.createFunction(CU, "noerr", "", File, 1, Type, false, true, 1,
367 DINode::FlagZero, true);
368 EXPECT_TRUE(!NoErr->getThrownTypes());
369 auto Int = DIB.createBasicType("Int", 64, dwarf::DW_ATE_signed);
370 auto Error = DIB.getOrCreateArray({Int});
371 auto Err =
372 DIB.createFunction(CU, "err", "", File, 1, Type, false, true, 1,
373 DINode::FlagZero, true, nullptr, nullptr, Error.get());
374 EXPECT_TRUE(Err->getThrownTypes().get() == Error.get());
375 DIB.finalize();
376}
377
Adrian Prantl1292e242015-01-21 18:32:56 +0000378TEST_F(IRBuilderTest, DIBuilder) {
379 IRBuilder<> Builder(BB);
380 DIBuilder DIB(*M);
381 auto File = DIB.createFile("F.CBL", "/");
Amjad Aboud43c8b6b2016-12-14 20:24:54 +0000382 auto CU = DIB.createCompileUnit(dwarf::DW_LANG_Cobol74,
383 DIB.createFile("F.CBL", "/"), "llvm-cobol74",
384 true, "", 0);
Eric Christopherbdafb3c2015-10-15 06:56:10 +0000385 auto Type = DIB.createSubroutineType(DIB.getOrCreateTypeArray(None));
Leny Kholodov40c62352016-09-06 17:03:02 +0000386 auto SP = DIB.createFunction(CU, "foo", "", File, 1, Type, false, true, 1,
387 DINode::FlagZero, true);
Peter Collingbourned4bff302015-11-05 22:03:56 +0000388 F->setSubprogram(SP);
Adrian Prantl1292e242015-01-21 18:32:56 +0000389 AllocaInst *I = Builder.CreateAlloca(Builder.getInt8Ty());
Leny Kholodov40c62352016-09-06 17:03:02 +0000390 auto BarSP = DIB.createFunction(CU, "bar", "", File, 1, Type, false, true, 1,
391 DINode::FlagZero, true);
Duncan P. N. Exon Smith0e202b92015-03-30 16:37:48 +0000392 auto BadScope = DIB.createLexicalBlockFile(BarSP, File, 0);
Adrian Prantl1292e242015-01-21 18:32:56 +0000393 I->setDebugLoc(DebugLoc::get(2, 0, BadScope));
Adrian Prantl53d382f2015-01-21 19:25:35 +0000394 DIB.finalize();
Duncan P. N. Exon Smith3ee34e12015-03-31 02:09:55 +0000395 EXPECT_TRUE(verifyModule(*M));
Adrian Prantl1292e242015-01-21 18:32:56 +0000396}
397
Pawel Bylica0c2ab372015-04-30 09:01:22 +0000398TEST_F(IRBuilderTest, InsertExtractElement) {
399 IRBuilder<> Builder(BB);
400
401 auto VecTy = VectorType::get(Builder.getInt64Ty(), 4);
402 auto Elt1 = Builder.getInt64(-1);
403 auto Elt2 = Builder.getInt64(-2);
404 Value *Vec = UndefValue::get(VecTy);
405 Vec = Builder.CreateInsertElement(Vec, Elt1, Builder.getInt8(1));
406 Vec = Builder.CreateInsertElement(Vec, Elt2, 2);
407 auto X1 = Builder.CreateExtractElement(Vec, 1);
408 auto X2 = Builder.CreateExtractElement(Vec, Builder.getInt32(2));
409 EXPECT_EQ(Elt1, X1);
410 EXPECT_EQ(Elt2, X2);
411}
412
Tobias Grosser94aed4f2015-06-19 07:19:17 +0000413TEST_F(IRBuilderTest, CreateGlobalStringPtr) {
414 IRBuilder<> Builder(BB);
Benjamin Kramerd36f1ab2013-09-30 15:39:48 +0000415
Tobias Grosser94aed4f2015-06-19 07:19:17 +0000416 auto String1a = Builder.CreateGlobalStringPtr("TestString", "String1a");
417 auto String1b = Builder.CreateGlobalStringPtr("TestString", "String1b", 0);
418 auto String2 = Builder.CreateGlobalStringPtr("TestString", "String2", 1);
419 auto String3 = Builder.CreateGlobalString("TestString", "String3", 2);
420
421 EXPECT_TRUE(String1a->getType()->getPointerAddressSpace() == 0);
422 EXPECT_TRUE(String1b->getType()->getPointerAddressSpace() == 0);
423 EXPECT_TRUE(String2->getType()->getPointerAddressSpace() == 1);
424 EXPECT_TRUE(String3->getType()->getPointerAddressSpace() == 2);
425}
Alexey Samsonove5039b72015-06-30 19:07:20 +0000426
427TEST_F(IRBuilderTest, DebugLoc) {
428 auto CalleeTy = FunctionType::get(Type::getVoidTy(Ctx),
429 /*isVarArg=*/false);
430 auto Callee =
431 Function::Create(CalleeTy, Function::ExternalLinkage, "", M.get());
432
433 DIBuilder DIB(*M);
434 auto File = DIB.createFile("tmp.cpp", "/");
Amjad Aboud43c8b6b2016-12-14 20:24:54 +0000435 auto CU = DIB.createCompileUnit(dwarf::DW_LANG_C_plus_plus_11,
436 DIB.createFile("tmp.cpp", "/"), "", true, "",
437 0);
Eric Christopherbdafb3c2015-10-15 06:56:10 +0000438 auto SPType = DIB.createSubroutineType(DIB.getOrCreateTypeArray(None));
Alexey Samsonove5039b72015-06-30 19:07:20 +0000439 auto SP =
Alexey Samsonovafadf062015-06-30 22:17:29 +0000440 DIB.createFunction(CU, "foo", "foo", File, 1, SPType, false, true, 1);
Alexey Samsonove5039b72015-06-30 19:07:20 +0000441 DebugLoc DL1 = DILocation::get(Ctx, 2, 0, SP);
442 DebugLoc DL2 = DILocation::get(Ctx, 3, 0, SP);
443
444 auto BB2 = BasicBlock::Create(Ctx, "bb2", F);
445 auto Br = BranchInst::Create(BB2, BB);
446 Br->setDebugLoc(DL1);
447
448 IRBuilder<> Builder(Ctx);
449 Builder.SetInsertPoint(Br);
450 EXPECT_EQ(DL1, Builder.getCurrentDebugLocation());
451 auto Call1 = Builder.CreateCall(Callee, None);
452 EXPECT_EQ(DL1, Call1->getDebugLoc());
453
454 Call1->setDebugLoc(DL2);
Duncan P. N. Exon Smithc8925b12015-10-20 18:30:20 +0000455 Builder.SetInsertPoint(Call1->getParent(), Call1->getIterator());
Alexey Samsonove5039b72015-06-30 19:07:20 +0000456 EXPECT_EQ(DL2, Builder.getCurrentDebugLocation());
457 auto Call2 = Builder.CreateCall(Callee, None);
458 EXPECT_EQ(DL2, Call2->getDebugLoc());
Alexey Samsonovafadf062015-06-30 22:17:29 +0000459
460 DIB.finalize();
Alexey Samsonove5039b72015-06-30 19:07:20 +0000461}
Amjad Aboud62f6f5c2016-03-13 11:11:39 +0000462
463TEST_F(IRBuilderTest, DIImportedEntity) {
464 IRBuilder<> Builder(BB);
465 DIBuilder DIB(*M);
Adrian Prantld63bfd22017-07-19 00:09:54 +0000466 auto F = DIB.createFile("F.CBL", "/");
Amjad Aboud43c8b6b2016-12-14 20:24:54 +0000467 auto CU = DIB.createCompileUnit(dwarf::DW_LANG_Cobol74,
Adrian Prantld63bfd22017-07-19 00:09:54 +0000468 F, "llvm-cobol74",
Amjad Aboud43c8b6b2016-12-14 20:24:54 +0000469 true, "", 0);
Adrian Prantld63bfd22017-07-19 00:09:54 +0000470 DIB.createImportedDeclaration(CU, nullptr, F, 1);
471 DIB.createImportedDeclaration(CU, nullptr, F, 1);
472 DIB.createImportedModule(CU, (DIImportedEntity *)nullptr, F, 2);
473 DIB.createImportedModule(CU, (DIImportedEntity *)nullptr, F, 2);
Amjad Aboud62f6f5c2016-03-13 11:11:39 +0000474 DIB.finalize();
475 EXPECT_TRUE(verifyModule(*M));
476 EXPECT_TRUE(CU->getImportedEntities().size() == 2);
477}
Amjad Aboud96075712017-01-12 15:49:46 +0000478
479// 0: #define M0 V0 <-- command line definition
480// 0: main.c <-- main file
481// 3: #define M1 V1 <-- M1 definition in main.c
482// 5: #include "file.h" <-- inclusion of file.h from main.c
483// 1: #define M2 <-- M2 definition in file.h with no value
484// 7: #undef M1 V1 <-- M1 un-definition in main.c
485TEST_F(IRBuilderTest, DIBuilderMacro) {
486 IRBuilder<> Builder(BB);
487 DIBuilder DIB(*M);
488 auto File1 = DIB.createFile("main.c", "/");
489 auto File2 = DIB.createFile("file.h", "/");
490 auto CU = DIB.createCompileUnit(
491 dwarf::DW_LANG_C, DIB.createFile("main.c", "/"), "llvm-c", true, "", 0);
492 auto MDef0 =
493 DIB.createMacro(nullptr, 0, dwarf::DW_MACINFO_define, "M0", "V0");
494 auto TMF1 = DIB.createTempMacroFile(nullptr, 0, File1);
495 auto MDef1 = DIB.createMacro(TMF1, 3, dwarf::DW_MACINFO_define, "M1", "V1");
496 auto TMF2 = DIB.createTempMacroFile(TMF1, 5, File2);
497 auto MDef2 = DIB.createMacro(TMF2, 1, dwarf::DW_MACINFO_define, "M2");
498 auto MUndef1 = DIB.createMacro(TMF1, 7, dwarf::DW_MACINFO_undef, "M1");
499
500 EXPECT_EQ(dwarf::DW_MACINFO_define, MDef1->getMacinfoType());
David Blaikiec7e51b12017-01-12 17:44:32 +0000501 EXPECT_EQ(3u, MDef1->getLine());
Amjad Aboud96075712017-01-12 15:49:46 +0000502 EXPECT_EQ("M1", MDef1->getName());
503 EXPECT_EQ("V1", MDef1->getValue());
504
505 EXPECT_EQ(dwarf::DW_MACINFO_undef, MUndef1->getMacinfoType());
David Blaikiec7e51b12017-01-12 17:44:32 +0000506 EXPECT_EQ(7u, MUndef1->getLine());
Amjad Aboud96075712017-01-12 15:49:46 +0000507 EXPECT_EQ("M1", MUndef1->getName());
508 EXPECT_EQ("", MUndef1->getValue());
509
510 EXPECT_EQ(dwarf::DW_MACINFO_start_file, TMF2->getMacinfoType());
David Blaikiec7e51b12017-01-12 17:44:32 +0000511 EXPECT_EQ(5u, TMF2->getLine());
Amjad Aboud96075712017-01-12 15:49:46 +0000512 EXPECT_EQ(File2, TMF2->getFile());
513
514 DIB.finalize();
515
516 SmallVector<Metadata *, 4> Elements;
517 Elements.push_back(MDef2);
518 auto MF2 = DIMacroFile::get(Ctx, dwarf::DW_MACINFO_start_file, 5, File2,
519 DIB.getOrCreateMacroArray(Elements));
520
521 Elements.clear();
522 Elements.push_back(MDef1);
523 Elements.push_back(MF2);
524 Elements.push_back(MUndef1);
525 auto MF1 = DIMacroFile::get(Ctx, dwarf::DW_MACINFO_start_file, 0, File1,
526 DIB.getOrCreateMacroArray(Elements));
527
528 Elements.clear();
529 Elements.push_back(MDef0);
530 Elements.push_back(MF1);
531 auto MN0 = MDTuple::get(Ctx, Elements);
532 EXPECT_EQ(MN0, CU->getRawMacros());
533
534 Elements.clear();
535 Elements.push_back(MDef1);
536 Elements.push_back(MF2);
537 Elements.push_back(MUndef1);
538 auto MN1 = MDTuple::get(Ctx, Elements);
539 EXPECT_EQ(MN1, MF1->getRawElements());
540
541 Elements.clear();
542 Elements.push_back(MDef2);
543 auto MN2 = MDTuple::get(Ctx, Elements);
544 EXPECT_EQ(MN2, MF2->getRawElements());
545 EXPECT_TRUE(verifyModule(*M));
546}
Chandler Carruth35e67062012-06-20 08:39:27 +0000547}