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