ConstantFolding.cpp revision 04594aeffa3360882eb09a888a0970321b987b16
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 5504594aeffa3360882eb09a888a0970321b987b16Rafael Espindola // Handle a vector->integer cast. 5604594aeffa3360882eb09a888a0970321b987b16Rafael Espindola if (IntegerType *IT = dyn_cast<IntegerType>(DestTy)) { 5704594aeffa3360882eb09a888a0970321b987b16Rafael Espindola // FIXME: Remove ConstantVector support. 5804594aeffa3360882eb09a888a0970321b987b16Rafael Espindola if ((!isa<ConstantDataVector>(C) && !isa<ConstantVector>(C)) || 5904594aeffa3360882eb09a888a0970321b987b16Rafael Espindola // TODO: Handle big endian someday. 6004594aeffa3360882eb09a888a0970321b987b16Rafael Espindola !TD.isLittleEndian()) 6104594aeffa3360882eb09a888a0970321b987b16Rafael Espindola return ConstantExpr::getBitCast(C, DestTy); 6204594aeffa3360882eb09a888a0970321b987b16Rafael Espindola 6304594aeffa3360882eb09a888a0970321b987b16Rafael Espindola unsigned NumSrcElts = C->getType()->getVectorNumElements(); 6404594aeffa3360882eb09a888a0970321b987b16Rafael Espindola 6504594aeffa3360882eb09a888a0970321b987b16Rafael Espindola // If the vector is a vector of floating point, convert it to vector of int 6604594aeffa3360882eb09a888a0970321b987b16Rafael Espindola // to simplify things. 6704594aeffa3360882eb09a888a0970321b987b16Rafael Espindola if (C->getType()->getVectorElementType()->isFloatingPointTy()) { 6804594aeffa3360882eb09a888a0970321b987b16Rafael Espindola unsigned FPWidth = 6904594aeffa3360882eb09a888a0970321b987b16Rafael Espindola C->getType()->getVectorElementType()->getPrimitiveSizeInBits(); 7004594aeffa3360882eb09a888a0970321b987b16Rafael Espindola Type *SrcIVTy = 7104594aeffa3360882eb09a888a0970321b987b16Rafael Espindola VectorType::get(IntegerType::get(C->getContext(), FPWidth), NumSrcElts); 7204594aeffa3360882eb09a888a0970321b987b16Rafael Espindola // Ask VMCore to do the conversion now that #elts line up. 7304594aeffa3360882eb09a888a0970321b987b16Rafael Espindola C = ConstantExpr::getBitCast(C, SrcIVTy); 7404594aeffa3360882eb09a888a0970321b987b16Rafael Espindola } 7504594aeffa3360882eb09a888a0970321b987b16Rafael Espindola 7604594aeffa3360882eb09a888a0970321b987b16Rafael Espindola // Now that we know that the input value is a vector of integers, just shift 7704594aeffa3360882eb09a888a0970321b987b16Rafael Espindola // and insert them into our result. 7804594aeffa3360882eb09a888a0970321b987b16Rafael Espindola unsigned BitShift = 7904594aeffa3360882eb09a888a0970321b987b16Rafael Espindola TD.getTypeAllocSizeInBits(C->getType()->getVectorElementType()); 8004594aeffa3360882eb09a888a0970321b987b16Rafael Espindola APInt Result(IT->getBitWidth(), 0); 8104594aeffa3360882eb09a888a0970321b987b16Rafael Espindola for (unsigned i = 0; i != NumSrcElts; ++i) { 8204594aeffa3360882eb09a888a0970321b987b16Rafael Espindola // FIXME: Rework when we have ConstantDataVector. 8304594aeffa3360882eb09a888a0970321b987b16Rafael Espindola ConstantInt *Elt=dyn_cast_or_null<ConstantInt>(C->getAggregateElement(i)); 8404594aeffa3360882eb09a888a0970321b987b16Rafael Espindola if (Elt == 0) // Elt must be a constant expr or something. 8504594aeffa3360882eb09a888a0970321b987b16Rafael Espindola return ConstantExpr::getBitCast(C, DestTy); 8604594aeffa3360882eb09a888a0970321b987b16Rafael Espindola 8704594aeffa3360882eb09a888a0970321b987b16Rafael Espindola Result |= Elt->getValue().zextOrSelf(IT->getBitWidth()) << i*BitShift; 8804594aeffa3360882eb09a888a0970321b987b16Rafael Espindola } 8904594aeffa3360882eb09a888a0970321b987b16Rafael Espindola 9004594aeffa3360882eb09a888a0970321b987b16Rafael Espindola return ConstantInt::get(IT, Result); 9104594aeffa3360882eb09a888a0970321b987b16Rafael Espindola } 9204594aeffa3360882eb09a888a0970321b987b16Rafael Espindola 931c9fe0361b3f9f5c814ce761b97ecd11637a1d7dNadav Rotem // The code below only handles casts to vectors currently. 94db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner VectorType *DestVTy = dyn_cast<VectorType>(DestTy); 956333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner if (DestVTy == 0) 966333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner return ConstantExpr::getBitCast(C, DestTy); 976333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner 9893798daa3ecbfb29624052d3e54d2404ab65897aChris Lattner // If this is a scalar -> vector cast, convert the input into a <1 x scalar> 9993798daa3ecbfb29624052d3e54d2404ab65897aChris Lattner // vector so the code below can handle it uniformly. 10093798daa3ecbfb29624052d3e54d2404ab65897aChris Lattner if (isa<ConstantFP>(C) || isa<ConstantInt>(C)) { 10193798daa3ecbfb29624052d3e54d2404ab65897aChris Lattner Constant *Ops = C; // don't take the address of C! 1022ca5c8644e6c35b3a7910a576ed89cddb7b82c3bChris Lattner return FoldBitCast(ConstantVector::get(Ops), DestTy, TD); 10393798daa3ecbfb29624052d3e54d2404ab65897aChris Lattner } 10493798daa3ecbfb29624052d3e54d2404ab65897aChris Lattner 10593798daa3ecbfb29624052d3e54d2404ab65897aChris Lattner // If this is a bitcast from constant vector -> vector, fold it. 1066b0dc92043ab1f63d78b8796098575e1d777b701Chris Lattner // FIXME: Remove ConstantVector support. 1076b0dc92043ab1f63d78b8796098575e1d777b701Chris Lattner if (!isa<ConstantDataVector>(C) && !isa<ConstantVector>(C)) 10893798daa3ecbfb29624052d3e54d2404ab65897aChris Lattner return ConstantExpr::getBitCast(C, DestTy); 10993798daa3ecbfb29624052d3e54d2404ab65897aChris Lattner 1106333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner // If the element types match, VMCore can fold it. 1116333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner unsigned NumDstElt = DestVTy->getNumElements(); 1126b0dc92043ab1f63d78b8796098575e1d777b701Chris Lattner unsigned NumSrcElt = C->getType()->getVectorNumElements(); 1136333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner if (NumDstElt == NumSrcElt) 1146333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner return ConstantExpr::getBitCast(C, DestTy); 1156333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner 1166b0dc92043ab1f63d78b8796098575e1d777b701Chris Lattner Type *SrcEltTy = C->getType()->getVectorElementType(); 117db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *DstEltTy = DestVTy->getElementType(); 1186333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner 1196333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner // Otherwise, we're changing the number of elements in a vector, which 1206333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner // requires endianness information to do the right thing. For example, 1216333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner // bitcast (<2 x i64> <i64 0, i64 1> to <4 x i32>) 1226333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner // folds to (little endian): 1236333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner // <4 x i32> <i32 0, i32 0, i32 1, i32 0> 1246333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner // and to (big endian): 1256333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner // <4 x i32> <i32 0, i32 0, i32 0, i32 1> 1266333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner 1276333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner // First thing is first. We only want to think about integer here, so if 1286333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner // we have something in FP form, recast it as integer. 129b0bc6c361da9009e8414efde317d9bbff755f6c0Duncan Sands if (DstEltTy->isFloatingPointTy()) { 1306333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner // Fold to an vector of integers with same size as our FP type. 1316333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner unsigned FPWidth = DstEltTy->getPrimitiveSizeInBits(); 132db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *DestIVTy = 1336333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner VectorType::get(IntegerType::get(C->getContext(), FPWidth), NumDstElt); 1346333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner // Recursively handle this integer conversion, if possible. 1356333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner C = FoldBitCast(C, DestIVTy, TD); 1366333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner 1376333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner // Finally, VMCore can handle this now that #elts line up. 1386333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner return ConstantExpr::getBitCast(C, DestTy); 1396333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner } 1406333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner 1416333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner // Okay, we know the destination is integer, if the input is FP, convert 1426333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner // it to integer first. 143b0bc6c361da9009e8414efde317d9bbff755f6c0Duncan Sands if (SrcEltTy->isFloatingPointTy()) { 1446333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner unsigned FPWidth = SrcEltTy->getPrimitiveSizeInBits(); 145db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *SrcIVTy = 1466333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner VectorType::get(IntegerType::get(C->getContext(), FPWidth), NumSrcElt); 1476333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner // Ask VMCore to do the conversion now that #elts line up. 1486333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner C = ConstantExpr::getBitCast(C, SrcIVTy); 1496b0dc92043ab1f63d78b8796098575e1d777b701Chris Lattner // If VMCore wasn't able to fold it, bail out. 1506b0dc92043ab1f63d78b8796098575e1d777b701Chris Lattner if (!isa<ConstantVector>(C) && // FIXME: Remove ConstantVector. 1516b0dc92043ab1f63d78b8796098575e1d777b701Chris Lattner !isa<ConstantDataVector>(C)) 1526333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner return C; 1536333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner } 1546333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner 1556333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner // Now we know that the input and output vectors are both integer vectors 1566333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner // of the same size, and that their #elements is not the same. Do the 1576333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner // conversion here, which depends on whether the input or output has 1586333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner // more elements. 1596333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner bool isLittleEndian = TD.isLittleEndian(); 1606333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner 1616333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner SmallVector<Constant*, 32> Result; 1626333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner if (NumDstElt < NumSrcElt) { 1636333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner // Handle: bitcast (<4 x i32> <i32 0, i32 1, i32 2, i32 3> to <2 x i64>) 1646333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner Constant *Zero = Constant::getNullValue(DstEltTy); 1656333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner unsigned Ratio = NumSrcElt/NumDstElt; 1666333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner unsigned SrcBitSize = SrcEltTy->getPrimitiveSizeInBits(); 1676333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner unsigned SrcElt = 0; 1686333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner for (unsigned i = 0; i != NumDstElt; ++i) { 1696333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner // Build each element of the result. 1706333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner Constant *Elt = Zero; 1716333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner unsigned ShiftAmt = isLittleEndian ? 0 : SrcBitSize*(Ratio-1); 1726333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner for (unsigned j = 0; j != Ratio; ++j) { 1736b0dc92043ab1f63d78b8796098575e1d777b701Chris Lattner Constant *Src =dyn_cast<ConstantInt>(C->getAggregateElement(SrcElt++)); 1746333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner if (!Src) // Reject constantexpr elements. 1756333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner return ConstantExpr::getBitCast(C, DestTy); 1766333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner 1776333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner // Zero extend the element to the right size. 1786333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner Src = ConstantExpr::getZExt(Src, Elt->getType()); 1796333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner 1806333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner // Shift it to the right place, depending on endianness. 1816333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner Src = ConstantExpr::getShl(Src, 1826333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner ConstantInt::get(Src->getType(), ShiftAmt)); 1836333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner ShiftAmt += isLittleEndian ? SrcBitSize : -SrcBitSize; 1846333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner 1856333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner // Mix it in. 1866333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner Elt = ConstantExpr::getOr(Elt, Src); 1876333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner } 1886333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner Result.push_back(Elt); 1896333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner } 1906b0dc92043ab1f63d78b8796098575e1d777b701Chris Lattner return ConstantVector::get(Result); 1916b0dc92043ab1f63d78b8796098575e1d777b701Chris Lattner } 1926b0dc92043ab1f63d78b8796098575e1d777b701Chris Lattner 1936b0dc92043ab1f63d78b8796098575e1d777b701Chris Lattner // Handle: bitcast (<2 x i64> <i64 0, i64 1> to <4 x i32>) 1946b0dc92043ab1f63d78b8796098575e1d777b701Chris Lattner unsigned Ratio = NumDstElt/NumSrcElt; 1956b0dc92043ab1f63d78b8796098575e1d777b701Chris Lattner unsigned DstBitSize = DstEltTy->getPrimitiveSizeInBits(); 1966b0dc92043ab1f63d78b8796098575e1d777b701Chris Lattner 1976b0dc92043ab1f63d78b8796098575e1d777b701Chris Lattner // Loop over each source value, expanding into multiple results. 1986b0dc92043ab1f63d78b8796098575e1d777b701Chris Lattner for (unsigned i = 0; i != NumSrcElt; ++i) { 1996b0dc92043ab1f63d78b8796098575e1d777b701Chris Lattner Constant *Src = dyn_cast<ConstantInt>(C->getAggregateElement(i)); 2006b0dc92043ab1f63d78b8796098575e1d777b701Chris Lattner if (!Src) // Reject constantexpr elements. 2016b0dc92043ab1f63d78b8796098575e1d777b701Chris Lattner return ConstantExpr::getBitCast(C, DestTy); 2026333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner 2036b0dc92043ab1f63d78b8796098575e1d777b701Chris Lattner unsigned ShiftAmt = isLittleEndian ? 0 : DstBitSize*(Ratio-1); 2046b0dc92043ab1f63d78b8796098575e1d777b701Chris Lattner for (unsigned j = 0; j != Ratio; ++j) { 2056b0dc92043ab1f63d78b8796098575e1d777b701Chris Lattner // Shift the piece of the value into the right place, depending on 2066b0dc92043ab1f63d78b8796098575e1d777b701Chris Lattner // endianness. 2076b0dc92043ab1f63d78b8796098575e1d777b701Chris Lattner Constant *Elt = ConstantExpr::getLShr(Src, 2086b0dc92043ab1f63d78b8796098575e1d777b701Chris Lattner ConstantInt::get(Src->getType(), ShiftAmt)); 2096b0dc92043ab1f63d78b8796098575e1d777b701Chris Lattner ShiftAmt += isLittleEndian ? DstBitSize : -DstBitSize; 2106333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner 2116b0dc92043ab1f63d78b8796098575e1d777b701Chris Lattner // Truncate and remember this piece. 2126b0dc92043ab1f63d78b8796098575e1d777b701Chris Lattner Result.push_back(ConstantExpr::getTrunc(Elt, DstEltTy)); 2136333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner } 2146333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner } 2156333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner 2162ca5c8644e6c35b3a7910a576ed89cddb7b82c3bChris Lattner return ConstantVector::get(Result); 2176333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner} 2186333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner 2196333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner 22003dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner/// IsConstantOffsetFromGlobal - If this constant is actually a constant offset 22103dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner/// from a global, return the global and the constant. Because of 22203dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner/// constantexprs, this function is recursive. 22303dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattnerstatic bool IsConstantOffsetFromGlobal(Constant *C, GlobalValue *&GV, 22403dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner int64_t &Offset, const TargetData &TD) { 22503dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner // Trivial case, constant is the global. 22603dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner if ((GV = dyn_cast<GlobalValue>(C))) { 22703dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner Offset = 0; 22803dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner return true; 22903dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner } 23003dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner 23103dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner // Otherwise, if this isn't a constant expr, bail out. 23203dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner ConstantExpr *CE = dyn_cast<ConstantExpr>(C); 23303dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner if (!CE) return false; 23403dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner 23503dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner // Look through ptr->int and ptr->ptr casts. 23603dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner if (CE->getOpcode() == Instruction::PtrToInt || 23703dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner CE->getOpcode() == Instruction::BitCast) 23803dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner return IsConstantOffsetFromGlobal(CE->getOperand(0), GV, Offset, TD); 23903dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner 24003dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner // i32* getelementptr ([5 x i32]* @a, i32 0, i32 5) 24103dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner if (CE->getOpcode() == Instruction::GetElementPtr) { 24203dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner // Cannot compute this if the element type of the pointer is missing size 24303dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner // info. 244f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner if (!cast<PointerType>(CE->getOperand(0)->getType()) 245f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner ->getElementType()->isSized()) 24603dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner return false; 24703dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner 24803dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner // If the base isn't a global+constant, we aren't either. 24903dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner if (!IsConstantOffsetFromGlobal(CE->getOperand(0), GV, Offset, TD)) 25003dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner return false; 25103dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner 25203dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner // Otherwise, add any offset that our operands provide. 25303dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner gep_type_iterator GTI = gep_type_begin(CE); 254de2d74b213844b17eb5327fba27f21e699d3af66Gabor Greif for (User::const_op_iterator i = CE->op_begin() + 1, e = CE->op_end(); 255785c6af9797fcf551feef70f2ecb5cd075b6e3c4Gabor Greif i != e; ++i, ++GTI) { 256de2d74b213844b17eb5327fba27f21e699d3af66Gabor Greif ConstantInt *CI = dyn_cast<ConstantInt>(*i); 25703dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner if (!CI) return false; // Index isn't a simple constant? 258e368b460a206fafa0d31d5d059b1779b94f7df8cDan Gohman if (CI->isZero()) continue; // Not adding anything. 25903dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner 260db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner if (StructType *ST = dyn_cast<StructType>(*GTI)) { 26103dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner // N = N + Offset 262b1919e2f08ecb37140af676fd2916f8d5ed7df3dChris Lattner Offset += TD.getStructLayout(ST)->getElementOffset(CI->getZExtValue()); 26303dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner } else { 264db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner SequentialType *SQT = cast<SequentialType>(*GTI); 265777d2306b36816a53bc1ae1244c0dc7d998ae691Duncan Sands Offset += TD.getTypeAllocSize(SQT->getElementType())*CI->getSExtValue(); 26603dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner } 26703dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner } 26803dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner return true; 26903dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner } 27003dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner 27103dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner return false; 27203dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner} 27303dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner 274fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner/// ReadDataFromGlobal - Recursive helper to read bits out of global. C is the 275fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner/// constant being copied out of. ByteOffset is an offset into C. CurPtr is the 276fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner/// pointer to copy results into and BytesLeft is the number of bytes left in 277fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner/// the CurPtr buffer. TD is the target data. 278fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattnerstatic bool ReadDataFromGlobal(Constant *C, uint64_t ByteOffset, 279fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner unsigned char *CurPtr, unsigned BytesLeft, 280fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner const TargetData &TD) { 281fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner assert(ByteOffset <= TD.getTypeAllocSize(C->getType()) && 282fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner "Out of range access"); 283fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 284c7b1382e351249774be63bd73839e8a0671635e1Chris Lattner // If this element is zero or undefined, we can just return since *CurPtr is 285c7b1382e351249774be63bd73839e8a0671635e1Chris Lattner // zero initialized. 286fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner if (isa<ConstantAggregateZero>(C) || isa<UndefValue>(C)) 287fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner return true; 288fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 289fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner if (ConstantInt *CI = dyn_cast<ConstantInt>(C)) { 290fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner if (CI->getBitWidth() > 64 || 291fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner (CI->getBitWidth() & 7) != 0) 292fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner return false; 293fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 294fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner uint64_t Val = CI->getZExtValue(); 295fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner unsigned IntBytes = unsigned(CI->getBitWidth()/8); 296fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 297fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner for (unsigned i = 0; i != BytesLeft && ByteOffset != IntBytes; ++i) { 298c7b1382e351249774be63bd73839e8a0671635e1Chris Lattner CurPtr[i] = (unsigned char)(Val >> (ByteOffset * 8)); 299fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner ++ByteOffset; 300fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner } 301fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner return true; 302fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner } 303fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 304fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner if (ConstantFP *CFP = dyn_cast<ConstantFP>(C)) { 305fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner if (CFP->getType()->isDoubleTy()) { 3066333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner C = FoldBitCast(C, Type::getInt64Ty(C->getContext()), TD); 307fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner return ReadDataFromGlobal(C, ByteOffset, CurPtr, BytesLeft, TD); 308fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner } 309fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner if (CFP->getType()->isFloatTy()){ 3106333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner C = FoldBitCast(C, Type::getInt32Ty(C->getContext()), TD); 311fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner return ReadDataFromGlobal(C, ByteOffset, CurPtr, BytesLeft, TD); 312fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner } 313c7b1382e351249774be63bd73839e8a0671635e1Chris Lattner return false; 314fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner } 315a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner 316fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner if (ConstantStruct *CS = dyn_cast<ConstantStruct>(C)) { 317fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner const StructLayout *SL = TD.getStructLayout(CS->getType()); 318fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner unsigned Index = SL->getElementContainingOffset(ByteOffset); 319fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner uint64_t CurEltOffset = SL->getElementOffset(Index); 320fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner ByteOffset -= CurEltOffset; 321fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 322fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner while (1) { 323fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner // If the element access is to the element itself and not to tail padding, 324fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner // read the bytes from the element. 325fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner uint64_t EltSize = TD.getTypeAllocSize(CS->getOperand(Index)->getType()); 326fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 327fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner if (ByteOffset < EltSize && 328fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner !ReadDataFromGlobal(CS->getOperand(Index), ByteOffset, CurPtr, 329fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner BytesLeft, TD)) 330fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner return false; 331fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 332fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner ++Index; 333fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 334fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner // Check to see if we read from the last struct element, if so we're done. 335fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner if (Index == CS->getType()->getNumElements()) 336fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner return true; 337fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 338fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner // If we read all of the bytes we needed from this element we're done. 339fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner uint64_t NextEltOffset = SL->getElementOffset(Index); 340fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 341fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner if (BytesLeft <= NextEltOffset-CurEltOffset-ByteOffset) 342fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner return true; 343fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 344fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner // Move to the next element of the struct. 345c5af649758ac8c6ffb896d7d5c570640a9fa1deeChris Lattner CurPtr += NextEltOffset-CurEltOffset-ByteOffset; 346fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner BytesLeft -= NextEltOffset-CurEltOffset-ByteOffset; 347fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner ByteOffset = 0; 348fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner CurEltOffset = NextEltOffset; 349fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner } 350fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner // not reached. 351fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner } 352fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 3536b0dc92043ab1f63d78b8796098575e1d777b701Chris Lattner // FIXME: Remove ConstantVector 354a1f00f4d488eb5daff52faaf99c62ee652fd3b85Chris Lattner if (isa<ConstantArray>(C) || isa<ConstantVector>(C) || 355a1f00f4d488eb5daff52faaf99c62ee652fd3b85Chris Lattner isa<ConstantDataSequential>(C)) { 356a1f00f4d488eb5daff52faaf99c62ee652fd3b85Chris Lattner Type *EltTy = cast<SequentialType>(C->getType())->getElementType(); 357a1f00f4d488eb5daff52faaf99c62ee652fd3b85Chris Lattner uint64_t EltSize = TD.getTypeAllocSize(EltTy); 358fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner uint64_t Index = ByteOffset / EltSize; 359fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner uint64_t Offset = ByteOffset - Index * EltSize; 360a1f00f4d488eb5daff52faaf99c62ee652fd3b85Chris Lattner uint64_t NumElts; 361a1f00f4d488eb5daff52faaf99c62ee652fd3b85Chris Lattner if (ArrayType *AT = dyn_cast<ArrayType>(C->getType())) 362a1f00f4d488eb5daff52faaf99c62ee652fd3b85Chris Lattner NumElts = AT->getNumElements(); 363a1f00f4d488eb5daff52faaf99c62ee652fd3b85Chris Lattner else 364a1f00f4d488eb5daff52faaf99c62ee652fd3b85Chris Lattner NumElts = cast<VectorType>(C->getType())->getNumElements(); 365a1f00f4d488eb5daff52faaf99c62ee652fd3b85Chris Lattner 366a1f00f4d488eb5daff52faaf99c62ee652fd3b85Chris Lattner for (; Index != NumElts; ++Index) { 367a1f00f4d488eb5daff52faaf99c62ee652fd3b85Chris Lattner if (!ReadDataFromGlobal(C->getAggregateElement(Index), Offset, CurPtr, 368fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner BytesLeft, TD)) 369fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner return false; 370fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner if (EltSize >= BytesLeft) 371fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner return true; 372fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 373fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner Offset = 0; 374fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner BytesLeft -= EltSize; 375fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner CurPtr += EltSize; 376fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner } 377fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner return true; 378fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner } 379a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner 3806475d9434ff3e981160f85d3f1aa07c9e0ace6f2Anders Carlsson if (ConstantExpr *CE = dyn_cast<ConstantExpr>(C)) { 3815d43ff4e7ea30eaf5a1c417f0ec528900f5c83f2Anders Carlsson if (CE->getOpcode() == Instruction::IntToPtr && 3825d43ff4e7ea30eaf5a1c417f0ec528900f5c83f2Anders Carlsson CE->getOperand(0)->getType() == TD.getIntPtrType(CE->getContext())) 383a1f00f4d488eb5daff52faaf99c62ee652fd3b85Chris Lattner return ReadDataFromGlobal(CE->getOperand(0), ByteOffset, CurPtr, 384a1f00f4d488eb5daff52faaf99c62ee652fd3b85Chris Lattner BytesLeft, TD); 3856475d9434ff3e981160f85d3f1aa07c9e0ace6f2Anders Carlsson } 3866475d9434ff3e981160f85d3f1aa07c9e0ace6f2Anders Carlsson 387fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner // Otherwise, unknown initializer type. 388fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner return false; 389fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner} 390fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 391fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattnerstatic Constant *FoldReinterpretLoadFromConstPtr(Constant *C, 392fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner const TargetData &TD) { 393db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *LoadTy = cast<PointerType>(C->getType())->getElementType(); 394db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner IntegerType *IntType = dyn_cast<IntegerType>(LoadTy); 395fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 396fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner // If this isn't an integer load we can't fold it directly. 397fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner if (!IntType) { 398fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner // If this is a float/double load, we can try folding it as an int32/64 load 39917f0cd3cc1432231f2bc5601e7e463986a59d817Chris Lattner // and then bitcast the result. This can be useful for union cases. Note 40017f0cd3cc1432231f2bc5601e7e463986a59d817Chris Lattner // that address spaces don't matter here since we're not going to result in 40117f0cd3cc1432231f2bc5601e7e463986a59d817Chris Lattner // an actual new load. 402db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *MapTy; 40317f0cd3cc1432231f2bc5601e7e463986a59d817Chris Lattner if (LoadTy->isFloatTy()) 404fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner MapTy = Type::getInt32PtrTy(C->getContext()); 40517f0cd3cc1432231f2bc5601e7e463986a59d817Chris Lattner else if (LoadTy->isDoubleTy()) 406fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner MapTy = Type::getInt64PtrTy(C->getContext()); 4071df9859c40492511b8aa4321eb76496005d3b75bDuncan Sands else if (LoadTy->isVectorTy()) { 40817f0cd3cc1432231f2bc5601e7e463986a59d817Chris Lattner MapTy = IntegerType::get(C->getContext(), 40917f0cd3cc1432231f2bc5601e7e463986a59d817Chris Lattner TD.getTypeAllocSizeInBits(LoadTy)); 41017f0cd3cc1432231f2bc5601e7e463986a59d817Chris Lattner MapTy = PointerType::getUnqual(MapTy); 41117f0cd3cc1432231f2bc5601e7e463986a59d817Chris Lattner } else 412fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner return 0; 413fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 4146333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner C = FoldBitCast(C, MapTy, TD); 415fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner if (Constant *Res = FoldReinterpretLoadFromConstPtr(C, TD)) 4166333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner return FoldBitCast(Res, LoadTy, TD); 417fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner return 0; 418fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner } 419fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 420fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner unsigned BytesLoaded = (IntType->getBitWidth() + 7) / 8; 421739208a790398cf1f9da05149c768371e48781e8Chris Lattner if (BytesLoaded > 32 || BytesLoaded == 0) return 0; 422fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 423fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner GlobalValue *GVal; 424fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner int64_t Offset; 425fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner if (!IsConstantOffsetFromGlobal(C, GVal, Offset, TD)) 426fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner return 0; 427fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 428fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner GlobalVariable *GV = dyn_cast<GlobalVariable>(GVal); 429c7b1382e351249774be63bd73839e8a0671635e1Chris Lattner if (!GV || !GV->isConstant() || !GV->hasDefinitiveInitializer() || 430fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner !GV->getInitializer()->getType()->isSized()) 431fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner return 0; 432fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 433fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner // If we're loading off the beginning of the global, some bytes may be valid, 434fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner // but we don't try to handle this. 435fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner if (Offset < 0) return 0; 436fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 437fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner // If we're not accessing anything in this constant, the result is undefined. 438fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner if (uint64_t(Offset) >= TD.getTypeAllocSize(GV->getInitializer()->getType())) 439fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner return UndefValue::get(IntType); 440fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 441739208a790398cf1f9da05149c768371e48781e8Chris Lattner unsigned char RawBytes[32] = {0}; 442fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner if (!ReadDataFromGlobal(GV->getInitializer(), Offset, RawBytes, 443fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner BytesLoaded, TD)) 444fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner return 0; 445fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 446b31189f26297fe30214da3e8d14a4575d5fdd262Chris Lattner APInt ResultVal = APInt(IntType->getBitWidth(), RawBytes[BytesLoaded-1]); 447b31189f26297fe30214da3e8d14a4575d5fdd262Chris Lattner for (unsigned i = 1; i != BytesLoaded; ++i) { 448739208a790398cf1f9da05149c768371e48781e8Chris Lattner ResultVal <<= 8; 449ce800511701135ed86845b451693a248c83a7973Dan Gohman ResultVal |= RawBytes[BytesLoaded-1-i]; 450739208a790398cf1f9da05149c768371e48781e8Chris Lattner } 451fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 452739208a790398cf1f9da05149c768371e48781e8Chris Lattner return ConstantInt::get(IntType->getContext(), ResultVal); 453fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner} 454fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 455878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner/// ConstantFoldLoadFromConstPtr - Return the value that a load from C would 456878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner/// produce if it is constant and determinable. If this is not determinable, 457878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner/// return null. 458878e4946700824954d7eb7f3ff660353db8e0d17Chris LattnerConstant *llvm::ConstantFoldLoadFromConstPtr(Constant *C, 459878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner const TargetData *TD) { 460878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner // First, try the easy cases: 461878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner if (GlobalVariable *GV = dyn_cast<GlobalVariable>(C)) 462878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner if (GV->isConstant() && GV->hasDefinitiveInitializer()) 463878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner return GV->getInitializer(); 464878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner 465e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner // If the loaded value isn't a constant expr, we can't handle it. 466e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner ConstantExpr *CE = dyn_cast<ConstantExpr>(C); 467e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner if (!CE) return 0; 468e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner 469e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner if (CE->getOpcode() == Instruction::GetElementPtr) { 470e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner if (GlobalVariable *GV = dyn_cast<GlobalVariable>(CE->getOperand(0))) 471e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner if (GV->isConstant() && GV->hasDefinitiveInitializer()) 472e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner if (Constant *V = 473e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner ConstantFoldLoadThroughGEPConstantExpr(GV->getInitializer(), CE)) 474e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner return V; 475e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner } 476e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner 477e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner // Instead of loading constant c string, use corresponding integer value 478e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner // directly if string length is small enough. 479e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner std::string Str; 48044a7a380aa9fbb303f57f4f8269062b7b56be980Chris Lattner if (TD && GetConstantStringInfo(CE, Str) && !Str.empty()) { 481fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner unsigned StrLen = Str.length(); 482db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *Ty = cast<PointerType>(CE->getType())->getElementType(); 483fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner unsigned NumBits = Ty->getPrimitiveSizeInBits(); 484eae2895120500d06bf6de9725efeeff498f60385Chris Lattner // Replace load with immediate integer if the result is an integer or fp 485eae2895120500d06bf6de9725efeeff498f60385Chris Lattner // value. 486eae2895120500d06bf6de9725efeeff498f60385Chris Lattner if ((NumBits >> 3) == StrLen + 1 && (NumBits & 7) == 0 && 487490b8f5d64823d998fc12ad3a91ecacd82c1901eChandler Carruth (isa<IntegerType>(Ty) || Ty->isFloatingPointTy())) { 488fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner APInt StrVal(NumBits, 0); 489fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner APInt SingleChar(NumBits, 0); 490e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner if (TD->isLittleEndian()) { 491fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner for (signed i = StrLen-1; i >= 0; i--) { 492e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner SingleChar = (uint64_t) Str[i] & UCHAR_MAX; 49362d327e241f59735eed836e98c8f11ffaa6cf6f8Chris Lattner StrVal = (StrVal << 8) | SingleChar; 49462d327e241f59735eed836e98c8f11ffaa6cf6f8Chris Lattner } 495e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner } else { 496fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner for (unsigned i = 0; i < StrLen; i++) { 497e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner SingleChar = (uint64_t) Str[i] & UCHAR_MAX; 498e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner StrVal = (StrVal << 8) | SingleChar; 499e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner } 500e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner // Append NULL at the end. 501e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner SingleChar = 0; 502e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner StrVal = (StrVal << 8) | SingleChar; 50362d327e241f59735eed836e98c8f11ffaa6cf6f8Chris Lattner } 504eae2895120500d06bf6de9725efeeff498f60385Chris Lattner 505eae2895120500d06bf6de9725efeeff498f60385Chris Lattner Constant *Res = ConstantInt::get(CE->getContext(), StrVal); 506eae2895120500d06bf6de9725efeeff498f60385Chris Lattner if (Ty->isFloatingPointTy()) 507eae2895120500d06bf6de9725efeeff498f60385Chris Lattner Res = ConstantExpr::getBitCast(Res, Ty); 508eae2895120500d06bf6de9725efeeff498f60385Chris Lattner return Res; 50962d327e241f59735eed836e98c8f11ffaa6cf6f8Chris Lattner } 510878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner } 511e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner 512e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner // If this load comes from anywhere in a constant global, and if the global 513e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner // is all undef or zero, we know what it loads. 514bd1801b5553c8be3960255a92738464e0010b6f6Dan Gohman if (GlobalVariable *GV = 515bd1801b5553c8be3960255a92738464e0010b6f6Dan Gohman dyn_cast<GlobalVariable>(GetUnderlyingObject(CE, TD))) { 516e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner if (GV->isConstant() && GV->hasDefinitiveInitializer()) { 517db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *ResTy = cast<PointerType>(C->getType())->getElementType(); 518e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner if (GV->getInitializer()->isNullValue()) 519e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner return Constant::getNullValue(ResTy); 520e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner if (isa<UndefValue>(GV->getInitializer())) 521e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner return UndefValue::get(ResTy); 522e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner } 523e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner } 524e00c43fc35d64b2ee8d2b4692c02375d2d7b4aa4Chris Lattner 525fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner // Try hard to fold loads from bitcasted strange and non-type-safe things. We 526fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner // currently don't do any of this for big endian systems. It can be 527fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner // generalized in the future if someone is interested. 528fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner if (TD && TD->isLittleEndian()) 529fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner return FoldReinterpretLoadFromConstPtr(CE, *TD); 530878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner return 0; 531878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner} 532878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner 533878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattnerstatic Constant *ConstantFoldLoadInst(const LoadInst *LI, const TargetData *TD){ 534878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner if (LI->isVolatile()) return 0; 535878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner 536878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner if (Constant *C = dyn_cast<Constant>(LI->getOperand(0))) 537878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner return ConstantFoldLoadFromConstPtr(C, TD); 538fe8c7c807c6f815561bc2bdddfd330b05dbdfc93Chris Lattner 539878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner return 0; 540878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner} 54103dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner 54203dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner/// SymbolicallyEvaluateBinop - One of Op0/Op1 is a constant expression. 54367e3566d8f6875e9c92d224f493023408a66227eNick Lewycky/// Attempt to symbolically evaluate the result of a binary operator merging 54403dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner/// these together. If target data info is available, it is provided as TD, 54503dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner/// otherwise TD is null. 54603dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattnerstatic Constant *SymbolicallyEvaluateBinop(unsigned Opc, Constant *Op0, 5477b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner Constant *Op1, const TargetData *TD){ 54803dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner // SROA 54903dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner 55003dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner // Fold (and 0xffffffff00000000, (shl x, 32)) -> shl. 55103dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner // Fold (lshr (or X, Y), 32) -> (lshr [X/Y], 32) if one doesn't contribute 55203dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner // bits. 55303dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner 55403dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner 55503dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner // If the constant expr is something like &A[123] - &A[4].f, fold this into a 55603dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner // constant. This happens frequently when iterating over a global array. 55703dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner if (Opc == Instruction::Sub && TD) { 55803dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner GlobalValue *GV1, *GV2; 55903dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner int64_t Offs1, Offs2; 56003dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner 56103dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner if (IsConstantOffsetFromGlobal(Op0, GV1, Offs1, *TD)) 56203dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner if (IsConstantOffsetFromGlobal(Op1, GV2, Offs2, *TD) && 56303dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner GV1 == GV2) { 56403dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner // (&GV+C1) - (&GV+C2) -> C1-C2, pointer arithmetic cannot overflow. 565eed707b1e6097aac2bb6b3d47271f6300ace7f2eOwen Anderson return ConstantInt::get(Op0->getType(), Offs1-Offs2); 56603dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner } 56703dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner } 56803dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner 56903dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner return 0; 57003dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner} 57103dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner 5724f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman/// CastGEPIndices - If array indices are not pointer-sized integers, 5734f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman/// explicitly cast them so that they aren't implicitly casted by the 5744f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman/// getelementptr. 5751d2f569c3428d70d0cf690c9adb459ad4a3ecff2Jay Foadstatic Constant *CastGEPIndices(ArrayRef<Constant *> Ops, 576618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier Type *ResultTy, const TargetData *TD, 577618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier const TargetLibraryInfo *TLI) { 5784f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman if (!TD) return 0; 579db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *IntPtrTy = TD->getIntPtrType(ResultTy->getContext()); 5804f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman 5814f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman bool Any = false; 5824f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman SmallVector<Constant*, 32> NewIdxs; 5831d2f569c3428d70d0cf690c9adb459ad4a3ecff2Jay Foad for (unsigned i = 1, e = Ops.size(); i != e; ++i) { 5844f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman if ((i == 1 || 5854f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman !isa<StructType>(GetElementPtrInst::getIndexedType(Ops[0]->getType(), 586a9203109f4ac95aa7e9624f2838e3d89623ec902Jay Foad Ops.slice(1, i-1)))) && 5874f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman Ops[i]->getType() != IntPtrTy) { 5884f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman Any = true; 5894f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman NewIdxs.push_back(ConstantExpr::getCast(CastInst::getCastOpcode(Ops[i], 5904f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman true, 5914f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman IntPtrTy, 5924f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman true), 5934f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman Ops[i], IntPtrTy)); 5944f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman } else 5954f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman NewIdxs.push_back(Ops[i]); 5964f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman } 5974f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman if (!Any) return 0; 5984f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman 5994f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman Constant *C = 600dab3d29605a5c83db41b28176273ef55961120c1Jay Foad ConstantExpr::getGetElementPtr(Ops[0], NewIdxs); 6014f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman if (ConstantExpr *CE = dyn_cast<ConstantExpr>(C)) 602618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier if (Constant *Folded = ConstantFoldConstantExpression(CE, TD, TLI)) 6034f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman C = Folded; 6044f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman return C; 6054f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman} 6064f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman 60703dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner/// SymbolicallyEvaluateGEP - If we can symbolically evaluate the specified GEP 60803dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner/// constant expression, do so. 6091d2f569c3428d70d0cf690c9adb459ad4a3ecff2Jay Foadstatic Constant *SymbolicallyEvaluateGEP(ArrayRef<Constant *> Ops, 610618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier Type *ResultTy, const TargetData *TD, 611618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier const TargetLibraryInfo *TLI) { 61203dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner Constant *Ptr = Ops[0]; 6131608769abeb1430dc34f31ffac0d9850f99ae36aNadav Rotem if (!TD || !cast<PointerType>(Ptr->getType())->getElementType()->isSized() || 6141608769abeb1430dc34f31ffac0d9850f99ae36aNadav Rotem !Ptr->getType()->isPointerTy()) 61503dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner return 0; 616f1cadf28991b2643128bfe38990ff235f2cb1491Chris Lattner 617db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *IntPtrTy = TD->getIntPtrType(Ptr->getContext()); 618268e7d7a9446bb89b80472cdb7b9803f16c0191eChris Lattner 619268e7d7a9446bb89b80472cdb7b9803f16c0191eChris Lattner // If this is a constant expr gep that is effectively computing an 620268e7d7a9446bb89b80472cdb7b9803f16c0191eChris Lattner // "offsetof", fold it into 'cast int Size to T*' instead of 'gep 0, 0, 12' 6211d2f569c3428d70d0cf690c9adb459ad4a3ecff2Jay Foad for (unsigned i = 1, e = Ops.size(); i != e; ++i) 6228cd4efb6a5e51b4e77039c17338f290ca3a8ee92Chris Lattner if (!isa<ConstantInt>(Ops[i])) { 6238cd4efb6a5e51b4e77039c17338f290ca3a8ee92Chris Lattner 6248cd4efb6a5e51b4e77039c17338f290ca3a8ee92Chris Lattner // If this is "gep i8* Ptr, (sub 0, V)", fold this as: 6258cd4efb6a5e51b4e77039c17338f290ca3a8ee92Chris Lattner // "inttoptr (sub (ptrtoint Ptr), V)" 6261d2f569c3428d70d0cf690c9adb459ad4a3ecff2Jay Foad if (Ops.size() == 2 && 6278cd4efb6a5e51b4e77039c17338f290ca3a8ee92Chris Lattner cast<PointerType>(ResultTy)->getElementType()->isIntegerTy(8)) { 6288cd4efb6a5e51b4e77039c17338f290ca3a8ee92Chris Lattner ConstantExpr *CE = dyn_cast<ConstantExpr>(Ops[1]); 629156eb0a569a1ebac86ad8438645f690d8a3894c4Chris Lattner assert((CE == 0 || CE->getType() == IntPtrTy) && 630f1cadf28991b2643128bfe38990ff235f2cb1491Chris Lattner "CastGEPIndices didn't canonicalize index types!"); 6318cd4efb6a5e51b4e77039c17338f290ca3a8ee92Chris Lattner if (CE && CE->getOpcode() == Instruction::Sub && 632f1cadf28991b2643128bfe38990ff235f2cb1491Chris Lattner CE->getOperand(0)->isNullValue()) { 6338cd4efb6a5e51b4e77039c17338f290ca3a8ee92Chris Lattner Constant *Res = ConstantExpr::getPtrToInt(Ptr, CE->getType()); 6348cd4efb6a5e51b4e77039c17338f290ca3a8ee92Chris Lattner Res = ConstantExpr::getSub(Res, CE->getOperand(1)); 6358cd4efb6a5e51b4e77039c17338f290ca3a8ee92Chris Lattner Res = ConstantExpr::getIntToPtr(Res, ResultTy); 6368cd4efb6a5e51b4e77039c17338f290ca3a8ee92Chris Lattner if (ConstantExpr *ResCE = dyn_cast<ConstantExpr>(Res)) 637618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier Res = ConstantFoldConstantExpression(ResCE, TD, TLI); 6388cd4efb6a5e51b4e77039c17338f290ca3a8ee92Chris Lattner return Res; 6398cd4efb6a5e51b4e77039c17338f290ca3a8ee92Chris Lattner } 6408cd4efb6a5e51b4e77039c17338f290ca3a8ee92Chris Lattner } 641de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman return 0; 6428cd4efb6a5e51b4e77039c17338f290ca3a8ee92Chris Lattner } 643268e7d7a9446bb89b80472cdb7b9803f16c0191eChris Lattner 644f1cadf28991b2643128bfe38990ff235f2cb1491Chris Lattner unsigned BitWidth = TD->getTypeSizeInBits(IntPtrTy); 6458fbbb3980755d74539a0aed02bc18842ed2bd18dJay Foad APInt Offset = 6468fbbb3980755d74539a0aed02bc18842ed2bd18dJay Foad APInt(BitWidth, TD->getIndexedOffset(Ptr->getType(), 6478fbbb3980755d74539a0aed02bc18842ed2bd18dJay Foad makeArrayRef((Value **)Ops.data() + 1, 6488fbbb3980755d74539a0aed02bc18842ed2bd18dJay Foad Ops.size() - 1))); 649890edda7c22dd8525f9fe44085cfd7080e67dfeeDuncan Sands Ptr = cast<Constant>(Ptr->stripPointerCasts()); 6500891d752a68a25025ffc3339aab1f0ad3221b0edDan Gohman 6510891d752a68a25025ffc3339aab1f0ad3221b0edDan Gohman // If this is a GEP of a GEP, fold it all into a single GEP. 6520891d752a68a25025ffc3339aab1f0ad3221b0edDan Gohman while (GEPOperator *GEP = dyn_cast<GEPOperator>(Ptr)) { 6530891d752a68a25025ffc3339aab1f0ad3221b0edDan Gohman SmallVector<Value *, 4> NestedOps(GEP->op_begin()+1, GEP->op_end()); 654890edda7c22dd8525f9fe44085cfd7080e67dfeeDuncan Sands 655890edda7c22dd8525f9fe44085cfd7080e67dfeeDuncan Sands // Do not try the incorporate the sub-GEP if some index is not a number. 656890edda7c22dd8525f9fe44085cfd7080e67dfeeDuncan Sands bool AllConstantInt = true; 657890edda7c22dd8525f9fe44085cfd7080e67dfeeDuncan Sands for (unsigned i = 0, e = NestedOps.size(); i != e; ++i) 658890edda7c22dd8525f9fe44085cfd7080e67dfeeDuncan Sands if (!isa<ConstantInt>(NestedOps[i])) { 659890edda7c22dd8525f9fe44085cfd7080e67dfeeDuncan Sands AllConstantInt = false; 660890edda7c22dd8525f9fe44085cfd7080e67dfeeDuncan Sands break; 661890edda7c22dd8525f9fe44085cfd7080e67dfeeDuncan Sands } 662890edda7c22dd8525f9fe44085cfd7080e67dfeeDuncan Sands if (!AllConstantInt) 663890edda7c22dd8525f9fe44085cfd7080e67dfeeDuncan Sands break; 664890edda7c22dd8525f9fe44085cfd7080e67dfeeDuncan Sands 6650891d752a68a25025ffc3339aab1f0ad3221b0edDan Gohman Ptr = cast<Constant>(GEP->getOperand(0)); 6660891d752a68a25025ffc3339aab1f0ad3221b0edDan Gohman Offset += APInt(BitWidth, 6678fbbb3980755d74539a0aed02bc18842ed2bd18dJay Foad TD->getIndexedOffset(Ptr->getType(), NestedOps)); 668890edda7c22dd8525f9fe44085cfd7080e67dfeeDuncan Sands Ptr = cast<Constant>(Ptr->stripPointerCasts()); 6690891d752a68a25025ffc3339aab1f0ad3221b0edDan Gohman } 6700891d752a68a25025ffc3339aab1f0ad3221b0edDan Gohman 671de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman // If the base value for this address is a literal integer value, fold the 672de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman // getelementptr to the resulting integer value casted to the pointer type. 67340b037ef611efc5c43ffc43a911d0eb57e0d444dDan Gohman APInt BasePtr(BitWidth, 0); 67440b037ef611efc5c43ffc43a911d0eb57e0d444dDan Gohman if (ConstantExpr *CE = dyn_cast<ConstantExpr>(Ptr)) 67540b037ef611efc5c43ffc43a911d0eb57e0d444dDan Gohman if (CE->getOpcode() == Instruction::IntToPtr) 67640f8f6264d5af2c38e797e0dc59827cd231e8ff7Jay Foad if (ConstantInt *Base = dyn_cast<ConstantInt>(CE->getOperand(0))) 67740f8f6264d5af2c38e797e0dc59827cd231e8ff7Jay Foad BasePtr = Base->getValue().zextOrTrunc(BitWidth); 67840b037ef611efc5c43ffc43a911d0eb57e0d444dDan Gohman if (Ptr->isNullValue() || BasePtr != 0) { 6797b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner Constant *C = ConstantInt::get(Ptr->getContext(), Offset+BasePtr); 680de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman return ConstantExpr::getIntToPtr(C, ResultTy); 681de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman } 682de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman 683de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman // Otherwise form a regular getelementptr. Recompute the indices so that 684de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman // we eliminate over-indexing of the notional static type array bounds. 685de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman // This makes it easy to determine if the getelementptr is "inbounds". 686de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman // Also, this helps GlobalOpt do SROA on GlobalVariables. 687db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *Ty = Ptr->getType(); 688de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman SmallVector<Constant*, 32> NewIdxs; 6893d01334e6fa5caba1361d088fabf728c64b23c58Dan Gohman do { 690db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner if (SequentialType *ATy = dyn_cast<SequentialType>(Ty)) { 6911df9859c40492511b8aa4321eb76496005d3b75bDuncan Sands if (ATy->isPointerTy()) { 692e568fa2d29c58aaedcaa832f805949d5e4371b4bChris Lattner // The only pointer indexing we'll do is on the first index of the GEP. 693e568fa2d29c58aaedcaa832f805949d5e4371b4bChris Lattner if (!NewIdxs.empty()) 694e568fa2d29c58aaedcaa832f805949d5e4371b4bChris Lattner break; 695e568fa2d29c58aaedcaa832f805949d5e4371b4bChris Lattner 696e568fa2d29c58aaedcaa832f805949d5e4371b4bChris Lattner // Only handle pointers to sized types, not pointers to functions. 697e568fa2d29c58aaedcaa832f805949d5e4371b4bChris Lattner if (!ATy->getElementType()->isSized()) 698e568fa2d29c58aaedcaa832f805949d5e4371b4bChris Lattner return 0; 699e568fa2d29c58aaedcaa832f805949d5e4371b4bChris Lattner } 700e568fa2d29c58aaedcaa832f805949d5e4371b4bChris Lattner 701de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman // Determine which element of the array the offset points into. 702cda9706cb74a65d2feb2175613cb0fe46acc5aa2Dan Gohman APInt ElemSize(BitWidth, TD->getTypeAllocSize(ATy->getElementType())); 703db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner IntegerType *IntPtrTy = TD->getIntPtrType(Ty->getContext()); 704de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman if (ElemSize == 0) 705c1da204c433459713cbf4c62f0addf458e6b4b9fChris Lattner // The element size is 0. This may be [0 x Ty]*, so just use a zero 706e58391073f6e17cd1024f737618a6f6b8d1323ceDuncan Sands // index for this level and proceed to the next level to see if it can 707e58391073f6e17cd1024f737618a6f6b8d1323ceDuncan Sands // accommodate the offset. 708c1da204c433459713cbf4c62f0addf458e6b4b9fChris Lattner NewIdxs.push_back(ConstantInt::get(IntPtrTy, 0)); 709c1da204c433459713cbf4c62f0addf458e6b4b9fChris Lattner else { 710c1da204c433459713cbf4c62f0addf458e6b4b9fChris Lattner // The element size is non-zero divide the offset by the element 711c1da204c433459713cbf4c62f0addf458e6b4b9fChris Lattner // size (rounding down), to compute the index at this level. 712c1da204c433459713cbf4c62f0addf458e6b4b9fChris Lattner APInt NewIdx = Offset.udiv(ElemSize); 713c1da204c433459713cbf4c62f0addf458e6b4b9fChris Lattner Offset -= NewIdx * ElemSize; 714c1da204c433459713cbf4c62f0addf458e6b4b9fChris Lattner NewIdxs.push_back(ConstantInt::get(IntPtrTy, NewIdx)); 715c1da204c433459713cbf4c62f0addf458e6b4b9fChris Lattner } 716de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman Ty = ATy->getElementType(); 717db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner } else if (StructType *STy = dyn_cast<StructType>(Ty)) { 718cda9706cb74a65d2feb2175613cb0fe46acc5aa2Dan Gohman // Determine which field of the struct the offset points into. The 719cda9706cb74a65d2feb2175613cb0fe46acc5aa2Dan Gohman // getZExtValue is at least as safe as the StructLayout API because we 720cda9706cb74a65d2feb2175613cb0fe46acc5aa2Dan Gohman // know the offset is within the struct at this point. 721de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman const StructLayout &SL = *TD->getStructLayout(STy); 722cda9706cb74a65d2feb2175613cb0fe46acc5aa2Dan Gohman unsigned ElIdx = SL.getElementContainingOffset(Offset.getZExtValue()); 7237b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner NewIdxs.push_back(ConstantInt::get(Type::getInt32Ty(Ty->getContext()), 7247b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner ElIdx)); 725cda9706cb74a65d2feb2175613cb0fe46acc5aa2Dan Gohman Offset -= APInt(BitWidth, SL.getElementOffset(ElIdx)); 726de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman Ty = STy->getTypeAtIndex(ElIdx); 727de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman } else { 7283d01334e6fa5caba1361d088fabf728c64b23c58Dan Gohman // We've reached some non-indexable type. 7293d01334e6fa5caba1361d088fabf728c64b23c58Dan Gohman break; 730de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman } 7313d01334e6fa5caba1361d088fabf728c64b23c58Dan Gohman } while (Ty != cast<PointerType>(ResultTy)->getElementType()); 7323d01334e6fa5caba1361d088fabf728c64b23c58Dan Gohman 7333d01334e6fa5caba1361d088fabf728c64b23c58Dan Gohman // If we haven't used up the entire offset by descending the static 7343d01334e6fa5caba1361d088fabf728c64b23c58Dan Gohman // type, then the offset is pointing into the middle of an indivisible 7353d01334e6fa5caba1361d088fabf728c64b23c58Dan Gohman // member, so we can't simplify it. 7363d01334e6fa5caba1361d088fabf728c64b23c58Dan Gohman if (Offset != 0) 7373d01334e6fa5caba1361d088fabf728c64b23c58Dan Gohman return 0; 738de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman 7393bfbc4587a7e79f08f8c126a9e62c3475fb90f8bDan Gohman // Create a GEP. 7403bfbc4587a7e79f08f8c126a9e62c3475fb90f8bDan Gohman Constant *C = 741dab3d29605a5c83db41b28176273ef55961120c1Jay Foad ConstantExpr::getGetElementPtr(Ptr, NewIdxs); 7426e7ad958683f34bf6c014c88fef723e5a2d741beDan Gohman assert(cast<PointerType>(C->getType())->getElementType() == Ty && 7436e7ad958683f34bf6c014c88fef723e5a2d741beDan Gohman "Computed GetElementPtr has unexpected type!"); 7443d01334e6fa5caba1361d088fabf728c64b23c58Dan Gohman 7453d01334e6fa5caba1361d088fabf728c64b23c58Dan Gohman // If we ended up indexing a member with a type that doesn't match 7464c0d5d5db876b0628bdf6a2174263a1c0a9130e2Dan Gohman // the type of what the original indices indexed, add a cast. 7473d01334e6fa5caba1361d088fabf728c64b23c58Dan Gohman if (Ty != cast<PointerType>(ResultTy)->getElementType()) 7486333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner C = FoldBitCast(C, ResultTy, *TD); 749de0e587e63f71afb2ac53c9880c262089fe798bbDan Gohman 7503d01334e6fa5caba1361d088fabf728c64b23c58Dan Gohman return C; 75103dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner} 75203dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner 7531afab9c1e0ae2d145ce01718aa14065393117e70Chris Lattner 75403dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner 75503dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner//===----------------------------------------------------------------------===// 75603dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner// Constant Folding public APIs 75703dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner//===----------------------------------------------------------------------===// 75803dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner 759b9b369fa9983843c4ed77b3a35b6e9b7933955bbDuncan Sands/// ConstantFoldInstruction - Try to constant fold the specified instruction. 760b9b369fa9983843c4ed77b3a35b6e9b7933955bbDuncan Sands/// If successful, the constant result is returned, if not, null is returned. 761b9b369fa9983843c4ed77b3a35b6e9b7933955bbDuncan Sands/// Note that this fails if not all of the operands are constant. Otherwise, 762b9b369fa9983843c4ed77b3a35b6e9b7933955bbDuncan Sands/// this function can only fail when attempting to fold instructions like loads 763b9b369fa9983843c4ed77b3a35b6e9b7933955bbDuncan Sands/// and stores, which have no constant expression form. 764618c1dbd293d15ee19f61b1156ab8086ad28311aChad RosierConstant *llvm::ConstantFoldInstruction(Instruction *I, 765618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier const TargetData *TD, 766618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier const TargetLibraryInfo *TLI) { 767b9b369fa9983843c4ed77b3a35b6e9b7933955bbDuncan Sands // Handle PHI nodes quickly here... 7685520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner if (PHINode *PN = dyn_cast<PHINode>(I)) { 769c0362d5c6e0066c741bce056a65d8b4c026de19fDuncan Sands Constant *CommonValue = 0; 770c0362d5c6e0066c741bce056a65d8b4c026de19fDuncan Sands 771c0362d5c6e0066c741bce056a65d8b4c026de19fDuncan Sands for (unsigned i = 0, e = PN->getNumIncomingValues(); i != e; ++i) { 772c0362d5c6e0066c741bce056a65d8b4c026de19fDuncan Sands Value *Incoming = PN->getIncomingValue(i); 773b9b369fa9983843c4ed77b3a35b6e9b7933955bbDuncan Sands // If the incoming value is undef then skip it. Note that while we could 774b9b369fa9983843c4ed77b3a35b6e9b7933955bbDuncan Sands // skip the value if it is equal to the phi node itself we choose not to 775b9b369fa9983843c4ed77b3a35b6e9b7933955bbDuncan Sands // because that would break the rule that constant folding only applies if 776b9b369fa9983843c4ed77b3a35b6e9b7933955bbDuncan Sands // all operands are constants. 777b9b369fa9983843c4ed77b3a35b6e9b7933955bbDuncan Sands if (isa<UndefValue>(Incoming)) 778c0362d5c6e0066c741bce056a65d8b4c026de19fDuncan Sands continue; 779c0362d5c6e0066c741bce056a65d8b4c026de19fDuncan Sands // If the incoming value is not a constant, or is a different constant to 780c0362d5c6e0066c741bce056a65d8b4c026de19fDuncan Sands // the one we saw previously, then give up. 781c0362d5c6e0066c741bce056a65d8b4c026de19fDuncan Sands Constant *C = dyn_cast<Constant>(Incoming); 782c0362d5c6e0066c741bce056a65d8b4c026de19fDuncan Sands if (!C || (CommonValue && C != CommonValue)) 783c0362d5c6e0066c741bce056a65d8b4c026de19fDuncan Sands return 0; 784c0362d5c6e0066c741bce056a65d8b4c026de19fDuncan Sands CommonValue = C; 785c0362d5c6e0066c741bce056a65d8b4c026de19fDuncan Sands } 7865520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner 787c0362d5c6e0066c741bce056a65d8b4c026de19fDuncan Sands // If we reach here, all incoming values are the same constant or undef. 788c0362d5c6e0066c741bce056a65d8b4c026de19fDuncan Sands return CommonValue ? CommonValue : UndefValue::get(PN->getType()); 7895520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner } 7905520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner 7915520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner // Scan the operand list, checking to see if they are all constants, if so, 7925520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner // hand off to ConstantFoldInstOperands. 7935520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner SmallVector<Constant*, 8> Ops; 794de2d74b213844b17eb5327fba27f21e699d3af66Gabor Greif for (User::op_iterator i = I->op_begin(), e = I->op_end(); i != e; ++i) 795de2d74b213844b17eb5327fba27f21e699d3af66Gabor Greif if (Constant *Op = dyn_cast<Constant>(*i)) 7965520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner Ops.push_back(Op); 7975520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner else 7985520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner return 0; // All operands not constant! 7995520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner 800f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner if (const CmpInst *CI = dyn_cast<CmpInst>(I)) 8018f73deaa8732a556046bf4ac6207be55972e3b74Chris Lattner return ConstantFoldCompareInstOperands(CI->getPredicate(), Ops[0], Ops[1], 802618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier TD, TLI); 80358665d47aea69875c433fb039c8f10106abc1ceeChris Lattner 804878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner if (const LoadInst *LI = dyn_cast<LoadInst>(I)) 805878e4946700824954d7eb7f3ff660353db8e0d17Chris Lattner return ConstantFoldLoadInst(LI, TD); 8063ee0af35842999c1781052fe74fdcb19997de950Frits van Bommel 8073ee0af35842999c1781052fe74fdcb19997de950Frits van Bommel if (InsertValueInst *IVI = dyn_cast<InsertValueInst>(I)) 8083ee0af35842999c1781052fe74fdcb19997de950Frits van Bommel return ConstantExpr::getInsertValue( 8093ee0af35842999c1781052fe74fdcb19997de950Frits van Bommel cast<Constant>(IVI->getAggregateOperand()), 8103ee0af35842999c1781052fe74fdcb19997de950Frits van Bommel cast<Constant>(IVI->getInsertedValueOperand()), 811fc6d3a49867cd38954dc40936a88f1907252c6d2Jay Foad IVI->getIndices()); 8123ee0af35842999c1781052fe74fdcb19997de950Frits van Bommel 8133ee0af35842999c1781052fe74fdcb19997de950Frits van Bommel if (ExtractValueInst *EVI = dyn_cast<ExtractValueInst>(I)) 8143ee0af35842999c1781052fe74fdcb19997de950Frits van Bommel return ConstantExpr::getExtractValue( 8153ee0af35842999c1781052fe74fdcb19997de950Frits van Bommel cast<Constant>(EVI->getAggregateOperand()), 816fc6d3a49867cd38954dc40936a88f1907252c6d2Jay Foad EVI->getIndices()); 8173ee0af35842999c1781052fe74fdcb19997de950Frits van Bommel 818618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier return ConstantFoldInstOperands(I->getOpcode(), I->getType(), Ops, TD, TLI); 8195520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner} 8205520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner 8213dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky/// ConstantFoldConstantExpression - Attempt to fold the constant expression 8223dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky/// using the specified TargetData. If successful, the constant result is 8233dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky/// result is returned, if not, null is returned. 824baf0c67988feec5b7698f89dce499ddb7d0c3c42Dan GohmanConstant *llvm::ConstantFoldConstantExpression(const ConstantExpr *CE, 825618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier const TargetData *TD, 826618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier const TargetLibraryInfo *TLI) { 8273dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky SmallVector<Constant*, 8> Ops; 828c1da204c433459713cbf4c62f0addf458e6b4b9fChris Lattner for (User::const_op_iterator i = CE->op_begin(), e = CE->op_end(); 829c1da204c433459713cbf4c62f0addf458e6b4b9fChris Lattner i != e; ++i) { 83001b97dd01eec1197d79fceb7229f5ec233994de3Dan Gohman Constant *NewC = cast<Constant>(*i); 83101b97dd01eec1197d79fceb7229f5ec233994de3Dan Gohman // Recursively fold the ConstantExpr's operands. 83201b97dd01eec1197d79fceb7229f5ec233994de3Dan Gohman if (ConstantExpr *NewCE = dyn_cast<ConstantExpr>(NewC)) 833618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier NewC = ConstantFoldConstantExpression(NewCE, TD, TLI); 83401b97dd01eec1197d79fceb7229f5ec233994de3Dan Gohman Ops.push_back(NewC); 83501b97dd01eec1197d79fceb7229f5ec233994de3Dan Gohman } 8363dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky 8373dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky if (CE->isCompare()) 8388f73deaa8732a556046bf4ac6207be55972e3b74Chris Lattner return ConstantFoldCompareInstOperands(CE->getPredicate(), Ops[0], Ops[1], 839618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier TD, TLI); 840618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier return ConstantFoldInstOperands(CE->getOpcode(), CE->getType(), Ops, TD, TLI); 8413dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky} 8423dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky 8435520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner/// ConstantFoldInstOperands - Attempt to constant fold an instruction with the 8445520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner/// specified opcode and operands. If successful, the constant result is 8455520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner/// returned, if not, null is returned. Note that this function can fail when 8465520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner/// attempting to fold instructions like loads and stores, which have no 8475520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner/// constant expression form. 8485520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner/// 84901b97dd01eec1197d79fceb7229f5ec233994de3Dan Gohman/// TODO: This function neither utilizes nor preserves nsw/nuw/inbounds/etc 85001b97dd01eec1197d79fceb7229f5ec233994de3Dan Gohman/// information, due to only being passed an opcode and operands. Constant 85101b97dd01eec1197d79fceb7229f5ec233994de3Dan Gohman/// folding using this function strips this information. 85201b97dd01eec1197d79fceb7229f5ec233994de3Dan Gohman/// 853db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris LattnerConstant *llvm::ConstantFoldInstOperands(unsigned Opcode, Type *DestTy, 8541d2f569c3428d70d0cf690c9adb459ad4a3ecff2Jay Foad ArrayRef<Constant *> Ops, 855618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier const TargetData *TD, 856618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier const TargetLibraryInfo *TLI) { 85703dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner // Handle easy binops first. 858f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner if (Instruction::isBinaryOp(Opcode)) { 85903dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner if (isa<ConstantExpr>(Ops[0]) || isa<ConstantExpr>(Ops[1])) 8607b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner if (Constant *C = SymbolicallyEvaluateBinop(Opcode, Ops[0], Ops[1], TD)) 86103dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner return C; 86203dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner 863baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson return ConstantExpr::get(Opcode, Ops[0], Ops[1]); 86403dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner } 8655520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner 866f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner switch (Opcode) { 8675520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner default: return 0; 86879fa3cf451a74dd1cec0be2d77d7ea9737dffc69Chris Lattner case Instruction::ICmp: 86979fa3cf451a74dd1cec0be2d77d7ea9737dffc69Chris Lattner case Instruction::FCmp: assert(0 && "Invalid for compares"); 8705520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner case Instruction::Call: 8711d2f569c3428d70d0cf690c9adb459ad4a3ecff2Jay Foad if (Function *F = dyn_cast<Function>(Ops.back())) 8725520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner if (canConstantFoldCallTo(F)) 873618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier return ConstantFoldCall(F, Ops.slice(0, Ops.size() - 1), TLI); 8745520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner return 0; 875001f7534e0311508cc93831a595785bb177f8a18Chris Lattner case Instruction::PtrToInt: 876001f7534e0311508cc93831a595785bb177f8a18Chris Lattner // If the input is a inttoptr, eliminate the pair. This requires knowing 877001f7534e0311508cc93831a595785bb177f8a18Chris Lattner // the width of a pointer, so it can't be done in ConstantExpr::getCast. 878001f7534e0311508cc93831a595785bb177f8a18Chris Lattner if (ConstantExpr *CE = dyn_cast<ConstantExpr>(Ops[0])) { 879001f7534e0311508cc93831a595785bb177f8a18Chris Lattner if (TD && CE->getOpcode() == Instruction::IntToPtr) { 880001f7534e0311508cc93831a595785bb177f8a18Chris Lattner Constant *Input = CE->getOperand(0); 8816de29f8d960505421d61c80cdb738e16720b6c0eDan Gohman unsigned InWidth = Input->getType()->getScalarSizeInBits(); 88204aa2c3520c049333ca216fb794d700310c4a33dNick Lewycky if (TD->getPointerSizeInBits() < InWidth) { 88304aa2c3520c049333ca216fb794d700310c4a33dNick Lewycky Constant *Mask = 8847b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner ConstantInt::get(CE->getContext(), APInt::getLowBitsSet(InWidth, 88504aa2c3520c049333ca216fb794d700310c4a33dNick Lewycky TD->getPointerSizeInBits())); 886baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson Input = ConstantExpr::getAnd(Input, Mask); 88704aa2c3520c049333ca216fb794d700310c4a33dNick Lewycky } 888001f7534e0311508cc93831a595785bb177f8a18Chris Lattner // Do a zext or trunc to get to the dest size. 889baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson return ConstantExpr::getIntegerCast(Input, DestTy, false); 890001f7534e0311508cc93831a595785bb177f8a18Chris Lattner } 891001f7534e0311508cc93831a595785bb177f8a18Chris Lattner } 892baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson return ConstantExpr::getCast(Opcode, Ops[0], DestTy); 893001f7534e0311508cc93831a595785bb177f8a18Chris Lattner case Instruction::IntToPtr: 89481b06be055e74a3c23a1c8f17ead97f9f76335eeDuncan Sands // If the input is a ptrtoint, turn the pair into a ptr to ptr bitcast if 89581b06be055e74a3c23a1c8f17ead97f9f76335eeDuncan Sands // the int size is >= the ptr size. This requires knowing the width of a 89681b06be055e74a3c23a1c8f17ead97f9f76335eeDuncan Sands // pointer, so it can't be done in ConstantExpr::getCast. 897b80a2a686fe76496d71397f8bdda394d5718ab01Dan Gohman if (ConstantExpr *CE = dyn_cast<ConstantExpr>(Ops[0])) 8989a38e3e3991ea443e555d8060f91202a786acdd4Dan Gohman if (TD && 899b80a2a686fe76496d71397f8bdda394d5718ab01Dan Gohman TD->getPointerSizeInBits() <= CE->getType()->getScalarSizeInBits() && 900b80a2a686fe76496d71397f8bdda394d5718ab01Dan Gohman CE->getOpcode() == Instruction::PtrToInt) 901b80a2a686fe76496d71397f8bdda394d5718ab01Dan Gohman return FoldBitCast(CE->getOperand(0), DestTy, *TD); 902b80a2a686fe76496d71397f8bdda394d5718ab01Dan Gohman 903baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson return ConstantExpr::getCast(Opcode, Ops[0], DestTy); 9045520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner case Instruction::Trunc: 9055520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner case Instruction::ZExt: 9065520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner case Instruction::SExt: 9075520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner case Instruction::FPTrunc: 9085520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner case Instruction::FPExt: 9095520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner case Instruction::UIToFP: 9105520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner case Instruction::SIToFP: 9115520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner case Instruction::FPToUI: 9125520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner case Instruction::FPToSI: 913baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson return ConstantExpr::getCast(Opcode, Ops[0], DestTy); 9145520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner case Instruction::BitCast: 9151afab9c1e0ae2d145ce01718aa14065393117e70Chris Lattner if (TD) 9166333c3959b7e3e43930ab523f0657791e9eb5d48Chris Lattner return FoldBitCast(Ops[0], DestTy, *TD); 917baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson return ConstantExpr::getBitCast(Ops[0], DestTy); 9185520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner case Instruction::Select: 919baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson return ConstantExpr::getSelect(Ops[0], Ops[1], Ops[2]); 9205520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner case Instruction::ExtractElement: 921baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson return ConstantExpr::getExtractElement(Ops[0], Ops[1]); 9225520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner case Instruction::InsertElement: 923baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson return ConstantExpr::getInsertElement(Ops[0], Ops[1], Ops[2]); 9245520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner case Instruction::ShuffleVector: 925baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson return ConstantExpr::getShuffleVector(Ops[0], Ops[1], Ops[2]); 9265520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner case Instruction::GetElementPtr: 927618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier if (Constant *C = CastGEPIndices(Ops, DestTy, TD, TLI)) 9284f8eea82d8967cffa85b9df6c9255717b059009eDan Gohman return C; 929618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier if (Constant *C = SymbolicallyEvaluateGEP(Ops, DestTy, TD, TLI)) 93003dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner return C; 93103dd25ca964813c8b9fe14479443b9c21fb92c55Chris Lattner 932dab3d29605a5c83db41b28176273ef55961120c1Jay Foad return ConstantExpr::getGetElementPtr(Ops[0], Ops.slice(1)); 9335520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner } 9345520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner} 9355520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner 936f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner/// ConstantFoldCompareInstOperands - Attempt to constant fold a compare 937f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner/// instruction (icmp/fcmp) with the specified operands. If it fails, it 938f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner/// returns a constant expression of the specified operands. 939f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner/// 940f286f6fd93d569befe6e77c94a947e6e04e95685Chris LattnerConstant *llvm::ConstantFoldCompareInstOperands(unsigned Predicate, 9418f73deaa8732a556046bf4ac6207be55972e3b74Chris Lattner Constant *Ops0, Constant *Ops1, 942618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier const TargetData *TD, 943618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier const TargetLibraryInfo *TLI) { 944f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner // fold: icmp (inttoptr x), null -> icmp x, 0 945f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner // fold: icmp (ptrtoint x), 0 -> icmp x, null 9463dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky // fold: icmp (inttoptr x), (inttoptr y) -> icmp trunc/zext x, trunc/zext y 947f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner // fold: icmp (ptrtoint x), (ptrtoint y) -> icmp x, y 948f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner // 949f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner // ConstantExpr::getCompare cannot do this, because it doesn't have TD 950f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner // around to know if bit truncation is happening. 9518f73deaa8732a556046bf4ac6207be55972e3b74Chris Lattner if (ConstantExpr *CE0 = dyn_cast<ConstantExpr>(Ops0)) { 9528f73deaa8732a556046bf4ac6207be55972e3b74Chris Lattner if (TD && Ops1->isNullValue()) { 953db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *IntPtrTy = TD->getIntPtrType(CE0->getContext()); 954f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner if (CE0->getOpcode() == Instruction::IntToPtr) { 955f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner // Convert the integer value to the right size to ensure we get the 956f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner // proper extension or truncation. 957baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson Constant *C = ConstantExpr::getIntegerCast(CE0->getOperand(0), 958f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner IntPtrTy, false); 9598f73deaa8732a556046bf4ac6207be55972e3b74Chris Lattner Constant *Null = Constant::getNullValue(C->getType()); 960618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier return ConstantFoldCompareInstOperands(Predicate, C, Null, TD, TLI); 961f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner } 962f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner 963f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner // Only do this transformation if the int is intptrty in size, otherwise 964f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner // there is a truncation or extension that we aren't modeling. 965f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner if (CE0->getOpcode() == Instruction::PtrToInt && 966f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner CE0->getType() == IntPtrTy) { 967f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner Constant *C = CE0->getOperand(0); 9688f73deaa8732a556046bf4ac6207be55972e3b74Chris Lattner Constant *Null = Constant::getNullValue(C->getType()); 969618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier return ConstantFoldCompareInstOperands(Predicate, C, Null, TD, TLI); 970f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner } 971f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner } 972f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner 9738f73deaa8732a556046bf4ac6207be55972e3b74Chris Lattner if (ConstantExpr *CE1 = dyn_cast<ConstantExpr>(Ops1)) { 9743dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky if (TD && CE0->getOpcode() == CE1->getOpcode()) { 975db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *IntPtrTy = TD->getIntPtrType(CE0->getContext()); 9763dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky 9773dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky if (CE0->getOpcode() == Instruction::IntToPtr) { 9783dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky // Convert the integer value to the right size to ensure we get the 9793dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky // proper extension or truncation. 980baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson Constant *C0 = ConstantExpr::getIntegerCast(CE0->getOperand(0), 9813dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky IntPtrTy, false); 982baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson Constant *C1 = ConstantExpr::getIntegerCast(CE1->getOperand(0), 9833dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky IntPtrTy, false); 984618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier return ConstantFoldCompareInstOperands(Predicate, C0, C1, TD, TLI); 9853dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky } 9863dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky 9873dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky // Only do this transformation if the int is intptrty in size, otherwise 9883dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky // there is a truncation or extension that we aren't modeling. 9893dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky if ((CE0->getOpcode() == Instruction::PtrToInt && 9903dfd7bf5110c47e99fd0fcce96122b90f699ca3aNick Lewycky CE0->getType() == IntPtrTy && 9918f73deaa8732a556046bf4ac6207be55972e3b74Chris Lattner CE0->getOperand(0)->getType() == CE1->getOperand(0)->getType())) 9928f73deaa8732a556046bf4ac6207be55972e3b74Chris Lattner return ConstantFoldCompareInstOperands(Predicate, CE0->getOperand(0), 993618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier CE1->getOperand(0), TD, TLI); 994f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner } 995f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner } 99679fa3cf451a74dd1cec0be2d77d7ea9737dffc69Chris Lattner 99779fa3cf451a74dd1cec0be2d77d7ea9737dffc69Chris Lattner // icmp eq (or x, y), 0 -> (icmp eq x, 0) & (icmp eq y, 0) 99879fa3cf451a74dd1cec0be2d77d7ea9737dffc69Chris Lattner // icmp ne (or x, y), 0 -> (icmp ne x, 0) | (icmp ne y, 0) 99979fa3cf451a74dd1cec0be2d77d7ea9737dffc69Chris Lattner if ((Predicate == ICmpInst::ICMP_EQ || Predicate == ICmpInst::ICMP_NE) && 100079fa3cf451a74dd1cec0be2d77d7ea9737dffc69Chris Lattner CE0->getOpcode() == Instruction::Or && Ops1->isNullValue()) { 100179fa3cf451a74dd1cec0be2d77d7ea9737dffc69Chris Lattner Constant *LHS = 1002618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier ConstantFoldCompareInstOperands(Predicate, CE0->getOperand(0), Ops1, 1003618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier TD, TLI); 100479fa3cf451a74dd1cec0be2d77d7ea9737dffc69Chris Lattner Constant *RHS = 1005618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier ConstantFoldCompareInstOperands(Predicate, CE0->getOperand(1), Ops1, 1006618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier TD, TLI); 100779fa3cf451a74dd1cec0be2d77d7ea9737dffc69Chris Lattner unsigned OpC = 100879fa3cf451a74dd1cec0be2d77d7ea9737dffc69Chris Lattner Predicate == ICmpInst::ICMP_EQ ? Instruction::And : Instruction::Or; 100979fa3cf451a74dd1cec0be2d77d7ea9737dffc69Chris Lattner Constant *Ops[] = { LHS, RHS }; 1010618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier return ConstantFoldInstOperands(OpC, LHS->getType(), Ops, TD, TLI); 101179fa3cf451a74dd1cec0be2d77d7ea9737dffc69Chris Lattner } 1012f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner } 10138f73deaa8732a556046bf4ac6207be55972e3b74Chris Lattner 10148f73deaa8732a556046bf4ac6207be55972e3b74Chris Lattner return ConstantExpr::getCompare(Predicate, Ops0, Ops1); 1015f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner} 1016f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner 1017f286f6fd93d569befe6e77c94a947e6e04e95685Chris Lattner 10185520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner/// ConstantFoldLoadThroughGEPConstantExpr - Given a constant and a 10195520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner/// getelementptr constantexpr, return the constant value being addressed by the 10205520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner/// constant expression, or null if something is funny and we can't decide. 10215520732b24a5a321140dd79af70d321c7ff3dec9Chris LattnerConstant *llvm::ConstantFoldLoadThroughGEPConstantExpr(Constant *C, 1022c6f69e94fa46f585b59bb9d7ace3224b0e638c95Dan Gohman ConstantExpr *CE) { 1023a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner if (!CE->getOperand(1)->isNullValue()) 10245520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner return 0; // Do not allow stepping over the value! 1025a1f00f4d488eb5daff52faaf99c62ee652fd3b85Chris Lattner 1026a1f00f4d488eb5daff52faaf99c62ee652fd3b85Chris Lattner // Loop over all of the operands, tracking down which value we are 1027a1f00f4d488eb5daff52faaf99c62ee652fd3b85Chris Lattner // addressing. 1028a1f00f4d488eb5daff52faaf99c62ee652fd3b85Chris Lattner for (unsigned i = 2, e = CE->getNumOperands(); i != e; ++i) { 1029a1f00f4d488eb5daff52faaf99c62ee652fd3b85Chris Lattner C = C->getAggregateElement(CE->getOperand(i)); 1030a1f00f4d488eb5daff52faaf99c62ee652fd3b85Chris Lattner if (C == 0) return 0; 1031a1f00f4d488eb5daff52faaf99c62ee652fd3b85Chris Lattner } 1032a1f00f4d488eb5daff52faaf99c62ee652fd3b85Chris Lattner return C; 1033a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner} 1034a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner 1035a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner/// ConstantFoldLoadThroughGEPIndices - Given a constant and getelementptr 1036a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner/// indices (with an *implied* zero pointer index that is not in the list), 1037a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner/// return the constant value being addressed by a virtual load, or null if 1038a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner/// something is funny and we can't decide. 1039a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris LattnerConstant *llvm::ConstantFoldLoadThroughGEPIndices(Constant *C, 1040a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner ArrayRef<Constant*> Indices) { 10415520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner // Loop over all of the operands, tracking down which value we are 1042a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner // addressing. 1043a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner for (unsigned i = 0, e = Indices.size(); i != e; ++i) { 1044a1f00f4d488eb5daff52faaf99c62ee652fd3b85Chris Lattner C = C->getAggregateElement(Indices[i]); 1045a1f00f4d488eb5daff52faaf99c62ee652fd3b85Chris Lattner if (C == 0) return 0; 1046a97ddd0cffb9a9eb1313343fd49097c9057a5850Chris Lattner } 10475520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner return C; 10485520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner} 10495520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner 10505520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner 1051bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell//===----------------------------------------------------------------------===// 10525520732b24a5a321140dd79af70d321c7ff3dec9Chris Lattner// Constant Folding for Calls 1053bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell// 1054bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell 1055bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell/// canConstantFoldCallTo - Return true if its even possible to fold a call to 1056bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell/// the specified function. 1057bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswellbool 1058fa9b80eb64127b3d9691e18537975635520e51e9Dan Gohmanllvm::canConstantFoldCallTo(const Function *F) { 1059bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell switch (F->getIntrinsicID()) { 10609ab7fb3ba47442d521a5bed09a27a5e8e7a786edDale Johannesen case Intrinsic::sqrt: 106124fbf2bf161b20b13d0a218345bdcd092bdd56f8Chad Rosier case Intrinsic::pow: 10629ab7fb3ba47442d521a5bed09a27a5e8e7a786edDale Johannesen case Intrinsic::powi: 1063e9391fd9b52e93717b365bdd05c471101323a4dfReid Spencer case Intrinsic::bswap: 1064e9391fd9b52e93717b365bdd05c471101323a4dfReid Spencer case Intrinsic::ctpop: 1065e9391fd9b52e93717b365bdd05c471101323a4dfReid Spencer case Intrinsic::ctlz: 1066e9391fd9b52e93717b365bdd05c471101323a4dfReid Spencer case Intrinsic::cttz: 10671614e50d9f6d48a38e4dafb4b1a51fd4b37614f3Evan Phoenix case Intrinsic::sadd_with_overflow: 10686208610fd602ebdb18bb793152899573d0b2b7abFrits van Bommel case Intrinsic::uadd_with_overflow: 10691614e50d9f6d48a38e4dafb4b1a51fd4b37614f3Evan Phoenix case Intrinsic::ssub_with_overflow: 10706208610fd602ebdb18bb793152899573d0b2b7abFrits van Bommel case Intrinsic::usub_with_overflow: 1071eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner case Intrinsic::smul_with_overflow: 10726208610fd602ebdb18bb793152899573d0b2b7abFrits van Bommel case Intrinsic::umul_with_overflow: 1073faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov case Intrinsic::convert_from_fp16: 1074faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov case Intrinsic::convert_to_fp16: 107515ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth case Intrinsic::x86_sse_cvtss2si: 107615ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth case Intrinsic::x86_sse_cvtss2si64: 107715ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth case Intrinsic::x86_sse_cvttss2si: 107815ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth case Intrinsic::x86_sse_cvttss2si64: 107915ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth case Intrinsic::x86_sse2_cvtsd2si: 108015ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth case Intrinsic::x86_sse2_cvtsd2si64: 108115ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth case Intrinsic::x86_sse2_cvttsd2si: 108215ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth case Intrinsic::x86_sse2_cvttsd2si64: 1083bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell return true; 108468a0603dc5d3d79cd4de2195e950af26c3380cd7Chris Lattner default: 108568a0603dc5d3d79cd4de2195e950af26c3380cd7Chris Lattner return false; 108668a0603dc5d3d79cd4de2195e950af26c3380cd7Chris Lattner case 0: break; 1087bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell } 1088bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell 10896f532a988e5c02fcb271fb65e9e7b83f0147d3f2Chris Lattner if (!F->hasName()) return false; 1090f0443c1eb44d737d9bd78962932fc80f74c6113cDaniel Dunbar StringRef Name = F->getName(); 1091c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner 1092c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner // In these cases, the check of the length is required. We don't want to 1093c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner // return true for a name like "cos\0blah" which strcmp would return equal to 1094c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner // "cos", but has length 8. 1095f0443c1eb44d737d9bd78962932fc80f74c6113cDaniel Dunbar switch (Name[0]) { 1096c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner default: return false; 1097c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner case 'a': 1098f0443c1eb44d737d9bd78962932fc80f74c6113cDaniel Dunbar return Name == "acos" || Name == "asin" || 1099f0443c1eb44d737d9bd78962932fc80f74c6113cDaniel Dunbar Name == "atan" || Name == "atan2"; 1100c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner case 'c': 1101f0443c1eb44d737d9bd78962932fc80f74c6113cDaniel Dunbar return Name == "cos" || Name == "ceil" || Name == "cosf" || Name == "cosh"; 1102c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner case 'e': 1103805fa97a0f5ff61624855da37d4881708fb8f0fbChris Lattner return Name == "exp" || Name == "exp2"; 1104c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner case 'f': 1105f0443c1eb44d737d9bd78962932fc80f74c6113cDaniel Dunbar return Name == "fabs" || Name == "fmod" || Name == "floor"; 1106c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner case 'l': 1107f0443c1eb44d737d9bd78962932fc80f74c6113cDaniel Dunbar return Name == "log" || Name == "log10"; 1108c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner case 'p': 1109f0443c1eb44d737d9bd78962932fc80f74c6113cDaniel Dunbar return Name == "pow"; 1110c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner case 's': 1111f0443c1eb44d737d9bd78962932fc80f74c6113cDaniel Dunbar return Name == "sin" || Name == "sinh" || Name == "sqrt" || 1112f0443c1eb44d737d9bd78962932fc80f74c6113cDaniel Dunbar Name == "sinf" || Name == "sqrtf"; 1113c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner case 't': 1114f0443c1eb44d737d9bd78962932fc80f74c6113cDaniel Dunbar return Name == "tan" || Name == "tanh"; 1115bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell } 1116bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell} 1117bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell 111872d88ae5447a3929c97e88f4c806213847b5d988Chris Lattnerstatic Constant *ConstantFoldFP(double (*NativeFP)(double), double V, 1119db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *Ty) { 11203f2f21e640682c604d9322ba0708e122d1fcf5eeDan Gohman sys::llvm_fenv_clearexcept(); 1121bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell V = NativeFP(V); 11223f2f21e640682c604d9322ba0708e122d1fcf5eeDan Gohman if (sys::llvm_fenv_testexcept()) { 11233f2f21e640682c604d9322ba0708e122d1fcf5eeDan Gohman sys::llvm_fenv_clearexcept(); 1124f19f58a9360927f8ef50b9a029e8efb780d3d8a8Chris Lattner return 0; 112543421b3dd70af5b70e71816521f37502c397cc65Dale Johannesen } 1126f19f58a9360927f8ef50b9a029e8efb780d3d8a8Chris Lattner 1127d0806a1519f3bc450d2dea3e6f22db73fc797e4cChris Lattner if (Ty->isFloatTy()) 11287b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFP::get(Ty->getContext(), APFloat((float)V)); 1129d0806a1519f3bc450d2dea3e6f22db73fc797e4cChris Lattner if (Ty->isDoubleTy()) 11307b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFP::get(Ty->getContext(), APFloat(V)); 1131c23197a26f34f559ea9797de51e187087c039c42Torok Edwin llvm_unreachable("Can only constant fold float/double"); 1132bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell} 1133bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell 1134384152444d22d356f7b4090a158da4cbc1f79464Dan Gohmanstatic Constant *ConstantFoldBinaryFP(double (*NativeFP)(double, double), 1135db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner double V, double W, Type *Ty) { 11363f2f21e640682c604d9322ba0708e122d1fcf5eeDan Gohman sys::llvm_fenv_clearexcept(); 1137384152444d22d356f7b4090a158da4cbc1f79464Dan Gohman V = NativeFP(V, W); 11383f2f21e640682c604d9322ba0708e122d1fcf5eeDan Gohman if (sys::llvm_fenv_testexcept()) { 11393f2f21e640682c604d9322ba0708e122d1fcf5eeDan Gohman sys::llvm_fenv_clearexcept(); 1140f19f58a9360927f8ef50b9a029e8efb780d3d8a8Chris Lattner return 0; 114143421b3dd70af5b70e71816521f37502c397cc65Dale Johannesen } 1142f19f58a9360927f8ef50b9a029e8efb780d3d8a8Chris Lattner 1143d0806a1519f3bc450d2dea3e6f22db73fc797e4cChris Lattner if (Ty->isFloatTy()) 11447b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFP::get(Ty->getContext(), APFloat((float)V)); 1145d0806a1519f3bc450d2dea3e6f22db73fc797e4cChris Lattner if (Ty->isDoubleTy()) 11467b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFP::get(Ty->getContext(), APFloat(V)); 1147c23197a26f34f559ea9797de51e187087c039c42Torok Edwin llvm_unreachable("Can only constant fold float/double"); 1148384152444d22d356f7b4090a158da4cbc1f79464Dan Gohman} 1149384152444d22d356f7b4090a158da4cbc1f79464Dan Gohman 115015ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth/// ConstantFoldConvertToInt - Attempt to an SSE floating point to integer 115115ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth/// conversion of a constant floating point. If roundTowardZero is false, the 115215ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth/// default IEEE rounding is used (toward nearest, ties to even). This matches 115315ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth/// the behavior of the non-truncating SSE instructions in the default rounding 115415ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth/// mode. The desired integer type Ty is used to select how many bits are 115515ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth/// available for the result. Returns null if the conversion cannot be 115615ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth/// performed, otherwise returns the Constant value resulting from the 115715ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth/// conversion. 11586b0dc92043ab1f63d78b8796098575e1d777b701Chris Lattnerstatic Constant *ConstantFoldConvertToInt(const APFloat &Val, 11596b0dc92043ab1f63d78b8796098575e1d777b701Chris Lattner bool roundTowardZero, Type *Ty) { 116015ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth // All of these conversion intrinsics form an integer of at most 64bits. 116115ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth unsigned ResultWidth = cast<IntegerType>(Ty)->getBitWidth(); 116215ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth assert(ResultWidth <= 64 && 116315ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth "Can only constant fold conversions to 64 and 32 bit ints"); 116415ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth 116515ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth uint64_t UIntVal; 116615ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth bool isExact = false; 116715ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth APFloat::roundingMode mode = roundTowardZero? APFloat::rmTowardZero 116815ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth : APFloat::rmNearestTiesToEven; 116915ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth APFloat::opStatus status = Val.convertToInteger(&UIntVal, ResultWidth, 117015ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth /*isSigned=*/true, mode, 117115ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth &isExact); 117215ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth if (status != APFloat::opOK && status != APFloat::opInexact) 117315ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth return 0; 117415ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth return ConstantInt::get(Ty, UIntVal, /*isSigned=*/true); 117515ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth} 117615ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth 1177bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell/// ConstantFoldCall - Attempt to constant fold a call to the specified function 1178bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell/// with the specified arguments, returning null if unsuccessful. 1179bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John CriswellConstant * 1180618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosierllvm::ConstantFoldCall(Function *F, ArrayRef<Constant *> Operands, 1181618c1dbd293d15ee19f61b1156ab8086ad28311aChad Rosier const TargetLibraryInfo *TLI) { 11826f532a988e5c02fcb271fb65e9e7b83f0147d3f2Chris Lattner if (!F->hasName()) return 0; 1183f0443c1eb44d737d9bd78962932fc80f74c6113cDaniel Dunbar StringRef Name = F->getName(); 1184e65cd40ab99c3bd43dc997873e500643d012bbadChris Lattner 1185db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *Ty = F->getReturnType(); 11861d2f569c3428d70d0cf690c9adb459ad4a3ecff2Jay Foad if (Operands.size() == 1) { 1187bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell if (ConstantFP *Op = dyn_cast<ConstantFP>(Operands[0])) { 1188f4db877cff62340aaf9f29d3994562ae1cee8d72Chandler Carruth if (F->getIntrinsicID() == Intrinsic::convert_to_fp16) { 1189faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov APFloat Val(Op->getValueAPF()); 1190faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov 1191faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov bool lost = false; 1192faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov Val.convert(APFloat::IEEEhalf, APFloat::rmNearestTiesToEven, &lost); 1193faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov 1194faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov return ConstantInt::get(F->getContext(), Val.bitcastToAPInt()); 1195faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov } 119621646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier if (!TLI) 119721646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier return 0; 1198faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov 1199d0806a1519f3bc450d2dea3e6f22db73fc797e4cChris Lattner if (!Ty->isFloatTy() && !Ty->isDoubleTy()) 120043421b3dd70af5b70e71816521f37502c397cc65Dale Johannesen return 0; 12011f386c4ce1983c448320aecca8d3cd27d1497867Jakob Stoklund Olesen 12021f386c4ce1983c448320aecca8d3cd27d1497867Jakob Stoklund Olesen /// We only fold functions with finite arguments. Folding NaN and inf is 12031f386c4ce1983c448320aecca8d3cd27d1497867Jakob Stoklund Olesen /// likely to be aborted with an exception anyway, and some host libms 12041f386c4ce1983c448320aecca8d3cd27d1497867Jakob Stoklund Olesen /// have known errors raising exceptions. 12051f386c4ce1983c448320aecca8d3cd27d1497867Jakob Stoklund Olesen if (Op->getValueAPF().isNaN() || Op->getValueAPF().isInfinity()) 12061f386c4ce1983c448320aecca8d3cd27d1497867Jakob Stoklund Olesen return 0; 12071f386c4ce1983c448320aecca8d3cd27d1497867Jakob Stoklund Olesen 120843421b3dd70af5b70e71816521f37502c397cc65Dale Johannesen /// Currently APFloat versions of these functions do not exist, so we use 120943421b3dd70af5b70e71816521f37502c397cc65Dale Johannesen /// the host native double versions. Float versions are not called 121043421b3dd70af5b70e71816521f37502c397cc65Dale Johannesen /// directly but for all these it is true (float)(f((double)arg)) == 121143421b3dd70af5b70e71816521f37502c397cc65Dale Johannesen /// f(arg). Long double not supported yet. 1212d0806a1519f3bc450d2dea3e6f22db73fc797e4cChris Lattner double V = Ty->isFloatTy() ? (double)Op->getValueAPF().convertToFloat() : 121343421b3dd70af5b70e71816521f37502c397cc65Dale Johannesen Op->getValueAPF().convertToDouble(); 1214f0443c1eb44d737d9bd78962932fc80f74c6113cDaniel Dunbar switch (Name[0]) { 1215c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner case 'a': 121621646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier if (Name == "acos" && TLI->has(LibFunc::acos)) 12177b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(acos, V, Ty); 121821646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier else if (Name == "asin" && TLI->has(LibFunc::asin)) 12197b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(asin, V, Ty); 122021646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier else if (Name == "atan" && TLI->has(LibFunc::atan)) 12217b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(atan, V, Ty); 1222c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner break; 1223c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner case 'c': 122421646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier if (Name == "ceil" && TLI->has(LibFunc::ceil)) 12257b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(ceil, V, Ty); 122621646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier else if (Name == "cos" && TLI->has(LibFunc::cos)) 12277b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(cos, V, Ty); 122821646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier else if (Name == "cosh" && TLI->has(LibFunc::cosh)) 12297b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(cosh, V, Ty); 123021646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier else if (Name == "cosf" && TLI->has(LibFunc::cosf)) 12317b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(cos, V, Ty); 1232c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner break; 1233c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner case 'e': 123421646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier if (Name == "exp" && TLI->has(LibFunc::exp)) 12357b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(exp, V, Ty); 1236805fa97a0f5ff61624855da37d4881708fb8f0fbChris Lattner 123721646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier if (Name == "exp2" && TLI->has(LibFunc::exp2)) { 1238805fa97a0f5ff61624855da37d4881708fb8f0fbChris Lattner // Constant fold exp2(x) as pow(2,x) in case the host doesn't have a 1239805fa97a0f5ff61624855da37d4881708fb8f0fbChris Lattner // C99 library. 1240805fa97a0f5ff61624855da37d4881708fb8f0fbChris Lattner return ConstantFoldBinaryFP(pow, 2.0, V, Ty); 1241805fa97a0f5ff61624855da37d4881708fb8f0fbChris Lattner } 1242c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner break; 1243c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner case 'f': 124421646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier if (Name == "fabs" && TLI->has(LibFunc::fabs)) 12457b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(fabs, V, Ty); 124621646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier else if (Name == "floor" && TLI->has(LibFunc::floor)) 12477b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(floor, V, Ty); 1248c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner break; 1249c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner case 'l': 125021646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier if (Name == "log" && V > 0 && TLI->has(LibFunc::log)) 12517b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(log, V, Ty); 125221646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier else if (Name == "log10" && V > 0 && TLI->has(LibFunc::log10)) 12537b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(log10, V, Ty); 1254f4db877cff62340aaf9f29d3994562ae1cee8d72Chandler Carruth else if (F->getIntrinsicID() == Intrinsic::sqrt && 1255f4db877cff62340aaf9f29d3994562ae1cee8d72Chandler Carruth (Ty->isFloatTy() || Ty->isDoubleTy())) { 1256c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner if (V >= -0.0) 12577b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(sqrt, V, Ty); 1258c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner else // Undefined 1259a7235ea7245028a0723e8ab7fd011386b3900777Owen Anderson return Constant::getNullValue(Ty); 1260c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner } 1261c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner break; 1262c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner case 's': 126321646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier if (Name == "sin" && TLI->has(LibFunc::sin)) 12647b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(sin, V, Ty); 126521646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier else if (Name == "sinh" && TLI->has(LibFunc::sinh)) 12667b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(sinh, V, Ty); 126721646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier else if (Name == "sqrt" && V >= 0 && TLI->has(LibFunc::sqrt)) 12687b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(sqrt, V, Ty); 126921646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier else if (Name == "sqrtf" && V >= 0 && TLI->has(LibFunc::sqrtf)) 12707b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(sqrt, V, Ty); 127121646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier else if (Name == "sinf" && TLI->has(LibFunc::sinf)) 12727b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(sin, V, Ty); 1273c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner break; 1274c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner case 't': 127521646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier if (Name == "tan" && TLI->has(LibFunc::tan)) 12767b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(tan, V, Ty); 127721646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier else if (Name == "tanh" && TLI->has(LibFunc::tanh)) 12787b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldFP(tanh, V, Ty); 1279c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner break; 1280c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner default: 1281c5f6a1f9d61d74017d90e149728cb3d283e0a0e0Chris Lattner break; 1282bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell } 128368a0603dc5d3d79cd4de2195e950af26c3380cd7Chris Lattner return 0; 128468a0603dc5d3d79cd4de2195e950af26c3380cd7Chris Lattner } 1285f4db877cff62340aaf9f29d3994562ae1cee8d72Chandler Carruth 128668a0603dc5d3d79cd4de2195e950af26c3380cd7Chris Lattner if (ConstantInt *Op = dyn_cast<ConstantInt>(Operands[0])) { 1287f4db877cff62340aaf9f29d3994562ae1cee8d72Chandler Carruth switch (F->getIntrinsicID()) { 1288f4db877cff62340aaf9f29d3994562ae1cee8d72Chandler Carruth case Intrinsic::bswap: 12897b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantInt::get(F->getContext(), Op->getValue().byteSwap()); 1290f4db877cff62340aaf9f29d3994562ae1cee8d72Chandler Carruth case Intrinsic::ctpop: 1291eed707b1e6097aac2bb6b3d47271f6300ace7f2eOwen Anderson return ConstantInt::get(Ty, Op->getValue().countPopulation()); 1292f4db877cff62340aaf9f29d3994562ae1cee8d72Chandler Carruth case Intrinsic::convert_from_fp16: { 1293faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov APFloat Val(Op->getValue()); 1294faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov 1295faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov bool lost = false; 1296faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov APFloat::opStatus status = 1297faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov Val.convert(APFloat::IEEEsingle, APFloat::rmNearestTiesToEven, &lost); 1298faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov 1299faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov // Conversion is always precise. 13008e68c3873549ca31533e2e3e40dda3a43cb79566Jeffrey Yasskin (void)status; 1301faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov assert(status == APFloat::opOK && !lost && 1302faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov "Precision lost during fp16 constfolding"); 1303faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov 1304faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov return ConstantFP::get(F->getContext(), Val); 1305faa75f6e47350f491a44d5e8773acd9457185d51Anton Korobeynikov } 1306f4db877cff62340aaf9f29d3994562ae1cee8d72Chandler Carruth default: 1307f4db877cff62340aaf9f29d3994562ae1cee8d72Chandler Carruth return 0; 1308f4db877cff62340aaf9f29d3994562ae1cee8d72Chandler Carruth } 1309bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell } 1310f4db877cff62340aaf9f29d3994562ae1cee8d72Chandler Carruth 13116b0dc92043ab1f63d78b8796098575e1d777b701Chris Lattner // Support ConstantVector in case we have an Undef in the top. 13126b0dc92043ab1f63d78b8796098575e1d777b701Chris Lattner if (isa<ConstantVector>(Operands[0]) || 13136b0dc92043ab1f63d78b8796098575e1d777b701Chris Lattner isa<ConstantDataVector>(Operands[0])) { 13146b0dc92043ab1f63d78b8796098575e1d777b701Chris Lattner Constant *Op = cast<Constant>(Operands[0]); 131515ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth switch (F->getIntrinsicID()) { 131615ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth default: break; 131715ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth case Intrinsic::x86_sse_cvtss2si: 131815ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth case Intrinsic::x86_sse_cvtss2si64: 131915ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth case Intrinsic::x86_sse2_cvtsd2si: 132015ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth case Intrinsic::x86_sse2_cvtsd2si64: 13216b0dc92043ab1f63d78b8796098575e1d777b701Chris Lattner if (ConstantFP *FPOp = 13226b0dc92043ab1f63d78b8796098575e1d777b701Chris Lattner dyn_cast_or_null<ConstantFP>(Op->getAggregateElement(0U))) 13236b0dc92043ab1f63d78b8796098575e1d777b701Chris Lattner return ConstantFoldConvertToInt(FPOp->getValueAPF(), 13246b0dc92043ab1f63d78b8796098575e1d777b701Chris Lattner /*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: 13296b0dc92043ab1f63d78b8796098575e1d777b701Chris Lattner if (ConstantFP *FPOp = 13306b0dc92043ab1f63d78b8796098575e1d777b701Chris Lattner dyn_cast_or_null<ConstantFP>(Op->getAggregateElement(0U))) 13316b0dc92043ab1f63d78b8796098575e1d777b701Chris Lattner return ConstantFoldConvertToInt(FPOp->getValueAPF(), 13326b0dc92043ab1f63d78b8796098575e1d777b701Chris Lattner /*roundTowardZero=*/true, Ty); 133315ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth } 133415ed90c859e5df11112c614c83d0d2e786d4c73aChandler Carruth } 13356b0dc92043ab1f63d78b8796098575e1d777b701Chris Lattner 13369ee7123ab462d97a2b242280e9bf0b47883531cbDan Gohman if (isa<UndefValue>(Operands[0])) { 1337f4db877cff62340aaf9f29d3994562ae1cee8d72Chandler Carruth if (F->getIntrinsicID() == Intrinsic::bswap) 13389ee7123ab462d97a2b242280e9bf0b47883531cbDan Gohman return Operands[0]; 13399ee7123ab462d97a2b242280e9bf0b47883531cbDan Gohman return 0; 13409ee7123ab462d97a2b242280e9bf0b47883531cbDan Gohman } 13419ee7123ab462d97a2b242280e9bf0b47883531cbDan Gohman 134268a0603dc5d3d79cd4de2195e950af26c3380cd7Chris Lattner return 0; 134368a0603dc5d3d79cd4de2195e950af26c3380cd7Chris Lattner } 1344f4db877cff62340aaf9f29d3994562ae1cee8d72Chandler Carruth 13451d2f569c3428d70d0cf690c9adb459ad4a3ecff2Jay Foad if (Operands.size() == 2) { 1346bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell if (ConstantFP *Op1 = dyn_cast<ConstantFP>(Operands[0])) { 1347d0806a1519f3bc450d2dea3e6f22db73fc797e4cChris Lattner if (!Ty->isFloatTy() && !Ty->isDoubleTy()) 13489ab7fb3ba47442d521a5bed09a27a5e8e7a786edDale Johannesen return 0; 1349d0806a1519f3bc450d2dea3e6f22db73fc797e4cChris Lattner double Op1V = Ty->isFloatTy() ? 1350d0806a1519f3bc450d2dea3e6f22db73fc797e4cChris Lattner (double)Op1->getValueAPF().convertToFloat() : 135143421b3dd70af5b70e71816521f37502c397cc65Dale Johannesen Op1->getValueAPF().convertToDouble(); 1352bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell if (ConstantFP *Op2 = dyn_cast<ConstantFP>(Operands[1])) { 1353d0806a1519f3bc450d2dea3e6f22db73fc797e4cChris Lattner if (Op2->getType() != Op1->getType()) 1354d0806a1519f3bc450d2dea3e6f22db73fc797e4cChris Lattner return 0; 135521646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier 1356d0806a1519f3bc450d2dea3e6f22db73fc797e4cChris Lattner double Op2V = Ty->isFloatTy() ? 135743421b3dd70af5b70e71816521f37502c397cc65Dale Johannesen (double)Op2->getValueAPF().convertToFloat(): 135843421b3dd70af5b70e71816521f37502c397cc65Dale Johannesen Op2->getValueAPF().convertToDouble(); 1359bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell 136024fbf2bf161b20b13d0a218345bdcd092bdd56f8Chad Rosier if (F->getIntrinsicID() == Intrinsic::pow) { 136124fbf2bf161b20b13d0a218345bdcd092bdd56f8Chad Rosier return ConstantFoldBinaryFP(pow, Op1V, Op2V, Ty); 136224fbf2bf161b20b13d0a218345bdcd092bdd56f8Chad Rosier } 136324fbf2bf161b20b13d0a218345bdcd092bdd56f8Chad Rosier if (!TLI) 136424fbf2bf161b20b13d0a218345bdcd092bdd56f8Chad Rosier return 0; 136521646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier if (Name == "pow" && TLI->has(LibFunc::pow)) 13667b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldBinaryFP(pow, Op1V, Op2V, Ty); 136721646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier if (Name == "fmod" && TLI->has(LibFunc::fmod)) 13687b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldBinaryFP(fmod, Op1V, Op2V, Ty); 136921646e8becaa13cb813490edcd7fe9150b11f88eChad Rosier if (Name == "atan2" && TLI->has(LibFunc::atan2)) 13707b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFoldBinaryFP(atan2, Op1V, Op2V, Ty); 1371b5282dcf4745be59046f440980a1c0f0a50c9c09Chris Lattner } else if (ConstantInt *Op2C = dyn_cast<ConstantInt>(Operands[1])) { 1372f4db877cff62340aaf9f29d3994562ae1cee8d72Chandler Carruth if (F->getIntrinsicID() == Intrinsic::powi && Ty->isFloatTy()) 13737b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFP::get(F->getContext(), 13747b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner APFloat((float)std::pow((float)Op1V, 137502a260aa11a2e1b2c14335274d3c42ca3f3eabc0Chris Lattner (int)Op2C->getZExtValue()))); 1376f4db877cff62340aaf9f29d3994562ae1cee8d72Chandler Carruth if (F->getIntrinsicID() == Intrinsic::powi && Ty->isDoubleTy()) 13777b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner return ConstantFP::get(F->getContext(), 13787b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner APFloat((double)std::pow((double)Op1V, 13797b550ccfc5a3346c17e0390a59e2d6d19bc52705Chris Lattner (int)Op2C->getZExtValue()))); 1380bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell } 138168a0603dc5d3d79cd4de2195e950af26c3380cd7Chris Lattner return 0; 1382bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell } 1383e65cd40ab99c3bd43dc997873e500643d012bbadChris Lattner 1384e65cd40ab99c3bd43dc997873e500643d012bbadChris Lattner if (ConstantInt *Op1 = dyn_cast<ConstantInt>(Operands[0])) { 1385e65cd40ab99c3bd43dc997873e500643d012bbadChris Lattner if (ConstantInt *Op2 = dyn_cast<ConstantInt>(Operands[1])) { 1386e65cd40ab99c3bd43dc997873e500643d012bbadChris Lattner switch (F->getIntrinsicID()) { 1387e65cd40ab99c3bd43dc997873e500643d012bbadChris Lattner default: break; 1388eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner case Intrinsic::sadd_with_overflow: 1389eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner case Intrinsic::uadd_with_overflow: 1390eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner case Intrinsic::ssub_with_overflow: 1391eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner case Intrinsic::usub_with_overflow: 13926208610fd602ebdb18bb793152899573d0b2b7abFrits van Bommel case Intrinsic::smul_with_overflow: 13936208610fd602ebdb18bb793152899573d0b2b7abFrits van Bommel case Intrinsic::umul_with_overflow: { 1394eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner APInt Res; 1395eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner bool Overflow; 1396eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner switch (F->getIntrinsicID()) { 1397eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner default: assert(0 && "Invalid case"); 1398eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner case Intrinsic::sadd_with_overflow: 1399eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner Res = Op1->getValue().sadd_ov(Op2->getValue(), Overflow); 1400eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner break; 1401eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner case Intrinsic::uadd_with_overflow: 1402eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner Res = Op1->getValue().uadd_ov(Op2->getValue(), Overflow); 1403eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner break; 1404eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner case Intrinsic::ssub_with_overflow: 1405eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner Res = Op1->getValue().ssub_ov(Op2->getValue(), Overflow); 1406eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner break; 1407eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner case Intrinsic::usub_with_overflow: 1408eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner Res = Op1->getValue().usub_ov(Op2->getValue(), Overflow); 1409eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner break; 1410eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner case Intrinsic::smul_with_overflow: 1411eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner Res = Op1->getValue().smul_ov(Op2->getValue(), Overflow); 1412eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner break; 14136208610fd602ebdb18bb793152899573d0b2b7abFrits van Bommel case Intrinsic::umul_with_overflow: 14146208610fd602ebdb18bb793152899573d0b2b7abFrits van Bommel Res = Op1->getValue().umul_ov(Op2->getValue(), Overflow); 14156208610fd602ebdb18bb793152899573d0b2b7abFrits van Bommel break; 1416eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner } 1417e65cd40ab99c3bd43dc997873e500643d012bbadChris Lattner Constant *Ops[] = { 1418eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner ConstantInt::get(F->getContext(), Res), 1419eafc5cb80d58cb9447623a557be4d4f55f42fbc3Chris Lattner ConstantInt::get(Type::getInt1Ty(F->getContext()), Overflow) 1420e65cd40ab99c3bd43dc997873e500643d012bbadChris Lattner }; 1421b065b06c12dba6001b8140df2744d0c856ef6ea1Chris Lattner return ConstantStruct::get(cast<StructType>(F->getReturnType()), Ops); 1422e65cd40ab99c3bd43dc997873e500643d012bbadChris Lattner } 1423ccbf1e36d34c2d66600748bdd8767dc122629a58Chandler Carruth case Intrinsic::cttz: 1424ccbf1e36d34c2d66600748bdd8767dc122629a58Chandler Carruth // FIXME: This should check for Op2 == 1, and become unreachable if 1425ccbf1e36d34c2d66600748bdd8767dc122629a58Chandler Carruth // Op1 == 0. 1426ccbf1e36d34c2d66600748bdd8767dc122629a58Chandler Carruth return ConstantInt::get(Ty, Op1->getValue().countTrailingZeros()); 1427ccbf1e36d34c2d66600748bdd8767dc122629a58Chandler Carruth case Intrinsic::ctlz: 1428ccbf1e36d34c2d66600748bdd8767dc122629a58Chandler Carruth // FIXME: This should check for Op2 == 1, and become unreachable if 1429ccbf1e36d34c2d66600748bdd8767dc122629a58Chandler Carruth // Op1 == 0. 1430ccbf1e36d34c2d66600748bdd8767dc122629a58Chandler Carruth return ConstantInt::get(Ty, Op1->getValue().countLeadingZeros()); 1431e65cd40ab99c3bd43dc997873e500643d012bbadChris Lattner } 1432e65cd40ab99c3bd43dc997873e500643d012bbadChris Lattner } 1433e65cd40ab99c3bd43dc997873e500643d012bbadChris Lattner 1434e65cd40ab99c3bd43dc997873e500643d012bbadChris Lattner return 0; 1435e65cd40ab99c3bd43dc997873e500643d012bbadChris Lattner } 1436e65cd40ab99c3bd43dc997873e500643d012bbadChris Lattner return 0; 1437bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell } 1438bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell return 0; 1439bd9d37026a5c17d9a51371a6a5446bf4761ee7d6John Criswell} 1440