1753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner//===- InstCombineCalls.cpp -----------------------------------------------===// 2753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner// 3753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner// The LLVM Compiler Infrastructure 4753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner// 5753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner// This file is distributed under the University of Illinois Open Source 6753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner// License. See LICENSE.TXT for details. 7753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner// 8753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner//===----------------------------------------------------------------------===// 9753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner// 10753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner// This file implements the visitCall and visitInvoke functions. 11753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner// 12753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner//===----------------------------------------------------------------------===// 13753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 14753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner#include "InstCombine.h" 1563f932ca3c91ea8ac5b592158f5e8ef7de550547Meador Inge#include "llvm/ADT/Statistic.h" 16753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner#include "llvm/Analysis/MemoryBuiltins.h" 1736b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines#include "llvm/IR/CallSite.h" 180b8c9a80f20772c3793201ab5b251d3520b9cea3Chandler Carruth#include "llvm/IR/DataLayout.h" 1936b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines#include "llvm/IR/PatternMatch.h" 2027ceaa13f820e33c44e29adc2095dd3e1e69d09fEric Christopher#include "llvm/Transforms/Utils/BuildLibCalls.h" 21687140c818ba4b896329a83324714140b6580ef8Chris Lattner#include "llvm/Transforms/Utils/Local.h" 22753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattnerusing namespace llvm; 238ad435fa4809b5a4ad1043435cbafd5c9ddf2d93Michael Ilsemanusing namespace PatternMatch; 24753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 25dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines#define DEBUG_TYPE "instcombine" 26dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines 2763f932ca3c91ea8ac5b592158f5e8ef7de550547Meador IngeSTATISTIC(NumSimplified, "Number of library calls simplified"); 2863f932ca3c91ea8ac5b592158f5e8ef7de550547Meador Inge 29753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner/// getPromotedType - Return the specified type promoted as it would be to pass 30753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner/// though a va_arg area. 31db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattnerstatic Type *getPromotedType(Type *Ty) { 32db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner if (IntegerType* ITy = dyn_cast<IntegerType>(Ty)) { 33753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (ITy->getBitWidth() < 32) 34753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return Type::getInt32Ty(Ty->getContext()); 35753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 36753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return Ty; 37753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner} 38753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 39ce52bc53538df8e5412ec507f2da3661c991baf1Dan Gohman/// reduceToSingleValueType - Given an aggregate type which ultimately holds a 40ce52bc53538df8e5412ec507f2da3661c991baf1Dan Gohman/// single scalar element, like {{{type}}} or [1 x type], return type. 41ce52bc53538df8e5412ec507f2da3661c991baf1Dan Gohmanstatic Type *reduceToSingleValueType(Type *T) { 42ce52bc53538df8e5412ec507f2da3661c991baf1Dan Gohman while (!T->isSingleValueType()) { 43ce52bc53538df8e5412ec507f2da3661c991baf1Dan Gohman if (StructType *STy = dyn_cast<StructType>(T)) { 44ce52bc53538df8e5412ec507f2da3661c991baf1Dan Gohman if (STy->getNumElements() == 1) 45ce52bc53538df8e5412ec507f2da3661c991baf1Dan Gohman T = STy->getElementType(0); 46ce52bc53538df8e5412ec507f2da3661c991baf1Dan Gohman else 47ce52bc53538df8e5412ec507f2da3661c991baf1Dan Gohman break; 48ce52bc53538df8e5412ec507f2da3661c991baf1Dan Gohman } else if (ArrayType *ATy = dyn_cast<ArrayType>(T)) { 49ce52bc53538df8e5412ec507f2da3661c991baf1Dan Gohman if (ATy->getNumElements() == 1) 50ce52bc53538df8e5412ec507f2da3661c991baf1Dan Gohman T = ATy->getElementType(); 51ce52bc53538df8e5412ec507f2da3661c991baf1Dan Gohman else 52ce52bc53538df8e5412ec507f2da3661c991baf1Dan Gohman break; 53ce52bc53538df8e5412ec507f2da3661c991baf1Dan Gohman } else 54ce52bc53538df8e5412ec507f2da3661c991baf1Dan Gohman break; 55ce52bc53538df8e5412ec507f2da3661c991baf1Dan Gohman } 56ce52bc53538df8e5412ec507f2da3661c991baf1Dan Gohman 57ce52bc53538df8e5412ec507f2da3661c991baf1Dan Gohman return T; 58ce52bc53538df8e5412ec507f2da3661c991baf1Dan Gohman} 59753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 60753a2b464d0628231bd3dce645edd05dfbaa3a06Chris LattnerInstruction *InstCombiner::SimplifyMemTransfer(MemIntrinsic *MI) { 6136b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines unsigned DstAlign = getKnownAlignment(MI->getArgOperand(0), DL); 6236b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines unsigned SrcAlign = getKnownAlignment(MI->getArgOperand(1), DL); 63753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner unsigned MinAlign = std::min(DstAlign, SrcAlign); 64753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner unsigned CopyAlign = MI->getAlignment(); 65753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 66753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (CopyAlign < MinAlign) { 6700e403abe384b8f61dd012242d3d81a23541ca25Jim Grosbach MI->setAlignment(ConstantInt::get(MI->getAlignmentType(), 68753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner MinAlign, false)); 69753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return MI; 70753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 7100e403abe384b8f61dd012242d3d81a23541ca25Jim Grosbach 72753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If MemCpyInst length is 1/2/4/8 bytes then replace memcpy with 73753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // load/store. 74bcda85c74304ffada586a9c58ce9bcff93b69a86Gabor Greif ConstantInt *MemOpLength = dyn_cast<ConstantInt>(MI->getArgOperand(2)); 75dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines if (!MemOpLength) return nullptr; 7600e403abe384b8f61dd012242d3d81a23541ca25Jim Grosbach 77753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Source and destination pointer types are always "i8*" for intrinsic. See 78753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // if the size is something we can handle with a single primitive load/store. 79753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // A single load+store correctly handles overlapping memory in the memmove 80753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // case. 819441ad0b6e02ee9558168b37a9620ef1afe79388Michael Liao uint64_t Size = MemOpLength->getLimitedValue(); 8236b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines assert(Size && "0-sized memory transferring should be removed already."); 8300e403abe384b8f61dd012242d3d81a23541ca25Jim Grosbach 84753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Size > 8 || (Size&(Size-1))) 85dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return nullptr; // If not 1/2/4/8 bytes, exit. 8600e403abe384b8f61dd012242d3d81a23541ca25Jim Grosbach 87753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Use an integer load+store unless we can find something better. 8820adc9dc4650313f017b27d9818eb2176238113dMon P Wang unsigned SrcAddrSp = 89bcda85c74304ffada586a9c58ce9bcff93b69a86Gabor Greif cast<PointerType>(MI->getArgOperand(1)->getType())->getAddressSpace(); 904ec2258ffb495d7ce00177e447740ef1123a27dbGabor Greif unsigned DstAddrSp = 91bcda85c74304ffada586a9c58ce9bcff93b69a86Gabor Greif cast<PointerType>(MI->getArgOperand(0)->getType())->getAddressSpace(); 9220adc9dc4650313f017b27d9818eb2176238113dMon P Wang 93db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner IntegerType* IntType = IntegerType::get(MI->getContext(), Size<<3); 9420adc9dc4650313f017b27d9818eb2176238113dMon P Wang Type *NewSrcPtrTy = PointerType::get(IntType, SrcAddrSp); 9520adc9dc4650313f017b27d9818eb2176238113dMon P Wang Type *NewDstPtrTy = PointerType::get(IntType, DstAddrSp); 9600e403abe384b8f61dd012242d3d81a23541ca25Jim Grosbach 97753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Memcpy forces the use of i8* for the source and destination. That means 98753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // that if you're using memcpy to move one double around, you'll get a cast 99753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // from double* to i8*. We'd much rather use a double load+store rather than 100753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // an i64 load+store, here because this improves the odds that the source or 101753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // dest address will be promotable. See if we can find a better type than the 102753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // integer datatype. 103cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif Value *StrippedDest = MI->getArgOperand(0)->stripPointerCasts(); 104dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines MDNode *CopyMD = nullptr; 105cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif if (StrippedDest != MI->getArgOperand(0)) { 106db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *SrcETy = cast<PointerType>(StrippedDest->getType()) 107753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ->getElementType(); 10836b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines if (DL && SrcETy->isSized() && DL->getTypeStoreSize(SrcETy) == Size) { 109753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // The SrcETy might be something like {{{double}}} or [1 x double]. Rip 110753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // down through these levels if so. 111ce52bc53538df8e5412ec507f2da3661c991baf1Dan Gohman SrcETy = reduceToSingleValueType(SrcETy); 11200e403abe384b8f61dd012242d3d81a23541ca25Jim Grosbach 11320adc9dc4650313f017b27d9818eb2176238113dMon P Wang if (SrcETy->isSingleValueType()) { 11420adc9dc4650313f017b27d9818eb2176238113dMon P Wang NewSrcPtrTy = PointerType::get(SrcETy, SrcAddrSp); 11520adc9dc4650313f017b27d9818eb2176238113dMon P Wang NewDstPtrTy = PointerType::get(SrcETy, DstAddrSp); 116b998913ff4aa58c3c342e167c785029cb331078eDan Gohman 117b998913ff4aa58c3c342e167c785029cb331078eDan Gohman // If the memcpy has metadata describing the members, see if we can 118b998913ff4aa58c3c342e167c785029cb331078eDan Gohman // get the TBAA tag describing our copy. 119b998913ff4aa58c3c342e167c785029cb331078eDan Gohman if (MDNode *M = MI->getMetadata(LLVMContext::MD_tbaa_struct)) { 120b998913ff4aa58c3c342e167c785029cb331078eDan Gohman if (M->getNumOperands() == 3 && 1215e01f80bf85b9a68352d4c146caa9ddcf6af6dcfNick Lewycky M->getOperand(0) && 122b998913ff4aa58c3c342e167c785029cb331078eDan Gohman isa<ConstantInt>(M->getOperand(0)) && 123b998913ff4aa58c3c342e167c785029cb331078eDan Gohman cast<ConstantInt>(M->getOperand(0))->isNullValue() && 1245e01f80bf85b9a68352d4c146caa9ddcf6af6dcfNick Lewycky M->getOperand(1) && 125b998913ff4aa58c3c342e167c785029cb331078eDan Gohman isa<ConstantInt>(M->getOperand(1)) && 126b998913ff4aa58c3c342e167c785029cb331078eDan Gohman cast<ConstantInt>(M->getOperand(1))->getValue() == Size && 1275e01f80bf85b9a68352d4c146caa9ddcf6af6dcfNick Lewycky M->getOperand(2) && 128b998913ff4aa58c3c342e167c785029cb331078eDan Gohman isa<MDNode>(M->getOperand(2))) 129b998913ff4aa58c3c342e167c785029cb331078eDan Gohman CopyMD = cast<MDNode>(M->getOperand(2)); 130b998913ff4aa58c3c342e167c785029cb331078eDan Gohman } 13120adc9dc4650313f017b27d9818eb2176238113dMon P Wang } 132753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 133753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 13400e403abe384b8f61dd012242d3d81a23541ca25Jim Grosbach 135753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If the memcpy/memmove provides better alignment info than we can 136753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // infer, use it. 137753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner SrcAlign = std::max(SrcAlign, CopyAlign); 138753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner DstAlign = std::max(DstAlign, CopyAlign); 13900e403abe384b8f61dd012242d3d81a23541ca25Jim Grosbach 1409c68a7bb45ebf71cec5ab4eab7c73aacdfce16dfGabor Greif Value *Src = Builder->CreateBitCast(MI->getArgOperand(1), NewSrcPtrTy); 1419c68a7bb45ebf71cec5ab4eab7c73aacdfce16dfGabor Greif Value *Dest = Builder->CreateBitCast(MI->getArgOperand(0), NewDstPtrTy); 14259f15918c28ad340bb9f0ef78e43c2952a5b5c1cEli Friedman LoadInst *L = Builder->CreateLoad(Src, MI->isVolatile()); 14359f15918c28ad340bb9f0ef78e43c2952a5b5c1cEli Friedman L->setAlignment(SrcAlign); 144b998913ff4aa58c3c342e167c785029cb331078eDan Gohman if (CopyMD) 145b998913ff4aa58c3c342e167c785029cb331078eDan Gohman L->setMetadata(LLVMContext::MD_tbaa, CopyMD); 14659f15918c28ad340bb9f0ef78e43c2952a5b5c1cEli Friedman StoreInst *S = Builder->CreateStore(L, Dest, MI->isVolatile()); 14759f15918c28ad340bb9f0ef78e43c2952a5b5c1cEli Friedman S->setAlignment(DstAlign); 148b998913ff4aa58c3c342e167c785029cb331078eDan Gohman if (CopyMD) 149b998913ff4aa58c3c342e167c785029cb331078eDan Gohman S->setMetadata(LLVMContext::MD_tbaa, CopyMD); 150753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 151753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Set the size of the copy to 0, it will be deleted on the next iteration. 152a90c5c7605e0d8df96e61d4bf0cc47a7eaea92cfGabor Greif MI->setArgOperand(2, Constant::getNullValue(MemOpLength->getType())); 153753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return MI; 154753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner} 155753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 156753a2b464d0628231bd3dce645edd05dfbaa3a06Chris LattnerInstruction *InstCombiner::SimplifyMemSet(MemSetInst *MI) { 15736b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines unsigned Alignment = getKnownAlignment(MI->getDest(), DL); 158753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (MI->getAlignment() < Alignment) { 159753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner MI->setAlignment(ConstantInt::get(MI->getAlignmentType(), 160753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Alignment, false)); 161753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return MI; 162753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 16300e403abe384b8f61dd012242d3d81a23541ca25Jim Grosbach 164753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Extract the length and alignment and fill if they are constant. 165753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ConstantInt *LenC = dyn_cast<ConstantInt>(MI->getLength()); 166753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ConstantInt *FillC = dyn_cast<ConstantInt>(MI->getValue()); 167b0bc6c361da9009e8414efde317d9bbff755f6c0Duncan Sands if (!LenC || !FillC || !FillC->getType()->isIntegerTy(8)) 168dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return nullptr; 1699441ad0b6e02ee9558168b37a9620ef1afe79388Michael Liao uint64_t Len = LenC->getLimitedValue(); 170753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Alignment = MI->getAlignment(); 1719441ad0b6e02ee9558168b37a9620ef1afe79388Michael Liao assert(Len && "0-sized memory setting should be removed already."); 17200e403abe384b8f61dd012242d3d81a23541ca25Jim Grosbach 173753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // memset(s,c,n) -> store s, c (for n=1,2,4,8) 174753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Len <= 8 && isPowerOf2_32((uint32_t)Len)) { 175db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *ITy = IntegerType::get(MI->getContext(), Len*8); // n=1 -> i8. 17600e403abe384b8f61dd012242d3d81a23541ca25Jim Grosbach 177753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Value *Dest = MI->getDest(); 17855fb9b0fcf89501a0abe74808f4e08e0b3a17207Mon P Wang unsigned DstAddrSp = cast<PointerType>(Dest->getType())->getAddressSpace(); 17955fb9b0fcf89501a0abe74808f4e08e0b3a17207Mon P Wang Type *NewDstPtrTy = PointerType::get(ITy, DstAddrSp); 18055fb9b0fcf89501a0abe74808f4e08e0b3a17207Mon P Wang Dest = Builder->CreateBitCast(Dest, NewDstPtrTy); 181753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 182753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Alignment 0 is identity for alignment 1 for memset, but not store. 183753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Alignment == 0) Alignment = 1; 18400e403abe384b8f61dd012242d3d81a23541ca25Jim Grosbach 185753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Extract the fill value and store. 186753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner uint64_t Fill = FillC->getZExtValue()*0x0101010101010101ULL; 18759f15918c28ad340bb9f0ef78e43c2952a5b5c1cEli Friedman StoreInst *S = Builder->CreateStore(ConstantInt::get(ITy, Fill), Dest, 18859f15918c28ad340bb9f0ef78e43c2952a5b5c1cEli Friedman MI->isVolatile()); 18959f15918c28ad340bb9f0ef78e43c2952a5b5c1cEli Friedman S->setAlignment(Alignment); 19000e403abe384b8f61dd012242d3d81a23541ca25Jim Grosbach 191753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Set the size of the copy to 0, it will be deleted on the next iteration. 192753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner MI->setLength(Constant::getNullValue(LenC->getType())); 193753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return MI; 194753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 195753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 196dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return nullptr; 197753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner} 198753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 19900e403abe384b8f61dd012242d3d81a23541ca25Jim Grosbach/// visitCallInst - CallInst simplification. This mostly only handles folding 200753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner/// of intrinsic instructions. For normal calls, it allows visitCallSite to do 201753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner/// the heavy lifting. 202753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner/// 203753a2b464d0628231bd3dce645edd05dfbaa3a06Chris LattnerInstruction *InstCombiner::visitCallInst(CallInst &CI) { 2048e0d1c03ca7fd86e6879b4e37d0d7f0e982feef6Benjamin Kramer if (isFreeCall(&CI, TLI)) 205753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return visitFree(CI); 206753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 207753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If the caller function is nounwind, mark the call as nounwind, even if the 208753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // callee isn't. 209753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (CI.getParent()->getParent()->doesNotThrow() && 210753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner !CI.doesNotThrow()) { 211753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner CI.setDoesNotThrow(); 212753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return &CI; 213753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 21400e403abe384b8f61dd012242d3d81a23541ca25Jim Grosbach 215753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner IntrinsicInst *II = dyn_cast<IntrinsicInst>(&CI); 216753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!II) return visitCallSite(&CI); 217cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif 218753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Intrinsics cannot occur in an invoke, so handle them here instead of in 219753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // visitCallSite. 220753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (MemIntrinsic *MI = dyn_cast<MemIntrinsic>(II)) { 221753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner bool Changed = false; 222753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 223753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // memmove/cpy/set of zero bytes is a noop. 224753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Constant *NumBytes = dyn_cast<Constant>(MI->getLength())) { 2256eff75104e660bae5b28963190a6940eeecb7484Chris Lattner if (NumBytes->isNullValue()) 2266eff75104e660bae5b28963190a6940eeecb7484Chris Lattner return EraseInstFromFunction(CI); 227753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 228753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (ConstantInt *CI = dyn_cast<ConstantInt>(NumBytes)) 229753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (CI->getZExtValue() == 1) { 230753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Replace the instruction with just byte operations. We would 231753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // transform other cases to loads/stores, but we don't know if 232753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // alignment is sufficient. 233753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 234753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 23500e403abe384b8f61dd012242d3d81a23541ca25Jim Grosbach 2366eff75104e660bae5b28963190a6940eeecb7484Chris Lattner // No other transformations apply to volatile transfers. 2376eff75104e660bae5b28963190a6940eeecb7484Chris Lattner if (MI->isVolatile()) 238dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return nullptr; 239753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 240753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If we have a memmove and the source operation is a constant global, 241753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // then the source and dest pointers can't alias, so we can change this 242753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // into a call to memcpy. 243753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (MemMoveInst *MMI = dyn_cast<MemMoveInst>(MI)) { 244753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (GlobalVariable *GVSrc = dyn_cast<GlobalVariable>(MMI->getSource())) 245753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (GVSrc->isConstant()) { 246551754c4958086cc6910da7c950f2875e212f5cfEric Christopher Module *M = CI.getParent()->getParent()->getParent(); 247753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Intrinsic::ID MemCpyID = Intrinsic::memcpy; 2485fdd6c8793462549e3593890ec61573da06e3346Jay Foad Type *Tys[3] = { CI.getArgOperand(0)->getType(), 2495fdd6c8793462549e3593890ec61573da06e3346Jay Foad CI.getArgOperand(1)->getType(), 2505fdd6c8793462549e3593890ec61573da06e3346Jay Foad CI.getArgOperand(2)->getType() }; 251eb9a85f09e18b3fe88499710404b38d3a9128f62Benjamin Kramer CI.setCalledFunction(Intrinsic::getDeclaration(M, MemCpyID, Tys)); 252753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Changed = true; 253753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 254753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 255753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 256753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (MemTransferInst *MTI = dyn_cast<MemTransferInst>(MI)) { 257753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // memmove(x,x,size) -> noop. 258753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (MTI->getSource() == MTI->getDest()) 259753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return EraseInstFromFunction(CI); 260551754c4958086cc6910da7c950f2875e212f5cfEric Christopher } 261753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 262551754c4958086cc6910da7c950f2875e212f5cfEric Christopher // If we can determine a pointer alignment that is bigger than currently 263551754c4958086cc6910da7c950f2875e212f5cfEric Christopher // set, update the alignment. 264551754c4958086cc6910da7c950f2875e212f5cfEric Christopher if (isa<MemTransferInst>(MI)) { 265551754c4958086cc6910da7c950f2875e212f5cfEric Christopher if (Instruction *I = SimplifyMemTransfer(MI)) 266753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return I; 267753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } else if (MemSetInst *MSI = dyn_cast<MemSetInst>(MI)) { 268753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Instruction *I = SimplifyMemSet(MSI)) 269753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return I; 270753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 271c310fcc909c21c94d1e5b05a97763911568a3ed8Gabor Greif 272753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Changed) return II; 273753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 27400e403abe384b8f61dd012242d3d81a23541ca25Jim Grosbach 275753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner switch (II->getIntrinsicID()) { 276753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner default: break; 277415326b4edcc967dfb03c5ab41923b195e7c3cb1Eric Christopher case Intrinsic::objectsize: { 2789e72a79ef4a9fcda482ce0b0e1f0bd6a4f16cffdNuno Lopes uint64_t Size; 27936b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines if (getObjectSize(II->getArgOperand(0), Size, DL, TLI)) 2809e72a79ef4a9fcda482ce0b0e1f0bd6a4f16cffdNuno Lopes return ReplaceInstUsesWith(CI, ConstantInt::get(CI.getType(), Size)); 281dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return nullptr; 282415326b4edcc967dfb03c5ab41923b195e7c3cb1Eric Christopher } 2838ad435fa4809b5a4ad1043435cbafd5c9ddf2d93Michael Ilseman case Intrinsic::bswap: { 2848ad435fa4809b5a4ad1043435cbafd5c9ddf2d93Michael Ilseman Value *IIOperand = II->getArgOperand(0); 285dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines Value *X = nullptr; 2868ad435fa4809b5a4ad1043435cbafd5c9ddf2d93Michael Ilseman 287753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // bswap(bswap(x)) -> x 2888ad435fa4809b5a4ad1043435cbafd5c9ddf2d93Michael Ilseman if (match(IIOperand, m_BSwap(m_Value(X)))) 2898ad435fa4809b5a4ad1043435cbafd5c9ddf2d93Michael Ilseman return ReplaceInstUsesWith(CI, X); 29000e403abe384b8f61dd012242d3d81a23541ca25Jim Grosbach 291753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // bswap(trunc(bswap(x))) -> trunc(lshr(x, c)) 2928ad435fa4809b5a4ad1043435cbafd5c9ddf2d93Michael Ilseman if (match(IIOperand, m_Trunc(m_BSwap(m_Value(X))))) { 2938ad435fa4809b5a4ad1043435cbafd5c9ddf2d93Michael Ilseman unsigned C = X->getType()->getPrimitiveSizeInBits() - 2948ad435fa4809b5a4ad1043435cbafd5c9ddf2d93Michael Ilseman IIOperand->getType()->getPrimitiveSizeInBits(); 2958ad435fa4809b5a4ad1043435cbafd5c9ddf2d93Michael Ilseman Value *CV = ConstantInt::get(X->getType(), C); 2968ad435fa4809b5a4ad1043435cbafd5c9ddf2d93Michael Ilseman Value *V = Builder->CreateLShr(X, CV); 2978ad435fa4809b5a4ad1043435cbafd5c9ddf2d93Michael Ilseman return new TruncInst(V, IIOperand->getType()); 298753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 299753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 3008ad435fa4809b5a4ad1043435cbafd5c9ddf2d93Michael Ilseman } 3018ad435fa4809b5a4ad1043435cbafd5c9ddf2d93Michael Ilseman 302753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::powi: 303cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif if (ConstantInt *Power = dyn_cast<ConstantInt>(II->getArgOperand(1))) { 304753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // powi(x, 0) -> 1.0 305753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Power->isZero()) 306753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return ReplaceInstUsesWith(CI, ConstantFP::get(CI.getType(), 1.0)); 307753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // powi(x, 1) -> x 308753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Power->isOne()) 309cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif return ReplaceInstUsesWith(CI, II->getArgOperand(0)); 310753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // powi(x, -1) -> 1/x 311753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Power->isAllOnesValue()) 312753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return BinaryOperator::CreateFDiv(ConstantFP::get(CI.getType(), 1.0), 313cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif II->getArgOperand(0)); 314753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 315753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 316753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::cttz: { 317753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If all bits below the first known one are known zero, 318753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // this value is constant. 319db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner IntegerType *IT = dyn_cast<IntegerType>(II->getArgOperand(0)->getType()); 320f1ac465b67d5fc11a0d9cd09b98ceb4ffa75dd97Owen Anderson // FIXME: Try to simplify vectors of integers. 321f1ac465b67d5fc11a0d9cd09b98ceb4ffa75dd97Owen Anderson if (!IT) break; 322753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner uint32_t BitWidth = IT->getBitWidth(); 323753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt KnownZero(BitWidth, 0); 324753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt KnownOne(BitWidth, 0); 325dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines computeKnownBits(II->getArgOperand(0), KnownZero, KnownOne); 326753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner unsigned TrailingZeros = KnownOne.countTrailingZeros(); 327753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt Mask(APInt::getLowBitsSet(BitWidth, TrailingZeros)); 328753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if ((Mask & KnownZero) == Mask) 329753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return ReplaceInstUsesWith(CI, ConstantInt::get(IT, 330753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt(BitWidth, TrailingZeros))); 33100e403abe384b8f61dd012242d3d81a23541ca25Jim Grosbach 332753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 333753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 334753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::ctlz: { 335753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If all bits above the first known one are known zero, 336753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // this value is constant. 337db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner IntegerType *IT = dyn_cast<IntegerType>(II->getArgOperand(0)->getType()); 338f1ac465b67d5fc11a0d9cd09b98ceb4ffa75dd97Owen Anderson // FIXME: Try to simplify vectors of integers. 339f1ac465b67d5fc11a0d9cd09b98ceb4ffa75dd97Owen Anderson if (!IT) break; 340753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner uint32_t BitWidth = IT->getBitWidth(); 341753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt KnownZero(BitWidth, 0); 342753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt KnownOne(BitWidth, 0); 343dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines computeKnownBits(II->getArgOperand(0), KnownZero, KnownOne); 344753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner unsigned LeadingZeros = KnownOne.countLeadingZeros(); 345753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt Mask(APInt::getHighBitsSet(BitWidth, LeadingZeros)); 346753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if ((Mask & KnownZero) == Mask) 347753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return ReplaceInstUsesWith(CI, ConstantInt::get(IT, 348753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt(BitWidth, LeadingZeros))); 34900e403abe384b8f61dd012242d3d81a23541ca25Jim Grosbach 350753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 351753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 352753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::uadd_with_overflow: { 353cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif Value *LHS = II->getArgOperand(0), *RHS = II->getArgOperand(1); 354db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner IntegerType *IT = cast<IntegerType>(II->getArgOperand(0)->getType()); 355753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner uint32_t BitWidth = IT->getBitWidth(); 356753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt LHSKnownZero(BitWidth, 0); 357753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt LHSKnownOne(BitWidth, 0); 358dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines computeKnownBits(LHS, LHSKnownZero, LHSKnownOne); 359753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner bool LHSKnownNegative = LHSKnownOne[BitWidth - 1]; 360753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner bool LHSKnownPositive = LHSKnownZero[BitWidth - 1]; 361753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 362753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (LHSKnownNegative || LHSKnownPositive) { 363753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt RHSKnownZero(BitWidth, 0); 364753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt RHSKnownOne(BitWidth, 0); 365dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines computeKnownBits(RHS, RHSKnownZero, RHSKnownOne); 366753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner bool RHSKnownNegative = RHSKnownOne[BitWidth - 1]; 367753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner bool RHSKnownPositive = RHSKnownZero[BitWidth - 1]; 368753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (LHSKnownNegative && RHSKnownNegative) { 369753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // The sign bit is set in both cases: this MUST overflow. 370753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Create a simple add instruction, and insert it into the struct. 37159f15918c28ad340bb9f0ef78e43c2952a5b5c1cEli Friedman Value *Add = Builder->CreateAdd(LHS, RHS); 37259f15918c28ad340bb9f0ef78e43c2952a5b5c1cEli Friedman Add->takeName(&CI); 373753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Constant *V[] = { 37459f15918c28ad340bb9f0ef78e43c2952a5b5c1cEli Friedman UndefValue::get(LHS->getType()), 37559f15918c28ad340bb9f0ef78e43c2952a5b5c1cEli Friedman ConstantInt::getTrue(II->getContext()) 376753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner }; 377db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner StructType *ST = cast<StructType>(II->getType()); 378b065b06c12dba6001b8140df2744d0c856ef6ea1Chris Lattner Constant *Struct = ConstantStruct::get(ST, V); 379753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return InsertValueInst::Create(Struct, Add, 0); 380753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 38159f15918c28ad340bb9f0ef78e43c2952a5b5c1cEli Friedman 382753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (LHSKnownPositive && RHSKnownPositive) { 383753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // The sign bit is clear in both cases: this CANNOT overflow. 384753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Create a simple add instruction, and insert it into the struct. 38559f15918c28ad340bb9f0ef78e43c2952a5b5c1cEli Friedman Value *Add = Builder->CreateNUWAdd(LHS, RHS); 38659f15918c28ad340bb9f0ef78e43c2952a5b5c1cEli Friedman Add->takeName(&CI); 387753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Constant *V[] = { 388753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner UndefValue::get(LHS->getType()), 389753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ConstantInt::getFalse(II->getContext()) 390753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner }; 391db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner StructType *ST = cast<StructType>(II->getType()); 392b065b06c12dba6001b8140df2744d0c856ef6ea1Chris Lattner Constant *Struct = ConstantStruct::get(ST, V); 393753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return InsertValueInst::Create(Struct, Add, 0); 394753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 395753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 396753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 397753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // FALL THROUGH uadd into sadd 398753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::sadd_with_overflow: 399753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Canonicalize constants into the RHS. 400a90c5c7605e0d8df96e61d4bf0cc47a7eaea92cfGabor Greif if (isa<Constant>(II->getArgOperand(0)) && 401a90c5c7605e0d8df96e61d4bf0cc47a7eaea92cfGabor Greif !isa<Constant>(II->getArgOperand(1))) { 402a90c5c7605e0d8df96e61d4bf0cc47a7eaea92cfGabor Greif Value *LHS = II->getArgOperand(0); 403a90c5c7605e0d8df96e61d4bf0cc47a7eaea92cfGabor Greif II->setArgOperand(0, II->getArgOperand(1)); 404a90c5c7605e0d8df96e61d4bf0cc47a7eaea92cfGabor Greif II->setArgOperand(1, LHS); 405753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return II; 406753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 407753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 408753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // X + undef -> undef 4099c68a7bb45ebf71cec5ab4eab7c73aacdfce16dfGabor Greif if (isa<UndefValue>(II->getArgOperand(1))) 410753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return ReplaceInstUsesWith(CI, UndefValue::get(II->getType())); 41100e403abe384b8f61dd012242d3d81a23541ca25Jim Grosbach 4129c68a7bb45ebf71cec5ab4eab7c73aacdfce16dfGabor Greif if (ConstantInt *RHS = dyn_cast<ConstantInt>(II->getArgOperand(1))) { 413753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // X + 0 -> {X, false} 414753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (RHS->isZero()) { 415753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Constant *V[] = { 4164fffb345ed20bd144a31c1c1724cffb6f519b814Eli Friedman UndefValue::get(II->getArgOperand(0)->getType()), 417753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ConstantInt::getFalse(II->getContext()) 418753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner }; 419b065b06c12dba6001b8140df2744d0c856ef6ea1Chris Lattner Constant *Struct = 420b065b06c12dba6001b8140df2744d0c856ef6ea1Chris Lattner ConstantStruct::get(cast<StructType>(II->getType()), V); 421cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif return InsertValueInst::Create(Struct, II->getArgOperand(0), 0); 422753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 423753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 424cd81d94322a39503e4a3e87b6ee03d4fcb3465fbStephen Hines 425cd81d94322a39503e4a3e87b6ee03d4fcb3465fbStephen Hines // We can strength reduce reduce this signed add into a regular add if we 426cd81d94322a39503e4a3e87b6ee03d4fcb3465fbStephen Hines // can prove that it will never overflow. 427cd81d94322a39503e4a3e87b6ee03d4fcb3465fbStephen Hines if (II->getIntrinsicID() == Intrinsic::sadd_with_overflow) { 428cd81d94322a39503e4a3e87b6ee03d4fcb3465fbStephen Hines Value *LHS = II->getArgOperand(0), *RHS = II->getArgOperand(1); 429cd81d94322a39503e4a3e87b6ee03d4fcb3465fbStephen Hines if (WillNotOverflowSignedAdd(LHS, RHS)) { 430cd81d94322a39503e4a3e87b6ee03d4fcb3465fbStephen Hines Value *Add = Builder->CreateNSWAdd(LHS, RHS); 431cd81d94322a39503e4a3e87b6ee03d4fcb3465fbStephen Hines Add->takeName(&CI); 432cd81d94322a39503e4a3e87b6ee03d4fcb3465fbStephen Hines Constant *V[] = {UndefValue::get(Add->getType()), Builder->getFalse()}; 433cd81d94322a39503e4a3e87b6ee03d4fcb3465fbStephen Hines StructType *ST = cast<StructType>(II->getType()); 434cd81d94322a39503e4a3e87b6ee03d4fcb3465fbStephen Hines Constant *Struct = ConstantStruct::get(ST, V); 435cd81d94322a39503e4a3e87b6ee03d4fcb3465fbStephen Hines return InsertValueInst::Create(Struct, Add, 0); 436cd81d94322a39503e4a3e87b6ee03d4fcb3465fbStephen Hines } 437cd81d94322a39503e4a3e87b6ee03d4fcb3465fbStephen Hines } 438cd81d94322a39503e4a3e87b6ee03d4fcb3465fbStephen Hines 439753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 440753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::usub_with_overflow: 441753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::ssub_with_overflow: 442753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // undef - X -> undef 443753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // X - undef -> undef 4449c68a7bb45ebf71cec5ab4eab7c73aacdfce16dfGabor Greif if (isa<UndefValue>(II->getArgOperand(0)) || 4459c68a7bb45ebf71cec5ab4eab7c73aacdfce16dfGabor Greif isa<UndefValue>(II->getArgOperand(1))) 446753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return ReplaceInstUsesWith(CI, UndefValue::get(II->getType())); 44700e403abe384b8f61dd012242d3d81a23541ca25Jim Grosbach 4489c68a7bb45ebf71cec5ab4eab7c73aacdfce16dfGabor Greif if (ConstantInt *RHS = dyn_cast<ConstantInt>(II->getArgOperand(1))) { 449753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // X - 0 -> {X, false} 450753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (RHS->isZero()) { 451753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Constant *V[] = { 4529c68a7bb45ebf71cec5ab4eab7c73aacdfce16dfGabor Greif UndefValue::get(II->getArgOperand(0)->getType()), 453753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ConstantInt::getFalse(II->getContext()) 454753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner }; 45500e403abe384b8f61dd012242d3d81a23541ca25Jim Grosbach Constant *Struct = 456b065b06c12dba6001b8140df2744d0c856ef6ea1Chris Lattner ConstantStruct::get(cast<StructType>(II->getType()), V); 4579c68a7bb45ebf71cec5ab4eab7c73aacdfce16dfGabor Greif return InsertValueInst::Create(Struct, II->getArgOperand(0), 0); 458753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 459753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 460753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 4616b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer case Intrinsic::umul_with_overflow: { 4626b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer Value *LHS = II->getArgOperand(0), *RHS = II->getArgOperand(1); 4636b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer unsigned BitWidth = cast<IntegerType>(LHS->getType())->getBitWidth(); 4646b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer 4656b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer APInt LHSKnownZero(BitWidth, 0); 4666b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer APInt LHSKnownOne(BitWidth, 0); 467dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines computeKnownBits(LHS, LHSKnownZero, LHSKnownOne); 4686b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer APInt RHSKnownZero(BitWidth, 0); 4696b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer APInt RHSKnownOne(BitWidth, 0); 470dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines computeKnownBits(RHS, RHSKnownZero, RHSKnownOne); 4716b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer 472d655e6e9dc0d21c3a2d76494dec255bb027572aeBenjamin Kramer // Get the largest possible values for each operand. 473d655e6e9dc0d21c3a2d76494dec255bb027572aeBenjamin Kramer APInt LHSMax = ~LHSKnownZero; 474d655e6e9dc0d21c3a2d76494dec255bb027572aeBenjamin Kramer APInt RHSMax = ~RHSKnownZero; 4756b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer 4766b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer // If multiplying the maximum values does not overflow then we can turn 4776b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer // this into a plain NUW mul. 478d655e6e9dc0d21c3a2d76494dec255bb027572aeBenjamin Kramer bool Overflow; 479d655e6e9dc0d21c3a2d76494dec255bb027572aeBenjamin Kramer LHSMax.umul_ov(RHSMax, Overflow); 480d655e6e9dc0d21c3a2d76494dec255bb027572aeBenjamin Kramer if (!Overflow) { 4816b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer Value *Mul = Builder->CreateNUWMul(LHS, RHS, "umul_with_overflow"); 4826b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer Constant *V[] = { 4836b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer UndefValue::get(LHS->getType()), 4846b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer Builder->getFalse() 4856b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer }; 486b065b06c12dba6001b8140df2744d0c856ef6ea1Chris Lattner Constant *Struct = ConstantStruct::get(cast<StructType>(II->getType()),V); 4876b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer return InsertValueInst::Create(Struct, Mul, 0); 4886b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer } 4896b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer } // FALL THROUGH 490753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::smul_with_overflow: 491753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Canonicalize constants into the RHS. 492a90c5c7605e0d8df96e61d4bf0cc47a7eaea92cfGabor Greif if (isa<Constant>(II->getArgOperand(0)) && 493a90c5c7605e0d8df96e61d4bf0cc47a7eaea92cfGabor Greif !isa<Constant>(II->getArgOperand(1))) { 494a90c5c7605e0d8df96e61d4bf0cc47a7eaea92cfGabor Greif Value *LHS = II->getArgOperand(0); 495a90c5c7605e0d8df96e61d4bf0cc47a7eaea92cfGabor Greif II->setArgOperand(0, II->getArgOperand(1)); 496a90c5c7605e0d8df96e61d4bf0cc47a7eaea92cfGabor Greif II->setArgOperand(1, LHS); 497753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return II; 498753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 499753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 500753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // X * undef -> undef 5019c68a7bb45ebf71cec5ab4eab7c73aacdfce16dfGabor Greif if (isa<UndefValue>(II->getArgOperand(1))) 502753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return ReplaceInstUsesWith(CI, UndefValue::get(II->getType())); 50300e403abe384b8f61dd012242d3d81a23541ca25Jim Grosbach 5049c68a7bb45ebf71cec5ab4eab7c73aacdfce16dfGabor Greif if (ConstantInt *RHSI = dyn_cast<ConstantInt>(II->getArgOperand(1))) { 505753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // X*0 -> {0, false} 506753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (RHSI->isZero()) 507753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return ReplaceInstUsesWith(CI, Constant::getNullValue(II->getType())); 50800e403abe384b8f61dd012242d3d81a23541ca25Jim Grosbach 509753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // X * 1 -> {X, false} 510753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (RHSI->equalsInt(1)) { 511753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Constant *V[] = { 512cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif UndefValue::get(II->getArgOperand(0)->getType()), 513753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ConstantInt::getFalse(II->getContext()) 514753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner }; 51500e403abe384b8f61dd012242d3d81a23541ca25Jim Grosbach Constant *Struct = 516b065b06c12dba6001b8140df2744d0c856ef6ea1Chris Lattner ConstantStruct::get(cast<StructType>(II->getType()), V); 517cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif return InsertValueInst::Create(Struct, II->getArgOperand(0), 0); 518753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 519753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 520753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 521753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::ppc_altivec_lvx: 522753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::ppc_altivec_lvxl: 523f93f7b2446bec3febc30b7136e18704664bd98ccBill Wendling // Turn PPC lvx -> load if the pointer is known aligned. 52436b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines if (getOrEnforceKnownAlignment(II->getArgOperand(0), 16, DL) >= 16) { 525cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif Value *Ptr = Builder->CreateBitCast(II->getArgOperand(0), 526753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner PointerType::getUnqual(II->getType())); 527753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return new LoadInst(Ptr); 528753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 529753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 530753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::ppc_altivec_stvx: 531753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::ppc_altivec_stvxl: 532753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Turn stvx -> store if the pointer is known aligned. 53336b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines if (getOrEnforceKnownAlignment(II->getArgOperand(1), 16, DL) >= 16) { 53400e403abe384b8f61dd012242d3d81a23541ca25Jim Grosbach Type *OpPtrTy = 5352f1ab74e763ba7f10a058f62393d0987f6f7e298Gabor Greif PointerType::getUnqual(II->getArgOperand(0)->getType()); 5362f1ab74e763ba7f10a058f62393d0987f6f7e298Gabor Greif Value *Ptr = Builder->CreateBitCast(II->getArgOperand(1), OpPtrTy); 5372f1ab74e763ba7f10a058f62393d0987f6f7e298Gabor Greif return new StoreInst(II->getArgOperand(0), Ptr); 538753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 539753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 540753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::x86_sse_storeu_ps: 541753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::x86_sse2_storeu_pd: 542753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::x86_sse2_storeu_dq: 543753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Turn X86 storeu -> store if the pointer is known aligned. 54436b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines if (getOrEnforceKnownAlignment(II->getArgOperand(0), 16, DL) >= 16) { 54500e403abe384b8f61dd012242d3d81a23541ca25Jim Grosbach Type *OpPtrTy = 5462f1ab74e763ba7f10a058f62393d0987f6f7e298Gabor Greif PointerType::getUnqual(II->getArgOperand(1)->getType()); 5472f1ab74e763ba7f10a058f62393d0987f6f7e298Gabor Greif Value *Ptr = Builder->CreateBitCast(II->getArgOperand(0), OpPtrTy); 5482f1ab74e763ba7f10a058f62393d0987f6f7e298Gabor Greif return new StoreInst(II->getArgOperand(1), Ptr); 549753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 550753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 5519cc9f50abc28f08fb6082b5ee4f5cc7e1af4969bChandler Carruth 5529cc9f50abc28f08fb6082b5ee4f5cc7e1af4969bChandler Carruth case Intrinsic::x86_sse_cvtss2si: 5539cc9f50abc28f08fb6082b5ee4f5cc7e1af4969bChandler Carruth case Intrinsic::x86_sse_cvtss2si64: 5549cc9f50abc28f08fb6082b5ee4f5cc7e1af4969bChandler Carruth case Intrinsic::x86_sse_cvttss2si: 5559cc9f50abc28f08fb6082b5ee4f5cc7e1af4969bChandler Carruth case Intrinsic::x86_sse_cvttss2si64: 5569cc9f50abc28f08fb6082b5ee4f5cc7e1af4969bChandler Carruth case Intrinsic::x86_sse2_cvtsd2si: 5579cc9f50abc28f08fb6082b5ee4f5cc7e1af4969bChandler Carruth case Intrinsic::x86_sse2_cvtsd2si64: 5589cc9f50abc28f08fb6082b5ee4f5cc7e1af4969bChandler Carruth case Intrinsic::x86_sse2_cvttsd2si: 5599cc9f50abc28f08fb6082b5ee4f5cc7e1af4969bChandler Carruth case Intrinsic::x86_sse2_cvttsd2si64: { 5609cc9f50abc28f08fb6082b5ee4f5cc7e1af4969bChandler Carruth // These intrinsics only demand the 0th element of their input vectors. If 561753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // we can simplify the input based on that, do so now. 562753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner unsigned VWidth = 5639c68a7bb45ebf71cec5ab4eab7c73aacdfce16dfGabor Greif cast<VectorType>(II->getArgOperand(0)->getType())->getNumElements(); 564753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt DemandedElts(VWidth, 1); 565753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt UndefElts(VWidth, 0); 566a399781289092fcdceb58b21174229f4373c4191Gabor Greif if (Value *V = SimplifyDemandedVectorElts(II->getArgOperand(0), 567a399781289092fcdceb58b21174229f4373c4191Gabor Greif DemandedElts, UndefElts)) { 568a90c5c7605e0d8df96e61d4bf0cc47a7eaea92cfGabor Greif II->setArgOperand(0, V); 569753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return II; 570753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 571753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 572753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 5739cc9f50abc28f08fb6082b5ee4f5cc7e1af4969bChandler Carruth 574dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines // Constant fold <A x Bi> << Ci. 575dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines // FIXME: We don't handle _dq because it's a shift of an i128, but is 576dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines // represented in the IR as <2 x i64>. A per element shift is wrong. 577dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines case Intrinsic::x86_sse2_psll_d: 578dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines case Intrinsic::x86_sse2_psll_q: 579dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines case Intrinsic::x86_sse2_psll_w: 580dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines case Intrinsic::x86_sse2_pslli_d: 581dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines case Intrinsic::x86_sse2_pslli_q: 582dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines case Intrinsic::x86_sse2_pslli_w: 583dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines case Intrinsic::x86_avx2_psll_d: 584dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines case Intrinsic::x86_avx2_psll_q: 585dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines case Intrinsic::x86_avx2_psll_w: 586dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines case Intrinsic::x86_avx2_pslli_d: 587dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines case Intrinsic::x86_avx2_pslli_q: 588dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines case Intrinsic::x86_avx2_pslli_w: 589dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines case Intrinsic::x86_sse2_psrl_d: 590dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines case Intrinsic::x86_sse2_psrl_q: 591dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines case Intrinsic::x86_sse2_psrl_w: 592dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines case Intrinsic::x86_sse2_psrli_d: 593dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines case Intrinsic::x86_sse2_psrli_q: 594dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines case Intrinsic::x86_sse2_psrli_w: 595dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines case Intrinsic::x86_avx2_psrl_d: 596dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines case Intrinsic::x86_avx2_psrl_q: 597dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines case Intrinsic::x86_avx2_psrl_w: 598dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines case Intrinsic::x86_avx2_psrli_d: 599dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines case Intrinsic::x86_avx2_psrli_q: 600dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines case Intrinsic::x86_avx2_psrli_w: { 601dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines // Simplify if count is constant. To 0 if >= BitWidth, 602dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines // otherwise to shl/lshr. 603dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines auto CDV = dyn_cast<ConstantDataVector>(II->getArgOperand(1)); 604dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines auto CInt = dyn_cast<ConstantInt>(II->getArgOperand(1)); 605dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines if (!CDV && !CInt) 606dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines break; 607dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines ConstantInt *Count; 608dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines if (CDV) 609dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines Count = cast<ConstantInt>(CDV->getElementAsConstant(0)); 610dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines else 611dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines Count = CInt; 612dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines 613dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines auto Vec = II->getArgOperand(0); 614dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines auto VT = cast<VectorType>(Vec->getType()); 615dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines if (Count->getZExtValue() > 616dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines VT->getElementType()->getPrimitiveSizeInBits() - 1) 617dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return ReplaceInstUsesWith( 618dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines CI, ConstantAggregateZero::get(Vec->getType())); 619dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines 620dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines bool isPackedShiftLeft = true; 621dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines switch (II->getIntrinsicID()) { 622dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines default : break; 623dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines case Intrinsic::x86_sse2_psrl_d: 624dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines case Intrinsic::x86_sse2_psrl_q: 625dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines case Intrinsic::x86_sse2_psrl_w: 626dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines case Intrinsic::x86_sse2_psrli_d: 627dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines case Intrinsic::x86_sse2_psrli_q: 628dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines case Intrinsic::x86_sse2_psrli_w: 629dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines case Intrinsic::x86_avx2_psrl_d: 630dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines case Intrinsic::x86_avx2_psrl_q: 631dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines case Intrinsic::x86_avx2_psrl_w: 632dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines case Intrinsic::x86_avx2_psrli_d: 633dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines case Intrinsic::x86_avx2_psrli_q: 634dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines case Intrinsic::x86_avx2_psrli_w: isPackedShiftLeft = false; break; 635dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines } 636dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines 637dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines unsigned VWidth = VT->getNumElements(); 638dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines // Get a constant vector of the same type as the first operand. 639dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines auto VTCI = ConstantInt::get(VT->getElementType(), Count->getZExtValue()); 640dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines if (isPackedShiftLeft) 641dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return BinaryOperator::CreateShl(Vec, 642dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines Builder->CreateVectorSplat(VWidth, VTCI)); 643dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines 644dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return BinaryOperator::CreateLShr(Vec, 645dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines Builder->CreateVectorSplat(VWidth, VTCI)); 646dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines } 647ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings 648ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings case Intrinsic::x86_sse41_pmovsxbw: 649ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings case Intrinsic::x86_sse41_pmovsxwd: 650ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings case Intrinsic::x86_sse41_pmovsxdq: 651ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings case Intrinsic::x86_sse41_pmovzxbw: 652ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings case Intrinsic::x86_sse41_pmovzxwd: 653ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings case Intrinsic::x86_sse41_pmovzxdq: { 654aaa7f499c1a2d46491a98d978ef45b0a06d73d31Evan Cheng // pmov{s|z}x ignores the upper half of their input vectors. 655ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings unsigned VWidth = 656ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings cast<VectorType>(II->getArgOperand(0)->getType())->getNumElements(); 657ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings unsigned LowHalfElts = VWidth / 2; 658d116611e0a884e53ea1524de3a1306c1eb447228Stuart Hastings APInt InputDemandedElts(APInt::getBitsSet(VWidth, 0, LowHalfElts)); 659ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings APInt UndefElts(VWidth, 0); 660ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings if (Value *TmpV = SimplifyDemandedVectorElts(II->getArgOperand(0), 661ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings InputDemandedElts, 662ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings UndefElts)) { 663ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings II->setArgOperand(0, TmpV); 664ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings return II; 665ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings } 666ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings break; 667ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings } 668ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings 669dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines case Intrinsic::x86_sse4a_insertqi: { 670dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines // insertqi x, y, 64, 0 can just copy y's lower bits and leave the top 671dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines // ones undef 672dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines // TODO: eventually we should lower this intrinsic to IR 673dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines if (auto CIWidth = dyn_cast<ConstantInt>(II->getArgOperand(2))) { 674dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines if (auto CIStart = dyn_cast<ConstantInt>(II->getArgOperand(3))) { 675dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines if (CIWidth->equalsInt(64) && CIStart->isZero()) { 676dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines Value *Vec = II->getArgOperand(1); 677dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines Value *Undef = UndefValue::get(Vec->getType()); 678dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines const uint32_t Mask[] = { 0, 2 }; 679dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return ReplaceInstUsesWith( 680dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines CI, 681dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines Builder->CreateShuffleVector( 682dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines Vec, Undef, ConstantDataVector::get( 683dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines II->getContext(), ArrayRef<uint32_t>(Mask)))); 684dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines 685dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines } else if (auto Source = 686dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines dyn_cast<IntrinsicInst>(II->getArgOperand(0))) { 687dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines if (Source->hasOneUse() && 688dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines Source->getArgOperand(1) == II->getArgOperand(1)) { 689dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines // If the source of the insert has only one use and it's another 690dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines // insert (and they're both inserting from the same vector), try to 691dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines // bundle both together. 692dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines auto CISourceWidth = 693dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines dyn_cast<ConstantInt>(Source->getArgOperand(2)); 694dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines auto CISourceStart = 695dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines dyn_cast<ConstantInt>(Source->getArgOperand(3)); 696dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines if (CISourceStart && CISourceWidth) { 697dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines unsigned Start = CIStart->getZExtValue(); 698dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines unsigned Width = CIWidth->getZExtValue(); 699dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines unsigned End = Start + Width; 700dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines unsigned SourceStart = CISourceStart->getZExtValue(); 701dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines unsigned SourceWidth = CISourceWidth->getZExtValue(); 702dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines unsigned SourceEnd = SourceStart + SourceWidth; 703dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines unsigned NewStart, NewWidth; 704dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines bool ShouldReplace = false; 705dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines if (Start <= SourceStart && SourceStart <= End) { 706dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines NewStart = Start; 707dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines NewWidth = std::max(End, SourceEnd) - NewStart; 708dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines ShouldReplace = true; 709dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines } else if (SourceStart <= Start && Start <= SourceEnd) { 710dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines NewStart = SourceStart; 711dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines NewWidth = std::max(SourceEnd, End) - NewStart; 712dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines ShouldReplace = true; 713dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines } 714dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines 715dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines if (ShouldReplace) { 716dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines Constant *ConstantWidth = ConstantInt::get( 717dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines II->getArgOperand(2)->getType(), NewWidth, false); 718dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines Constant *ConstantStart = ConstantInt::get( 719dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines II->getArgOperand(3)->getType(), NewStart, false); 720dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines Value *Args[4] = { Source->getArgOperand(0), 721dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines II->getArgOperand(1), ConstantWidth, 722dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines ConstantStart }; 723dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines Module *M = CI.getParent()->getParent()->getParent(); 724dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines Value *F = 725dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines Intrinsic::getDeclaration(M, Intrinsic::x86_sse4a_insertqi); 726dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return ReplaceInstUsesWith(CI, Builder->CreateCall(F, Args)); 727dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines } 728dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines } 729dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines } 730dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines } 731dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines } 732dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines } 733dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines break; 734dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines } 735dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines 736dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines case Intrinsic::x86_sse41_pblendvb: 737dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines case Intrinsic::x86_sse41_blendvps: 738dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines case Intrinsic::x86_sse41_blendvpd: 739dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines case Intrinsic::x86_avx_blendv_ps_256: 740dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines case Intrinsic::x86_avx_blendv_pd_256: 741dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines case Intrinsic::x86_avx2_pblendvb: { 742dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines // Convert blendv* to vector selects if the mask is constant. 743dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines // This optimization is convoluted because the intrinsic is defined as 744dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines // getting a vector of floats or doubles for the ps and pd versions. 745dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines // FIXME: That should be changed. 746dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines Value *Mask = II->getArgOperand(2); 747dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines if (auto C = dyn_cast<ConstantDataVector>(Mask)) { 748dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines auto Tyi1 = Builder->getInt1Ty(); 749dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines auto SelectorType = cast<VectorType>(Mask->getType()); 750dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines auto EltTy = SelectorType->getElementType(); 751dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines unsigned Size = SelectorType->getNumElements(); 752dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines unsigned BitWidth = 753dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines EltTy->isFloatTy() 754dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines ? 32 755dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines : (EltTy->isDoubleTy() ? 64 : EltTy->getIntegerBitWidth()); 756dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines assert((BitWidth == 64 || BitWidth == 32 || BitWidth == 8) && 757dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines "Wrong arguments for variable blend intrinsic"); 758dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines SmallVector<Constant *, 32> Selectors; 759dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines for (unsigned I = 0; I < Size; ++I) { 760dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines // The intrinsics only read the top bit 761dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines uint64_t Selector; 762dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines if (BitWidth == 8) 763dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines Selector = C->getElementAsInteger(I); 764dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines else 765dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines Selector = C->getElementAsAPFloat(I).bitcastToAPInt().getZExtValue(); 766dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines Selectors.push_back(ConstantInt::get(Tyi1, Selector >> (BitWidth - 1))); 767dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines } 768dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines auto NewSelector = ConstantVector::get(Selectors); 769dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return SelectInst::Create(NewSelector, II->getArgOperand(1), 770dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines II->getArgOperand(0), "blendv"); 771dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines } else { 772dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines break; 773dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines } 774dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines } 775dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines 776dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines case Intrinsic::x86_avx_vpermilvar_ps: 777dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines case Intrinsic::x86_avx_vpermilvar_ps_256: 778dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines case Intrinsic::x86_avx_vpermilvar_pd: 779dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines case Intrinsic::x86_avx_vpermilvar_pd_256: { 780dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines // Convert vpermil* to shufflevector if the mask is constant. 781dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines Value *V = II->getArgOperand(1); 782dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines unsigned Size = cast<VectorType>(V->getType())->getNumElements(); 783dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines assert(Size == 8 || Size == 4 || Size == 2); 784dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines uint32_t Indexes[8]; 785dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines if (auto C = dyn_cast<ConstantDataVector>(V)) { 786dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines // The intrinsics only read one or two bits, clear the rest. 787dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines for (unsigned I = 0; I < Size; ++I) { 788dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines uint32_t Index = C->getElementAsInteger(I) & 0x3; 789dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines if (II->getIntrinsicID() == Intrinsic::x86_avx_vpermilvar_pd || 790dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines II->getIntrinsicID() == Intrinsic::x86_avx_vpermilvar_pd_256) 791dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines Index >>= 1; 792dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines Indexes[I] = Index; 793dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines } 794dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines } else if (isa<ConstantAggregateZero>(V)) { 795dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines for (unsigned I = 0; I < Size; ++I) 796dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines Indexes[I] = 0; 797dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines } else { 798dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines break; 799dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines } 800dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines // The _256 variants are a bit trickier since the mask bits always index 801dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines // into the corresponding 128 half. In order to convert to a generic 802dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines // shuffle, we have to make that explicit. 803dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines if (II->getIntrinsicID() == Intrinsic::x86_avx_vpermilvar_ps_256 || 804dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines II->getIntrinsicID() == Intrinsic::x86_avx_vpermilvar_pd_256) { 805dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines for (unsigned I = Size / 2; I < Size; ++I) 806dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines Indexes[I] += Size / 2; 807dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines } 808dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines auto NewC = 809dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines ConstantDataVector::get(V->getContext(), makeArrayRef(Indexes, Size)); 810dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines auto V1 = II->getArgOperand(0); 811dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines auto V2 = UndefValue::get(V1->getType()); 812dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines auto Shuffle = Builder->CreateShuffleVector(V1, V2, NewC); 813dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return ReplaceInstUsesWith(CI, Shuffle); 814dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines } 815dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines 816753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::ppc_altivec_vperm: 817753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Turn vperm(V1,V2,mask) -> shuffle(V1,V2,mask) if mask is a constant. 818cd81d94322a39503e4a3e87b6ee03d4fcb3465fbStephen Hines // Note that ppc_altivec_vperm has a big-endian bias, so when creating 819cd81d94322a39503e4a3e87b6ee03d4fcb3465fbStephen Hines // a vectorshuffle for little endian, we must undo the transformation 820cd81d94322a39503e4a3e87b6ee03d4fcb3465fbStephen Hines // performed on vec_perm in altivec.h. That is, we must complement 821cd81d94322a39503e4a3e87b6ee03d4fcb3465fbStephen Hines // the permutation mask with respect to 31 and reverse the order of 822cd81d94322a39503e4a3e87b6ee03d4fcb3465fbStephen Hines // V1 and V2. 823a78fa8cc2dd6d2ffe5e4fe605f38aae7b3d2fb7aChris Lattner if (Constant *Mask = dyn_cast<Constant>(II->getArgOperand(2))) { 824a78fa8cc2dd6d2ffe5e4fe605f38aae7b3d2fb7aChris Lattner assert(Mask->getType()->getVectorNumElements() == 16 && 825a78fa8cc2dd6d2ffe5e4fe605f38aae7b3d2fb7aChris Lattner "Bad type for intrinsic!"); 82600e403abe384b8f61dd012242d3d81a23541ca25Jim Grosbach 827753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Check that all of the elements are integer constants or undefs. 828753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner bool AllEltsOk = true; 829753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner for (unsigned i = 0; i != 16; ++i) { 830a78fa8cc2dd6d2ffe5e4fe605f38aae7b3d2fb7aChris Lattner Constant *Elt = Mask->getAggregateElement(i); 831dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines if (!Elt || !(isa<ConstantInt>(Elt) || isa<UndefValue>(Elt))) { 832753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner AllEltsOk = false; 833753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 834753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 835753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 83600e403abe384b8f61dd012242d3d81a23541ca25Jim Grosbach 837753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (AllEltsOk) { 838753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Cast the input vectors to byte vectors. 839a399781289092fcdceb58b21174229f4373c4191Gabor Greif Value *Op0 = Builder->CreateBitCast(II->getArgOperand(0), 840a399781289092fcdceb58b21174229f4373c4191Gabor Greif Mask->getType()); 841a399781289092fcdceb58b21174229f4373c4191Gabor Greif Value *Op1 = Builder->CreateBitCast(II->getArgOperand(1), 842a399781289092fcdceb58b21174229f4373c4191Gabor Greif Mask->getType()); 843753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Value *Result = UndefValue::get(Op0->getType()); 84400e403abe384b8f61dd012242d3d81a23541ca25Jim Grosbach 845753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Only extract each element once. 846753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Value *ExtractedElts[32]; 847753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner memset(ExtractedElts, 0, sizeof(ExtractedElts)); 84800e403abe384b8f61dd012242d3d81a23541ca25Jim Grosbach 849753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner for (unsigned i = 0; i != 16; ++i) { 850a78fa8cc2dd6d2ffe5e4fe605f38aae7b3d2fb7aChris Lattner if (isa<UndefValue>(Mask->getAggregateElement(i))) 851753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner continue; 85200e403abe384b8f61dd012242d3d81a23541ca25Jim Grosbach unsigned Idx = 853a78fa8cc2dd6d2ffe5e4fe605f38aae7b3d2fb7aChris Lattner cast<ConstantInt>(Mask->getAggregateElement(i))->getZExtValue(); 854753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Idx &= 31; // Match the hardware behavior. 855cd81d94322a39503e4a3e87b6ee03d4fcb3465fbStephen Hines if (DL && DL->isLittleEndian()) 856cd81d94322a39503e4a3e87b6ee03d4fcb3465fbStephen Hines Idx = 31 - Idx; 85700e403abe384b8f61dd012242d3d81a23541ca25Jim Grosbach 858dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines if (!ExtractedElts[Idx]) { 859cd81d94322a39503e4a3e87b6ee03d4fcb3465fbStephen Hines Value *Op0ToUse = (DL && DL->isLittleEndian()) ? Op1 : Op0; 860cd81d94322a39503e4a3e87b6ee03d4fcb3465fbStephen Hines Value *Op1ToUse = (DL && DL->isLittleEndian()) ? Op0 : Op1; 86100e403abe384b8f61dd012242d3d81a23541ca25Jim Grosbach ExtractedElts[Idx] = 862cd81d94322a39503e4a3e87b6ee03d4fcb3465fbStephen Hines Builder->CreateExtractElement(Idx < 16 ? Op0ToUse : Op1ToUse, 863a9390a4d5f5d568059a80970d22194b165d097a7Benjamin Kramer Builder->getInt32(Idx&15)); 864753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 86500e403abe384b8f61dd012242d3d81a23541ca25Jim Grosbach 866753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Insert this value into the result vector. 867753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Result = Builder->CreateInsertElement(Result, ExtractedElts[Idx], 868a9390a4d5f5d568059a80970d22194b165d097a7Benjamin Kramer Builder->getInt32(i)); 869753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 870753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return CastInst::Create(Instruction::BitCast, Result, CI.getType()); 871753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 872753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 873753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 874753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 875364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson case Intrinsic::arm_neon_vld1: 876364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson case Intrinsic::arm_neon_vld2: 877364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson case Intrinsic::arm_neon_vld3: 878364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson case Intrinsic::arm_neon_vld4: 879364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson case Intrinsic::arm_neon_vld2lane: 880364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson case Intrinsic::arm_neon_vld3lane: 881364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson case Intrinsic::arm_neon_vld4lane: 882364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson case Intrinsic::arm_neon_vst1: 883364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson case Intrinsic::arm_neon_vst2: 884364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson case Intrinsic::arm_neon_vst3: 885364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson case Intrinsic::arm_neon_vst4: 886364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson case Intrinsic::arm_neon_vst2lane: 887364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson case Intrinsic::arm_neon_vst3lane: 888364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson case Intrinsic::arm_neon_vst4lane: { 88936b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines unsigned MemAlign = getKnownAlignment(II->getArgOperand(0), DL); 890364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson unsigned AlignArg = II->getNumArgOperands() - 1; 891364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson ConstantInt *IntrAlign = dyn_cast<ConstantInt>(II->getArgOperand(AlignArg)); 892364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson if (IntrAlign && IntrAlign->getZExtValue() < MemAlign) { 893364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson II->setArgOperand(AlignArg, 894364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson ConstantInt::get(Type::getInt32Ty(II->getContext()), 895364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson MemAlign, false)); 896364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson return II; 897364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson } 898364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson break; 899364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson } 900364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson 901973f72a29aeafb1fdc4f8dafc3f6c6651cbb0c99Lang Hames case Intrinsic::arm_neon_vmulls: 90236b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines case Intrinsic::arm_neon_vmullu: 903dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines case Intrinsic::aarch64_neon_smull: 904dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines case Intrinsic::aarch64_neon_umull: { 905973f72a29aeafb1fdc4f8dafc3f6c6651cbb0c99Lang Hames Value *Arg0 = II->getArgOperand(0); 906973f72a29aeafb1fdc4f8dafc3f6c6651cbb0c99Lang Hames Value *Arg1 = II->getArgOperand(1); 907973f72a29aeafb1fdc4f8dafc3f6c6651cbb0c99Lang Hames 908973f72a29aeafb1fdc4f8dafc3f6c6651cbb0c99Lang Hames // Handle mul by zero first: 909973f72a29aeafb1fdc4f8dafc3f6c6651cbb0c99Lang Hames if (isa<ConstantAggregateZero>(Arg0) || isa<ConstantAggregateZero>(Arg1)) { 910973f72a29aeafb1fdc4f8dafc3f6c6651cbb0c99Lang Hames return ReplaceInstUsesWith(CI, ConstantAggregateZero::get(II->getType())); 911973f72a29aeafb1fdc4f8dafc3f6c6651cbb0c99Lang Hames } 912973f72a29aeafb1fdc4f8dafc3f6c6651cbb0c99Lang Hames 913973f72a29aeafb1fdc4f8dafc3f6c6651cbb0c99Lang Hames // Check for constant LHS & RHS - in this case we just simplify. 91436b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines bool Zext = (II->getIntrinsicID() == Intrinsic::arm_neon_vmullu || 915dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines II->getIntrinsicID() == Intrinsic::aarch64_neon_umull); 916973f72a29aeafb1fdc4f8dafc3f6c6651cbb0c99Lang Hames VectorType *NewVT = cast<VectorType>(II->getType()); 91736b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines if (Constant *CV0 = dyn_cast<Constant>(Arg0)) { 91836b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines if (Constant *CV1 = dyn_cast<Constant>(Arg1)) { 91936b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines CV0 = ConstantExpr::getIntegerCast(CV0, NewVT, /*isSigned=*/!Zext); 92036b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines CV1 = ConstantExpr::getIntegerCast(CV1, NewVT, /*isSigned=*/!Zext); 92136b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines 92236b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines return ReplaceInstUsesWith(CI, ConstantExpr::getMul(CV0, CV1)); 923973f72a29aeafb1fdc4f8dafc3f6c6651cbb0c99Lang Hames } 924973f72a29aeafb1fdc4f8dafc3f6c6651cbb0c99Lang Hames 92536b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines // Couldn't simplify - canonicalize constant to the RHS. 926973f72a29aeafb1fdc4f8dafc3f6c6651cbb0c99Lang Hames std::swap(Arg0, Arg1); 927973f72a29aeafb1fdc4f8dafc3f6c6651cbb0c99Lang Hames } 928973f72a29aeafb1fdc4f8dafc3f6c6651cbb0c99Lang Hames 929973f72a29aeafb1fdc4f8dafc3f6c6651cbb0c99Lang Hames // Handle mul by one: 93036b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines if (Constant *CV1 = dyn_cast<Constant>(Arg1)) 931973f72a29aeafb1fdc4f8dafc3f6c6651cbb0c99Lang Hames if (ConstantInt *Splat = 93236b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines dyn_cast_or_null<ConstantInt>(CV1->getSplatValue())) 93336b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines if (Splat->isOne()) 93436b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines return CastInst::CreateIntegerCast(Arg0, II->getType(), 93536b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines /*isSigned=*/!Zext); 936973f72a29aeafb1fdc4f8dafc3f6c6651cbb0c99Lang Hames 937973f72a29aeafb1fdc4f8dafc3f6c6651cbb0c99Lang Hames break; 938973f72a29aeafb1fdc4f8dafc3f6c6651cbb0c99Lang Hames } 939973f72a29aeafb1fdc4f8dafc3f6c6651cbb0c99Lang Hames 940cd81d94322a39503e4a3e87b6ee03d4fcb3465fbStephen Hines case Intrinsic::AMDGPU_rcp: { 941cd81d94322a39503e4a3e87b6ee03d4fcb3465fbStephen Hines if (const ConstantFP *C = dyn_cast<ConstantFP>(II->getArgOperand(0))) { 942cd81d94322a39503e4a3e87b6ee03d4fcb3465fbStephen Hines const APFloat &ArgVal = C->getValueAPF(); 943cd81d94322a39503e4a3e87b6ee03d4fcb3465fbStephen Hines APFloat Val(ArgVal.getSemantics(), 1.0); 944cd81d94322a39503e4a3e87b6ee03d4fcb3465fbStephen Hines APFloat::opStatus Status = Val.divide(ArgVal, 945cd81d94322a39503e4a3e87b6ee03d4fcb3465fbStephen Hines APFloat::rmNearestTiesToEven); 946cd81d94322a39503e4a3e87b6ee03d4fcb3465fbStephen Hines // Only do this if it was exact and therefore not dependent on the 947cd81d94322a39503e4a3e87b6ee03d4fcb3465fbStephen Hines // rounding mode. 948cd81d94322a39503e4a3e87b6ee03d4fcb3465fbStephen Hines if (Status == APFloat::opOK) 949cd81d94322a39503e4a3e87b6ee03d4fcb3465fbStephen Hines return ReplaceInstUsesWith(CI, ConstantFP::get(II->getContext(), Val)); 950cd81d94322a39503e4a3e87b6ee03d4fcb3465fbStephen Hines } 951cd81d94322a39503e4a3e87b6ee03d4fcb3465fbStephen Hines 952cd81d94322a39503e4a3e87b6ee03d4fcb3465fbStephen Hines break; 953cd81d94322a39503e4a3e87b6ee03d4fcb3465fbStephen Hines } 954753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::stackrestore: { 955753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If the save is right next to the restore, remove the restore. This can 956753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // happen when variable allocas are DCE'd. 957cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif if (IntrinsicInst *SS = dyn_cast<IntrinsicInst>(II->getArgOperand(0))) { 958753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (SS->getIntrinsicID() == Intrinsic::stacksave) { 959753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner BasicBlock::iterator BI = SS; 960753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (&*++BI == II) 961753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return EraseInstFromFunction(CI); 962753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 963753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 96400e403abe384b8f61dd012242d3d81a23541ca25Jim Grosbach 965753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Scan down this block to see if there is another stack restore in the 966753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // same block without an intervening call/alloca. 967753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner BasicBlock::iterator BI = II; 968753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner TerminatorInst *TI = II->getParent()->getTerminator(); 969753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner bool CannotRemove = false; 970753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner for (++BI; &*BI != TI; ++BI) { 9719e72a79ef4a9fcda482ce0b0e1f0bd6a4f16cffdNuno Lopes if (isa<AllocaInst>(BI)) { 972753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner CannotRemove = true; 973753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 974753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 975753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (CallInst *BCI = dyn_cast<CallInst>(BI)) { 976753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (IntrinsicInst *II = dyn_cast<IntrinsicInst>(BCI)) { 977753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If there is a stackrestore below this one, remove this one. 978753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (II->getIntrinsicID() == Intrinsic::stackrestore) 979753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return EraseInstFromFunction(CI); 980753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Otherwise, ignore the intrinsic. 981753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } else { 982753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If we found a non-intrinsic call, we can't remove the stack 983753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // restore. 984753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner CannotRemove = true; 985753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 986753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 987753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 988753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 98900e403abe384b8f61dd012242d3d81a23541ca25Jim Grosbach 990dccc03b2423fe65efb5963ae816b99c24fc53374Bill Wendling // If the stack restore is in a return, resume, or unwind block and if there 991dccc03b2423fe65efb5963ae816b99c24fc53374Bill Wendling // are no allocas or calls between the restore and the return, nuke the 992dccc03b2423fe65efb5963ae816b99c24fc53374Bill Wendling // restore. 993aa5abe88d6aa445afa593476a665e3ab14b3524cBill Wendling if (!CannotRemove && (isa<ReturnInst>(TI) || isa<ResumeInst>(TI))) 994753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return EraseInstFromFunction(CI); 995753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 996753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 997753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 998753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 999753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return visitCallSite(II); 1000753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner} 1001753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1002753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner// InvokeInst simplification 1003753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner// 1004753a2b464d0628231bd3dce645edd05dfbaa3a06Chris LattnerInstruction *InstCombiner::visitInvokeInst(InvokeInst &II) { 1005753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return visitCallSite(&II); 1006753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner} 1007753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 100800e403abe384b8f61dd012242d3d81a23541ca25Jim Grosbach/// isSafeToEliminateVarargsCast - If this cast does not affect the value 1009753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner/// passed through the varargs area, we can eliminate the use of the cast. 1010753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattnerstatic bool isSafeToEliminateVarargsCast(const CallSite CS, 1011753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const CastInst * const CI, 101236b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines const DataLayout * const DL, 1013753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const int ix) { 1014753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!CI->isLosslessCast()) 1015753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return false; 1016753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 101736b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines // The size of ByVal or InAlloca arguments is derived from the type, so we 1018753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // can't change to a type with a different size. If the size were 1019753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // passed explicitly we could avoid this check. 102036b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines if (!CS.isByValOrInAllocaArgument(ix)) 1021753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return true; 1022753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 102300e403abe384b8f61dd012242d3d81a23541ca25Jim Grosbach Type* SrcTy = 1024753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner cast<PointerType>(CI->getOperand(0)->getType())->getElementType(); 1025db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type* DstTy = cast<PointerType>(CI->getType())->getElementType(); 1026753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!SrcTy->isSized() || !DstTy->isSized()) 1027753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return false; 102836b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines if (!DL || DL->getTypeAllocSize(SrcTy) != DL->getTypeAllocSize(DstTy)) 1029753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return false; 1030753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return true; 1031753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner} 1032753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 103327ceaa13f820e33c44e29adc2095dd3e1e69d09fEric Christopher// Try to fold some different type of calls here. 103400e403abe384b8f61dd012242d3d81a23541ca25Jim Grosbach// Currently we're only working with the checking functions, memcpy_chk, 103527ceaa13f820e33c44e29adc2095dd3e1e69d09fEric Christopher// mempcpy_chk, memmove_chk, memset_chk, strcpy_chk, stpcpy_chk, strncpy_chk, 103627ceaa13f820e33c44e29adc2095dd3e1e69d09fEric Christopher// strcat_chk and strncat_chk. 103736b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen HinesInstruction *InstCombiner::tryOptimizeCall(CallInst *CI, const DataLayout *DL) { 1038dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines if (!CI->getCalledFunction()) return nullptr; 103927ceaa13f820e33c44e29adc2095dd3e1e69d09fEric Christopher 104063f932ca3c91ea8ac5b592158f5e8ef7de550547Meador Inge if (Value *With = Simplifier->optimizeCall(CI)) { 104163f932ca3c91ea8ac5b592158f5e8ef7de550547Meador Inge ++NumSimplified; 1042a241b58f8d5d71eebfb2cb9be3689fe72b454474Meador Inge return CI->use_empty() ? CI : ReplaceInstUsesWith(*CI, With); 104363f932ca3c91ea8ac5b592158f5e8ef7de550547Meador Inge } 10445e8904576a5260cfd5b14596e338a4bb25b9817eMeador Inge 1045dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return nullptr; 104627ceaa13f820e33c44e29adc2095dd3e1e69d09fEric Christopher} 104727ceaa13f820e33c44e29adc2095dd3e1e69d09fEric Christopher 10484a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sandsstatic IntrinsicInst *FindInitTrampolineFromAlloca(Value *TrampMem) { 10494a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands // Strip off at most one level of pointer casts, looking for an alloca. This 10504a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands // is good enough in practice and simpler than handling any number of casts. 10514a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands Value *Underlying = TrampMem->stripPointerCasts(); 10524a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands if (Underlying != TrampMem && 105336b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines (!Underlying->hasOneUse() || Underlying->user_back() != TrampMem)) 1054dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return nullptr; 10554a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands if (!isa<AllocaInst>(Underlying)) 1056dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return nullptr; 10574a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands 1058dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines IntrinsicInst *InitTrampoline = nullptr; 105936b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines for (User *U : TrampMem->users()) { 106036b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines IntrinsicInst *II = dyn_cast<IntrinsicInst>(U); 10614a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands if (!II) 1062dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return nullptr; 10634a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands if (II->getIntrinsicID() == Intrinsic::init_trampoline) { 10644a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands if (InitTrampoline) 10654a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands // More than one init_trampoline writes to this value. Give up. 1066dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return nullptr; 10674a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands InitTrampoline = II; 10684a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands continue; 10694a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands } 10704a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands if (II->getIntrinsicID() == Intrinsic::adjust_trampoline) 10714a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands // Allow any number of calls to adjust.trampoline. 10724a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands continue; 1073dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return nullptr; 10744a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands } 10754a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands 10764a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands // No call to init.trampoline found. 10774a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands if (!InitTrampoline) 1078dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return nullptr; 10794a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands 10804a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands // Check that the alloca is being used in the expected way. 10814a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands if (InitTrampoline->getOperand(0) != TrampMem) 1082dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return nullptr; 10834a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands 10844a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands return InitTrampoline; 10854a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands} 10864a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands 10874a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sandsstatic IntrinsicInst *FindInitTrampolineFromBB(IntrinsicInst *AdjustTramp, 10884a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands Value *TrampMem) { 10894a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands // Visit all the previous instructions in the basic block, and try to find a 10904a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands // init.trampoline which has a direct path to the adjust.trampoline. 10914a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands for (BasicBlock::iterator I = AdjustTramp, 10924a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands E = AdjustTramp->getParent()->begin(); I != E; ) { 10934a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands Instruction *Inst = --I; 10944a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands if (IntrinsicInst *II = dyn_cast<IntrinsicInst>(I)) 10954a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands if (II->getIntrinsicID() == Intrinsic::init_trampoline && 10964a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands II->getOperand(0) == TrampMem) 10974a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands return II; 10984a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands if (Inst->mayWriteToMemory()) 1099dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return nullptr; 11004a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands } 1101dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return nullptr; 11024a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands} 11034a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands 11044a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands// Given a call to llvm.adjust.trampoline, find and return the corresponding 11054a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands// call to llvm.init.trampoline if the call to the trampoline can be optimized 11064a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands// to a direct call to a function. Otherwise return NULL. 11074a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands// 11084a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sandsstatic IntrinsicInst *FindInitTrampoline(Value *Callee) { 11094a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands Callee = Callee->stripPointerCasts(); 11104a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands IntrinsicInst *AdjustTramp = dyn_cast<IntrinsicInst>(Callee); 11114a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands if (!AdjustTramp || 11124a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands AdjustTramp->getIntrinsicID() != Intrinsic::adjust_trampoline) 1113dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return nullptr; 11144a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands 11154a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands Value *TrampMem = AdjustTramp->getOperand(0); 11164a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands 11174a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands if (IntrinsicInst *IT = FindInitTrampolineFromAlloca(TrampMem)) 11184a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands return IT; 11194a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands if (IntrinsicInst *IT = FindInitTrampolineFromBB(AdjustTramp, TrampMem)) 11204a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands return IT; 1121dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return nullptr; 11224a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands} 11234a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands 1124753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner// visitCallSite - Improvements for call and invoke instructions. 1125753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner// 1126753a2b464d0628231bd3dce645edd05dfbaa3a06Chris LattnerInstruction *InstCombiner::visitCallSite(CallSite CS) { 11278e0d1c03ca7fd86e6879b4e37d0d7f0e982feef6Benjamin Kramer if (isAllocLikeFn(CS.getInstruction(), TLI)) 112878f8ef42173a3a9867ed789073d4ddc652fb7ff2Nuno Lopes return visitAllocSite(*CS.getInstruction()); 11292b3e9580536dfb5666b9d91e99baebf6d45bfa5fNuno Lopes 1130753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner bool Changed = false; 1131753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1132ab215bc371677cdeac6eacb3e890e5af666d1ab4Chris Lattner // If the callee is a pointer to a function, attempt to move any casts to the 1133ab215bc371677cdeac6eacb3e890e5af666d1ab4Chris Lattner // arguments of the call/invoke. 1134753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Value *Callee = CS.getCalledValue(); 1135ab215bc371677cdeac6eacb3e890e5af666d1ab4Chris Lattner if (!isa<Function>(Callee) && transformConstExprCastCall(CS)) 1136dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return nullptr; 1137753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1138753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Function *CalleeF = dyn_cast<Function>(Callee)) 1139d569561835b0fa4dbbb0fca1b1f0a8de6c01439aChris Lattner // If the call and callee calling conventions don't match, this call must 1140d569561835b0fa4dbbb0fca1b1f0a8de6c01439aChris Lattner // be unreachable, as the call is undefined. 1141d569561835b0fa4dbbb0fca1b1f0a8de6c01439aChris Lattner if (CalleeF->getCallingConv() != CS.getCallingConv() && 1142d569561835b0fa4dbbb0fca1b1f0a8de6c01439aChris Lattner // Only do this for calls to a function with a body. A prototype may 1143d569561835b0fa4dbbb0fca1b1f0a8de6c01439aChris Lattner // not actually end up matching the implementation's calling conv for a 1144d569561835b0fa4dbbb0fca1b1f0a8de6c01439aChris Lattner // variety of reasons (e.g. it may be written in assembly). 1145d569561835b0fa4dbbb0fca1b1f0a8de6c01439aChris Lattner !CalleeF->isDeclaration()) { 1146753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Instruction *OldCall = CS.getInstruction(); 1147753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner new StoreInst(ConstantInt::getTrue(Callee->getContext()), 114800e403abe384b8f61dd012242d3d81a23541ca25Jim Grosbach UndefValue::get(Type::getInt1PtrTy(Callee->getContext())), 1149753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner OldCall); 11500fabd08066a29f4e5060539d0d0605cbfe2760b8Chad Rosier // If OldCall does not return void then replaceAllUsesWith undef. 1151753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // This allows ValueHandlers and custom metadata to adjust itself. 1152753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!OldCall->getType()->isVoidTy()) 11533e22cb9ec30cd9b1be9b0f50e400f512124997e5Eli Friedman ReplaceInstUsesWith(*OldCall, UndefValue::get(OldCall->getType())); 1154830f3f205d8a77acbf8838fe78c2c733ca924d9cChris Lattner if (isa<CallInst>(OldCall)) 1155753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return EraseInstFromFunction(*OldCall); 115600e403abe384b8f61dd012242d3d81a23541ca25Jim Grosbach 1157830f3f205d8a77acbf8838fe78c2c733ca924d9cChris Lattner // We cannot remove an invoke, because it would change the CFG, just 1158830f3f205d8a77acbf8838fe78c2c733ca924d9cChris Lattner // change the callee to a null pointer. 1159654c06f6457f80428e40810ad251c412462731e4Gabor Greif cast<InvokeInst>(OldCall)->setCalledFunction( 1160830f3f205d8a77acbf8838fe78c2c733ca924d9cChris Lattner Constant::getNullValue(CalleeF->getType())); 1161dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return nullptr; 1162753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1163753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1164753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (isa<ConstantPointerNull>(Callee) || isa<UndefValue>(Callee)) { 1165cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif // If CS does not return void then replaceAllUsesWith undef. 1166753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // This allows ValueHandlers and custom metadata to adjust itself. 1167753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!CS.getInstruction()->getType()->isVoidTy()) 11683e22cb9ec30cd9b1be9b0f50e400f512124997e5Eli Friedman ReplaceInstUsesWith(*CS.getInstruction(), 11693e22cb9ec30cd9b1be9b0f50e400f512124997e5Eli Friedman UndefValue::get(CS.getInstruction()->getType())); 1170753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1171f1fb6c836940d1b92c0e3df27f4c9ca6569ff968Nuno Lopes if (isa<InvokeInst>(CS.getInstruction())) { 1172f1fb6c836940d1b92c0e3df27f4c9ca6569ff968Nuno Lopes // Can't remove an invoke because we cannot change the CFG. 1173dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return nullptr; 1174753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1175f1fb6c836940d1b92c0e3df27f4c9ca6569ff968Nuno Lopes 1176f1fb6c836940d1b92c0e3df27f4c9ca6569ff968Nuno Lopes // This instruction is not reachable, just remove it. We insert a store to 1177f1fb6c836940d1b92c0e3df27f4c9ca6569ff968Nuno Lopes // undef so that we know that this code is not reachable, despite the fact 1178f1fb6c836940d1b92c0e3df27f4c9ca6569ff968Nuno Lopes // that we can't modify the CFG here. 1179f1fb6c836940d1b92c0e3df27f4c9ca6569ff968Nuno Lopes new StoreInst(ConstantInt::getTrue(Callee->getContext()), 1180f1fb6c836940d1b92c0e3df27f4c9ca6569ff968Nuno Lopes UndefValue::get(Type::getInt1PtrTy(Callee->getContext())), 1181f1fb6c836940d1b92c0e3df27f4c9ca6569ff968Nuno Lopes CS.getInstruction()); 1182f1fb6c836940d1b92c0e3df27f4c9ca6569ff968Nuno Lopes 1183753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return EraseInstFromFunction(*CS.getInstruction()); 1184753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1185753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 11864a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands if (IntrinsicInst *II = FindInitTrampoline(Callee)) 11874a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands return transformCallThroughTrampoline(CS, II); 1188753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1189db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner PointerType *PTy = cast<PointerType>(Callee->getType()); 1190db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner FunctionType *FTy = cast<FunctionType>(PTy->getElementType()); 1191753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (FTy->isVarArg()) { 1192ba78c883d4c89bd807fa7ebea7ac6f22f3503223Eli Friedman int ix = FTy->getNumParams(); 1193753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // See if we can optimize any arguments passed through the varargs area of 1194753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // the call. 11951bde770ccd35b27e0421e54fcdde1f33cdba0f7aMatt Arsenault for (CallSite::arg_iterator I = CS.arg_begin() + FTy->getNumParams(), 1196753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner E = CS.arg_end(); I != E; ++I, ++ix) { 1197753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner CastInst *CI = dyn_cast<CastInst>(*I); 119836b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines if (CI && isSafeToEliminateVarargsCast(CS, CI, DL, ix)) { 1199753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner *I = CI->getOperand(0); 1200753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Changed = true; 1201753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1202753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1203753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1204753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1205753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (isa<InlineAsm>(Callee) && !CS.doesNotThrow()) { 1206753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Inline asm calls cannot throw - mark them 'nounwind'. 1207753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner CS.setDoesNotThrow(); 1208753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Changed = true; 1209753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1210753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 12113574eca1b02600bac4e625297f4ecf745f4c4f32Micah Villmow // Try to optimize the call if possible, we require DataLayout for most of 121227ceaa13f820e33c44e29adc2095dd3e1e69d09fEric Christopher // this. None of these calls are seen as possibly dead so go ahead and 121327ceaa13f820e33c44e29adc2095dd3e1e69d09fEric Christopher // delete the instruction now. 121427ceaa13f820e33c44e29adc2095dd3e1e69d09fEric Christopher if (CallInst *CI = dyn_cast<CallInst>(CS.getInstruction())) { 121536b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines Instruction *I = tryOptimizeCall(CI, DL); 12167b323a34fa034389e9e439f6c02eeb73e87ab9dbEric Christopher // If we changed something return the result, etc. Otherwise let 12177b323a34fa034389e9e439f6c02eeb73e87ab9dbEric Christopher // the fallthrough check. 12187b323a34fa034389e9e439f6c02eeb73e87ab9dbEric Christopher if (I) return EraseInstFromFunction(*I); 121927ceaa13f820e33c44e29adc2095dd3e1e69d09fEric Christopher } 122027ceaa13f820e33c44e29adc2095dd3e1e69d09fEric Christopher 1221dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return Changed ? CS.getInstruction() : nullptr; 1222753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner} 1223753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1224753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner// transformConstExprCastCall - If the callee is a constexpr cast of a function, 1225753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner// attempt to move the cast to the arguments of the call/invoke. 1226753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner// 1227753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattnerbool InstCombiner::transformConstExprCastCall(CallSite CS) { 1228ab215bc371677cdeac6eacb3e890e5af666d1ab4Chris Lattner Function *Callee = 1229ab215bc371677cdeac6eacb3e890e5af666d1ab4Chris Lattner dyn_cast<Function>(CS.getCalledValue()->stripPointerCasts()); 1230dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines if (!Callee) 1231753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return false; 1232753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Instruction *Caller = CS.getInstruction(); 123399faa3b4ec6d03ac7808fe4ff3fbf3d04e375502Bill Wendling const AttributeSet &CallerPAL = CS.getAttributes(); 1234753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1235753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Okay, this is a cast from a function to a different type. Unless doing so 1236753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // would cause a type conversion of one of our arguments, change this call to 1237753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // be a direct call with arguments casted to the appropriate types. 1238753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // 1239db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner FunctionType *FT = Callee->getFunctionType(); 1240db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *OldRetTy = Caller->getType(); 1241db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *NewRetTy = FT->getReturnType(); 1242753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1243753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Check to see if we are changing the return type... 1244753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (OldRetTy != NewRetTy) { 124536b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines 124636b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines if (NewRetTy->isStructTy()) 124736b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines return false; // TODO: Handle multiple return values. 124836b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines 12493e1c40de7f33db639698982bb58d8c2c8d7d5780Matt Arsenault if (!CastInst::isBitCastable(NewRetTy, OldRetTy)) { 12503e1c40de7f33db639698982bb58d8c2c8d7d5780Matt Arsenault if (Callee->isDeclaration()) 12513e1c40de7f33db639698982bb58d8c2c8d7d5780Matt Arsenault return false; // Cannot transform this return value. 1252753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 12533e1c40de7f33db639698982bb58d8c2c8d7d5780Matt Arsenault if (!Caller->use_empty() && 12543e1c40de7f33db639698982bb58d8c2c8d7d5780Matt Arsenault // void -> non-void is handled specially 12553e1c40de7f33db639698982bb58d8c2c8d7d5780Matt Arsenault !NewRetTy->isVoidTy()) 1256753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return false; // Cannot transform this return value. 12573e1c40de7f33db639698982bb58d8c2c8d7d5780Matt Arsenault } 1258753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1259753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!CallerPAL.isEmpty() && !Caller->use_empty()) { 12601b0c54f1c5dd61e56cb7cbc435fcb3319cff628fBill Wendling AttrBuilder RAttrs(CallerPAL, AttributeSet::ReturnIndex); 1261e74365462a39529ae48ef4d34ec76b4543b8ea29Bill Wendling if (RAttrs. 1262e74365462a39529ae48ef4d34ec76b4543b8ea29Bill Wendling hasAttributes(AttributeFuncs:: 1263e74365462a39529ae48ef4d34ec76b4543b8ea29Bill Wendling typeIncompatible(NewRetTy, AttributeSet::ReturnIndex), 1264e74365462a39529ae48ef4d34ec76b4543b8ea29Bill Wendling AttributeSet::ReturnIndex)) 1265753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return false; // Attribute not compatible with transformed value. 1266753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1267753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1268753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If the callsite is an invoke instruction, and the return value is used by 1269753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // a PHI node in a successor, we cannot change the return type of the call 1270753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // because there is no place to put the cast instruction (without breaking 1271753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // the critical edge). Bail out in this case. 1272753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!Caller->use_empty()) 1273753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (InvokeInst *II = dyn_cast<InvokeInst>(Caller)) 127436b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines for (User *U : II->users()) 127536b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines if (PHINode *PN = dyn_cast<PHINode>(U)) 1276753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (PN->getParent() == II->getNormalDest() || 1277753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner PN->getParent() == II->getUnwindDest()) 1278753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return false; 1279753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1280753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 12811bde770ccd35b27e0421e54fcdde1f33cdba0f7aMatt Arsenault unsigned NumActualArgs = CS.arg_size(); 1282753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner unsigned NumCommonArgs = std::min(FT->getNumParams(), NumActualArgs); 1283753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1284753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner CallSite::arg_iterator AI = CS.arg_begin(); 1285753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner for (unsigned i = 0, e = NumCommonArgs; i != e; ++i, ++AI) { 1286db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *ParamTy = FT->getParamType(i); 1287db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *ActTy = (*AI)->getType(); 1288753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 12893e1c40de7f33db639698982bb58d8c2c8d7d5780Matt Arsenault if (!CastInst::isBitCastable(ActTy, ParamTy)) 1290753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return false; // Cannot transform this parameter value. 1291753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 129228d65722d6f283b327b5815914382077fe9c0ab4Bill Wendling if (AttrBuilder(CallerPAL.getParamAttributes(i + 1), i + 1). 1293e74365462a39529ae48ef4d34ec76b4543b8ea29Bill Wendling hasAttributes(AttributeFuncs:: 1294e74365462a39529ae48ef4d34ec76b4543b8ea29Bill Wendling typeIncompatible(ParamTy, i + 1), i + 1)) 1295753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return false; // Attribute not compatible with transformed value. 129600e403abe384b8f61dd012242d3d81a23541ca25Jim Grosbach 129736b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines if (CS.isInAllocaArgument(i)) 129836b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines return false; // Cannot transform to and from inalloca. 129936b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines 13002b9375e44b0e99ce23d2cef8de78f66347eef88aChris Lattner // If the parameter is passed as a byval argument, then we have to have a 13012b9375e44b0e99ce23d2cef8de78f66347eef88aChris Lattner // sized type and the sized type has to have the same size as the old type. 130228d65722d6f283b327b5815914382077fe9c0ab4Bill Wendling if (ParamTy != ActTy && 130328d65722d6f283b327b5815914382077fe9c0ab4Bill Wendling CallerPAL.getParamAttributes(i + 1).hasAttribute(i + 1, 130428d65722d6f283b327b5815914382077fe9c0ab4Bill Wendling Attribute::ByVal)) { 1305db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner PointerType *ParamPTy = dyn_cast<PointerType>(ParamTy); 1306dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines if (!ParamPTy || !ParamPTy->getElementType()->isSized() || !DL) 13072b9375e44b0e99ce23d2cef8de78f66347eef88aChris Lattner return false; 130800e403abe384b8f61dd012242d3d81a23541ca25Jim Grosbach 1309b060a46b0de75d6ab711b60d123ab42a9a6d7b3cMatt Arsenault Type *CurElTy = ActTy->getPointerElementType(); 131036b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines if (DL->getTypeAllocSize(CurElTy) != 131136b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines DL->getTypeAllocSize(ParamPTy->getElementType())) 13122b9375e44b0e99ce23d2cef8de78f66347eef88aChris Lattner return false; 13132b9375e44b0e99ce23d2cef8de78f66347eef88aChris Lattner } 1314753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1315753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1316091b1e3c74fee17e33aa87478ab07bfbba0a2f78Chris Lattner if (Callee->isDeclaration()) { 1317091b1e3c74fee17e33aa87478ab07bfbba0a2f78Chris Lattner // Do not delete arguments unless we have a function body. 1318091b1e3c74fee17e33aa87478ab07bfbba0a2f78Chris Lattner if (FT->getNumParams() < NumActualArgs && !FT->isVarArg()) 1319091b1e3c74fee17e33aa87478ab07bfbba0a2f78Chris Lattner return false; 1320091b1e3c74fee17e33aa87478ab07bfbba0a2f78Chris Lattner 1321091b1e3c74fee17e33aa87478ab07bfbba0a2f78Chris Lattner // If the callee is just a declaration, don't change the varargsness of the 1322091b1e3c74fee17e33aa87478ab07bfbba0a2f78Chris Lattner // call. We don't want to introduce a varargs call where one doesn't 1323091b1e3c74fee17e33aa87478ab07bfbba0a2f78Chris Lattner // already exist. 1324db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner PointerType *APTy = cast<PointerType>(CS.getCalledValue()->getType()); 1325091b1e3c74fee17e33aa87478ab07bfbba0a2f78Chris Lattner if (FT->isVarArg()!=cast<FunctionType>(APTy->getElementType())->isVarArg()) 1326091b1e3c74fee17e33aa87478ab07bfbba0a2f78Chris Lattner return false; 1327f374486659822e93054363bef05e781acee16f3bJim Grosbach 1328f374486659822e93054363bef05e781acee16f3bJim Grosbach // If both the callee and the cast type are varargs, we still have to make 1329f374486659822e93054363bef05e781acee16f3bJim Grosbach // sure the number of fixed parameters are the same or we have the same 1330f374486659822e93054363bef05e781acee16f3bJim Grosbach // ABI issues as if we introduce a varargs call. 1331871a2051f7d49827259d719a97b8cc86163edbd9Jim Grosbach if (FT->isVarArg() && 1332871a2051f7d49827259d719a97b8cc86163edbd9Jim Grosbach cast<FunctionType>(APTy->getElementType())->isVarArg() && 1333871a2051f7d49827259d719a97b8cc86163edbd9Jim Grosbach FT->getNumParams() != 1334f374486659822e93054363bef05e781acee16f3bJim Grosbach cast<FunctionType>(APTy->getElementType())->getNumParams()) 1335f374486659822e93054363bef05e781acee16f3bJim Grosbach return false; 1336091b1e3c74fee17e33aa87478ab07bfbba0a2f78Chris Lattner } 133700e403abe384b8f61dd012242d3d81a23541ca25Jim Grosbach 1338d5917f0b4d8f87985bfdf130c269e1929d0085d1Jim Grosbach if (FT->getNumParams() < NumActualArgs && FT->isVarArg() && 1339d5917f0b4d8f87985bfdf130c269e1929d0085d1Jim Grosbach !CallerPAL.isEmpty()) 1340d5917f0b4d8f87985bfdf130c269e1929d0085d1Jim Grosbach // In this case we have more arguments than the new function type, but we 1341d5917f0b4d8f87985bfdf130c269e1929d0085d1Jim Grosbach // won't be dropping them. Check that these extra arguments have attributes 1342d5917f0b4d8f87985bfdf130c269e1929d0085d1Jim Grosbach // that are compatible with being a vararg call argument. 1343d5917f0b4d8f87985bfdf130c269e1929d0085d1Jim Grosbach for (unsigned i = CallerPAL.getNumSlots(); i; --i) { 13448e47daf2858e980210f3e1f007036b24da342c29Bill Wendling unsigned Index = CallerPAL.getSlotIndex(i - 1); 13458e47daf2858e980210f3e1f007036b24da342c29Bill Wendling if (Index <= FT->getNumParams()) 1346d5917f0b4d8f87985bfdf130c269e1929d0085d1Jim Grosbach break; 13478e47daf2858e980210f3e1f007036b24da342c29Bill Wendling 13481d3dcfe4246b4d45fa78a8dfd0a11c7fff842c15Bill Wendling // Check if it has an attribute that's incompatible with varargs. 13498e47daf2858e980210f3e1f007036b24da342c29Bill Wendling AttributeSet PAttrs = CallerPAL.getSlotAttributes(i - 1); 13508e47daf2858e980210f3e1f007036b24da342c29Bill Wendling if (PAttrs.hasAttribute(Index, Attribute::StructRet)) 1351d5917f0b4d8f87985bfdf130c269e1929d0085d1Jim Grosbach return false; 1352d5917f0b4d8f87985bfdf130c269e1929d0085d1Jim Grosbach } 1353753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 135400e403abe384b8f61dd012242d3d81a23541ca25Jim Grosbach 1355753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Okay, we decided that this is a safe thing to do: go ahead and start 1356091b1e3c74fee17e33aa87478ab07bfbba0a2f78Chris Lattner // inserting cast instructions as necessary. 1357753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner std::vector<Value*> Args; 1358753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Args.reserve(NumActualArgs); 1359ac90696722bef3993e5fb9f72a6fa4209bbc8763Bill Wendling SmallVector<AttributeSet, 8> attrVec; 1360753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner attrVec.reserve(NumCommonArgs); 1361753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1362753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Get any return attributes. 13631b0c54f1c5dd61e56cb7cbc435fcb3319cff628fBill Wendling AttrBuilder RAttrs(CallerPAL, AttributeSet::ReturnIndex); 1364753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1365753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If the return value is not being used, the type may not be compatible 1366753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // with the existing attributes. Wipe out any problematic attributes. 1367e74365462a39529ae48ef4d34ec76b4543b8ea29Bill Wendling RAttrs. 1368e74365462a39529ae48ef4d34ec76b4543b8ea29Bill Wendling removeAttributes(AttributeFuncs:: 1369e74365462a39529ae48ef4d34ec76b4543b8ea29Bill Wendling typeIncompatible(NewRetTy, AttributeSet::ReturnIndex), 1370e74365462a39529ae48ef4d34ec76b4543b8ea29Bill Wendling AttributeSet::ReturnIndex); 1371753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1372753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Add the new return attributes. 13738831c0605bbc0c82ce56c2fb85bd681d1c013925Bill Wendling if (RAttrs.hasAttributes()) 1374ac90696722bef3993e5fb9f72a6fa4209bbc8763Bill Wendling attrVec.push_back(AttributeSet::get(Caller->getContext(), 1375ac90696722bef3993e5fb9f72a6fa4209bbc8763Bill Wendling AttributeSet::ReturnIndex, RAttrs)); 1376753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1377753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner AI = CS.arg_begin(); 1378753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner for (unsigned i = 0; i != NumCommonArgs; ++i, ++AI) { 1379db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *ParamTy = FT->getParamType(i); 1380f34dc428fa577d6d5d71ab3a1f9765b4e5da5a4fMatt Arsenault 1381753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if ((*AI)->getType() == ParamTy) { 1382753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Args.push_back(*AI); 1383753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } else { 1384f34dc428fa577d6d5d71ab3a1f9765b4e5da5a4fMatt Arsenault Args.push_back(Builder->CreateBitCast(*AI, ParamTy)); 1385753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1386753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1387753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Add any parameter attributes. 138828d65722d6f283b327b5815914382077fe9c0ab4Bill Wendling AttrBuilder PAttrs(CallerPAL.getParamAttributes(i + 1), i + 1); 13897be7848e17f60825f5fbc177b8a25909a30ddb00Bill Wendling if (PAttrs.hasAttributes()) 1390ac90696722bef3993e5fb9f72a6fa4209bbc8763Bill Wendling attrVec.push_back(AttributeSet::get(Caller->getContext(), i + 1, 1391ac90696722bef3993e5fb9f72a6fa4209bbc8763Bill Wendling PAttrs)); 1392753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1393753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1394753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If the function takes more arguments than the call was taking, add them 1395753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // now. 1396753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner for (unsigned i = NumCommonArgs; i != FT->getNumParams(); ++i) 1397753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Args.push_back(Constant::getNullValue(FT->getParamType(i))); 1398753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1399753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If we are removing arguments to the function, emit an obnoxious warning. 1400753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (FT->getNumParams() < NumActualArgs) { 14011dec62ed43fef82b1cafbcc1fc3d63cacd90cad9Nick Lewycky // TODO: if (!FT->isVarArg()) this call may be unreachable. PR14722 14021dec62ed43fef82b1cafbcc1fc3d63cacd90cad9Nick Lewycky if (FT->isVarArg()) { 1403753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Add all of the arguments in their promoted form to the arg list. 1404753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner for (unsigned i = FT->getNumParams(); i != NumActualArgs; ++i, ++AI) { 1405db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *PTy = getPromotedType((*AI)->getType()); 1406753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (PTy != (*AI)->getType()) { 1407753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Must promote to pass through va_arg area! 1408753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Instruction::CastOps opcode = 1409753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner CastInst::getCastOpcode(*AI, false, PTy, false); 1410a9390a4d5f5d568059a80970d22194b165d097a7Benjamin Kramer Args.push_back(Builder->CreateCast(opcode, *AI, PTy)); 1411753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } else { 1412753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Args.push_back(*AI); 1413753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1414753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1415753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Add any parameter attributes. 141628d65722d6f283b327b5815914382077fe9c0ab4Bill Wendling AttrBuilder PAttrs(CallerPAL.getParamAttributes(i + 1), i + 1); 14177be7848e17f60825f5fbc177b8a25909a30ddb00Bill Wendling if (PAttrs.hasAttributes()) 1418ac90696722bef3993e5fb9f72a6fa4209bbc8763Bill Wendling attrVec.push_back(AttributeSet::get(FT->getContext(), i + 1, 1419ac90696722bef3993e5fb9f72a6fa4209bbc8763Bill Wendling PAttrs)); 1420753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1421753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1422753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1423753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1424c5f1bc88a2eb7ad9ff924ca90cf88494e5f947b9Bill Wendling AttributeSet FnAttrs = CallerPAL.getFnAttributes(); 1425956f13440a4aa0297606a4412f4aa091d931592aBill Wendling if (CallerPAL.hasAttributes(AttributeSet::FunctionIndex)) 1426ac90696722bef3993e5fb9f72a6fa4209bbc8763Bill Wendling attrVec.push_back(AttributeSet::get(Callee->getContext(), FnAttrs)); 1427753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1428753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (NewRetTy->isVoidTy()) 1429753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Caller->setName(""); // Void type should not have a name. 1430753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 143199faa3b4ec6d03ac7808fe4ff3fbf3d04e375502Bill Wendling const AttributeSet &NewCallerPAL = AttributeSet::get(Callee->getContext(), 1432c5f1bc88a2eb7ad9ff924ca90cf88494e5f947b9Bill Wendling attrVec); 1433753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1434753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Instruction *NC; 1435753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (InvokeInst *II = dyn_cast<InvokeInst>(Caller)) { 1436ef819d0ed8c9df3da633e8c1240e582a7879bb94Eli Friedman NC = Builder->CreateInvoke(Callee, II->getNormalDest(), 1437a3efbb15ddd5aa9006564cd79086723640084878Jay Foad II->getUnwindDest(), Args); 1438ef819d0ed8c9df3da633e8c1240e582a7879bb94Eli Friedman NC->takeName(II); 1439753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner cast<InvokeInst>(NC)->setCallingConv(II->getCallingConv()); 1440753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner cast<InvokeInst>(NC)->setAttributes(NewCallerPAL); 1441753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } else { 1442753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner CallInst *CI = cast<CallInst>(Caller); 1443a3efbb15ddd5aa9006564cd79086723640084878Jay Foad NC = Builder->CreateCall(Callee, Args); 1444ef819d0ed8c9df3da633e8c1240e582a7879bb94Eli Friedman NC->takeName(CI); 1445753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (CI->isTailCall()) 1446753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner cast<CallInst>(NC)->setTailCall(); 1447753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner cast<CallInst>(NC)->setCallingConv(CI->getCallingConv()); 1448753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner cast<CallInst>(NC)->setAttributes(NewCallerPAL); 1449753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1450753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1451753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Insert a cast of the return type as necessary. 1452753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Value *NV = NC; 1453753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (OldRetTy != NV->getType() && !Caller->use_empty()) { 1454753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!NV->getType()->isVoidTy()) { 1455f34dc428fa577d6d5d71ab3a1f9765b4e5da5a4fMatt Arsenault NV = NC = CastInst::Create(CastInst::BitCast, NC, OldRetTy); 1456a311c34d2af7c750f016ef5e4c41bee77a1dfac7Eli Friedman NC->setDebugLoc(Caller->getDebugLoc()); 1457753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1458753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If this is an invoke instruction, we should insert it after the first 1459753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // non-phi, instruction in the normal successor block. 1460753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (InvokeInst *II = dyn_cast<InvokeInst>(Caller)) { 146189d4411cef736898047aa7e3bc159da39cacf8e6Bill Wendling BasicBlock::iterator I = II->getNormalDest()->getFirstInsertionPt(); 1462753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner InsertNewInstBefore(NC, *I); 1463753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } else { 1464ab215bc371677cdeac6eacb3e890e5af666d1ab4Chris Lattner // Otherwise, it's a call, just insert cast right after the call. 1465753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner InsertNewInstBefore(NC, *Caller); 1466753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1467753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Worklist.AddUsersToWorkList(*Caller); 1468753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } else { 1469753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NV = UndefValue::get(Caller->getType()); 1470753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1471753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1472753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1473753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!Caller->use_empty()) 14743e22cb9ec30cd9b1be9b0f50e400f512124997e5Eli Friedman ReplaceInstUsesWith(*Caller, NV); 147536b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines else if (Caller->hasValueHandle()) 147636b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines ValueHandleBase::ValueIsRAUWd(Caller, NV); 14773e22cb9ec30cd9b1be9b0f50e400f512124997e5Eli Friedman 1478753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner EraseInstFromFunction(*Caller); 1479753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return true; 1480753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner} 1481753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 14824a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands// transformCallThroughTrampoline - Turn a call to a function created by 14834a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands// init_trampoline / adjust_trampoline intrinsic pair into a direct call to the 14844a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands// underlying function. 1485753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner// 14864a544a79bd735967f1d33fe675ae4566dbd17813Duncan SandsInstruction * 14874a544a79bd735967f1d33fe675ae4566dbd17813Duncan SandsInstCombiner::transformCallThroughTrampoline(CallSite CS, 14884a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands IntrinsicInst *Tramp) { 1489753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Value *Callee = CS.getCalledValue(); 1490db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner PointerType *PTy = cast<PointerType>(Callee->getType()); 1491db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner FunctionType *FTy = cast<FunctionType>(PTy->getElementType()); 149299faa3b4ec6d03ac7808fe4ff3fbf3d04e375502Bill Wendling const AttributeSet &Attrs = CS.getAttributes(); 1493753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1494753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If the call already has the 'nest' attribute somewhere then give up - 1495753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // otherwise 'nest' would occur twice after splicing in the chain. 14969d30e7208e6b2bc3fa48305e3ae371188f643425Bill Wendling if (Attrs.hasAttrSomewhere(Attribute::Nest)) 1497dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return nullptr; 1498753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 14994a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands assert(Tramp && 15004a544a79bd735967f1d33fe675ae4566dbd17813Duncan Sands "transformCallThroughTrampoline called with incorrect CallSite."); 1501753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1502a399781289092fcdceb58b21174229f4373c4191Gabor Greif Function *NestF =cast<Function>(Tramp->getArgOperand(1)->stripPointerCasts()); 1503db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner PointerType *NestFPTy = cast<PointerType>(NestF->getType()); 1504db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner FunctionType *NestFTy = cast<FunctionType>(NestFPTy->getElementType()); 1505753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 150699faa3b4ec6d03ac7808fe4ff3fbf3d04e375502Bill Wendling const AttributeSet &NestAttrs = NestF->getAttributes(); 1507753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!NestAttrs.isEmpty()) { 1508753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner unsigned NestIdx = 1; 1509dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines Type *NestTy = nullptr; 151028d65722d6f283b327b5815914382077fe9c0ab4Bill Wendling AttributeSet NestAttr; 1511753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1512753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Look for a parameter marked with the 'nest' attribute. 1513753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner for (FunctionType::param_iterator I = NestFTy->param_begin(), 1514753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner E = NestFTy->param_end(); I != E; ++NestIdx, ++I) 151528d65722d6f283b327b5815914382077fe9c0ab4Bill Wendling if (NestAttrs.hasAttribute(NestIdx, Attribute::Nest)) { 1516753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Record the parameter type and any other attributes. 1517753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NestTy = *I; 1518753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NestAttr = NestAttrs.getParamAttributes(NestIdx); 1519753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 1520753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1521753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1522753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (NestTy) { 1523753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Instruction *Caller = CS.getInstruction(); 1524753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner std::vector<Value*> NewArgs; 15251bde770ccd35b27e0421e54fcdde1f33cdba0f7aMatt Arsenault NewArgs.reserve(CS.arg_size() + 1); 1526753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1527ac90696722bef3993e5fb9f72a6fa4209bbc8763Bill Wendling SmallVector<AttributeSet, 8> NewAttrs; 1528753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NewAttrs.reserve(Attrs.getNumSlots() + 1); 1529753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1530753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Insert the nest argument into the call argument list, which may 1531753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // mean appending it. Likewise for attributes. 1532753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1533753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Add any result attributes. 15341b0c54f1c5dd61e56cb7cbc435fcb3319cff628fBill Wendling if (Attrs.hasAttributes(AttributeSet::ReturnIndex)) 1535ac90696722bef3993e5fb9f72a6fa4209bbc8763Bill Wendling NewAttrs.push_back(AttributeSet::get(Caller->getContext(), 1536ac90696722bef3993e5fb9f72a6fa4209bbc8763Bill Wendling Attrs.getRetAttributes())); 1537753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1538753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner { 1539753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner unsigned Idx = 1; 1540753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner CallSite::arg_iterator I = CS.arg_begin(), E = CS.arg_end(); 1541753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner do { 1542753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Idx == NestIdx) { 1543753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Add the chain argument and attributes. 1544cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif Value *NestVal = Tramp->getArgOperand(2); 1545753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (NestVal->getType() != NestTy) 1546e6f364b6c44eda14cd4ad54366ea5cc7246b9500Eli Friedman NestVal = Builder->CreateBitCast(NestVal, NestTy, "nest"); 1547753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NewArgs.push_back(NestVal); 1548ac90696722bef3993e5fb9f72a6fa4209bbc8763Bill Wendling NewAttrs.push_back(AttributeSet::get(Caller->getContext(), 1549ac90696722bef3993e5fb9f72a6fa4209bbc8763Bill Wendling NestAttr)); 1550753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1551753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1552753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (I == E) 1553753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 1554753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1555753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Add the original argument and attributes. 1556753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NewArgs.push_back(*I); 155728d65722d6f283b327b5815914382077fe9c0ab4Bill Wendling AttributeSet Attr = Attrs.getParamAttributes(Idx); 155828d65722d6f283b327b5815914382077fe9c0ab4Bill Wendling if (Attr.hasAttributes(Idx)) { 1559ac90696722bef3993e5fb9f72a6fa4209bbc8763Bill Wendling AttrBuilder B(Attr, Idx); 1560ac90696722bef3993e5fb9f72a6fa4209bbc8763Bill Wendling NewAttrs.push_back(AttributeSet::get(Caller->getContext(), 1561ac90696722bef3993e5fb9f72a6fa4209bbc8763Bill Wendling Idx + (Idx >= NestIdx), B)); 156228d65722d6f283b327b5815914382077fe9c0ab4Bill Wendling } 1563753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1564753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ++Idx, ++I; 1565753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } while (1); 1566753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1567753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1568753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Add any function attributes. 1569956f13440a4aa0297606a4412f4aa091d931592aBill Wendling if (Attrs.hasAttributes(AttributeSet::FunctionIndex)) 1570ac90696722bef3993e5fb9f72a6fa4209bbc8763Bill Wendling NewAttrs.push_back(AttributeSet::get(FTy->getContext(), 1571ac90696722bef3993e5fb9f72a6fa4209bbc8763Bill Wendling Attrs.getFnAttributes())); 1572753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1573753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // The trampoline may have been bitcast to a bogus type (FTy). 1574753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Handle this by synthesizing a new function type, equal to FTy 1575753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // with the chain parameter inserted. 1576753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 15775fdd6c8793462549e3593890ec61573da06e3346Jay Foad std::vector<Type*> NewTypes; 1578753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NewTypes.reserve(FTy->getNumParams()+1); 1579753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1580753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Insert the chain's type into the list of parameter types, which may 1581753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // mean appending it. 1582753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner { 1583753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner unsigned Idx = 1; 1584753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner FunctionType::param_iterator I = FTy->param_begin(), 1585753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner E = FTy->param_end(); 1586753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1587753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner do { 1588753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Idx == NestIdx) 1589753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Add the chain's type. 1590753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NewTypes.push_back(NestTy); 1591753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1592753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (I == E) 1593753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 1594753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1595753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Add the original type. 1596753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NewTypes.push_back(*I); 1597753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1598753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ++Idx, ++I; 1599753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } while (1); 1600753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1601753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1602753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Replace the trampoline call with a direct call. Let the generic 1603753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // code sort out any function type mismatches. 160400e403abe384b8f61dd012242d3d81a23541ca25Jim Grosbach FunctionType *NewFTy = FunctionType::get(FTy->getReturnType(), NewTypes, 1605753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner FTy->isVarArg()); 1606753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Constant *NewCallee = 1607753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NestF->getType() == PointerType::getUnqual(NewFTy) ? 160800e403abe384b8f61dd012242d3d81a23541ca25Jim Grosbach NestF : ConstantExpr::getBitCast(NestF, 1609753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner PointerType::getUnqual(NewFTy)); 161003fceff6f69a0261a767aab8e62de8aa9301b86cJim Grosbach const AttributeSet &NewPAL = 161103fceff6f69a0261a767aab8e62de8aa9301b86cJim Grosbach AttributeSet::get(FTy->getContext(), NewAttrs); 1612753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1613753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Instruction *NewCaller; 1614753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (InvokeInst *II = dyn_cast<InvokeInst>(Caller)) { 1615753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NewCaller = InvokeInst::Create(NewCallee, 1616753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner II->getNormalDest(), II->getUnwindDest(), 1617a3efbb15ddd5aa9006564cd79086723640084878Jay Foad NewArgs); 1618753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner cast<InvokeInst>(NewCaller)->setCallingConv(II->getCallingConv()); 1619753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner cast<InvokeInst>(NewCaller)->setAttributes(NewPAL); 1620753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } else { 1621a3efbb15ddd5aa9006564cd79086723640084878Jay Foad NewCaller = CallInst::Create(NewCallee, NewArgs); 1622753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (cast<CallInst>(Caller)->isTailCall()) 1623753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner cast<CallInst>(NewCaller)->setTailCall(); 1624753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner cast<CallInst>(NewCaller)-> 1625753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner setCallingConv(cast<CallInst>(Caller)->getCallingConv()); 1626753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner cast<CallInst>(NewCaller)->setAttributes(NewPAL); 1627753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 162859f15918c28ad340bb9f0ef78e43c2952a5b5c1cEli Friedman 162959f15918c28ad340bb9f0ef78e43c2952a5b5c1cEli Friedman return NewCaller; 1630753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1631753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1632753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1633753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Replace the trampoline call with a direct call. Since there is no 'nest' 1634753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // parameter, there is no need to adjust the argument list. Let the generic 1635753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // code sort out any function type mismatches. 1636753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Constant *NewCallee = 163700e403abe384b8f61dd012242d3d81a23541ca25Jim Grosbach NestF->getType() == PTy ? NestF : 1638753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ConstantExpr::getBitCast(NestF, PTy); 1639753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner CS.setCalledFunction(NewCallee); 1640753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return CS.getInstruction(); 1641753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner} 1642