1 //===-- ConstantFolding.cpp - Fold instructions into constants ------------===// 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 // This file defines routines for folding instructions into constants. 10 // 11 // Also, to supplement the basic IR ConstantExpr simplifications, 12 // this file defines some additional folding routines that can make use of 13 // DataLayout information. These functions cannot go in IR due to library 14 // dependency issues. 15 // 16 //===----------------------------------------------------------------------===// 17 18 #include "llvm/Analysis/ConstantFolding.h" 19 #include "llvm/ADT/APFloat.h" 20 #include "llvm/ADT/APInt.h" 21 #include "llvm/ADT/APSInt.h" 22 #include "llvm/ADT/ArrayRef.h" 23 #include "llvm/ADT/DenseMap.h" 24 #include "llvm/ADT/STLExtras.h" 25 #include "llvm/ADT/SmallVector.h" 26 #include "llvm/ADT/StringRef.h" 27 #include "llvm/Analysis/TargetFolder.h" 28 #include "llvm/Analysis/TargetLibraryInfo.h" 29 #include "llvm/Analysis/ValueTracking.h" 30 #include "llvm/Analysis/VectorUtils.h" 31 #include "llvm/Config/config.h" 32 #include "llvm/IR/Constant.h" 33 #include "llvm/IR/Constants.h" 34 #include "llvm/IR/DataLayout.h" 35 #include "llvm/IR/DerivedTypes.h" 36 #include "llvm/IR/Function.h" 37 #include "llvm/IR/GlobalValue.h" 38 #include "llvm/IR/GlobalVariable.h" 39 #include "llvm/IR/InstrTypes.h" 40 #include "llvm/IR/Instruction.h" 41 #include "llvm/IR/Instructions.h" 42 #include "llvm/IR/IntrinsicInst.h" 43 #include "llvm/IR/Intrinsics.h" 44 #include "llvm/IR/IntrinsicsAArch64.h" 45 #include "llvm/IR/IntrinsicsAMDGPU.h" 46 #include "llvm/IR/IntrinsicsARM.h" 47 #include "llvm/IR/IntrinsicsWebAssembly.h" 48 #include "llvm/IR/IntrinsicsX86.h" 49 #include "llvm/IR/Operator.h" 50 #include "llvm/IR/Type.h" 51 #include "llvm/IR/Value.h" 52 #include "llvm/Support/Casting.h" 53 #include "llvm/Support/ErrorHandling.h" 54 #include "llvm/Support/KnownBits.h" 55 #include "llvm/Support/MathExtras.h" 56 #include <cassert> 57 #include <cerrno> 58 #include <cfenv> 59 #include <cmath> 60 #include <cstddef> 61 #include <cstdint> 62 63 using namespace llvm; 64 65 namespace { 66 67 //===----------------------------------------------------------------------===// 68 // Constant Folding internal helper functions 69 //===----------------------------------------------------------------------===// 70 71 static Constant *foldConstVectorToAPInt(APInt &Result, Type *DestTy, 72 Constant *C, Type *SrcEltTy, 73 unsigned NumSrcElts, 74 const DataLayout &DL) { 75 // Now that we know that the input value is a vector of integers, just shift 76 // and insert them into our result. 77 unsigned BitShift = DL.getTypeSizeInBits(SrcEltTy); 78 for (unsigned i = 0; i != NumSrcElts; ++i) { 79 Constant *Element; 80 if (DL.isLittleEndian()) 81 Element = C->getAggregateElement(NumSrcElts - i - 1); 82 else 83 Element = C->getAggregateElement(i); 84 85 if (Element && isa<UndefValue>(Element)) { 86 Result <<= BitShift; 87 continue; 88 } 89 90 auto *ElementCI = dyn_cast_or_null<ConstantInt>(Element); 91 if (!ElementCI) 92 return ConstantExpr::getBitCast(C, DestTy); 93 94 Result <<= BitShift; 95 Result |= ElementCI->getValue().zextOrSelf(Result.getBitWidth()); 96 } 97 98 return nullptr; 99 } 100 101 /// Constant fold bitcast, symbolically evaluating it with DataLayout. 102 /// This always returns a non-null constant, but it may be a 103 /// ConstantExpr if unfoldable. 104 Constant *FoldBitCast(Constant *C, Type *DestTy, const DataLayout &DL) { 105 assert(CastInst::castIsValid(Instruction::BitCast, C, DestTy) && 106 "Invalid constantexpr bitcast!"); 107 108 // Catch the obvious splat cases. 109 if (C->isNullValue() && !DestTy->isX86_MMXTy() && !DestTy->isX86_AMXTy()) 110 return Constant::getNullValue(DestTy); 111 if (C->isAllOnesValue() && !DestTy->isX86_MMXTy() && !DestTy->isX86_AMXTy() && 112 !DestTy->isPtrOrPtrVectorTy()) // Don't get ones for ptr types! 113 return Constant::getAllOnesValue(DestTy); 114 115 if (auto *VTy = dyn_cast<VectorType>(C->getType())) { 116 // Handle a vector->scalar integer/fp cast. 117 if (isa<IntegerType>(DestTy) || DestTy->isFloatingPointTy()) { 118 unsigned NumSrcElts = cast<FixedVectorType>(VTy)->getNumElements(); 119 Type *SrcEltTy = VTy->getElementType(); 120 121 // If the vector is a vector of floating point, convert it to vector of int 122 // to simplify things. 123 if (SrcEltTy->isFloatingPointTy()) { 124 unsigned FPWidth = SrcEltTy->getPrimitiveSizeInBits(); 125 auto *SrcIVTy = FixedVectorType::get( 126 IntegerType::get(C->getContext(), FPWidth), NumSrcElts); 127 // Ask IR to do the conversion now that #elts line up. 128 C = ConstantExpr::getBitCast(C, SrcIVTy); 129 } 130 131 APInt Result(DL.getTypeSizeInBits(DestTy), 0); 132 if (Constant *CE = foldConstVectorToAPInt(Result, DestTy, C, 133 SrcEltTy, NumSrcElts, DL)) 134 return CE; 135 136 if (isa<IntegerType>(DestTy)) 137 return ConstantInt::get(DestTy, Result); 138 139 APFloat FP(DestTy->getFltSemantics(), Result); 140 return ConstantFP::get(DestTy->getContext(), FP); 141 } 142 } 143 144 // The code below only handles casts to vectors currently. 145 auto *DestVTy = dyn_cast<VectorType>(DestTy); 146 if (!DestVTy) 147 return ConstantExpr::getBitCast(C, DestTy); 148 149 // If this is a scalar -> vector cast, convert the input into a <1 x scalar> 150 // vector so the code below can handle it uniformly. 151 if (isa<ConstantFP>(C) || isa<ConstantInt>(C)) { 152 Constant *Ops = C; // don't take the address of C! 153 return FoldBitCast(ConstantVector::get(Ops), DestTy, DL); 154 } 155 156 // If this is a bitcast from constant vector -> vector, fold it. 157 if (!isa<ConstantDataVector>(C) && !isa<ConstantVector>(C)) 158 return ConstantExpr::getBitCast(C, DestTy); 159 160 // If the element types match, IR can fold it. 161 unsigned NumDstElt = cast<FixedVectorType>(DestVTy)->getNumElements(); 162 unsigned NumSrcElt = cast<FixedVectorType>(C->getType())->getNumElements(); 163 if (NumDstElt == NumSrcElt) 164 return ConstantExpr::getBitCast(C, DestTy); 165 166 Type *SrcEltTy = cast<VectorType>(C->getType())->getElementType(); 167 Type *DstEltTy = DestVTy->getElementType(); 168 169 // Otherwise, we're changing the number of elements in a vector, which 170 // requires endianness information to do the right thing. For example, 171 // bitcast (<2 x i64> <i64 0, i64 1> to <4 x i32>) 172 // folds to (little endian): 173 // <4 x i32> <i32 0, i32 0, i32 1, i32 0> 174 // and to (big endian): 175 // <4 x i32> <i32 0, i32 0, i32 0, i32 1> 176 177 // First thing is first. We only want to think about integer here, so if 178 // we have something in FP form, recast it as integer. 179 if (DstEltTy->isFloatingPointTy()) { 180 // Fold to an vector of integers with same size as our FP type. 181 unsigned FPWidth = DstEltTy->getPrimitiveSizeInBits(); 182 auto *DestIVTy = FixedVectorType::get( 183 IntegerType::get(C->getContext(), FPWidth), NumDstElt); 184 // Recursively handle this integer conversion, if possible. 185 C = FoldBitCast(C, DestIVTy, DL); 186 187 // Finally, IR can handle this now that #elts line up. 188 return ConstantExpr::getBitCast(C, DestTy); 189 } 190 191 // Okay, we know the destination is integer, if the input is FP, convert 192 // it to integer first. 193 if (SrcEltTy->isFloatingPointTy()) { 194 unsigned FPWidth = SrcEltTy->getPrimitiveSizeInBits(); 195 auto *SrcIVTy = FixedVectorType::get( 196 IntegerType::get(C->getContext(), FPWidth), NumSrcElt); 197 // Ask IR to do the conversion now that #elts line up. 198 C = ConstantExpr::getBitCast(C, SrcIVTy); 199 // If IR wasn't able to fold it, bail out. 200 if (!isa<ConstantVector>(C) && // FIXME: Remove ConstantVector. 201 !isa<ConstantDataVector>(C)) 202 return C; 203 } 204 205 // Now we know that the input and output vectors are both integer vectors 206 // of the same size, and that their #elements is not the same. Do the 207 // conversion here, which depends on whether the input or output has 208 // more elements. 209 bool isLittleEndian = DL.isLittleEndian(); 210 211 SmallVector<Constant*, 32> Result; 212 if (NumDstElt < NumSrcElt) { 213 // Handle: bitcast (<4 x i32> <i32 0, i32 1, i32 2, i32 3> to <2 x i64>) 214 Constant *Zero = Constant::getNullValue(DstEltTy); 215 unsigned Ratio = NumSrcElt/NumDstElt; 216 unsigned SrcBitSize = SrcEltTy->getPrimitiveSizeInBits(); 217 unsigned SrcElt = 0; 218 for (unsigned i = 0; i != NumDstElt; ++i) { 219 // Build each element of the result. 220 Constant *Elt = Zero; 221 unsigned ShiftAmt = isLittleEndian ? 0 : SrcBitSize*(Ratio-1); 222 for (unsigned j = 0; j != Ratio; ++j) { 223 Constant *Src = C->getAggregateElement(SrcElt++); 224 if (Src && isa<UndefValue>(Src)) 225 Src = Constant::getNullValue( 226 cast<VectorType>(C->getType())->getElementType()); 227 else 228 Src = dyn_cast_or_null<ConstantInt>(Src); 229 if (!Src) // Reject constantexpr elements. 230 return ConstantExpr::getBitCast(C, DestTy); 231 232 // Zero extend the element to the right size. 233 Src = ConstantExpr::getZExt(Src, Elt->getType()); 234 235 // Shift it to the right place, depending on endianness. 236 Src = ConstantExpr::getShl(Src, 237 ConstantInt::get(Src->getType(), ShiftAmt)); 238 ShiftAmt += isLittleEndian ? SrcBitSize : -SrcBitSize; 239 240 // Mix it in. 241 Elt = ConstantExpr::getOr(Elt, Src); 242 } 243 Result.push_back(Elt); 244 } 245 return ConstantVector::get(Result); 246 } 247 248 // Handle: bitcast (<2 x i64> <i64 0, i64 1> to <4 x i32>) 249 unsigned Ratio = NumDstElt/NumSrcElt; 250 unsigned DstBitSize = DL.getTypeSizeInBits(DstEltTy); 251 252 // Loop over each source value, expanding into multiple results. 253 for (unsigned i = 0; i != NumSrcElt; ++i) { 254 auto *Element = C->getAggregateElement(i); 255 256 if (!Element) // Reject constantexpr elements. 257 return ConstantExpr::getBitCast(C, DestTy); 258 259 if (isa<UndefValue>(Element)) { 260 // Correctly Propagate undef values. 261 Result.append(Ratio, UndefValue::get(DstEltTy)); 262 continue; 263 } 264 265 auto *Src = dyn_cast<ConstantInt>(Element); 266 if (!Src) 267 return ConstantExpr::getBitCast(C, DestTy); 268 269 unsigned ShiftAmt = isLittleEndian ? 0 : DstBitSize*(Ratio-1); 270 for (unsigned j = 0; j != Ratio; ++j) { 271 // Shift the piece of the value into the right place, depending on 272 // endianness. 273 Constant *Elt = ConstantExpr::getLShr(Src, 274 ConstantInt::get(Src->getType(), ShiftAmt)); 275 ShiftAmt += isLittleEndian ? DstBitSize : -DstBitSize; 276 277 // Truncate the element to an integer with the same pointer size and 278 // convert the element back to a pointer using a inttoptr. 279 if (DstEltTy->isPointerTy()) { 280 IntegerType *DstIntTy = Type::getIntNTy(C->getContext(), DstBitSize); 281 Constant *CE = ConstantExpr::getTrunc(Elt, DstIntTy); 282 Result.push_back(ConstantExpr::getIntToPtr(CE, DstEltTy)); 283 continue; 284 } 285 286 // Truncate and remember this piece. 287 Result.push_back(ConstantExpr::getTrunc(Elt, DstEltTy)); 288 } 289 } 290 291 return ConstantVector::get(Result); 292 } 293 294 } // end anonymous namespace 295 296 /// If this constant is a constant offset from a global, return the global and 297 /// the constant. Because of constantexprs, this function is recursive. 298 bool llvm::IsConstantOffsetFromGlobal(Constant *C, GlobalValue *&GV, 299 APInt &Offset, const DataLayout &DL, 300 DSOLocalEquivalent **DSOEquiv) { 301 if (DSOEquiv) 302 *DSOEquiv = nullptr; 303 304 // Trivial case, constant is the global. 305 if ((GV = dyn_cast<GlobalValue>(C))) { 306 unsigned BitWidth = DL.getIndexTypeSizeInBits(GV->getType()); 307 Offset = APInt(BitWidth, 0); 308 return true; 309 } 310 311 if (auto *FoundDSOEquiv = dyn_cast<DSOLocalEquivalent>(C)) { 312 if (DSOEquiv) 313 *DSOEquiv = FoundDSOEquiv; 314 GV = FoundDSOEquiv->getGlobalValue(); 315 unsigned BitWidth = DL.getIndexTypeSizeInBits(GV->getType()); 316 Offset = APInt(BitWidth, 0); 317 return true; 318 } 319 320 // Otherwise, if this isn't a constant expr, bail out. 321 auto *CE = dyn_cast<ConstantExpr>(C); 322 if (!CE) return false; 323 324 // Look through ptr->int and ptr->ptr casts. 325 if (CE->getOpcode() == Instruction::PtrToInt || 326 CE->getOpcode() == Instruction::BitCast) 327 return IsConstantOffsetFromGlobal(CE->getOperand(0), GV, Offset, DL, 328 DSOEquiv); 329 330 // i32* getelementptr ([5 x i32]* @a, i32 0, i32 5) 331 auto *GEP = dyn_cast<GEPOperator>(CE); 332 if (!GEP) 333 return false; 334 335 unsigned BitWidth = DL.getIndexTypeSizeInBits(GEP->getType()); 336 APInt TmpOffset(BitWidth, 0); 337 338 // If the base isn't a global+constant, we aren't either. 339 if (!IsConstantOffsetFromGlobal(CE->getOperand(0), GV, TmpOffset, DL, 340 DSOEquiv)) 341 return false; 342 343 // Otherwise, add any offset that our operands provide. 344 if (!GEP->accumulateConstantOffset(DL, TmpOffset)) 345 return false; 346 347 Offset = TmpOffset; 348 return true; 349 } 350 351 Constant *llvm::ConstantFoldLoadThroughBitcast(Constant *C, Type *DestTy, 352 const DataLayout &DL) { 353 do { 354 Type *SrcTy = C->getType(); 355 if (SrcTy == DestTy) 356 return C; 357 358 TypeSize DestSize = DL.getTypeSizeInBits(DestTy); 359 TypeSize SrcSize = DL.getTypeSizeInBits(SrcTy); 360 if (!TypeSize::isKnownGE(SrcSize, DestSize)) 361 return nullptr; 362 363 // Catch the obvious splat cases (since all-zeros can coerce non-integral 364 // pointers legally). 365 if (C->isNullValue() && !DestTy->isX86_MMXTy() && !DestTy->isX86_AMXTy()) 366 return Constant::getNullValue(DestTy); 367 if (C->isAllOnesValue() && 368 (DestTy->isIntegerTy() || DestTy->isFloatingPointTy() || 369 DestTy->isVectorTy()) && 370 !DestTy->isX86_AMXTy() && !DestTy->isX86_MMXTy() && 371 !DestTy->isPtrOrPtrVectorTy()) 372 // Get ones when the input is trivial, but 373 // only for supported types inside getAllOnesValue. 374 return Constant::getAllOnesValue(DestTy); 375 376 // If the type sizes are the same and a cast is legal, just directly 377 // cast the constant. 378 // But be careful not to coerce non-integral pointers illegally. 379 if (SrcSize == DestSize && 380 DL.isNonIntegralPointerType(SrcTy->getScalarType()) == 381 DL.isNonIntegralPointerType(DestTy->getScalarType())) { 382 Instruction::CastOps Cast = Instruction::BitCast; 383 // If we are going from a pointer to int or vice versa, we spell the cast 384 // differently. 385 if (SrcTy->isIntegerTy() && DestTy->isPointerTy()) 386 Cast = Instruction::IntToPtr; 387 else if (SrcTy->isPointerTy() && DestTy->isIntegerTy()) 388 Cast = Instruction::PtrToInt; 389 390 if (CastInst::castIsValid(Cast, C, DestTy)) 391 return ConstantExpr::getCast(Cast, C, DestTy); 392 } 393 394 // If this isn't an aggregate type, there is nothing we can do to drill down 395 // and find a bitcastable constant. 396 if (!SrcTy->isAggregateType() && !SrcTy->isVectorTy()) 397 return nullptr; 398 399 // We're simulating a load through a pointer that was bitcast to point to 400 // a different type, so we can try to walk down through the initial 401 // elements of an aggregate to see if some part of the aggregate is 402 // castable to implement the "load" semantic model. 403 if (SrcTy->isStructTy()) { 404 // Struct types might have leading zero-length elements like [0 x i32], 405 // which are certainly not what we are looking for, so skip them. 406 unsigned Elem = 0; 407 Constant *ElemC; 408 do { 409 ElemC = C->getAggregateElement(Elem++); 410 } while (ElemC && DL.getTypeSizeInBits(ElemC->getType()).isZero()); 411 C = ElemC; 412 } else { 413 C = C->getAggregateElement(0u); 414 } 415 } while (C); 416 417 return nullptr; 418 } 419 420 namespace { 421 422 /// Recursive helper to read bits out of global. C is the constant being copied 423 /// out of. ByteOffset is an offset into C. CurPtr is the pointer to copy 424 /// results into and BytesLeft is the number of bytes left in 425 /// the CurPtr buffer. DL is the DataLayout. 426 bool ReadDataFromGlobal(Constant *C, uint64_t ByteOffset, unsigned char *CurPtr, 427 unsigned BytesLeft, const DataLayout &DL) { 428 assert(ByteOffset <= DL.getTypeAllocSize(C->getType()) && 429 "Out of range access"); 430 431 // If this element is zero or undefined, we can just return since *CurPtr is 432 // zero initialized. 433 if (isa<ConstantAggregateZero>(C) || isa<UndefValue>(C)) 434 return true; 435 436 if (auto *CI = dyn_cast<ConstantInt>(C)) { 437 if (CI->getBitWidth() > 64 || 438 (CI->getBitWidth() & 7) != 0) 439 return false; 440 441 uint64_t Val = CI->getZExtValue(); 442 unsigned IntBytes = unsigned(CI->getBitWidth()/8); 443 444 for (unsigned i = 0; i != BytesLeft && ByteOffset != IntBytes; ++i) { 445 int n = ByteOffset; 446 if (!DL.isLittleEndian()) 447 n = IntBytes - n - 1; 448 CurPtr[i] = (unsigned char)(Val >> (n * 8)); 449 ++ByteOffset; 450 } 451 return true; 452 } 453 454 if (auto *CFP = dyn_cast<ConstantFP>(C)) { 455 if (CFP->getType()->isDoubleTy()) { 456 C = FoldBitCast(C, Type::getInt64Ty(C->getContext()), DL); 457 return ReadDataFromGlobal(C, ByteOffset, CurPtr, BytesLeft, DL); 458 } 459 if (CFP->getType()->isFloatTy()){ 460 C = FoldBitCast(C, Type::getInt32Ty(C->getContext()), DL); 461 return ReadDataFromGlobal(C, ByteOffset, CurPtr, BytesLeft, DL); 462 } 463 if (CFP->getType()->isHalfTy()){ 464 C = FoldBitCast(C, Type::getInt16Ty(C->getContext()), DL); 465 return ReadDataFromGlobal(C, ByteOffset, CurPtr, BytesLeft, DL); 466 } 467 return false; 468 } 469 470 if (auto *CS = dyn_cast<ConstantStruct>(C)) { 471 const StructLayout *SL = DL.getStructLayout(CS->getType()); 472 unsigned Index = SL->getElementContainingOffset(ByteOffset); 473 uint64_t CurEltOffset = SL->getElementOffset(Index); 474 ByteOffset -= CurEltOffset; 475 476 while (true) { 477 // If the element access is to the element itself and not to tail padding, 478 // read the bytes from the element. 479 uint64_t EltSize = DL.getTypeAllocSize(CS->getOperand(Index)->getType()); 480 481 if (ByteOffset < EltSize && 482 !ReadDataFromGlobal(CS->getOperand(Index), ByteOffset, CurPtr, 483 BytesLeft, DL)) 484 return false; 485 486 ++Index; 487 488 // Check to see if we read from the last struct element, if so we're done. 489 if (Index == CS->getType()->getNumElements()) 490 return true; 491 492 // If we read all of the bytes we needed from this element we're done. 493 uint64_t NextEltOffset = SL->getElementOffset(Index); 494 495 if (BytesLeft <= NextEltOffset - CurEltOffset - ByteOffset) 496 return true; 497 498 // Move to the next element of the struct. 499 CurPtr += NextEltOffset - CurEltOffset - ByteOffset; 500 BytesLeft -= NextEltOffset - CurEltOffset - ByteOffset; 501 ByteOffset = 0; 502 CurEltOffset = NextEltOffset; 503 } 504 // not reached. 505 } 506 507 if (isa<ConstantArray>(C) || isa<ConstantVector>(C) || 508 isa<ConstantDataSequential>(C)) { 509 uint64_t NumElts; 510 Type *EltTy; 511 if (auto *AT = dyn_cast<ArrayType>(C->getType())) { 512 NumElts = AT->getNumElements(); 513 EltTy = AT->getElementType(); 514 } else { 515 NumElts = cast<FixedVectorType>(C->getType())->getNumElements(); 516 EltTy = cast<FixedVectorType>(C->getType())->getElementType(); 517 } 518 uint64_t EltSize = DL.getTypeAllocSize(EltTy); 519 uint64_t Index = ByteOffset / EltSize; 520 uint64_t Offset = ByteOffset - Index * EltSize; 521 522 for (; Index != NumElts; ++Index) { 523 if (!ReadDataFromGlobal(C->getAggregateElement(Index), Offset, CurPtr, 524 BytesLeft, DL)) 525 return false; 526 527 uint64_t BytesWritten = EltSize - Offset; 528 assert(BytesWritten <= EltSize && "Not indexing into this element?"); 529 if (BytesWritten >= BytesLeft) 530 return true; 531 532 Offset = 0; 533 BytesLeft -= BytesWritten; 534 CurPtr += BytesWritten; 535 } 536 return true; 537 } 538 539 if (auto *CE = dyn_cast<ConstantExpr>(C)) { 540 if (CE->getOpcode() == Instruction::IntToPtr && 541 CE->getOperand(0)->getType() == DL.getIntPtrType(CE->getType())) { 542 return ReadDataFromGlobal(CE->getOperand(0), ByteOffset, CurPtr, 543 BytesLeft, DL); 544 } 545 } 546 547 // Otherwise, unknown initializer type. 548 return false; 549 } 550 551 Constant *FoldReinterpretLoadFromConst(Constant *C, Type *LoadTy, 552 int64_t Offset, const DataLayout &DL) { 553 // Bail out early. Not expect to load from scalable global variable. 554 if (isa<ScalableVectorType>(LoadTy)) 555 return nullptr; 556 557 auto *IntType = dyn_cast<IntegerType>(LoadTy); 558 559 // If this isn't an integer load we can't fold it directly. 560 if (!IntType) { 561 // If this is a float/double load, we can try folding it as an int32/64 load 562 // and then bitcast the result. This can be useful for union cases. Note 563 // that address spaces don't matter here since we're not going to result in 564 // an actual new load. 565 Type *MapTy; 566 if (LoadTy->isHalfTy()) 567 MapTy = Type::getInt16Ty(C->getContext()); 568 else if (LoadTy->isFloatTy()) 569 MapTy = Type::getInt32Ty(C->getContext()); 570 else if (LoadTy->isDoubleTy()) 571 MapTy = Type::getInt64Ty(C->getContext()); 572 else if (LoadTy->isVectorTy()) { 573 MapTy = PointerType::getIntNTy( 574 C->getContext(), DL.getTypeSizeInBits(LoadTy).getFixedSize()); 575 } else 576 return nullptr; 577 578 if (Constant *Res = FoldReinterpretLoadFromConst(C, MapTy, Offset, DL)) { 579 if (Res->isNullValue() && !LoadTy->isX86_MMXTy() && 580 !LoadTy->isX86_AMXTy()) 581 // Materializing a zero can be done trivially without a bitcast 582 return Constant::getNullValue(LoadTy); 583 Type *CastTy = LoadTy->isPtrOrPtrVectorTy() ? DL.getIntPtrType(LoadTy) : LoadTy; 584 Res = FoldBitCast(Res, CastTy, DL); 585 if (LoadTy->isPtrOrPtrVectorTy()) { 586 // For vector of pointer, we needed to first convert to a vector of integer, then do vector inttoptr 587 if (Res->isNullValue() && !LoadTy->isX86_MMXTy() && 588 !LoadTy->isX86_AMXTy()) 589 return Constant::getNullValue(LoadTy); 590 if (DL.isNonIntegralPointerType(LoadTy->getScalarType())) 591 // Be careful not to replace a load of an addrspace value with an inttoptr here 592 return nullptr; 593 Res = ConstantExpr::getCast(Instruction::IntToPtr, Res, LoadTy); 594 } 595 return Res; 596 } 597 return nullptr; 598 } 599 600 unsigned BytesLoaded = (IntType->getBitWidth() + 7) / 8; 601 if (BytesLoaded > 32 || BytesLoaded == 0) 602 return nullptr; 603 604 int64_t InitializerSize = DL.getTypeAllocSize(C->getType()).getFixedSize(); 605 606 // If we're not accessing anything in this constant, the result is undefined. 607 if (Offset <= -1 * static_cast<int64_t>(BytesLoaded)) 608 return UndefValue::get(IntType); 609 610 // If we're not accessing anything in this constant, the result is undefined. 611 if (Offset >= InitializerSize) 612 return UndefValue::get(IntType); 613 614 unsigned char RawBytes[32] = {0}; 615 unsigned char *CurPtr = RawBytes; 616 unsigned BytesLeft = BytesLoaded; 617 618 // If we're loading off the beginning of the global, some bytes may be valid. 619 if (Offset < 0) { 620 CurPtr += -Offset; 621 BytesLeft += Offset; 622 Offset = 0; 623 } 624 625 if (!ReadDataFromGlobal(C, Offset, CurPtr, BytesLeft, DL)) 626 return nullptr; 627 628 APInt ResultVal = APInt(IntType->getBitWidth(), 0); 629 if (DL.isLittleEndian()) { 630 ResultVal = RawBytes[BytesLoaded - 1]; 631 for (unsigned i = 1; i != BytesLoaded; ++i) { 632 ResultVal <<= 8; 633 ResultVal |= RawBytes[BytesLoaded - 1 - i]; 634 } 635 } else { 636 ResultVal = RawBytes[0]; 637 for (unsigned i = 1; i != BytesLoaded; ++i) { 638 ResultVal <<= 8; 639 ResultVal |= RawBytes[i]; 640 } 641 } 642 643 return ConstantInt::get(IntType->getContext(), ResultVal); 644 } 645 646 /// If this Offset points exactly to the start of an aggregate element, return 647 /// that element, otherwise return nullptr. 648 Constant *getConstantAtOffset(Constant *Base, APInt Offset, 649 const DataLayout &DL) { 650 if (Offset.isZero()) 651 return Base; 652 653 if (!isa<ConstantAggregate>(Base) && !isa<ConstantDataSequential>(Base)) 654 return nullptr; 655 656 Type *ElemTy = Base->getType(); 657 SmallVector<APInt> Indices = DL.getGEPIndicesForOffset(ElemTy, Offset); 658 if (!Offset.isZero() || !Indices[0].isZero()) 659 return nullptr; 660 661 Constant *C = Base; 662 for (const APInt &Index : drop_begin(Indices)) { 663 if (Index.isNegative() || Index.getActiveBits() >= 32) 664 return nullptr; 665 666 C = C->getAggregateElement(Index.getZExtValue()); 667 if (!C) 668 return nullptr; 669 } 670 671 return C; 672 } 673 674 } // end anonymous namespace 675 676 Constant *llvm::ConstantFoldLoadFromConst(Constant *C, Type *Ty, 677 const APInt &Offset, 678 const DataLayout &DL) { 679 if (Constant *AtOffset = getConstantAtOffset(C, Offset, DL)) 680 if (Constant *Result = ConstantFoldLoadThroughBitcast(AtOffset, Ty, DL)) 681 return Result; 682 683 // Try hard to fold loads from bitcasted strange and non-type-safe things. 684 if (Offset.getMinSignedBits() <= 64) 685 return FoldReinterpretLoadFromConst(C, Ty, Offset.getSExtValue(), DL); 686 687 return nullptr; 688 } 689 690 Constant *llvm::ConstantFoldLoadFromConst(Constant *C, Type *Ty, 691 const DataLayout &DL) { 692 return ConstantFoldLoadFromConst(C, Ty, APInt(64, 0), DL); 693 } 694 695 Constant *llvm::ConstantFoldLoadFromConstPtr(Constant *C, Type *Ty, 696 APInt Offset, 697 const DataLayout &DL) { 698 C = cast<Constant>(C->stripAndAccumulateConstantOffsets( 699 DL, Offset, /* AllowNonInbounds */ true)); 700 701 if (auto *GV = dyn_cast<GlobalVariable>(C)) 702 if (GV->isConstant() && GV->hasDefinitiveInitializer()) 703 if (Constant *Result = ConstantFoldLoadFromConst(GV->getInitializer(), Ty, 704 Offset, DL)) 705 return Result; 706 707 // If this load comes from anywhere in a constant global, and if the global 708 // is all undef or zero, we know what it loads. 709 if (auto *GV = dyn_cast<GlobalVariable>(getUnderlyingObject(C))) { 710 if (GV->isConstant() && GV->hasDefinitiveInitializer()) { 711 if (GV->getInitializer()->isNullValue()) 712 return Constant::getNullValue(Ty); 713 if (isa<UndefValue>(GV->getInitializer())) 714 return UndefValue::get(Ty); 715 } 716 } 717 718 return nullptr; 719 } 720 721 Constant *llvm::ConstantFoldLoadFromConstPtr(Constant *C, Type *Ty, 722 const DataLayout &DL) { 723 APInt Offset(DL.getIndexTypeSizeInBits(C->getType()), 0); 724 return ConstantFoldLoadFromConstPtr(C, Ty, Offset, DL); 725 } 726 727 namespace { 728 729 /// One of Op0/Op1 is a constant expression. 730 /// Attempt to symbolically evaluate the result of a binary operator merging 731 /// these together. If target data info is available, it is provided as DL, 732 /// otherwise DL is null. 733 Constant *SymbolicallyEvaluateBinop(unsigned Opc, Constant *Op0, Constant *Op1, 734 const DataLayout &DL) { 735 // SROA 736 737 // Fold (and 0xffffffff00000000, (shl x, 32)) -> shl. 738 // Fold (lshr (or X, Y), 32) -> (lshr [X/Y], 32) if one doesn't contribute 739 // bits. 740 741 if (Opc == Instruction::And) { 742 KnownBits Known0 = computeKnownBits(Op0, DL); 743 KnownBits Known1 = computeKnownBits(Op1, DL); 744 if ((Known1.One | Known0.Zero).isAllOnes()) { 745 // All the bits of Op0 that the 'and' could be masking are already zero. 746 return Op0; 747 } 748 if ((Known0.One | Known1.Zero).isAllOnes()) { 749 // All the bits of Op1 that the 'and' could be masking are already zero. 750 return Op1; 751 } 752 753 Known0 &= Known1; 754 if (Known0.isConstant()) 755 return ConstantInt::get(Op0->getType(), Known0.getConstant()); 756 } 757 758 // If the constant expr is something like &A[123] - &A[4].f, fold this into a 759 // constant. This happens frequently when iterating over a global array. 760 if (Opc == Instruction::Sub) { 761 GlobalValue *GV1, *GV2; 762 APInt Offs1, Offs2; 763 764 if (IsConstantOffsetFromGlobal(Op0, GV1, Offs1, DL)) 765 if (IsConstantOffsetFromGlobal(Op1, GV2, Offs2, DL) && GV1 == GV2) { 766 unsigned OpSize = DL.getTypeSizeInBits(Op0->getType()); 767 768 // (&GV+C1) - (&GV+C2) -> C1-C2, pointer arithmetic cannot overflow. 769 // PtrToInt may change the bitwidth so we have convert to the right size 770 // first. 771 return ConstantInt::get(Op0->getType(), Offs1.zextOrTrunc(OpSize) - 772 Offs2.zextOrTrunc(OpSize)); 773 } 774 } 775 776 return nullptr; 777 } 778 779 /// If array indices are not pointer-sized integers, explicitly cast them so 780 /// that they aren't implicitly casted by the getelementptr. 781 Constant *CastGEPIndices(Type *SrcElemTy, ArrayRef<Constant *> Ops, 782 Type *ResultTy, Optional<unsigned> InRangeIndex, 783 const DataLayout &DL, const TargetLibraryInfo *TLI) { 784 Type *IntIdxTy = DL.getIndexType(ResultTy); 785 Type *IntIdxScalarTy = IntIdxTy->getScalarType(); 786 787 bool Any = false; 788 SmallVector<Constant*, 32> NewIdxs; 789 for (unsigned i = 1, e = Ops.size(); i != e; ++i) { 790 if ((i == 1 || 791 !isa<StructType>(GetElementPtrInst::getIndexedType( 792 SrcElemTy, Ops.slice(1, i - 1)))) && 793 Ops[i]->getType()->getScalarType() != IntIdxScalarTy) { 794 Any = true; 795 Type *NewType = Ops[i]->getType()->isVectorTy() 796 ? IntIdxTy 797 : IntIdxScalarTy; 798 NewIdxs.push_back(ConstantExpr::getCast(CastInst::getCastOpcode(Ops[i], 799 true, 800 NewType, 801 true), 802 Ops[i], NewType)); 803 } else 804 NewIdxs.push_back(Ops[i]); 805 } 806 807 if (!Any) 808 return nullptr; 809 810 Constant *C = ConstantExpr::getGetElementPtr( 811 SrcElemTy, Ops[0], NewIdxs, /*InBounds=*/false, InRangeIndex); 812 return ConstantFoldConstant(C, DL, TLI); 813 } 814 815 /// Strip the pointer casts, but preserve the address space information. 816 Constant *StripPtrCastKeepAS(Constant *Ptr) { 817 assert(Ptr->getType()->isPointerTy() && "Not a pointer type"); 818 auto *OldPtrTy = cast<PointerType>(Ptr->getType()); 819 Ptr = cast<Constant>(Ptr->stripPointerCasts()); 820 auto *NewPtrTy = cast<PointerType>(Ptr->getType()); 821 822 // Preserve the address space number of the pointer. 823 if (NewPtrTy->getAddressSpace() != OldPtrTy->getAddressSpace()) { 824 Ptr = ConstantExpr::getPointerCast( 825 Ptr, PointerType::getWithSamePointeeType(NewPtrTy, 826 OldPtrTy->getAddressSpace())); 827 } 828 return Ptr; 829 } 830 831 /// If we can symbolically evaluate the GEP constant expression, do so. 832 Constant *SymbolicallyEvaluateGEP(const GEPOperator *GEP, 833 ArrayRef<Constant *> Ops, 834 const DataLayout &DL, 835 const TargetLibraryInfo *TLI) { 836 const GEPOperator *InnermostGEP = GEP; 837 bool InBounds = GEP->isInBounds(); 838 839 Type *SrcElemTy = GEP->getSourceElementType(); 840 Type *ResElemTy = GEP->getResultElementType(); 841 Type *ResTy = GEP->getType(); 842 if (!SrcElemTy->isSized() || isa<ScalableVectorType>(SrcElemTy)) 843 return nullptr; 844 845 if (Constant *C = CastGEPIndices(SrcElemTy, Ops, ResTy, 846 GEP->getInRangeIndex(), DL, TLI)) 847 return C; 848 849 Constant *Ptr = Ops[0]; 850 if (!Ptr->getType()->isPointerTy()) 851 return nullptr; 852 853 Type *IntIdxTy = DL.getIndexType(Ptr->getType()); 854 855 // If this is "gep i8* Ptr, (sub 0, V)", fold this as: 856 // "inttoptr (sub (ptrtoint Ptr), V)" 857 if (Ops.size() == 2 && ResElemTy->isIntegerTy(8)) { 858 auto *CE = dyn_cast<ConstantExpr>(Ops[1]); 859 assert((!CE || CE->getType() == IntIdxTy) && 860 "CastGEPIndices didn't canonicalize index types!"); 861 if (CE && CE->getOpcode() == Instruction::Sub && 862 CE->getOperand(0)->isNullValue()) { 863 Constant *Res = ConstantExpr::getPtrToInt(Ptr, CE->getType()); 864 Res = ConstantExpr::getSub(Res, CE->getOperand(1)); 865 Res = ConstantExpr::getIntToPtr(Res, ResTy); 866 return ConstantFoldConstant(Res, DL, TLI); 867 } 868 } 869 870 for (unsigned i = 1, e = Ops.size(); i != e; ++i) 871 if (!isa<ConstantInt>(Ops[i])) 872 return nullptr; 873 874 unsigned BitWidth = DL.getTypeSizeInBits(IntIdxTy); 875 APInt Offset = 876 APInt(BitWidth, 877 DL.getIndexedOffsetInType( 878 SrcElemTy, 879 makeArrayRef((Value * const *)Ops.data() + 1, Ops.size() - 1))); 880 Ptr = StripPtrCastKeepAS(Ptr); 881 882 // If this is a GEP of a GEP, fold it all into a single GEP. 883 while (auto *GEP = dyn_cast<GEPOperator>(Ptr)) { 884 InnermostGEP = GEP; 885 InBounds &= GEP->isInBounds(); 886 887 SmallVector<Value *, 4> NestedOps(GEP->op_begin() + 1, GEP->op_end()); 888 889 // Do not try the incorporate the sub-GEP if some index is not a number. 890 bool AllConstantInt = true; 891 for (Value *NestedOp : NestedOps) 892 if (!isa<ConstantInt>(NestedOp)) { 893 AllConstantInt = false; 894 break; 895 } 896 if (!AllConstantInt) 897 break; 898 899 Ptr = cast<Constant>(GEP->getOperand(0)); 900 SrcElemTy = GEP->getSourceElementType(); 901 Offset += APInt(BitWidth, DL.getIndexedOffsetInType(SrcElemTy, NestedOps)); 902 Ptr = StripPtrCastKeepAS(Ptr); 903 } 904 905 // If the base value for this address is a literal integer value, fold the 906 // getelementptr to the resulting integer value casted to the pointer type. 907 APInt BasePtr(BitWidth, 0); 908 if (auto *CE = dyn_cast<ConstantExpr>(Ptr)) { 909 if (CE->getOpcode() == Instruction::IntToPtr) { 910 if (auto *Base = dyn_cast<ConstantInt>(CE->getOperand(0))) 911 BasePtr = Base->getValue().zextOrTrunc(BitWidth); 912 } 913 } 914 915 auto *PTy = cast<PointerType>(Ptr->getType()); 916 if ((Ptr->isNullValue() || BasePtr != 0) && 917 !DL.isNonIntegralPointerType(PTy)) { 918 Constant *C = ConstantInt::get(Ptr->getContext(), Offset + BasePtr); 919 return ConstantExpr::getIntToPtr(C, ResTy); 920 } 921 922 // Otherwise form a regular getelementptr. Recompute the indices so that 923 // we eliminate over-indexing of the notional static type array bounds. 924 // This makes it easy to determine if the getelementptr is "inbounds". 925 // Also, this helps GlobalOpt do SROA on GlobalVariables. 926 927 // For GEPs of GlobalValues, use the value type even for opaque pointers. 928 // Otherwise use an i8 GEP. 929 if (auto *GV = dyn_cast<GlobalValue>(Ptr)) 930 SrcElemTy = GV->getValueType(); 931 else if (!PTy->isOpaque()) 932 SrcElemTy = PTy->getElementType(); 933 else 934 SrcElemTy = Type::getInt8Ty(Ptr->getContext()); 935 936 if (!SrcElemTy->isSized()) 937 return nullptr; 938 939 Type *ElemTy = SrcElemTy; 940 SmallVector<APInt> Indices = DL.getGEPIndicesForOffset(ElemTy, Offset); 941 if (Offset != 0) 942 return nullptr; 943 944 // Try to add additional zero indices to reach the desired result element 945 // type. 946 // TODO: Should we avoid extra zero indices if ResElemTy can't be reached and 947 // we'll have to insert a bitcast anyway? 948 while (ElemTy != ResElemTy) { 949 Type *NextTy = GetElementPtrInst::getTypeAtIndex(ElemTy, (uint64_t)0); 950 if (!NextTy) 951 break; 952 953 Indices.push_back(APInt::getZero(isa<StructType>(ElemTy) ? 32 : BitWidth)); 954 ElemTy = NextTy; 955 } 956 957 SmallVector<Constant *, 32> NewIdxs; 958 for (const APInt &Index : Indices) 959 NewIdxs.push_back(ConstantInt::get( 960 Type::getIntNTy(Ptr->getContext(), Index.getBitWidth()), Index)); 961 962 // Preserve the inrange index from the innermost GEP if possible. We must 963 // have calculated the same indices up to and including the inrange index. 964 Optional<unsigned> InRangeIndex; 965 if (Optional<unsigned> LastIRIndex = InnermostGEP->getInRangeIndex()) 966 if (SrcElemTy == InnermostGEP->getSourceElementType() && 967 NewIdxs.size() > *LastIRIndex) { 968 InRangeIndex = LastIRIndex; 969 for (unsigned I = 0; I <= *LastIRIndex; ++I) 970 if (NewIdxs[I] != InnermostGEP->getOperand(I + 1)) 971 return nullptr; 972 } 973 974 // Create a GEP. 975 Constant *C = ConstantExpr::getGetElementPtr(SrcElemTy, Ptr, NewIdxs, 976 InBounds, InRangeIndex); 977 assert( 978 cast<PointerType>(C->getType())->isOpaqueOrPointeeTypeMatches(ElemTy) && 979 "Computed GetElementPtr has unexpected type!"); 980 981 // If we ended up indexing a member with a type that doesn't match 982 // the type of what the original indices indexed, add a cast. 983 if (C->getType() != ResTy) 984 C = FoldBitCast(C, ResTy, DL); 985 986 return C; 987 } 988 989 /// Attempt to constant fold an instruction with the 990 /// specified opcode and operands. If successful, the constant result is 991 /// returned, if not, null is returned. Note that this function can fail when 992 /// attempting to fold instructions like loads and stores, which have no 993 /// constant expression form. 994 Constant *ConstantFoldInstOperandsImpl(const Value *InstOrCE, unsigned Opcode, 995 ArrayRef<Constant *> Ops, 996 const DataLayout &DL, 997 const TargetLibraryInfo *TLI) { 998 Type *DestTy = InstOrCE->getType(); 999 1000 if (Instruction::isUnaryOp(Opcode)) 1001 return ConstantFoldUnaryOpOperand(Opcode, Ops[0], DL); 1002 1003 if (Instruction::isBinaryOp(Opcode)) 1004 return ConstantFoldBinaryOpOperands(Opcode, Ops[0], Ops[1], DL); 1005 1006 if (Instruction::isCast(Opcode)) 1007 return ConstantFoldCastOperand(Opcode, Ops[0], DestTy, DL); 1008 1009 if (auto *GEP = dyn_cast<GEPOperator>(InstOrCE)) { 1010 if (Constant *C = SymbolicallyEvaluateGEP(GEP, Ops, DL, TLI)) 1011 return C; 1012 1013 return ConstantExpr::getGetElementPtr(GEP->getSourceElementType(), Ops[0], 1014 Ops.slice(1), GEP->isInBounds(), 1015 GEP->getInRangeIndex()); 1016 } 1017 1018 if (auto *CE = dyn_cast<ConstantExpr>(InstOrCE)) 1019 return CE->getWithOperands(Ops); 1020 1021 switch (Opcode) { 1022 default: return nullptr; 1023 case Instruction::ICmp: 1024 case Instruction::FCmp: llvm_unreachable("Invalid for compares"); 1025 case Instruction::Freeze: 1026 return isGuaranteedNotToBeUndefOrPoison(Ops[0]) ? Ops[0] : nullptr; 1027 case Instruction::Call: 1028 if (auto *F = dyn_cast<Function>(Ops.back())) { 1029 const auto *Call = cast<CallBase>(InstOrCE); 1030 if (canConstantFoldCallTo(Call, F)) 1031 return ConstantFoldCall(Call, F, Ops.slice(0, Ops.size() - 1), TLI); 1032 } 1033 return nullptr; 1034 case Instruction::Select: 1035 return ConstantExpr::getSelect(Ops[0], Ops[1], Ops[2]); 1036 case Instruction::ExtractElement: 1037 return ConstantExpr::getExtractElement(Ops[0], Ops[1]); 1038 case Instruction::ExtractValue: 1039 return ConstantExpr::getExtractValue( 1040 Ops[0], cast<ExtractValueInst>(InstOrCE)->getIndices()); 1041 case Instruction::InsertElement: 1042 return ConstantExpr::getInsertElement(Ops[0], Ops[1], Ops[2]); 1043 case Instruction::ShuffleVector: 1044 return ConstantExpr::getShuffleVector( 1045 Ops[0], Ops[1], cast<ShuffleVectorInst>(InstOrCE)->getShuffleMask()); 1046 } 1047 } 1048 1049 } // end anonymous namespace 1050 1051 //===----------------------------------------------------------------------===// 1052 // Constant Folding public APIs 1053 //===----------------------------------------------------------------------===// 1054 1055 namespace { 1056 1057 Constant * 1058 ConstantFoldConstantImpl(const Constant *C, const DataLayout &DL, 1059 const TargetLibraryInfo *TLI, 1060 SmallDenseMap<Constant *, Constant *> &FoldedOps) { 1061 if (!isa<ConstantVector>(C) && !isa<ConstantExpr>(C)) 1062 return const_cast<Constant *>(C); 1063 1064 SmallVector<Constant *, 8> Ops; 1065 for (const Use &OldU : C->operands()) { 1066 Constant *OldC = cast<Constant>(&OldU); 1067 Constant *NewC = OldC; 1068 // Recursively fold the ConstantExpr's operands. If we have already folded 1069 // a ConstantExpr, we don't have to process it again. 1070 if (isa<ConstantVector>(OldC) || isa<ConstantExpr>(OldC)) { 1071 auto It = FoldedOps.find(OldC); 1072 if (It == FoldedOps.end()) { 1073 NewC = ConstantFoldConstantImpl(OldC, DL, TLI, FoldedOps); 1074 FoldedOps.insert({OldC, NewC}); 1075 } else { 1076 NewC = It->second; 1077 } 1078 } 1079 Ops.push_back(NewC); 1080 } 1081 1082 if (auto *CE = dyn_cast<ConstantExpr>(C)) { 1083 if (CE->isCompare()) 1084 return ConstantFoldCompareInstOperands(CE->getPredicate(), Ops[0], Ops[1], 1085 DL, TLI); 1086 1087 return ConstantFoldInstOperandsImpl(CE, CE->getOpcode(), Ops, DL, TLI); 1088 } 1089 1090 assert(isa<ConstantVector>(C)); 1091 return ConstantVector::get(Ops); 1092 } 1093 1094 } // end anonymous namespace 1095 1096 Constant *llvm::ConstantFoldInstruction(Instruction *I, const DataLayout &DL, 1097 const TargetLibraryInfo *TLI) { 1098 // Handle PHI nodes quickly here... 1099 if (auto *PN = dyn_cast<PHINode>(I)) { 1100 Constant *CommonValue = nullptr; 1101 1102 SmallDenseMap<Constant *, Constant *> FoldedOps; 1103 for (Value *Incoming : PN->incoming_values()) { 1104 // If the incoming value is undef then skip it. Note that while we could 1105 // skip the value if it is equal to the phi node itself we choose not to 1106 // because that would break the rule that constant folding only applies if 1107 // all operands are constants. 1108 if (isa<UndefValue>(Incoming)) 1109 continue; 1110 // If the incoming value is not a constant, then give up. 1111 auto *C = dyn_cast<Constant>(Incoming); 1112 if (!C) 1113 return nullptr; 1114 // Fold the PHI's operands. 1115 C = ConstantFoldConstantImpl(C, DL, TLI, FoldedOps); 1116 // If the incoming value is a different constant to 1117 // the one we saw previously, then give up. 1118 if (CommonValue && C != CommonValue) 1119 return nullptr; 1120 CommonValue = C; 1121 } 1122 1123 // If we reach here, all incoming values are the same constant or undef. 1124 return CommonValue ? CommonValue : UndefValue::get(PN->getType()); 1125 } 1126 1127 // Scan the operand list, checking to see if they are all constants, if so, 1128 // hand off to ConstantFoldInstOperandsImpl. 1129 if (!all_of(I->operands(), [](Use &U) { return isa<Constant>(U); })) 1130 return nullptr; 1131 1132 SmallDenseMap<Constant *, Constant *> FoldedOps; 1133 SmallVector<Constant *, 8> Ops; 1134 for (const Use &OpU : I->operands()) { 1135 auto *Op = cast<Constant>(&OpU); 1136 // Fold the Instruction's operands. 1137 Op = ConstantFoldConstantImpl(Op, DL, TLI, FoldedOps); 1138 Ops.push_back(Op); 1139 } 1140 1141 if (const auto *CI = dyn_cast<CmpInst>(I)) 1142 return ConstantFoldCompareInstOperands(CI->getPredicate(), Ops[0], Ops[1], 1143 DL, TLI); 1144 1145 if (const auto *LI = dyn_cast<LoadInst>(I)) { 1146 if (LI->isVolatile()) 1147 return nullptr; 1148 return ConstantFoldLoadFromConstPtr(Ops[0], LI->getType(), DL); 1149 } 1150 1151 if (auto *IVI = dyn_cast<InsertValueInst>(I)) 1152 return ConstantExpr::getInsertValue(Ops[0], Ops[1], IVI->getIndices()); 1153 1154 if (auto *EVI = dyn_cast<ExtractValueInst>(I)) 1155 return ConstantExpr::getExtractValue(Ops[0], EVI->getIndices()); 1156 1157 return ConstantFoldInstOperands(I, Ops, DL, TLI); 1158 } 1159 1160 Constant *llvm::ConstantFoldConstant(const Constant *C, const DataLayout &DL, 1161 const TargetLibraryInfo *TLI) { 1162 SmallDenseMap<Constant *, Constant *> FoldedOps; 1163 return ConstantFoldConstantImpl(C, DL, TLI, FoldedOps); 1164 } 1165 1166 Constant *llvm::ConstantFoldInstOperands(Instruction *I, 1167 ArrayRef<Constant *> Ops, 1168 const DataLayout &DL, 1169 const TargetLibraryInfo *TLI) { 1170 return ConstantFoldInstOperandsImpl(I, I->getOpcode(), Ops, DL, TLI); 1171 } 1172 1173 Constant *llvm::ConstantFoldCompareInstOperands(unsigned Predicate, 1174 Constant *Ops0, Constant *Ops1, 1175 const DataLayout &DL, 1176 const TargetLibraryInfo *TLI) { 1177 // fold: icmp (inttoptr x), null -> icmp x, 0 1178 // fold: icmp null, (inttoptr x) -> icmp 0, x 1179 // fold: icmp (ptrtoint x), 0 -> icmp x, null 1180 // fold: icmp 0, (ptrtoint x) -> icmp null, x 1181 // fold: icmp (inttoptr x), (inttoptr y) -> icmp trunc/zext x, trunc/zext y 1182 // fold: icmp (ptrtoint x), (ptrtoint y) -> icmp x, y 1183 // 1184 // FIXME: The following comment is out of data and the DataLayout is here now. 1185 // ConstantExpr::getCompare cannot do this, because it doesn't have DL 1186 // around to know if bit truncation is happening. 1187 if (auto *CE0 = dyn_cast<ConstantExpr>(Ops0)) { 1188 if (Ops1->isNullValue()) { 1189 if (CE0->getOpcode() == Instruction::IntToPtr) { 1190 Type *IntPtrTy = DL.getIntPtrType(CE0->getType()); 1191 // Convert the integer value to the right size to ensure we get the 1192 // proper extension or truncation. 1193 Constant *C = ConstantExpr::getIntegerCast(CE0->getOperand(0), 1194 IntPtrTy, false); 1195 Constant *Null = Constant::getNullValue(C->getType()); 1196 return ConstantFoldCompareInstOperands(Predicate, C, Null, DL, TLI); 1197 } 1198 1199 // Only do this transformation if the int is intptrty in size, otherwise 1200 // there is a truncation or extension that we aren't modeling. 1201 if (CE0->getOpcode() == Instruction::PtrToInt) { 1202 Type *IntPtrTy = DL.getIntPtrType(CE0->getOperand(0)->getType()); 1203 if (CE0->getType() == IntPtrTy) { 1204 Constant *C = CE0->getOperand(0); 1205 Constant *Null = Constant::getNullValue(C->getType()); 1206 return ConstantFoldCompareInstOperands(Predicate, C, Null, DL, TLI); 1207 } 1208 } 1209 } 1210 1211 if (auto *CE1 = dyn_cast<ConstantExpr>(Ops1)) { 1212 if (CE0->getOpcode() == CE1->getOpcode()) { 1213 if (CE0->getOpcode() == Instruction::IntToPtr) { 1214 Type *IntPtrTy = DL.getIntPtrType(CE0->getType()); 1215 1216 // Convert the integer value to the right size to ensure we get the 1217 // proper extension or truncation. 1218 Constant *C0 = ConstantExpr::getIntegerCast(CE0->getOperand(0), 1219 IntPtrTy, false); 1220 Constant *C1 = ConstantExpr::getIntegerCast(CE1->getOperand(0), 1221 IntPtrTy, false); 1222 return ConstantFoldCompareInstOperands(Predicate, C0, C1, DL, TLI); 1223 } 1224 1225 // Only do this transformation if the int is intptrty in size, otherwise 1226 // there is a truncation or extension that we aren't modeling. 1227 if (CE0->getOpcode() == Instruction::PtrToInt) { 1228 Type *IntPtrTy = DL.getIntPtrType(CE0->getOperand(0)->getType()); 1229 if (CE0->getType() == IntPtrTy && 1230 CE0->getOperand(0)->getType() == CE1->getOperand(0)->getType()) { 1231 return ConstantFoldCompareInstOperands( 1232 Predicate, CE0->getOperand(0), CE1->getOperand(0), DL, TLI); 1233 } 1234 } 1235 } 1236 } 1237 1238 // icmp eq (or x, y), 0 -> (icmp eq x, 0) & (icmp eq y, 0) 1239 // icmp ne (or x, y), 0 -> (icmp ne x, 0) | (icmp ne y, 0) 1240 if ((Predicate == ICmpInst::ICMP_EQ || Predicate == ICmpInst::ICMP_NE) && 1241 CE0->getOpcode() == Instruction::Or && Ops1->isNullValue()) { 1242 Constant *LHS = ConstantFoldCompareInstOperands( 1243 Predicate, CE0->getOperand(0), Ops1, DL, TLI); 1244 Constant *RHS = ConstantFoldCompareInstOperands( 1245 Predicate, CE0->getOperand(1), Ops1, DL, TLI); 1246 unsigned OpC = 1247 Predicate == ICmpInst::ICMP_EQ ? Instruction::And : Instruction::Or; 1248 return ConstantFoldBinaryOpOperands(OpC, LHS, RHS, DL); 1249 } 1250 } else if (isa<ConstantExpr>(Ops1)) { 1251 // If RHS is a constant expression, but the left side isn't, swap the 1252 // operands and try again. 1253 Predicate = ICmpInst::getSwappedPredicate((ICmpInst::Predicate)Predicate); 1254 return ConstantFoldCompareInstOperands(Predicate, Ops1, Ops0, DL, TLI); 1255 } 1256 1257 return ConstantExpr::getCompare(Predicate, Ops0, Ops1); 1258 } 1259 1260 Constant *llvm::ConstantFoldUnaryOpOperand(unsigned Opcode, Constant *Op, 1261 const DataLayout &DL) { 1262 assert(Instruction::isUnaryOp(Opcode)); 1263 1264 return ConstantExpr::get(Opcode, Op); 1265 } 1266 1267 Constant *llvm::ConstantFoldBinaryOpOperands(unsigned Opcode, Constant *LHS, 1268 Constant *RHS, 1269 const DataLayout &DL) { 1270 assert(Instruction::isBinaryOp(Opcode)); 1271 if (isa<ConstantExpr>(LHS) || isa<ConstantExpr>(RHS)) 1272 if (Constant *C = SymbolicallyEvaluateBinop(Opcode, LHS, RHS, DL)) 1273 return C; 1274 1275 return ConstantExpr::get(Opcode, LHS, RHS); 1276 } 1277 1278 Constant *llvm::ConstantFoldCastOperand(unsigned Opcode, Constant *C, 1279 Type *DestTy, const DataLayout &DL) { 1280 assert(Instruction::isCast(Opcode)); 1281 switch (Opcode) { 1282 default: 1283 llvm_unreachable("Missing case"); 1284 case Instruction::PtrToInt: 1285 if (auto *CE = dyn_cast<ConstantExpr>(C)) { 1286 Constant *FoldedValue = nullptr; 1287 // If the input is a inttoptr, eliminate the pair. This requires knowing 1288 // the width of a pointer, so it can't be done in ConstantExpr::getCast. 1289 if (CE->getOpcode() == Instruction::IntToPtr) { 1290 // zext/trunc the inttoptr to pointer size. 1291 FoldedValue = ConstantExpr::getIntegerCast( 1292 CE->getOperand(0), DL.getIntPtrType(CE->getType()), 1293 /*IsSigned=*/false); 1294 } else if (auto *GEP = dyn_cast<GEPOperator>(CE)) { 1295 // If we have GEP, we can perform the following folds: 1296 // (ptrtoint (gep null, x)) -> x 1297 // (ptrtoint (gep (gep null, x), y) -> x + y, etc. 1298 unsigned BitWidth = DL.getIndexTypeSizeInBits(GEP->getType()); 1299 APInt BaseOffset(BitWidth, 0); 1300 auto *Base = cast<Constant>(GEP->stripAndAccumulateConstantOffsets( 1301 DL, BaseOffset, /*AllowNonInbounds=*/true)); 1302 if (Base->isNullValue()) { 1303 FoldedValue = ConstantInt::get(CE->getContext(), BaseOffset); 1304 } 1305 } 1306 if (FoldedValue) { 1307 // Do a zext or trunc to get to the ptrtoint dest size. 1308 return ConstantExpr::getIntegerCast(FoldedValue, DestTy, 1309 /*IsSigned=*/false); 1310 } 1311 } 1312 return ConstantExpr::getCast(Opcode, C, DestTy); 1313 case Instruction::IntToPtr: 1314 // If the input is a ptrtoint, turn the pair into a ptr to ptr bitcast if 1315 // the int size is >= the ptr size and the address spaces are the same. 1316 // This requires knowing the width of a pointer, so it can't be done in 1317 // ConstantExpr::getCast. 1318 if (auto *CE = dyn_cast<ConstantExpr>(C)) { 1319 if (CE->getOpcode() == Instruction::PtrToInt) { 1320 Constant *SrcPtr = CE->getOperand(0); 1321 unsigned SrcPtrSize = DL.getPointerTypeSizeInBits(SrcPtr->getType()); 1322 unsigned MidIntSize = CE->getType()->getScalarSizeInBits(); 1323 1324 if (MidIntSize >= SrcPtrSize) { 1325 unsigned SrcAS = SrcPtr->getType()->getPointerAddressSpace(); 1326 if (SrcAS == DestTy->getPointerAddressSpace()) 1327 return FoldBitCast(CE->getOperand(0), DestTy, DL); 1328 } 1329 } 1330 } 1331 1332 return ConstantExpr::getCast(Opcode, C, DestTy); 1333 case Instruction::Trunc: 1334 case Instruction::ZExt: 1335 case Instruction::SExt: 1336 case Instruction::FPTrunc: 1337 case Instruction::FPExt: 1338 case Instruction::UIToFP: 1339 case Instruction::SIToFP: 1340 case Instruction::FPToUI: 1341 case Instruction::FPToSI: 1342 case Instruction::AddrSpaceCast: 1343 return ConstantExpr::getCast(Opcode, C, DestTy); 1344 case Instruction::BitCast: 1345 return FoldBitCast(C, DestTy, DL); 1346 } 1347 } 1348 1349 Constant *llvm::ConstantFoldLoadThroughGEPConstantExpr(Constant *C, 1350 ConstantExpr *CE, 1351 Type *Ty, 1352 const DataLayout &DL) { 1353 if (!CE->getOperand(1)->isNullValue()) 1354 return nullptr; // Do not allow stepping over the value! 1355 1356 // Loop over all of the operands, tracking down which value we are 1357 // addressing. 1358 for (unsigned i = 2, e = CE->getNumOperands(); i != e; ++i) { 1359 C = C->getAggregateElement(CE->getOperand(i)); 1360 if (!C) 1361 return nullptr; 1362 } 1363 return ConstantFoldLoadThroughBitcast(C, Ty, DL); 1364 } 1365 1366 //===----------------------------------------------------------------------===// 1367 // Constant Folding for Calls 1368 // 1369 1370 bool llvm::canConstantFoldCallTo(const CallBase *Call, const Function *F) { 1371 if (Call->isNoBuiltin()) 1372 return false; 1373 switch (F->getIntrinsicID()) { 1374 // Operations that do not operate floating-point numbers and do not depend on 1375 // FP environment can be folded even in strictfp functions. 1376 case Intrinsic::bswap: 1377 case Intrinsic::ctpop: 1378 case Intrinsic::ctlz: 1379 case Intrinsic::cttz: 1380 case Intrinsic::fshl: 1381 case Intrinsic::fshr: 1382 case Intrinsic::launder_invariant_group: 1383 case Intrinsic::strip_invariant_group: 1384 case Intrinsic::masked_load: 1385 case Intrinsic::get_active_lane_mask: 1386 case Intrinsic::abs: 1387 case Intrinsic::smax: 1388 case Intrinsic::smin: 1389 case Intrinsic::umax: 1390 case Intrinsic::umin: 1391 case Intrinsic::sadd_with_overflow: 1392 case Intrinsic::uadd_with_overflow: 1393 case Intrinsic::ssub_with_overflow: 1394 case Intrinsic::usub_with_overflow: 1395 case Intrinsic::smul_with_overflow: 1396 case Intrinsic::umul_with_overflow: 1397 case Intrinsic::sadd_sat: 1398 case Intrinsic::uadd_sat: 1399 case Intrinsic::ssub_sat: 1400 case Intrinsic::usub_sat: 1401 case Intrinsic::smul_fix: 1402 case Intrinsic::smul_fix_sat: 1403 case Intrinsic::bitreverse: 1404 case Intrinsic::is_constant: 1405 case Intrinsic::vector_reduce_add: 1406 case Intrinsic::vector_reduce_mul: 1407 case Intrinsic::vector_reduce_and: 1408 case Intrinsic::vector_reduce_or: 1409 case Intrinsic::vector_reduce_xor: 1410 case Intrinsic::vector_reduce_smin: 1411 case Intrinsic::vector_reduce_smax: 1412 case Intrinsic::vector_reduce_umin: 1413 case Intrinsic::vector_reduce_umax: 1414 // Target intrinsics 1415 case Intrinsic::amdgcn_perm: 1416 case Intrinsic::arm_mve_vctp8: 1417 case Intrinsic::arm_mve_vctp16: 1418 case Intrinsic::arm_mve_vctp32: 1419 case Intrinsic::arm_mve_vctp64: 1420 case Intrinsic::aarch64_sve_convert_from_svbool: 1421 // WebAssembly float semantics are always known 1422 case Intrinsic::wasm_trunc_signed: 1423 case Intrinsic::wasm_trunc_unsigned: 1424 return true; 1425 1426 // Floating point operations cannot be folded in strictfp functions in 1427 // general case. They can be folded if FP environment is known to compiler. 1428 case Intrinsic::minnum: 1429 case Intrinsic::maxnum: 1430 case Intrinsic::minimum: 1431 case Intrinsic::maximum: 1432 case Intrinsic::log: 1433 case Intrinsic::log2: 1434 case Intrinsic::log10: 1435 case Intrinsic::exp: 1436 case Intrinsic::exp2: 1437 case Intrinsic::sqrt: 1438 case Intrinsic::sin: 1439 case Intrinsic::cos: 1440 case Intrinsic::pow: 1441 case Intrinsic::powi: 1442 case Intrinsic::fma: 1443 case Intrinsic::fmuladd: 1444 case Intrinsic::fptoui_sat: 1445 case Intrinsic::fptosi_sat: 1446 case Intrinsic::convert_from_fp16: 1447 case Intrinsic::convert_to_fp16: 1448 case Intrinsic::amdgcn_cos: 1449 case Intrinsic::amdgcn_cubeid: 1450 case Intrinsic::amdgcn_cubema: 1451 case Intrinsic::amdgcn_cubesc: 1452 case Intrinsic::amdgcn_cubetc: 1453 case Intrinsic::amdgcn_fmul_legacy: 1454 case Intrinsic::amdgcn_fma_legacy: 1455 case Intrinsic::amdgcn_fract: 1456 case Intrinsic::amdgcn_ldexp: 1457 case Intrinsic::amdgcn_sin: 1458 // The intrinsics below depend on rounding mode in MXCSR. 1459 case Intrinsic::x86_sse_cvtss2si: 1460 case Intrinsic::x86_sse_cvtss2si64: 1461 case Intrinsic::x86_sse_cvttss2si: 1462 case Intrinsic::x86_sse_cvttss2si64: 1463 case Intrinsic::x86_sse2_cvtsd2si: 1464 case Intrinsic::x86_sse2_cvtsd2si64: 1465 case Intrinsic::x86_sse2_cvttsd2si: 1466 case Intrinsic::x86_sse2_cvttsd2si64: 1467 case Intrinsic::x86_avx512_vcvtss2si32: 1468 case Intrinsic::x86_avx512_vcvtss2si64: 1469 case Intrinsic::x86_avx512_cvttss2si: 1470 case Intrinsic::x86_avx512_cvttss2si64: 1471 case Intrinsic::x86_avx512_vcvtsd2si32: 1472 case Intrinsic::x86_avx512_vcvtsd2si64: 1473 case Intrinsic::x86_avx512_cvttsd2si: 1474 case Intrinsic::x86_avx512_cvttsd2si64: 1475 case Intrinsic::x86_avx512_vcvtss2usi32: 1476 case Intrinsic::x86_avx512_vcvtss2usi64: 1477 case Intrinsic::x86_avx512_cvttss2usi: 1478 case Intrinsic::x86_avx512_cvttss2usi64: 1479 case Intrinsic::x86_avx512_vcvtsd2usi32: 1480 case Intrinsic::x86_avx512_vcvtsd2usi64: 1481 case Intrinsic::x86_avx512_cvttsd2usi: 1482 case Intrinsic::x86_avx512_cvttsd2usi64: 1483 return !Call->isStrictFP(); 1484 1485 // Sign operations are actually bitwise operations, they do not raise 1486 // exceptions even for SNANs. 1487 case Intrinsic::fabs: 1488 case Intrinsic::copysign: 1489 // Non-constrained variants of rounding operations means default FP 1490 // environment, they can be folded in any case. 1491 case Intrinsic::ceil: 1492 case Intrinsic::floor: 1493 case Intrinsic::round: 1494 case Intrinsic::roundeven: 1495 case Intrinsic::trunc: 1496 case Intrinsic::nearbyint: 1497 case Intrinsic::rint: 1498 // Constrained intrinsics can be folded if FP environment is known 1499 // to compiler. 1500 case Intrinsic::experimental_constrained_fma: 1501 case Intrinsic::experimental_constrained_fmuladd: 1502 case Intrinsic::experimental_constrained_fadd: 1503 case Intrinsic::experimental_constrained_fsub: 1504 case Intrinsic::experimental_constrained_fmul: 1505 case Intrinsic::experimental_constrained_fdiv: 1506 case Intrinsic::experimental_constrained_frem: 1507 case Intrinsic::experimental_constrained_ceil: 1508 case Intrinsic::experimental_constrained_floor: 1509 case Intrinsic::experimental_constrained_round: 1510 case Intrinsic::experimental_constrained_roundeven: 1511 case Intrinsic::experimental_constrained_trunc: 1512 case Intrinsic::experimental_constrained_nearbyint: 1513 case Intrinsic::experimental_constrained_rint: 1514 return true; 1515 default: 1516 return false; 1517 case Intrinsic::not_intrinsic: break; 1518 } 1519 1520 if (!F->hasName() || Call->isStrictFP()) 1521 return false; 1522 1523 // In these cases, the check of the length is required. We don't want to 1524 // return true for a name like "cos\0blah" which strcmp would return equal to 1525 // "cos", but has length 8. 1526 StringRef Name = F->getName(); 1527 switch (Name[0]) { 1528 default: 1529 return false; 1530 case 'a': 1531 return Name == "acos" || Name == "acosf" || 1532 Name == "asin" || Name == "asinf" || 1533 Name == "atan" || Name == "atanf" || 1534 Name == "atan2" || Name == "atan2f"; 1535 case 'c': 1536 return Name == "ceil" || Name == "ceilf" || 1537 Name == "cos" || Name == "cosf" || 1538 Name == "cosh" || Name == "coshf"; 1539 case 'e': 1540 return Name == "exp" || Name == "expf" || 1541 Name == "exp2" || Name == "exp2f"; 1542 case 'f': 1543 return Name == "fabs" || Name == "fabsf" || 1544 Name == "floor" || Name == "floorf" || 1545 Name == "fmod" || Name == "fmodf"; 1546 case 'l': 1547 return Name == "log" || Name == "logf" || 1548 Name == "log2" || Name == "log2f" || 1549 Name == "log10" || Name == "log10f"; 1550 case 'n': 1551 return Name == "nearbyint" || Name == "nearbyintf"; 1552 case 'p': 1553 return Name == "pow" || Name == "powf"; 1554 case 'r': 1555 return Name == "remainder" || Name == "remainderf" || 1556 Name == "rint" || Name == "rintf" || 1557 Name == "round" || Name == "roundf"; 1558 case 's': 1559 return Name == "sin" || Name == "sinf" || 1560 Name == "sinh" || Name == "sinhf" || 1561 Name == "sqrt" || Name == "sqrtf"; 1562 case 't': 1563 return Name == "tan" || Name == "tanf" || 1564 Name == "tanh" || Name == "tanhf" || 1565 Name == "trunc" || Name == "truncf"; 1566 case '_': 1567 // Check for various function names that get used for the math functions 1568 // when the header files are preprocessed with the macro 1569 // __FINITE_MATH_ONLY__ enabled. 1570 // The '12' here is the length of the shortest name that can match. 1571 // We need to check the size before looking at Name[1] and Name[2] 1572 // so we may as well check a limit that will eliminate mismatches. 1573 if (Name.size() < 12 || Name[1] != '_') 1574 return false; 1575 switch (Name[2]) { 1576 default: 1577 return false; 1578 case 'a': 1579 return Name == "__acos_finite" || Name == "__acosf_finite" || 1580 Name == "__asin_finite" || Name == "__asinf_finite" || 1581 Name == "__atan2_finite" || Name == "__atan2f_finite"; 1582 case 'c': 1583 return Name == "__cosh_finite" || Name == "__coshf_finite"; 1584 case 'e': 1585 return Name == "__exp_finite" || Name == "__expf_finite" || 1586 Name == "__exp2_finite" || Name == "__exp2f_finite"; 1587 case 'l': 1588 return Name == "__log_finite" || Name == "__logf_finite" || 1589 Name == "__log10_finite" || Name == "__log10f_finite"; 1590 case 'p': 1591 return Name == "__pow_finite" || Name == "__powf_finite"; 1592 case 's': 1593 return Name == "__sinh_finite" || Name == "__sinhf_finite"; 1594 } 1595 } 1596 } 1597 1598 namespace { 1599 1600 Constant *GetConstantFoldFPValue(double V, Type *Ty) { 1601 if (Ty->isHalfTy() || Ty->isFloatTy()) { 1602 APFloat APF(V); 1603 bool unused; 1604 APF.convert(Ty->getFltSemantics(), APFloat::rmNearestTiesToEven, &unused); 1605 return ConstantFP::get(Ty->getContext(), APF); 1606 } 1607 if (Ty->isDoubleTy()) 1608 return ConstantFP::get(Ty->getContext(), APFloat(V)); 1609 llvm_unreachable("Can only constant fold half/float/double"); 1610 } 1611 1612 /// Clear the floating-point exception state. 1613 inline void llvm_fenv_clearexcept() { 1614 #if defined(HAVE_FENV_H) && HAVE_DECL_FE_ALL_EXCEPT 1615 feclearexcept(FE_ALL_EXCEPT); 1616 #endif 1617 errno = 0; 1618 } 1619 1620 /// Test if a floating-point exception was raised. 1621 inline bool llvm_fenv_testexcept() { 1622 int errno_val = errno; 1623 if (errno_val == ERANGE || errno_val == EDOM) 1624 return true; 1625 #if defined(HAVE_FENV_H) && HAVE_DECL_FE_ALL_EXCEPT && HAVE_DECL_FE_INEXACT 1626 if (fetestexcept(FE_ALL_EXCEPT & ~FE_INEXACT)) 1627 return true; 1628 #endif 1629 return false; 1630 } 1631 1632 Constant *ConstantFoldFP(double (*NativeFP)(double), const APFloat &V, 1633 Type *Ty) { 1634 llvm_fenv_clearexcept(); 1635 double Result = NativeFP(V.convertToDouble()); 1636 if (llvm_fenv_testexcept()) { 1637 llvm_fenv_clearexcept(); 1638 return nullptr; 1639 } 1640 1641 return GetConstantFoldFPValue(Result, Ty); 1642 } 1643 1644 Constant *ConstantFoldBinaryFP(double (*NativeFP)(double, double), 1645 const APFloat &V, const APFloat &W, Type *Ty) { 1646 llvm_fenv_clearexcept(); 1647 double Result = NativeFP(V.convertToDouble(), W.convertToDouble()); 1648 if (llvm_fenv_testexcept()) { 1649 llvm_fenv_clearexcept(); 1650 return nullptr; 1651 } 1652 1653 return GetConstantFoldFPValue(Result, Ty); 1654 } 1655 1656 Constant *constantFoldVectorReduce(Intrinsic::ID IID, Constant *Op) { 1657 FixedVectorType *VT = dyn_cast<FixedVectorType>(Op->getType()); 1658 if (!VT) 1659 return nullptr; 1660 1661 // This isn't strictly necessary, but handle the special/common case of zero: 1662 // all integer reductions of a zero input produce zero. 1663 if (isa<ConstantAggregateZero>(Op)) 1664 return ConstantInt::get(VT->getElementType(), 0); 1665 1666 // This is the same as the underlying binops - poison propagates. 1667 if (isa<PoisonValue>(Op) || Op->containsPoisonElement()) 1668 return PoisonValue::get(VT->getElementType()); 1669 1670 // TODO: Handle undef. 1671 if (!isa<ConstantVector>(Op) && !isa<ConstantDataVector>(Op)) 1672 return nullptr; 1673 1674 auto *EltC = dyn_cast<ConstantInt>(Op->getAggregateElement(0U)); 1675 if (!EltC) 1676 return nullptr; 1677 1678 APInt Acc = EltC->getValue(); 1679 for (unsigned I = 1, E = VT->getNumElements(); I != E; I++) { 1680 if (!(EltC = dyn_cast<ConstantInt>(Op->getAggregateElement(I)))) 1681 return nullptr; 1682 const APInt &X = EltC->getValue(); 1683 switch (IID) { 1684 case Intrinsic::vector_reduce_add: 1685 Acc = Acc + X; 1686 break; 1687 case Intrinsic::vector_reduce_mul: 1688 Acc = Acc * X; 1689 break; 1690 case Intrinsic::vector_reduce_and: 1691 Acc = Acc & X; 1692 break; 1693 case Intrinsic::vector_reduce_or: 1694 Acc = Acc | X; 1695 break; 1696 case Intrinsic::vector_reduce_xor: 1697 Acc = Acc ^ X; 1698 break; 1699 case Intrinsic::vector_reduce_smin: 1700 Acc = APIntOps::smin(Acc, X); 1701 break; 1702 case Intrinsic::vector_reduce_smax: 1703 Acc = APIntOps::smax(Acc, X); 1704 break; 1705 case Intrinsic::vector_reduce_umin: 1706 Acc = APIntOps::umin(Acc, X); 1707 break; 1708 case Intrinsic::vector_reduce_umax: 1709 Acc = APIntOps::umax(Acc, X); 1710 break; 1711 } 1712 } 1713 1714 return ConstantInt::get(Op->getContext(), Acc); 1715 } 1716 1717 /// Attempt to fold an SSE floating point to integer conversion of a constant 1718 /// floating point. If roundTowardZero is false, the default IEEE rounding is 1719 /// used (toward nearest, ties to even). This matches the behavior of the 1720 /// non-truncating SSE instructions in the default rounding mode. The desired 1721 /// integer type Ty is used to select how many bits are available for the 1722 /// result. Returns null if the conversion cannot be performed, otherwise 1723 /// returns the Constant value resulting from the conversion. 1724 Constant *ConstantFoldSSEConvertToInt(const APFloat &Val, bool roundTowardZero, 1725 Type *Ty, bool IsSigned) { 1726 // All of these conversion intrinsics form an integer of at most 64bits. 1727 unsigned ResultWidth = Ty->getIntegerBitWidth(); 1728 assert(ResultWidth <= 64 && 1729 "Can only constant fold conversions to 64 and 32 bit ints"); 1730 1731 uint64_t UIntVal; 1732 bool isExact = false; 1733 APFloat::roundingMode mode = roundTowardZero? APFloat::rmTowardZero 1734 : APFloat::rmNearestTiesToEven; 1735 APFloat::opStatus status = 1736 Val.convertToInteger(makeMutableArrayRef(UIntVal), ResultWidth, 1737 IsSigned, mode, &isExact); 1738 if (status != APFloat::opOK && 1739 (!roundTowardZero || status != APFloat::opInexact)) 1740 return nullptr; 1741 return ConstantInt::get(Ty, UIntVal, IsSigned); 1742 } 1743 1744 double getValueAsDouble(ConstantFP *Op) { 1745 Type *Ty = Op->getType(); 1746 1747 if (Ty->isBFloatTy() || Ty->isHalfTy() || Ty->isFloatTy() || Ty->isDoubleTy()) 1748 return Op->getValueAPF().convertToDouble(); 1749 1750 bool unused; 1751 APFloat APF = Op->getValueAPF(); 1752 APF.convert(APFloat::IEEEdouble(), APFloat::rmNearestTiesToEven, &unused); 1753 return APF.convertToDouble(); 1754 } 1755 1756 static bool getConstIntOrUndef(Value *Op, const APInt *&C) { 1757 if (auto *CI = dyn_cast<ConstantInt>(Op)) { 1758 C = &CI->getValue(); 1759 return true; 1760 } 1761 if (isa<UndefValue>(Op)) { 1762 C = nullptr; 1763 return true; 1764 } 1765 return false; 1766 } 1767 1768 /// Checks if the given intrinsic call, which evaluates to constant, is allowed 1769 /// to be folded. 1770 /// 1771 /// \param CI Constrained intrinsic call. 1772 /// \param St Exception flags raised during constant evaluation. 1773 static bool mayFoldConstrained(ConstrainedFPIntrinsic *CI, 1774 APFloat::opStatus St) { 1775 Optional<RoundingMode> ORM = CI->getRoundingMode(); 1776 Optional<fp::ExceptionBehavior> EB = CI->getExceptionBehavior(); 1777 1778 // If the operation does not change exception status flags, it is safe 1779 // to fold. 1780 if (St == APFloat::opStatus::opOK) { 1781 // When FP exceptions are not ignored, intrinsic call will not be 1782 // eliminated, because it is considered as having side effect. But we 1783 // know that its evaluation does not raise exceptions, so side effect 1784 // is absent. To allow removing the call, mark it as not accessing memory. 1785 if (EB && *EB != fp::ExceptionBehavior::ebIgnore) 1786 CI->addFnAttr(Attribute::ReadNone); 1787 return true; 1788 } 1789 1790 // If evaluation raised FP exception, the result can depend on rounding 1791 // mode. If the latter is unknown, folding is not possible. 1792 if (!ORM || *ORM == RoundingMode::Dynamic) 1793 return false; 1794 1795 // If FP exceptions are ignored, fold the call, even if such exception is 1796 // raised. 1797 if (!EB || *EB != fp::ExceptionBehavior::ebStrict) 1798 return true; 1799 1800 // Leave the calculation for runtime so that exception flags be correctly set 1801 // in hardware. 1802 return false; 1803 } 1804 1805 /// Returns the rounding mode that should be used for constant evaluation. 1806 static RoundingMode 1807 getEvaluationRoundingMode(const ConstrainedFPIntrinsic *CI) { 1808 Optional<RoundingMode> ORM = CI->getRoundingMode(); 1809 if (!ORM || *ORM == RoundingMode::Dynamic) 1810 // Even if the rounding mode is unknown, try evaluating the operation. 1811 // If it does not raise inexact exception, rounding was not applied, 1812 // so the result is exact and does not depend on rounding mode. Whether 1813 // other FP exceptions are raised, it does not depend on rounding mode. 1814 return RoundingMode::NearestTiesToEven; 1815 return *ORM; 1816 } 1817 1818 static Constant *ConstantFoldScalarCall1(StringRef Name, 1819 Intrinsic::ID IntrinsicID, 1820 Type *Ty, 1821 ArrayRef<Constant *> Operands, 1822 const TargetLibraryInfo *TLI, 1823 const CallBase *Call) { 1824 assert(Operands.size() == 1 && "Wrong number of operands."); 1825 1826 if (IntrinsicID == Intrinsic::is_constant) { 1827 // We know we have a "Constant" argument. But we want to only 1828 // return true for manifest constants, not those that depend on 1829 // constants with unknowable values, e.g. GlobalValue or BlockAddress. 1830 if (Operands[0]->isManifestConstant()) 1831 return ConstantInt::getTrue(Ty->getContext()); 1832 return nullptr; 1833 } 1834 if (isa<UndefValue>(Operands[0])) { 1835 // cosine(arg) is between -1 and 1. cosine(invalid arg) is NaN. 1836 // ctpop() is between 0 and bitwidth, pick 0 for undef. 1837 // fptoui.sat and fptosi.sat can always fold to zero (for a zero input). 1838 if (IntrinsicID == Intrinsic::cos || 1839 IntrinsicID == Intrinsic::ctpop || 1840 IntrinsicID == Intrinsic::fptoui_sat || 1841 IntrinsicID == Intrinsic::fptosi_sat) 1842 return Constant::getNullValue(Ty); 1843 if (IntrinsicID == Intrinsic::bswap || 1844 IntrinsicID == Intrinsic::bitreverse || 1845 IntrinsicID == Intrinsic::launder_invariant_group || 1846 IntrinsicID == Intrinsic::strip_invariant_group) 1847 return Operands[0]; 1848 } 1849 1850 if (isa<ConstantPointerNull>(Operands[0])) { 1851 // launder(null) == null == strip(null) iff in addrspace 0 1852 if (IntrinsicID == Intrinsic::launder_invariant_group || 1853 IntrinsicID == Intrinsic::strip_invariant_group) { 1854 // If instruction is not yet put in a basic block (e.g. when cloning 1855 // a function during inlining), Call's caller may not be available. 1856 // So check Call's BB first before querying Call->getCaller. 1857 const Function *Caller = 1858 Call->getParent() ? Call->getCaller() : nullptr; 1859 if (Caller && 1860 !NullPointerIsDefined( 1861 Caller, Operands[0]->getType()->getPointerAddressSpace())) { 1862 return Operands[0]; 1863 } 1864 return nullptr; 1865 } 1866 } 1867 1868 if (auto *Op = dyn_cast<ConstantFP>(Operands[0])) { 1869 if (IntrinsicID == Intrinsic::convert_to_fp16) { 1870 APFloat Val(Op->getValueAPF()); 1871 1872 bool lost = false; 1873 Val.convert(APFloat::IEEEhalf(), APFloat::rmNearestTiesToEven, &lost); 1874 1875 return ConstantInt::get(Ty->getContext(), Val.bitcastToAPInt()); 1876 } 1877 1878 APFloat U = Op->getValueAPF(); 1879 1880 if (IntrinsicID == Intrinsic::wasm_trunc_signed || 1881 IntrinsicID == Intrinsic::wasm_trunc_unsigned) { 1882 bool Signed = IntrinsicID == Intrinsic::wasm_trunc_signed; 1883 1884 if (U.isNaN()) 1885 return nullptr; 1886 1887 unsigned Width = Ty->getIntegerBitWidth(); 1888 APSInt Int(Width, !Signed); 1889 bool IsExact = false; 1890 APFloat::opStatus Status = 1891 U.convertToInteger(Int, APFloat::rmTowardZero, &IsExact); 1892 1893 if (Status == APFloat::opOK || Status == APFloat::opInexact) 1894 return ConstantInt::get(Ty, Int); 1895 1896 return nullptr; 1897 } 1898 1899 if (IntrinsicID == Intrinsic::fptoui_sat || 1900 IntrinsicID == Intrinsic::fptosi_sat) { 1901 // convertToInteger() already has the desired saturation semantics. 1902 APSInt Int(Ty->getIntegerBitWidth(), 1903 IntrinsicID == Intrinsic::fptoui_sat); 1904 bool IsExact; 1905 U.convertToInteger(Int, APFloat::rmTowardZero, &IsExact); 1906 return ConstantInt::get(Ty, Int); 1907 } 1908 1909 if (!Ty->isHalfTy() && !Ty->isFloatTy() && !Ty->isDoubleTy()) 1910 return nullptr; 1911 1912 // Use internal versions of these intrinsics. 1913 1914 if (IntrinsicID == Intrinsic::nearbyint || IntrinsicID == Intrinsic::rint) { 1915 U.roundToIntegral(APFloat::rmNearestTiesToEven); 1916 return ConstantFP::get(Ty->getContext(), U); 1917 } 1918 1919 if (IntrinsicID == Intrinsic::round) { 1920 U.roundToIntegral(APFloat::rmNearestTiesToAway); 1921 return ConstantFP::get(Ty->getContext(), U); 1922 } 1923 1924 if (IntrinsicID == Intrinsic::roundeven) { 1925 U.roundToIntegral(APFloat::rmNearestTiesToEven); 1926 return ConstantFP::get(Ty->getContext(), U); 1927 } 1928 1929 if (IntrinsicID == Intrinsic::ceil) { 1930 U.roundToIntegral(APFloat::rmTowardPositive); 1931 return ConstantFP::get(Ty->getContext(), U); 1932 } 1933 1934 if (IntrinsicID == Intrinsic::floor) { 1935 U.roundToIntegral(APFloat::rmTowardNegative); 1936 return ConstantFP::get(Ty->getContext(), U); 1937 } 1938 1939 if (IntrinsicID == Intrinsic::trunc) { 1940 U.roundToIntegral(APFloat::rmTowardZero); 1941 return ConstantFP::get(Ty->getContext(), U); 1942 } 1943 1944 if (IntrinsicID == Intrinsic::fabs) { 1945 U.clearSign(); 1946 return ConstantFP::get(Ty->getContext(), U); 1947 } 1948 1949 if (IntrinsicID == Intrinsic::amdgcn_fract) { 1950 // The v_fract instruction behaves like the OpenCL spec, which defines 1951 // fract(x) as fmin(x - floor(x), 0x1.fffffep-1f): "The min() operator is 1952 // there to prevent fract(-small) from returning 1.0. It returns the 1953 // largest positive floating-point number less than 1.0." 1954 APFloat FloorU(U); 1955 FloorU.roundToIntegral(APFloat::rmTowardNegative); 1956 APFloat FractU(U - FloorU); 1957 APFloat AlmostOne(U.getSemantics(), 1); 1958 AlmostOne.next(/*nextDown*/ true); 1959 return ConstantFP::get(Ty->getContext(), minimum(FractU, AlmostOne)); 1960 } 1961 1962 // Rounding operations (floor, trunc, ceil, round and nearbyint) do not 1963 // raise FP exceptions, unless the argument is signaling NaN. 1964 1965 Optional<APFloat::roundingMode> RM; 1966 switch (IntrinsicID) { 1967 default: 1968 break; 1969 case Intrinsic::experimental_constrained_nearbyint: 1970 case Intrinsic::experimental_constrained_rint: { 1971 auto CI = cast<ConstrainedFPIntrinsic>(Call); 1972 RM = CI->getRoundingMode(); 1973 if (!RM || RM.getValue() == RoundingMode::Dynamic) 1974 return nullptr; 1975 break; 1976 } 1977 case Intrinsic::experimental_constrained_round: 1978 RM = APFloat::rmNearestTiesToAway; 1979 break; 1980 case Intrinsic::experimental_constrained_ceil: 1981 RM = APFloat::rmTowardPositive; 1982 break; 1983 case Intrinsic::experimental_constrained_floor: 1984 RM = APFloat::rmTowardNegative; 1985 break; 1986 case Intrinsic::experimental_constrained_trunc: 1987 RM = APFloat::rmTowardZero; 1988 break; 1989 } 1990 if (RM) { 1991 auto CI = cast<ConstrainedFPIntrinsic>(Call); 1992 if (U.isFinite()) { 1993 APFloat::opStatus St = U.roundToIntegral(*RM); 1994 if (IntrinsicID == Intrinsic::experimental_constrained_rint && 1995 St == APFloat::opInexact) { 1996 Optional<fp::ExceptionBehavior> EB = CI->getExceptionBehavior(); 1997 if (EB && *EB == fp::ebStrict) 1998 return nullptr; 1999 } 2000 } else if (U.isSignaling()) { 2001 Optional<fp::ExceptionBehavior> EB = CI->getExceptionBehavior(); 2002 if (EB && *EB != fp::ebIgnore) 2003 return nullptr; 2004 U = APFloat::getQNaN(U.getSemantics()); 2005 } 2006 return ConstantFP::get(Ty->getContext(), U); 2007 } 2008 2009 /// We only fold functions with finite arguments. Folding NaN and inf is 2010 /// likely to be aborted with an exception anyway, and some host libms 2011 /// have known errors raising exceptions. 2012 if (!U.isFinite()) 2013 return nullptr; 2014 2015 /// Currently APFloat versions of these functions do not exist, so we use 2016 /// the host native double versions. Float versions are not called 2017 /// directly but for all these it is true (float)(f((double)arg)) == 2018 /// f(arg). Long double not supported yet. 2019 const APFloat &APF = Op->getValueAPF(); 2020 2021 switch (IntrinsicID) { 2022 default: break; 2023 case Intrinsic::log: 2024 return ConstantFoldFP(log, APF, Ty); 2025 case Intrinsic::log2: 2026 // TODO: What about hosts that lack a C99 library? 2027 return ConstantFoldFP(Log2, APF, Ty); 2028 case Intrinsic::log10: 2029 // TODO: What about hosts that lack a C99 library? 2030 return ConstantFoldFP(log10, APF, Ty); 2031 case Intrinsic::exp: 2032 return ConstantFoldFP(exp, APF, Ty); 2033 case Intrinsic::exp2: 2034 // Fold exp2(x) as pow(2, x), in case the host lacks a C99 library. 2035 return ConstantFoldBinaryFP(pow, APFloat(2.0), APF, Ty); 2036 case Intrinsic::sin: 2037 return ConstantFoldFP(sin, APF, Ty); 2038 case Intrinsic::cos: 2039 return ConstantFoldFP(cos, APF, Ty); 2040 case Intrinsic::sqrt: 2041 return ConstantFoldFP(sqrt, APF, Ty); 2042 case Intrinsic::amdgcn_cos: 2043 case Intrinsic::amdgcn_sin: { 2044 double V = getValueAsDouble(Op); 2045 if (V < -256.0 || V > 256.0) 2046 // The gfx8 and gfx9 architectures handle arguments outside the range 2047 // [-256, 256] differently. This should be a rare case so bail out 2048 // rather than trying to handle the difference. 2049 return nullptr; 2050 bool IsCos = IntrinsicID == Intrinsic::amdgcn_cos; 2051 double V4 = V * 4.0; 2052 if (V4 == floor(V4)) { 2053 // Force exact results for quarter-integer inputs. 2054 const double SinVals[4] = { 0.0, 1.0, 0.0, -1.0 }; 2055 V = SinVals[((int)V4 + (IsCos ? 1 : 0)) & 3]; 2056 } else { 2057 if (IsCos) 2058 V = cos(V * 2.0 * numbers::pi); 2059 else 2060 V = sin(V * 2.0 * numbers::pi); 2061 } 2062 return GetConstantFoldFPValue(V, Ty); 2063 } 2064 } 2065 2066 if (!TLI) 2067 return nullptr; 2068 2069 LibFunc Func = NotLibFunc; 2070 if (!TLI->getLibFunc(Name, Func)) 2071 return nullptr; 2072 2073 switch (Func) { 2074 default: 2075 break; 2076 case LibFunc_acos: 2077 case LibFunc_acosf: 2078 case LibFunc_acos_finite: 2079 case LibFunc_acosf_finite: 2080 if (TLI->has(Func)) 2081 return ConstantFoldFP(acos, APF, Ty); 2082 break; 2083 case LibFunc_asin: 2084 case LibFunc_asinf: 2085 case LibFunc_asin_finite: 2086 case LibFunc_asinf_finite: 2087 if (TLI->has(Func)) 2088 return ConstantFoldFP(asin, APF, Ty); 2089 break; 2090 case LibFunc_atan: 2091 case LibFunc_atanf: 2092 if (TLI->has(Func)) 2093 return ConstantFoldFP(atan, APF, Ty); 2094 break; 2095 case LibFunc_ceil: 2096 case LibFunc_ceilf: 2097 if (TLI->has(Func)) { 2098 U.roundToIntegral(APFloat::rmTowardPositive); 2099 return ConstantFP::get(Ty->getContext(), U); 2100 } 2101 break; 2102 case LibFunc_cos: 2103 case LibFunc_cosf: 2104 if (TLI->has(Func)) 2105 return ConstantFoldFP(cos, APF, Ty); 2106 break; 2107 case LibFunc_cosh: 2108 case LibFunc_coshf: 2109 case LibFunc_cosh_finite: 2110 case LibFunc_coshf_finite: 2111 if (TLI->has(Func)) 2112 return ConstantFoldFP(cosh, APF, Ty); 2113 break; 2114 case LibFunc_exp: 2115 case LibFunc_expf: 2116 case LibFunc_exp_finite: 2117 case LibFunc_expf_finite: 2118 if (TLI->has(Func)) 2119 return ConstantFoldFP(exp, APF, Ty); 2120 break; 2121 case LibFunc_exp2: 2122 case LibFunc_exp2f: 2123 case LibFunc_exp2_finite: 2124 case LibFunc_exp2f_finite: 2125 if (TLI->has(Func)) 2126 // Fold exp2(x) as pow(2, x), in case the host lacks a C99 library. 2127 return ConstantFoldBinaryFP(pow, APFloat(2.0), APF, Ty); 2128 break; 2129 case LibFunc_fabs: 2130 case LibFunc_fabsf: 2131 if (TLI->has(Func)) { 2132 U.clearSign(); 2133 return ConstantFP::get(Ty->getContext(), U); 2134 } 2135 break; 2136 case LibFunc_floor: 2137 case LibFunc_floorf: 2138 if (TLI->has(Func)) { 2139 U.roundToIntegral(APFloat::rmTowardNegative); 2140 return ConstantFP::get(Ty->getContext(), U); 2141 } 2142 break; 2143 case LibFunc_log: 2144 case LibFunc_logf: 2145 case LibFunc_log_finite: 2146 case LibFunc_logf_finite: 2147 if (!APF.isNegative() && !APF.isZero() && TLI->has(Func)) 2148 return ConstantFoldFP(log, APF, Ty); 2149 break; 2150 case LibFunc_log2: 2151 case LibFunc_log2f: 2152 case LibFunc_log2_finite: 2153 case LibFunc_log2f_finite: 2154 if (!APF.isNegative() && !APF.isZero() && TLI->has(Func)) 2155 // TODO: What about hosts that lack a C99 library? 2156 return ConstantFoldFP(Log2, APF, Ty); 2157 break; 2158 case LibFunc_log10: 2159 case LibFunc_log10f: 2160 case LibFunc_log10_finite: 2161 case LibFunc_log10f_finite: 2162 if (!APF.isNegative() && !APF.isZero() && TLI->has(Func)) 2163 // TODO: What about hosts that lack a C99 library? 2164 return ConstantFoldFP(log10, APF, Ty); 2165 break; 2166 case LibFunc_nearbyint: 2167 case LibFunc_nearbyintf: 2168 case LibFunc_rint: 2169 case LibFunc_rintf: 2170 if (TLI->has(Func)) { 2171 U.roundToIntegral(APFloat::rmNearestTiesToEven); 2172 return ConstantFP::get(Ty->getContext(), U); 2173 } 2174 break; 2175 case LibFunc_round: 2176 case LibFunc_roundf: 2177 if (TLI->has(Func)) { 2178 U.roundToIntegral(APFloat::rmNearestTiesToAway); 2179 return ConstantFP::get(Ty->getContext(), U); 2180 } 2181 break; 2182 case LibFunc_sin: 2183 case LibFunc_sinf: 2184 if (TLI->has(Func)) 2185 return ConstantFoldFP(sin, APF, Ty); 2186 break; 2187 case LibFunc_sinh: 2188 case LibFunc_sinhf: 2189 case LibFunc_sinh_finite: 2190 case LibFunc_sinhf_finite: 2191 if (TLI->has(Func)) 2192 return ConstantFoldFP(sinh, APF, Ty); 2193 break; 2194 case LibFunc_sqrt: 2195 case LibFunc_sqrtf: 2196 if (!APF.isNegative() && TLI->has(Func)) 2197 return ConstantFoldFP(sqrt, APF, Ty); 2198 break; 2199 case LibFunc_tan: 2200 case LibFunc_tanf: 2201 if (TLI->has(Func)) 2202 return ConstantFoldFP(tan, APF, Ty); 2203 break; 2204 case LibFunc_tanh: 2205 case LibFunc_tanhf: 2206 if (TLI->has(Func)) 2207 return ConstantFoldFP(tanh, APF, Ty); 2208 break; 2209 case LibFunc_trunc: 2210 case LibFunc_truncf: 2211 if (TLI->has(Func)) { 2212 U.roundToIntegral(APFloat::rmTowardZero); 2213 return ConstantFP::get(Ty->getContext(), U); 2214 } 2215 break; 2216 } 2217 return nullptr; 2218 } 2219 2220 if (auto *Op = dyn_cast<ConstantInt>(Operands[0])) { 2221 switch (IntrinsicID) { 2222 case Intrinsic::bswap: 2223 return ConstantInt::get(Ty->getContext(), Op->getValue().byteSwap()); 2224 case Intrinsic::ctpop: 2225 return ConstantInt::get(Ty, Op->getValue().countPopulation()); 2226 case Intrinsic::bitreverse: 2227 return ConstantInt::get(Ty->getContext(), Op->getValue().reverseBits()); 2228 case Intrinsic::convert_from_fp16: { 2229 APFloat Val(APFloat::IEEEhalf(), Op->getValue()); 2230 2231 bool lost = false; 2232 APFloat::opStatus status = Val.convert( 2233 Ty->getFltSemantics(), APFloat::rmNearestTiesToEven, &lost); 2234 2235 // Conversion is always precise. 2236 (void)status; 2237 assert(status == APFloat::opOK && !lost && 2238 "Precision lost during fp16 constfolding"); 2239 2240 return ConstantFP::get(Ty->getContext(), Val); 2241 } 2242 default: 2243 return nullptr; 2244 } 2245 } 2246 2247 switch (IntrinsicID) { 2248 default: break; 2249 case Intrinsic::vector_reduce_add: 2250 case Intrinsic::vector_reduce_mul: 2251 case Intrinsic::vector_reduce_and: 2252 case Intrinsic::vector_reduce_or: 2253 case Intrinsic::vector_reduce_xor: 2254 case Intrinsic::vector_reduce_smin: 2255 case Intrinsic::vector_reduce_smax: 2256 case Intrinsic::vector_reduce_umin: 2257 case Intrinsic::vector_reduce_umax: 2258 if (Constant *C = constantFoldVectorReduce(IntrinsicID, Operands[0])) 2259 return C; 2260 break; 2261 } 2262 2263 // Support ConstantVector in case we have an Undef in the top. 2264 if (isa<ConstantVector>(Operands[0]) || 2265 isa<ConstantDataVector>(Operands[0])) { 2266 auto *Op = cast<Constant>(Operands[0]); 2267 switch (IntrinsicID) { 2268 default: break; 2269 case Intrinsic::x86_sse_cvtss2si: 2270 case Intrinsic::x86_sse_cvtss2si64: 2271 case Intrinsic::x86_sse2_cvtsd2si: 2272 case Intrinsic::x86_sse2_cvtsd2si64: 2273 if (ConstantFP *FPOp = 2274 dyn_cast_or_null<ConstantFP>(Op->getAggregateElement(0U))) 2275 return ConstantFoldSSEConvertToInt(FPOp->getValueAPF(), 2276 /*roundTowardZero=*/false, Ty, 2277 /*IsSigned*/true); 2278 break; 2279 case Intrinsic::x86_sse_cvttss2si: 2280 case Intrinsic::x86_sse_cvttss2si64: 2281 case Intrinsic::x86_sse2_cvttsd2si: 2282 case Intrinsic::x86_sse2_cvttsd2si64: 2283 if (ConstantFP *FPOp = 2284 dyn_cast_or_null<ConstantFP>(Op->getAggregateElement(0U))) 2285 return ConstantFoldSSEConvertToInt(FPOp->getValueAPF(), 2286 /*roundTowardZero=*/true, Ty, 2287 /*IsSigned*/true); 2288 break; 2289 } 2290 } 2291 2292 return nullptr; 2293 } 2294 2295 static Constant *ConstantFoldScalarCall2(StringRef Name, 2296 Intrinsic::ID IntrinsicID, 2297 Type *Ty, 2298 ArrayRef<Constant *> Operands, 2299 const TargetLibraryInfo *TLI, 2300 const CallBase *Call) { 2301 assert(Operands.size() == 2 && "Wrong number of operands."); 2302 2303 if (Ty->isFloatingPointTy()) { 2304 // TODO: We should have undef handling for all of the FP intrinsics that 2305 // are attempted to be folded in this function. 2306 bool IsOp0Undef = isa<UndefValue>(Operands[0]); 2307 bool IsOp1Undef = isa<UndefValue>(Operands[1]); 2308 switch (IntrinsicID) { 2309 case Intrinsic::maxnum: 2310 case Intrinsic::minnum: 2311 case Intrinsic::maximum: 2312 case Intrinsic::minimum: 2313 // If one argument is undef, return the other argument. 2314 if (IsOp0Undef) 2315 return Operands[1]; 2316 if (IsOp1Undef) 2317 return Operands[0]; 2318 break; 2319 } 2320 } 2321 2322 if (const auto *Op1 = dyn_cast<ConstantFP>(Operands[0])) { 2323 if (!Ty->isFloatingPointTy()) 2324 return nullptr; 2325 const APFloat &Op1V = Op1->getValueAPF(); 2326 2327 if (const auto *Op2 = dyn_cast<ConstantFP>(Operands[1])) { 2328 if (Op2->getType() != Op1->getType()) 2329 return nullptr; 2330 const APFloat &Op2V = Op2->getValueAPF(); 2331 2332 if (const auto *ConstrIntr = dyn_cast<ConstrainedFPIntrinsic>(Call)) { 2333 RoundingMode RM = getEvaluationRoundingMode(ConstrIntr); 2334 APFloat Res = Op1V; 2335 APFloat::opStatus St; 2336 switch (IntrinsicID) { 2337 default: 2338 return nullptr; 2339 case Intrinsic::experimental_constrained_fadd: 2340 St = Res.add(Op2V, RM); 2341 break; 2342 case Intrinsic::experimental_constrained_fsub: 2343 St = Res.subtract(Op2V, RM); 2344 break; 2345 case Intrinsic::experimental_constrained_fmul: 2346 St = Res.multiply(Op2V, RM); 2347 break; 2348 case Intrinsic::experimental_constrained_fdiv: 2349 St = Res.divide(Op2V, RM); 2350 break; 2351 case Intrinsic::experimental_constrained_frem: 2352 St = Res.mod(Op2V); 2353 break; 2354 } 2355 if (mayFoldConstrained(const_cast<ConstrainedFPIntrinsic *>(ConstrIntr), 2356 St)) 2357 return ConstantFP::get(Ty->getContext(), Res); 2358 return nullptr; 2359 } 2360 2361 switch (IntrinsicID) { 2362 default: 2363 break; 2364 case Intrinsic::copysign: 2365 return ConstantFP::get(Ty->getContext(), APFloat::copySign(Op1V, Op2V)); 2366 case Intrinsic::minnum: 2367 return ConstantFP::get(Ty->getContext(), minnum(Op1V, Op2V)); 2368 case Intrinsic::maxnum: 2369 return ConstantFP::get(Ty->getContext(), maxnum(Op1V, Op2V)); 2370 case Intrinsic::minimum: 2371 return ConstantFP::get(Ty->getContext(), minimum(Op1V, Op2V)); 2372 case Intrinsic::maximum: 2373 return ConstantFP::get(Ty->getContext(), maximum(Op1V, Op2V)); 2374 } 2375 2376 if (!Ty->isHalfTy() && !Ty->isFloatTy() && !Ty->isDoubleTy()) 2377 return nullptr; 2378 2379 switch (IntrinsicID) { 2380 default: 2381 break; 2382 case Intrinsic::pow: 2383 return ConstantFoldBinaryFP(pow, Op1V, Op2V, Ty); 2384 case Intrinsic::amdgcn_fmul_legacy: 2385 // The legacy behaviour is that multiplying +/- 0.0 by anything, even 2386 // NaN or infinity, gives +0.0. 2387 if (Op1V.isZero() || Op2V.isZero()) 2388 return ConstantFP::getNullValue(Ty); 2389 return ConstantFP::get(Ty->getContext(), Op1V * Op2V); 2390 } 2391 2392 if (!TLI) 2393 return nullptr; 2394 2395 LibFunc Func = NotLibFunc; 2396 if (!TLI->getLibFunc(Name, Func)) 2397 return nullptr; 2398 2399 switch (Func) { 2400 default: 2401 break; 2402 case LibFunc_pow: 2403 case LibFunc_powf: 2404 case LibFunc_pow_finite: 2405 case LibFunc_powf_finite: 2406 if (TLI->has(Func)) 2407 return ConstantFoldBinaryFP(pow, Op1V, Op2V, Ty); 2408 break; 2409 case LibFunc_fmod: 2410 case LibFunc_fmodf: 2411 if (TLI->has(Func)) { 2412 APFloat V = Op1->getValueAPF(); 2413 if (APFloat::opStatus::opOK == V.mod(Op2->getValueAPF())) 2414 return ConstantFP::get(Ty->getContext(), V); 2415 } 2416 break; 2417 case LibFunc_remainder: 2418 case LibFunc_remainderf: 2419 if (TLI->has(Func)) { 2420 APFloat V = Op1->getValueAPF(); 2421 if (APFloat::opStatus::opOK == V.remainder(Op2->getValueAPF())) 2422 return ConstantFP::get(Ty->getContext(), V); 2423 } 2424 break; 2425 case LibFunc_atan2: 2426 case LibFunc_atan2f: 2427 case LibFunc_atan2_finite: 2428 case LibFunc_atan2f_finite: 2429 if (TLI->has(Func)) 2430 return ConstantFoldBinaryFP(atan2, Op1V, Op2V, Ty); 2431 break; 2432 } 2433 } else if (auto *Op2C = dyn_cast<ConstantInt>(Operands[1])) { 2434 if (!Ty->isHalfTy() && !Ty->isFloatTy() && !Ty->isDoubleTy()) 2435 return nullptr; 2436 if (IntrinsicID == Intrinsic::powi && Ty->isHalfTy()) 2437 return ConstantFP::get( 2438 Ty->getContext(), 2439 APFloat((float)std::pow((float)Op1V.convertToDouble(), 2440 (int)Op2C->getZExtValue()))); 2441 if (IntrinsicID == Intrinsic::powi && Ty->isFloatTy()) 2442 return ConstantFP::get( 2443 Ty->getContext(), 2444 APFloat((float)std::pow((float)Op1V.convertToDouble(), 2445 (int)Op2C->getZExtValue()))); 2446 if (IntrinsicID == Intrinsic::powi && Ty->isDoubleTy()) 2447 return ConstantFP::get( 2448 Ty->getContext(), 2449 APFloat((double)std::pow(Op1V.convertToDouble(), 2450 (int)Op2C->getZExtValue()))); 2451 2452 if (IntrinsicID == Intrinsic::amdgcn_ldexp) { 2453 // FIXME: Should flush denorms depending on FP mode, but that's ignored 2454 // everywhere else. 2455 2456 // scalbn is equivalent to ldexp with float radix 2 2457 APFloat Result = scalbn(Op1->getValueAPF(), Op2C->getSExtValue(), 2458 APFloat::rmNearestTiesToEven); 2459 return ConstantFP::get(Ty->getContext(), Result); 2460 } 2461 } 2462 return nullptr; 2463 } 2464 2465 if (Operands[0]->getType()->isIntegerTy() && 2466 Operands[1]->getType()->isIntegerTy()) { 2467 const APInt *C0, *C1; 2468 if (!getConstIntOrUndef(Operands[0], C0) || 2469 !getConstIntOrUndef(Operands[1], C1)) 2470 return nullptr; 2471 2472 unsigned BitWidth = Ty->getScalarSizeInBits(); 2473 switch (IntrinsicID) { 2474 default: break; 2475 case Intrinsic::smax: 2476 if (!C0 && !C1) 2477 return UndefValue::get(Ty); 2478 if (!C0 || !C1) 2479 return ConstantInt::get(Ty, APInt::getSignedMaxValue(BitWidth)); 2480 return ConstantInt::get(Ty, C0->sgt(*C1) ? *C0 : *C1); 2481 2482 case Intrinsic::smin: 2483 if (!C0 && !C1) 2484 return UndefValue::get(Ty); 2485 if (!C0 || !C1) 2486 return ConstantInt::get(Ty, APInt::getSignedMinValue(BitWidth)); 2487 return ConstantInt::get(Ty, C0->slt(*C1) ? *C0 : *C1); 2488 2489 case Intrinsic::umax: 2490 if (!C0 && !C1) 2491 return UndefValue::get(Ty); 2492 if (!C0 || !C1) 2493 return ConstantInt::get(Ty, APInt::getMaxValue(BitWidth)); 2494 return ConstantInt::get(Ty, C0->ugt(*C1) ? *C0 : *C1); 2495 2496 case Intrinsic::umin: 2497 if (!C0 && !C1) 2498 return UndefValue::get(Ty); 2499 if (!C0 || !C1) 2500 return ConstantInt::get(Ty, APInt::getMinValue(BitWidth)); 2501 return ConstantInt::get(Ty, C0->ult(*C1) ? *C0 : *C1); 2502 2503 case Intrinsic::usub_with_overflow: 2504 case Intrinsic::ssub_with_overflow: 2505 // X - undef -> { 0, false } 2506 // undef - X -> { 0, false } 2507 if (!C0 || !C1) 2508 return Constant::getNullValue(Ty); 2509 LLVM_FALLTHROUGH; 2510 case Intrinsic::uadd_with_overflow: 2511 case Intrinsic::sadd_with_overflow: 2512 // X + undef -> { -1, false } 2513 // undef + x -> { -1, false } 2514 if (!C0 || !C1) { 2515 return ConstantStruct::get( 2516 cast<StructType>(Ty), 2517 {Constant::getAllOnesValue(Ty->getStructElementType(0)), 2518 Constant::getNullValue(Ty->getStructElementType(1))}); 2519 } 2520 LLVM_FALLTHROUGH; 2521 case Intrinsic::smul_with_overflow: 2522 case Intrinsic::umul_with_overflow: { 2523 // undef * X -> { 0, false } 2524 // X * undef -> { 0, false } 2525 if (!C0 || !C1) 2526 return Constant::getNullValue(Ty); 2527 2528 APInt Res; 2529 bool Overflow; 2530 switch (IntrinsicID) { 2531 default: llvm_unreachable("Invalid case"); 2532 case Intrinsic::sadd_with_overflow: 2533 Res = C0->sadd_ov(*C1, Overflow); 2534 break; 2535 case Intrinsic::uadd_with_overflow: 2536 Res = C0->uadd_ov(*C1, Overflow); 2537 break; 2538 case Intrinsic::ssub_with_overflow: 2539 Res = C0->ssub_ov(*C1, Overflow); 2540 break; 2541 case Intrinsic::usub_with_overflow: 2542 Res = C0->usub_ov(*C1, Overflow); 2543 break; 2544 case Intrinsic::smul_with_overflow: 2545 Res = C0->smul_ov(*C1, Overflow); 2546 break; 2547 case Intrinsic::umul_with_overflow: 2548 Res = C0->umul_ov(*C1, Overflow); 2549 break; 2550 } 2551 Constant *Ops[] = { 2552 ConstantInt::get(Ty->getContext(), Res), 2553 ConstantInt::get(Type::getInt1Ty(Ty->getContext()), Overflow) 2554 }; 2555 return ConstantStruct::get(cast<StructType>(Ty), Ops); 2556 } 2557 case Intrinsic::uadd_sat: 2558 case Intrinsic::sadd_sat: 2559 if (!C0 && !C1) 2560 return UndefValue::get(Ty); 2561 if (!C0 || !C1) 2562 return Constant::getAllOnesValue(Ty); 2563 if (IntrinsicID == Intrinsic::uadd_sat) 2564 return ConstantInt::get(Ty, C0->uadd_sat(*C1)); 2565 else 2566 return ConstantInt::get(Ty, C0->sadd_sat(*C1)); 2567 case Intrinsic::usub_sat: 2568 case Intrinsic::ssub_sat: 2569 if (!C0 && !C1) 2570 return UndefValue::get(Ty); 2571 if (!C0 || !C1) 2572 return Constant::getNullValue(Ty); 2573 if (IntrinsicID == Intrinsic::usub_sat) 2574 return ConstantInt::get(Ty, C0->usub_sat(*C1)); 2575 else 2576 return ConstantInt::get(Ty, C0->ssub_sat(*C1)); 2577 case Intrinsic::cttz: 2578 case Intrinsic::ctlz: 2579 assert(C1 && "Must be constant int"); 2580 2581 // cttz(0, 1) and ctlz(0, 1) are undef. 2582 if (C1->isOne() && (!C0 || C0->isZero())) 2583 return UndefValue::get(Ty); 2584 if (!C0) 2585 return Constant::getNullValue(Ty); 2586 if (IntrinsicID == Intrinsic::cttz) 2587 return ConstantInt::get(Ty, C0->countTrailingZeros()); 2588 else 2589 return ConstantInt::get(Ty, C0->countLeadingZeros()); 2590 2591 case Intrinsic::abs: 2592 // Undef or minimum val operand with poison min --> undef 2593 assert(C1 && "Must be constant int"); 2594 if (C1->isOne() && (!C0 || C0->isMinSignedValue())) 2595 return UndefValue::get(Ty); 2596 2597 // Undef operand with no poison min --> 0 (sign bit must be clear) 2598 if (C1->isZero() && !C0) 2599 return Constant::getNullValue(Ty); 2600 2601 return ConstantInt::get(Ty, C0->abs()); 2602 } 2603 2604 return nullptr; 2605 } 2606 2607 // Support ConstantVector in case we have an Undef in the top. 2608 if ((isa<ConstantVector>(Operands[0]) || 2609 isa<ConstantDataVector>(Operands[0])) && 2610 // Check for default rounding mode. 2611 // FIXME: Support other rounding modes? 2612 isa<ConstantInt>(Operands[1]) && 2613 cast<ConstantInt>(Operands[1])->getValue() == 4) { 2614 auto *Op = cast<Constant>(Operands[0]); 2615 switch (IntrinsicID) { 2616 default: break; 2617 case Intrinsic::x86_avx512_vcvtss2si32: 2618 case Intrinsic::x86_avx512_vcvtss2si64: 2619 case Intrinsic::x86_avx512_vcvtsd2si32: 2620 case Intrinsic::x86_avx512_vcvtsd2si64: 2621 if (ConstantFP *FPOp = 2622 dyn_cast_or_null<ConstantFP>(Op->getAggregateElement(0U))) 2623 return ConstantFoldSSEConvertToInt(FPOp->getValueAPF(), 2624 /*roundTowardZero=*/false, Ty, 2625 /*IsSigned*/true); 2626 break; 2627 case Intrinsic::x86_avx512_vcvtss2usi32: 2628 case Intrinsic::x86_avx512_vcvtss2usi64: 2629 case Intrinsic::x86_avx512_vcvtsd2usi32: 2630 case Intrinsic::x86_avx512_vcvtsd2usi64: 2631 if (ConstantFP *FPOp = 2632 dyn_cast_or_null<ConstantFP>(Op->getAggregateElement(0U))) 2633 return ConstantFoldSSEConvertToInt(FPOp->getValueAPF(), 2634 /*roundTowardZero=*/false, Ty, 2635 /*IsSigned*/false); 2636 break; 2637 case Intrinsic::x86_avx512_cvttss2si: 2638 case Intrinsic::x86_avx512_cvttss2si64: 2639 case Intrinsic::x86_avx512_cvttsd2si: 2640 case Intrinsic::x86_avx512_cvttsd2si64: 2641 if (ConstantFP *FPOp = 2642 dyn_cast_or_null<ConstantFP>(Op->getAggregateElement(0U))) 2643 return ConstantFoldSSEConvertToInt(FPOp->getValueAPF(), 2644 /*roundTowardZero=*/true, Ty, 2645 /*IsSigned*/true); 2646 break; 2647 case Intrinsic::x86_avx512_cvttss2usi: 2648 case Intrinsic::x86_avx512_cvttss2usi64: 2649 case Intrinsic::x86_avx512_cvttsd2usi: 2650 case Intrinsic::x86_avx512_cvttsd2usi64: 2651 if (ConstantFP *FPOp = 2652 dyn_cast_or_null<ConstantFP>(Op->getAggregateElement(0U))) 2653 return ConstantFoldSSEConvertToInt(FPOp->getValueAPF(), 2654 /*roundTowardZero=*/true, Ty, 2655 /*IsSigned*/false); 2656 break; 2657 } 2658 } 2659 return nullptr; 2660 } 2661 2662 static APFloat ConstantFoldAMDGCNCubeIntrinsic(Intrinsic::ID IntrinsicID, 2663 const APFloat &S0, 2664 const APFloat &S1, 2665 const APFloat &S2) { 2666 unsigned ID; 2667 const fltSemantics &Sem = S0.getSemantics(); 2668 APFloat MA(Sem), SC(Sem), TC(Sem); 2669 if (abs(S2) >= abs(S0) && abs(S2) >= abs(S1)) { 2670 if (S2.isNegative() && S2.isNonZero() && !S2.isNaN()) { 2671 // S2 < 0 2672 ID = 5; 2673 SC = -S0; 2674 } else { 2675 ID = 4; 2676 SC = S0; 2677 } 2678 MA = S2; 2679 TC = -S1; 2680 } else if (abs(S1) >= abs(S0)) { 2681 if (S1.isNegative() && S1.isNonZero() && !S1.isNaN()) { 2682 // S1 < 0 2683 ID = 3; 2684 TC = -S2; 2685 } else { 2686 ID = 2; 2687 TC = S2; 2688 } 2689 MA = S1; 2690 SC = S0; 2691 } else { 2692 if (S0.isNegative() && S0.isNonZero() && !S0.isNaN()) { 2693 // S0 < 0 2694 ID = 1; 2695 SC = S2; 2696 } else { 2697 ID = 0; 2698 SC = -S2; 2699 } 2700 MA = S0; 2701 TC = -S1; 2702 } 2703 switch (IntrinsicID) { 2704 default: 2705 llvm_unreachable("unhandled amdgcn cube intrinsic"); 2706 case Intrinsic::amdgcn_cubeid: 2707 return APFloat(Sem, ID); 2708 case Intrinsic::amdgcn_cubema: 2709 return MA + MA; 2710 case Intrinsic::amdgcn_cubesc: 2711 return SC; 2712 case Intrinsic::amdgcn_cubetc: 2713 return TC; 2714 } 2715 } 2716 2717 static Constant *ConstantFoldAMDGCNPermIntrinsic(ArrayRef<Constant *> Operands, 2718 Type *Ty) { 2719 const APInt *C0, *C1, *C2; 2720 if (!getConstIntOrUndef(Operands[0], C0) || 2721 !getConstIntOrUndef(Operands[1], C1) || 2722 !getConstIntOrUndef(Operands[2], C2)) 2723 return nullptr; 2724 2725 if (!C2) 2726 return UndefValue::get(Ty); 2727 2728 APInt Val(32, 0); 2729 unsigned NumUndefBytes = 0; 2730 for (unsigned I = 0; I < 32; I += 8) { 2731 unsigned Sel = C2->extractBitsAsZExtValue(8, I); 2732 unsigned B = 0; 2733 2734 if (Sel >= 13) 2735 B = 0xff; 2736 else if (Sel == 12) 2737 B = 0x00; 2738 else { 2739 const APInt *Src = ((Sel & 10) == 10 || (Sel & 12) == 4) ? C0 : C1; 2740 if (!Src) 2741 ++NumUndefBytes; 2742 else if (Sel < 8) 2743 B = Src->extractBitsAsZExtValue(8, (Sel & 3) * 8); 2744 else 2745 B = Src->extractBitsAsZExtValue(1, (Sel & 1) ? 31 : 15) * 0xff; 2746 } 2747 2748 Val.insertBits(B, I, 8); 2749 } 2750 2751 if (NumUndefBytes == 4) 2752 return UndefValue::get(Ty); 2753 2754 return ConstantInt::get(Ty, Val); 2755 } 2756 2757 static Constant *ConstantFoldScalarCall3(StringRef Name, 2758 Intrinsic::ID IntrinsicID, 2759 Type *Ty, 2760 ArrayRef<Constant *> Operands, 2761 const TargetLibraryInfo *TLI, 2762 const CallBase *Call) { 2763 assert(Operands.size() == 3 && "Wrong number of operands."); 2764 2765 if (const auto *Op1 = dyn_cast<ConstantFP>(Operands[0])) { 2766 if (const auto *Op2 = dyn_cast<ConstantFP>(Operands[1])) { 2767 if (const auto *Op3 = dyn_cast<ConstantFP>(Operands[2])) { 2768 const APFloat &C1 = Op1->getValueAPF(); 2769 const APFloat &C2 = Op2->getValueAPF(); 2770 const APFloat &C3 = Op3->getValueAPF(); 2771 2772 if (const auto *ConstrIntr = dyn_cast<ConstrainedFPIntrinsic>(Call)) { 2773 RoundingMode RM = getEvaluationRoundingMode(ConstrIntr); 2774 APFloat Res = C1; 2775 APFloat::opStatus St; 2776 switch (IntrinsicID) { 2777 default: 2778 return nullptr; 2779 case Intrinsic::experimental_constrained_fma: 2780 case Intrinsic::experimental_constrained_fmuladd: 2781 St = Res.fusedMultiplyAdd(C2, C3, RM); 2782 break; 2783 } 2784 if (mayFoldConstrained( 2785 const_cast<ConstrainedFPIntrinsic *>(ConstrIntr), St)) 2786 return ConstantFP::get(Ty->getContext(), Res); 2787 return nullptr; 2788 } 2789 2790 switch (IntrinsicID) { 2791 default: break; 2792 case Intrinsic::amdgcn_fma_legacy: { 2793 // The legacy behaviour is that multiplying +/- 0.0 by anything, even 2794 // NaN or infinity, gives +0.0. 2795 if (C1.isZero() || C2.isZero()) { 2796 // It's tempting to just return C3 here, but that would give the 2797 // wrong result if C3 was -0.0. 2798 return ConstantFP::get(Ty->getContext(), APFloat(0.0f) + C3); 2799 } 2800 LLVM_FALLTHROUGH; 2801 } 2802 case Intrinsic::fma: 2803 case Intrinsic::fmuladd: { 2804 APFloat V = C1; 2805 V.fusedMultiplyAdd(C2, C3, APFloat::rmNearestTiesToEven); 2806 return ConstantFP::get(Ty->getContext(), V); 2807 } 2808 case Intrinsic::amdgcn_cubeid: 2809 case Intrinsic::amdgcn_cubema: 2810 case Intrinsic::amdgcn_cubesc: 2811 case Intrinsic::amdgcn_cubetc: { 2812 APFloat V = ConstantFoldAMDGCNCubeIntrinsic(IntrinsicID, C1, C2, C3); 2813 return ConstantFP::get(Ty->getContext(), V); 2814 } 2815 } 2816 } 2817 } 2818 } 2819 2820 if (IntrinsicID == Intrinsic::smul_fix || 2821 IntrinsicID == Intrinsic::smul_fix_sat) { 2822 // poison * C -> poison 2823 // C * poison -> poison 2824 if (isa<PoisonValue>(Operands[0]) || isa<PoisonValue>(Operands[1])) 2825 return PoisonValue::get(Ty); 2826 2827 const APInt *C0, *C1; 2828 if (!getConstIntOrUndef(Operands[0], C0) || 2829 !getConstIntOrUndef(Operands[1], C1)) 2830 return nullptr; 2831 2832 // undef * C -> 0 2833 // C * undef -> 0 2834 if (!C0 || !C1) 2835 return Constant::getNullValue(Ty); 2836 2837 // This code performs rounding towards negative infinity in case the result 2838 // cannot be represented exactly for the given scale. Targets that do care 2839 // about rounding should use a target hook for specifying how rounding 2840 // should be done, and provide their own folding to be consistent with 2841 // rounding. This is the same approach as used by 2842 // DAGTypeLegalizer::ExpandIntRes_MULFIX. 2843 unsigned Scale = cast<ConstantInt>(Operands[2])->getZExtValue(); 2844 unsigned Width = C0->getBitWidth(); 2845 assert(Scale < Width && "Illegal scale."); 2846 unsigned ExtendedWidth = Width * 2; 2847 APInt Product = (C0->sextOrSelf(ExtendedWidth) * 2848 C1->sextOrSelf(ExtendedWidth)).ashr(Scale); 2849 if (IntrinsicID == Intrinsic::smul_fix_sat) { 2850 APInt Max = APInt::getSignedMaxValue(Width).sextOrSelf(ExtendedWidth); 2851 APInt Min = APInt::getSignedMinValue(Width).sextOrSelf(ExtendedWidth); 2852 Product = APIntOps::smin(Product, Max); 2853 Product = APIntOps::smax(Product, Min); 2854 } 2855 return ConstantInt::get(Ty->getContext(), Product.sextOrTrunc(Width)); 2856 } 2857 2858 if (IntrinsicID == Intrinsic::fshl || IntrinsicID == Intrinsic::fshr) { 2859 const APInt *C0, *C1, *C2; 2860 if (!getConstIntOrUndef(Operands[0], C0) || 2861 !getConstIntOrUndef(Operands[1], C1) || 2862 !getConstIntOrUndef(Operands[2], C2)) 2863 return nullptr; 2864 2865 bool IsRight = IntrinsicID == Intrinsic::fshr; 2866 if (!C2) 2867 return Operands[IsRight ? 1 : 0]; 2868 if (!C0 && !C1) 2869 return UndefValue::get(Ty); 2870 2871 // The shift amount is interpreted as modulo the bitwidth. If the shift 2872 // amount is effectively 0, avoid UB due to oversized inverse shift below. 2873 unsigned BitWidth = C2->getBitWidth(); 2874 unsigned ShAmt = C2->urem(BitWidth); 2875 if (!ShAmt) 2876 return Operands[IsRight ? 1 : 0]; 2877 2878 // (C0 << ShlAmt) | (C1 >> LshrAmt) 2879 unsigned LshrAmt = IsRight ? ShAmt : BitWidth - ShAmt; 2880 unsigned ShlAmt = !IsRight ? ShAmt : BitWidth - ShAmt; 2881 if (!C0) 2882 return ConstantInt::get(Ty, C1->lshr(LshrAmt)); 2883 if (!C1) 2884 return ConstantInt::get(Ty, C0->shl(ShlAmt)); 2885 return ConstantInt::get(Ty, C0->shl(ShlAmt) | C1->lshr(LshrAmt)); 2886 } 2887 2888 if (IntrinsicID == Intrinsic::amdgcn_perm) 2889 return ConstantFoldAMDGCNPermIntrinsic(Operands, Ty); 2890 2891 return nullptr; 2892 } 2893 2894 static Constant *ConstantFoldScalarCall(StringRef Name, 2895 Intrinsic::ID IntrinsicID, 2896 Type *Ty, 2897 ArrayRef<Constant *> Operands, 2898 const TargetLibraryInfo *TLI, 2899 const CallBase *Call) { 2900 if (Operands.size() == 1) 2901 return ConstantFoldScalarCall1(Name, IntrinsicID, Ty, Operands, TLI, Call); 2902 2903 if (Operands.size() == 2) 2904 return ConstantFoldScalarCall2(Name, IntrinsicID, Ty, Operands, TLI, Call); 2905 2906 if (Operands.size() == 3) 2907 return ConstantFoldScalarCall3(Name, IntrinsicID, Ty, Operands, TLI, Call); 2908 2909 return nullptr; 2910 } 2911 2912 static Constant *ConstantFoldFixedVectorCall( 2913 StringRef Name, Intrinsic::ID IntrinsicID, FixedVectorType *FVTy, 2914 ArrayRef<Constant *> Operands, const DataLayout &DL, 2915 const TargetLibraryInfo *TLI, const CallBase *Call) { 2916 SmallVector<Constant *, 4> Result(FVTy->getNumElements()); 2917 SmallVector<Constant *, 4> Lane(Operands.size()); 2918 Type *Ty = FVTy->getElementType(); 2919 2920 switch (IntrinsicID) { 2921 case Intrinsic::masked_load: { 2922 auto *SrcPtr = Operands[0]; 2923 auto *Mask = Operands[2]; 2924 auto *Passthru = Operands[3]; 2925 2926 Constant *VecData = ConstantFoldLoadFromConstPtr(SrcPtr, FVTy, DL); 2927 2928 SmallVector<Constant *, 32> NewElements; 2929 for (unsigned I = 0, E = FVTy->getNumElements(); I != E; ++I) { 2930 auto *MaskElt = Mask->getAggregateElement(I); 2931 if (!MaskElt) 2932 break; 2933 auto *PassthruElt = Passthru->getAggregateElement(I); 2934 auto *VecElt = VecData ? VecData->getAggregateElement(I) : nullptr; 2935 if (isa<UndefValue>(MaskElt)) { 2936 if (PassthruElt) 2937 NewElements.push_back(PassthruElt); 2938 else if (VecElt) 2939 NewElements.push_back(VecElt); 2940 else 2941 return nullptr; 2942 } 2943 if (MaskElt->isNullValue()) { 2944 if (!PassthruElt) 2945 return nullptr; 2946 NewElements.push_back(PassthruElt); 2947 } else if (MaskElt->isOneValue()) { 2948 if (!VecElt) 2949 return nullptr; 2950 NewElements.push_back(VecElt); 2951 } else { 2952 return nullptr; 2953 } 2954 } 2955 if (NewElements.size() != FVTy->getNumElements()) 2956 return nullptr; 2957 return ConstantVector::get(NewElements); 2958 } 2959 case Intrinsic::arm_mve_vctp8: 2960 case Intrinsic::arm_mve_vctp16: 2961 case Intrinsic::arm_mve_vctp32: 2962 case Intrinsic::arm_mve_vctp64: { 2963 if (auto *Op = dyn_cast<ConstantInt>(Operands[0])) { 2964 unsigned Lanes = FVTy->getNumElements(); 2965 uint64_t Limit = Op->getZExtValue(); 2966 2967 SmallVector<Constant *, 16> NCs; 2968 for (unsigned i = 0; i < Lanes; i++) { 2969 if (i < Limit) 2970 NCs.push_back(ConstantInt::getTrue(Ty)); 2971 else 2972 NCs.push_back(ConstantInt::getFalse(Ty)); 2973 } 2974 return ConstantVector::get(NCs); 2975 } 2976 break; 2977 } 2978 case Intrinsic::get_active_lane_mask: { 2979 auto *Op0 = dyn_cast<ConstantInt>(Operands[0]); 2980 auto *Op1 = dyn_cast<ConstantInt>(Operands[1]); 2981 if (Op0 && Op1) { 2982 unsigned Lanes = FVTy->getNumElements(); 2983 uint64_t Base = Op0->getZExtValue(); 2984 uint64_t Limit = Op1->getZExtValue(); 2985 2986 SmallVector<Constant *, 16> NCs; 2987 for (unsigned i = 0; i < Lanes; i++) { 2988 if (Base + i < Limit) 2989 NCs.push_back(ConstantInt::getTrue(Ty)); 2990 else 2991 NCs.push_back(ConstantInt::getFalse(Ty)); 2992 } 2993 return ConstantVector::get(NCs); 2994 } 2995 break; 2996 } 2997 default: 2998 break; 2999 } 3000 3001 for (unsigned I = 0, E = FVTy->getNumElements(); I != E; ++I) { 3002 // Gather a column of constants. 3003 for (unsigned J = 0, JE = Operands.size(); J != JE; ++J) { 3004 // Some intrinsics use a scalar type for certain arguments. 3005 if (hasVectorInstrinsicScalarOpd(IntrinsicID, J)) { 3006 Lane[J] = Operands[J]; 3007 continue; 3008 } 3009 3010 Constant *Agg = Operands[J]->getAggregateElement(I); 3011 if (!Agg) 3012 return nullptr; 3013 3014 Lane[J] = Agg; 3015 } 3016 3017 // Use the regular scalar folding to simplify this column. 3018 Constant *Folded = 3019 ConstantFoldScalarCall(Name, IntrinsicID, Ty, Lane, TLI, Call); 3020 if (!Folded) 3021 return nullptr; 3022 Result[I] = Folded; 3023 } 3024 3025 return ConstantVector::get(Result); 3026 } 3027 3028 static Constant *ConstantFoldScalableVectorCall( 3029 StringRef Name, Intrinsic::ID IntrinsicID, ScalableVectorType *SVTy, 3030 ArrayRef<Constant *> Operands, const DataLayout &DL, 3031 const TargetLibraryInfo *TLI, const CallBase *Call) { 3032 switch (IntrinsicID) { 3033 case Intrinsic::aarch64_sve_convert_from_svbool: { 3034 auto *Src = dyn_cast<Constant>(Operands[0]); 3035 if (!Src || !Src->isNullValue()) 3036 break; 3037 3038 return ConstantInt::getFalse(SVTy); 3039 } 3040 default: 3041 break; 3042 } 3043 return nullptr; 3044 } 3045 3046 } // end anonymous namespace 3047 3048 Constant *llvm::ConstantFoldCall(const CallBase *Call, Function *F, 3049 ArrayRef<Constant *> Operands, 3050 const TargetLibraryInfo *TLI) { 3051 if (Call->isNoBuiltin()) 3052 return nullptr; 3053 if (!F->hasName()) 3054 return nullptr; 3055 3056 // If this is not an intrinsic and not recognized as a library call, bail out. 3057 if (F->getIntrinsicID() == Intrinsic::not_intrinsic) { 3058 if (!TLI) 3059 return nullptr; 3060 LibFunc LibF; 3061 if (!TLI->getLibFunc(*F, LibF)) 3062 return nullptr; 3063 } 3064 3065 StringRef Name = F->getName(); 3066 Type *Ty = F->getReturnType(); 3067 if (auto *FVTy = dyn_cast<FixedVectorType>(Ty)) 3068 return ConstantFoldFixedVectorCall( 3069 Name, F->getIntrinsicID(), FVTy, Operands, 3070 F->getParent()->getDataLayout(), TLI, Call); 3071 3072 if (auto *SVTy = dyn_cast<ScalableVectorType>(Ty)) 3073 return ConstantFoldScalableVectorCall( 3074 Name, F->getIntrinsicID(), SVTy, Operands, 3075 F->getParent()->getDataLayout(), TLI, Call); 3076 3077 // TODO: If this is a library function, we already discovered that above, 3078 // so we should pass the LibFunc, not the name (and it might be better 3079 // still to separate intrinsic handling from libcalls). 3080 return ConstantFoldScalarCall(Name, F->getIntrinsicID(), Ty, Operands, TLI, 3081 Call); 3082 } 3083 3084 bool llvm::isMathLibCallNoop(const CallBase *Call, 3085 const TargetLibraryInfo *TLI) { 3086 // FIXME: Refactor this code; this duplicates logic in LibCallsShrinkWrap 3087 // (and to some extent ConstantFoldScalarCall). 3088 if (Call->isNoBuiltin() || Call->isStrictFP()) 3089 return false; 3090 Function *F = Call->getCalledFunction(); 3091 if (!F) 3092 return false; 3093 3094 LibFunc Func; 3095 if (!TLI || !TLI->getLibFunc(*F, Func)) 3096 return false; 3097 3098 if (Call->arg_size() == 1) { 3099 if (ConstantFP *OpC = dyn_cast<ConstantFP>(Call->getArgOperand(0))) { 3100 const APFloat &Op = OpC->getValueAPF(); 3101 switch (Func) { 3102 case LibFunc_logl: 3103 case LibFunc_log: 3104 case LibFunc_logf: 3105 case LibFunc_log2l: 3106 case LibFunc_log2: 3107 case LibFunc_log2f: 3108 case LibFunc_log10l: 3109 case LibFunc_log10: 3110 case LibFunc_log10f: 3111 return Op.isNaN() || (!Op.isZero() && !Op.isNegative()); 3112 3113 case LibFunc_expl: 3114 case LibFunc_exp: 3115 case LibFunc_expf: 3116 // FIXME: These boundaries are slightly conservative. 3117 if (OpC->getType()->isDoubleTy()) 3118 return !(Op < APFloat(-745.0) || Op > APFloat(709.0)); 3119 if (OpC->getType()->isFloatTy()) 3120 return !(Op < APFloat(-103.0f) || Op > APFloat(88.0f)); 3121 break; 3122 3123 case LibFunc_exp2l: 3124 case LibFunc_exp2: 3125 case LibFunc_exp2f: 3126 // FIXME: These boundaries are slightly conservative. 3127 if (OpC->getType()->isDoubleTy()) 3128 return !(Op < APFloat(-1074.0) || Op > APFloat(1023.0)); 3129 if (OpC->getType()->isFloatTy()) 3130 return !(Op < APFloat(-149.0f) || Op > APFloat(127.0f)); 3131 break; 3132 3133 case LibFunc_sinl: 3134 case LibFunc_sin: 3135 case LibFunc_sinf: 3136 case LibFunc_cosl: 3137 case LibFunc_cos: 3138 case LibFunc_cosf: 3139 return !Op.isInfinity(); 3140 3141 case LibFunc_tanl: 3142 case LibFunc_tan: 3143 case LibFunc_tanf: { 3144 // FIXME: Stop using the host math library. 3145 // FIXME: The computation isn't done in the right precision. 3146 Type *Ty = OpC->getType(); 3147 if (Ty->isDoubleTy() || Ty->isFloatTy() || Ty->isHalfTy()) 3148 return ConstantFoldFP(tan, OpC->getValueAPF(), Ty) != nullptr; 3149 break; 3150 } 3151 3152 case LibFunc_asinl: 3153 case LibFunc_asin: 3154 case LibFunc_asinf: 3155 case LibFunc_acosl: 3156 case LibFunc_acos: 3157 case LibFunc_acosf: 3158 return !(Op < APFloat(Op.getSemantics(), "-1") || 3159 Op > APFloat(Op.getSemantics(), "1")); 3160 3161 case LibFunc_sinh: 3162 case LibFunc_cosh: 3163 case LibFunc_sinhf: 3164 case LibFunc_coshf: 3165 case LibFunc_sinhl: 3166 case LibFunc_coshl: 3167 // FIXME: These boundaries are slightly conservative. 3168 if (OpC->getType()->isDoubleTy()) 3169 return !(Op < APFloat(-710.0) || Op > APFloat(710.0)); 3170 if (OpC->getType()->isFloatTy()) 3171 return !(Op < APFloat(-89.0f) || Op > APFloat(89.0f)); 3172 break; 3173 3174 case LibFunc_sqrtl: 3175 case LibFunc_sqrt: 3176 case LibFunc_sqrtf: 3177 return Op.isNaN() || Op.isZero() || !Op.isNegative(); 3178 3179 // FIXME: Add more functions: sqrt_finite, atanh, expm1, log1p, 3180 // maybe others? 3181 default: 3182 break; 3183 } 3184 } 3185 } 3186 3187 if (Call->arg_size() == 2) { 3188 ConstantFP *Op0C = dyn_cast<ConstantFP>(Call->getArgOperand(0)); 3189 ConstantFP *Op1C = dyn_cast<ConstantFP>(Call->getArgOperand(1)); 3190 if (Op0C && Op1C) { 3191 const APFloat &Op0 = Op0C->getValueAPF(); 3192 const APFloat &Op1 = Op1C->getValueAPF(); 3193 3194 switch (Func) { 3195 case LibFunc_powl: 3196 case LibFunc_pow: 3197 case LibFunc_powf: { 3198 // FIXME: Stop using the host math library. 3199 // FIXME: The computation isn't done in the right precision. 3200 Type *Ty = Op0C->getType(); 3201 if (Ty->isDoubleTy() || Ty->isFloatTy() || Ty->isHalfTy()) { 3202 if (Ty == Op1C->getType()) 3203 return ConstantFoldBinaryFP(pow, Op0, Op1, Ty) != nullptr; 3204 } 3205 break; 3206 } 3207 3208 case LibFunc_fmodl: 3209 case LibFunc_fmod: 3210 case LibFunc_fmodf: 3211 case LibFunc_remainderl: 3212 case LibFunc_remainder: 3213 case LibFunc_remainderf: 3214 return Op0.isNaN() || Op1.isNaN() || 3215 (!Op0.isInfinity() && !Op1.isZero()); 3216 3217 default: 3218 break; 3219 } 3220 } 3221 } 3222 3223 return false; 3224 } 3225 3226 void TargetFolder::anchor() {} 3227