ConstantFolding.cpp revision 1608769abeb1430dc34f31ffac0d9850f99ae36a
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 } 276fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris 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 3506475d9434ff3e981160f85d3f1aa07c9e0ace6f2Anders Carlsson if (ConstantExpr *CE = dyn_cast<ConstantExpr>(C)) { 3515d43ff4e7ea30eaf5a1c417f0ec528900f5c83f2Anders Carlsson if (CE->getOpcode() == Instruction::IntToPtr && 3525d43ff4e7ea30eaf5a1c417f0ec528900f5c83f2Anders Carlsson CE->getOperand(0)->getType() == TD.getIntPtrType(CE->getContext())) 3536475d9434ff3e981160f85d3f1aa07c9e0ace6f2Anders Carlsson return ReadDataFromGlobal(CE->getOperand(0), ByteOffset, CurPtr, 3546475d9434ff3e981160f85d3f1aa07c9e0ace6f2Anders Carlsson BytesLeft, TD); 3556475d9434ff3e981160f85d3f1aa07c9e0ace6f2Anders Carlsson } 3566475d9434ff3e981160f85d3f1aa07c9e0ace6f2Anders Carlsson 357fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner // Otherwise, unknown initializer type. 358fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner return false; 359fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner} 360fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 361fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattnerstatic Constant *FoldReinterpretLoadFromConstPtr(Constant *C, 362fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner const TargetData &TD) { 363db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *LoadTy = cast<PointerType>(C->getType())->getElementType(); 364db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner IntegerType *IntType = dyn_cast<IntegerType>(LoadTy); 365fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 366fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner // If this isn't an integer load we can't fold it directly. 367fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner if (!IntType) { 368fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner // If this is a float/double load, we can try folding it as an int32/64 load 36917f0cd3cc1432231f2bc5601e7e463986a59d817Chris Lattner // and then bitcast the result. This can be useful for union cases. Note 37017f0cd3cc1432231f2bc5601e7e463986a59d817Chris Lattner // that address spaces don't matter here since we're not going to result in 37117f0cd3cc1432231f2bc5601e7e463986a59d817Chris Lattner // an actual new load. 372db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *MapTy; 37317f0cd3cc1432231f2bc5601e7e463986a59d817Chris Lattner if (LoadTy->isFloatTy()) 374fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner MapTy = Type::getInt32PtrTy(C->getContext()); 37517f0cd3cc1432231f2bc5601e7e463986a59d817Chris Lattner else if (LoadTy->isDoubleTy()) 376fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner MapTy = Type::getInt64PtrTy(C->getContext()); 3771df9859c40492511b8aa4321eb76496005d3b75bDuncan Sands else if (LoadTy->isVectorTy()) { 37817f0cd3cc1432231f2bc5601e7e463986a59d817Chris Lattner MapTy = IntegerType::get(C->getContext(), 37917f0cd3cc1432231f2bc5601e7e463986a59d817Chris Lattner TD.getTypeAllocSizeInBits(LoadTy)); 38017f0cd3cc1432231f2bc5601e7e463986a59d817Chris Lattner MapTy = PointerType::getUnqual(MapTy); 38117f0cd3cc1432231f2bc5601e7e463986a59d817Chris Lattner } else 382fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner return 0; 383fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 3846333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner C = FoldBitCast(C, MapTy, TD); 385fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner if (Constant *Res = FoldReinterpretLoadFromConstPtr(C, TD)) 3866333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner return FoldBitCast(Res, LoadTy, TD); 387fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner return 0; 388fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner } 389fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 390fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner unsigned BytesLoaded = (IntType->getBitWidth() + 7) / 8; 391739208a790398cf1f9da05149c768371e48781e8Chris Lattner if (BytesLoaded > 32 || BytesLoaded == 0) return 0; 392fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 393fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner GlobalValue *GVal; 394fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner int64_t Offset; 395fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner if (!IsConstantOffsetFromGlobal(C, GVal, Offset, TD)) 396fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner return 0; 397fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 398fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner GlobalVariable *GV = dyn_cast<GlobalVariable>(GVal); 399c7b1382e351249774be63bd73839e8a0671635e1Chris Lattner if (!GV || !GV->isConstant() || !GV->hasDefinitiveInitializer() || 400fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner !GV->getInitializer()->getType()->isSized()) 401fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner return 0; 402fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 403fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner // If we're loading off the beginning of the global, some bytes may be valid, 404fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner // but we don't try to handle this. 405fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner if (Offset < 0) return 0; 406fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 407fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner // If we're not accessing anything in this constant, the result is undefined. 408fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner if (uint64_t(Offset) >= TD.getTypeAllocSize(GV->getInitializer()->getType())) 409fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner return UndefValue::get(IntType); 410fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 411739208a790398cf1f9da05149c768371e48781e8Chris Lattner unsigned char RawBytes[32] = {0}; 412fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner if (!ReadDataFromGlobal(GV->getInitializer(), Offset, RawBytes, 413fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner BytesLoaded, TD)) 414fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner return 0; 415fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 416b31189f26297fe30214da3e8d14a4575d5fdd262Chris Lattner APInt ResultVal = APInt(IntType->getBitWidth(), RawBytes[BytesLoaded-1]); 417b31189f26297fe30214da3e8d14a4575d5fdd262Chris Lattner for (unsigned i = 1; i != BytesLoaded; ++i) { 418739208a790398cf1f9da05149c768371e48781e8Chris Lattner ResultVal <<= 8; 419ce800511701135ed86845b451693a248c83a7973Dan Gohman ResultVal |= RawBytes[BytesLoaded-1-i]; 420739208a790398cf1f9da05149c768371e48781e8Chris Lattner } 421fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 422739208a790398cf1f9da05149c768371e48781e8Chris Lattner return ConstantInt::get(IntType->getContext(), ResultVal); 423fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner} 424fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 425878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner/// ConstantFoldLoadFromConstPtr - Return the value that a load from C would 426878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner/// produce if it is constant and determinable. If this is not determinable, 427878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner/// return null. 428878e4946700824954d7eb7f3ff660353db8e0d17Chris LattnerConstant *llvm::ConstantFoldLoadFromConstPtr(Constant *C, 429878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner const TargetData *TD) { 430878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner // First, try the easy cases: 431878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner if (GlobalVariable *GV = dyn_cast<GlobalVariable>(C)) 432878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner if (GV->isConstant() && GV->hasDefinitiveInitializer()) 433878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner return GV->getInitializer(); 434878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner 435e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner // If the loaded value isn't a constant expr, we can't handle it. 436e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner ConstantExpr *CE = dyn_cast<ConstantExpr>(C); 437e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner if (!CE) return 0; 438e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner 439e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner if (CE->getOpcode() == Instruction::GetElementPtr) { 440e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner if (GlobalVariable *GV = dyn_cast<GlobalVariable>(CE->getOperand(0))) 441e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner if (GV->isConstant() && GV->hasDefinitiveInitializer()) 442e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner if (Constant *V = 443e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner ConstantFoldLoadThroughGEPConstantExpr(GV->getInitializer(), CE)) 444e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner return V; 445e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner } 446e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner 447e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner // Instead of loading constant c string, use corresponding integer value 448e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner // directly if string length is small enough. 449e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner std::string Str; 45044a7a380aa9fbb303f57f4f8269062b7b56be980Chris Lattner if (TD && GetConstantStringInfo(CE, Str) && !Str.empty()) { 451fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner unsigned StrLen = Str.length(); 452db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *Ty = cast<PointerType>(CE->getType())->getElementType(); 453fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner unsigned NumBits = Ty->getPrimitiveSizeInBits(); 454eae2895120500d06bf6de9725efeeff498f60385Chris Lattner // Replace load with immediate integer if the result is an integer or fp 455eae2895120500d06bf6de9725efeeff498f60385Chris Lattner // value. 456eae2895120500d06bf6de9725efeeff498f60385Chris Lattner if ((NumBits >> 3) == StrLen + 1 && (NumBits & 7) == 0 && 457490b8f5d64823d998fc12ad3a91ecacd82c1901eChandler Carruth (isa<IntegerType>(Ty) || Ty->isFloatingPointTy())) { 458fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner APInt StrVal(NumBits, 0); 459fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner APInt SingleChar(NumBits, 0); 460e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner if (TD->isLittleEndian()) { 461fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner for (signed i = StrLen-1; i >= 0; i--) { 462e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner SingleChar = (uint64_t) Str[i] & UCHAR_MAX; 46362d327e241f59735eed836e98c8f11ffaa6cf6f8Chris Lattner StrVal = (StrVal << 8) | SingleChar; 46462d327e241f59735eed836e98c8f11ffaa6cf6f8Chris Lattner } 465e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner } else { 466fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner for (unsigned i = 0; i < StrLen; i++) { 467e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner SingleChar = (uint64_t) Str[i] & UCHAR_MAX; 468e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner StrVal = (StrVal << 8) | SingleChar; 469e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner } 470e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner // Append NULL at the end. 471e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner SingleChar = 0; 472e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner StrVal = (StrVal << 8) | SingleChar; 47362d327e241f59735eed836e98c8f11ffaa6cf6f8Chris Lattner } 474eae2895120500d06bf6de9725efeeff498f60385Chris Lattner 475eae2895120500d06bf6de9725efeeff498f60385Chris Lattner Constant *Res = ConstantInt::get(CE->getContext(), StrVal); 476eae2895120500d06bf6de9725efeeff498f60385Chris Lattner if (Ty->isFloatingPointTy()) 477eae2895120500d06bf6de9725efeeff498f60385Chris Lattner Res = ConstantExpr::getBitCast(Res, Ty); 478eae2895120500d06bf6de9725efeeff498f60385Chris Lattner return Res; 47962d327e241f59735eed836e98c8f11ffaa6cf6f8Chris Lattner } 480878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner } 481e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner 482e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner // If this load comes from anywhere in a constant global, and if the global 483e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner // is all undef or zero, we know what it loads. 484bd1801b5553c8be3960255a92738464e0010b6f6Dan Gohman if (GlobalVariable *GV = 485bd1801b5553c8be3960255a92738464e0010b6f6Dan Gohman dyn_cast<GlobalVariable>(GetUnderlyingObject(CE, TD))) { 486e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner if (GV->isConstant() && GV->hasDefinitiveInitializer()) { 487db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *ResTy = cast<PointerType>(C->getType())->getElementType(); 488e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner if (GV->getInitializer()->isNullValue()) 489e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner return Constant::getNullValue(ResTy); 490e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner if (isa<UndefValue>(GV->getInitializer())) 491e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner return UndefValue::get(ResTy); 492e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner } 493e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner } 494e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner 495fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner // Try hard to fold loads from bitcasted strange and non-type-safe things. We 496fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner // currently don't do any of this for big endian systems. It can be 497fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner // generalized in the future if someone is interested. 498fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner if (TD && TD->isLittleEndian()) 499fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner return FoldReinterpretLoadFromConstPtr(CE, *TD); 500878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner return 0; 501878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner} 502878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner 503878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattnerstatic Constant *ConstantFoldLoadInst(const LoadInst *LI, const TargetData *TD){ 504878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner if (LI->isVolatile()) return 0; 505878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner 506878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner if (Constant *C = dyn_cast<Constant>(LI->getOperand(0))) 507878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner return ConstantFoldLoadFromConstPtr(C, TD); 508fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 509878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner return 0; 510878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner} 51103dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner 51203dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner/// SymbolicallyEvaluateBinop - One of Op0/Op1 is a constant expression. 51367e3566d8f6875e9c92d224f493023408a66227eNick Lewycky/// Attempt to symbolically evaluate the result of a binary operator merging 51403dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner/// these together. If target data info is available, it is provided as TD, 51503dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner/// otherwise TD is null. 51603dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattnerstatic Constant *SymbolicallyEvaluateBinop(unsigned Opc, Constant *Op0, 5177b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner Constant *Op1, const TargetData *TD){ 51803dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner // SROA 51903dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner 52003dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner // Fold (and 0xffffffff00000000, (shl x, 32)) -> shl. 52103dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner // Fold (lshr (or X, Y), 32) -> (lshr [X/Y], 32) if one doesn't contribute 52203dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner // bits. 52303dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner 52403dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner 52503dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner // If the constant expr is something like &A[123] - &A[4].f, fold this into a 52603dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner // constant. This happens frequently when iterating over a global array. 52703dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner if (Opc == Instruction::Sub && TD) { 52803dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner GlobalValue *GV1, *GV2; 52903dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner int64_t Offs1, Offs2; 53003dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner 53103dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner if (IsConstantOffsetFromGlobal(Op0, GV1, Offs1, *TD)) 53203dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner if (IsConstantOffsetFromGlobal(Op1, GV2, Offs2, *TD) && 53303dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner GV1 == GV2) { 53403dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner // (&GV+C1) - (&GV+C2) -> C1-C2, pointer arithmetic cannot overflow. 535eed707b1e6097aac2bb6b3d47271f6300ace7f2eOwen Anderson return ConstantInt::get(Op0->getType(), Offs1-Offs2); 53603dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner } 53703dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner } 53803dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner 53903dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner return 0; 54003dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner} 54103dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner 5424f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman/// CastGEPIndices - If array indices are not pointer-sized integers, 5434f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman/// explicitly cast them so that they aren't implicitly casted by the 5444f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman/// getelementptr. 5451d2f569c3428d70d0cf690c9adb459ad4a3ecff2Jay Foadstatic Constant *CastGEPIndices(ArrayRef<Constant *> Ops, 546618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier Type *ResultTy, const TargetData *TD, 547618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier const TargetLibraryInfo *TLI) { 5484f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman if (!TD) return 0; 549db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *IntPtrTy = TD->getIntPtrType(ResultTy->getContext()); 5504f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman 5514f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman bool Any = false; 5524f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman SmallVector<Constant*, 32> NewIdxs; 5531d2f569c3428d70d0cf690c9adb459ad4a3ecff2Jay Foad for (unsigned i = 1, e = Ops.size(); i != e; ++i) { 5544f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman if ((i == 1 || 5554f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman !isa<StructType>(GetElementPtrInst::getIndexedType(Ops[0]->getType(), 556a9203109f4ac95aa7e9624f2838e3d89623ec902Jay Foad Ops.slice(1, i-1)))) && 5574f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman Ops[i]->getType() != IntPtrTy) { 5584f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman Any = true; 5594f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman NewIdxs.push_back(ConstantExpr::getCast(CastInst::getCastOpcode(Ops[i], 5604f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman true, 5614f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman IntPtrTy, 5624f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman true), 5634f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman Ops[i], IntPtrTy)); 5644f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman } else 5654f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman NewIdxs.push_back(Ops[i]); 5664f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman } 5674f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman if (!Any) return 0; 5684f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman 5694f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman Constant *C = 570dab3d29605a5c83db41b28176273ef55961120c1Jay Foad ConstantExpr::getGetElementPtr(Ops[0], NewIdxs); 5714f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman if (ConstantExpr *CE = dyn_cast<ConstantExpr>(C)) 572618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier if (Constant *Folded = ConstantFoldConstantExpression(CE, TD, TLI)) 5734f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman C = Folded; 5744f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman return C; 5754f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman} 5764f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman 57703dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner/// SymbolicallyEvaluateGEP - If we can symbolically evaluate the specified GEP 57803dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner/// constant expression, do so. 5791d2f569c3428d70d0cf690c9adb459ad4a3ecff2Jay Foadstatic Constant *SymbolicallyEvaluateGEP(ArrayRef<Constant *> Ops, 580618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier Type *ResultTy, const TargetData *TD, 581618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier const TargetLibraryInfo *TLI) { 58203dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner Constant *Ptr = Ops[0]; 5831608769abeb1430dc34f31ffac0d9850f99ae36aNadav Rotem if (!TD || !cast<PointerType>(Ptr->getType())->getElementType()->isSized() || 5841608769abeb1430dc34f31ffac0d9850f99ae36aNadav Rotem !Ptr->getType()->isPointerTy()) 58503dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner return 0; 586f1cadf28991b2643128bfe38990ff235f2cb1491Chris Lattner 587db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *IntPtrTy = TD->getIntPtrType(Ptr->getContext()); 588268e7d7a9446bb89b80472cdb7b9803f16c0191eChris Lattner 589268e7d7a9446bb89b80472cdb7b9803f16c0191eChris Lattner // If this is a constant expr gep that is effectively computing an 590268e7d7a9446bb89b80472cdb7b9803f16c0191eChris Lattner // "offsetof", fold it into 'cast int Size to T*' instead of 'gep 0, 0, 12' 5911d2f569c3428d70d0cf690c9adb459ad4a3ecff2Jay Foad for (unsigned i = 1, e = Ops.size(); i != e; ++i) 5928cd4efb6a5e51b4e77039c17338f290ca3a8ee92Chris Lattner if (!isa<ConstantInt>(Ops[i])) { 5938cd4efb6a5e51b4e77039c17338f290ca3a8ee92Chris Lattner 5948cd4efb6a5e51b4e77039c17338f290ca3a8ee92Chris Lattner // If this is "gep i8* Ptr, (sub 0, V)", fold this as: 5958cd4efb6a5e51b4e77039c17338f290ca3a8ee92Chris Lattner // "inttoptr (sub (ptrtoint Ptr), V)" 5961d2f569c3428d70d0cf690c9adb459ad4a3ecff2Jay Foad if (Ops.size() == 2 && 5978cd4efb6a5e51b4e77039c17338f290ca3a8ee92Chris Lattner cast<PointerType>(ResultTy)->getElementType()->isIntegerTy(8)) { 5988cd4efb6a5e51b4e77039c17338f290ca3a8ee92Chris Lattner ConstantExpr *CE = dyn_cast<ConstantExpr>(Ops[1]); 599156eb0a569a1ebac86ad8438645f690d8a3894c4Chris Lattner assert((CE == 0 || CE->getType() == IntPtrTy) && 600f1cadf28991b2643128bfe38990ff235f2cb1491Chris Lattner "CastGEPIndices didn't canonicalize index types!"); 6018cd4efb6a5e51b4e77039c17338f290ca3a8ee92Chris Lattner if (CE && CE->getOpcode() == Instruction::Sub && 602f1cadf28991b2643128bfe38990ff235f2cb1491Chris Lattner CE->getOperand(0)->isNullValue()) { 6038cd4efb6a5e51b4e77039c17338f290ca3a8ee92Chris Lattner Constant *Res = ConstantExpr::getPtrToInt(Ptr, CE->getType()); 6048cd4efb6a5e51b4e77039c17338f290ca3a8ee92Chris Lattner Res = ConstantExpr::getSub(Res, CE->getOperand(1)); 6058cd4efb6a5e51b4e77039c17338f290ca3a8ee92Chris Lattner Res = ConstantExpr::getIntToPtr(Res, ResultTy); 6068cd4efb6a5e51b4e77039c17338f290ca3a8ee92Chris Lattner if (ConstantExpr *ResCE = dyn_cast<ConstantExpr>(Res)) 607618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier Res = ConstantFoldConstantExpression(ResCE, TD, TLI); 6088cd4efb6a5e51b4e77039c17338f290ca3a8ee92Chris Lattner return Res; 6098cd4efb6a5e51b4e77039c17338f290ca3a8ee92Chris Lattner } 6108cd4efb6a5e51b4e77039c17338f290ca3a8ee92Chris Lattner } 611de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman return 0; 6128cd4efb6a5e51b4e77039c17338f290ca3a8ee92Chris Lattner } 613268e7d7a9446bb89b80472cdb7b9803f16c0191eChris Lattner 614f1cadf28991b2643128bfe38990ff235f2cb1491Chris Lattner unsigned BitWidth = TD->getTypeSizeInBits(IntPtrTy); 6158fbbb3980755d74539a0aed02bc18842ed2bd18dJay Foad APInt Offset = 6168fbbb3980755d74539a0aed02bc18842ed2bd18dJay Foad APInt(BitWidth, TD->getIndexedOffset(Ptr->getType(), 6178fbbb3980755d74539a0aed02bc18842ed2bd18dJay Foad makeArrayRef((Value **)Ops.data() + 1, 6188fbbb3980755d74539a0aed02bc18842ed2bd18dJay Foad Ops.size() - 1))); 619890edda7c22dd8525f9fe44085cfd7080e67dfeeDuncan Sands Ptr = cast<Constant>(Ptr->stripPointerCasts()); 6200891d752a68a25025ffc3339aab1f0ad3221b0edDan Gohman 6210891d752a68a25025ffc3339aab1f0ad3221b0edDan Gohman // If this is a GEP of a GEP, fold it all into a single GEP. 6220891d752a68a25025ffc3339aab1f0ad3221b0edDan Gohman while (GEPOperator *GEP = dyn_cast<GEPOperator>(Ptr)) { 6230891d752a68a25025ffc3339aab1f0ad3221b0edDan Gohman SmallVector<Value *, 4> NestedOps(GEP->op_begin()+1, GEP->op_end()); 624890edda7c22dd8525f9fe44085cfd7080e67dfeeDuncan Sands 625890edda7c22dd8525f9fe44085cfd7080e67dfeeDuncan Sands // Do not try the incorporate the sub-GEP if some index is not a number. 626890edda7c22dd8525f9fe44085cfd7080e67dfeeDuncan Sands bool AllConstantInt = true; 627890edda7c22dd8525f9fe44085cfd7080e67dfeeDuncan Sands for (unsigned i = 0, e = NestedOps.size(); i != e; ++i) 628890edda7c22dd8525f9fe44085cfd7080e67dfeeDuncan Sands if (!isa<ConstantInt>(NestedOps[i])) { 629890edda7c22dd8525f9fe44085cfd7080e67dfeeDuncan Sands AllConstantInt = false; 630890edda7c22dd8525f9fe44085cfd7080e67dfeeDuncan Sands break; 631890edda7c22dd8525f9fe44085cfd7080e67dfeeDuncan Sands } 632890edda7c22dd8525f9fe44085cfd7080e67dfeeDuncan Sands if (!AllConstantInt) 633890edda7c22dd8525f9fe44085cfd7080e67dfeeDuncan Sands break; 634890edda7c22dd8525f9fe44085cfd7080e67dfeeDuncan Sands 6350891d752a68a25025ffc3339aab1f0ad3221b0edDan Gohman Ptr = cast<Constant>(GEP->getOperand(0)); 6360891d752a68a25025ffc3339aab1f0ad3221b0edDan Gohman Offset += APInt(BitWidth, 6378fbbb3980755d74539a0aed02bc18842ed2bd18dJay Foad TD->getIndexedOffset(Ptr->getType(), NestedOps)); 638890edda7c22dd8525f9fe44085cfd7080e67dfeeDuncan Sands Ptr = cast<Constant>(Ptr->stripPointerCasts()); 6390891d752a68a25025ffc3339aab1f0ad3221b0edDan Gohman } 6400891d752a68a25025ffc3339aab1f0ad3221b0edDan Gohman 641de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman // If the base value for this address is a literal integer value, fold the 642de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman // getelementptr to the resulting integer value casted to the pointer type. 64340b037ef611efc5c43ffc43a911d0eb57e0d444dDan Gohman APInt BasePtr(BitWidth, 0); 64440b037ef611efc5c43ffc43a911d0eb57e0d444dDan Gohman if (ConstantExpr *CE = dyn_cast<ConstantExpr>(Ptr)) 64540b037ef611efc5c43ffc43a911d0eb57e0d444dDan Gohman if (CE->getOpcode() == Instruction::IntToPtr) 64640f8f6264d5af2c38e797e0dc59827cd231e8ff7Jay Foad if (ConstantInt *Base = dyn_cast<ConstantInt>(CE->getOperand(0))) 64740f8f6264d5af2c38e797e0dc59827cd231e8ff7Jay Foad BasePtr = Base->getValue().zextOrTrunc(BitWidth); 64840b037ef611efc5c43ffc43a911d0eb57e0d444dDan Gohman if (Ptr->isNullValue() || BasePtr != 0) { 6497b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner Constant *C = ConstantInt::get(Ptr->getContext(), Offset+BasePtr); 650de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman return ConstantExpr::getIntToPtr(C, ResultTy); 651de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman } 652de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman 653de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman // Otherwise form a regular getelementptr. Recompute the indices so that 654de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman // we eliminate over-indexing of the notional static type array bounds. 655de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman // This makes it easy to determine if the getelementptr is "inbounds". 656de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman // Also, this helps GlobalOpt do SROA on GlobalVariables. 657db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *Ty = Ptr->getType(); 658de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman SmallVector<Constant*, 32> NewIdxs; 6593d01334e6fa5caba1361d088fabf728c64b23c58Dan Gohman do { 660db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner if (SequentialType *ATy = dyn_cast<SequentialType>(Ty)) { 6611df9859c40492511b8aa4321eb76496005d3b75bDuncan Sands if (ATy->isPointerTy()) { 662e568fa2d29c58aaedcaa832f805949d5e4371b4bChris Lattner // The only pointer indexing we'll do is on the first index of the GEP. 663e568fa2d29c58aaedcaa832f805949d5e4371b4bChris Lattner if (!NewIdxs.empty()) 664e568fa2d29c58aaedcaa832f805949d5e4371b4bChris Lattner break; 665e568fa2d29c58aaedcaa832f805949d5e4371b4bChris Lattner 666e568fa2d29c58aaedcaa832f805949d5e4371b4bChris Lattner // Only handle pointers to sized types, not pointers to functions. 667e568fa2d29c58aaedcaa832f805949d5e4371b4bChris Lattner if (!ATy->getElementType()->isSized()) 668e568fa2d29c58aaedcaa832f805949d5e4371b4bChris Lattner return 0; 669e568fa2d29c58aaedcaa832f805949d5e4371b4bChris Lattner } 670e568fa2d29c58aaedcaa832f805949d5e4371b4bChris Lattner 671de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman // Determine which element of the array the offset points into. 672cda9706cb74a65d2feb2175613cb0fe46acc5aa2Dan Gohman APInt ElemSize(BitWidth, TD->getTypeAllocSize(ATy->getElementType())); 673db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner IntegerType *IntPtrTy = TD->getIntPtrType(Ty->getContext()); 674de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman if (ElemSize == 0) 675c1da204c433459713cbf4c62f0addf458e6b4b9fChris Lattner // The element size is 0. This may be [0 x Ty]*, so just use a zero 676e58391073f6e17cd1024f737618a6f6b8d1323ceDuncan Sands // index for this level and proceed to the next level to see if it can 677e58391073f6e17cd1024f737618a6f6b8d1323ceDuncan Sands // accommodate the offset. 678c1da204c433459713cbf4c62f0addf458e6b4b9fChris Lattner NewIdxs.push_back(ConstantInt::get(IntPtrTy, 0)); 679c1da204c433459713cbf4c62f0addf458e6b4b9fChris Lattner else { 680c1da204c433459713cbf4c62f0addf458e6b4b9fChris Lattner // The element size is non-zero divide the offset by the element 681c1da204c433459713cbf4c62f0addf458e6b4b9fChris Lattner // size (rounding down), to compute the index at this level. 682c1da204c433459713cbf4c62f0addf458e6b4b9fChris Lattner APInt NewIdx = Offset.udiv(ElemSize); 683c1da204c433459713cbf4c62f0addf458e6b4b9fChris Lattner Offset -= NewIdx * ElemSize; 684c1da204c433459713cbf4c62f0addf458e6b4b9fChris Lattner NewIdxs.push_back(ConstantInt::get(IntPtrTy, NewIdx)); 685c1da204c433459713cbf4c62f0addf458e6b4b9fChris Lattner } 686de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman Ty = ATy->getElementType(); 687db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner } else if (StructType *STy = dyn_cast<StructType>(Ty)) { 688cda9706cb74a65d2feb2175613cb0fe46acc5aa2Dan Gohman // Determine which field of the struct the offset points into. The 689cda9706cb74a65d2feb2175613cb0fe46acc5aa2Dan Gohman // getZExtValue is at least as safe as the StructLayout API because we 690cda9706cb74a65d2feb2175613cb0fe46acc5aa2Dan Gohman // know the offset is within the struct at this point. 691de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman const StructLayout &SL = *TD->getStructLayout(STy); 692cda9706cb74a65d2feb2175613cb0fe46acc5aa2Dan Gohman unsigned ElIdx = SL.getElementContainingOffset(Offset.getZExtValue()); 6937b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner NewIdxs.push_back(ConstantInt::get(Type::getInt32Ty(Ty->getContext()), 6947b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner ElIdx)); 695cda9706cb74a65d2feb2175613cb0fe46acc5aa2Dan Gohman Offset -= APInt(BitWidth, SL.getElementOffset(ElIdx)); 696de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman Ty = STy->getTypeAtIndex(ElIdx); 697de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman } else { 6983d01334e6fa5caba1361d088fabf728c64b23c58Dan Gohman // We've reached some non-indexable type. 6993d01334e6fa5caba1361d088fabf728c64b23c58Dan Gohman break; 700de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman } 7013d01334e6fa5caba1361d088fabf728c64b23c58Dan Gohman } while (Ty != cast<PointerType>(ResultTy)->getElementType()); 7023d01334e6fa5caba1361d088fabf728c64b23c58Dan Gohman 7033d01334e6fa5caba1361d088fabf728c64b23c58Dan Gohman // If we haven't used up the entire offset by descending the static 7043d01334e6fa5caba1361d088fabf728c64b23c58Dan Gohman // type, then the offset is pointing into the middle of an indivisible 7053d01334e6fa5caba1361d088fabf728c64b23c58Dan Gohman // member, so we can't simplify it. 7063d01334e6fa5caba1361d088fabf728c64b23c58Dan Gohman if (Offset != 0) 7073d01334e6fa5caba1361d088fabf728c64b23c58Dan Gohman return 0; 708de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman 7093bfbc4587a7e79f08f8c126a9e62c3475fb90f8bDan Gohman // Create a GEP. 7103bfbc4587a7e79f08f8c126a9e62c3475fb90f8bDan Gohman Constant *C = 711dab3d29605a5c83db41b28176273ef55961120c1Jay Foad ConstantExpr::getGetElementPtr(Ptr, NewIdxs); 7126e7ad958683f34bf6c014c88fef723e5a2d741beDan Gohman assert(cast<PointerType>(C->getType())->getElementType() == Ty && 7136e7ad958683f34bf6c014c88fef723e5a2d741beDan Gohman "Computed GetElementPtr has unexpected type!"); 7143d01334e6fa5caba1361d088fabf728c64b23c58Dan Gohman 7153d01334e6fa5caba1361d088fabf728c64b23c58Dan Gohman // If we ended up indexing a member with a type that doesn't match 7164c0d5d5db876b0628bdf6a2174263a1c0a9130e2Dan Gohman // the type of what the original indices indexed, add a cast. 7173d01334e6fa5caba1361d088fabf728c64b23c58Dan Gohman if (Ty != cast<PointerType>(ResultTy)->getElementType()) 7186333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner C = FoldBitCast(C, ResultTy, *TD); 719de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman 7203d01334e6fa5caba1361d088fabf728c64b23c58Dan Gohman return C; 72103dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner} 72203dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner 7231afab9c1e0ae2d145ce01718aa14065393117e70Chris Lattner 72403dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner 72503dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner//===----------------------------------------------------------------------===// 72603dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner// Constant Folding public APIs 72703dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner//===----------------------------------------------------------------------===// 72803dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner 729b9b369fa9983843c4ed77b3a35b6e9b7933955bbDuncan Sands/// ConstantFoldInstruction - Try to constant fold the specified instruction. 730b9b369fa9983843c4ed77b3a35b6e9b7933955bbDuncan Sands/// If successful, the constant result is returned, if not, null is returned. 731b9b369fa9983843c4ed77b3a35b6e9b7933955bbDuncan Sands/// Note that this fails if not all of the operands are constant. Otherwise, 732b9b369fa9983843c4ed77b3a35b6e9b7933955bbDuncan Sands/// this function can only fail when attempting to fold instructions like loads 733b9b369fa9983843c4ed77b3a35b6e9b7933955bbDuncan Sands/// and stores, which have no constant expression form. 734618c1dbd293d15ee19f61b1156ab8086ad28311aChad RosierConstant *llvm::ConstantFoldInstruction(Instruction *I, 735618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier const TargetData *TD, 736618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier const TargetLibraryInfo *TLI) { 737b9b369fa9983843c4ed77b3a35b6e9b7933955bbDuncan Sands // Handle PHI nodes quickly here... 7385520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner if (PHINode *PN = dyn_cast<PHINode>(I)) { 739c0362d5c6e0066c741bce056a65d8b4c026de19fDuncan Sands Constant *CommonValue = 0; 740c0362d5c6e0066c741bce056a65d8b4c026de19fDuncan Sands 741c0362d5c6e0066c741bce056a65d8b4c026de19fDuncan Sands for (unsigned i = 0, e = PN->getNumIncomingValues(); i != e; ++i) { 742c0362d5c6e0066c741bce056a65d8b4c026de19fDuncan Sands Value *Incoming = PN->getIncomingValue(i); 743b9b369fa9983843c4ed77b3a35b6e9b7933955bbDuncan Sands // If the incoming value is undef then skip it. Note that while we could 744b9b369fa9983843c4ed77b3a35b6e9b7933955bbDuncan Sands // skip the value if it is equal to the phi node itself we choose not to 745b9b369fa9983843c4ed77b3a35b6e9b7933955bbDuncan Sands // because that would break the rule that constant folding only applies if 746b9b369fa9983843c4ed77b3a35b6e9b7933955bbDuncan Sands // all operands are constants. 747b9b369fa9983843c4ed77b3a35b6e9b7933955bbDuncan Sands if (isa<UndefValue>(Incoming)) 748c0362d5c6e0066c741bce056a65d8b4c026de19fDuncan Sands continue; 749c0362d5c6e0066c741bce056a65d8b4c026de19fDuncan Sands // If the incoming value is not a constant, or is a different constant to 750c0362d5c6e0066c741bce056a65d8b4c026de19fDuncan Sands // the one we saw previously, then give up. 751c0362d5c6e0066c741bce056a65d8b4c026de19fDuncan Sands Constant *C = dyn_cast<Constant>(Incoming); 752c0362d5c6e0066c741bce056a65d8b4c026de19fDuncan Sands if (!C || (CommonValue && C != CommonValue)) 753c0362d5c6e0066c741bce056a65d8b4c026de19fDuncan Sands return 0; 754c0362d5c6e0066c741bce056a65d8b4c026de19fDuncan Sands CommonValue = C; 755c0362d5c6e0066c741bce056a65d8b4c026de19fDuncan Sands } 7565520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner 757c0362d5c6e0066c741bce056a65d8b4c026de19fDuncan Sands // If we reach here, all incoming values are the same constant or undef. 758c0362d5c6e0066c741bce056a65d8b4c026de19fDuncan Sands return CommonValue ? CommonValue : UndefValue::get(PN->getType()); 7595520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner } 7605520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner 7615520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner // Scan the operand list, checking to see if they are all constants, if so, 7625520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner // hand off to ConstantFoldInstOperands. 7635520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner SmallVector<Constant*, 8> Ops; 764de2d74b213844b17eb5327fba27f21e699d3af66Gabor Greif for (User::op_iterator i = I->op_begin(), e = I->op_end(); i != e; ++i) 765de2d74b213844b17eb5327fba27f21e699d3af66Gabor Greif if (Constant *Op = dyn_cast<Constant>(*i)) 7665520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner Ops.push_back(Op); 7675520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner else 7685520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner return 0; // All operands not constant! 7695520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner 770f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner if (const CmpInst *CI = dyn_cast<CmpInst>(I)) 7718f73deaa8732a556046bf4ac6207be55972e3b74Chris Lattner return ConstantFoldCompareInstOperands(CI->getPredicate(), Ops[0], Ops[1], 772618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier TD, TLI); 77358665d47aea69875c433fb039c8f10106abc1ceeChris Lattner 774878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner if (const LoadInst *LI = dyn_cast<LoadInst>(I)) 775878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner return ConstantFoldLoadInst(LI, TD); 7763ee0af35842999c1781052fe74fdcb19997de950Frits van Bommel 7773ee0af35842999c1781052fe74fdcb19997de950Frits van Bommel if (InsertValueInst *IVI = dyn_cast<InsertValueInst>(I)) 7783ee0af35842999c1781052fe74fdcb19997de950Frits van Bommel return ConstantExpr::getInsertValue( 7793ee0af35842999c1781052fe74fdcb19997de950Frits van Bommel cast<Constant>(IVI->getAggregateOperand()), 7803ee0af35842999c1781052fe74fdcb19997de950Frits van Bommel cast<Constant>(IVI->getInsertedValueOperand()), 781fc6d3a49867cd38954dc40936a88f1907252c6d2Jay Foad IVI->getIndices()); 7823ee0af35842999c1781052fe74fdcb19997de950Frits van Bommel 7833ee0af35842999c1781052fe74fdcb19997de950Frits van Bommel if (ExtractValueInst *EVI = dyn_cast<ExtractValueInst>(I)) 7843ee0af35842999c1781052fe74fdcb19997de950Frits van Bommel return ConstantExpr::getExtractValue( 7853ee0af35842999c1781052fe74fdcb19997de950Frits van Bommel cast<Constant>(EVI->getAggregateOperand()), 786fc6d3a49867cd38954dc40936a88f1907252c6d2Jay Foad EVI->getIndices()); 7873ee0af35842999c1781052fe74fdcb19997de950Frits van Bommel 788618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier return ConstantFoldInstOperands(I->getOpcode(), I->getType(), Ops, TD, TLI); 7895520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner} 7905520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner 7913dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky/// ConstantFoldConstantExpression - Attempt to fold the constant expression 7923dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky/// using the specified TargetData. If successful, the constant result is 7933dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky/// result is returned, if not, null is returned. 794baf0c67988feec5b7698f89dce499ddb7d0c3c42Dan GohmanConstant *llvm::ConstantFoldConstantExpression(const ConstantExpr *CE, 795618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier const TargetData *TD, 796618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier const TargetLibraryInfo *TLI) { 7973dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky SmallVector<Constant*, 8> Ops; 798c1da204c433459713cbf4c62f0addf458e6b4b9fChris Lattner for (User::const_op_iterator i = CE->op_begin(), e = CE->op_end(); 799c1da204c433459713cbf4c62f0addf458e6b4b9fChris Lattner i != e; ++i) { 80001b97dd01eec1197d79fceb7229f5ec233994de3Dan Gohman Constant *NewC = cast<Constant>(*i); 80101b97dd01eec1197d79fceb7229f5ec233994de3Dan Gohman // Recursively fold the ConstantExpr's operands. 80201b97dd01eec1197d79fceb7229f5ec233994de3Dan Gohman if (ConstantExpr *NewCE = dyn_cast<ConstantExpr>(NewC)) 803618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier NewC = ConstantFoldConstantExpression(NewCE, TD, TLI); 80401b97dd01eec1197d79fceb7229f5ec233994de3Dan Gohman Ops.push_back(NewC); 80501b97dd01eec1197d79fceb7229f5ec233994de3Dan Gohman } 8063dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky 8073dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky if (CE->isCompare()) 8088f73deaa8732a556046bf4ac6207be55972e3b74Chris Lattner return ConstantFoldCompareInstOperands(CE->getPredicate(), Ops[0], Ops[1], 809618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier TD, TLI); 810618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier return ConstantFoldInstOperands(CE->getOpcode(), CE->getType(), Ops, TD, TLI); 8113dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky} 8123dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky 8135520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner/// ConstantFoldInstOperands - Attempt to constant fold an instruction with the 8145520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner/// specified opcode and operands. If successful, the constant result is 8155520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner/// returned, if not, null is returned. Note that this function can fail when 8165520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner/// attempting to fold instructions like loads and stores, which have no 8175520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner/// constant expression form. 8185520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner/// 81901b97dd01eec1197d79fceb7229f5ec233994de3Dan Gohman/// TODO: This function neither utilizes nor preserves nsw/nuw/inbounds/etc 82001b97dd01eec1197d79fceb7229f5ec233994de3Dan Gohman/// information, due to only being passed an opcode and operands. Constant 82101b97dd01eec1197d79fceb7229f5ec233994de3Dan Gohman/// folding using this function strips this information. 82201b97dd01eec1197d79fceb7229f5ec233994de3Dan Gohman/// 823db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris LattnerConstant *llvm::ConstantFoldInstOperands(unsigned Opcode, Type *DestTy, 8241d2f569c3428d70d0cf690c9adb459ad4a3ecff2Jay Foad ArrayRef<Constant *> Ops, 825618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier const TargetData *TD, 826618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier const TargetLibraryInfo *TLI) { 82703dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner // Handle easy binops first. 828f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner if (Instruction::isBinaryOp(Opcode)) { 82903dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner if (isa<ConstantExpr>(Ops[0]) || isa<ConstantExpr>(Ops[1])) 8307b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner if (Constant *C = SymbolicallyEvaluateBinop(Opcode, Ops[0], Ops[1], TD)) 83103dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner return C; 83203dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner 833baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson return ConstantExpr::get(Opcode, Ops[0], Ops[1]); 83403dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner } 8355520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner 836f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner switch (Opcode) { 8375520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner default: return 0; 83879fa3cf451a74dd1cec0be2d77d7ea9737dffc69Chris Lattner case Instruction::ICmp: 83979fa3cf451a74dd1cec0be2d77d7ea9737dffc69Chris Lattner case Instruction::FCmp: assert(0 && "Invalid for compares"); 8405520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner case Instruction::Call: 8411d2f569c3428d70d0cf690c9adb459ad4a3ecff2Jay Foad if (Function *F = dyn_cast<Function>(Ops.back())) 8425520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner if (canConstantFoldCallTo(F)) 843618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier return ConstantFoldCall(F, Ops.slice(0, Ops.size() - 1), TLI); 8445520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner return 0; 845001f7534e0311508cc93831a595785bb177f8a18Chris Lattner case Instruction::PtrToInt: 846001f7534e0311508cc93831a595785bb177f8a18Chris Lattner // If the input is a inttoptr, eliminate the pair. This requires knowing 847001f7534e0311508cc93831a595785bb177f8a18Chris Lattner // the width of a pointer, so it can't be done in ConstantExpr::getCast. 848001f7534e0311508cc93831a595785bb177f8a18Chris Lattner if (ConstantExpr *CE = dyn_cast<ConstantExpr>(Ops[0])) { 849001f7534e0311508cc93831a595785bb177f8a18Chris Lattner if (TD && CE->getOpcode() == Instruction::IntToPtr) { 850001f7534e0311508cc93831a595785bb177f8a18Chris Lattner Constant *Input = CE->getOperand(0); 8516de29f8d960505421d61c80cdb738e16720b6c0eDan Gohman unsigned InWidth = Input->getType()->getScalarSizeInBits(); 85204aa2c3520c049333ca216fb794d700310c4a33dNick Lewycky if (TD->getPointerSizeInBits() < InWidth) { 85304aa2c3520c049333ca216fb794d700310c4a33dNick Lewycky Constant *Mask = 8547b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner ConstantInt::get(CE->getContext(), APInt::getLowBitsSet(InWidth, 85504aa2c3520c049333ca216fb794d700310c4a33dNick Lewycky TD->getPointerSizeInBits())); 856baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson Input = ConstantExpr::getAnd(Input, Mask); 85704aa2c3520c049333ca216fb794d700310c4a33dNick Lewycky } 858001f7534e0311508cc93831a595785bb177f8a18Chris Lattner // Do a zext or trunc to get to the dest size. 859baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson return ConstantExpr::getIntegerCast(Input, DestTy, false); 860001f7534e0311508cc93831a595785bb177f8a18Chris Lattner } 861001f7534e0311508cc93831a595785bb177f8a18Chris Lattner } 862baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson return ConstantExpr::getCast(Opcode, Ops[0], DestTy); 863001f7534e0311508cc93831a595785bb177f8a18Chris Lattner case Instruction::IntToPtr: 86481b06be055e74a3c23a1c8f17ead97f9f76335eeDuncan Sands // If the input is a ptrtoint, turn the pair into a ptr to ptr bitcast if 86581b06be055e74a3c23a1c8f17ead97f9f76335eeDuncan Sands // the int size is >= the ptr size. This requires knowing the width of a 86681b06be055e74a3c23a1c8f17ead97f9f76335eeDuncan Sands // pointer, so it can't be done in ConstantExpr::getCast. 867b80a2a686fe76496d71397f8bdda394d5718ab01Dan Gohman if (ConstantExpr *CE = dyn_cast<ConstantExpr>(Ops[0])) 8689a38e3e3991ea443e555d8060f91202a786acdd4Dan Gohman if (TD && 869b80a2a686fe76496d71397f8bdda394d5718ab01Dan Gohman TD->getPointerSizeInBits() <= CE->getType()->getScalarSizeInBits() && 870b80a2a686fe76496d71397f8bdda394d5718ab01Dan Gohman CE->getOpcode() == Instruction::PtrToInt) 871b80a2a686fe76496d71397f8bdda394d5718ab01Dan Gohman return FoldBitCast(CE->getOperand(0), DestTy, *TD); 872b80a2a686fe76496d71397f8bdda394d5718ab01Dan Gohman 873baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson return ConstantExpr::getCast(Opcode, Ops[0], DestTy); 8745520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner case Instruction::Trunc: 8755520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner case Instruction::ZExt: 8765520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner case Instruction::SExt: 8775520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner case Instruction::FPTrunc: 8785520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner case Instruction::FPExt: 8795520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner case Instruction::UIToFP: 8805520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner case Instruction::SIToFP: 8815520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner case Instruction::FPToUI: 8825520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner case Instruction::FPToSI: 883baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson return ConstantExpr::getCast(Opcode, Ops[0], DestTy); 8845520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner case Instruction::BitCast: 8851afab9c1e0ae2d145ce01718aa14065393117e70Chris Lattner if (TD) 8866333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner return FoldBitCast(Ops[0], DestTy, *TD); 887baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson return ConstantExpr::getBitCast(Ops[0], DestTy); 8885520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner case Instruction::Select: 889baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson return ConstantExpr::getSelect(Ops[0], Ops[1], Ops[2]); 8905520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner case Instruction::ExtractElement: 891baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson return ConstantExpr::getExtractElement(Ops[0], Ops[1]); 8925520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner case Instruction::InsertElement: 893baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson return ConstantExpr::getInsertElement(Ops[0], Ops[1], Ops[2]); 8945520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner case Instruction::ShuffleVector: 895baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson return ConstantExpr::getShuffleVector(Ops[0], Ops[1], Ops[2]); 8965520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner case Instruction::GetElementPtr: 897618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier if (Constant *C = CastGEPIndices(Ops, DestTy, TD, TLI)) 8984f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman return C; 899618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier if (Constant *C = SymbolicallyEvaluateGEP(Ops, DestTy, TD, TLI)) 90003dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner return C; 90103dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner 902dab3d29605a5c83db41b28176273ef55961120c1Jay Foad return ConstantExpr::getGetElementPtr(Ops[0], Ops.slice(1)); 9035520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner } 9045520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner} 9055520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner 906f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner/// ConstantFoldCompareInstOperands - Attempt to constant fold a compare 907f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner/// instruction (icmp/fcmp) with the specified operands. If it fails, it 908f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner/// returns a constant expression of the specified operands. 909f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner/// 910f286f6fd93d569befe6e77c94a947e6e04e95685Chris LattnerConstant *llvm::ConstantFoldCompareInstOperands(unsigned Predicate, 9118f73deaa8732a556046bf4ac6207be55972e3b74Chris Lattner Constant *Ops0, Constant *Ops1, 912618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier const TargetData *TD, 913618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier const TargetLibraryInfo *TLI) { 914f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner // fold: icmp (inttoptr x), null -> icmp x, 0 915f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner // fold: icmp (ptrtoint x), 0 -> icmp x, null 9163dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky // fold: icmp (inttoptr x), (inttoptr y) -> icmp trunc/zext x, trunc/zext y 917f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner // fold: icmp (ptrtoint x), (ptrtoint y) -> icmp x, y 918f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner // 919f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner // ConstantExpr::getCompare cannot do this, because it doesn't have TD 920f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner // around to know if bit truncation is happening. 9218f73deaa8732a556046bf4ac6207be55972e3b74Chris Lattner if (ConstantExpr *CE0 = dyn_cast<ConstantExpr>(Ops0)) { 9228f73deaa8732a556046bf4ac6207be55972e3b74Chris Lattner if (TD && Ops1->isNullValue()) { 923db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *IntPtrTy = TD->getIntPtrType(CE0->getContext()); 924f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner if (CE0->getOpcode() == Instruction::IntToPtr) { 925f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner // Convert the integer value to the right size to ensure we get the 926f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner // proper extension or truncation. 927baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson Constant *C = ConstantExpr::getIntegerCast(CE0->getOperand(0), 928f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner IntPtrTy, false); 9298f73deaa8732a556046bf4ac6207be55972e3b74Chris Lattner Constant *Null = Constant::getNullValue(C->getType()); 930618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier return ConstantFoldCompareInstOperands(Predicate, C, Null, TD, TLI); 931f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner } 932f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner 933f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner // Only do this transformation if the int is intptrty in size, otherwise 934f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner // there is a truncation or extension that we aren't modeling. 935f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner if (CE0->getOpcode() == Instruction::PtrToInt && 936f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner CE0->getType() == IntPtrTy) { 937f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner Constant *C = CE0->getOperand(0); 9388f73deaa8732a556046bf4ac6207be55972e3b74Chris Lattner Constant *Null = Constant::getNullValue(C->getType()); 939618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier return ConstantFoldCompareInstOperands(Predicate, C, Null, TD, TLI); 940f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner } 941f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner } 942f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner 9438f73deaa8732a556046bf4ac6207be55972e3b74Chris Lattner if (ConstantExpr *CE1 = dyn_cast<ConstantExpr>(Ops1)) { 9443dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky if (TD && CE0->getOpcode() == CE1->getOpcode()) { 945db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *IntPtrTy = TD->getIntPtrType(CE0->getContext()); 9463dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky 9473dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky if (CE0->getOpcode() == Instruction::IntToPtr) { 9483dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky // Convert the integer value to the right size to ensure we get the 9493dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky // proper extension or truncation. 950baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson Constant *C0 = ConstantExpr::getIntegerCast(CE0->getOperand(0), 9513dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky IntPtrTy, false); 952baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson Constant *C1 = ConstantExpr::getIntegerCast(CE1->getOperand(0), 9533dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky IntPtrTy, false); 954618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier return ConstantFoldCompareInstOperands(Predicate, C0, C1, TD, TLI); 9553dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky } 9563dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky 9573dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky // Only do this transformation if the int is intptrty in size, otherwise 9583dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky // there is a truncation or extension that we aren't modeling. 9593dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky if ((CE0->getOpcode() == Instruction::PtrToInt && 9603dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky CE0->getType() == IntPtrTy && 9618f73deaa8732a556046bf4ac6207be55972e3b74Chris Lattner CE0->getOperand(0)->getType() == CE1->getOperand(0)->getType())) 9628f73deaa8732a556046bf4ac6207be55972e3b74Chris Lattner return ConstantFoldCompareInstOperands(Predicate, CE0->getOperand(0), 963618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier CE1->getOperand(0), TD, TLI); 964f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner } 965f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner } 96679fa3cf451a74dd1cec0be2d77d7ea9737dffc69Chris Lattner 96779fa3cf451a74dd1cec0be2d77d7ea9737dffc69Chris Lattner // icmp eq (or x, y), 0 -> (icmp eq x, 0) & (icmp eq y, 0) 96879fa3cf451a74dd1cec0be2d77d7ea9737dffc69Chris Lattner // icmp ne (or x, y), 0 -> (icmp ne x, 0) | (icmp ne y, 0) 96979fa3cf451a74dd1cec0be2d77d7ea9737dffc69Chris Lattner if ((Predicate == ICmpInst::ICMP_EQ || Predicate == ICmpInst::ICMP_NE) && 97079fa3cf451a74dd1cec0be2d77d7ea9737dffc69Chris Lattner CE0->getOpcode() == Instruction::Or && Ops1->isNullValue()) { 97179fa3cf451a74dd1cec0be2d77d7ea9737dffc69Chris Lattner Constant *LHS = 972618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier ConstantFoldCompareInstOperands(Predicate, CE0->getOperand(0), Ops1, 973618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier TD, TLI); 97479fa3cf451a74dd1cec0be2d77d7ea9737dffc69Chris Lattner Constant *RHS = 975618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier ConstantFoldCompareInstOperands(Predicate, CE0->getOperand(1), Ops1, 976618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier TD, TLI); 97779fa3cf451a74dd1cec0be2d77d7ea9737dffc69Chris Lattner unsigned OpC = 97879fa3cf451a74dd1cec0be2d77d7ea9737dffc69Chris Lattner Predicate == ICmpInst::ICMP_EQ ? Instruction::And : Instruction::Or; 97979fa3cf451a74dd1cec0be2d77d7ea9737dffc69Chris Lattner Constant *Ops[] = { LHS, RHS }; 980618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier return ConstantFoldInstOperands(OpC, LHS->getType(), Ops, TD, TLI); 98179fa3cf451a74dd1cec0be2d77d7ea9737dffc69Chris Lattner } 982f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner } 9838f73deaa8732a556046bf4ac6207be55972e3b74Chris Lattner 9848f73deaa8732a556046bf4ac6207be55972e3b74Chris Lattner return ConstantExpr::getCompare(Predicate, Ops0, Ops1); 985f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner} 986f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner 987f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner 9885520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner/// ConstantFoldLoadThroughGEPConstantExpr - Given a constant and a 9895520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner/// getelementptr constantexpr, return the constant value being addressed by the 9905520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner/// constant expression, or null if something is funny and we can't decide. 9915520732b24a5a321140dd79af70d321c7ff3dec9Chris LattnerConstant *llvm::ConstantFoldLoadThroughGEPConstantExpr(Constant *C, 992c6f69e94fa46f585b59bb9d7ace3224b0e638c95Dan Gohman ConstantExpr *CE) { 993a7235ea7245028a0723e8ab7fd011386b3900777Owen Anderson if (CE->getOperand(1) != Constant::getNullValue(CE->getOperand(1)->getType())) 9945520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner return 0; // Do not allow stepping over the value! 9955520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner 9965520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner // Loop over all of the operands, tracking down which value we are 9975520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner // addressing... 9985520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner gep_type_iterator I = gep_type_begin(CE), E = gep_type_end(CE); 9995520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner for (++I; I != E; ++I) 1000db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner if (StructType *STy = dyn_cast<StructType>(*I)) { 10015520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner ConstantInt *CU = cast<ConstantInt>(I.getOperand()); 10025520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner assert(CU->getZExtValue() < STy->getNumElements() && 10035520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner "Struct index out of range!"); 10045520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner unsigned El = (unsigned)CU->getZExtValue(); 10055520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner if (ConstantStruct *CS = dyn_cast<ConstantStruct>(C)) { 10065520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner C = CS->getOperand(El); 10075520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner } else if (isa<ConstantAggregateZero>(C)) { 1008a7235ea7245028a0723e8ab7fd011386b3900777Owen Anderson C = Constant::getNullValue(STy->getElementType(El)); 10095520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner } else if (isa<UndefValue>(C)) { 10109e9a0d5fc26878e51a58a8b57900fcbf952c2691Owen Anderson C = UndefValue::get(STy->getElementType(El)); 10115520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner } else { 10125520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner return 0; 10135520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner } 10145520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner } else if (ConstantInt *CI = dyn_cast<ConstantInt>(I.getOperand())) { 1015db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner if (ArrayType *ATy = dyn_cast<ArrayType>(*I)) { 10165520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner if (CI->getZExtValue() >= ATy->getNumElements()) 10175520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner return 0; 10185520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner if (ConstantArray *CA = dyn_cast<ConstantArray>(C)) 10195520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner C = CA->getOperand(CI->getZExtValue()); 10205520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner else if (isa<ConstantAggregateZero>(C)) 1021a7235ea7245028a0723e8ab7fd011386b3900777Owen Anderson C = Constant::getNullValue(ATy->getElementType()); 10225520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner else if (isa<UndefValue>(C)) 10239e9a0d5fc26878e51a58a8b57900fcbf952c2691Owen Anderson C = UndefValue::get(ATy->getElementType()); 10245520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner else 10255520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner return 0; 1026db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner } else if (VectorType *VTy = dyn_cast<VectorType>(*I)) { 102762d327e241f59735eed836e98c8f11ffaa6cf6f8Chris Lattner if (CI->getZExtValue() >= VTy->getNumElements()) 10285520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner return 0; 10299d6565a5b1fbc4286d6ee638d8f47a3171a9ed7eReid Spencer if (ConstantVector *CP = dyn_cast<ConstantVector>(C)) 10305520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner C = CP->getOperand(CI->getZExtValue()); 10315520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner else if (isa<ConstantAggregateZero>(C)) 103262d327e241f59735eed836e98c8f11ffaa6cf6f8Chris Lattner C = Constant::getNullValue(VTy->getElementType()); 10335520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner else if (isa<UndefValue>(C)) 103462d327e241f59735eed836e98c8f11ffaa6cf6f8Chris Lattner C = UndefValue::get(VTy->getElementType()); 10355520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner else 10365520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner return 0; 10375520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner } else { 10385520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner return 0; 10395520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner } 10405520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner } else { 10415520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner return 0; 10425520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner } 10435520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner return C; 10445520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner} 10455520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner 10465520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner 1047bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell//===----------------------------------------------------------------------===// 10485520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner// Constant Folding for Calls 1049bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell// 1050bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell 1051bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell/// canConstantFoldCallTo - Return true if its even possible to fold a call to 1052bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell/// the specified function. 1053bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswellbool 1054fa9b80eb64127b3d9691e18537975635520e51e9Dan Gohmanllvm::canConstantFoldCallTo(const Function *F) { 1055bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell switch (F->getIntrinsicID()) { 10569ab7fb3ba47442d521a5bed09a27a5e8e7a786edDale Johannesen case Intrinsic::sqrt: 105724fbf2bf161b20b13d0a218345bdcd092bdd56f8Chad Rosier case Intrinsic::pow: 10589ab7fb3ba47442d521a5bed09a27a5e8e7a786edDale Johannesen case Intrinsic::powi: 1059e9391fd9b52e93717b365bdd05c471101323a4dfReid Spencer case Intrinsic::bswap: 1060e9391fd9b52e93717b365bdd05c471101323a4dfReid Spencer case Intrinsic::ctpop: 1061e9391fd9b52e93717b365bdd05c471101323a4dfReid Spencer case Intrinsic::ctlz: 1062e9391fd9b52e93717b365bdd05c471101323a4dfReid Spencer case Intrinsic::cttz: 10631614e50d9f6d48a38e4dafb4b1a51fd4b37614f3Evan Phoenix case Intrinsic::sadd_with_overflow: 10646208610fd602ebdb18bb793152899573d0b2b7abFrits van Bommel case Intrinsic::uadd_with_overflow: 10651614e50d9f6d48a38e4dafb4b1a51fd4b37614f3Evan Phoenix case Intrinsic::ssub_with_overflow: 10666208610fd602ebdb18bb793152899573d0b2b7abFrits van Bommel case Intrinsic::usub_with_overflow: 1067eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner case Intrinsic::smul_with_overflow: 10686208610fd602ebdb18bb793152899573d0b2b7abFrits van Bommel case Intrinsic::umul_with_overflow: 1069faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov case Intrinsic::convert_from_fp16: 1070faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov case Intrinsic::convert_to_fp16: 107115ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth case Intrinsic::x86_sse_cvtss2si: 107215ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth case Intrinsic::x86_sse_cvtss2si64: 107315ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth case Intrinsic::x86_sse_cvttss2si: 107415ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth case Intrinsic::x86_sse_cvttss2si64: 107515ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth case Intrinsic::x86_sse2_cvtsd2si: 107615ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth case Intrinsic::x86_sse2_cvtsd2si64: 107715ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth case Intrinsic::x86_sse2_cvttsd2si: 107815ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth case Intrinsic::x86_sse2_cvttsd2si64: 1079bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell return true; 108068a0603dc5d3d79cd4de2195e950af26c3380cd7Chris Lattner default: 108168a0603dc5d3d79cd4de2195e950af26c3380cd7Chris Lattner return false; 108268a0603dc5d3d79cd4de2195e950af26c3380cd7Chris Lattner case 0: break; 1083bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell } 1084bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell 10856f532a988e5c02fcb271fb65e9e7b83f0147d3f2Chris Lattner if (!F->hasName()) return false; 1086f0443c1eb44d737d9bd78962932fc80f74c6113cDaniel Dunbar StringRef Name = F->getName(); 1087c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner 1088c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner // In these cases, the check of the length is required. We don't want to 1089c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner // return true for a name like "cos\0blah" which strcmp would return equal to 1090c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner // "cos", but has length 8. 1091f0443c1eb44d737d9bd78962932fc80f74c6113cDaniel Dunbar switch (Name[0]) { 1092c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner default: return false; 1093c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner case 'a': 1094f0443c1eb44d737d9bd78962932fc80f74c6113cDaniel Dunbar return Name == "acos" || Name == "asin" || 1095f0443c1eb44d737d9bd78962932fc80f74c6113cDaniel Dunbar Name == "atan" || Name == "atan2"; 1096c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner case 'c': 1097f0443c1eb44d737d9bd78962932fc80f74c6113cDaniel Dunbar return Name == "cos" || Name == "ceil" || Name == "cosf" || Name == "cosh"; 1098c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner case 'e': 1099805fa97a0f5ff61624855da37d4881708fb8f0fbChris Lattner return Name == "exp" || Name == "exp2"; 1100c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner case 'f': 1101f0443c1eb44d737d9bd78962932fc80f74c6113cDaniel Dunbar return Name == "fabs" || Name == "fmod" || Name == "floor"; 1102c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner case 'l': 1103f0443c1eb44d737d9bd78962932fc80f74c6113cDaniel Dunbar return Name == "log" || Name == "log10"; 1104c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner case 'p': 1105f0443c1eb44d737d9bd78962932fc80f74c6113cDaniel Dunbar return Name == "pow"; 1106c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner case 's': 1107f0443c1eb44d737d9bd78962932fc80f74c6113cDaniel Dunbar return Name == "sin" || Name == "sinh" || Name == "sqrt" || 1108f0443c1eb44d737d9bd78962932fc80f74c6113cDaniel Dunbar Name == "sinf" || Name == "sqrtf"; 1109c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner case 't': 1110f0443c1eb44d737d9bd78962932fc80f74c6113cDaniel Dunbar return Name == "tan" || Name == "tanh"; 1111bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell } 1112bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell} 1113bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell 111472d88ae5447a3929c97e88f4c806213847b5d988Chris Lattnerstatic Constant *ConstantFoldFP(double (*NativeFP)(double), double V, 1115db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *Ty) { 11163f2f21e640682c604d9322ba0708e122d1fcf5eeDan Gohman sys::llvm_fenv_clearexcept(); 1117bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell V = NativeFP(V); 11183f2f21e640682c604d9322ba0708e122d1fcf5eeDan Gohman if (sys::llvm_fenv_testexcept()) { 11193f2f21e640682c604d9322ba0708e122d1fcf5eeDan Gohman sys::llvm_fenv_clearexcept(); 1120f19f58a9360927f8ef50b9a029e8efb780d3d8a8Chris Lattner return 0; 112143421b3dd70af5b70e71816521f37502c397cc65Dale Johannesen } 1122f19f58a9360927f8ef50b9a029e8efb780d3d8a8Chris Lattner 1123d0806a1519f3bc450d2dea3e6f22db73fc797e4cChris Lattner if (Ty->isFloatTy()) 11247b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFP::get(Ty->getContext(), APFloat((float)V)); 1125d0806a1519f3bc450d2dea3e6f22db73fc797e4cChris Lattner if (Ty->isDoubleTy()) 11267b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFP::get(Ty->getContext(), APFloat(V)); 1127c23197a26f34f559ea9797de51e187087c039c42Torok Edwin llvm_unreachable("Can only constant fold float/double"); 112833e456d5f3909e0c7b96e04e5674c25a182da100Gabor Greif return 0; // dummy return to suppress warning 1129bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell} 1130bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell 1131384152444d22d356f7b4090a158da4cbc1f79464Dan Gohmanstatic Constant *ConstantFoldBinaryFP(double (*NativeFP)(double, double), 1132db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner double V, double W, Type *Ty) { 11333f2f21e640682c604d9322ba0708e122d1fcf5eeDan Gohman sys::llvm_fenv_clearexcept(); 1134384152444d22d356f7b4090a158da4cbc1f79464Dan Gohman V = NativeFP(V, W); 11353f2f21e640682c604d9322ba0708e122d1fcf5eeDan Gohman if (sys::llvm_fenv_testexcept()) { 11363f2f21e640682c604d9322ba0708e122d1fcf5eeDan Gohman sys::llvm_fenv_clearexcept(); 1137f19f58a9360927f8ef50b9a029e8efb780d3d8a8Chris Lattner return 0; 113843421b3dd70af5b70e71816521f37502c397cc65Dale Johannesen } 1139f19f58a9360927f8ef50b9a029e8efb780d3d8a8Chris Lattner 1140d0806a1519f3bc450d2dea3e6f22db73fc797e4cChris Lattner if (Ty->isFloatTy()) 11417b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFP::get(Ty->getContext(), APFloat((float)V)); 1142d0806a1519f3bc450d2dea3e6f22db73fc797e4cChris Lattner if (Ty->isDoubleTy()) 11437b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFP::get(Ty->getContext(), APFloat(V)); 1144c23197a26f34f559ea9797de51e187087c039c42Torok Edwin llvm_unreachable("Can only constant fold float/double"); 114533e456d5f3909e0c7b96e04e5674c25a182da100Gabor Greif return 0; // dummy return to suppress warning 1146384152444d22d356f7b4090a158da4cbc1f79464Dan Gohman} 1147384152444d22d356f7b4090a158da4cbc1f79464Dan Gohman 114815ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth/// ConstantFoldConvertToInt - Attempt to an SSE floating point to integer 114915ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth/// conversion of a constant floating point. If roundTowardZero is false, the 115015ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth/// default IEEE rounding is used (toward nearest, ties to even). This matches 115115ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth/// the behavior of the non-truncating SSE instructions in the default rounding 115215ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth/// mode. The desired integer type Ty is used to select how many bits are 115315ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth/// available for the result. Returns null if the conversion cannot be 115415ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth/// performed, otherwise returns the Constant value resulting from the 115515ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth/// conversion. 115615ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruthstatic Constant *ConstantFoldConvertToInt(ConstantFP *Op, bool roundTowardZero, 1157db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *Ty) { 115815ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth assert(Op && "Called with NULL operand"); 115915ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth APFloat Val(Op->getValueAPF()); 116015ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth 116115ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth // All of these conversion intrinsics form an integer of at most 64bits. 116215ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth unsigned ResultWidth = cast<IntegerType>(Ty)->getBitWidth(); 116315ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth assert(ResultWidth <= 64 && 116415ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth "Can only constant fold conversions to 64 and 32 bit ints"); 116515ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth 116615ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth uint64_t UIntVal; 116715ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth bool isExact = false; 116815ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth APFloat::roundingMode mode = roundTowardZero? APFloat::rmTowardZero 116915ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth : APFloat::rmNearestTiesToEven; 117015ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth APFloat::opStatus status = Val.convertToInteger(&UIntVal, ResultWidth, 117115ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth /*isSigned=*/true, mode, 117215ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth &isExact); 117315ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth if (status != APFloat::opOK && status != APFloat::opInexact) 117415ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth return 0; 117515ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth return ConstantInt::get(Ty, UIntVal, /*isSigned=*/true); 117615ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth} 117715ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth 1178bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell/// ConstantFoldCall - Attempt to constant fold a call to the specified function 1179bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell/// with the specified arguments, returning null if unsuccessful. 1180bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John CriswellConstant * 1181618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosierllvm::ConstantFoldCall(Function *F, ArrayRef<Constant *> Operands, 1182618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier const TargetLibraryInfo *TLI) { 11836f532a988e5c02fcb271fb65e9e7b83f0147d3f2Chris Lattner if (!F->hasName()) return 0; 1184f0443c1eb44d737d9bd78962932fc80f74c6113cDaniel Dunbar StringRef Name = F->getName(); 1185e65cd40ab99c3bd43dc997873e500643d012bbadChris Lattner 1186db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *Ty = F->getReturnType(); 11871d2f569c3428d70d0cf690c9adb459ad4a3ecff2Jay Foad if (Operands.size() == 1) { 1188bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell if (ConstantFP *Op = dyn_cast<ConstantFP>(Operands[0])) { 1189f4db877cff62340aaf9f29d3994562ae1cee8d72Chandler Carruth if (F->getIntrinsicID() == Intrinsic::convert_to_fp16) { 1190faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov APFloat Val(Op->getValueAPF()); 1191faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov 1192faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov bool lost = false; 1193faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov Val.convert(APFloat::IEEEhalf, APFloat::rmNearestTiesToEven, &lost); 1194faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov 1195faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov return ConstantInt::get(F->getContext(), Val.bitcastToAPInt()); 1196faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov } 119721646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier if (!TLI) 119821646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier return 0; 1199faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov 1200d0806a1519f3bc450d2dea3e6f22db73fc797e4cChris Lattner if (!Ty->isFloatTy() && !Ty->isDoubleTy()) 120143421b3dd70af5b70e71816521f37502c397cc65Dale Johannesen return 0; 12021f386c4ce1983c448320aecca8d3cd27d1497867Jakob Stoklund Olesen 12031f386c4ce1983c448320aecca8d3cd27d1497867Jakob Stoklund Olesen /// We only fold functions with finite arguments. Folding NaN and inf is 12041f386c4ce1983c448320aecca8d3cd27d1497867Jakob Stoklund Olesen /// likely to be aborted with an exception anyway, and some host libms 12051f386c4ce1983c448320aecca8d3cd27d1497867Jakob Stoklund Olesen /// have known errors raising exceptions. 12061f386c4ce1983c448320aecca8d3cd27d1497867Jakob Stoklund Olesen if (Op->getValueAPF().isNaN() || Op->getValueAPF().isInfinity()) 12071f386c4ce1983c448320aecca8d3cd27d1497867Jakob Stoklund Olesen return 0; 12081f386c4ce1983c448320aecca8d3cd27d1497867Jakob Stoklund Olesen 120943421b3dd70af5b70e71816521f37502c397cc65Dale Johannesen /// Currently APFloat versions of these functions do not exist, so we use 121043421b3dd70af5b70e71816521f37502c397cc65Dale Johannesen /// the host native double versions. Float versions are not called 121143421b3dd70af5b70e71816521f37502c397cc65Dale Johannesen /// directly but for all these it is true (float)(f((double)arg)) == 121243421b3dd70af5b70e71816521f37502c397cc65Dale Johannesen /// f(arg). Long double not supported yet. 1213d0806a1519f3bc450d2dea3e6f22db73fc797e4cChris Lattner double V = Ty->isFloatTy() ? (double)Op->getValueAPF().convertToFloat() : 121443421b3dd70af5b70e71816521f37502c397cc65Dale Johannesen Op->getValueAPF().convertToDouble(); 1215f0443c1eb44d737d9bd78962932fc80f74c6113cDaniel Dunbar switch (Name[0]) { 1216c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner case 'a': 121721646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier if (Name == "acos" && TLI->has(LibFunc::acos)) 12187b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(acos, V, Ty); 121921646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier else if (Name == "asin" && TLI->has(LibFunc::asin)) 12207b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(asin, V, Ty); 122121646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier else if (Name == "atan" && TLI->has(LibFunc::atan)) 12227b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(atan, V, Ty); 1223c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner break; 1224c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner case 'c': 122521646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier if (Name == "ceil" && TLI->has(LibFunc::ceil)) 12267b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(ceil, V, Ty); 122721646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier else if (Name == "cos" && TLI->has(LibFunc::cos)) 12287b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(cos, V, Ty); 122921646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier else if (Name == "cosh" && TLI->has(LibFunc::cosh)) 12307b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(cosh, V, Ty); 123121646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier else if (Name == "cosf" && TLI->has(LibFunc::cosf)) 12327b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(cos, V, Ty); 1233c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner break; 1234c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner case 'e': 123521646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier if (Name == "exp" && TLI->has(LibFunc::exp)) 12367b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(exp, V, Ty); 1237805fa97a0f5ff61624855da37d4881708fb8f0fbChris Lattner 123821646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier if (Name == "exp2" && TLI->has(LibFunc::exp2)) { 1239805fa97a0f5ff61624855da37d4881708fb8f0fbChris Lattner // Constant fold exp2(x) as pow(2,x) in case the host doesn't have a 1240805fa97a0f5ff61624855da37d4881708fb8f0fbChris Lattner // C99 library. 1241805fa97a0f5ff61624855da37d4881708fb8f0fbChris Lattner return ConstantFoldBinaryFP(pow, 2.0, V, Ty); 1242805fa97a0f5ff61624855da37d4881708fb8f0fbChris Lattner } 1243c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner break; 1244c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner case 'f': 124521646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier if (Name == "fabs" && TLI->has(LibFunc::fabs)) 12467b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(fabs, V, Ty); 124721646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier else if (Name == "floor" && TLI->has(LibFunc::floor)) 12487b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(floor, V, Ty); 1249c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner break; 1250c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner case 'l': 125121646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier if (Name == "log" && V > 0 && TLI->has(LibFunc::log)) 12527b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(log, V, Ty); 125321646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier else if (Name == "log10" && V > 0 && TLI->has(LibFunc::log10)) 12547b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(log10, V, Ty); 1255f4db877cff62340aaf9f29d3994562ae1cee8d72Chandler Carruth else if (F->getIntrinsicID() == Intrinsic::sqrt && 1256f4db877cff62340aaf9f29d3994562ae1cee8d72Chandler Carruth (Ty->isFloatTy() || Ty->isDoubleTy())) { 1257c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner if (V >= -0.0) 12587b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(sqrt, V, Ty); 1259c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner else // Undefined 1260a7235ea7245028a0723e8ab7fd011386b3900777Owen Anderson return Constant::getNullValue(Ty); 1261c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner } 1262c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner break; 1263c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner case 's': 126421646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier if (Name == "sin" && TLI->has(LibFunc::sin)) 12657b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(sin, V, Ty); 126621646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier else if (Name == "sinh" && TLI->has(LibFunc::sinh)) 12677b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(sinh, V, Ty); 126821646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier else if (Name == "sqrt" && V >= 0 && TLI->has(LibFunc::sqrt)) 12697b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(sqrt, V, Ty); 127021646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier else if (Name == "sqrtf" && V >= 0 && TLI->has(LibFunc::sqrtf)) 12717b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(sqrt, V, Ty); 127221646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier else if (Name == "sinf" && TLI->has(LibFunc::sinf)) 12737b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(sin, V, Ty); 1274c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner break; 1275c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner case 't': 127621646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier if (Name == "tan" && TLI->has(LibFunc::tan)) 12777b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(tan, V, Ty); 127821646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier else if (Name == "tanh" && TLI->has(LibFunc::tanh)) 12797b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(tanh, V, Ty); 1280c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner break; 1281c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner default: 1282c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner break; 1283bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell } 128468a0603dc5d3d79cd4de2195e950af26c3380cd7Chris Lattner return 0; 128568a0603dc5d3d79cd4de2195e950af26c3380cd7Chris Lattner } 1286f4db877cff62340aaf9f29d3994562ae1cee8d72Chandler Carruth 128768a0603dc5d3d79cd4de2195e950af26c3380cd7Chris Lattner if (ConstantInt *Op = dyn_cast<ConstantInt>(Operands[0])) { 1288f4db877cff62340aaf9f29d3994562ae1cee8d72Chandler Carruth switch (F->getIntrinsicID()) { 1289f4db877cff62340aaf9f29d3994562ae1cee8d72Chandler Carruth case Intrinsic::bswap: 12907b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantInt::get(F->getContext(), Op->getValue().byteSwap()); 1291f4db877cff62340aaf9f29d3994562ae1cee8d72Chandler Carruth case Intrinsic::ctpop: 1292eed707b1e6097aac2bb6b3d47271f6300ace7f2eOwen Anderson return ConstantInt::get(Ty, Op->getValue().countPopulation()); 1293f4db877cff62340aaf9f29d3994562ae1cee8d72Chandler Carruth case Intrinsic::cttz: 1294eed707b1e6097aac2bb6b3d47271f6300ace7f2eOwen Anderson return ConstantInt::get(Ty, Op->getValue().countTrailingZeros()); 1295f4db877cff62340aaf9f29d3994562ae1cee8d72Chandler Carruth case Intrinsic::ctlz: 1296eed707b1e6097aac2bb6b3d47271f6300ace7f2eOwen Anderson return ConstantInt::get(Ty, Op->getValue().countLeadingZeros()); 1297f4db877cff62340aaf9f29d3994562ae1cee8d72Chandler Carruth case Intrinsic::convert_from_fp16: { 1298faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov APFloat Val(Op->getValue()); 1299faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov 1300faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov bool lost = false; 1301faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov APFloat::opStatus status = 1302faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov Val.convert(APFloat::IEEEsingle, APFloat::rmNearestTiesToEven, &lost); 1303faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov 1304faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov // Conversion is always precise. 13058e68c3873549ca31533e2e3e40dda3a43cb79566Jeffrey Yasskin (void)status; 1306faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov assert(status == APFloat::opOK && !lost && 1307faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov "Precision lost during fp16 constfolding"); 1308faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov 1309faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov return ConstantFP::get(F->getContext(), Val); 1310faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov } 1311f4db877cff62340aaf9f29d3994562ae1cee8d72Chandler Carruth default: 1312f4db877cff62340aaf9f29d3994562ae1cee8d72Chandler Carruth return 0; 1313f4db877cff62340aaf9f29d3994562ae1cee8d72Chandler Carruth } 1314bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell } 1315f4db877cff62340aaf9f29d3994562ae1cee8d72Chandler Carruth 131615ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth if (ConstantVector *Op = dyn_cast<ConstantVector>(Operands[0])) { 131715ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth switch (F->getIntrinsicID()) { 131815ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth default: break; 131915ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth case Intrinsic::x86_sse_cvtss2si: 132015ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth case Intrinsic::x86_sse_cvtss2si64: 132115ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth case Intrinsic::x86_sse2_cvtsd2si: 132215ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth case Intrinsic::x86_sse2_cvtsd2si64: 132315ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth if (ConstantFP *FPOp = dyn_cast<ConstantFP>(Op->getOperand(0))) 132415ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth return ConstantFoldConvertToInt(FPOp, /*roundTowardZero=*/false, Ty); 132515ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth case Intrinsic::x86_sse_cvttss2si: 132615ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth case Intrinsic::x86_sse_cvttss2si64: 132715ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth case Intrinsic::x86_sse2_cvttsd2si: 132815ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth case Intrinsic::x86_sse2_cvttsd2si64: 132915ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth if (ConstantFP *FPOp = dyn_cast<ConstantFP>(Op->getOperand(0))) 133015ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth return ConstantFoldConvertToInt(FPOp, /*roundTowardZero=*/true, Ty); 133115ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth } 133215ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth } 133315ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth 13349ee7123ab462d97a2b242280e9bf0b47883531cbDan Gohman if (isa<UndefValue>(Operands[0])) { 1335f4db877cff62340aaf9f29d3994562ae1cee8d72Chandler Carruth if (F->getIntrinsicID() == Intrinsic::bswap) 13369ee7123ab462d97a2b242280e9bf0b47883531cbDan Gohman return Operands[0]; 13379ee7123ab462d97a2b242280e9bf0b47883531cbDan Gohman return 0; 13389ee7123ab462d97a2b242280e9bf0b47883531cbDan Gohman } 13399ee7123ab462d97a2b242280e9bf0b47883531cbDan Gohman 134068a0603dc5d3d79cd4de2195e950af26c3380cd7Chris Lattner return 0; 134168a0603dc5d3d79cd4de2195e950af26c3380cd7Chris Lattner } 1342f4db877cff62340aaf9f29d3994562ae1cee8d72Chandler Carruth 13431d2f569c3428d70d0cf690c9adb459ad4a3ecff2Jay Foad if (Operands.size() == 2) { 1344bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell if (ConstantFP *Op1 = dyn_cast<ConstantFP>(Operands[0])) { 1345d0806a1519f3bc450d2dea3e6f22db73fc797e4cChris Lattner if (!Ty->isFloatTy() && !Ty->isDoubleTy()) 13469ab7fb3ba47442d521a5bed09a27a5e8e7a786edDale Johannesen return 0; 1347d0806a1519f3bc450d2dea3e6f22db73fc797e4cChris Lattner double Op1V = Ty->isFloatTy() ? 1348d0806a1519f3bc450d2dea3e6f22db73fc797e4cChris Lattner (double)Op1->getValueAPF().convertToFloat() : 134943421b3dd70af5b70e71816521f37502c397cc65Dale Johannesen Op1->getValueAPF().convertToDouble(); 1350bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell if (ConstantFP *Op2 = dyn_cast<ConstantFP>(Operands[1])) { 1351d0806a1519f3bc450d2dea3e6f22db73fc797e4cChris Lattner if (Op2->getType() != Op1->getType()) 1352d0806a1519f3bc450d2dea3e6f22db73fc797e4cChris Lattner return 0; 135321646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier 1354d0806a1519f3bc450d2dea3e6f22db73fc797e4cChris Lattner double Op2V = Ty->isFloatTy() ? 135543421b3dd70af5b70e71816521f37502c397cc65Dale Johannesen (double)Op2->getValueAPF().convertToFloat(): 135643421b3dd70af5b70e71816521f37502c397cc65Dale Johannesen Op2->getValueAPF().convertToDouble(); 1357bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell 135824fbf2bf161b20b13d0a218345bdcd092bdd56f8Chad Rosier if (F->getIntrinsicID() == Intrinsic::pow) { 135924fbf2bf161b20b13d0a218345bdcd092bdd56f8Chad Rosier return ConstantFoldBinaryFP(pow, Op1V, Op2V, Ty); 136024fbf2bf161b20b13d0a218345bdcd092bdd56f8Chad Rosier } 136124fbf2bf161b20b13d0a218345bdcd092bdd56f8Chad Rosier if (!TLI) 136224fbf2bf161b20b13d0a218345bdcd092bdd56f8Chad Rosier return 0; 136321646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier if (Name == "pow" && TLI->has(LibFunc::pow)) 13647b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldBinaryFP(pow, Op1V, Op2V, Ty); 136521646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier if (Name == "fmod" && TLI->has(LibFunc::fmod)) 13667b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldBinaryFP(fmod, Op1V, Op2V, Ty); 136721646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier if (Name == "atan2" && TLI->has(LibFunc::atan2)) 13687b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldBinaryFP(atan2, Op1V, Op2V, Ty); 1369b5282dcf4745be59046f440980a1c0f0a50c9c09Chris Lattner } else if (ConstantInt *Op2C = dyn_cast<ConstantInt>(Operands[1])) { 1370f4db877cff62340aaf9f29d3994562ae1cee8d72Chandler Carruth if (F->getIntrinsicID() == Intrinsic::powi && Ty->isFloatTy()) 13717b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFP::get(F->getContext(), 13727b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner APFloat((float)std::pow((float)Op1V, 137302a260aa11a2e1b2c14335274d3c42ca3f3eabc0Chris Lattner (int)Op2C->getZExtValue()))); 1374f4db877cff62340aaf9f29d3994562ae1cee8d72Chandler Carruth if (F->getIntrinsicID() == Intrinsic::powi && Ty->isDoubleTy()) 13757b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFP::get(F->getContext(), 13767b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner APFloat((double)std::pow((double)Op1V, 13777b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner (int)Op2C->getZExtValue()))); 1378bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell } 137968a0603dc5d3d79cd4de2195e950af26c3380cd7Chris Lattner return 0; 1380bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell } 1381e65cd40ab99c3bd43dc997873e500643d012bbadChris Lattner 1382e65cd40ab99c3bd43dc997873e500643d012bbadChris Lattner if (ConstantInt *Op1 = dyn_cast<ConstantInt>(Operands[0])) { 1383e65cd40ab99c3bd43dc997873e500643d012bbadChris Lattner if (ConstantInt *Op2 = dyn_cast<ConstantInt>(Operands[1])) { 1384e65cd40ab99c3bd43dc997873e500643d012bbadChris Lattner switch (F->getIntrinsicID()) { 1385e65cd40ab99c3bd43dc997873e500643d012bbadChris Lattner default: break; 1386eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner case Intrinsic::sadd_with_overflow: 1387eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner case Intrinsic::uadd_with_overflow: 1388eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner case Intrinsic::ssub_with_overflow: 1389eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner case Intrinsic::usub_with_overflow: 13906208610fd602ebdb18bb793152899573d0b2b7abFrits van Bommel case Intrinsic::smul_with_overflow: 13916208610fd602ebdb18bb793152899573d0b2b7abFrits van Bommel case Intrinsic::umul_with_overflow: { 1392eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner APInt Res; 1393eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner bool Overflow; 1394eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner switch (F->getIntrinsicID()) { 1395eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner default: assert(0 && "Invalid case"); 1396eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner case Intrinsic::sadd_with_overflow: 1397eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner Res = Op1->getValue().sadd_ov(Op2->getValue(), Overflow); 1398eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner break; 1399eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner case Intrinsic::uadd_with_overflow: 1400eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner Res = Op1->getValue().uadd_ov(Op2->getValue(), Overflow); 1401eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner break; 1402eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner case Intrinsic::ssub_with_overflow: 1403eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner Res = Op1->getValue().ssub_ov(Op2->getValue(), Overflow); 1404eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner break; 1405eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner case Intrinsic::usub_with_overflow: 1406eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner Res = Op1->getValue().usub_ov(Op2->getValue(), Overflow); 1407eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner break; 1408eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner case Intrinsic::smul_with_overflow: 1409eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner Res = Op1->getValue().smul_ov(Op2->getValue(), Overflow); 1410eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner break; 14116208610fd602ebdb18bb793152899573d0b2b7abFrits van Bommel case Intrinsic::umul_with_overflow: 14126208610fd602ebdb18bb793152899573d0b2b7abFrits van Bommel Res = Op1->getValue().umul_ov(Op2->getValue(), Overflow); 14136208610fd602ebdb18bb793152899573d0b2b7abFrits van Bommel break; 1414eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner } 1415e65cd40ab99c3bd43dc997873e500643d012bbadChris Lattner Constant *Ops[] = { 1416eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner ConstantInt::get(F->getContext(), Res), 1417eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner ConstantInt::get(Type::getInt1Ty(F->getContext()), Overflow) 1418e65cd40ab99c3bd43dc997873e500643d012bbadChris Lattner }; 1419b065b06c12dba6001b8140df2744d0c856ef6ea1Chris Lattner return ConstantStruct::get(cast<StructType>(F->getReturnType()), Ops); 1420e65cd40ab99c3bd43dc997873e500643d012bbadChris Lattner } 1421e65cd40ab99c3bd43dc997873e500643d012bbadChris Lattner } 1422e65cd40ab99c3bd43dc997873e500643d012bbadChris Lattner } 1423e65cd40ab99c3bd43dc997873e500643d012bbadChris Lattner 1424e65cd40ab99c3bd43dc997873e500643d012bbadChris Lattner return 0; 1425e65cd40ab99c3bd43dc997873e500643d012bbadChris Lattner } 1426e65cd40ab99c3bd43dc997873e500643d012bbadChris Lattner return 0; 1427bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell } 1428bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell return 0; 1429bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell} 1430