1 //===- Local.cpp - Unit tests for Local -----------------------------------===// 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 #include "llvm/BasicBlock.h" 11 #include "llvm/IRBuilder.h" 12 #include "llvm/Instructions.h" 13 #include "llvm/LLVMContext.h" 14 #include "llvm/Transforms/Utils/Local.h" 15 16 #include "gtest/gtest.h" 17 18 using namespace llvm; 19 20 TEST(Local, RecursivelyDeleteDeadPHINodes) { 21 LLVMContext &C(getGlobalContext()); 22 23 IRBuilder<> builder(C); 24 25 // Make blocks 26 BasicBlock *bb0 = BasicBlock::Create(C); 27 BasicBlock *bb1 = BasicBlock::Create(C); 28 29 builder.SetInsertPoint(bb0); 30 PHINode *phi = builder.CreatePHI(Type::getInt32Ty(C), 2); 31 BranchInst *br0 = builder.CreateCondBr(builder.getTrue(), bb0, bb1); 32 33 builder.SetInsertPoint(bb1); 34 BranchInst *br1 = builder.CreateBr(bb0); 35 36 phi->addIncoming(phi, bb0); 37 phi->addIncoming(phi, bb1); 38 39 // The PHI will be removed 40 EXPECT_TRUE(RecursivelyDeleteDeadPHINode(phi)); 41 42 // Make sure the blocks only contain the branches 43 EXPECT_EQ(&bb0->front(), br0); 44 EXPECT_EQ(&bb1->front(), br1); 45 46 builder.SetInsertPoint(bb0); 47 phi = builder.CreatePHI(Type::getInt32Ty(C), 0); 48 49 EXPECT_TRUE(RecursivelyDeleteDeadPHINode(phi)); 50 51 builder.SetInsertPoint(bb0); 52 phi = builder.CreatePHI(Type::getInt32Ty(C), 0); 53 builder.CreateAdd(phi, phi); 54 55 EXPECT_TRUE(RecursivelyDeleteDeadPHINode(phi)); 56 57 bb0->dropAllReferences(); 58 bb1->dropAllReferences(); 59 delete bb0; 60 delete bb1; 61 } 62