1 //===- unittests/IR/ModuleTest.cpp - Module unit tests --------------------===// 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/IR/Module.h" 11 #include "llvm/IR/GlobalVariable.h" 12 #include "llvm/Support/RandomNumberGenerator.h" 13 #include "gtest/gtest.h" 14 15 #include <random> 16 17 using namespace llvm; 18 19 namespace { 20 21 bool sortByName(const GlobalVariable &L, const GlobalVariable &R) { 22 return L.getName() < R.getName(); 23 } 24 25 bool sortByNameReverse(const GlobalVariable &L, const GlobalVariable &R) { 26 return sortByName(R, L); 27 } 28 29 TEST(ModuleTest, sortGlobalsByName) { 30 LLVMContext Context; 31 for (auto compare : {&sortByName, &sortByNameReverse}) { 32 Module M("M", Context); 33 Type *T = Type::getInt8Ty(Context); 34 GlobalValue::LinkageTypes L = GlobalValue::ExternalLinkage; 35 (void)new GlobalVariable(M, T, false, L, nullptr, "A"); 36 (void)new GlobalVariable(M, T, false, L, nullptr, "F"); 37 (void)new GlobalVariable(M, T, false, L, nullptr, "G"); 38 (void)new GlobalVariable(M, T, false, L, nullptr, "E"); 39 (void)new GlobalVariable(M, T, false, L, nullptr, "B"); 40 (void)new GlobalVariable(M, T, false, L, nullptr, "H"); 41 (void)new GlobalVariable(M, T, false, L, nullptr, "C"); 42 (void)new GlobalVariable(M, T, false, L, nullptr, "D"); 43 44 // Sort the globals by name. 45 EXPECT_FALSE(std::is_sorted(M.global_begin(), M.global_end(), compare)); 46 M.getGlobalList().sort(compare); 47 EXPECT_TRUE(std::is_sorted(M.global_begin(), M.global_end(), compare)); 48 } 49 } 50 51 TEST(ModuleTest, randomNumberGenerator) { 52 LLVMContext Context; 53 static char ID; 54 struct DummyPass : ModulePass { 55 DummyPass() : ModulePass(ID) {} 56 bool runOnModule(Module &) { return true; } 57 } DP; 58 59 Module M("R", Context); 60 61 std::uniform_int_distribution<int> dist; 62 const size_t NBCheck = 10; 63 64 std::array<int, NBCheck> RandomStreams[2]; 65 for (auto &RandomStream : RandomStreams) { 66 std::unique_ptr<RandomNumberGenerator> RNG = M.createRNG(&DP); 67 std::generate(RandomStream.begin(), RandomStream.end(), 68 [&]() { return dist(*RNG); }); 69 } 70 71 EXPECT_TRUE(std::equal(RandomStreams[0].begin(), RandomStreams[0].end(), 72 RandomStreams[1].begin())); 73 } 74 75 } // end namespace 76