ConstantFolding.cpp revision a97ddd0cffb9a9eb1313343fd49097c9057a5850
183e3c4f9d796a5da223a3ebd5d4ba985c7ecc39dDan Gohman//===-- ConstantFolding.cpp - Fold instructions into constants ------------===// 2bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell// 3bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell// The LLVM Compiler Infrastructure 4bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell// 54ee451de366474b9c228b4e5fa573795a715216dChris Lattner// This file is distributed under the University of Illinois Open Source 64ee451de366474b9c228b4e5fa573795a715216dChris Lattner// License. See LICENSE.TXT for details. 7bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell// 8bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell//===----------------------------------------------------------------------===// 9bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell// 1083e3c4f9d796a5da223a3ebd5d4ba985c7ecc39dDan Gohman// This file defines routines for folding instructions into constants. 1183e3c4f9d796a5da223a3ebd5d4ba985c7ecc39dDan Gohman// 1283e3c4f9d796a5da223a3ebd5d4ba985c7ecc39dDan Gohman// Also, to supplement the basic VMCore ConstantExpr simplifications, 1383e3c4f9d796a5da223a3ebd5d4ba985c7ecc39dDan Gohman// this file defines some additional folding routines that can make use of 1483e3c4f9d796a5da223a3ebd5d4ba985c7ecc39dDan Gohman// TargetData information. These functions cannot go in VMCore due to library 1583e3c4f9d796a5da223a3ebd5d4ba985c7ecc39dDan Gohman// dependency issues. 16bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell// 17bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell//===----------------------------------------------------------------------===// 18bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell 19bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell#include "llvm/Analysis/ConstantFolding.h" 20bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell#include "llvm/Constants.h" 21bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell#include "llvm/DerivedTypes.h" 225520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner#include "llvm/Function.h" 239a38e3e3991ea443e555d8060f91202a786acdd4Dan Gohman#include "llvm/GlobalVariable.h" 24bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell#include "llvm/Instructions.h" 25bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell#include "llvm/Intrinsics.h" 26562b84b3aea359d1f918184e355da82bf05eb290Jay Foad#include "llvm/Operator.h" 2762d327e241f59735eed836e98c8f11ffaa6cf6f8Chris Lattner#include "llvm/Analysis/ValueTracking.h" 2862d327e241f59735eed836e98c8f11ffaa6cf6f8Chris Lattner#include "llvm/Target/TargetData.h" 29618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier#include "llvm/Target/TargetLibraryInfo.h" 305520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner#include "llvm/ADT/SmallVector.h" 31c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner#include "llvm/ADT/StringMap.h" 32c25e7581b9b8088910da31702d4ca21c4734c6d7Torok Edwin#include "llvm/Support/ErrorHandling.h" 33bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell#include "llvm/Support/GetElementPtrTypeIterator.h" 34bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell#include "llvm/Support/MathExtras.h" 3583471853b1d5a9efe6b9e7565c457fc901f84926Rafael Espindola#include "llvm/Support/FEnv.h" 36bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell#include <cerrno> 3797af751deb9b26fd42fbcee082da9ccc4ded5b45Jeff Cohen#include <cmath> 38bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswellusing namespace llvm; 39bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell 4003dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner//===----------------------------------------------------------------------===// 4103dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner// Constant Folding internal helper functions 4203dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner//===----------------------------------------------------------------------===// 4303dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner 446333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner/// FoldBitCast - Constant fold bitcast, symbolically evaluating it with 456333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner/// TargetData. This always returns a non-null constant, but it may be a 466333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner/// ConstantExpr if unfoldable. 47db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattnerstatic Constant *FoldBitCast(Constant *C, Type *DestTy, 486333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner const TargetData &TD) { 494c7c0f23533565c7e2ddf71e01bf50f2aede5f1bNadav Rotem // Catch the obvious splat cases. 504c7c0f23533565c7e2ddf71e01bf50f2aede5f1bNadav Rotem if (C->isNullValue() && !DestTy->isX86_MMXTy()) 514c7c0f23533565c7e2ddf71e01bf50f2aede5f1bNadav Rotem return Constant::getNullValue(DestTy); 524c7c0f23533565c7e2ddf71e01bf50f2aede5f1bNadav Rotem if (C->isAllOnesValue() && !DestTy->isX86_MMXTy()) 534c7c0f23533565c7e2ddf71e01bf50f2aede5f1bNadav Rotem return Constant::getAllOnesValue(DestTy); 544c7c0f23533565c7e2ddf71e01bf50f2aede5f1bNadav Rotem 551c9fe0361b3f9f5c814ce761b97ecd11637a1d7dNadav Rotem // The code below only handles casts to vectors currently. 56db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner VectorType *DestVTy = dyn_cast<VectorType>(DestTy); 576333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner if (DestVTy == 0) 586333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner return ConstantExpr::getBitCast(C, DestTy); 596333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner 6093798daa3ecbfb29624052d3e54d2404ab65897aChris Lattner // If this is a scalar -> vector cast, convert the input into a <1 x scalar> 6193798daa3ecbfb29624052d3e54d2404ab65897aChris Lattner // vector so the code below can handle it uniformly. 6293798daa3ecbfb29624052d3e54d2404ab65897aChris Lattner if (isa<ConstantFP>(C) || isa<ConstantInt>(C)) { 6393798daa3ecbfb29624052d3e54d2404ab65897aChris Lattner Constant *Ops = C; // don't take the address of C! 642ca5c8644e6c35b3a7910a576ed89cddb7b82c3bChris Lattner return FoldBitCast(ConstantVector::get(Ops), DestTy, TD); 6593798daa3ecbfb29624052d3e54d2404ab65897aChris Lattner } 6693798daa3ecbfb29624052d3e54d2404ab65897aChris Lattner 6793798daa3ecbfb29624052d3e54d2404ab65897aChris Lattner // If this is a bitcast from constant vector -> vector, fold it. 684c7c0f23533565c7e2ddf71e01bf50f2aede5f1bNadav Rotem ConstantVector *CV = dyn_cast<ConstantVector>(C); 6993798daa3ecbfb29624052d3e54d2404ab65897aChris Lattner if (CV == 0) 7093798daa3ecbfb29624052d3e54d2404ab65897aChris Lattner return ConstantExpr::getBitCast(C, DestTy); 7193798daa3ecbfb29624052d3e54d2404ab65897aChris Lattner 726333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner // If the element types match, VMCore can fold it. 736333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner unsigned NumDstElt = DestVTy->getNumElements(); 746333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner unsigned NumSrcElt = CV->getNumOperands(); 756333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner if (NumDstElt == NumSrcElt) 766333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner return ConstantExpr::getBitCast(C, DestTy); 776333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner 78db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *SrcEltTy = CV->getType()->getElementType(); 79db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *DstEltTy = DestVTy->getElementType(); 806333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner 816333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner // Otherwise, we're changing the number of elements in a vector, which 826333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner // requires endianness information to do the right thing. For example, 836333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner // bitcast (<2 x i64> <i64 0, i64 1> to <4 x i32>) 846333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner // folds to (little endian): 856333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner // <4 x i32> <i32 0, i32 0, i32 1, i32 0> 866333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner // and to (big endian): 876333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner // <4 x i32> <i32 0, i32 0, i32 0, i32 1> 886333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner 896333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner // First thing is first. We only want to think about integer here, so if 906333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner // we have something in FP form, recast it as integer. 91b0bc6c361da9009e8414efde317d9bbff755f6c0Duncan Sands if (DstEltTy->isFloatingPointTy()) { 926333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner // Fold to an vector of integers with same size as our FP type. 936333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner unsigned FPWidth = DstEltTy->getPrimitiveSizeInBits(); 94db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *DestIVTy = 956333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner VectorType::get(IntegerType::get(C->getContext(), FPWidth), NumDstElt); 966333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner // Recursively handle this integer conversion, if possible. 976333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner C = FoldBitCast(C, DestIVTy, TD); 986333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner if (!C) return ConstantExpr::getBitCast(C, DestTy); 996333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner 1006333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner // Finally, VMCore can handle this now that #elts line up. 1016333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner return ConstantExpr::getBitCast(C, DestTy); 1026333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner } 1036333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner 1046333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner // Okay, we know the destination is integer, if the input is FP, convert 1056333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner // it to integer first. 106b0bc6c361da9009e8414efde317d9bbff755f6c0Duncan Sands if (SrcEltTy->isFloatingPointTy()) { 1076333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner unsigned FPWidth = SrcEltTy->getPrimitiveSizeInBits(); 108db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *SrcIVTy = 1096333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner VectorType::get(IntegerType::get(C->getContext(), FPWidth), NumSrcElt); 1106333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner // Ask VMCore to do the conversion now that #elts line up. 1116333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner C = ConstantExpr::getBitCast(C, SrcIVTy); 1126333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner CV = dyn_cast<ConstantVector>(C); 1136333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner if (!CV) // If VMCore wasn't able to fold it, bail out. 1146333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner return C; 1156333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner } 1166333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner 1176333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner // Now we know that the input and output vectors are both integer vectors 1186333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner // of the same size, and that their #elements is not the same. Do the 1196333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner // conversion here, which depends on whether the input or output has 1206333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner // more elements. 1216333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner bool isLittleEndian = TD.isLittleEndian(); 1226333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner 1236333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner SmallVector<Constant*, 32> Result; 1246333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner if (NumDstElt < NumSrcElt) { 1256333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner // Handle: bitcast (<4 x i32> <i32 0, i32 1, i32 2, i32 3> to <2 x i64>) 1266333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner Constant *Zero = Constant::getNullValue(DstEltTy); 1276333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner unsigned Ratio = NumSrcElt/NumDstElt; 1286333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner unsigned SrcBitSize = SrcEltTy->getPrimitiveSizeInBits(); 1296333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner unsigned SrcElt = 0; 1306333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner for (unsigned i = 0; i != NumDstElt; ++i) { 1316333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner // Build each element of the result. 1326333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner Constant *Elt = Zero; 1336333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner unsigned ShiftAmt = isLittleEndian ? 0 : SrcBitSize*(Ratio-1); 1346333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner for (unsigned j = 0; j != Ratio; ++j) { 1356333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner Constant *Src = dyn_cast<ConstantInt>(CV->getOperand(SrcElt++)); 1366333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner if (!Src) // Reject constantexpr elements. 1376333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner return ConstantExpr::getBitCast(C, DestTy); 1386333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner 1396333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner // Zero extend the element to the right size. 1406333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner Src = ConstantExpr::getZExt(Src, Elt->getType()); 1416333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner 1426333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner // Shift it to the right place, depending on endianness. 1436333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner Src = ConstantExpr::getShl(Src, 1446333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner ConstantInt::get(Src->getType(), ShiftAmt)); 1456333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner ShiftAmt += isLittleEndian ? SrcBitSize : -SrcBitSize; 1466333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner 1476333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner // Mix it in. 1486333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner Elt = ConstantExpr::getOr(Elt, Src); 1496333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner } 1506333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner Result.push_back(Elt); 1516333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner } 1526333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner } else { 1536333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner // Handle: bitcast (<2 x i64> <i64 0, i64 1> to <4 x i32>) 1546333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner unsigned Ratio = NumDstElt/NumSrcElt; 1556333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner unsigned DstBitSize = DstEltTy->getPrimitiveSizeInBits(); 1566333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner 1576333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner // Loop over each source value, expanding into multiple results. 1586333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner for (unsigned i = 0; i != NumSrcElt; ++i) { 1596333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner Constant *Src = dyn_cast<ConstantInt>(CV->getOperand(i)); 1606333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner if (!Src) // Reject constantexpr elements. 1616333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner return ConstantExpr::getBitCast(C, DestTy); 1626333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner 1636333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner unsigned ShiftAmt = isLittleEndian ? 0 : DstBitSize*(Ratio-1); 1646333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner for (unsigned j = 0; j != Ratio; ++j) { 1656333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner // Shift the piece of the value into the right place, depending on 1666333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner // endianness. 1676333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner Constant *Elt = ConstantExpr::getLShr(Src, 1686333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner ConstantInt::get(Src->getType(), ShiftAmt)); 1696333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner ShiftAmt += isLittleEndian ? DstBitSize : -DstBitSize; 1706333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner 1716333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner // Truncate and remember this piece. 1726333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner Result.push_back(ConstantExpr::getTrunc(Elt, DstEltTy)); 1736333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner } 1746333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner } 1756333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner } 1766333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner 1772ca5c8644e6c35b3a7910a576ed89cddb7b82c3bChris Lattner return ConstantVector::get(Result); 1786333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner} 1796333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner 1806333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner 18103dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner/// IsConstantOffsetFromGlobal - If this constant is actually a constant offset 18203dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner/// from a global, return the global and the constant. Because of 18303dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner/// constantexprs, this function is recursive. 18403dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattnerstatic bool IsConstantOffsetFromGlobal(Constant *C, GlobalValue *&GV, 18503dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner int64_t &Offset, const TargetData &TD) { 18603dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner // Trivial case, constant is the global. 18703dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner if ((GV = dyn_cast<GlobalValue>(C))) { 18803dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner Offset = 0; 18903dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner return true; 19003dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner } 19103dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner 19203dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner // Otherwise, if this isn't a constant expr, bail out. 19303dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner ConstantExpr *CE = dyn_cast<ConstantExpr>(C); 19403dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner if (!CE) return false; 19503dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner 19603dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner // Look through ptr->int and ptr->ptr casts. 19703dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner if (CE->getOpcode() == Instruction::PtrToInt || 19803dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner CE->getOpcode() == Instruction::BitCast) 19903dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner return IsConstantOffsetFromGlobal(CE->getOperand(0), GV, Offset, TD); 20003dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner 20103dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner // i32* getelementptr ([5 x i32]* @a, i32 0, i32 5) 20203dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner if (CE->getOpcode() == Instruction::GetElementPtr) { 20303dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner // Cannot compute this if the element type of the pointer is missing size 20403dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner // info. 205f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner if (!cast<PointerType>(CE->getOperand(0)->getType()) 206f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner ->getElementType()->isSized()) 20703dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner return false; 20803dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner 20903dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner // If the base isn't a global+constant, we aren't either. 21003dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner if (!IsConstantOffsetFromGlobal(CE->getOperand(0), GV, Offset, TD)) 21103dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner return false; 21203dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner 21303dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner // Otherwise, add any offset that our operands provide. 21403dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner gep_type_iterator GTI = gep_type_begin(CE); 215de2d74b213844b17eb5327fba27f21e699d3af66Gabor Greif for (User::const_op_iterator i = CE->op_begin() + 1, e = CE->op_end(); 216785c6af9797fcf551feef70f2ecb5cd075b6e3c4Gabor Greif i != e; ++i, ++GTI) { 217de2d74b213844b17eb5327fba27f21e699d3af66Gabor Greif ConstantInt *CI = dyn_cast<ConstantInt>(*i); 21803dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner if (!CI) return false; // Index isn't a simple constant? 219e368b460a206fafa0d31d5d059b1779b94f7df8cDan Gohman if (CI->isZero()) continue; // Not adding anything. 22003dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner 221db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner if (StructType *ST = dyn_cast<StructType>(*GTI)) { 22203dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner // N = N + Offset 223b1919e2f08ecb37140af676fd2916f8d5ed7df3dChris Lattner Offset += TD.getStructLayout(ST)->getElementOffset(CI->getZExtValue()); 22403dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner } else { 225db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner SequentialType *SQT = cast<SequentialType>(*GTI); 226777d2306b36816a53bc1ae1244c0dc7d998ae691Duncan Sands Offset += TD.getTypeAllocSize(SQT->getElementType())*CI->getSExtValue(); 22703dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner } 22803dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner } 22903dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner return true; 23003dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner } 23103dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner 23203dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner return false; 23303dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner} 23403dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner 235fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner/// ReadDataFromGlobal - Recursive helper to read bits out of global. C is the 236fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner/// constant being copied out of. ByteOffset is an offset into C. CurPtr is the 237fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner/// pointer to copy results into and BytesLeft is the number of bytes left in 238fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner/// the CurPtr buffer. TD is the target data. 239fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattnerstatic bool ReadDataFromGlobal(Constant *C, uint64_t ByteOffset, 240fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner unsigned char *CurPtr, unsigned BytesLeft, 241fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner const TargetData &TD) { 242fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner assert(ByteOffset <= TD.getTypeAllocSize(C->getType()) && 243fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner "Out of range access"); 244fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 245c7b1382e351249774be63bd73839e8a0671635e1Chris Lattner // If this element is zero or undefined, we can just return since *CurPtr is 246c7b1382e351249774be63bd73839e8a0671635e1Chris Lattner // zero initialized. 247fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner if (isa<ConstantAggregateZero>(C) || isa<UndefValue>(C)) 248fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner return true; 249fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 250fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner if (ConstantInt *CI = dyn_cast<ConstantInt>(C)) { 251fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner if (CI->getBitWidth() > 64 || 252fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner (CI->getBitWidth() & 7) != 0) 253fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner return false; 254fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 255fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner uint64_t Val = CI->getZExtValue(); 256fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner unsigned IntBytes = unsigned(CI->getBitWidth()/8); 257fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 258fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner for (unsigned i = 0; i != BytesLeft && ByteOffset != IntBytes; ++i) { 259c7b1382e351249774be63bd73839e8a0671635e1Chris Lattner CurPtr[i] = (unsigned char)(Val >> (ByteOffset * 8)); 260fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner ++ByteOffset; 261fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner } 262fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner return true; 263fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner } 264fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 265fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner if (ConstantFP *CFP = dyn_cast<ConstantFP>(C)) { 266fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner if (CFP->getType()->isDoubleTy()) { 2676333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner C = FoldBitCast(C, Type::getInt64Ty(C->getContext()), TD); 268fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner return ReadDataFromGlobal(C, ByteOffset, CurPtr, BytesLeft, TD); 269fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner } 270fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner if (CFP->getType()->isFloatTy()){ 2716333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner C = FoldBitCast(C, Type::getInt32Ty(C->getContext()), TD); 272fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner return ReadDataFromGlobal(C, ByteOffset, CurPtr, BytesLeft, TD); 273fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner } 274c7b1382e351249774be63bd73839e8a0671635e1Chris Lattner return false; 275fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner } 276a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner 277fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner if (ConstantStruct *CS = dyn_cast<ConstantStruct>(C)) { 278fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner const StructLayout *SL = TD.getStructLayout(CS->getType()); 279fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner unsigned Index = SL->getElementContainingOffset(ByteOffset); 280fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner uint64_t CurEltOffset = SL->getElementOffset(Index); 281fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner ByteOffset -= CurEltOffset; 282fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 283fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner while (1) { 284fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner // If the element access is to the element itself and not to tail padding, 285fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner // read the bytes from the element. 286fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner uint64_t EltSize = TD.getTypeAllocSize(CS->getOperand(Index)->getType()); 287fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 288fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner if (ByteOffset < EltSize && 289fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner !ReadDataFromGlobal(CS->getOperand(Index), ByteOffset, CurPtr, 290fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner BytesLeft, TD)) 291fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner return false; 292fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 293fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner ++Index; 294fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 295fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner // Check to see if we read from the last struct element, if so we're done. 296fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner if (Index == CS->getType()->getNumElements()) 297fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner return true; 298fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 299fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner // If we read all of the bytes we needed from this element we're done. 300fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner uint64_t NextEltOffset = SL->getElementOffset(Index); 301fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 302fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner if (BytesLeft <= NextEltOffset-CurEltOffset-ByteOffset) 303fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner return true; 304fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 305fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner // Move to the next element of the struct. 306c5af649758ac8c6ffb896d7d5c570640a9fa1deeChris Lattner CurPtr += NextEltOffset-CurEltOffset-ByteOffset; 307fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner BytesLeft -= NextEltOffset-CurEltOffset-ByteOffset; 308fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner ByteOffset = 0; 309fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner CurEltOffset = NextEltOffset; 310fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner } 311fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner // not reached. 312fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner } 313fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 314fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner if (ConstantArray *CA = dyn_cast<ConstantArray>(C)) { 315fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner uint64_t EltSize = TD.getTypeAllocSize(CA->getType()->getElementType()); 316fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner uint64_t Index = ByteOffset / EltSize; 317fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner uint64_t Offset = ByteOffset - Index * EltSize; 318fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner for (; Index != CA->getType()->getNumElements(); ++Index) { 319fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner if (!ReadDataFromGlobal(CA->getOperand(Index), Offset, CurPtr, 320fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner BytesLeft, TD)) 321fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner return false; 322fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner if (EltSize >= BytesLeft) 323fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner return true; 324fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 325fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner Offset = 0; 326fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner BytesLeft -= EltSize; 327fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner CurPtr += EltSize; 328fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner } 329fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner return true; 330fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner } 331fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 332fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner if (ConstantVector *CV = dyn_cast<ConstantVector>(C)) { 333fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner uint64_t EltSize = TD.getTypeAllocSize(CV->getType()->getElementType()); 334fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner uint64_t Index = ByteOffset / EltSize; 335fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner uint64_t Offset = ByteOffset - Index * EltSize; 336fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner for (; Index != CV->getType()->getNumElements(); ++Index) { 337fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner if (!ReadDataFromGlobal(CV->getOperand(Index), Offset, CurPtr, 338fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner BytesLeft, TD)) 339fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner return false; 340fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner if (EltSize >= BytesLeft) 341fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner return true; 342fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 343fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner Offset = 0; 344fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner BytesLeft -= EltSize; 345fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner CurPtr += EltSize; 346fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner } 347fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner return true; 348fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner } 349fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 350a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner if (ConstantDataSequential *CDS = dyn_cast<ConstantDataSequential>(C)) { 351a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner uint64_t EltSize = CDS->getElementByteSize(); 352a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner uint64_t Index = ByteOffset / EltSize; 353a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner uint64_t Offset = ByteOffset - Index * EltSize; 354a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner for (; Index != CDS->getType()->getNumElements(); ++Index) { 355a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner if (!ReadDataFromGlobal(CDS->getElementAsConstant(Index), Offset, CurPtr, 356a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner BytesLeft, TD)) 357a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner return false; 358a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner if (EltSize >= BytesLeft) 359a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner return true; 360a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner 361a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner Offset = 0; 362a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner BytesLeft -= EltSize; 363a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner CurPtr += EltSize; 364a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner } 365a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner return true; 366a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner } 367a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner 3686475d9434ff3e981160f85d3f1aa07c9e0ace6f2Anders Carlsson if (ConstantExpr *CE = dyn_cast<ConstantExpr>(C)) { 3695d43ff4e7ea30eaf5a1c417f0ec528900f5c83f2Anders Carlsson if (CE->getOpcode() == Instruction::IntToPtr && 3705d43ff4e7ea30eaf5a1c417f0ec528900f5c83f2Anders Carlsson CE->getOperand(0)->getType() == TD.getIntPtrType(CE->getContext())) 3716475d9434ff3e981160f85d3f1aa07c9e0ace6f2Anders Carlsson return ReadDataFromGlobal(CE->getOperand(0), ByteOffset, CurPtr, 3726475d9434ff3e981160f85d3f1aa07c9e0ace6f2Anders Carlsson BytesLeft, TD); 3736475d9434ff3e981160f85d3f1aa07c9e0ace6f2Anders Carlsson } 3746475d9434ff3e981160f85d3f1aa07c9e0ace6f2Anders Carlsson 375fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner // Otherwise, unknown initializer type. 376fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner return false; 377fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner} 378fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 379fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattnerstatic Constant *FoldReinterpretLoadFromConstPtr(Constant *C, 380fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner const TargetData &TD) { 381db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *LoadTy = cast<PointerType>(C->getType())->getElementType(); 382db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner IntegerType *IntType = dyn_cast<IntegerType>(LoadTy); 383fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 384fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner // If this isn't an integer load we can't fold it directly. 385fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner if (!IntType) { 386fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner // If this is a float/double load, we can try folding it as an int32/64 load 38717f0cd3cc1432231f2bc5601e7e463986a59d817Chris Lattner // and then bitcast the result. This can be useful for union cases. Note 38817f0cd3cc1432231f2bc5601e7e463986a59d817Chris Lattner // that address spaces don't matter here since we're not going to result in 38917f0cd3cc1432231f2bc5601e7e463986a59d817Chris Lattner // an actual new load. 390db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *MapTy; 39117f0cd3cc1432231f2bc5601e7e463986a59d817Chris Lattner if (LoadTy->isFloatTy()) 392fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner MapTy = Type::getInt32PtrTy(C->getContext()); 39317f0cd3cc1432231f2bc5601e7e463986a59d817Chris Lattner else if (LoadTy->isDoubleTy()) 394fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner MapTy = Type::getInt64PtrTy(C->getContext()); 3951df9859c40492511b8aa4321eb76496005d3b75bDuncan Sands else if (LoadTy->isVectorTy()) { 39617f0cd3cc1432231f2bc5601e7e463986a59d817Chris Lattner MapTy = IntegerType::get(C->getContext(), 39717f0cd3cc1432231f2bc5601e7e463986a59d817Chris Lattner TD.getTypeAllocSizeInBits(LoadTy)); 39817f0cd3cc1432231f2bc5601e7e463986a59d817Chris Lattner MapTy = PointerType::getUnqual(MapTy); 39917f0cd3cc1432231f2bc5601e7e463986a59d817Chris Lattner } else 400fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner return 0; 401fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 4026333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner C = FoldBitCast(C, MapTy, TD); 403fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner if (Constant *Res = FoldReinterpretLoadFromConstPtr(C, TD)) 4046333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner return FoldBitCast(Res, LoadTy, TD); 405fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner return 0; 406fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner } 407fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 408fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner unsigned BytesLoaded = (IntType->getBitWidth() + 7) / 8; 409739208a790398cf1f9da05149c768371e48781e8Chris Lattner if (BytesLoaded > 32 || BytesLoaded == 0) return 0; 410fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 411fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner GlobalValue *GVal; 412fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner int64_t Offset; 413fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner if (!IsConstantOffsetFromGlobal(C, GVal, Offset, TD)) 414fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner return 0; 415fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 416fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner GlobalVariable *GV = dyn_cast<GlobalVariable>(GVal); 417c7b1382e351249774be63bd73839e8a0671635e1Chris Lattner if (!GV || !GV->isConstant() || !GV->hasDefinitiveInitializer() || 418fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner !GV->getInitializer()->getType()->isSized()) 419fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner return 0; 420fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 421fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner // If we're loading off the beginning of the global, some bytes may be valid, 422fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner // but we don't try to handle this. 423fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner if (Offset < 0) return 0; 424fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 425fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner // If we're not accessing anything in this constant, the result is undefined. 426fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner if (uint64_t(Offset) >= TD.getTypeAllocSize(GV->getInitializer()->getType())) 427fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner return UndefValue::get(IntType); 428fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 429739208a790398cf1f9da05149c768371e48781e8Chris Lattner unsigned char RawBytes[32] = {0}; 430fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner if (!ReadDataFromGlobal(GV->getInitializer(), Offset, RawBytes, 431fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner BytesLoaded, TD)) 432fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner return 0; 433fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 434b31189f26297fe30214da3e8d14a4575d5fdd262Chris Lattner APInt ResultVal = APInt(IntType->getBitWidth(), RawBytes[BytesLoaded-1]); 435b31189f26297fe30214da3e8d14a4575d5fdd262Chris Lattner for (unsigned i = 1; i != BytesLoaded; ++i) { 436739208a790398cf1f9da05149c768371e48781e8Chris Lattner ResultVal <<= 8; 437ce800511701135ed86845b451693a248c83a7973Dan Gohman ResultVal |= RawBytes[BytesLoaded-1-i]; 438739208a790398cf1f9da05149c768371e48781e8Chris Lattner } 439fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 440739208a790398cf1f9da05149c768371e48781e8Chris Lattner return ConstantInt::get(IntType->getContext(), ResultVal); 441fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner} 442fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 443878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner/// ConstantFoldLoadFromConstPtr - Return the value that a load from C would 444878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner/// produce if it is constant and determinable. If this is not determinable, 445878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner/// return null. 446878e4946700824954d7eb7f3ff660353db8e0d17Chris LattnerConstant *llvm::ConstantFoldLoadFromConstPtr(Constant *C, 447878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner const TargetData *TD) { 448878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner // First, try the easy cases: 449878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner if (GlobalVariable *GV = dyn_cast<GlobalVariable>(C)) 450878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner if (GV->isConstant() && GV->hasDefinitiveInitializer()) 451878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner return GV->getInitializer(); 452878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner 453e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner // If the loaded value isn't a constant expr, we can't handle it. 454e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner ConstantExpr *CE = dyn_cast<ConstantExpr>(C); 455e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner if (!CE) return 0; 456e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner 457e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner if (CE->getOpcode() == Instruction::GetElementPtr) { 458e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner if (GlobalVariable *GV = dyn_cast<GlobalVariable>(CE->getOperand(0))) 459e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner if (GV->isConstant() && GV->hasDefinitiveInitializer()) 460e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner if (Constant *V = 461e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner ConstantFoldLoadThroughGEPConstantExpr(GV->getInitializer(), CE)) 462e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner return V; 463e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner } 464e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner 465e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner // Instead of loading constant c string, use corresponding integer value 466e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner // directly if string length is small enough. 467e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner std::string Str; 46844a7a380aa9fbb303f57f4f8269062b7b56be980Chris Lattner if (TD && GetConstantStringInfo(CE, Str) && !Str.empty()) { 469fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner unsigned StrLen = Str.length(); 470db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *Ty = cast<PointerType>(CE->getType())->getElementType(); 471fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner unsigned NumBits = Ty->getPrimitiveSizeInBits(); 472eae2895120500d06bf6de9725efeeff498f60385Chris Lattner // Replace load with immediate integer if the result is an integer or fp 473eae2895120500d06bf6de9725efeeff498f60385Chris Lattner // value. 474eae2895120500d06bf6de9725efeeff498f60385Chris Lattner if ((NumBits >> 3) == StrLen + 1 && (NumBits & 7) == 0 && 475490b8f5d64823d998fc12ad3a91ecacd82c1901eChandler Carruth (isa<IntegerType>(Ty) || Ty->isFloatingPointTy())) { 476fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner APInt StrVal(NumBits, 0); 477fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner APInt SingleChar(NumBits, 0); 478e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner if (TD->isLittleEndian()) { 479fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner for (signed i = StrLen-1; i >= 0; i--) { 480e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner SingleChar = (uint64_t) Str[i] & UCHAR_MAX; 48162d327e241f59735eed836e98c8f11ffaa6cf6f8Chris Lattner StrVal = (StrVal << 8) | SingleChar; 48262d327e241f59735eed836e98c8f11ffaa6cf6f8Chris Lattner } 483e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner } else { 484fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner for (unsigned i = 0; i < StrLen; i++) { 485e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner SingleChar = (uint64_t) Str[i] & UCHAR_MAX; 486e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner StrVal = (StrVal << 8) | SingleChar; 487e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner } 488e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner // Append NULL at the end. 489e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner SingleChar = 0; 490e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner StrVal = (StrVal << 8) | SingleChar; 49162d327e241f59735eed836e98c8f11ffaa6cf6f8Chris Lattner } 492eae2895120500d06bf6de9725efeeff498f60385Chris Lattner 493eae2895120500d06bf6de9725efeeff498f60385Chris Lattner Constant *Res = ConstantInt::get(CE->getContext(), StrVal); 494eae2895120500d06bf6de9725efeeff498f60385Chris Lattner if (Ty->isFloatingPointTy()) 495eae2895120500d06bf6de9725efeeff498f60385Chris Lattner Res = ConstantExpr::getBitCast(Res, Ty); 496eae2895120500d06bf6de9725efeeff498f60385Chris Lattner return Res; 49762d327e241f59735eed836e98c8f11ffaa6cf6f8Chris Lattner } 498878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner } 499e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner 500e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner // If this load comes from anywhere in a constant global, and if the global 501e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner // is all undef or zero, we know what it loads. 502bd1801b5553c8be3960255a92738464e0010b6f6Dan Gohman if (GlobalVariable *GV = 503bd1801b5553c8be3960255a92738464e0010b6f6Dan Gohman dyn_cast<GlobalVariable>(GetUnderlyingObject(CE, TD))) { 504e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner if (GV->isConstant() && GV->hasDefinitiveInitializer()) { 505db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *ResTy = cast<PointerType>(C->getType())->getElementType(); 506e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner if (GV->getInitializer()->isNullValue()) 507e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner return Constant::getNullValue(ResTy); 508e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner if (isa<UndefValue>(GV->getInitializer())) 509e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner return UndefValue::get(ResTy); 510e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner } 511e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner } 512e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner 513fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner // Try hard to fold loads from bitcasted strange and non-type-safe things. We 514fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner // currently don't do any of this for big endian systems. It can be 515fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner // generalized in the future if someone is interested. 516fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner if (TD && TD->isLittleEndian()) 517fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner return FoldReinterpretLoadFromConstPtr(CE, *TD); 518878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner return 0; 519878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner} 520878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner 521878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattnerstatic Constant *ConstantFoldLoadInst(const LoadInst *LI, const TargetData *TD){ 522878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner if (LI->isVolatile()) return 0; 523878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner 524878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner if (Constant *C = dyn_cast<Constant>(LI->getOperand(0))) 525878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner return ConstantFoldLoadFromConstPtr(C, TD); 526fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 527878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner return 0; 528878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner} 52903dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner 53003dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner/// SymbolicallyEvaluateBinop - One of Op0/Op1 is a constant expression. 53167e3566d8f6875e9c92d224f493023408a66227eNick Lewycky/// Attempt to symbolically evaluate the result of a binary operator merging 53203dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner/// these together. If target data info is available, it is provided as TD, 53303dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner/// otherwise TD is null. 53403dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattnerstatic Constant *SymbolicallyEvaluateBinop(unsigned Opc, Constant *Op0, 5357b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner Constant *Op1, const TargetData *TD){ 53603dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner // SROA 53703dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner 53803dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner // Fold (and 0xffffffff00000000, (shl x, 32)) -> shl. 53903dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner // Fold (lshr (or X, Y), 32) -> (lshr [X/Y], 32) if one doesn't contribute 54003dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner // bits. 54103dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner 54203dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner 54303dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner // If the constant expr is something like &A[123] - &A[4].f, fold this into a 54403dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner // constant. This happens frequently when iterating over a global array. 54503dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner if (Opc == Instruction::Sub && TD) { 54603dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner GlobalValue *GV1, *GV2; 54703dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner int64_t Offs1, Offs2; 54803dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner 54903dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner if (IsConstantOffsetFromGlobal(Op0, GV1, Offs1, *TD)) 55003dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner if (IsConstantOffsetFromGlobal(Op1, GV2, Offs2, *TD) && 55103dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner GV1 == GV2) { 55203dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner // (&GV+C1) - (&GV+C2) -> C1-C2, pointer arithmetic cannot overflow. 553eed707b1e6097aac2bb6b3d47271f6300ace7f2eOwen Anderson return ConstantInt::get(Op0->getType(), Offs1-Offs2); 55403dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner } 55503dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner } 55603dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner 55703dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner return 0; 55803dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner} 55903dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner 5604f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman/// CastGEPIndices - If array indices are not pointer-sized integers, 5614f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman/// explicitly cast them so that they aren't implicitly casted by the 5624f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman/// getelementptr. 5631d2f569c3428d70d0cf690c9adb459ad4a3ecff2Jay Foadstatic Constant *CastGEPIndices(ArrayRef<Constant *> Ops, 564618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier Type *ResultTy, const TargetData *TD, 565618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier const TargetLibraryInfo *TLI) { 5664f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman if (!TD) return 0; 567db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *IntPtrTy = TD->getIntPtrType(ResultTy->getContext()); 5684f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman 5694f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman bool Any = false; 5704f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman SmallVector<Constant*, 32> NewIdxs; 5711d2f569c3428d70d0cf690c9adb459ad4a3ecff2Jay Foad for (unsigned i = 1, e = Ops.size(); i != e; ++i) { 5724f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman if ((i == 1 || 5734f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman !isa<StructType>(GetElementPtrInst::getIndexedType(Ops[0]->getType(), 574a9203109f4ac95aa7e9624f2838e3d89623ec902Jay Foad Ops.slice(1, i-1)))) && 5754f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman Ops[i]->getType() != IntPtrTy) { 5764f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman Any = true; 5774f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman NewIdxs.push_back(ConstantExpr::getCast(CastInst::getCastOpcode(Ops[i], 5784f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman true, 5794f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman IntPtrTy, 5804f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman true), 5814f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman Ops[i], IntPtrTy)); 5824f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman } else 5834f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman NewIdxs.push_back(Ops[i]); 5844f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman } 5854f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman if (!Any) return 0; 5864f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman 5874f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman Constant *C = 588dab3d29605a5c83db41b28176273ef55961120c1Jay Foad ConstantExpr::getGetElementPtr(Ops[0], NewIdxs); 5894f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman if (ConstantExpr *CE = dyn_cast<ConstantExpr>(C)) 590618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier if (Constant *Folded = ConstantFoldConstantExpression(CE, TD, TLI)) 5914f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman C = Folded; 5924f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman return C; 5934f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman} 5944f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman 59503dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner/// SymbolicallyEvaluateGEP - If we can symbolically evaluate the specified GEP 59603dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner/// constant expression, do so. 5971d2f569c3428d70d0cf690c9adb459ad4a3ecff2Jay Foadstatic Constant *SymbolicallyEvaluateGEP(ArrayRef<Constant *> Ops, 598618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier Type *ResultTy, const TargetData *TD, 599618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier const TargetLibraryInfo *TLI) { 60003dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner Constant *Ptr = Ops[0]; 6011608769abeb1430dc34f31ffac0d9850f99ae36aNadav Rotem if (!TD || !cast<PointerType>(Ptr->getType())->getElementType()->isSized() || 6021608769abeb1430dc34f31ffac0d9850f99ae36aNadav Rotem !Ptr->getType()->isPointerTy()) 60303dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner return 0; 604f1cadf28991b2643128bfe38990ff235f2cb1491Chris Lattner 605db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *IntPtrTy = TD->getIntPtrType(Ptr->getContext()); 606268e7d7a9446bb89b80472cdb7b9803f16c0191eChris Lattner 607268e7d7a9446bb89b80472cdb7b9803f16c0191eChris Lattner // If this is a constant expr gep that is effectively computing an 608268e7d7a9446bb89b80472cdb7b9803f16c0191eChris Lattner // "offsetof", fold it into 'cast int Size to T*' instead of 'gep 0, 0, 12' 6091d2f569c3428d70d0cf690c9adb459ad4a3ecff2Jay Foad for (unsigned i = 1, e = Ops.size(); i != e; ++i) 6108cd4efb6a5e51b4e77039c17338f290ca3a8ee92Chris Lattner if (!isa<ConstantInt>(Ops[i])) { 6118cd4efb6a5e51b4e77039c17338f290ca3a8ee92Chris Lattner 6128cd4efb6a5e51b4e77039c17338f290ca3a8ee92Chris Lattner // If this is "gep i8* Ptr, (sub 0, V)", fold this as: 6138cd4efb6a5e51b4e77039c17338f290ca3a8ee92Chris Lattner // "inttoptr (sub (ptrtoint Ptr), V)" 6141d2f569c3428d70d0cf690c9adb459ad4a3ecff2Jay Foad if (Ops.size() == 2 && 6158cd4efb6a5e51b4e77039c17338f290ca3a8ee92Chris Lattner cast<PointerType>(ResultTy)->getElementType()->isIntegerTy(8)) { 6168cd4efb6a5e51b4e77039c17338f290ca3a8ee92Chris Lattner ConstantExpr *CE = dyn_cast<ConstantExpr>(Ops[1]); 617156eb0a569a1ebac86ad8438645f690d8a3894c4Chris Lattner assert((CE == 0 || CE->getType() == IntPtrTy) && 618f1cadf28991b2643128bfe38990ff235f2cb1491Chris Lattner "CastGEPIndices didn't canonicalize index types!"); 6198cd4efb6a5e51b4e77039c17338f290ca3a8ee92Chris Lattner if (CE && CE->getOpcode() == Instruction::Sub && 620f1cadf28991b2643128bfe38990ff235f2cb1491Chris Lattner CE->getOperand(0)->isNullValue()) { 6218cd4efb6a5e51b4e77039c17338f290ca3a8ee92Chris Lattner Constant *Res = ConstantExpr::getPtrToInt(Ptr, CE->getType()); 6228cd4efb6a5e51b4e77039c17338f290ca3a8ee92Chris Lattner Res = ConstantExpr::getSub(Res, CE->getOperand(1)); 6238cd4efb6a5e51b4e77039c17338f290ca3a8ee92Chris Lattner Res = ConstantExpr::getIntToPtr(Res, ResultTy); 6248cd4efb6a5e51b4e77039c17338f290ca3a8ee92Chris Lattner if (ConstantExpr *ResCE = dyn_cast<ConstantExpr>(Res)) 625618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier Res = ConstantFoldConstantExpression(ResCE, TD, TLI); 6268cd4efb6a5e51b4e77039c17338f290ca3a8ee92Chris Lattner return Res; 6278cd4efb6a5e51b4e77039c17338f290ca3a8ee92Chris Lattner } 6288cd4efb6a5e51b4e77039c17338f290ca3a8ee92Chris Lattner } 629de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman return 0; 6308cd4efb6a5e51b4e77039c17338f290ca3a8ee92Chris Lattner } 631268e7d7a9446bb89b80472cdb7b9803f16c0191eChris Lattner 632f1cadf28991b2643128bfe38990ff235f2cb1491Chris Lattner unsigned BitWidth = TD->getTypeSizeInBits(IntPtrTy); 6338fbbb3980755d74539a0aed02bc18842ed2bd18dJay Foad APInt Offset = 6348fbbb3980755d74539a0aed02bc18842ed2bd18dJay Foad APInt(BitWidth, TD->getIndexedOffset(Ptr->getType(), 6358fbbb3980755d74539a0aed02bc18842ed2bd18dJay Foad makeArrayRef((Value **)Ops.data() + 1, 6368fbbb3980755d74539a0aed02bc18842ed2bd18dJay Foad Ops.size() - 1))); 637890edda7c22dd8525f9fe44085cfd7080e67dfeeDuncan Sands Ptr = cast<Constant>(Ptr->stripPointerCasts()); 6380891d752a68a25025ffc3339aab1f0ad3221b0edDan Gohman 6390891d752a68a25025ffc3339aab1f0ad3221b0edDan Gohman // If this is a GEP of a GEP, fold it all into a single GEP. 6400891d752a68a25025ffc3339aab1f0ad3221b0edDan Gohman while (GEPOperator *GEP = dyn_cast<GEPOperator>(Ptr)) { 6410891d752a68a25025ffc3339aab1f0ad3221b0edDan Gohman SmallVector<Value *, 4> NestedOps(GEP->op_begin()+1, GEP->op_end()); 642890edda7c22dd8525f9fe44085cfd7080e67dfeeDuncan Sands 643890edda7c22dd8525f9fe44085cfd7080e67dfeeDuncan Sands // Do not try the incorporate the sub-GEP if some index is not a number. 644890edda7c22dd8525f9fe44085cfd7080e67dfeeDuncan Sands bool AllConstantInt = true; 645890edda7c22dd8525f9fe44085cfd7080e67dfeeDuncan Sands for (unsigned i = 0, e = NestedOps.size(); i != e; ++i) 646890edda7c22dd8525f9fe44085cfd7080e67dfeeDuncan Sands if (!isa<ConstantInt>(NestedOps[i])) { 647890edda7c22dd8525f9fe44085cfd7080e67dfeeDuncan Sands AllConstantInt = false; 648890edda7c22dd8525f9fe44085cfd7080e67dfeeDuncan Sands break; 649890edda7c22dd8525f9fe44085cfd7080e67dfeeDuncan Sands } 650890edda7c22dd8525f9fe44085cfd7080e67dfeeDuncan Sands if (!AllConstantInt) 651890edda7c22dd8525f9fe44085cfd7080e67dfeeDuncan Sands break; 652890edda7c22dd8525f9fe44085cfd7080e67dfeeDuncan Sands 6530891d752a68a25025ffc3339aab1f0ad3221b0edDan Gohman Ptr = cast<Constant>(GEP->getOperand(0)); 6540891d752a68a25025ffc3339aab1f0ad3221b0edDan Gohman Offset += APInt(BitWidth, 6558fbbb3980755d74539a0aed02bc18842ed2bd18dJay Foad TD->getIndexedOffset(Ptr->getType(), NestedOps)); 656890edda7c22dd8525f9fe44085cfd7080e67dfeeDuncan Sands Ptr = cast<Constant>(Ptr->stripPointerCasts()); 6570891d752a68a25025ffc3339aab1f0ad3221b0edDan Gohman } 6580891d752a68a25025ffc3339aab1f0ad3221b0edDan Gohman 659de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman // If the base value for this address is a literal integer value, fold the 660de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman // getelementptr to the resulting integer value casted to the pointer type. 66140b037ef611efc5c43ffc43a911d0eb57e0d444dDan Gohman APInt BasePtr(BitWidth, 0); 66240b037ef611efc5c43ffc43a911d0eb57e0d444dDan Gohman if (ConstantExpr *CE = dyn_cast<ConstantExpr>(Ptr)) 66340b037ef611efc5c43ffc43a911d0eb57e0d444dDan Gohman if (CE->getOpcode() == Instruction::IntToPtr) 66440f8f6264d5af2c38e797e0dc59827cd231e8ff7Jay Foad if (ConstantInt *Base = dyn_cast<ConstantInt>(CE->getOperand(0))) 66540f8f6264d5af2c38e797e0dc59827cd231e8ff7Jay Foad BasePtr = Base->getValue().zextOrTrunc(BitWidth); 66640b037ef611efc5c43ffc43a911d0eb57e0d444dDan Gohman if (Ptr->isNullValue() || BasePtr != 0) { 6677b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner Constant *C = ConstantInt::get(Ptr->getContext(), Offset+BasePtr); 668de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman return ConstantExpr::getIntToPtr(C, ResultTy); 669de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman } 670de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman 671de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman // Otherwise form a regular getelementptr. Recompute the indices so that 672de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman // we eliminate over-indexing of the notional static type array bounds. 673de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman // This makes it easy to determine if the getelementptr is "inbounds". 674de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman // Also, this helps GlobalOpt do SROA on GlobalVariables. 675db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *Ty = Ptr->getType(); 676de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman SmallVector<Constant*, 32> NewIdxs; 6773d01334e6fa5caba1361d088fabf728c64b23c58Dan Gohman do { 678db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner if (SequentialType *ATy = dyn_cast<SequentialType>(Ty)) { 6791df9859c40492511b8aa4321eb76496005d3b75bDuncan Sands if (ATy->isPointerTy()) { 680e568fa2d29c58aaedcaa832f805949d5e4371b4bChris Lattner // The only pointer indexing we'll do is on the first index of the GEP. 681e568fa2d29c58aaedcaa832f805949d5e4371b4bChris Lattner if (!NewIdxs.empty()) 682e568fa2d29c58aaedcaa832f805949d5e4371b4bChris Lattner break; 683e568fa2d29c58aaedcaa832f805949d5e4371b4bChris Lattner 684e568fa2d29c58aaedcaa832f805949d5e4371b4bChris Lattner // Only handle pointers to sized types, not pointers to functions. 685e568fa2d29c58aaedcaa832f805949d5e4371b4bChris Lattner if (!ATy->getElementType()->isSized()) 686e568fa2d29c58aaedcaa832f805949d5e4371b4bChris Lattner return 0; 687e568fa2d29c58aaedcaa832f805949d5e4371b4bChris Lattner } 688e568fa2d29c58aaedcaa832f805949d5e4371b4bChris Lattner 689de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman // Determine which element of the array the offset points into. 690cda9706cb74a65d2feb2175613cb0fe46acc5aa2Dan Gohman APInt ElemSize(BitWidth, TD->getTypeAllocSize(ATy->getElementType())); 691db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner IntegerType *IntPtrTy = TD->getIntPtrType(Ty->getContext()); 692de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman if (ElemSize == 0) 693c1da204c433459713cbf4c62f0addf458e6b4b9fChris Lattner // The element size is 0. This may be [0 x Ty]*, so just use a zero 694e58391073f6e17cd1024f737618a6f6b8d1323ceDuncan Sands // index for this level and proceed to the next level to see if it can 695e58391073f6e17cd1024f737618a6f6b8d1323ceDuncan Sands // accommodate the offset. 696c1da204c433459713cbf4c62f0addf458e6b4b9fChris Lattner NewIdxs.push_back(ConstantInt::get(IntPtrTy, 0)); 697c1da204c433459713cbf4c62f0addf458e6b4b9fChris Lattner else { 698c1da204c433459713cbf4c62f0addf458e6b4b9fChris Lattner // The element size is non-zero divide the offset by the element 699c1da204c433459713cbf4c62f0addf458e6b4b9fChris Lattner // size (rounding down), to compute the index at this level. 700c1da204c433459713cbf4c62f0addf458e6b4b9fChris Lattner APInt NewIdx = Offset.udiv(ElemSize); 701c1da204c433459713cbf4c62f0addf458e6b4b9fChris Lattner Offset -= NewIdx * ElemSize; 702c1da204c433459713cbf4c62f0addf458e6b4b9fChris Lattner NewIdxs.push_back(ConstantInt::get(IntPtrTy, NewIdx)); 703c1da204c433459713cbf4c62f0addf458e6b4b9fChris Lattner } 704de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman Ty = ATy->getElementType(); 705db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner } else if (StructType *STy = dyn_cast<StructType>(Ty)) { 706cda9706cb74a65d2feb2175613cb0fe46acc5aa2Dan Gohman // Determine which field of the struct the offset points into. The 707cda9706cb74a65d2feb2175613cb0fe46acc5aa2Dan Gohman // getZExtValue is at least as safe as the StructLayout API because we 708cda9706cb74a65d2feb2175613cb0fe46acc5aa2Dan Gohman // know the offset is within the struct at this point. 709de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman const StructLayout &SL = *TD->getStructLayout(STy); 710cda9706cb74a65d2feb2175613cb0fe46acc5aa2Dan Gohman unsigned ElIdx = SL.getElementContainingOffset(Offset.getZExtValue()); 7117b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner NewIdxs.push_back(ConstantInt::get(Type::getInt32Ty(Ty->getContext()), 7127b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner ElIdx)); 713cda9706cb74a65d2feb2175613cb0fe46acc5aa2Dan Gohman Offset -= APInt(BitWidth, SL.getElementOffset(ElIdx)); 714de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman Ty = STy->getTypeAtIndex(ElIdx); 715de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman } else { 7163d01334e6fa5caba1361d088fabf728c64b23c58Dan Gohman // We've reached some non-indexable type. 7173d01334e6fa5caba1361d088fabf728c64b23c58Dan Gohman break; 718de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman } 7193d01334e6fa5caba1361d088fabf728c64b23c58Dan Gohman } while (Ty != cast<PointerType>(ResultTy)->getElementType()); 7203d01334e6fa5caba1361d088fabf728c64b23c58Dan Gohman 7213d01334e6fa5caba1361d088fabf728c64b23c58Dan Gohman // If we haven't used up the entire offset by descending the static 7223d01334e6fa5caba1361d088fabf728c64b23c58Dan Gohman // type, then the offset is pointing into the middle of an indivisible 7233d01334e6fa5caba1361d088fabf728c64b23c58Dan Gohman // member, so we can't simplify it. 7243d01334e6fa5caba1361d088fabf728c64b23c58Dan Gohman if (Offset != 0) 7253d01334e6fa5caba1361d088fabf728c64b23c58Dan Gohman return 0; 726de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman 7273bfbc4587a7e79f08f8c126a9e62c3475fb90f8bDan Gohman // Create a GEP. 7283bfbc4587a7e79f08f8c126a9e62c3475fb90f8bDan Gohman Constant *C = 729dab3d29605a5c83db41b28176273ef55961120c1Jay Foad ConstantExpr::getGetElementPtr(Ptr, NewIdxs); 7306e7ad958683f34bf6c014c88fef723e5a2d741beDan Gohman assert(cast<PointerType>(C->getType())->getElementType() == Ty && 7316e7ad958683f34bf6c014c88fef723e5a2d741beDan Gohman "Computed GetElementPtr has unexpected type!"); 7323d01334e6fa5caba1361d088fabf728c64b23c58Dan Gohman 7333d01334e6fa5caba1361d088fabf728c64b23c58Dan Gohman // If we ended up indexing a member with a type that doesn't match 7344c0d5d5db876b0628bdf6a2174263a1c0a9130e2Dan Gohman // the type of what the original indices indexed, add a cast. 7353d01334e6fa5caba1361d088fabf728c64b23c58Dan Gohman if (Ty != cast<PointerType>(ResultTy)->getElementType()) 7366333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner C = FoldBitCast(C, ResultTy, *TD); 737de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman 7383d01334e6fa5caba1361d088fabf728c64b23c58Dan Gohman return C; 73903dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner} 74003dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner 7411afab9c1e0ae2d145ce01718aa14065393117e70Chris Lattner 74203dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner 74303dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner//===----------------------------------------------------------------------===// 74403dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner// Constant Folding public APIs 74503dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner//===----------------------------------------------------------------------===// 74603dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner 747b9b369fa9983843c4ed77b3a35b6e9b7933955bbDuncan Sands/// ConstantFoldInstruction - Try to constant fold the specified instruction. 748b9b369fa9983843c4ed77b3a35b6e9b7933955bbDuncan Sands/// If successful, the constant result is returned, if not, null is returned. 749b9b369fa9983843c4ed77b3a35b6e9b7933955bbDuncan Sands/// Note that this fails if not all of the operands are constant. Otherwise, 750b9b369fa9983843c4ed77b3a35b6e9b7933955bbDuncan Sands/// this function can only fail when attempting to fold instructions like loads 751b9b369fa9983843c4ed77b3a35b6e9b7933955bbDuncan Sands/// and stores, which have no constant expression form. 752618c1dbd293d15ee19f61b1156ab8086ad28311aChad RosierConstant *llvm::ConstantFoldInstruction(Instruction *I, 753618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier const TargetData *TD, 754618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier const TargetLibraryInfo *TLI) { 755b9b369fa9983843c4ed77b3a35b6e9b7933955bbDuncan Sands // Handle PHI nodes quickly here... 7565520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner if (PHINode *PN = dyn_cast<PHINode>(I)) { 757c0362d5c6e0066c741bce056a65d8b4c026de19fDuncan Sands Constant *CommonValue = 0; 758c0362d5c6e0066c741bce056a65d8b4c026de19fDuncan Sands 759c0362d5c6e0066c741bce056a65d8b4c026de19fDuncan Sands for (unsigned i = 0, e = PN->getNumIncomingValues(); i != e; ++i) { 760c0362d5c6e0066c741bce056a65d8b4c026de19fDuncan Sands Value *Incoming = PN->getIncomingValue(i); 761b9b369fa9983843c4ed77b3a35b6e9b7933955bbDuncan Sands // If the incoming value is undef then skip it. Note that while we could 762b9b369fa9983843c4ed77b3a35b6e9b7933955bbDuncan Sands // skip the value if it is equal to the phi node itself we choose not to 763b9b369fa9983843c4ed77b3a35b6e9b7933955bbDuncan Sands // because that would break the rule that constant folding only applies if 764b9b369fa9983843c4ed77b3a35b6e9b7933955bbDuncan Sands // all operands are constants. 765b9b369fa9983843c4ed77b3a35b6e9b7933955bbDuncan Sands if (isa<UndefValue>(Incoming)) 766c0362d5c6e0066c741bce056a65d8b4c026de19fDuncan Sands continue; 767c0362d5c6e0066c741bce056a65d8b4c026de19fDuncan Sands // If the incoming value is not a constant, or is a different constant to 768c0362d5c6e0066c741bce056a65d8b4c026de19fDuncan Sands // the one we saw previously, then give up. 769c0362d5c6e0066c741bce056a65d8b4c026de19fDuncan Sands Constant *C = dyn_cast<Constant>(Incoming); 770c0362d5c6e0066c741bce056a65d8b4c026de19fDuncan Sands if (!C || (CommonValue && C != CommonValue)) 771c0362d5c6e0066c741bce056a65d8b4c026de19fDuncan Sands return 0; 772c0362d5c6e0066c741bce056a65d8b4c026de19fDuncan Sands CommonValue = C; 773c0362d5c6e0066c741bce056a65d8b4c026de19fDuncan Sands } 7745520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner 775c0362d5c6e0066c741bce056a65d8b4c026de19fDuncan Sands // If we reach here, all incoming values are the same constant or undef. 776c0362d5c6e0066c741bce056a65d8b4c026de19fDuncan Sands return CommonValue ? CommonValue : UndefValue::get(PN->getType()); 7775520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner } 7785520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner 7795520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner // Scan the operand list, checking to see if they are all constants, if so, 7805520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner // hand off to ConstantFoldInstOperands. 7815520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner SmallVector<Constant*, 8> Ops; 782de2d74b213844b17eb5327fba27f21e699d3af66Gabor Greif for (User::op_iterator i = I->op_begin(), e = I->op_end(); i != e; ++i) 783de2d74b213844b17eb5327fba27f21e699d3af66Gabor Greif if (Constant *Op = dyn_cast<Constant>(*i)) 7845520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner Ops.push_back(Op); 7855520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner else 7865520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner return 0; // All operands not constant! 7875520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner 788f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner if (const CmpInst *CI = dyn_cast<CmpInst>(I)) 7898f73deaa8732a556046bf4ac6207be55972e3b74Chris Lattner return ConstantFoldCompareInstOperands(CI->getPredicate(), Ops[0], Ops[1], 790618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier TD, TLI); 79158665d47aea69875c433fb039c8f10106abc1ceeChris Lattner 792878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner if (const LoadInst *LI = dyn_cast<LoadInst>(I)) 793878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner return ConstantFoldLoadInst(LI, TD); 7943ee0af35842999c1781052fe74fdcb19997de950Frits van Bommel 7953ee0af35842999c1781052fe74fdcb19997de950Frits van Bommel if (InsertValueInst *IVI = dyn_cast<InsertValueInst>(I)) 7963ee0af35842999c1781052fe74fdcb19997de950Frits van Bommel return ConstantExpr::getInsertValue( 7973ee0af35842999c1781052fe74fdcb19997de950Frits van Bommel cast<Constant>(IVI->getAggregateOperand()), 7983ee0af35842999c1781052fe74fdcb19997de950Frits van Bommel cast<Constant>(IVI->getInsertedValueOperand()), 799fc6d3a49867cd38954dc40936a88f1907252c6d2Jay Foad IVI->getIndices()); 8003ee0af35842999c1781052fe74fdcb19997de950Frits van Bommel 8013ee0af35842999c1781052fe74fdcb19997de950Frits van Bommel if (ExtractValueInst *EVI = dyn_cast<ExtractValueInst>(I)) 8023ee0af35842999c1781052fe74fdcb19997de950Frits van Bommel return ConstantExpr::getExtractValue( 8033ee0af35842999c1781052fe74fdcb19997de950Frits van Bommel cast<Constant>(EVI->getAggregateOperand()), 804fc6d3a49867cd38954dc40936a88f1907252c6d2Jay Foad EVI->getIndices()); 8053ee0af35842999c1781052fe74fdcb19997de950Frits van Bommel 806618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier return ConstantFoldInstOperands(I->getOpcode(), I->getType(), Ops, TD, TLI); 8075520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner} 8085520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner 8093dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky/// ConstantFoldConstantExpression - Attempt to fold the constant expression 8103dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky/// using the specified TargetData. If successful, the constant result is 8113dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky/// result is returned, if not, null is returned. 812baf0c67988feec5b7698f89dce499ddb7d0c3c42Dan GohmanConstant *llvm::ConstantFoldConstantExpression(const ConstantExpr *CE, 813618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier const TargetData *TD, 814618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier const TargetLibraryInfo *TLI) { 8153dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky SmallVector<Constant*, 8> Ops; 816c1da204c433459713cbf4c62f0addf458e6b4b9fChris Lattner for (User::const_op_iterator i = CE->op_begin(), e = CE->op_end(); 817c1da204c433459713cbf4c62f0addf458e6b4b9fChris Lattner i != e; ++i) { 81801b97dd01eec1197d79fceb7229f5ec233994de3Dan Gohman Constant *NewC = cast<Constant>(*i); 81901b97dd01eec1197d79fceb7229f5ec233994de3Dan Gohman // Recursively fold the ConstantExpr's operands. 82001b97dd01eec1197d79fceb7229f5ec233994de3Dan Gohman if (ConstantExpr *NewCE = dyn_cast<ConstantExpr>(NewC)) 821618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier NewC = ConstantFoldConstantExpression(NewCE, TD, TLI); 82201b97dd01eec1197d79fceb7229f5ec233994de3Dan Gohman Ops.push_back(NewC); 82301b97dd01eec1197d79fceb7229f5ec233994de3Dan Gohman } 8243dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky 8253dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky if (CE->isCompare()) 8268f73deaa8732a556046bf4ac6207be55972e3b74Chris Lattner return ConstantFoldCompareInstOperands(CE->getPredicate(), Ops[0], Ops[1], 827618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier TD, TLI); 828618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier return ConstantFoldInstOperands(CE->getOpcode(), CE->getType(), Ops, TD, TLI); 8293dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky} 8303dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky 8315520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner/// ConstantFoldInstOperands - Attempt to constant fold an instruction with the 8325520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner/// specified opcode and operands. If successful, the constant result is 8335520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner/// returned, if not, null is returned. Note that this function can fail when 8345520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner/// attempting to fold instructions like loads and stores, which have no 8355520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner/// constant expression form. 8365520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner/// 83701b97dd01eec1197d79fceb7229f5ec233994de3Dan Gohman/// TODO: This function neither utilizes nor preserves nsw/nuw/inbounds/etc 83801b97dd01eec1197d79fceb7229f5ec233994de3Dan Gohman/// information, due to only being passed an opcode and operands. Constant 83901b97dd01eec1197d79fceb7229f5ec233994de3Dan Gohman/// folding using this function strips this information. 84001b97dd01eec1197d79fceb7229f5ec233994de3Dan Gohman/// 841db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris LattnerConstant *llvm::ConstantFoldInstOperands(unsigned Opcode, Type *DestTy, 8421d2f569c3428d70d0cf690c9adb459ad4a3ecff2Jay Foad ArrayRef<Constant *> Ops, 843618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier const TargetData *TD, 844618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier const TargetLibraryInfo *TLI) { 84503dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner // Handle easy binops first. 846f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner if (Instruction::isBinaryOp(Opcode)) { 84703dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner if (isa<ConstantExpr>(Ops[0]) || isa<ConstantExpr>(Ops[1])) 8487b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner if (Constant *C = SymbolicallyEvaluateBinop(Opcode, Ops[0], Ops[1], TD)) 84903dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner return C; 85003dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner 851baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson return ConstantExpr::get(Opcode, Ops[0], Ops[1]); 85203dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner } 8535520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner 854f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner switch (Opcode) { 8555520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner default: return 0; 85679fa3cf451a74dd1cec0be2d77d7ea9737dffc69Chris Lattner case Instruction::ICmp: 85779fa3cf451a74dd1cec0be2d77d7ea9737dffc69Chris Lattner case Instruction::FCmp: assert(0 && "Invalid for compares"); 8585520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner case Instruction::Call: 8591d2f569c3428d70d0cf690c9adb459ad4a3ecff2Jay Foad if (Function *F = dyn_cast<Function>(Ops.back())) 8605520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner if (canConstantFoldCallTo(F)) 861618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier return ConstantFoldCall(F, Ops.slice(0, Ops.size() - 1), TLI); 8625520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner return 0; 863001f7534e0311508cc93831a595785bb177f8a18Chris Lattner case Instruction::PtrToInt: 864001f7534e0311508cc93831a595785bb177f8a18Chris Lattner // If the input is a inttoptr, eliminate the pair. This requires knowing 865001f7534e0311508cc93831a595785bb177f8a18Chris Lattner // the width of a pointer, so it can't be done in ConstantExpr::getCast. 866001f7534e0311508cc93831a595785bb177f8a18Chris Lattner if (ConstantExpr *CE = dyn_cast<ConstantExpr>(Ops[0])) { 867001f7534e0311508cc93831a595785bb177f8a18Chris Lattner if (TD && CE->getOpcode() == Instruction::IntToPtr) { 868001f7534e0311508cc93831a595785bb177f8a18Chris Lattner Constant *Input = CE->getOperand(0); 8696de29f8d960505421d61c80cdb738e16720b6c0eDan Gohman unsigned InWidth = Input->getType()->getScalarSizeInBits(); 87004aa2c3520c049333ca216fb794d700310c4a33dNick Lewycky if (TD->getPointerSizeInBits() < InWidth) { 87104aa2c3520c049333ca216fb794d700310c4a33dNick Lewycky Constant *Mask = 8727b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner ConstantInt::get(CE->getContext(), APInt::getLowBitsSet(InWidth, 87304aa2c3520c049333ca216fb794d700310c4a33dNick Lewycky TD->getPointerSizeInBits())); 874baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson Input = ConstantExpr::getAnd(Input, Mask); 87504aa2c3520c049333ca216fb794d700310c4a33dNick Lewycky } 876001f7534e0311508cc93831a595785bb177f8a18Chris Lattner // Do a zext or trunc to get to the dest size. 877baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson return ConstantExpr::getIntegerCast(Input, DestTy, false); 878001f7534e0311508cc93831a595785bb177f8a18Chris Lattner } 879001f7534e0311508cc93831a595785bb177f8a18Chris Lattner } 880baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson return ConstantExpr::getCast(Opcode, Ops[0], DestTy); 881001f7534e0311508cc93831a595785bb177f8a18Chris Lattner case Instruction::IntToPtr: 88281b06be055e74a3c23a1c8f17ead97f9f76335eeDuncan Sands // If the input is a ptrtoint, turn the pair into a ptr to ptr bitcast if 88381b06be055e74a3c23a1c8f17ead97f9f76335eeDuncan Sands // the int size is >= the ptr size. This requires knowing the width of a 88481b06be055e74a3c23a1c8f17ead97f9f76335eeDuncan Sands // pointer, so it can't be done in ConstantExpr::getCast. 885b80a2a686fe76496d71397f8bdda394d5718ab01Dan Gohman if (ConstantExpr *CE = dyn_cast<ConstantExpr>(Ops[0])) 8869a38e3e3991ea443e555d8060f91202a786acdd4Dan Gohman if (TD && 887b80a2a686fe76496d71397f8bdda394d5718ab01Dan Gohman TD->getPointerSizeInBits() <= CE->getType()->getScalarSizeInBits() && 888b80a2a686fe76496d71397f8bdda394d5718ab01Dan Gohman CE->getOpcode() == Instruction::PtrToInt) 889b80a2a686fe76496d71397f8bdda394d5718ab01Dan Gohman return FoldBitCast(CE->getOperand(0), DestTy, *TD); 890b80a2a686fe76496d71397f8bdda394d5718ab01Dan Gohman 891baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson return ConstantExpr::getCast(Opcode, Ops[0], DestTy); 8925520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner case Instruction::Trunc: 8935520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner case Instruction::ZExt: 8945520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner case Instruction::SExt: 8955520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner case Instruction::FPTrunc: 8965520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner case Instruction::FPExt: 8975520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner case Instruction::UIToFP: 8985520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner case Instruction::SIToFP: 8995520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner case Instruction::FPToUI: 9005520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner case Instruction::FPToSI: 901baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson return ConstantExpr::getCast(Opcode, Ops[0], DestTy); 9025520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner case Instruction::BitCast: 9031afab9c1e0ae2d145ce01718aa14065393117e70Chris Lattner if (TD) 9046333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner return FoldBitCast(Ops[0], DestTy, *TD); 905baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson return ConstantExpr::getBitCast(Ops[0], DestTy); 9065520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner case Instruction::Select: 907baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson return ConstantExpr::getSelect(Ops[0], Ops[1], Ops[2]); 9085520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner case Instruction::ExtractElement: 909baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson return ConstantExpr::getExtractElement(Ops[0], Ops[1]); 9105520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner case Instruction::InsertElement: 911baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson return ConstantExpr::getInsertElement(Ops[0], Ops[1], Ops[2]); 9125520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner case Instruction::ShuffleVector: 913baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson return ConstantExpr::getShuffleVector(Ops[0], Ops[1], Ops[2]); 9145520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner case Instruction::GetElementPtr: 915618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier if (Constant *C = CastGEPIndices(Ops, DestTy, TD, TLI)) 9164f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman return C; 917618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier if (Constant *C = SymbolicallyEvaluateGEP(Ops, DestTy, TD, TLI)) 91803dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner return C; 91903dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner 920dab3d29605a5c83db41b28176273ef55961120c1Jay Foad return ConstantExpr::getGetElementPtr(Ops[0], Ops.slice(1)); 9215520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner } 9225520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner} 9235520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner 924f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner/// ConstantFoldCompareInstOperands - Attempt to constant fold a compare 925f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner/// instruction (icmp/fcmp) with the specified operands. If it fails, it 926f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner/// returns a constant expression of the specified operands. 927f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner/// 928f286f6fd93d569befe6e77c94a947e6e04e95685Chris LattnerConstant *llvm::ConstantFoldCompareInstOperands(unsigned Predicate, 9298f73deaa8732a556046bf4ac6207be55972e3b74Chris Lattner Constant *Ops0, Constant *Ops1, 930618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier const TargetData *TD, 931618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier const TargetLibraryInfo *TLI) { 932f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner // fold: icmp (inttoptr x), null -> icmp x, 0 933f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner // fold: icmp (ptrtoint x), 0 -> icmp x, null 9343dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky // fold: icmp (inttoptr x), (inttoptr y) -> icmp trunc/zext x, trunc/zext y 935f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner // fold: icmp (ptrtoint x), (ptrtoint y) -> icmp x, y 936f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner // 937f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner // ConstantExpr::getCompare cannot do this, because it doesn't have TD 938f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner // around to know if bit truncation is happening. 9398f73deaa8732a556046bf4ac6207be55972e3b74Chris Lattner if (ConstantExpr *CE0 = dyn_cast<ConstantExpr>(Ops0)) { 9408f73deaa8732a556046bf4ac6207be55972e3b74Chris Lattner if (TD && Ops1->isNullValue()) { 941db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *IntPtrTy = TD->getIntPtrType(CE0->getContext()); 942f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner if (CE0->getOpcode() == Instruction::IntToPtr) { 943f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner // Convert the integer value to the right size to ensure we get the 944f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner // proper extension or truncation. 945baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson Constant *C = ConstantExpr::getIntegerCast(CE0->getOperand(0), 946f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner IntPtrTy, false); 9478f73deaa8732a556046bf4ac6207be55972e3b74Chris Lattner Constant *Null = Constant::getNullValue(C->getType()); 948618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier return ConstantFoldCompareInstOperands(Predicate, C, Null, TD, TLI); 949f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner } 950f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner 951f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner // Only do this transformation if the int is intptrty in size, otherwise 952f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner // there is a truncation or extension that we aren't modeling. 953f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner if (CE0->getOpcode() == Instruction::PtrToInt && 954f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner CE0->getType() == IntPtrTy) { 955f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner Constant *C = CE0->getOperand(0); 9568f73deaa8732a556046bf4ac6207be55972e3b74Chris Lattner Constant *Null = Constant::getNullValue(C->getType()); 957618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier return ConstantFoldCompareInstOperands(Predicate, C, Null, TD, TLI); 958f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner } 959f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner } 960f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner 9618f73deaa8732a556046bf4ac6207be55972e3b74Chris Lattner if (ConstantExpr *CE1 = dyn_cast<ConstantExpr>(Ops1)) { 9623dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky if (TD && CE0->getOpcode() == CE1->getOpcode()) { 963db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *IntPtrTy = TD->getIntPtrType(CE0->getContext()); 9643dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky 9653dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky if (CE0->getOpcode() == Instruction::IntToPtr) { 9663dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky // Convert the integer value to the right size to ensure we get the 9673dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky // proper extension or truncation. 968baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson Constant *C0 = ConstantExpr::getIntegerCast(CE0->getOperand(0), 9693dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky IntPtrTy, false); 970baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson Constant *C1 = ConstantExpr::getIntegerCast(CE1->getOperand(0), 9713dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky IntPtrTy, false); 972618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier return ConstantFoldCompareInstOperands(Predicate, C0, C1, TD, TLI); 9733dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky } 9743dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky 9753dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky // Only do this transformation if the int is intptrty in size, otherwise 9763dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky // there is a truncation or extension that we aren't modeling. 9773dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky if ((CE0->getOpcode() == Instruction::PtrToInt && 9783dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky CE0->getType() == IntPtrTy && 9798f73deaa8732a556046bf4ac6207be55972e3b74Chris Lattner CE0->getOperand(0)->getType() == CE1->getOperand(0)->getType())) 9808f73deaa8732a556046bf4ac6207be55972e3b74Chris Lattner return ConstantFoldCompareInstOperands(Predicate, CE0->getOperand(0), 981618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier CE1->getOperand(0), TD, TLI); 982f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner } 983f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner } 98479fa3cf451a74dd1cec0be2d77d7ea9737dffc69Chris Lattner 98579fa3cf451a74dd1cec0be2d77d7ea9737dffc69Chris Lattner // icmp eq (or x, y), 0 -> (icmp eq x, 0) & (icmp eq y, 0) 98679fa3cf451a74dd1cec0be2d77d7ea9737dffc69Chris Lattner // icmp ne (or x, y), 0 -> (icmp ne x, 0) | (icmp ne y, 0) 98779fa3cf451a74dd1cec0be2d77d7ea9737dffc69Chris Lattner if ((Predicate == ICmpInst::ICMP_EQ || Predicate == ICmpInst::ICMP_NE) && 98879fa3cf451a74dd1cec0be2d77d7ea9737dffc69Chris Lattner CE0->getOpcode() == Instruction::Or && Ops1->isNullValue()) { 98979fa3cf451a74dd1cec0be2d77d7ea9737dffc69Chris Lattner Constant *LHS = 990618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier ConstantFoldCompareInstOperands(Predicate, CE0->getOperand(0), Ops1, 991618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier TD, TLI); 99279fa3cf451a74dd1cec0be2d77d7ea9737dffc69Chris Lattner Constant *RHS = 993618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier ConstantFoldCompareInstOperands(Predicate, CE0->getOperand(1), Ops1, 994618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier TD, TLI); 99579fa3cf451a74dd1cec0be2d77d7ea9737dffc69Chris Lattner unsigned OpC = 99679fa3cf451a74dd1cec0be2d77d7ea9737dffc69Chris Lattner Predicate == ICmpInst::ICMP_EQ ? Instruction::And : Instruction::Or; 99779fa3cf451a74dd1cec0be2d77d7ea9737dffc69Chris Lattner Constant *Ops[] = { LHS, RHS }; 998618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier return ConstantFoldInstOperands(OpC, LHS->getType(), Ops, TD, TLI); 99979fa3cf451a74dd1cec0be2d77d7ea9737dffc69Chris Lattner } 1000f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner } 10018f73deaa8732a556046bf4ac6207be55972e3b74Chris Lattner 10028f73deaa8732a556046bf4ac6207be55972e3b74Chris Lattner return ConstantExpr::getCompare(Predicate, Ops0, Ops1); 1003f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner} 1004f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner 1005f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner 10065520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner/// ConstantFoldLoadThroughGEPConstantExpr - Given a constant and a 10075520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner/// getelementptr constantexpr, return the constant value being addressed by the 10085520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner/// constant expression, or null if something is funny and we can't decide. 10095520732b24a5a321140dd79af70d321c7ff3dec9Chris LattnerConstant *llvm::ConstantFoldLoadThroughGEPConstantExpr(Constant *C, 1010c6f69e94fa46f585b59bb9d7ace3224b0e638c95Dan Gohman ConstantExpr *CE) { 1011a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner if (!CE->getOperand(1)->isNullValue()) 10125520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner return 0; // Do not allow stepping over the value! 10135520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner 1014a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner SmallVector<Constant*, 8> Indices(CE->getNumOperands()-2); 1015a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner for (unsigned i = 2, e = CE->getNumOperands(); i != e; ++i) 1016a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner Indices[i-2] = CE->getOperand(i); 1017a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner return ConstantFoldLoadThroughGEPIndices(C, Indices); 1018a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner} 1019a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner 1020a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner/// ConstantFoldLoadThroughGEPIndices - Given a constant and getelementptr 1021a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner/// indices (with an *implied* zero pointer index that is not in the list), 1022a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner/// return the constant value being addressed by a virtual load, or null if 1023a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner/// something is funny and we can't decide. 1024a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris LattnerConstant *llvm::ConstantFoldLoadThroughGEPIndices(Constant *C, 1025a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner ArrayRef<Constant*> Indices) { 10265520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner // Loop over all of the operands, tracking down which value we are 1027a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner // addressing. 1028a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner for (unsigned i = 0, e = Indices.size(); i != e; ++i) { 1029a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner ConstantInt *Idx = dyn_cast<ConstantInt>(Indices[i]); 1030a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner if (Idx == 0) return 0; 1031a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner 1032a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner uint64_t IdxVal = Idx->getZExtValue(); 1033a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner 1034a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner if (ConstantStruct *CS = dyn_cast<ConstantStruct>(C)) { 1035a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner C = CS->getOperand(IdxVal); 1036a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner } else if (ConstantAggregateZero *CAZ = dyn_cast<ConstantAggregateZero>(C)){ 1037a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner C = CAZ->getElementValue(Idx); 1038a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner } else if (UndefValue *UV = dyn_cast<UndefValue>(C)) { 1039a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner C = UV->getElementValue(Idx); 1040a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner } else if (ConstantArray *CA = dyn_cast<ConstantArray>(C)) { 1041a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner if (IdxVal >= CA->getType()->getNumElements()) 10425520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner return 0; 1043a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner C = CA->getOperand(IdxVal); 1044a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner } else if (ConstantDataSequential *CDS=dyn_cast<ConstantDataSequential>(C)){ 1045a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner if (IdxVal >= CDS->getType()->getNumElements()) 10465520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner return 0; 1047a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner C = CDS->getElementAsConstant(IdxVal); 1048a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner } else if (ConstantVector *CV = dyn_cast<ConstantVector>(C)) { 1049a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner if (IdxVal >= CV->getType()->getNumElements()) 1050a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner return 0; 1051a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner C = CV->getOperand(IdxVal); 10525520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner } else { 10535520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner return 0; 10545520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner } 1055a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner } 10565520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner return C; 10575520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner} 10585520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner 10595520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner 1060bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell//===----------------------------------------------------------------------===// 10615520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner// Constant Folding for Calls 1062bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell// 1063bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell 1064bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell/// canConstantFoldCallTo - Return true if its even possible to fold a call to 1065bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell/// the specified function. 1066bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswellbool 1067fa9b80eb64127b3d9691e18537975635520e51e9Dan Gohmanllvm::canConstantFoldCallTo(const Function *F) { 1068bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell switch (F->getIntrinsicID()) { 10699ab7fb3ba47442d521a5bed09a27a5e8e7a786edDale Johannesen case Intrinsic::sqrt: 107024fbf2bf161b20b13d0a218345bdcd092bdd56f8Chad Rosier case Intrinsic::pow: 10719ab7fb3ba47442d521a5bed09a27a5e8e7a786edDale Johannesen case Intrinsic::powi: 1072e9391fd9b52e93717b365bdd05c471101323a4dfReid Spencer case Intrinsic::bswap: 1073e9391fd9b52e93717b365bdd05c471101323a4dfReid Spencer case Intrinsic::ctpop: 1074e9391fd9b52e93717b365bdd05c471101323a4dfReid Spencer case Intrinsic::ctlz: 1075e9391fd9b52e93717b365bdd05c471101323a4dfReid Spencer case Intrinsic::cttz: 10761614e50d9f6d48a38e4dafb4b1a51fd4b37614f3Evan Phoenix case Intrinsic::sadd_with_overflow: 10776208610fd602ebdb18bb793152899573d0b2b7abFrits van Bommel case Intrinsic::uadd_with_overflow: 10781614e50d9f6d48a38e4dafb4b1a51fd4b37614f3Evan Phoenix case Intrinsic::ssub_with_overflow: 10796208610fd602ebdb18bb793152899573d0b2b7abFrits van Bommel case Intrinsic::usub_with_overflow: 1080eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner case Intrinsic::smul_with_overflow: 10816208610fd602ebdb18bb793152899573d0b2b7abFrits van Bommel case Intrinsic::umul_with_overflow: 1082faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov case Intrinsic::convert_from_fp16: 1083faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov case Intrinsic::convert_to_fp16: 108415ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth case Intrinsic::x86_sse_cvtss2si: 108515ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth case Intrinsic::x86_sse_cvtss2si64: 108615ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth case Intrinsic::x86_sse_cvttss2si: 108715ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth case Intrinsic::x86_sse_cvttss2si64: 108815ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth case Intrinsic::x86_sse2_cvtsd2si: 108915ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth case Intrinsic::x86_sse2_cvtsd2si64: 109015ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth case Intrinsic::x86_sse2_cvttsd2si: 109115ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth case Intrinsic::x86_sse2_cvttsd2si64: 1092bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell return true; 109368a0603dc5d3d79cd4de2195e950af26c3380cd7Chris Lattner default: 109468a0603dc5d3d79cd4de2195e950af26c3380cd7Chris Lattner return false; 109568a0603dc5d3d79cd4de2195e950af26c3380cd7Chris Lattner case 0: break; 1096bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell } 1097bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell 10986f532a988e5c02fcb271fb65e9e7b83f0147d3f2Chris Lattner if (!F->hasName()) return false; 1099f0443c1eb44d737d9bd78962932fc80f74c6113cDaniel Dunbar StringRef Name = F->getName(); 1100c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner 1101c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner // In these cases, the check of the length is required. We don't want to 1102c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner // return true for a name like "cos\0blah" which strcmp would return equal to 1103c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner // "cos", but has length 8. 1104f0443c1eb44d737d9bd78962932fc80f74c6113cDaniel Dunbar switch (Name[0]) { 1105c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner default: return false; 1106c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner case 'a': 1107f0443c1eb44d737d9bd78962932fc80f74c6113cDaniel Dunbar return Name == "acos" || Name == "asin" || 1108f0443c1eb44d737d9bd78962932fc80f74c6113cDaniel Dunbar Name == "atan" || Name == "atan2"; 1109c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner case 'c': 1110f0443c1eb44d737d9bd78962932fc80f74c6113cDaniel Dunbar return Name == "cos" || Name == "ceil" || Name == "cosf" || Name == "cosh"; 1111c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner case 'e': 1112805fa97a0f5ff61624855da37d4881708fb8f0fbChris Lattner return Name == "exp" || Name == "exp2"; 1113c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner case 'f': 1114f0443c1eb44d737d9bd78962932fc80f74c6113cDaniel Dunbar return Name == "fabs" || Name == "fmod" || Name == "floor"; 1115c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner case 'l': 1116f0443c1eb44d737d9bd78962932fc80f74c6113cDaniel Dunbar return Name == "log" || Name == "log10"; 1117c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner case 'p': 1118f0443c1eb44d737d9bd78962932fc80f74c6113cDaniel Dunbar return Name == "pow"; 1119c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner case 's': 1120f0443c1eb44d737d9bd78962932fc80f74c6113cDaniel Dunbar return Name == "sin" || Name == "sinh" || Name == "sqrt" || 1121f0443c1eb44d737d9bd78962932fc80f74c6113cDaniel Dunbar Name == "sinf" || Name == "sqrtf"; 1122c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner case 't': 1123f0443c1eb44d737d9bd78962932fc80f74c6113cDaniel Dunbar return Name == "tan" || Name == "tanh"; 1124bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell } 1125bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell} 1126bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell 112772d88ae5447a3929c97e88f4c806213847b5d988Chris Lattnerstatic Constant *ConstantFoldFP(double (*NativeFP)(double), double V, 1128db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *Ty) { 11293f2f21e640682c604d9322ba0708e122d1fcf5eeDan Gohman sys::llvm_fenv_clearexcept(); 1130bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell V = NativeFP(V); 11313f2f21e640682c604d9322ba0708e122d1fcf5eeDan Gohman if (sys::llvm_fenv_testexcept()) { 11323f2f21e640682c604d9322ba0708e122d1fcf5eeDan Gohman sys::llvm_fenv_clearexcept(); 1133f19f58a9360927f8ef50b9a029e8efb780d3d8a8Chris Lattner return 0; 113443421b3dd70af5b70e71816521f37502c397cc65Dale Johannesen } 1135f19f58a9360927f8ef50b9a029e8efb780d3d8a8Chris Lattner 1136d0806a1519f3bc450d2dea3e6f22db73fc797e4cChris Lattner if (Ty->isFloatTy()) 11377b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFP::get(Ty->getContext(), APFloat((float)V)); 1138d0806a1519f3bc450d2dea3e6f22db73fc797e4cChris Lattner if (Ty->isDoubleTy()) 11397b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFP::get(Ty->getContext(), APFloat(V)); 1140c23197a26f34f559ea9797de51e187087c039c42Torok Edwin llvm_unreachable("Can only constant fold float/double"); 1141bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell} 1142bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell 1143384152444d22d356f7b4090a158da4cbc1f79464Dan Gohmanstatic Constant *ConstantFoldBinaryFP(double (*NativeFP)(double, double), 1144db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner double V, double W, Type *Ty) { 11453f2f21e640682c604d9322ba0708e122d1fcf5eeDan Gohman sys::llvm_fenv_clearexcept(); 1146384152444d22d356f7b4090a158da4cbc1f79464Dan Gohman V = NativeFP(V, W); 11473f2f21e640682c604d9322ba0708e122d1fcf5eeDan Gohman if (sys::llvm_fenv_testexcept()) { 11483f2f21e640682c604d9322ba0708e122d1fcf5eeDan Gohman sys::llvm_fenv_clearexcept(); 1149f19f58a9360927f8ef50b9a029e8efb780d3d8a8Chris Lattner return 0; 115043421b3dd70af5b70e71816521f37502c397cc65Dale Johannesen } 1151f19f58a9360927f8ef50b9a029e8efb780d3d8a8Chris Lattner 1152d0806a1519f3bc450d2dea3e6f22db73fc797e4cChris Lattner if (Ty->isFloatTy()) 11537b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFP::get(Ty->getContext(), APFloat((float)V)); 1154d0806a1519f3bc450d2dea3e6f22db73fc797e4cChris Lattner if (Ty->isDoubleTy()) 11557b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFP::get(Ty->getContext(), APFloat(V)); 1156c23197a26f34f559ea9797de51e187087c039c42Torok Edwin llvm_unreachable("Can only constant fold float/double"); 1157384152444d22d356f7b4090a158da4cbc1f79464Dan Gohman} 1158384152444d22d356f7b4090a158da4cbc1f79464Dan Gohman 115915ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth/// ConstantFoldConvertToInt - Attempt to an SSE floating point to integer 116015ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth/// conversion of a constant floating point. If roundTowardZero is false, the 116115ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth/// default IEEE rounding is used (toward nearest, ties to even). This matches 116215ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth/// the behavior of the non-truncating SSE instructions in the default rounding 116315ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth/// mode. The desired integer type Ty is used to select how many bits are 116415ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth/// available for the result. Returns null if the conversion cannot be 116515ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth/// performed, otherwise returns the Constant value resulting from the 116615ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth/// conversion. 116715ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruthstatic Constant *ConstantFoldConvertToInt(ConstantFP *Op, bool roundTowardZero, 1168db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *Ty) { 116915ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth assert(Op && "Called with NULL operand"); 117015ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth APFloat Val(Op->getValueAPF()); 117115ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth 117215ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth // All of these conversion intrinsics form an integer of at most 64bits. 117315ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth unsigned ResultWidth = cast<IntegerType>(Ty)->getBitWidth(); 117415ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth assert(ResultWidth <= 64 && 117515ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth "Can only constant fold conversions to 64 and 32 bit ints"); 117615ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth 117715ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth uint64_t UIntVal; 117815ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth bool isExact = false; 117915ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth APFloat::roundingMode mode = roundTowardZero? APFloat::rmTowardZero 118015ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth : APFloat::rmNearestTiesToEven; 118115ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth APFloat::opStatus status = Val.convertToInteger(&UIntVal, ResultWidth, 118215ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth /*isSigned=*/true, mode, 118315ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth &isExact); 118415ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth if (status != APFloat::opOK && status != APFloat::opInexact) 118515ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth return 0; 118615ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth return ConstantInt::get(Ty, UIntVal, /*isSigned=*/true); 118715ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth} 118815ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth 1189bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell/// ConstantFoldCall - Attempt to constant fold a call to the specified function 1190bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell/// with the specified arguments, returning null if unsuccessful. 1191bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John CriswellConstant * 1192618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosierllvm::ConstantFoldCall(Function *F, ArrayRef<Constant *> Operands, 1193618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier const TargetLibraryInfo *TLI) { 11946f532a988e5c02fcb271fb65e9e7b83f0147d3f2Chris Lattner if (!F->hasName()) return 0; 1195f0443c1eb44d737d9bd78962932fc80f74c6113cDaniel Dunbar StringRef Name = F->getName(); 1196e65cd40ab99c3bd43dc997873e500643d012bbadChris Lattner 1197db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *Ty = F->getReturnType(); 11981d2f569c3428d70d0cf690c9adb459ad4a3ecff2Jay Foad if (Operands.size() == 1) { 1199bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell if (ConstantFP *Op = dyn_cast<ConstantFP>(Operands[0])) { 1200f4db877cff62340aaf9f29d3994562ae1cee8d72Chandler Carruth if (F->getIntrinsicID() == Intrinsic::convert_to_fp16) { 1201faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov APFloat Val(Op->getValueAPF()); 1202faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov 1203faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov bool lost = false; 1204faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov Val.convert(APFloat::IEEEhalf, APFloat::rmNearestTiesToEven, &lost); 1205faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov 1206faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov return ConstantInt::get(F->getContext(), Val.bitcastToAPInt()); 1207faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov } 120821646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier if (!TLI) 120921646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier return 0; 1210faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov 1211d0806a1519f3bc450d2dea3e6f22db73fc797e4cChris Lattner if (!Ty->isFloatTy() && !Ty->isDoubleTy()) 121243421b3dd70af5b70e71816521f37502c397cc65Dale Johannesen return 0; 12131f386c4ce1983c448320aecca8d3cd27d1497867Jakob Stoklund Olesen 12141f386c4ce1983c448320aecca8d3cd27d1497867Jakob Stoklund Olesen /// We only fold functions with finite arguments. Folding NaN and inf is 12151f386c4ce1983c448320aecca8d3cd27d1497867Jakob Stoklund Olesen /// likely to be aborted with an exception anyway, and some host libms 12161f386c4ce1983c448320aecca8d3cd27d1497867Jakob Stoklund Olesen /// have known errors raising exceptions. 12171f386c4ce1983c448320aecca8d3cd27d1497867Jakob Stoklund Olesen if (Op->getValueAPF().isNaN() || Op->getValueAPF().isInfinity()) 12181f386c4ce1983c448320aecca8d3cd27d1497867Jakob Stoklund Olesen return 0; 12191f386c4ce1983c448320aecca8d3cd27d1497867Jakob Stoklund Olesen 122043421b3dd70af5b70e71816521f37502c397cc65Dale Johannesen /// Currently APFloat versions of these functions do not exist, so we use 122143421b3dd70af5b70e71816521f37502c397cc65Dale Johannesen /// the host native double versions. Float versions are not called 122243421b3dd70af5b70e71816521f37502c397cc65Dale Johannesen /// directly but for all these it is true (float)(f((double)arg)) == 122343421b3dd70af5b70e71816521f37502c397cc65Dale Johannesen /// f(arg). Long double not supported yet. 1224d0806a1519f3bc450d2dea3e6f22db73fc797e4cChris Lattner double V = Ty->isFloatTy() ? (double)Op->getValueAPF().convertToFloat() : 122543421b3dd70af5b70e71816521f37502c397cc65Dale Johannesen Op->getValueAPF().convertToDouble(); 1226f0443c1eb44d737d9bd78962932fc80f74c6113cDaniel Dunbar switch (Name[0]) { 1227c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner case 'a': 122821646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier if (Name == "acos" && TLI->has(LibFunc::acos)) 12297b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(acos, V, Ty); 123021646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier else if (Name == "asin" && TLI->has(LibFunc::asin)) 12317b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(asin, V, Ty); 123221646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier else if (Name == "atan" && TLI->has(LibFunc::atan)) 12337b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(atan, V, Ty); 1234c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner break; 1235c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner case 'c': 123621646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier if (Name == "ceil" && TLI->has(LibFunc::ceil)) 12377b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(ceil, V, Ty); 123821646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier else if (Name == "cos" && TLI->has(LibFunc::cos)) 12397b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(cos, V, Ty); 124021646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier else if (Name == "cosh" && TLI->has(LibFunc::cosh)) 12417b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(cosh, V, Ty); 124221646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier else if (Name == "cosf" && TLI->has(LibFunc::cosf)) 12437b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(cos, V, Ty); 1244c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner break; 1245c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner case 'e': 124621646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier if (Name == "exp" && TLI->has(LibFunc::exp)) 12477b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(exp, V, Ty); 1248805fa97a0f5ff61624855da37d4881708fb8f0fbChris Lattner 124921646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier if (Name == "exp2" && TLI->has(LibFunc::exp2)) { 1250805fa97a0f5ff61624855da37d4881708fb8f0fbChris Lattner // Constant fold exp2(x) as pow(2,x) in case the host doesn't have a 1251805fa97a0f5ff61624855da37d4881708fb8f0fbChris Lattner // C99 library. 1252805fa97a0f5ff61624855da37d4881708fb8f0fbChris Lattner return ConstantFoldBinaryFP(pow, 2.0, V, Ty); 1253805fa97a0f5ff61624855da37d4881708fb8f0fbChris Lattner } 1254c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner break; 1255c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner case 'f': 125621646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier if (Name == "fabs" && TLI->has(LibFunc::fabs)) 12577b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(fabs, V, Ty); 125821646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier else if (Name == "floor" && TLI->has(LibFunc::floor)) 12597b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(floor, V, Ty); 1260c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner break; 1261c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner case 'l': 126221646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier if (Name == "log" && V > 0 && TLI->has(LibFunc::log)) 12637b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(log, V, Ty); 126421646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier else if (Name == "log10" && V > 0 && TLI->has(LibFunc::log10)) 12657b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(log10, V, Ty); 1266f4db877cff62340aaf9f29d3994562ae1cee8d72Chandler Carruth else if (F->getIntrinsicID() == Intrinsic::sqrt && 1267f4db877cff62340aaf9f29d3994562ae1cee8d72Chandler Carruth (Ty->isFloatTy() || Ty->isDoubleTy())) { 1268c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner if (V >= -0.0) 12697b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(sqrt, V, Ty); 1270c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner else // Undefined 1271a7235ea7245028a0723e8ab7fd011386b3900777Owen Anderson return Constant::getNullValue(Ty); 1272c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner } 1273c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner break; 1274c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner case 's': 127521646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier if (Name == "sin" && TLI->has(LibFunc::sin)) 12767b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(sin, V, Ty); 127721646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier else if (Name == "sinh" && TLI->has(LibFunc::sinh)) 12787b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(sinh, V, Ty); 127921646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier else if (Name == "sqrt" && V >= 0 && TLI->has(LibFunc::sqrt)) 12807b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(sqrt, V, Ty); 128121646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier else if (Name == "sqrtf" && V >= 0 && TLI->has(LibFunc::sqrtf)) 12827b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(sqrt, V, Ty); 128321646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier else if (Name == "sinf" && TLI->has(LibFunc::sinf)) 12847b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(sin, V, Ty); 1285c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner break; 1286c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner case 't': 128721646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier if (Name == "tan" && TLI->has(LibFunc::tan)) 12887b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(tan, V, Ty); 128921646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier else if (Name == "tanh" && TLI->has(LibFunc::tanh)) 12907b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(tanh, V, Ty); 1291c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner break; 1292c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner default: 1293c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner break; 1294bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell } 129568a0603dc5d3d79cd4de2195e950af26c3380cd7Chris Lattner return 0; 129668a0603dc5d3d79cd4de2195e950af26c3380cd7Chris Lattner } 1297f4db877cff62340aaf9f29d3994562ae1cee8d72Chandler Carruth 129868a0603dc5d3d79cd4de2195e950af26c3380cd7Chris Lattner if (ConstantInt *Op = dyn_cast<ConstantInt>(Operands[0])) { 1299f4db877cff62340aaf9f29d3994562ae1cee8d72Chandler Carruth switch (F->getIntrinsicID()) { 1300f4db877cff62340aaf9f29d3994562ae1cee8d72Chandler Carruth case Intrinsic::bswap: 13017b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantInt::get(F->getContext(), Op->getValue().byteSwap()); 1302f4db877cff62340aaf9f29d3994562ae1cee8d72Chandler Carruth case Intrinsic::ctpop: 1303eed707b1e6097aac2bb6b3d47271f6300ace7f2eOwen Anderson return ConstantInt::get(Ty, Op->getValue().countPopulation()); 1304f4db877cff62340aaf9f29d3994562ae1cee8d72Chandler Carruth case Intrinsic::convert_from_fp16: { 1305faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov APFloat Val(Op->getValue()); 1306faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov 1307faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov bool lost = false; 1308faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov APFloat::opStatus status = 1309faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov Val.convert(APFloat::IEEEsingle, APFloat::rmNearestTiesToEven, &lost); 1310faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov 1311faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov // Conversion is always precise. 13128e68c3873549ca31533e2e3e40dda3a43cb79566Jeffrey Yasskin (void)status; 1313faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov assert(status == APFloat::opOK && !lost && 1314faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov "Precision lost during fp16 constfolding"); 1315faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov 1316faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov return ConstantFP::get(F->getContext(), Val); 1317faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov } 1318f4db877cff62340aaf9f29d3994562ae1cee8d72Chandler Carruth default: 1319f4db877cff62340aaf9f29d3994562ae1cee8d72Chandler Carruth return 0; 1320f4db877cff62340aaf9f29d3994562ae1cee8d72Chandler Carruth } 1321bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell } 1322f4db877cff62340aaf9f29d3994562ae1cee8d72Chandler Carruth 132315ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth if (ConstantVector *Op = dyn_cast<ConstantVector>(Operands[0])) { 132415ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth switch (F->getIntrinsicID()) { 132515ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth default: break; 132615ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth case Intrinsic::x86_sse_cvtss2si: 132715ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth case Intrinsic::x86_sse_cvtss2si64: 132815ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth case Intrinsic::x86_sse2_cvtsd2si: 132915ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth case Intrinsic::x86_sse2_cvtsd2si64: 133015ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth if (ConstantFP *FPOp = dyn_cast<ConstantFP>(Op->getOperand(0))) 133115ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth return ConstantFoldConvertToInt(FPOp, /*roundTowardZero=*/false, Ty); 133215ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth case Intrinsic::x86_sse_cvttss2si: 133315ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth case Intrinsic::x86_sse_cvttss2si64: 133415ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth case Intrinsic::x86_sse2_cvttsd2si: 133515ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth case Intrinsic::x86_sse2_cvttsd2si64: 133615ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth if (ConstantFP *FPOp = dyn_cast<ConstantFP>(Op->getOperand(0))) 133715ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth return ConstantFoldConvertToInt(FPOp, /*roundTowardZero=*/true, Ty); 133815ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth } 133915ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth } 134015ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth 13419ee7123ab462d97a2b242280e9bf0b47883531cbDan Gohman if (isa<UndefValue>(Operands[0])) { 1342f4db877cff62340aaf9f29d3994562ae1cee8d72Chandler Carruth if (F->getIntrinsicID() == Intrinsic::bswap) 13439ee7123ab462d97a2b242280e9bf0b47883531cbDan Gohman return Operands[0]; 13449ee7123ab462d97a2b242280e9bf0b47883531cbDan Gohman return 0; 13459ee7123ab462d97a2b242280e9bf0b47883531cbDan Gohman } 13469ee7123ab462d97a2b242280e9bf0b47883531cbDan Gohman 134768a0603dc5d3d79cd4de2195e950af26c3380cd7Chris Lattner return 0; 134868a0603dc5d3d79cd4de2195e950af26c3380cd7Chris Lattner } 1349f4db877cff62340aaf9f29d3994562ae1cee8d72Chandler Carruth 13501d2f569c3428d70d0cf690c9adb459ad4a3ecff2Jay Foad if (Operands.size() == 2) { 1351bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell if (ConstantFP *Op1 = dyn_cast<ConstantFP>(Operands[0])) { 1352d0806a1519f3bc450d2dea3e6f22db73fc797e4cChris Lattner if (!Ty->isFloatTy() && !Ty->isDoubleTy()) 13539ab7fb3ba47442d521a5bed09a27a5e8e7a786edDale Johannesen return 0; 1354d0806a1519f3bc450d2dea3e6f22db73fc797e4cChris Lattner double Op1V = Ty->isFloatTy() ? 1355d0806a1519f3bc450d2dea3e6f22db73fc797e4cChris Lattner (double)Op1->getValueAPF().convertToFloat() : 135643421b3dd70af5b70e71816521f37502c397cc65Dale Johannesen Op1->getValueAPF().convertToDouble(); 1357bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell if (ConstantFP *Op2 = dyn_cast<ConstantFP>(Operands[1])) { 1358d0806a1519f3bc450d2dea3e6f22db73fc797e4cChris Lattner if (Op2->getType() != Op1->getType()) 1359d0806a1519f3bc450d2dea3e6f22db73fc797e4cChris Lattner return 0; 136021646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier 1361d0806a1519f3bc450d2dea3e6f22db73fc797e4cChris Lattner double Op2V = Ty->isFloatTy() ? 136243421b3dd70af5b70e71816521f37502c397cc65Dale Johannesen (double)Op2->getValueAPF().convertToFloat(): 136343421b3dd70af5b70e71816521f37502c397cc65Dale Johannesen Op2->getValueAPF().convertToDouble(); 1364bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell 136524fbf2bf161b20b13d0a218345bdcd092bdd56f8Chad Rosier if (F->getIntrinsicID() == Intrinsic::pow) { 136624fbf2bf161b20b13d0a218345bdcd092bdd56f8Chad Rosier return ConstantFoldBinaryFP(pow, Op1V, Op2V, Ty); 136724fbf2bf161b20b13d0a218345bdcd092bdd56f8Chad Rosier } 136824fbf2bf161b20b13d0a218345bdcd092bdd56f8Chad Rosier if (!TLI) 136924fbf2bf161b20b13d0a218345bdcd092bdd56f8Chad Rosier return 0; 137021646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier if (Name == "pow" && TLI->has(LibFunc::pow)) 13717b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldBinaryFP(pow, Op1V, Op2V, Ty); 137221646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier if (Name == "fmod" && TLI->has(LibFunc::fmod)) 13737b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldBinaryFP(fmod, Op1V, Op2V, Ty); 137421646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier if (Name == "atan2" && TLI->has(LibFunc::atan2)) 13757b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldBinaryFP(atan2, Op1V, Op2V, Ty); 1376b5282dcf4745be59046f440980a1c0f0a50c9c09Chris Lattner } else if (ConstantInt *Op2C = dyn_cast<ConstantInt>(Operands[1])) { 1377f4db877cff62340aaf9f29d3994562ae1cee8d72Chandler Carruth if (F->getIntrinsicID() == Intrinsic::powi && Ty->isFloatTy()) 13787b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFP::get(F->getContext(), 13797b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner APFloat((float)std::pow((float)Op1V, 138002a260aa11a2e1b2c14335274d3c42ca3f3eabc0Chris Lattner (int)Op2C->getZExtValue()))); 1381f4db877cff62340aaf9f29d3994562ae1cee8d72Chandler Carruth if (F->getIntrinsicID() == Intrinsic::powi && Ty->isDoubleTy()) 13827b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFP::get(F->getContext(), 13837b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner APFloat((double)std::pow((double)Op1V, 13847b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner (int)Op2C->getZExtValue()))); 1385bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell } 138668a0603dc5d3d79cd4de2195e950af26c3380cd7Chris Lattner return 0; 1387bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell } 1388e65cd40ab99c3bd43dc997873e500643d012bbadChris Lattner 1389e65cd40ab99c3bd43dc997873e500643d012bbadChris Lattner if (ConstantInt *Op1 = dyn_cast<ConstantInt>(Operands[0])) { 1390e65cd40ab99c3bd43dc997873e500643d012bbadChris Lattner if (ConstantInt *Op2 = dyn_cast<ConstantInt>(Operands[1])) { 1391e65cd40ab99c3bd43dc997873e500643d012bbadChris Lattner switch (F->getIntrinsicID()) { 1392e65cd40ab99c3bd43dc997873e500643d012bbadChris Lattner default: break; 1393eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner case Intrinsic::sadd_with_overflow: 1394eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner case Intrinsic::uadd_with_overflow: 1395eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner case Intrinsic::ssub_with_overflow: 1396eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner case Intrinsic::usub_with_overflow: 13976208610fd602ebdb18bb793152899573d0b2b7abFrits van Bommel case Intrinsic::smul_with_overflow: 13986208610fd602ebdb18bb793152899573d0b2b7abFrits van Bommel case Intrinsic::umul_with_overflow: { 1399eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner APInt Res; 1400eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner bool Overflow; 1401eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner switch (F->getIntrinsicID()) { 1402eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner default: assert(0 && "Invalid case"); 1403eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner case Intrinsic::sadd_with_overflow: 1404eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner Res = Op1->getValue().sadd_ov(Op2->getValue(), Overflow); 1405eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner break; 1406eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner case Intrinsic::uadd_with_overflow: 1407eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner Res = Op1->getValue().uadd_ov(Op2->getValue(), Overflow); 1408eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner break; 1409eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner case Intrinsic::ssub_with_overflow: 1410eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner Res = Op1->getValue().ssub_ov(Op2->getValue(), Overflow); 1411eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner break; 1412eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner case Intrinsic::usub_with_overflow: 1413eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner Res = Op1->getValue().usub_ov(Op2->getValue(), Overflow); 1414eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner break; 1415eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner case Intrinsic::smul_with_overflow: 1416eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner Res = Op1->getValue().smul_ov(Op2->getValue(), Overflow); 1417eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner break; 14186208610fd602ebdb18bb793152899573d0b2b7abFrits van Bommel case Intrinsic::umul_with_overflow: 14196208610fd602ebdb18bb793152899573d0b2b7abFrits van Bommel Res = Op1->getValue().umul_ov(Op2->getValue(), Overflow); 14206208610fd602ebdb18bb793152899573d0b2b7abFrits van Bommel break; 1421eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner } 1422e65cd40ab99c3bd43dc997873e500643d012bbadChris Lattner Constant *Ops[] = { 1423eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner ConstantInt::get(F->getContext(), Res), 1424eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner ConstantInt::get(Type::getInt1Ty(F->getContext()), Overflow) 1425e65cd40ab99c3bd43dc997873e500643d012bbadChris Lattner }; 1426b065b06c12dba6001b8140df2744d0c856ef6ea1Chris Lattner return ConstantStruct::get(cast<StructType>(F->getReturnType()), Ops); 1427e65cd40ab99c3bd43dc997873e500643d012bbadChris Lattner } 1428ccbf1e36d34c2d66600748bdd8767dc122629a58Chandler Carruth case Intrinsic::cttz: 1429ccbf1e36d34c2d66600748bdd8767dc122629a58Chandler Carruth // FIXME: This should check for Op2 == 1, and become unreachable if 1430ccbf1e36d34c2d66600748bdd8767dc122629a58Chandler Carruth // Op1 == 0. 1431ccbf1e36d34c2d66600748bdd8767dc122629a58Chandler Carruth return ConstantInt::get(Ty, Op1->getValue().countTrailingZeros()); 1432ccbf1e36d34c2d66600748bdd8767dc122629a58Chandler Carruth case Intrinsic::ctlz: 1433ccbf1e36d34c2d66600748bdd8767dc122629a58Chandler Carruth // FIXME: This should check for Op2 == 1, and become unreachable if 1434ccbf1e36d34c2d66600748bdd8767dc122629a58Chandler Carruth // Op1 == 0. 1435ccbf1e36d34c2d66600748bdd8767dc122629a58Chandler Carruth return ConstantInt::get(Ty, Op1->getValue().countLeadingZeros()); 1436e65cd40ab99c3bd43dc997873e500643d012bbadChris Lattner } 1437e65cd40ab99c3bd43dc997873e500643d012bbadChris Lattner } 1438e65cd40ab99c3bd43dc997873e500643d012bbadChris Lattner 1439e65cd40ab99c3bd43dc997873e500643d012bbadChris Lattner return 0; 1440e65cd40ab99c3bd43dc997873e500643d012bbadChris Lattner } 1441e65cd40ab99c3bd43dc997873e500643d012bbadChris Lattner return 0; 1442bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell } 1443bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell return 0; 1444bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell} 1445