1 //===- llvm/unittest/IR/ConstantsTest.cpp - Constants unit tests ----------===// 2 // 3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4 // See https://llvm.org/LICENSE.txt for license information. 5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6 // 7 //===----------------------------------------------------------------------===// 8 9 #include "llvm/IR/Constants.h" 10 #include "llvm-c/Core.h" 11 #include "llvm/AsmParser/Parser.h" 12 #include "llvm/IR/DerivedTypes.h" 13 #include "llvm/IR/InstrTypes.h" 14 #include "llvm/IR/Instruction.h" 15 #include "llvm/IR/LLVMContext.h" 16 #include "llvm/IR/Module.h" 17 #include "llvm/Support/SourceMgr.h" 18 #include "gtest/gtest.h" 19 20 namespace llvm { 21 namespace { 22 23 TEST(ConstantsTest, Integer_i1) { 24 LLVMContext Context; 25 IntegerType *Int1 = IntegerType::get(Context, 1); 26 Constant *One = ConstantInt::get(Int1, 1, true); 27 Constant *Zero = ConstantInt::get(Int1, 0); 28 Constant *NegOne = ConstantInt::get(Int1, static_cast<uint64_t>(-1), true); 29 EXPECT_EQ(NegOne, ConstantInt::getSigned(Int1, -1)); 30 Constant *Poison = PoisonValue::get(Int1); 31 32 // Input: @b = constant i1 add(i1 1 , i1 1) 33 // Output: @b = constant i1 false 34 EXPECT_EQ(Zero, ConstantExpr::getAdd(One, One)); 35 36 // @c = constant i1 add(i1 -1, i1 1) 37 // @c = constant i1 false 38 EXPECT_EQ(Zero, ConstantExpr::getAdd(NegOne, One)); 39 40 // @d = constant i1 add(i1 -1, i1 -1) 41 // @d = constant i1 false 42 EXPECT_EQ(Zero, ConstantExpr::getAdd(NegOne, NegOne)); 43 44 // @e = constant i1 sub(i1 -1, i1 1) 45 // @e = constant i1 false 46 EXPECT_EQ(Zero, ConstantExpr::getSub(NegOne, One)); 47 48 // @f = constant i1 sub(i1 1 , i1 -1) 49 // @f = constant i1 false 50 EXPECT_EQ(Zero, ConstantExpr::getSub(One, NegOne)); 51 52 // @g = constant i1 sub(i1 1 , i1 1) 53 // @g = constant i1 false 54 EXPECT_EQ(Zero, ConstantExpr::getSub(One, One)); 55 56 // @h = constant i1 shl(i1 1 , i1 1) ; poison 57 // @h = constant i1 poison 58 EXPECT_EQ(Poison, ConstantExpr::getShl(One, One)); 59 60 // @i = constant i1 shl(i1 1 , i1 0) 61 // @i = constant i1 true 62 EXPECT_EQ(One, ConstantExpr::getShl(One, Zero)); 63 64 // @j = constant i1 lshr(i1 1, i1 1) ; poison 65 // @j = constant i1 poison 66 EXPECT_EQ(Poison, ConstantExpr::getLShr(One, One)); 67 68 // @m = constant i1 ashr(i1 1, i1 1) ; poison 69 // @m = constant i1 poison 70 EXPECT_EQ(Poison, ConstantExpr::getAShr(One, One)); 71 72 // @n = constant i1 mul(i1 -1, i1 1) 73 // @n = constant i1 true 74 EXPECT_EQ(One, ConstantExpr::getMul(NegOne, One)); 75 76 // @o = constant i1 sdiv(i1 -1, i1 1) ; overflow 77 // @o = constant i1 true 78 EXPECT_EQ(One, ConstantExpr::getSDiv(NegOne, One)); 79 80 // @p = constant i1 sdiv(i1 1 , i1 -1); overflow 81 // @p = constant i1 true 82 EXPECT_EQ(One, ConstantExpr::getSDiv(One, NegOne)); 83 84 // @q = constant i1 udiv(i1 -1, i1 1) 85 // @q = constant i1 true 86 EXPECT_EQ(One, ConstantExpr::getUDiv(NegOne, One)); 87 88 // @r = constant i1 udiv(i1 1, i1 -1) 89 // @r = constant i1 true 90 EXPECT_EQ(One, ConstantExpr::getUDiv(One, NegOne)); 91 92 // @s = constant i1 srem(i1 -1, i1 1) ; overflow 93 // @s = constant i1 false 94 EXPECT_EQ(Zero, ConstantExpr::getSRem(NegOne, One)); 95 96 // @t = constant i1 urem(i1 -1, i1 1) 97 // @t = constant i1 false 98 EXPECT_EQ(Zero, ConstantExpr::getURem(NegOne, One)); 99 100 // @u = constant i1 srem(i1 1, i1 -1) ; overflow 101 // @u = constant i1 false 102 EXPECT_EQ(Zero, ConstantExpr::getSRem(One, NegOne)); 103 } 104 105 TEST(ConstantsTest, IntSigns) { 106 LLVMContext Context; 107 IntegerType *Int8Ty = Type::getInt8Ty(Context); 108 EXPECT_EQ(100, ConstantInt::get(Int8Ty, 100, false)->getSExtValue()); 109 EXPECT_EQ(100, ConstantInt::get(Int8Ty, 100, true)->getSExtValue()); 110 EXPECT_EQ(100, ConstantInt::getSigned(Int8Ty, 100)->getSExtValue()); 111 EXPECT_EQ(-50, ConstantInt::get(Int8Ty, 206)->getSExtValue()); 112 EXPECT_EQ(-50, ConstantInt::getSigned(Int8Ty, -50)->getSExtValue()); 113 EXPECT_EQ(206U, ConstantInt::getSigned(Int8Ty, -50)->getZExtValue()); 114 115 // Overflow is handled by truncation. 116 EXPECT_EQ(0x3b, ConstantInt::get(Int8Ty, 0x13b)->getSExtValue()); 117 } 118 119 TEST(ConstantsTest, FP128Test) { 120 LLVMContext Context; 121 Type *FP128Ty = Type::getFP128Ty(Context); 122 123 IntegerType *Int128Ty = Type::getIntNTy(Context, 128); 124 Constant *Zero128 = Constant::getNullValue(Int128Ty); 125 Constant *X = ConstantExpr::getUIToFP(Zero128, FP128Ty); 126 EXPECT_TRUE(isa<ConstantFP>(X)); 127 } 128 129 TEST(ConstantsTest, PointerCast) { 130 LLVMContext C; 131 Type *Int8PtrTy = Type::getInt8PtrTy(C); 132 Type *Int32PtrTy = Type::getInt32PtrTy(C); 133 Type *Int64Ty = Type::getInt64Ty(C); 134 VectorType *Int8PtrVecTy = FixedVectorType::get(Int8PtrTy, 4); 135 VectorType *Int32PtrVecTy = FixedVectorType::get(Int32PtrTy, 4); 136 VectorType *Int64VecTy = FixedVectorType::get(Int64Ty, 4); 137 138 // ptrtoint i8* to i64 139 EXPECT_EQ( 140 Constant::getNullValue(Int64Ty), 141 ConstantExpr::getPointerCast(Constant::getNullValue(Int8PtrTy), Int64Ty)); 142 143 // bitcast i8* to i32* 144 EXPECT_EQ(Constant::getNullValue(Int32PtrTy), 145 ConstantExpr::getPointerCast(Constant::getNullValue(Int8PtrTy), 146 Int32PtrTy)); 147 148 // ptrtoint <4 x i8*> to <4 x i64> 149 EXPECT_EQ(Constant::getNullValue(Int64VecTy), 150 ConstantExpr::getPointerCast(Constant::getNullValue(Int8PtrVecTy), 151 Int64VecTy)); 152 153 // bitcast <4 x i8*> to <4 x i32*> 154 EXPECT_EQ(Constant::getNullValue(Int32PtrVecTy), 155 ConstantExpr::getPointerCast(Constant::getNullValue(Int8PtrVecTy), 156 Int32PtrVecTy)); 157 158 Type *Int32Ptr1Ty = Type::getInt32PtrTy(C, 1); 159 ConstantInt *K = ConstantInt::get(Type::getInt64Ty(C), 1234); 160 161 // Make sure that addrspacecast of inttoptr is not folded away. 162 EXPECT_NE(K, ConstantExpr::getAddrSpaceCast( 163 ConstantExpr::getIntToPtr(K, Int32PtrTy), Int32Ptr1Ty)); 164 EXPECT_NE(K, ConstantExpr::getAddrSpaceCast( 165 ConstantExpr::getIntToPtr(K, Int32Ptr1Ty), Int32PtrTy)); 166 167 Constant *NullInt32Ptr0 = Constant::getNullValue(Int32PtrTy); 168 Constant *NullInt32Ptr1 = Constant::getNullValue(Int32Ptr1Ty); 169 170 // Make sure that addrspacecast of null is not folded away. 171 EXPECT_NE(Constant::getNullValue(Int32PtrTy), 172 ConstantExpr::getAddrSpaceCast(NullInt32Ptr0, Int32Ptr1Ty)); 173 174 EXPECT_NE(Constant::getNullValue(Int32Ptr1Ty), 175 ConstantExpr::getAddrSpaceCast(NullInt32Ptr1, Int32PtrTy)); 176 } 177 178 #define CHECK(x, y) \ 179 { \ 180 std::string __s; \ 181 raw_string_ostream __o(__s); \ 182 Instruction *__I = cast<ConstantExpr>(x)->getAsInstruction(); \ 183 __I->print(__o); \ 184 __I->deleteValue(); \ 185 __o.flush(); \ 186 EXPECT_EQ(std::string(" <badref> = " y), __s); \ 187 } 188 189 TEST(ConstantsTest, AsInstructionsTest) { 190 LLVMContext Context; 191 std::unique_ptr<Module> M(new Module("MyModule", Context)); 192 193 Type *Int64Ty = Type::getInt64Ty(Context); 194 Type *Int32Ty = Type::getInt32Ty(Context); 195 Type *Int16Ty = Type::getInt16Ty(Context); 196 Type *Int1Ty = Type::getInt1Ty(Context); 197 Type *FloatTy = Type::getFloatTy(Context); 198 Type *DoubleTy = Type::getDoubleTy(Context); 199 200 Constant *Global = 201 M->getOrInsertGlobal("dummy", PointerType::getUnqual(Int32Ty)); 202 Constant *Global2 = 203 M->getOrInsertGlobal("dummy2", PointerType::getUnqual(Int32Ty)); 204 205 Constant *P0 = ConstantExpr::getPtrToInt(Global, Int32Ty); 206 Constant *P1 = ConstantExpr::getUIToFP(P0, FloatTy); 207 Constant *P2 = ConstantExpr::getUIToFP(P0, DoubleTy); 208 Constant *P3 = ConstantExpr::getTrunc(P0, Int1Ty); 209 Constant *P4 = ConstantExpr::getPtrToInt(Global2, Int32Ty); 210 Constant *P5 = ConstantExpr::getUIToFP(P4, FloatTy); 211 Constant *P6 = ConstantExpr::getBitCast(P4, FixedVectorType::get(Int16Ty, 2)); 212 213 Constant *One = ConstantInt::get(Int32Ty, 1); 214 Constant *Two = ConstantInt::get(Int64Ty, 2); 215 Constant *Big = ConstantInt::get(Context, APInt{256, uint64_t(-1), true}); 216 Constant *Elt = ConstantInt::get(Int16Ty, 2015); 217 Constant *Poison16 = PoisonValue::get(Int16Ty); 218 Constant *Undef64 = UndefValue::get(Int64Ty); 219 Constant *PoisonV16 = PoisonValue::get(P6->getType()); 220 221 #define P0STR "ptrtoint (i32** @dummy to i32)" 222 #define P1STR "uitofp (i32 ptrtoint (i32** @dummy to i32) to float)" 223 #define P2STR "uitofp (i32 ptrtoint (i32** @dummy to i32) to double)" 224 #define P3STR "ptrtoint (i32** @dummy to i1)" 225 #define P4STR "ptrtoint (i32** @dummy2 to i32)" 226 #define P5STR "uitofp (i32 ptrtoint (i32** @dummy2 to i32) to float)" 227 #define P6STR "bitcast (i32 ptrtoint (i32** @dummy2 to i32) to <2 x i16>)" 228 229 CHECK(ConstantExpr::getNeg(P0), "sub i32 0, " P0STR); 230 CHECK(ConstantExpr::getFNeg(P1), "fneg float " P1STR); 231 CHECK(ConstantExpr::getNot(P0), "xor i32 " P0STR ", -1"); 232 CHECK(ConstantExpr::getAdd(P0, P0), "add i32 " P0STR ", " P0STR); 233 CHECK(ConstantExpr::getAdd(P0, P0, false, true), 234 "add nsw i32 " P0STR ", " P0STR); 235 CHECK(ConstantExpr::getAdd(P0, P0, true, true), 236 "add nuw nsw i32 " P0STR ", " P0STR); 237 CHECK(ConstantExpr::getFAdd(P1, P1), "fadd float " P1STR ", " P1STR); 238 CHECK(ConstantExpr::getSub(P0, P0), "sub i32 " P0STR ", " P0STR); 239 CHECK(ConstantExpr::getFSub(P1, P1), "fsub float " P1STR ", " P1STR); 240 CHECK(ConstantExpr::getMul(P0, P0), "mul i32 " P0STR ", " P0STR); 241 CHECK(ConstantExpr::getFMul(P1, P1), "fmul float " P1STR ", " P1STR); 242 CHECK(ConstantExpr::getUDiv(P0, P0), "udiv i32 " P0STR ", " P0STR); 243 CHECK(ConstantExpr::getSDiv(P0, P0), "sdiv i32 " P0STR ", " P0STR); 244 CHECK(ConstantExpr::getFDiv(P1, P1), "fdiv float " P1STR ", " P1STR); 245 CHECK(ConstantExpr::getURem(P0, P0), "urem i32 " P0STR ", " P0STR); 246 CHECK(ConstantExpr::getSRem(P0, P0), "srem i32 " P0STR ", " P0STR); 247 CHECK(ConstantExpr::getFRem(P1, P1), "frem float " P1STR ", " P1STR); 248 CHECK(ConstantExpr::getAnd(P0, P0), "and i32 " P0STR ", " P0STR); 249 CHECK(ConstantExpr::getOr(P0, P0), "or i32 " P0STR ", " P0STR); 250 CHECK(ConstantExpr::getXor(P0, P0), "xor i32 " P0STR ", " P0STR); 251 CHECK(ConstantExpr::getShl(P0, P0), "shl i32 " P0STR ", " P0STR); 252 CHECK(ConstantExpr::getShl(P0, P0, true), "shl nuw i32 " P0STR ", " P0STR); 253 CHECK(ConstantExpr::getShl(P0, P0, false, true), 254 "shl nsw i32 " P0STR ", " P0STR); 255 CHECK(ConstantExpr::getLShr(P0, P0, false), "lshr i32 " P0STR ", " P0STR); 256 CHECK(ConstantExpr::getLShr(P0, P0, true), 257 "lshr exact i32 " P0STR ", " P0STR); 258 CHECK(ConstantExpr::getAShr(P0, P0, false), "ashr i32 " P0STR ", " P0STR); 259 CHECK(ConstantExpr::getAShr(P0, P0, true), 260 "ashr exact i32 " P0STR ", " P0STR); 261 262 CHECK(ConstantExpr::getSExt(P0, Int64Ty), "sext i32 " P0STR " to i64"); 263 CHECK(ConstantExpr::getZExt(P0, Int64Ty), "zext i32 " P0STR " to i64"); 264 CHECK(ConstantExpr::getFPTrunc(P2, FloatTy), 265 "fptrunc double " P2STR " to float"); 266 CHECK(ConstantExpr::getFPExtend(P1, DoubleTy), 267 "fpext float " P1STR " to double"); 268 269 CHECK(ConstantExpr::getExactUDiv(P0, P0), "udiv exact i32 " P0STR ", " P0STR); 270 271 CHECK(ConstantExpr::getSelect(P3, P0, P4), 272 "select i1 " P3STR ", i32 " P0STR ", i32 " P4STR); 273 CHECK(ConstantExpr::getICmp(CmpInst::ICMP_EQ, P0, P4), 274 "icmp eq i32 " P0STR ", " P4STR); 275 CHECK(ConstantExpr::getFCmp(CmpInst::FCMP_ULT, P1, P5), 276 "fcmp ult float " P1STR ", " P5STR); 277 278 std::vector<Constant *> V; 279 V.push_back(One); 280 // FIXME: getGetElementPtr() actually creates an inbounds ConstantGEP, 281 // not a normal one! 282 // CHECK(ConstantExpr::getGetElementPtr(Global, V, false), 283 // "getelementptr i32*, i32** @dummy, i32 1"); 284 CHECK(ConstantExpr::getInBoundsGetElementPtr(PointerType::getUnqual(Int32Ty), 285 Global, V), 286 "getelementptr inbounds i32*, i32** @dummy, i32 1"); 287 288 CHECK(ConstantExpr::getExtractElement(P6, One), 289 "extractelement <2 x i16> " P6STR ", i32 1"); 290 291 EXPECT_EQ(Poison16, ConstantExpr::getExtractElement(P6, Two)); 292 EXPECT_EQ(Poison16, ConstantExpr::getExtractElement(P6, Big)); 293 EXPECT_EQ(Poison16, ConstantExpr::getExtractElement(P6, Undef64)); 294 295 EXPECT_EQ(Elt, ConstantExpr::getExtractElement( 296 ConstantExpr::getInsertElement(P6, Elt, One), One)); 297 EXPECT_EQ(PoisonV16, ConstantExpr::getInsertElement(P6, Elt, Two)); 298 EXPECT_EQ(PoisonV16, ConstantExpr::getInsertElement(P6, Elt, Big)); 299 EXPECT_EQ(PoisonV16, ConstantExpr::getInsertElement(P6, Elt, Undef64)); 300 } 301 302 #ifdef GTEST_HAS_DEATH_TEST 303 #ifndef NDEBUG 304 TEST(ConstantsTest, ReplaceWithConstantTest) { 305 LLVMContext Context; 306 std::unique_ptr<Module> M(new Module("MyModule", Context)); 307 308 Type *Int32Ty = Type::getInt32Ty(Context); 309 Constant *One = ConstantInt::get(Int32Ty, 1); 310 311 Constant *Global = 312 M->getOrInsertGlobal("dummy", PointerType::getUnqual(Int32Ty)); 313 Constant *GEP = ConstantExpr::getGetElementPtr( 314 PointerType::getUnqual(Int32Ty), Global, One); 315 EXPECT_DEATH(Global->replaceAllUsesWith(GEP), 316 "this->replaceAllUsesWith\\(expr\\(this\\)\\) is NOT valid!"); 317 } 318 319 #endif 320 #endif 321 322 #undef CHECK 323 324 TEST(ConstantsTest, ConstantArrayReplaceWithConstant) { 325 LLVMContext Context; 326 std::unique_ptr<Module> M(new Module("MyModule", Context)); 327 328 Type *IntTy = Type::getInt8Ty(Context); 329 ArrayType *ArrayTy = ArrayType::get(IntTy, 2); 330 Constant *A01Vals[2] = {ConstantInt::get(IntTy, 0), 331 ConstantInt::get(IntTy, 1)}; 332 Constant *A01 = ConstantArray::get(ArrayTy, A01Vals); 333 334 Constant *Global = new GlobalVariable(*M, IntTy, false, 335 GlobalValue::ExternalLinkage, nullptr); 336 Constant *GlobalInt = ConstantExpr::getPtrToInt(Global, IntTy); 337 Constant *A0GVals[2] = {ConstantInt::get(IntTy, 0), GlobalInt}; 338 Constant *A0G = ConstantArray::get(ArrayTy, A0GVals); 339 ASSERT_NE(A01, A0G); 340 341 GlobalVariable *RefArray = 342 new GlobalVariable(*M, ArrayTy, false, GlobalValue::ExternalLinkage, A0G); 343 ASSERT_EQ(A0G, RefArray->getInitializer()); 344 345 GlobalInt->replaceAllUsesWith(ConstantInt::get(IntTy, 1)); 346 ASSERT_EQ(A01, RefArray->getInitializer()); 347 } 348 349 TEST(ConstantsTest, ConstantExprReplaceWithConstant) { 350 LLVMContext Context; 351 std::unique_ptr<Module> M(new Module("MyModule", Context)); 352 353 Type *IntTy = Type::getInt8Ty(Context); 354 Constant *G1 = new GlobalVariable(*M, IntTy, false, 355 GlobalValue::ExternalLinkage, nullptr); 356 Constant *G2 = new GlobalVariable(*M, IntTy, false, 357 GlobalValue::ExternalLinkage, nullptr); 358 ASSERT_NE(G1, G2); 359 360 Constant *Int1 = ConstantExpr::getPtrToInt(G1, IntTy); 361 Constant *Int2 = ConstantExpr::getPtrToInt(G2, IntTy); 362 ASSERT_NE(Int1, Int2); 363 364 GlobalVariable *Ref = 365 new GlobalVariable(*M, IntTy, false, GlobalValue::ExternalLinkage, Int1); 366 ASSERT_EQ(Int1, Ref->getInitializer()); 367 368 G1->replaceAllUsesWith(G2); 369 ASSERT_EQ(Int2, Ref->getInitializer()); 370 } 371 372 TEST(ConstantsTest, GEPReplaceWithConstant) { 373 LLVMContext Context; 374 std::unique_ptr<Module> M(new Module("MyModule", Context)); 375 376 Type *IntTy = Type::getInt32Ty(Context); 377 Type *PtrTy = PointerType::get(IntTy, 0); 378 auto *C1 = ConstantInt::get(IntTy, 1); 379 auto *Placeholder = new GlobalVariable( 380 *M, IntTy, false, GlobalValue::ExternalWeakLinkage, nullptr); 381 auto *GEP = ConstantExpr::getGetElementPtr(IntTy, Placeholder, C1); 382 ASSERT_EQ(GEP->getOperand(0), Placeholder); 383 384 auto *Ref = 385 new GlobalVariable(*M, PtrTy, false, GlobalValue::ExternalLinkage, GEP); 386 ASSERT_EQ(GEP, Ref->getInitializer()); 387 388 auto *Global = new GlobalVariable(*M, IntTy, false, 389 GlobalValue::ExternalLinkage, nullptr); 390 auto *Alias = GlobalAlias::create(IntTy, 0, GlobalValue::ExternalLinkage, 391 "alias", Global, M.get()); 392 Placeholder->replaceAllUsesWith(Alias); 393 ASSERT_EQ(GEP, Ref->getInitializer()); 394 ASSERT_EQ(GEP->getOperand(0), Alias); 395 } 396 397 TEST(ConstantsTest, AliasCAPI) { 398 LLVMContext Context; 399 SMDiagnostic Error; 400 std::unique_ptr<Module> M = 401 parseAssemblyString("@g = global i32 42", Error, Context); 402 GlobalVariable *G = M->getGlobalVariable("g"); 403 Type *I16Ty = Type::getInt16Ty(Context); 404 Type *I16PTy = PointerType::get(I16Ty, 0); 405 Constant *Aliasee = ConstantExpr::getBitCast(G, I16PTy); 406 LLVMValueRef AliasRef = 407 LLVMAddAlias2(wrap(M.get()), wrap(I16Ty), 0, wrap(Aliasee), "a"); 408 ASSERT_EQ(unwrap<GlobalAlias>(AliasRef)->getAliasee(), Aliasee); 409 } 410 411 static std::string getNameOfType(Type *T) { 412 std::string S; 413 raw_string_ostream RSOS(S); 414 T->print(RSOS); 415 return S; 416 } 417 418 TEST(ConstantsTest, BuildConstantDataArrays) { 419 LLVMContext Context; 420 421 for (Type *T : {Type::getInt8Ty(Context), Type::getInt16Ty(Context), 422 Type::getInt32Ty(Context), Type::getInt64Ty(Context)}) { 423 ArrayType *ArrayTy = ArrayType::get(T, 2); 424 Constant *Vals[] = {ConstantInt::get(T, 0), ConstantInt::get(T, 1)}; 425 Constant *CA = ConstantArray::get(ArrayTy, Vals); 426 ASSERT_TRUE(isa<ConstantDataArray>(CA)) << " T = " << getNameOfType(T); 427 auto *CDA = cast<ConstantDataArray>(CA); 428 Constant *CA2 = ConstantDataArray::getRaw( 429 CDA->getRawDataValues(), CDA->getNumElements(), CDA->getElementType()); 430 ASSERT_TRUE(CA == CA2) << " T = " << getNameOfType(T); 431 } 432 433 for (Type *T : {Type::getHalfTy(Context), Type::getBFloatTy(Context), 434 Type::getFloatTy(Context), Type::getDoubleTy(Context)}) { 435 ArrayType *ArrayTy = ArrayType::get(T, 2); 436 Constant *Vals[] = {ConstantFP::get(T, 0), ConstantFP::get(T, 1)}; 437 Constant *CA = ConstantArray::get(ArrayTy, Vals); 438 ASSERT_TRUE(isa<ConstantDataArray>(CA)) << " T = " << getNameOfType(T); 439 auto *CDA = cast<ConstantDataArray>(CA); 440 Constant *CA2 = ConstantDataArray::getRaw( 441 CDA->getRawDataValues(), CDA->getNumElements(), CDA->getElementType()); 442 ASSERT_TRUE(CA == CA2) << " T = " << getNameOfType(T); 443 } 444 } 445 446 TEST(ConstantsTest, BuildConstantDataVectors) { 447 LLVMContext Context; 448 449 for (Type *T : {Type::getInt8Ty(Context), Type::getInt16Ty(Context), 450 Type::getInt32Ty(Context), Type::getInt64Ty(Context)}) { 451 Constant *Vals[] = {ConstantInt::get(T, 0), ConstantInt::get(T, 1)}; 452 Constant *CV = ConstantVector::get(Vals); 453 ASSERT_TRUE(isa<ConstantDataVector>(CV)) << " T = " << getNameOfType(T); 454 auto *CDV = cast<ConstantDataVector>(CV); 455 Constant *CV2 = ConstantDataVector::getRaw( 456 CDV->getRawDataValues(), CDV->getNumElements(), CDV->getElementType()); 457 ASSERT_TRUE(CV == CV2) << " T = " << getNameOfType(T); 458 } 459 460 for (Type *T : {Type::getHalfTy(Context), Type::getBFloatTy(Context), 461 Type::getFloatTy(Context), Type::getDoubleTy(Context)}) { 462 Constant *Vals[] = {ConstantFP::get(T, 0), ConstantFP::get(T, 1)}; 463 Constant *CV = ConstantVector::get(Vals); 464 ASSERT_TRUE(isa<ConstantDataVector>(CV)) << " T = " << getNameOfType(T); 465 auto *CDV = cast<ConstantDataVector>(CV); 466 Constant *CV2 = ConstantDataVector::getRaw( 467 CDV->getRawDataValues(), CDV->getNumElements(), CDV->getElementType()); 468 ASSERT_TRUE(CV == CV2) << " T = " << getNameOfType(T); 469 } 470 } 471 472 TEST(ConstantsTest, BitcastToGEP) { 473 LLVMContext Context; 474 std::unique_ptr<Module> M(new Module("MyModule", Context)); 475 476 auto *i32 = Type::getInt32Ty(Context); 477 auto *U = StructType::create(Context, "Unsized"); 478 Type *EltTys[] = {i32, U}; 479 auto *S = StructType::create(EltTys); 480 481 auto *G = 482 new GlobalVariable(*M, S, false, GlobalValue::ExternalLinkage, nullptr); 483 auto *PtrTy = PointerType::get(i32, 0); 484 auto *C = ConstantExpr::getBitCast(G, PtrTy); 485 if (Context.supportsTypedPointers()) { 486 EXPECT_EQ(cast<ConstantExpr>(C)->getOpcode(), Instruction::BitCast); 487 } else { 488 /* With opaque pointers, no cast is necessary. */ 489 EXPECT_EQ(C, G); 490 } 491 } 492 493 bool foldFuncPtrAndConstToNull(LLVMContext &Context, Module *TheModule, 494 uint64_t AndValue, 495 MaybeAlign FunctionAlign = llvm::None) { 496 Type *VoidType(Type::getVoidTy(Context)); 497 FunctionType *FuncType(FunctionType::get(VoidType, false)); 498 Function *Func( 499 Function::Create(FuncType, GlobalValue::ExternalLinkage, "", TheModule)); 500 501 if (FunctionAlign) 502 Func->setAlignment(*FunctionAlign); 503 504 IntegerType *ConstantIntType(Type::getInt32Ty(Context)); 505 ConstantInt *TheConstant(ConstantInt::get(ConstantIntType, AndValue)); 506 507 Constant *TheConstantExpr(ConstantExpr::getPtrToInt(Func, ConstantIntType)); 508 509 bool Result = 510 ConstantExpr::get(Instruction::And, TheConstantExpr, TheConstant) 511 ->isNullValue(); 512 513 if (!TheModule) { 514 // If the Module exists then it will delete the Function. 515 delete Func; 516 } 517 518 return Result; 519 } 520 521 TEST(ConstantsTest, FoldFunctionPtrAlignUnknownAnd2) { 522 LLVMContext Context; 523 Module TheModule("TestModule", Context); 524 // When the DataLayout doesn't specify a function pointer alignment we 525 // assume in this case that it is 4 byte aligned. This is a bug but we can't 526 // fix it directly because it causes a code size regression on X86. 527 // FIXME: This test should be changed once existing targets have 528 // appropriate defaults. See associated FIXME in ConstantFoldBinaryInstruction 529 ASSERT_TRUE(foldFuncPtrAndConstToNull(Context, &TheModule, 2)); 530 } 531 532 TEST(ConstantsTest, DontFoldFunctionPtrAlignUnknownAnd4) { 533 LLVMContext Context; 534 Module TheModule("TestModule", Context); 535 ASSERT_FALSE(foldFuncPtrAndConstToNull(Context, &TheModule, 4)); 536 } 537 538 TEST(ConstantsTest, FoldFunctionPtrAlign4) { 539 LLVMContext Context; 540 Module TheModule("TestModule", Context); 541 const char *AlignmentStrings[] = {"Fi32", "Fn32"}; 542 543 for (unsigned AndValue = 1; AndValue <= 2; ++AndValue) { 544 for (const char *AlignmentString : AlignmentStrings) { 545 TheModule.setDataLayout(AlignmentString); 546 ASSERT_TRUE(foldFuncPtrAndConstToNull(Context, &TheModule, AndValue)); 547 } 548 } 549 } 550 551 TEST(ConstantsTest, DontFoldFunctionPtrAlign1) { 552 LLVMContext Context; 553 Module TheModule("TestModule", Context); 554 const char *AlignmentStrings[] = {"Fi8", "Fn8"}; 555 556 for (const char *AlignmentString : AlignmentStrings) { 557 TheModule.setDataLayout(AlignmentString); 558 ASSERT_FALSE(foldFuncPtrAndConstToNull(Context, &TheModule, 2)); 559 } 560 } 561 562 TEST(ConstantsTest, FoldFunctionAlign4PtrAlignMultiple) { 563 LLVMContext Context; 564 Module TheModule("TestModule", Context); 565 TheModule.setDataLayout("Fn8"); 566 ASSERT_TRUE(foldFuncPtrAndConstToNull(Context, &TheModule, 2, Align(4))); 567 } 568 569 TEST(ConstantsTest, DontFoldFunctionAlign4PtrAlignIndependent) { 570 LLVMContext Context; 571 Module TheModule("TestModule", Context); 572 TheModule.setDataLayout("Fi8"); 573 ASSERT_FALSE(foldFuncPtrAndConstToNull(Context, &TheModule, 2, Align(4))); 574 } 575 576 TEST(ConstantsTest, DontFoldFunctionPtrIfNoModule) { 577 LLVMContext Context; 578 // Even though the function is explicitly 4 byte aligned, in the absence of a 579 // DataLayout we can't assume that the function pointer is aligned. 580 ASSERT_FALSE(foldFuncPtrAndConstToNull(Context, nullptr, 2, Align(4))); 581 } 582 583 TEST(ConstantsTest, FoldGlobalVariablePtr) { 584 LLVMContext Context; 585 586 IntegerType *IntType(Type::getInt32Ty(Context)); 587 588 std::unique_ptr<GlobalVariable> Global( 589 new GlobalVariable(IntType, true, GlobalValue::ExternalLinkage)); 590 591 Global->setAlignment(Align(4)); 592 593 ConstantInt *TheConstant(ConstantInt::get(IntType, 2)); 594 595 Constant *TheConstantExpr(ConstantExpr::getPtrToInt(Global.get(), IntType)); 596 597 ASSERT_TRUE(ConstantExpr::get(Instruction::And, TheConstantExpr, TheConstant) 598 ->isNullValue()); 599 } 600 601 // Check that containsUndefOrPoisonElement and containsPoisonElement is working 602 // great 603 604 TEST(ConstantsTest, containsUndefElemTest) { 605 LLVMContext Context; 606 607 Type *Int32Ty = Type::getInt32Ty(Context); 608 Constant *CU = UndefValue::get(Int32Ty); 609 Constant *CP = PoisonValue::get(Int32Ty); 610 Constant *C1 = ConstantInt::get(Int32Ty, 1); 611 Constant *C2 = ConstantInt::get(Int32Ty, 2); 612 613 { 614 Constant *V1 = ConstantVector::get({C1, C2}); 615 EXPECT_FALSE(V1->containsUndefOrPoisonElement()); 616 EXPECT_FALSE(V1->containsPoisonElement()); 617 } 618 619 { 620 Constant *V2 = ConstantVector::get({C1, CU}); 621 EXPECT_TRUE(V2->containsUndefOrPoisonElement()); 622 EXPECT_FALSE(V2->containsPoisonElement()); 623 } 624 625 { 626 Constant *V3 = ConstantVector::get({C1, CP}); 627 EXPECT_TRUE(V3->containsUndefOrPoisonElement()); 628 EXPECT_TRUE(V3->containsPoisonElement()); 629 } 630 631 { 632 Constant *V4 = ConstantVector::get({CU, CP}); 633 EXPECT_TRUE(V4->containsUndefOrPoisonElement()); 634 EXPECT_TRUE(V4->containsPoisonElement()); 635 } 636 } 637 638 // Check that undefined elements in vector constants are matched 639 // correctly for both integer and floating-point types. Just don't 640 // crash on vectors of pointers (could be handled?). 641 642 TEST(ConstantsTest, isElementWiseEqual) { 643 LLVMContext Context; 644 645 Type *Int32Ty = Type::getInt32Ty(Context); 646 Constant *CU = UndefValue::get(Int32Ty); 647 Constant *C1 = ConstantInt::get(Int32Ty, 1); 648 Constant *C2 = ConstantInt::get(Int32Ty, 2); 649 650 Constant *C1211 = ConstantVector::get({C1, C2, C1, C1}); 651 Constant *C12U1 = ConstantVector::get({C1, C2, CU, C1}); 652 Constant *C12U2 = ConstantVector::get({C1, C2, CU, C2}); 653 Constant *C12U21 = ConstantVector::get({C1, C2, CU, C2, C1}); 654 655 EXPECT_TRUE(C1211->isElementWiseEqual(C12U1)); 656 EXPECT_TRUE(C12U1->isElementWiseEqual(C1211)); 657 EXPECT_FALSE(C12U2->isElementWiseEqual(C12U1)); 658 EXPECT_FALSE(C12U1->isElementWiseEqual(C12U2)); 659 EXPECT_FALSE(C12U21->isElementWiseEqual(C12U2)); 660 661 Type *FltTy = Type::getFloatTy(Context); 662 Constant *CFU = UndefValue::get(FltTy); 663 Constant *CF1 = ConstantFP::get(FltTy, 1.0); 664 Constant *CF2 = ConstantFP::get(FltTy, 2.0); 665 666 Constant *CF1211 = ConstantVector::get({CF1, CF2, CF1, CF1}); 667 Constant *CF12U1 = ConstantVector::get({CF1, CF2, CFU, CF1}); 668 Constant *CF12U2 = ConstantVector::get({CF1, CF2, CFU, CF2}); 669 Constant *CFUU1U = ConstantVector::get({CFU, CFU, CF1, CFU}); 670 671 EXPECT_TRUE(CF1211->isElementWiseEqual(CF12U1)); 672 EXPECT_TRUE(CF12U1->isElementWiseEqual(CF1211)); 673 EXPECT_TRUE(CFUU1U->isElementWiseEqual(CF12U1)); 674 EXPECT_FALSE(CF12U2->isElementWiseEqual(CF12U1)); 675 EXPECT_FALSE(CF12U1->isElementWiseEqual(CF12U2)); 676 677 PointerType *PtrTy = Type::getInt8PtrTy(Context); 678 Constant *CPU = UndefValue::get(PtrTy); 679 Constant *CP0 = ConstantPointerNull::get(PtrTy); 680 681 Constant *CP0000 = ConstantVector::get({CP0, CP0, CP0, CP0}); 682 Constant *CP00U0 = ConstantVector::get({CP0, CP0, CPU, CP0}); 683 Constant *CP00U = ConstantVector::get({CP0, CP0, CPU}); 684 685 EXPECT_FALSE(CP0000->isElementWiseEqual(CP00U0)); 686 EXPECT_FALSE(CP00U0->isElementWiseEqual(CP0000)); 687 EXPECT_FALSE(CP0000->isElementWiseEqual(CP00U)); 688 EXPECT_FALSE(CP00U->isElementWiseEqual(CP00U0)); 689 } 690 691 // Check that vector/aggregate constants correctly store undef and poison 692 // elements. 693 694 TEST(ConstantsTest, CheckElementWiseUndefPoison) { 695 LLVMContext Context; 696 697 Type *Int32Ty = Type::getInt32Ty(Context); 698 StructType *STy = StructType::get(Int32Ty, Int32Ty); 699 ArrayType *ATy = ArrayType::get(Int32Ty, 2); 700 Constant *CU = UndefValue::get(Int32Ty); 701 Constant *CP = PoisonValue::get(Int32Ty); 702 703 { 704 Constant *CUU = ConstantVector::get({CU, CU}); 705 Constant *CPP = ConstantVector::get({CP, CP}); 706 Constant *CUP = ConstantVector::get({CU, CP}); 707 Constant *CPU = ConstantVector::get({CP, CU}); 708 EXPECT_EQ(CUU, UndefValue::get(CUU->getType())); 709 EXPECT_EQ(CPP, PoisonValue::get(CPP->getType())); 710 EXPECT_NE(CUP, UndefValue::get(CUP->getType())); 711 EXPECT_NE(CPU, UndefValue::get(CPU->getType())); 712 } 713 714 { 715 Constant *CUU = ConstantStruct::get(STy, {CU, CU}); 716 Constant *CPP = ConstantStruct::get(STy, {CP, CP}); 717 Constant *CUP = ConstantStruct::get(STy, {CU, CP}); 718 Constant *CPU = ConstantStruct::get(STy, {CP, CU}); 719 EXPECT_EQ(CUU, UndefValue::get(CUU->getType())); 720 EXPECT_EQ(CPP, PoisonValue::get(CPP->getType())); 721 EXPECT_NE(CUP, UndefValue::get(CUP->getType())); 722 EXPECT_NE(CPU, UndefValue::get(CPU->getType())); 723 } 724 725 { 726 Constant *CUU = ConstantArray::get(ATy, {CU, CU}); 727 Constant *CPP = ConstantArray::get(ATy, {CP, CP}); 728 Constant *CUP = ConstantArray::get(ATy, {CU, CP}); 729 Constant *CPU = ConstantArray::get(ATy, {CP, CU}); 730 EXPECT_EQ(CUU, UndefValue::get(CUU->getType())); 731 EXPECT_EQ(CPP, PoisonValue::get(CPP->getType())); 732 EXPECT_NE(CUP, UndefValue::get(CUP->getType())); 733 EXPECT_NE(CPU, UndefValue::get(CPU->getType())); 734 } 735 } 736 737 TEST(ConstantsTest, GetSplatValueRoundTrip) { 738 LLVMContext Context; 739 740 Type *FloatTy = Type::getFloatTy(Context); 741 Type *Int32Ty = Type::getInt32Ty(Context); 742 Type *Int8Ty = Type::getInt8Ty(Context); 743 744 for (unsigned Min : {1, 2, 8}) { 745 auto ScalableEC = ElementCount::getScalable(Min); 746 auto FixedEC = ElementCount::getFixed(Min); 747 748 for (auto EC : {ScalableEC, FixedEC}) { 749 for (auto *Ty : {FloatTy, Int32Ty, Int8Ty}) { 750 Constant *Zero = Constant::getNullValue(Ty); 751 Constant *One = Constant::getAllOnesValue(Ty); 752 753 for (auto *C : {Zero, One}) { 754 Constant *Splat = ConstantVector::getSplat(EC, C); 755 ASSERT_NE(nullptr, Splat); 756 757 Constant *SplatVal = Splat->getSplatValue(); 758 EXPECT_NE(nullptr, SplatVal); 759 EXPECT_EQ(SplatVal, C); 760 } 761 } 762 } 763 } 764 } 765 766 } // end anonymous namespace 767 } // end namespace llvm 768