1 //===- InstCombineCasts.cpp -----------------------------------------------===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 //
10 // This file implements the visit functions for cast operations.
11 //
12 //===----------------------------------------------------------------------===//
13 
14 #include "InstCombineInternal.h"
15 #include "llvm/ADT/SetVector.h"
16 #include "llvm/Analysis/ConstantFolding.h"
17 #include "llvm/Analysis/TargetLibraryInfo.h"
18 #include "llvm/IR/DataLayout.h"
19 #include "llvm/IR/PatternMatch.h"
20 #include "llvm/Support/KnownBits.h"
21 using namespace llvm;
22 using namespace PatternMatch;
23 
24 #define DEBUG_TYPE "instcombine"
25 
26 /// Analyze 'Val', seeing if it is a simple linear expression.
27 /// If so, decompose it, returning some value X, such that Val is
28 /// X*Scale+Offset.
29 ///
30 static Value *decomposeSimpleLinearExpr(Value *Val, unsigned &Scale,
31                                         uint64_t &Offset) {
32   if (ConstantInt *CI = dyn_cast<ConstantInt>(Val)) {
33     Offset = CI->getZExtValue();
34     Scale  = 0;
35     return ConstantInt::get(Val->getType(), 0);
36   }
37 
38   if (BinaryOperator *I = dyn_cast<BinaryOperator>(Val)) {
39     // Cannot look past anything that might overflow.
40     OverflowingBinaryOperator *OBI = dyn_cast<OverflowingBinaryOperator>(Val);
41     if (OBI && !OBI->hasNoUnsignedWrap() && !OBI->hasNoSignedWrap()) {
42       Scale = 1;
43       Offset = 0;
44       return Val;
45     }
46 
47     if (ConstantInt *RHS = dyn_cast<ConstantInt>(I->getOperand(1))) {
48       if (I->getOpcode() == Instruction::Shl) {
49         // This is a value scaled by '1 << the shift amt'.
50         Scale = UINT64_C(1) << RHS->getZExtValue();
51         Offset = 0;
52         return I->getOperand(0);
53       }
54 
55       if (I->getOpcode() == Instruction::Mul) {
56         // This value is scaled by 'RHS'.
57         Scale = RHS->getZExtValue();
58         Offset = 0;
59         return I->getOperand(0);
60       }
61 
62       if (I->getOpcode() == Instruction::Add) {
63         // We have X+C.  Check to see if we really have (X*C2)+C1,
64         // where C1 is divisible by C2.
65         unsigned SubScale;
66         Value *SubVal =
67           decomposeSimpleLinearExpr(I->getOperand(0), SubScale, Offset);
68         Offset += RHS->getZExtValue();
69         Scale = SubScale;
70         return SubVal;
71       }
72     }
73   }
74 
75   // Otherwise, we can't look past this.
76   Scale = 1;
77   Offset = 0;
78   return Val;
79 }
80 
81 /// If we find a cast of an allocation instruction, try to eliminate the cast by
82 /// moving the type information into the alloc.
83 Instruction *InstCombiner::PromoteCastOfAllocation(BitCastInst &CI,
84                                                    AllocaInst &AI) {
85   PointerType *PTy = cast<PointerType>(CI.getType());
86 
87   BuilderTy AllocaBuilder(Builder);
88   AllocaBuilder.SetInsertPoint(&AI);
89 
90   // Get the type really allocated and the type casted to.
91   Type *AllocElTy = AI.getAllocatedType();
92   Type *CastElTy = PTy->getElementType();
93   if (!AllocElTy->isSized() || !CastElTy->isSized()) return nullptr;
94 
95   unsigned AllocElTyAlign = DL.getABITypeAlignment(AllocElTy);
96   unsigned CastElTyAlign = DL.getABITypeAlignment(CastElTy);
97   if (CastElTyAlign < AllocElTyAlign) return nullptr;
98 
99   // If the allocation has multiple uses, only promote it if we are strictly
100   // increasing the alignment of the resultant allocation.  If we keep it the
101   // same, we open the door to infinite loops of various kinds.
102   if (!AI.hasOneUse() && CastElTyAlign == AllocElTyAlign) return nullptr;
103 
104   uint64_t AllocElTySize = DL.getTypeAllocSize(AllocElTy);
105   uint64_t CastElTySize = DL.getTypeAllocSize(CastElTy);
106   if (CastElTySize == 0 || AllocElTySize == 0) return nullptr;
107 
108   // If the allocation has multiple uses, only promote it if we're not
109   // shrinking the amount of memory being allocated.
110   uint64_t AllocElTyStoreSize = DL.getTypeStoreSize(AllocElTy);
111   uint64_t CastElTyStoreSize = DL.getTypeStoreSize(CastElTy);
112   if (!AI.hasOneUse() && CastElTyStoreSize < AllocElTyStoreSize) return nullptr;
113 
114   // See if we can satisfy the modulus by pulling a scale out of the array
115   // size argument.
116   unsigned ArraySizeScale;
117   uint64_t ArrayOffset;
118   Value *NumElements = // See if the array size is a decomposable linear expr.
119     decomposeSimpleLinearExpr(AI.getOperand(0), ArraySizeScale, ArrayOffset);
120 
121   // If we can now satisfy the modulus, by using a non-1 scale, we really can
122   // do the xform.
123   if ((AllocElTySize*ArraySizeScale) % CastElTySize != 0 ||
124       (AllocElTySize*ArrayOffset   ) % CastElTySize != 0) return nullptr;
125 
126   unsigned Scale = (AllocElTySize*ArraySizeScale)/CastElTySize;
127   Value *Amt = nullptr;
128   if (Scale == 1) {
129     Amt = NumElements;
130   } else {
131     Amt = ConstantInt::get(AI.getArraySize()->getType(), Scale);
132     // Insert before the alloca, not before the cast.
133     Amt = AllocaBuilder.CreateMul(Amt, NumElements);
134   }
135 
136   if (uint64_t Offset = (AllocElTySize*ArrayOffset)/CastElTySize) {
137     Value *Off = ConstantInt::get(AI.getArraySize()->getType(),
138                                   Offset, true);
139     Amt = AllocaBuilder.CreateAdd(Amt, Off);
140   }
141 
142   AllocaInst *New = AllocaBuilder.CreateAlloca(CastElTy, Amt);
143   New->setAlignment(AI.getAlignment());
144   New->takeName(&AI);
145   New->setUsedWithInAlloca(AI.isUsedWithInAlloca());
146 
147   // If the allocation has multiple real uses, insert a cast and change all
148   // things that used it to use the new cast.  This will also hack on CI, but it
149   // will die soon.
150   if (!AI.hasOneUse()) {
151     // New is the allocation instruction, pointer typed. AI is the original
152     // allocation instruction, also pointer typed. Thus, cast to use is BitCast.
153     Value *NewCast = AllocaBuilder.CreateBitCast(New, AI.getType(), "tmpcast");
154     replaceInstUsesWith(AI, NewCast);
155   }
156   return replaceInstUsesWith(CI, New);
157 }
158 
159 /// Given an expression that CanEvaluateTruncated or CanEvaluateSExtd returns
160 /// true for, actually insert the code to evaluate the expression.
161 Value *InstCombiner::EvaluateInDifferentType(Value *V, Type *Ty,
162                                              bool isSigned) {
163   if (Constant *C = dyn_cast<Constant>(V)) {
164     C = ConstantExpr::getIntegerCast(C, Ty, isSigned /*Sext or ZExt*/);
165     // If we got a constantexpr back, try to simplify it with DL info.
166     if (Constant *FoldedC = ConstantFoldConstant(C, DL, &TLI))
167       C = FoldedC;
168     return C;
169   }
170 
171   // Otherwise, it must be an instruction.
172   Instruction *I = cast<Instruction>(V);
173   Instruction *Res = nullptr;
174   unsigned Opc = I->getOpcode();
175   switch (Opc) {
176   case Instruction::Add:
177   case Instruction::Sub:
178   case Instruction::Mul:
179   case Instruction::And:
180   case Instruction::Or:
181   case Instruction::Xor:
182   case Instruction::AShr:
183   case Instruction::LShr:
184   case Instruction::Shl:
185   case Instruction::UDiv:
186   case Instruction::URem: {
187     Value *LHS = EvaluateInDifferentType(I->getOperand(0), Ty, isSigned);
188     Value *RHS = EvaluateInDifferentType(I->getOperand(1), Ty, isSigned);
189     Res = BinaryOperator::Create((Instruction::BinaryOps)Opc, LHS, RHS);
190     break;
191   }
192   case Instruction::Trunc:
193   case Instruction::ZExt:
194   case Instruction::SExt:
195     // If the source type of the cast is the type we're trying for then we can
196     // just return the source.  There's no need to insert it because it is not
197     // new.
198     if (I->getOperand(0)->getType() == Ty)
199       return I->getOperand(0);
200 
201     // Otherwise, must be the same type of cast, so just reinsert a new one.
202     // This also handles the case of zext(trunc(x)) -> zext(x).
203     Res = CastInst::CreateIntegerCast(I->getOperand(0), Ty,
204                                       Opc == Instruction::SExt);
205     break;
206   case Instruction::Select: {
207     Value *True = EvaluateInDifferentType(I->getOperand(1), Ty, isSigned);
208     Value *False = EvaluateInDifferentType(I->getOperand(2), Ty, isSigned);
209     Res = SelectInst::Create(I->getOperand(0), True, False);
210     break;
211   }
212   case Instruction::PHI: {
213     PHINode *OPN = cast<PHINode>(I);
214     PHINode *NPN = PHINode::Create(Ty, OPN->getNumIncomingValues());
215     for (unsigned i = 0, e = OPN->getNumIncomingValues(); i != e; ++i) {
216       Value *V =
217           EvaluateInDifferentType(OPN->getIncomingValue(i), Ty, isSigned);
218       NPN->addIncoming(V, OPN->getIncomingBlock(i));
219     }
220     Res = NPN;
221     break;
222   }
223   default:
224     // TODO: Can handle more cases here.
225     llvm_unreachable("Unreachable!");
226   }
227 
228   Res->takeName(I);
229   return InsertNewInstWith(Res, *I);
230 }
231 
232 Instruction::CastOps InstCombiner::isEliminableCastPair(const CastInst *CI1,
233                                                         const CastInst *CI2) {
234   Type *SrcTy = CI1->getSrcTy();
235   Type *MidTy = CI1->getDestTy();
236   Type *DstTy = CI2->getDestTy();
237 
238   Instruction::CastOps firstOp = CI1->getOpcode();
239   Instruction::CastOps secondOp = CI2->getOpcode();
240   Type *SrcIntPtrTy =
241       SrcTy->isPtrOrPtrVectorTy() ? DL.getIntPtrType(SrcTy) : nullptr;
242   Type *MidIntPtrTy =
243       MidTy->isPtrOrPtrVectorTy() ? DL.getIntPtrType(MidTy) : nullptr;
244   Type *DstIntPtrTy =
245       DstTy->isPtrOrPtrVectorTy() ? DL.getIntPtrType(DstTy) : nullptr;
246   unsigned Res = CastInst::isEliminableCastPair(firstOp, secondOp, SrcTy, MidTy,
247                                                 DstTy, SrcIntPtrTy, MidIntPtrTy,
248                                                 DstIntPtrTy);
249 
250   // We don't want to form an inttoptr or ptrtoint that converts to an integer
251   // type that differs from the pointer size.
252   if ((Res == Instruction::IntToPtr && SrcTy != DstIntPtrTy) ||
253       (Res == Instruction::PtrToInt && DstTy != SrcIntPtrTy))
254     Res = 0;
255 
256   return Instruction::CastOps(Res);
257 }
258 
259 /// @brief Implement the transforms common to all CastInst visitors.
260 Instruction *InstCombiner::commonCastTransforms(CastInst &CI) {
261   Value *Src = CI.getOperand(0);
262 
263   // Try to eliminate a cast of a cast.
264   if (auto *CSrc = dyn_cast<CastInst>(Src)) {   // A->B->C cast
265     if (Instruction::CastOps NewOpc = isEliminableCastPair(CSrc, &CI)) {
266       // The first cast (CSrc) is eliminable so we need to fix up or replace
267       // the second cast (CI). CSrc will then have a good chance of being dead.
268       return CastInst::Create(NewOpc, CSrc->getOperand(0), CI.getType());
269     }
270   }
271 
272   // If we are casting a select, then fold the cast into the select.
273   if (auto *SI = dyn_cast<SelectInst>(Src))
274     if (Instruction *NV = FoldOpIntoSelect(CI, SI))
275       return NV;
276 
277   // If we are casting a PHI, then fold the cast into the PHI.
278   if (auto *PN = dyn_cast<PHINode>(Src)) {
279     // Don't do this if it would create a PHI node with an illegal type from a
280     // legal type.
281     if (!Src->getType()->isIntegerTy() || !CI.getType()->isIntegerTy() ||
282         shouldChangeType(CI.getType(), Src->getType()))
283       if (Instruction *NV = foldOpIntoPhi(CI, PN))
284         return NV;
285   }
286 
287   return nullptr;
288 }
289 
290 /// Return true if we can evaluate the specified expression tree as type Ty
291 /// instead of its larger type, and arrive with the same value.
292 /// This is used by code that tries to eliminate truncates.
293 ///
294 /// Ty will always be a type smaller than V.  We should return true if trunc(V)
295 /// can be computed by computing V in the smaller type.  If V is an instruction,
296 /// then trunc(inst(x,y)) can be computed as inst(trunc(x),trunc(y)), which only
297 /// makes sense if x and y can be efficiently truncated.
298 ///
299 /// This function works on both vectors and scalars.
300 ///
301 static bool canEvaluateTruncated(Value *V, Type *Ty, InstCombiner &IC,
302                                  Instruction *CxtI) {
303   // We can always evaluate constants in another type.
304   if (isa<Constant>(V))
305     return true;
306 
307   Instruction *I = dyn_cast<Instruction>(V);
308   if (!I) return false;
309 
310   Type *OrigTy = V->getType();
311 
312   // If this is an extension from the dest type, we can eliminate it, even if it
313   // has multiple uses.
314   if ((isa<ZExtInst>(I) || isa<SExtInst>(I)) &&
315       I->getOperand(0)->getType() == Ty)
316     return true;
317 
318   // We can't extend or shrink something that has multiple uses: doing so would
319   // require duplicating the instruction in general, which isn't profitable.
320   if (!I->hasOneUse()) return false;
321 
322   unsigned Opc = I->getOpcode();
323   switch (Opc) {
324   case Instruction::Add:
325   case Instruction::Sub:
326   case Instruction::Mul:
327   case Instruction::And:
328   case Instruction::Or:
329   case Instruction::Xor:
330     // These operators can all arbitrarily be extended or truncated.
331     return canEvaluateTruncated(I->getOperand(0), Ty, IC, CxtI) &&
332            canEvaluateTruncated(I->getOperand(1), Ty, IC, CxtI);
333 
334   case Instruction::UDiv:
335   case Instruction::URem: {
336     // UDiv and URem can be truncated if all the truncated bits are zero.
337     uint32_t OrigBitWidth = OrigTy->getScalarSizeInBits();
338     uint32_t BitWidth = Ty->getScalarSizeInBits();
339     if (BitWidth < OrigBitWidth) {
340       APInt Mask = APInt::getHighBitsSet(OrigBitWidth, OrigBitWidth-BitWidth);
341       if (IC.MaskedValueIsZero(I->getOperand(0), Mask, 0, CxtI) &&
342           IC.MaskedValueIsZero(I->getOperand(1), Mask, 0, CxtI)) {
343         return canEvaluateTruncated(I->getOperand(0), Ty, IC, CxtI) &&
344                canEvaluateTruncated(I->getOperand(1), Ty, IC, CxtI);
345       }
346     }
347     break;
348   }
349   case Instruction::Shl: {
350     // If we are truncating the result of this SHL, and if it's a shift of a
351     // constant amount, we can always perform a SHL in a smaller type.
352     const APInt *Amt;
353     if (match(I->getOperand(1), m_APInt(Amt))) {
354       uint32_t BitWidth = Ty->getScalarSizeInBits();
355       if (Amt->getLimitedValue(BitWidth) < BitWidth)
356         return canEvaluateTruncated(I->getOperand(0), Ty, IC, CxtI);
357     }
358     break;
359   }
360   case Instruction::LShr: {
361     // If this is a truncate of a logical shr, we can truncate it to a smaller
362     // lshr iff we know that the bits we would otherwise be shifting in are
363     // already zeros.
364     const APInt *Amt;
365     if (match(I->getOperand(1), m_APInt(Amt))) {
366       uint32_t OrigBitWidth = OrigTy->getScalarSizeInBits();
367       uint32_t BitWidth = Ty->getScalarSizeInBits();
368       if (IC.MaskedValueIsZero(I->getOperand(0),
369             APInt::getHighBitsSet(OrigBitWidth, OrigBitWidth-BitWidth), 0, CxtI) &&
370           Amt->getLimitedValue(BitWidth) < BitWidth) {
371         return canEvaluateTruncated(I->getOperand(0), Ty, IC, CxtI);
372       }
373     }
374     break;
375   }
376   case Instruction::AShr: {
377     // If this is a truncate of an arithmetic shr, we can truncate it to a
378     // smaller ashr iff we know that all the bits from the sign bit of the
379     // original type and the sign bit of the truncate type are similar.
380     // TODO: It is enough to check that the bits we would be shifting in are
381     //       similar to sign bit of the truncate type.
382     const APInt *Amt;
383     if (match(I->getOperand(1), m_APInt(Amt))) {
384       uint32_t OrigBitWidth = OrigTy->getScalarSizeInBits();
385       uint32_t BitWidth = Ty->getScalarSizeInBits();
386       if (Amt->getLimitedValue(BitWidth) < BitWidth &&
387           OrigBitWidth - BitWidth <
388               IC.ComputeNumSignBits(I->getOperand(0), 0, CxtI))
389         return canEvaluateTruncated(I->getOperand(0), Ty, IC, CxtI);
390     }
391     break;
392   }
393   case Instruction::Trunc:
394     // trunc(trunc(x)) -> trunc(x)
395     return true;
396   case Instruction::ZExt:
397   case Instruction::SExt:
398     // trunc(ext(x)) -> ext(x) if the source type is smaller than the new dest
399     // trunc(ext(x)) -> trunc(x) if the source type is larger than the new dest
400     return true;
401   case Instruction::Select: {
402     SelectInst *SI = cast<SelectInst>(I);
403     return canEvaluateTruncated(SI->getTrueValue(), Ty, IC, CxtI) &&
404            canEvaluateTruncated(SI->getFalseValue(), Ty, IC, CxtI);
405   }
406   case Instruction::PHI: {
407     // We can change a phi if we can change all operands.  Note that we never
408     // get into trouble with cyclic PHIs here because we only consider
409     // instructions with a single use.
410     PHINode *PN = cast<PHINode>(I);
411     for (Value *IncValue : PN->incoming_values())
412       if (!canEvaluateTruncated(IncValue, Ty, IC, CxtI))
413         return false;
414     return true;
415   }
416   default:
417     // TODO: Can handle more cases here.
418     break;
419   }
420 
421   return false;
422 }
423 
424 /// Given a vector that is bitcast to an integer, optionally logically
425 /// right-shifted, and truncated, convert it to an extractelement.
426 /// Example (big endian):
427 ///   trunc (lshr (bitcast <4 x i32> %X to i128), 32) to i32
428 ///   --->
429 ///   extractelement <4 x i32> %X, 1
430 static Instruction *foldVecTruncToExtElt(TruncInst &Trunc, InstCombiner &IC) {
431   Value *TruncOp = Trunc.getOperand(0);
432   Type *DestType = Trunc.getType();
433   if (!TruncOp->hasOneUse() || !isa<IntegerType>(DestType))
434     return nullptr;
435 
436   Value *VecInput = nullptr;
437   ConstantInt *ShiftVal = nullptr;
438   if (!match(TruncOp, m_CombineOr(m_BitCast(m_Value(VecInput)),
439                                   m_LShr(m_BitCast(m_Value(VecInput)),
440                                          m_ConstantInt(ShiftVal)))) ||
441       !isa<VectorType>(VecInput->getType()))
442     return nullptr;
443 
444   VectorType *VecType = cast<VectorType>(VecInput->getType());
445   unsigned VecWidth = VecType->getPrimitiveSizeInBits();
446   unsigned DestWidth = DestType->getPrimitiveSizeInBits();
447   unsigned ShiftAmount = ShiftVal ? ShiftVal->getZExtValue() : 0;
448 
449   if ((VecWidth % DestWidth != 0) || (ShiftAmount % DestWidth != 0))
450     return nullptr;
451 
452   // If the element type of the vector doesn't match the result type,
453   // bitcast it to a vector type that we can extract from.
454   unsigned NumVecElts = VecWidth / DestWidth;
455   if (VecType->getElementType() != DestType) {
456     VecType = VectorType::get(DestType, NumVecElts);
457     VecInput = IC.Builder.CreateBitCast(VecInput, VecType, "bc");
458   }
459 
460   unsigned Elt = ShiftAmount / DestWidth;
461   if (IC.getDataLayout().isBigEndian())
462     Elt = NumVecElts - 1 - Elt;
463 
464   return ExtractElementInst::Create(VecInput, IC.Builder.getInt32(Elt));
465 }
466 
467 /// Rotate left/right may occur in a wider type than necessary because of type
468 /// promotion rules. Try to narrow all of the component instructions.
469 Instruction *InstCombiner::narrowRotate(TruncInst &Trunc) {
470   assert((isa<VectorType>(Trunc.getSrcTy()) ||
471           shouldChangeType(Trunc.getSrcTy(), Trunc.getType())) &&
472          "Don't narrow to an illegal scalar type");
473 
474   // First, find an or'd pair of opposite shifts with the same shifted operand:
475   // trunc (or (lshr ShVal, ShAmt0), (shl ShVal, ShAmt1))
476   Value *Or0, *Or1;
477   if (!match(Trunc.getOperand(0), m_OneUse(m_Or(m_Value(Or0), m_Value(Or1)))))
478     return nullptr;
479 
480   Value *ShVal, *ShAmt0, *ShAmt1;
481   if (!match(Or0, m_OneUse(m_LogicalShift(m_Value(ShVal), m_Value(ShAmt0)))) ||
482       !match(Or1, m_OneUse(m_LogicalShift(m_Specific(ShVal), m_Value(ShAmt1)))))
483     return nullptr;
484 
485   auto ShiftOpcode0 = cast<BinaryOperator>(Or0)->getOpcode();
486   auto ShiftOpcode1 = cast<BinaryOperator>(Or1)->getOpcode();
487   if (ShiftOpcode0 == ShiftOpcode1)
488     return nullptr;
489 
490   // The shift amounts must add up to the narrow bit width.
491   Value *ShAmt;
492   bool SubIsOnLHS;
493   Type *DestTy = Trunc.getType();
494   unsigned NarrowWidth = DestTy->getScalarSizeInBits();
495   if (match(ShAmt0,
496             m_OneUse(m_Sub(m_SpecificInt(NarrowWidth), m_Specific(ShAmt1))))) {
497     ShAmt = ShAmt1;
498     SubIsOnLHS = true;
499   } else if (match(ShAmt1, m_OneUse(m_Sub(m_SpecificInt(NarrowWidth),
500                                           m_Specific(ShAmt0))))) {
501     ShAmt = ShAmt0;
502     SubIsOnLHS = false;
503   } else {
504     return nullptr;
505   }
506 
507   // The shifted value must have high zeros in the wide type. Typically, this
508   // will be a zext, but it could also be the result of an 'and' or 'shift'.
509   unsigned WideWidth = Trunc.getSrcTy()->getScalarSizeInBits();
510   APInt HiBitMask = APInt::getHighBitsSet(WideWidth, WideWidth - NarrowWidth);
511   if (!MaskedValueIsZero(ShVal, HiBitMask, 0, &Trunc))
512     return nullptr;
513 
514   // We have an unnecessarily wide rotate!
515   // trunc (or (lshr ShVal, ShAmt), (shl ShVal, BitWidth - ShAmt))
516   // Narrow it down to eliminate the zext/trunc:
517   // or (lshr trunc(ShVal), ShAmt0'), (shl trunc(ShVal), ShAmt1')
518   Value *NarrowShAmt = Builder.CreateTrunc(ShAmt, DestTy);
519   Value *NegShAmt = Builder.CreateNeg(NarrowShAmt);
520 
521   // Mask both shift amounts to ensure there's no UB from oversized shifts.
522   Constant *MaskC = ConstantInt::get(DestTy, NarrowWidth - 1);
523   Value *MaskedShAmt = Builder.CreateAnd(NarrowShAmt, MaskC);
524   Value *MaskedNegShAmt = Builder.CreateAnd(NegShAmt, MaskC);
525 
526   // Truncate the original value and use narrow ops.
527   Value *X = Builder.CreateTrunc(ShVal, DestTy);
528   Value *NarrowShAmt0 = SubIsOnLHS ? MaskedNegShAmt : MaskedShAmt;
529   Value *NarrowShAmt1 = SubIsOnLHS ? MaskedShAmt : MaskedNegShAmt;
530   Value *NarrowSh0 = Builder.CreateBinOp(ShiftOpcode0, X, NarrowShAmt0);
531   Value *NarrowSh1 = Builder.CreateBinOp(ShiftOpcode1, X, NarrowShAmt1);
532   return BinaryOperator::CreateOr(NarrowSh0, NarrowSh1);
533 }
534 
535 /// Try to narrow the width of math or bitwise logic instructions by pulling a
536 /// truncate ahead of binary operators.
537 /// TODO: Transforms for truncated shifts should be moved into here.
538 Instruction *InstCombiner::narrowBinOp(TruncInst &Trunc) {
539   Type *SrcTy = Trunc.getSrcTy();
540   Type *DestTy = Trunc.getType();
541   if (!isa<VectorType>(SrcTy) && !shouldChangeType(SrcTy, DestTy))
542     return nullptr;
543 
544   BinaryOperator *BinOp;
545   if (!match(Trunc.getOperand(0), m_OneUse(m_BinOp(BinOp))))
546     return nullptr;
547 
548   switch (BinOp->getOpcode()) {
549   case Instruction::And:
550   case Instruction::Or:
551   case Instruction::Xor:
552   case Instruction::Add:
553   case Instruction::Mul: {
554     Constant *C;
555     if (match(BinOp->getOperand(1), m_Constant(C))) {
556       // trunc (binop X, C) --> binop (trunc X, C')
557       Constant *NarrowC = ConstantExpr::getTrunc(C, DestTy);
558       Value *TruncX = Builder.CreateTrunc(BinOp->getOperand(0), DestTy);
559       return BinaryOperator::Create(BinOp->getOpcode(), TruncX, NarrowC);
560     }
561     break;
562   }
563   case Instruction::Sub: {
564     Constant *C;
565     if (match(BinOp->getOperand(0), m_Constant(C))) {
566       // trunc (binop C, X) --> binop (trunc C', X)
567       Constant *NarrowC = ConstantExpr::getTrunc(C, DestTy);
568       Value *TruncX = Builder.CreateTrunc(BinOp->getOperand(1), DestTy);
569       return BinaryOperator::Create(BinOp->getOpcode(), NarrowC, TruncX);
570     }
571     break;
572   }
573 
574   default: break;
575   }
576 
577   if (Instruction *NarrowOr = narrowRotate(Trunc))
578     return NarrowOr;
579 
580   return nullptr;
581 }
582 
583 /// Try to narrow the width of a splat shuffle. This could be generalized to any
584 /// shuffle with a constant operand, but we limit the transform to avoid
585 /// creating a shuffle type that targets may not be able to lower effectively.
586 static Instruction *shrinkSplatShuffle(TruncInst &Trunc,
587                                        InstCombiner::BuilderTy &Builder) {
588   auto *Shuf = dyn_cast<ShuffleVectorInst>(Trunc.getOperand(0));
589   if (Shuf && Shuf->hasOneUse() && isa<UndefValue>(Shuf->getOperand(1)) &&
590       Shuf->getMask()->getSplatValue() &&
591       Shuf->getType() == Shuf->getOperand(0)->getType()) {
592     // trunc (shuf X, Undef, SplatMask) --> shuf (trunc X), Undef, SplatMask
593     Constant *NarrowUndef = UndefValue::get(Trunc.getType());
594     Value *NarrowOp = Builder.CreateTrunc(Shuf->getOperand(0), Trunc.getType());
595     return new ShuffleVectorInst(NarrowOp, NarrowUndef, Shuf->getMask());
596   }
597 
598   return nullptr;
599 }
600 
601 /// Try to narrow the width of an insert element. This could be generalized for
602 /// any vector constant, but we limit the transform to insertion into undef to
603 /// avoid potential backend problems from unsupported insertion widths. This
604 /// could also be extended to handle the case of inserting a scalar constant
605 /// into a vector variable.
606 static Instruction *shrinkInsertElt(CastInst &Trunc,
607                                     InstCombiner::BuilderTy &Builder) {
608   Instruction::CastOps Opcode = Trunc.getOpcode();
609   assert((Opcode == Instruction::Trunc || Opcode == Instruction::FPTrunc) &&
610          "Unexpected instruction for shrinking");
611 
612   auto *InsElt = dyn_cast<InsertElementInst>(Trunc.getOperand(0));
613   if (!InsElt || !InsElt->hasOneUse())
614     return nullptr;
615 
616   Type *DestTy = Trunc.getType();
617   Type *DestScalarTy = DestTy->getScalarType();
618   Value *VecOp = InsElt->getOperand(0);
619   Value *ScalarOp = InsElt->getOperand(1);
620   Value *Index = InsElt->getOperand(2);
621 
622   if (isa<UndefValue>(VecOp)) {
623     // trunc   (inselt undef, X, Index) --> inselt undef,   (trunc X), Index
624     // fptrunc (inselt undef, X, Index) --> inselt undef, (fptrunc X), Index
625     UndefValue *NarrowUndef = UndefValue::get(DestTy);
626     Value *NarrowOp = Builder.CreateCast(Opcode, ScalarOp, DestScalarTy);
627     return InsertElementInst::Create(NarrowUndef, NarrowOp, Index);
628   }
629 
630   return nullptr;
631 }
632 
633 Instruction *InstCombiner::visitTrunc(TruncInst &CI) {
634   if (Instruction *Result = commonCastTransforms(CI))
635     return Result;
636 
637   // Test if the trunc is the user of a select which is part of a
638   // minimum or maximum operation. If so, don't do any more simplification.
639   // Even simplifying demanded bits can break the canonical form of a
640   // min/max.
641   Value *LHS, *RHS;
642   if (SelectInst *SI = dyn_cast<SelectInst>(CI.getOperand(0)))
643     if (matchSelectPattern(SI, LHS, RHS).Flavor != SPF_UNKNOWN)
644       return nullptr;
645 
646   // See if we can simplify any instructions used by the input whose sole
647   // purpose is to compute bits we don't care about.
648   if (SimplifyDemandedInstructionBits(CI))
649     return &CI;
650 
651   Value *Src = CI.getOperand(0);
652   Type *DestTy = CI.getType(), *SrcTy = Src->getType();
653 
654   // Attempt to truncate the entire input expression tree to the destination
655   // type.   Only do this if the dest type is a simple type, don't convert the
656   // expression tree to something weird like i93 unless the source is also
657   // strange.
658   if ((DestTy->isVectorTy() || shouldChangeType(SrcTy, DestTy)) &&
659       canEvaluateTruncated(Src, DestTy, *this, &CI)) {
660 
661     // If this cast is a truncate, evaluting in a different type always
662     // eliminates the cast, so it is always a win.
663     DEBUG(dbgs() << "ICE: EvaluateInDifferentType converting expression type"
664           " to avoid cast: " << CI << '\n');
665     Value *Res = EvaluateInDifferentType(Src, DestTy, false);
666     assert(Res->getType() == DestTy);
667     return replaceInstUsesWith(CI, Res);
668   }
669 
670   // Canonicalize trunc x to i1 -> (icmp ne (and x, 1), 0), likewise for vector.
671   if (DestTy->getScalarSizeInBits() == 1) {
672     Constant *One = ConstantInt::get(SrcTy, 1);
673     Src = Builder.CreateAnd(Src, One);
674     Value *Zero = Constant::getNullValue(Src->getType());
675     return new ICmpInst(ICmpInst::ICMP_NE, Src, Zero);
676   }
677 
678   // FIXME: Maybe combine the next two transforms to handle the no cast case
679   // more efficiently. Support vector types. Cleanup code by using m_OneUse.
680 
681   // Transform trunc(lshr (zext A), Cst) to eliminate one type conversion.
682   Value *A = nullptr; ConstantInt *Cst = nullptr;
683   if (Src->hasOneUse() &&
684       match(Src, m_LShr(m_ZExt(m_Value(A)), m_ConstantInt(Cst)))) {
685     // We have three types to worry about here, the type of A, the source of
686     // the truncate (MidSize), and the destination of the truncate. We know that
687     // ASize < MidSize   and MidSize > ResultSize, but don't know the relation
688     // between ASize and ResultSize.
689     unsigned ASize = A->getType()->getPrimitiveSizeInBits();
690 
691     // If the shift amount is larger than the size of A, then the result is
692     // known to be zero because all the input bits got shifted out.
693     if (Cst->getZExtValue() >= ASize)
694       return replaceInstUsesWith(CI, Constant::getNullValue(DestTy));
695 
696     // Since we're doing an lshr and a zero extend, and know that the shift
697     // amount is smaller than ASize, it is always safe to do the shift in A's
698     // type, then zero extend or truncate to the result.
699     Value *Shift = Builder.CreateLShr(A, Cst->getZExtValue());
700     Shift->takeName(Src);
701     return CastInst::CreateIntegerCast(Shift, DestTy, false);
702   }
703 
704   // FIXME: We should canonicalize to zext/trunc and remove this transform.
705   // Transform trunc(lshr (sext A), Cst) to ashr A, Cst to eliminate type
706   // conversion.
707   // It works because bits coming from sign extension have the same value as
708   // the sign bit of the original value; performing ashr instead of lshr
709   // generates bits of the same value as the sign bit.
710   if (Src->hasOneUse() &&
711       match(Src, m_LShr(m_SExt(m_Value(A)), m_ConstantInt(Cst)))) {
712     Value *SExt = cast<Instruction>(Src)->getOperand(0);
713     const unsigned SExtSize = SExt->getType()->getPrimitiveSizeInBits();
714     const unsigned ASize = A->getType()->getPrimitiveSizeInBits();
715     const unsigned CISize = CI.getType()->getPrimitiveSizeInBits();
716     const unsigned MaxAmt = SExtSize - std::max(CISize, ASize);
717     unsigned ShiftAmt = Cst->getZExtValue();
718 
719     // This optimization can be only performed when zero bits generated by
720     // the original lshr aren't pulled into the value after truncation, so we
721     // can only shift by values no larger than the number of extension bits.
722     // FIXME: Instead of bailing when the shift is too large, use and to clear
723     // the extra bits.
724     if (ShiftAmt <= MaxAmt) {
725       if (CISize == ASize)
726         return BinaryOperator::CreateAShr(A, ConstantInt::get(CI.getType(),
727                                           std::min(ShiftAmt, ASize - 1)));
728       if (SExt->hasOneUse()) {
729         Value *Shift = Builder.CreateAShr(A, std::min(ShiftAmt, ASize - 1));
730         Shift->takeName(Src);
731         return CastInst::CreateIntegerCast(Shift, CI.getType(), true);
732       }
733     }
734   }
735 
736   if (Instruction *I = narrowBinOp(CI))
737     return I;
738 
739   if (Instruction *I = shrinkSplatShuffle(CI, Builder))
740     return I;
741 
742   if (Instruction *I = shrinkInsertElt(CI, Builder))
743     return I;
744 
745   if (Src->hasOneUse() && isa<IntegerType>(SrcTy) &&
746       shouldChangeType(SrcTy, DestTy)) {
747     // Transform "trunc (shl X, cst)" -> "shl (trunc X), cst" so long as the
748     // dest type is native and cst < dest size.
749     if (match(Src, m_Shl(m_Value(A), m_ConstantInt(Cst))) &&
750         !match(A, m_Shr(m_Value(), m_Constant()))) {
751       // Skip shifts of shift by constants. It undoes a combine in
752       // FoldShiftByConstant and is the extend in reg pattern.
753       const unsigned DestSize = DestTy->getScalarSizeInBits();
754       if (Cst->getValue().ult(DestSize)) {
755         Value *NewTrunc = Builder.CreateTrunc(A, DestTy, A->getName() + ".tr");
756 
757         return BinaryOperator::Create(
758           Instruction::Shl, NewTrunc,
759           ConstantInt::get(DestTy, Cst->getValue().trunc(DestSize)));
760       }
761     }
762   }
763 
764   if (Instruction *I = foldVecTruncToExtElt(CI, *this))
765     return I;
766 
767   return nullptr;
768 }
769 
770 Instruction *InstCombiner::transformZExtICmp(ICmpInst *ICI, ZExtInst &CI,
771                                              bool DoTransform) {
772   // If we are just checking for a icmp eq of a single bit and zext'ing it
773   // to an integer, then shift the bit to the appropriate place and then
774   // cast to integer to avoid the comparison.
775   const APInt *Op1CV;
776   if (match(ICI->getOperand(1), m_APInt(Op1CV))) {
777 
778     // zext (x <s  0) to i32 --> x>>u31      true if signbit set.
779     // zext (x >s -1) to i32 --> (x>>u31)^1  true if signbit clear.
780     if ((ICI->getPredicate() == ICmpInst::ICMP_SLT && Op1CV->isNullValue()) ||
781         (ICI->getPredicate() == ICmpInst::ICMP_SGT && Op1CV->isAllOnesValue())) {
782       if (!DoTransform) return ICI;
783 
784       Value *In = ICI->getOperand(0);
785       Value *Sh = ConstantInt::get(In->getType(),
786                                    In->getType()->getScalarSizeInBits() - 1);
787       In = Builder.CreateLShr(In, Sh, In->getName() + ".lobit");
788       if (In->getType() != CI.getType())
789         In = Builder.CreateIntCast(In, CI.getType(), false /*ZExt*/);
790 
791       if (ICI->getPredicate() == ICmpInst::ICMP_SGT) {
792         Constant *One = ConstantInt::get(In->getType(), 1);
793         In = Builder.CreateXor(In, One, In->getName() + ".not");
794       }
795 
796       return replaceInstUsesWith(CI, In);
797     }
798 
799     // zext (X == 0) to i32 --> X^1      iff X has only the low bit set.
800     // zext (X == 0) to i32 --> (X>>1)^1 iff X has only the 2nd bit set.
801     // zext (X == 1) to i32 --> X        iff X has only the low bit set.
802     // zext (X == 2) to i32 --> X>>1     iff X has only the 2nd bit set.
803     // zext (X != 0) to i32 --> X        iff X has only the low bit set.
804     // zext (X != 0) to i32 --> X>>1     iff X has only the 2nd bit set.
805     // zext (X != 1) to i32 --> X^1      iff X has only the low bit set.
806     // zext (X != 2) to i32 --> (X>>1)^1 iff X has only the 2nd bit set.
807     if ((Op1CV->isNullValue() || Op1CV->isPowerOf2()) &&
808         // This only works for EQ and NE
809         ICI->isEquality()) {
810       // If Op1C some other power of two, convert:
811       KnownBits Known = computeKnownBits(ICI->getOperand(0), 0, &CI);
812 
813       APInt KnownZeroMask(~Known.Zero);
814       if (KnownZeroMask.isPowerOf2()) { // Exactly 1 possible 1?
815         if (!DoTransform) return ICI;
816 
817         bool isNE = ICI->getPredicate() == ICmpInst::ICMP_NE;
818         if (!Op1CV->isNullValue() && (*Op1CV != KnownZeroMask)) {
819           // (X&4) == 2 --> false
820           // (X&4) != 2 --> true
821           Constant *Res = ConstantInt::get(CI.getType(), isNE);
822           return replaceInstUsesWith(CI, Res);
823         }
824 
825         uint32_t ShAmt = KnownZeroMask.logBase2();
826         Value *In = ICI->getOperand(0);
827         if (ShAmt) {
828           // Perform a logical shr by shiftamt.
829           // Insert the shift to put the result in the low bit.
830           In = Builder.CreateLShr(In, ConstantInt::get(In->getType(), ShAmt),
831                                   In->getName() + ".lobit");
832         }
833 
834         if (!Op1CV->isNullValue() == isNE) { // Toggle the low bit.
835           Constant *One = ConstantInt::get(In->getType(), 1);
836           In = Builder.CreateXor(In, One);
837         }
838 
839         if (CI.getType() == In->getType())
840           return replaceInstUsesWith(CI, In);
841 
842         Value *IntCast = Builder.CreateIntCast(In, CI.getType(), false);
843         return replaceInstUsesWith(CI, IntCast);
844       }
845     }
846   }
847 
848   // icmp ne A, B is equal to xor A, B when A and B only really have one bit.
849   // It is also profitable to transform icmp eq into not(xor(A, B)) because that
850   // may lead to additional simplifications.
851   if (ICI->isEquality() && CI.getType() == ICI->getOperand(0)->getType()) {
852     if (IntegerType *ITy = dyn_cast<IntegerType>(CI.getType())) {
853       Value *LHS = ICI->getOperand(0);
854       Value *RHS = ICI->getOperand(1);
855 
856       KnownBits KnownLHS = computeKnownBits(LHS, 0, &CI);
857       KnownBits KnownRHS = computeKnownBits(RHS, 0, &CI);
858 
859       if (KnownLHS.Zero == KnownRHS.Zero && KnownLHS.One == KnownRHS.One) {
860         APInt KnownBits = KnownLHS.Zero | KnownLHS.One;
861         APInt UnknownBit = ~KnownBits;
862         if (UnknownBit.countPopulation() == 1) {
863           if (!DoTransform) return ICI;
864 
865           Value *Result = Builder.CreateXor(LHS, RHS);
866 
867           // Mask off any bits that are set and won't be shifted away.
868           if (KnownLHS.One.uge(UnknownBit))
869             Result = Builder.CreateAnd(Result,
870                                         ConstantInt::get(ITy, UnknownBit));
871 
872           // Shift the bit we're testing down to the lsb.
873           Result = Builder.CreateLShr(
874                Result, ConstantInt::get(ITy, UnknownBit.countTrailingZeros()));
875 
876           if (ICI->getPredicate() == ICmpInst::ICMP_EQ)
877             Result = Builder.CreateXor(Result, ConstantInt::get(ITy, 1));
878           Result->takeName(ICI);
879           return replaceInstUsesWith(CI, Result);
880         }
881       }
882     }
883   }
884 
885   return nullptr;
886 }
887 
888 /// Determine if the specified value can be computed in the specified wider type
889 /// and produce the same low bits. If not, return false.
890 ///
891 /// If this function returns true, it can also return a non-zero number of bits
892 /// (in BitsToClear) which indicates that the value it computes is correct for
893 /// the zero extend, but that the additional BitsToClear bits need to be zero'd
894 /// out.  For example, to promote something like:
895 ///
896 ///   %B = trunc i64 %A to i32
897 ///   %C = lshr i32 %B, 8
898 ///   %E = zext i32 %C to i64
899 ///
900 /// CanEvaluateZExtd for the 'lshr' will return true, and BitsToClear will be
901 /// set to 8 to indicate that the promoted value needs to have bits 24-31
902 /// cleared in addition to bits 32-63.  Since an 'and' will be generated to
903 /// clear the top bits anyway, doing this has no extra cost.
904 ///
905 /// This function works on both vectors and scalars.
906 static bool canEvaluateZExtd(Value *V, Type *Ty, unsigned &BitsToClear,
907                              InstCombiner &IC, Instruction *CxtI) {
908   BitsToClear = 0;
909   if (isa<Constant>(V))
910     return true;
911 
912   Instruction *I = dyn_cast<Instruction>(V);
913   if (!I) return false;
914 
915   // If the input is a truncate from the destination type, we can trivially
916   // eliminate it.
917   if (isa<TruncInst>(I) && I->getOperand(0)->getType() == Ty)
918     return true;
919 
920   // We can't extend or shrink something that has multiple uses: doing so would
921   // require duplicating the instruction in general, which isn't profitable.
922   if (!I->hasOneUse()) return false;
923 
924   unsigned Opc = I->getOpcode(), Tmp;
925   switch (Opc) {
926   case Instruction::ZExt:  // zext(zext(x)) -> zext(x).
927   case Instruction::SExt:  // zext(sext(x)) -> sext(x).
928   case Instruction::Trunc: // zext(trunc(x)) -> trunc(x) or zext(x)
929     return true;
930   case Instruction::And:
931   case Instruction::Or:
932   case Instruction::Xor:
933   case Instruction::Add:
934   case Instruction::Sub:
935   case Instruction::Mul:
936     if (!canEvaluateZExtd(I->getOperand(0), Ty, BitsToClear, IC, CxtI) ||
937         !canEvaluateZExtd(I->getOperand(1), Ty, Tmp, IC, CxtI))
938       return false;
939     // These can all be promoted if neither operand has 'bits to clear'.
940     if (BitsToClear == 0 && Tmp == 0)
941       return true;
942 
943     // If the operation is an AND/OR/XOR and the bits to clear are zero in the
944     // other side, BitsToClear is ok.
945     if (Tmp == 0 && I->isBitwiseLogicOp()) {
946       // We use MaskedValueIsZero here for generality, but the case we care
947       // about the most is constant RHS.
948       unsigned VSize = V->getType()->getScalarSizeInBits();
949       if (IC.MaskedValueIsZero(I->getOperand(1),
950                                APInt::getHighBitsSet(VSize, BitsToClear),
951                                0, CxtI)) {
952         // If this is an And instruction and all of the BitsToClear are
953         // known to be zero we can reset BitsToClear.
954         if (Opc == Instruction::And)
955           BitsToClear = 0;
956         return true;
957       }
958     }
959 
960     // Otherwise, we don't know how to analyze this BitsToClear case yet.
961     return false;
962 
963   case Instruction::Shl: {
964     // We can promote shl(x, cst) if we can promote x.  Since shl overwrites the
965     // upper bits we can reduce BitsToClear by the shift amount.
966     const APInt *Amt;
967     if (match(I->getOperand(1), m_APInt(Amt))) {
968       if (!canEvaluateZExtd(I->getOperand(0), Ty, BitsToClear, IC, CxtI))
969         return false;
970       uint64_t ShiftAmt = Amt->getZExtValue();
971       BitsToClear = ShiftAmt < BitsToClear ? BitsToClear - ShiftAmt : 0;
972       return true;
973     }
974     return false;
975   }
976   case Instruction::LShr: {
977     // We can promote lshr(x, cst) if we can promote x.  This requires the
978     // ultimate 'and' to clear out the high zero bits we're clearing out though.
979     const APInt *Amt;
980     if (match(I->getOperand(1), m_APInt(Amt))) {
981       if (!canEvaluateZExtd(I->getOperand(0), Ty, BitsToClear, IC, CxtI))
982         return false;
983       BitsToClear += Amt->getZExtValue();
984       if (BitsToClear > V->getType()->getScalarSizeInBits())
985         BitsToClear = V->getType()->getScalarSizeInBits();
986       return true;
987     }
988     // Cannot promote variable LSHR.
989     return false;
990   }
991   case Instruction::Select:
992     if (!canEvaluateZExtd(I->getOperand(1), Ty, Tmp, IC, CxtI) ||
993         !canEvaluateZExtd(I->getOperand(2), Ty, BitsToClear, IC, CxtI) ||
994         // TODO: If important, we could handle the case when the BitsToClear are
995         // known zero in the disagreeing side.
996         Tmp != BitsToClear)
997       return false;
998     return true;
999 
1000   case Instruction::PHI: {
1001     // We can change a phi if we can change all operands.  Note that we never
1002     // get into trouble with cyclic PHIs here because we only consider
1003     // instructions with a single use.
1004     PHINode *PN = cast<PHINode>(I);
1005     if (!canEvaluateZExtd(PN->getIncomingValue(0), Ty, BitsToClear, IC, CxtI))
1006       return false;
1007     for (unsigned i = 1, e = PN->getNumIncomingValues(); i != e; ++i)
1008       if (!canEvaluateZExtd(PN->getIncomingValue(i), Ty, Tmp, IC, CxtI) ||
1009           // TODO: If important, we could handle the case when the BitsToClear
1010           // are known zero in the disagreeing input.
1011           Tmp != BitsToClear)
1012         return false;
1013     return true;
1014   }
1015   default:
1016     // TODO: Can handle more cases here.
1017     return false;
1018   }
1019 }
1020 
1021 Instruction *InstCombiner::visitZExt(ZExtInst &CI) {
1022   // If this zero extend is only used by a truncate, let the truncate be
1023   // eliminated before we try to optimize this zext.
1024   if (CI.hasOneUse() && isa<TruncInst>(CI.user_back()))
1025     return nullptr;
1026 
1027   // If one of the common conversion will work, do it.
1028   if (Instruction *Result = commonCastTransforms(CI))
1029     return Result;
1030 
1031   Value *Src = CI.getOperand(0);
1032   Type *SrcTy = Src->getType(), *DestTy = CI.getType();
1033 
1034   // Attempt to extend the entire input expression tree to the destination
1035   // type.   Only do this if the dest type is a simple type, don't convert the
1036   // expression tree to something weird like i93 unless the source is also
1037   // strange.
1038   unsigned BitsToClear;
1039   if ((DestTy->isVectorTy() || shouldChangeType(SrcTy, DestTy)) &&
1040       canEvaluateZExtd(Src, DestTy, BitsToClear, *this, &CI)) {
1041     assert(BitsToClear <= SrcTy->getScalarSizeInBits() &&
1042            "Can't clear more bits than in SrcTy");
1043 
1044     // Okay, we can transform this!  Insert the new expression now.
1045     DEBUG(dbgs() << "ICE: EvaluateInDifferentType converting expression type"
1046           " to avoid zero extend: " << CI << '\n');
1047     Value *Res = EvaluateInDifferentType(Src, DestTy, false);
1048     assert(Res->getType() == DestTy);
1049 
1050     uint32_t SrcBitsKept = SrcTy->getScalarSizeInBits()-BitsToClear;
1051     uint32_t DestBitSize = DestTy->getScalarSizeInBits();
1052 
1053     // If the high bits are already filled with zeros, just replace this
1054     // cast with the result.
1055     if (MaskedValueIsZero(Res,
1056                           APInt::getHighBitsSet(DestBitSize,
1057                                                 DestBitSize-SrcBitsKept),
1058                              0, &CI))
1059       return replaceInstUsesWith(CI, Res);
1060 
1061     // We need to emit an AND to clear the high bits.
1062     Constant *C = ConstantInt::get(Res->getType(),
1063                                APInt::getLowBitsSet(DestBitSize, SrcBitsKept));
1064     return BinaryOperator::CreateAnd(Res, C);
1065   }
1066 
1067   // If this is a TRUNC followed by a ZEXT then we are dealing with integral
1068   // types and if the sizes are just right we can convert this into a logical
1069   // 'and' which will be much cheaper than the pair of casts.
1070   if (TruncInst *CSrc = dyn_cast<TruncInst>(Src)) {   // A->B->C cast
1071     // TODO: Subsume this into EvaluateInDifferentType.
1072 
1073     // Get the sizes of the types involved.  We know that the intermediate type
1074     // will be smaller than A or C, but don't know the relation between A and C.
1075     Value *A = CSrc->getOperand(0);
1076     unsigned SrcSize = A->getType()->getScalarSizeInBits();
1077     unsigned MidSize = CSrc->getType()->getScalarSizeInBits();
1078     unsigned DstSize = CI.getType()->getScalarSizeInBits();
1079     // If we're actually extending zero bits, then if
1080     // SrcSize <  DstSize: zext(a & mask)
1081     // SrcSize == DstSize: a & mask
1082     // SrcSize  > DstSize: trunc(a) & mask
1083     if (SrcSize < DstSize) {
1084       APInt AndValue(APInt::getLowBitsSet(SrcSize, MidSize));
1085       Constant *AndConst = ConstantInt::get(A->getType(), AndValue);
1086       Value *And = Builder.CreateAnd(A, AndConst, CSrc->getName() + ".mask");
1087       return new ZExtInst(And, CI.getType());
1088     }
1089 
1090     if (SrcSize == DstSize) {
1091       APInt AndValue(APInt::getLowBitsSet(SrcSize, MidSize));
1092       return BinaryOperator::CreateAnd(A, ConstantInt::get(A->getType(),
1093                                                            AndValue));
1094     }
1095     if (SrcSize > DstSize) {
1096       Value *Trunc = Builder.CreateTrunc(A, CI.getType());
1097       APInt AndValue(APInt::getLowBitsSet(DstSize, MidSize));
1098       return BinaryOperator::CreateAnd(Trunc,
1099                                        ConstantInt::get(Trunc->getType(),
1100                                                         AndValue));
1101     }
1102   }
1103 
1104   if (ICmpInst *ICI = dyn_cast<ICmpInst>(Src))
1105     return transformZExtICmp(ICI, CI);
1106 
1107   BinaryOperator *SrcI = dyn_cast<BinaryOperator>(Src);
1108   if (SrcI && SrcI->getOpcode() == Instruction::Or) {
1109     // zext (or icmp, icmp) -> or (zext icmp), (zext icmp) if at least one
1110     // of the (zext icmp) can be eliminated. If so, immediately perform the
1111     // according elimination.
1112     ICmpInst *LHS = dyn_cast<ICmpInst>(SrcI->getOperand(0));
1113     ICmpInst *RHS = dyn_cast<ICmpInst>(SrcI->getOperand(1));
1114     if (LHS && RHS && LHS->hasOneUse() && RHS->hasOneUse() &&
1115         (transformZExtICmp(LHS, CI, false) ||
1116          transformZExtICmp(RHS, CI, false))) {
1117       // zext (or icmp, icmp) -> or (zext icmp), (zext icmp)
1118       Value *LCast = Builder.CreateZExt(LHS, CI.getType(), LHS->getName());
1119       Value *RCast = Builder.CreateZExt(RHS, CI.getType(), RHS->getName());
1120       BinaryOperator *Or = BinaryOperator::Create(Instruction::Or, LCast, RCast);
1121 
1122       // Perform the elimination.
1123       if (auto *LZExt = dyn_cast<ZExtInst>(LCast))
1124         transformZExtICmp(LHS, *LZExt);
1125       if (auto *RZExt = dyn_cast<ZExtInst>(RCast))
1126         transformZExtICmp(RHS, *RZExt);
1127 
1128       return Or;
1129     }
1130   }
1131 
1132   // zext(trunc(X) & C) -> (X & zext(C)).
1133   Constant *C;
1134   Value *X;
1135   if (SrcI &&
1136       match(SrcI, m_OneUse(m_And(m_Trunc(m_Value(X)), m_Constant(C)))) &&
1137       X->getType() == CI.getType())
1138     return BinaryOperator::CreateAnd(X, ConstantExpr::getZExt(C, CI.getType()));
1139 
1140   // zext((trunc(X) & C) ^ C) -> ((X & zext(C)) ^ zext(C)).
1141   Value *And;
1142   if (SrcI && match(SrcI, m_OneUse(m_Xor(m_Value(And), m_Constant(C)))) &&
1143       match(And, m_OneUse(m_And(m_Trunc(m_Value(X)), m_Specific(C)))) &&
1144       X->getType() == CI.getType()) {
1145     Constant *ZC = ConstantExpr::getZExt(C, CI.getType());
1146     return BinaryOperator::CreateXor(Builder.CreateAnd(X, ZC), ZC);
1147   }
1148 
1149   return nullptr;
1150 }
1151 
1152 /// Transform (sext icmp) to bitwise / integer operations to eliminate the icmp.
1153 Instruction *InstCombiner::transformSExtICmp(ICmpInst *ICI, Instruction &CI) {
1154   Value *Op0 = ICI->getOperand(0), *Op1 = ICI->getOperand(1);
1155   ICmpInst::Predicate Pred = ICI->getPredicate();
1156 
1157   // Don't bother if Op1 isn't of vector or integer type.
1158   if (!Op1->getType()->isIntOrIntVectorTy())
1159     return nullptr;
1160 
1161   if (Constant *Op1C = dyn_cast<Constant>(Op1)) {
1162     // (x <s  0) ? -1 : 0 -> ashr x, 31        -> all ones if negative
1163     // (x >s -1) ? -1 : 0 -> not (ashr x, 31)  -> all ones if positive
1164     if ((Pred == ICmpInst::ICMP_SLT && Op1C->isNullValue()) ||
1165         (Pred == ICmpInst::ICMP_SGT && Op1C->isAllOnesValue())) {
1166 
1167       Value *Sh = ConstantInt::get(Op0->getType(),
1168                                    Op0->getType()->getScalarSizeInBits()-1);
1169       Value *In = Builder.CreateAShr(Op0, Sh, Op0->getName() + ".lobit");
1170       if (In->getType() != CI.getType())
1171         In = Builder.CreateIntCast(In, CI.getType(), true /*SExt*/);
1172 
1173       if (Pred == ICmpInst::ICMP_SGT)
1174         In = Builder.CreateNot(In, In->getName() + ".not");
1175       return replaceInstUsesWith(CI, In);
1176     }
1177   }
1178 
1179   if (ConstantInt *Op1C = dyn_cast<ConstantInt>(Op1)) {
1180     // If we know that only one bit of the LHS of the icmp can be set and we
1181     // have an equality comparison with zero or a power of 2, we can transform
1182     // the icmp and sext into bitwise/integer operations.
1183     if (ICI->hasOneUse() &&
1184         ICI->isEquality() && (Op1C->isZero() || Op1C->getValue().isPowerOf2())){
1185       KnownBits Known = computeKnownBits(Op0, 0, &CI);
1186 
1187       APInt KnownZeroMask(~Known.Zero);
1188       if (KnownZeroMask.isPowerOf2()) {
1189         Value *In = ICI->getOperand(0);
1190 
1191         // If the icmp tests for a known zero bit we can constant fold it.
1192         if (!Op1C->isZero() && Op1C->getValue() != KnownZeroMask) {
1193           Value *V = Pred == ICmpInst::ICMP_NE ?
1194                        ConstantInt::getAllOnesValue(CI.getType()) :
1195                        ConstantInt::getNullValue(CI.getType());
1196           return replaceInstUsesWith(CI, V);
1197         }
1198 
1199         if (!Op1C->isZero() == (Pred == ICmpInst::ICMP_NE)) {
1200           // sext ((x & 2^n) == 0)   -> (x >> n) - 1
1201           // sext ((x & 2^n) != 2^n) -> (x >> n) - 1
1202           unsigned ShiftAmt = KnownZeroMask.countTrailingZeros();
1203           // Perform a right shift to place the desired bit in the LSB.
1204           if (ShiftAmt)
1205             In = Builder.CreateLShr(In,
1206                                     ConstantInt::get(In->getType(), ShiftAmt));
1207 
1208           // At this point "In" is either 1 or 0. Subtract 1 to turn
1209           // {1, 0} -> {0, -1}.
1210           In = Builder.CreateAdd(In,
1211                                  ConstantInt::getAllOnesValue(In->getType()),
1212                                  "sext");
1213         } else {
1214           // sext ((x & 2^n) != 0)   -> (x << bitwidth-n) a>> bitwidth-1
1215           // sext ((x & 2^n) == 2^n) -> (x << bitwidth-n) a>> bitwidth-1
1216           unsigned ShiftAmt = KnownZeroMask.countLeadingZeros();
1217           // Perform a left shift to place the desired bit in the MSB.
1218           if (ShiftAmt)
1219             In = Builder.CreateShl(In,
1220                                    ConstantInt::get(In->getType(), ShiftAmt));
1221 
1222           // Distribute the bit over the whole bit width.
1223           In = Builder.CreateAShr(In, ConstantInt::get(In->getType(),
1224                                   KnownZeroMask.getBitWidth() - 1), "sext");
1225         }
1226 
1227         if (CI.getType() == In->getType())
1228           return replaceInstUsesWith(CI, In);
1229         return CastInst::CreateIntegerCast(In, CI.getType(), true/*SExt*/);
1230       }
1231     }
1232   }
1233 
1234   return nullptr;
1235 }
1236 
1237 /// Return true if we can take the specified value and return it as type Ty
1238 /// without inserting any new casts and without changing the value of the common
1239 /// low bits.  This is used by code that tries to promote integer operations to
1240 /// a wider types will allow us to eliminate the extension.
1241 ///
1242 /// This function works on both vectors and scalars.
1243 ///
1244 static bool canEvaluateSExtd(Value *V, Type *Ty) {
1245   assert(V->getType()->getScalarSizeInBits() < Ty->getScalarSizeInBits() &&
1246          "Can't sign extend type to a smaller type");
1247   // If this is a constant, it can be trivially promoted.
1248   if (isa<Constant>(V))
1249     return true;
1250 
1251   Instruction *I = dyn_cast<Instruction>(V);
1252   if (!I) return false;
1253 
1254   // If this is a truncate from the dest type, we can trivially eliminate it.
1255   if (isa<TruncInst>(I) && I->getOperand(0)->getType() == Ty)
1256     return true;
1257 
1258   // We can't extend or shrink something that has multiple uses: doing so would
1259   // require duplicating the instruction in general, which isn't profitable.
1260   if (!I->hasOneUse()) return false;
1261 
1262   switch (I->getOpcode()) {
1263   case Instruction::SExt:  // sext(sext(x)) -> sext(x)
1264   case Instruction::ZExt:  // sext(zext(x)) -> zext(x)
1265   case Instruction::Trunc: // sext(trunc(x)) -> trunc(x) or sext(x)
1266     return true;
1267   case Instruction::And:
1268   case Instruction::Or:
1269   case Instruction::Xor:
1270   case Instruction::Add:
1271   case Instruction::Sub:
1272   case Instruction::Mul:
1273     // These operators can all arbitrarily be extended if their inputs can.
1274     return canEvaluateSExtd(I->getOperand(0), Ty) &&
1275            canEvaluateSExtd(I->getOperand(1), Ty);
1276 
1277   //case Instruction::Shl:   TODO
1278   //case Instruction::LShr:  TODO
1279 
1280   case Instruction::Select:
1281     return canEvaluateSExtd(I->getOperand(1), Ty) &&
1282            canEvaluateSExtd(I->getOperand(2), Ty);
1283 
1284   case Instruction::PHI: {
1285     // We can change a phi if we can change all operands.  Note that we never
1286     // get into trouble with cyclic PHIs here because we only consider
1287     // instructions with a single use.
1288     PHINode *PN = cast<PHINode>(I);
1289     for (Value *IncValue : PN->incoming_values())
1290       if (!canEvaluateSExtd(IncValue, Ty)) return false;
1291     return true;
1292   }
1293   default:
1294     // TODO: Can handle more cases here.
1295     break;
1296   }
1297 
1298   return false;
1299 }
1300 
1301 Instruction *InstCombiner::visitSExt(SExtInst &CI) {
1302   // If this sign extend is only used by a truncate, let the truncate be
1303   // eliminated before we try to optimize this sext.
1304   if (CI.hasOneUse() && isa<TruncInst>(CI.user_back()))
1305     return nullptr;
1306 
1307   if (Instruction *I = commonCastTransforms(CI))
1308     return I;
1309 
1310   Value *Src = CI.getOperand(0);
1311   Type *SrcTy = Src->getType(), *DestTy = CI.getType();
1312 
1313   // If we know that the value being extended is positive, we can use a zext
1314   // instead.
1315   KnownBits Known = computeKnownBits(Src, 0, &CI);
1316   if (Known.isNonNegative()) {
1317     Value *ZExt = Builder.CreateZExt(Src, DestTy);
1318     return replaceInstUsesWith(CI, ZExt);
1319   }
1320 
1321   // Attempt to extend the entire input expression tree to the destination
1322   // type.   Only do this if the dest type is a simple type, don't convert the
1323   // expression tree to something weird like i93 unless the source is also
1324   // strange.
1325   if ((DestTy->isVectorTy() || shouldChangeType(SrcTy, DestTy)) &&
1326       canEvaluateSExtd(Src, DestTy)) {
1327     // Okay, we can transform this!  Insert the new expression now.
1328     DEBUG(dbgs() << "ICE: EvaluateInDifferentType converting expression type"
1329           " to avoid sign extend: " << CI << '\n');
1330     Value *Res = EvaluateInDifferentType(Src, DestTy, true);
1331     assert(Res->getType() == DestTy);
1332 
1333     uint32_t SrcBitSize = SrcTy->getScalarSizeInBits();
1334     uint32_t DestBitSize = DestTy->getScalarSizeInBits();
1335 
1336     // If the high bits are already filled with sign bit, just replace this
1337     // cast with the result.
1338     if (ComputeNumSignBits(Res, 0, &CI) > DestBitSize - SrcBitSize)
1339       return replaceInstUsesWith(CI, Res);
1340 
1341     // We need to emit a shl + ashr to do the sign extend.
1342     Value *ShAmt = ConstantInt::get(DestTy, DestBitSize-SrcBitSize);
1343     return BinaryOperator::CreateAShr(Builder.CreateShl(Res, ShAmt, "sext"),
1344                                       ShAmt);
1345   }
1346 
1347   // If the input is a trunc from the destination type, then turn sext(trunc(x))
1348   // into shifts.
1349   Value *X;
1350   if (match(Src, m_OneUse(m_Trunc(m_Value(X)))) && X->getType() == DestTy) {
1351     // sext(trunc(X)) --> ashr(shl(X, C), C)
1352     unsigned SrcBitSize = SrcTy->getScalarSizeInBits();
1353     unsigned DestBitSize = DestTy->getScalarSizeInBits();
1354     Constant *ShAmt = ConstantInt::get(DestTy, DestBitSize - SrcBitSize);
1355     return BinaryOperator::CreateAShr(Builder.CreateShl(X, ShAmt), ShAmt);
1356   }
1357 
1358   if (ICmpInst *ICI = dyn_cast<ICmpInst>(Src))
1359     return transformSExtICmp(ICI, CI);
1360 
1361   // If the input is a shl/ashr pair of a same constant, then this is a sign
1362   // extension from a smaller value.  If we could trust arbitrary bitwidth
1363   // integers, we could turn this into a truncate to the smaller bit and then
1364   // use a sext for the whole extension.  Since we don't, look deeper and check
1365   // for a truncate.  If the source and dest are the same type, eliminate the
1366   // trunc and extend and just do shifts.  For example, turn:
1367   //   %a = trunc i32 %i to i8
1368   //   %b = shl i8 %a, 6
1369   //   %c = ashr i8 %b, 6
1370   //   %d = sext i8 %c to i32
1371   // into:
1372   //   %a = shl i32 %i, 30
1373   //   %d = ashr i32 %a, 30
1374   Value *A = nullptr;
1375   // TODO: Eventually this could be subsumed by EvaluateInDifferentType.
1376   ConstantInt *BA = nullptr, *CA = nullptr;
1377   if (match(Src, m_AShr(m_Shl(m_Trunc(m_Value(A)), m_ConstantInt(BA)),
1378                         m_ConstantInt(CA))) &&
1379       BA == CA && A->getType() == CI.getType()) {
1380     unsigned MidSize = Src->getType()->getScalarSizeInBits();
1381     unsigned SrcDstSize = CI.getType()->getScalarSizeInBits();
1382     unsigned ShAmt = CA->getZExtValue()+SrcDstSize-MidSize;
1383     Constant *ShAmtV = ConstantInt::get(CI.getType(), ShAmt);
1384     A = Builder.CreateShl(A, ShAmtV, CI.getName());
1385     return BinaryOperator::CreateAShr(A, ShAmtV);
1386   }
1387 
1388   return nullptr;
1389 }
1390 
1391 
1392 /// Return a Constant* for the specified floating-point constant if it fits
1393 /// in the specified FP type without changing its value.
1394 static Constant *fitsInFPType(ConstantFP *CFP, const fltSemantics &Sem) {
1395   bool losesInfo;
1396   APFloat F = CFP->getValueAPF();
1397   (void)F.convert(Sem, APFloat::rmNearestTiesToEven, &losesInfo);
1398   if (!losesInfo)
1399     return ConstantFP::get(CFP->getContext(), F);
1400   return nullptr;
1401 }
1402 
1403 /// Look through floating-point extensions until we get the source value.
1404 static Value *lookThroughFPExtensions(Value *V) {
1405   while (auto *FPExt = dyn_cast<FPExtInst>(V))
1406     V = FPExt->getOperand(0);
1407 
1408   // If this value is a constant, return the constant in the smallest FP type
1409   // that can accurately represent it.  This allows us to turn
1410   // (float)((double)X+2.0) into x+2.0f.
1411   if (auto *CFP = dyn_cast<ConstantFP>(V)) {
1412     if (CFP->getType() == Type::getPPC_FP128Ty(V->getContext()))
1413       return V;  // No constant folding of this.
1414     // See if the value can be truncated to half and then reextended.
1415     if (Value *V = fitsInFPType(CFP, APFloat::IEEEhalf()))
1416       return V;
1417     // See if the value can be truncated to float and then reextended.
1418     if (Value *V = fitsInFPType(CFP, APFloat::IEEEsingle()))
1419       return V;
1420     if (CFP->getType()->isDoubleTy())
1421       return V;  // Won't shrink.
1422     if (Value *V = fitsInFPType(CFP, APFloat::IEEEdouble()))
1423       return V;
1424     // Don't try to shrink to various long double types.
1425   }
1426 
1427   return V;
1428 }
1429 
1430 Instruction *InstCombiner::visitFPTrunc(FPTruncInst &CI) {
1431   if (Instruction *I = commonCastTransforms(CI))
1432     return I;
1433   // If we have fptrunc(OpI (fpextend x), (fpextend y)), we would like to
1434   // simplify this expression to avoid one or more of the trunc/extend
1435   // operations if we can do so without changing the numerical results.
1436   //
1437   // The exact manner in which the widths of the operands interact to limit
1438   // what we can and cannot do safely varies from operation to operation, and
1439   // is explained below in the various case statements.
1440   BinaryOperator *OpI = dyn_cast<BinaryOperator>(CI.getOperand(0));
1441   if (OpI && OpI->hasOneUse()) {
1442     Value *LHSOrig = lookThroughFPExtensions(OpI->getOperand(0));
1443     Value *RHSOrig = lookThroughFPExtensions(OpI->getOperand(1));
1444     unsigned OpWidth = OpI->getType()->getFPMantissaWidth();
1445     unsigned LHSWidth = LHSOrig->getType()->getFPMantissaWidth();
1446     unsigned RHSWidth = RHSOrig->getType()->getFPMantissaWidth();
1447     unsigned SrcWidth = std::max(LHSWidth, RHSWidth);
1448     unsigned DstWidth = CI.getType()->getFPMantissaWidth();
1449     switch (OpI->getOpcode()) {
1450       default: break;
1451       case Instruction::FAdd:
1452       case Instruction::FSub:
1453         // For addition and subtraction, the infinitely precise result can
1454         // essentially be arbitrarily wide; proving that double rounding
1455         // will not occur because the result of OpI is exact (as we will for
1456         // FMul, for example) is hopeless.  However, we *can* nonetheless
1457         // frequently know that double rounding cannot occur (or that it is
1458         // innocuous) by taking advantage of the specific structure of
1459         // infinitely-precise results that admit double rounding.
1460         //
1461         // Specifically, if OpWidth >= 2*DstWdith+1 and DstWidth is sufficient
1462         // to represent both sources, we can guarantee that the double
1463         // rounding is innocuous (See p50 of Figueroa's 2000 PhD thesis,
1464         // "A Rigorous Framework for Fully Supporting the IEEE Standard ..."
1465         // for proof of this fact).
1466         //
1467         // Note: Figueroa does not consider the case where DstFormat !=
1468         // SrcFormat.  It's possible (likely even!) that this analysis
1469         // could be tightened for those cases, but they are rare (the main
1470         // case of interest here is (float)((double)float + float)).
1471         if (OpWidth >= 2*DstWidth+1 && DstWidth >= SrcWidth) {
1472           if (LHSOrig->getType() != CI.getType())
1473             LHSOrig = Builder.CreateFPExt(LHSOrig, CI.getType());
1474           if (RHSOrig->getType() != CI.getType())
1475             RHSOrig = Builder.CreateFPExt(RHSOrig, CI.getType());
1476           Instruction *RI =
1477             BinaryOperator::Create(OpI->getOpcode(), LHSOrig, RHSOrig);
1478           RI->copyFastMathFlags(OpI);
1479           return RI;
1480         }
1481         break;
1482       case Instruction::FMul:
1483         // For multiplication, the infinitely precise result has at most
1484         // LHSWidth + RHSWidth significant bits; if OpWidth is sufficient
1485         // that such a value can be exactly represented, then no double
1486         // rounding can possibly occur; we can safely perform the operation
1487         // in the destination format if it can represent both sources.
1488         if (OpWidth >= LHSWidth + RHSWidth && DstWidth >= SrcWidth) {
1489           if (LHSOrig->getType() != CI.getType())
1490             LHSOrig = Builder.CreateFPExt(LHSOrig, CI.getType());
1491           if (RHSOrig->getType() != CI.getType())
1492             RHSOrig = Builder.CreateFPExt(RHSOrig, CI.getType());
1493           Instruction *RI =
1494             BinaryOperator::CreateFMul(LHSOrig, RHSOrig);
1495           RI->copyFastMathFlags(OpI);
1496           return RI;
1497         }
1498         break;
1499       case Instruction::FDiv:
1500         // For division, we use again use the bound from Figueroa's
1501         // dissertation.  I am entirely certain that this bound can be
1502         // tightened in the unbalanced operand case by an analysis based on
1503         // the diophantine rational approximation bound, but the well-known
1504         // condition used here is a good conservative first pass.
1505         // TODO: Tighten bound via rigorous analysis of the unbalanced case.
1506         if (OpWidth >= 2*DstWidth && DstWidth >= SrcWidth) {
1507           if (LHSOrig->getType() != CI.getType())
1508             LHSOrig = Builder.CreateFPExt(LHSOrig, CI.getType());
1509           if (RHSOrig->getType() != CI.getType())
1510             RHSOrig = Builder.CreateFPExt(RHSOrig, CI.getType());
1511           Instruction *RI =
1512             BinaryOperator::CreateFDiv(LHSOrig, RHSOrig);
1513           RI->copyFastMathFlags(OpI);
1514           return RI;
1515         }
1516         break;
1517       case Instruction::FRem:
1518         // Remainder is straightforward.  Remainder is always exact, so the
1519         // type of OpI doesn't enter into things at all.  We simply evaluate
1520         // in whichever source type is larger, then convert to the
1521         // destination type.
1522         if (SrcWidth == OpWidth)
1523           break;
1524         if (LHSWidth < SrcWidth)
1525           LHSOrig = Builder.CreateFPExt(LHSOrig, RHSOrig->getType());
1526         else if (RHSWidth <= SrcWidth)
1527           RHSOrig = Builder.CreateFPExt(RHSOrig, LHSOrig->getType());
1528         if (LHSOrig != OpI->getOperand(0) || RHSOrig != OpI->getOperand(1)) {
1529           Value *ExactResult = Builder.CreateFRem(LHSOrig, RHSOrig);
1530           if (Instruction *RI = dyn_cast<Instruction>(ExactResult))
1531             RI->copyFastMathFlags(OpI);
1532           return CastInst::CreateFPCast(ExactResult, CI.getType());
1533         }
1534     }
1535 
1536     // (fptrunc (fneg x)) -> (fneg (fptrunc x))
1537     if (BinaryOperator::isFNeg(OpI)) {
1538       Value *InnerTrunc = Builder.CreateFPTrunc(OpI->getOperand(1),
1539                                                 CI.getType());
1540       Instruction *RI = BinaryOperator::CreateFNeg(InnerTrunc);
1541       RI->copyFastMathFlags(OpI);
1542       return RI;
1543     }
1544   }
1545 
1546   // (fptrunc (select cond, R1, Cst)) -->
1547   // (select cond, (fptrunc R1), (fptrunc Cst))
1548   //
1549   //  - but only if this isn't part of a min/max operation, else we'll
1550   // ruin min/max canonical form which is to have the select and
1551   // compare's operands be of the same type with no casts to look through.
1552   Value *LHS, *RHS;
1553   SelectInst *SI = dyn_cast<SelectInst>(CI.getOperand(0));
1554   if (SI &&
1555       (isa<ConstantFP>(SI->getOperand(1)) ||
1556        isa<ConstantFP>(SI->getOperand(2))) &&
1557       matchSelectPattern(SI, LHS, RHS).Flavor == SPF_UNKNOWN) {
1558     Value *LHSTrunc = Builder.CreateFPTrunc(SI->getOperand(1), CI.getType());
1559     Value *RHSTrunc = Builder.CreateFPTrunc(SI->getOperand(2), CI.getType());
1560     return SelectInst::Create(SI->getOperand(0), LHSTrunc, RHSTrunc);
1561   }
1562 
1563   IntrinsicInst *II = dyn_cast<IntrinsicInst>(CI.getOperand(0));
1564   if (II) {
1565     switch (II->getIntrinsicID()) {
1566     default: break;
1567     case Intrinsic::fabs:
1568     case Intrinsic::ceil:
1569     case Intrinsic::floor:
1570     case Intrinsic::rint:
1571     case Intrinsic::round:
1572     case Intrinsic::nearbyint:
1573     case Intrinsic::trunc: {
1574       Value *Src = II->getArgOperand(0);
1575       if (!Src->hasOneUse())
1576         break;
1577 
1578       // Except for fabs, this transformation requires the input of the unary FP
1579       // operation to be itself an fpext from the type to which we're
1580       // truncating.
1581       if (II->getIntrinsicID() != Intrinsic::fabs) {
1582         FPExtInst *FPExtSrc = dyn_cast<FPExtInst>(Src);
1583         if (!FPExtSrc || FPExtSrc->getOperand(0)->getType() != CI.getType())
1584           break;
1585       }
1586 
1587       // Do unary FP operation on smaller type.
1588       // (fptrunc (fabs x)) -> (fabs (fptrunc x))
1589       Value *InnerTrunc = Builder.CreateFPTrunc(Src, CI.getType());
1590       Type *IntrinsicType[] = { CI.getType() };
1591       Function *Overload = Intrinsic::getDeclaration(
1592         CI.getModule(), II->getIntrinsicID(), IntrinsicType);
1593 
1594       SmallVector<OperandBundleDef, 1> OpBundles;
1595       II->getOperandBundlesAsDefs(OpBundles);
1596 
1597       Value *Args[] = { InnerTrunc };
1598       CallInst *NewCI =  CallInst::Create(Overload, Args,
1599                                           OpBundles, II->getName());
1600       NewCI->copyFastMathFlags(II);
1601       return NewCI;
1602     }
1603     }
1604   }
1605 
1606   if (Instruction *I = shrinkInsertElt(CI, Builder))
1607     return I;
1608 
1609   return nullptr;
1610 }
1611 
1612 Instruction *InstCombiner::visitFPExt(CastInst &CI) {
1613   return commonCastTransforms(CI);
1614 }
1615 
1616 // fpto{s/u}i({u/s}itofp(X)) --> X or zext(X) or sext(X) or trunc(X)
1617 // This is safe if the intermediate type has enough bits in its mantissa to
1618 // accurately represent all values of X.  For example, this won't work with
1619 // i64 -> float -> i64.
1620 Instruction *InstCombiner::FoldItoFPtoI(Instruction &FI) {
1621   if (!isa<UIToFPInst>(FI.getOperand(0)) && !isa<SIToFPInst>(FI.getOperand(0)))
1622     return nullptr;
1623   Instruction *OpI = cast<Instruction>(FI.getOperand(0));
1624 
1625   Value *SrcI = OpI->getOperand(0);
1626   Type *FITy = FI.getType();
1627   Type *OpITy = OpI->getType();
1628   Type *SrcTy = SrcI->getType();
1629   bool IsInputSigned = isa<SIToFPInst>(OpI);
1630   bool IsOutputSigned = isa<FPToSIInst>(FI);
1631 
1632   // We can safely assume the conversion won't overflow the output range,
1633   // because (for example) (uint8_t)18293.f is undefined behavior.
1634 
1635   // Since we can assume the conversion won't overflow, our decision as to
1636   // whether the input will fit in the float should depend on the minimum
1637   // of the input range and output range.
1638 
1639   // This means this is also safe for a signed input and unsigned output, since
1640   // a negative input would lead to undefined behavior.
1641   int InputSize = (int)SrcTy->getScalarSizeInBits() - IsInputSigned;
1642   int OutputSize = (int)FITy->getScalarSizeInBits() - IsOutputSigned;
1643   int ActualSize = std::min(InputSize, OutputSize);
1644 
1645   if (ActualSize <= OpITy->getFPMantissaWidth()) {
1646     if (FITy->getScalarSizeInBits() > SrcTy->getScalarSizeInBits()) {
1647       if (IsInputSigned && IsOutputSigned)
1648         return new SExtInst(SrcI, FITy);
1649       return new ZExtInst(SrcI, FITy);
1650     }
1651     if (FITy->getScalarSizeInBits() < SrcTy->getScalarSizeInBits())
1652       return new TruncInst(SrcI, FITy);
1653     if (SrcTy == FITy)
1654       return replaceInstUsesWith(FI, SrcI);
1655     return new BitCastInst(SrcI, FITy);
1656   }
1657   return nullptr;
1658 }
1659 
1660 Instruction *InstCombiner::visitFPToUI(FPToUIInst &FI) {
1661   Instruction *OpI = dyn_cast<Instruction>(FI.getOperand(0));
1662   if (!OpI)
1663     return commonCastTransforms(FI);
1664 
1665   if (Instruction *I = FoldItoFPtoI(FI))
1666     return I;
1667 
1668   return commonCastTransforms(FI);
1669 }
1670 
1671 Instruction *InstCombiner::visitFPToSI(FPToSIInst &FI) {
1672   Instruction *OpI = dyn_cast<Instruction>(FI.getOperand(0));
1673   if (!OpI)
1674     return commonCastTransforms(FI);
1675 
1676   if (Instruction *I = FoldItoFPtoI(FI))
1677     return I;
1678 
1679   return commonCastTransforms(FI);
1680 }
1681 
1682 Instruction *InstCombiner::visitUIToFP(CastInst &CI) {
1683   return commonCastTransforms(CI);
1684 }
1685 
1686 Instruction *InstCombiner::visitSIToFP(CastInst &CI) {
1687   return commonCastTransforms(CI);
1688 }
1689 
1690 Instruction *InstCombiner::visitIntToPtr(IntToPtrInst &CI) {
1691   // If the source integer type is not the intptr_t type for this target, do a
1692   // trunc or zext to the intptr_t type, then inttoptr of it.  This allows the
1693   // cast to be exposed to other transforms.
1694   unsigned AS = CI.getAddressSpace();
1695   if (CI.getOperand(0)->getType()->getScalarSizeInBits() !=
1696       DL.getPointerSizeInBits(AS)) {
1697     Type *Ty = DL.getIntPtrType(CI.getContext(), AS);
1698     if (CI.getType()->isVectorTy()) // Handle vectors of pointers.
1699       Ty = VectorType::get(Ty, CI.getType()->getVectorNumElements());
1700 
1701     Value *P = Builder.CreateZExtOrTrunc(CI.getOperand(0), Ty);
1702     return new IntToPtrInst(P, CI.getType());
1703   }
1704 
1705   if (Instruction *I = commonCastTransforms(CI))
1706     return I;
1707 
1708   return nullptr;
1709 }
1710 
1711 /// @brief Implement the transforms for cast of pointer (bitcast/ptrtoint)
1712 Instruction *InstCombiner::commonPointerCastTransforms(CastInst &CI) {
1713   Value *Src = CI.getOperand(0);
1714 
1715   if (GetElementPtrInst *GEP = dyn_cast<GetElementPtrInst>(Src)) {
1716     // If casting the result of a getelementptr instruction with no offset, turn
1717     // this into a cast of the original pointer!
1718     if (GEP->hasAllZeroIndices() &&
1719         // If CI is an addrspacecast and GEP changes the poiner type, merging
1720         // GEP into CI would undo canonicalizing addrspacecast with different
1721         // pointer types, causing infinite loops.
1722         (!isa<AddrSpaceCastInst>(CI) ||
1723          GEP->getType() == GEP->getPointerOperandType())) {
1724       // Changing the cast operand is usually not a good idea but it is safe
1725       // here because the pointer operand is being replaced with another
1726       // pointer operand so the opcode doesn't need to change.
1727       Worklist.Add(GEP);
1728       CI.setOperand(0, GEP->getOperand(0));
1729       return &CI;
1730     }
1731   }
1732 
1733   return commonCastTransforms(CI);
1734 }
1735 
1736 Instruction *InstCombiner::visitPtrToInt(PtrToIntInst &CI) {
1737   // If the destination integer type is not the intptr_t type for this target,
1738   // do a ptrtoint to intptr_t then do a trunc or zext.  This allows the cast
1739   // to be exposed to other transforms.
1740 
1741   Type *Ty = CI.getType();
1742   unsigned AS = CI.getPointerAddressSpace();
1743 
1744   if (Ty->getScalarSizeInBits() == DL.getPointerSizeInBits(AS))
1745     return commonPointerCastTransforms(CI);
1746 
1747   Type *PtrTy = DL.getIntPtrType(CI.getContext(), AS);
1748   if (Ty->isVectorTy()) // Handle vectors of pointers.
1749     PtrTy = VectorType::get(PtrTy, Ty->getVectorNumElements());
1750 
1751   Value *P = Builder.CreatePtrToInt(CI.getOperand(0), PtrTy);
1752   return CastInst::CreateIntegerCast(P, Ty, /*isSigned=*/false);
1753 }
1754 
1755 /// This input value (which is known to have vector type) is being zero extended
1756 /// or truncated to the specified vector type.
1757 /// Try to replace it with a shuffle (and vector/vector bitcast) if possible.
1758 ///
1759 /// The source and destination vector types may have different element types.
1760 static Instruction *optimizeVectorResize(Value *InVal, VectorType *DestTy,
1761                                          InstCombiner &IC) {
1762   // We can only do this optimization if the output is a multiple of the input
1763   // element size, or the input is a multiple of the output element size.
1764   // Convert the input type to have the same element type as the output.
1765   VectorType *SrcTy = cast<VectorType>(InVal->getType());
1766 
1767   if (SrcTy->getElementType() != DestTy->getElementType()) {
1768     // The input types don't need to be identical, but for now they must be the
1769     // same size.  There is no specific reason we couldn't handle things like
1770     // <4 x i16> -> <4 x i32> by bitcasting to <2 x i32> but haven't gotten
1771     // there yet.
1772     if (SrcTy->getElementType()->getPrimitiveSizeInBits() !=
1773         DestTy->getElementType()->getPrimitiveSizeInBits())
1774       return nullptr;
1775 
1776     SrcTy = VectorType::get(DestTy->getElementType(), SrcTy->getNumElements());
1777     InVal = IC.Builder.CreateBitCast(InVal, SrcTy);
1778   }
1779 
1780   // Now that the element types match, get the shuffle mask and RHS of the
1781   // shuffle to use, which depends on whether we're increasing or decreasing the
1782   // size of the input.
1783   SmallVector<uint32_t, 16> ShuffleMask;
1784   Value *V2;
1785 
1786   if (SrcTy->getNumElements() > DestTy->getNumElements()) {
1787     // If we're shrinking the number of elements, just shuffle in the low
1788     // elements from the input and use undef as the second shuffle input.
1789     V2 = UndefValue::get(SrcTy);
1790     for (unsigned i = 0, e = DestTy->getNumElements(); i != e; ++i)
1791       ShuffleMask.push_back(i);
1792 
1793   } else {
1794     // If we're increasing the number of elements, shuffle in all of the
1795     // elements from InVal and fill the rest of the result elements with zeros
1796     // from a constant zero.
1797     V2 = Constant::getNullValue(SrcTy);
1798     unsigned SrcElts = SrcTy->getNumElements();
1799     for (unsigned i = 0, e = SrcElts; i != e; ++i)
1800       ShuffleMask.push_back(i);
1801 
1802     // The excess elements reference the first element of the zero input.
1803     for (unsigned i = 0, e = DestTy->getNumElements()-SrcElts; i != e; ++i)
1804       ShuffleMask.push_back(SrcElts);
1805   }
1806 
1807   return new ShuffleVectorInst(InVal, V2,
1808                                ConstantDataVector::get(V2->getContext(),
1809                                                        ShuffleMask));
1810 }
1811 
1812 static bool isMultipleOfTypeSize(unsigned Value, Type *Ty) {
1813   return Value % Ty->getPrimitiveSizeInBits() == 0;
1814 }
1815 
1816 static unsigned getTypeSizeIndex(unsigned Value, Type *Ty) {
1817   return Value / Ty->getPrimitiveSizeInBits();
1818 }
1819 
1820 /// V is a value which is inserted into a vector of VecEltTy.
1821 /// Look through the value to see if we can decompose it into
1822 /// insertions into the vector.  See the example in the comment for
1823 /// OptimizeIntegerToVectorInsertions for the pattern this handles.
1824 /// The type of V is always a non-zero multiple of VecEltTy's size.
1825 /// Shift is the number of bits between the lsb of V and the lsb of
1826 /// the vector.
1827 ///
1828 /// This returns false if the pattern can't be matched or true if it can,
1829 /// filling in Elements with the elements found here.
1830 static bool collectInsertionElements(Value *V, unsigned Shift,
1831                                      SmallVectorImpl<Value *> &Elements,
1832                                      Type *VecEltTy, bool isBigEndian) {
1833   assert(isMultipleOfTypeSize(Shift, VecEltTy) &&
1834          "Shift should be a multiple of the element type size");
1835 
1836   // Undef values never contribute useful bits to the result.
1837   if (isa<UndefValue>(V)) return true;
1838 
1839   // If we got down to a value of the right type, we win, try inserting into the
1840   // right element.
1841   if (V->getType() == VecEltTy) {
1842     // Inserting null doesn't actually insert any elements.
1843     if (Constant *C = dyn_cast<Constant>(V))
1844       if (C->isNullValue())
1845         return true;
1846 
1847     unsigned ElementIndex = getTypeSizeIndex(Shift, VecEltTy);
1848     if (isBigEndian)
1849       ElementIndex = Elements.size() - ElementIndex - 1;
1850 
1851     // Fail if multiple elements are inserted into this slot.
1852     if (Elements[ElementIndex])
1853       return false;
1854 
1855     Elements[ElementIndex] = V;
1856     return true;
1857   }
1858 
1859   if (Constant *C = dyn_cast<Constant>(V)) {
1860     // Figure out the # elements this provides, and bitcast it or slice it up
1861     // as required.
1862     unsigned NumElts = getTypeSizeIndex(C->getType()->getPrimitiveSizeInBits(),
1863                                         VecEltTy);
1864     // If the constant is the size of a vector element, we just need to bitcast
1865     // it to the right type so it gets properly inserted.
1866     if (NumElts == 1)
1867       return collectInsertionElements(ConstantExpr::getBitCast(C, VecEltTy),
1868                                       Shift, Elements, VecEltTy, isBigEndian);
1869 
1870     // Okay, this is a constant that covers multiple elements.  Slice it up into
1871     // pieces and insert each element-sized piece into the vector.
1872     if (!isa<IntegerType>(C->getType()))
1873       C = ConstantExpr::getBitCast(C, IntegerType::get(V->getContext(),
1874                                        C->getType()->getPrimitiveSizeInBits()));
1875     unsigned ElementSize = VecEltTy->getPrimitiveSizeInBits();
1876     Type *ElementIntTy = IntegerType::get(C->getContext(), ElementSize);
1877 
1878     for (unsigned i = 0; i != NumElts; ++i) {
1879       unsigned ShiftI = Shift+i*ElementSize;
1880       Constant *Piece = ConstantExpr::getLShr(C, ConstantInt::get(C->getType(),
1881                                                                   ShiftI));
1882       Piece = ConstantExpr::getTrunc(Piece, ElementIntTy);
1883       if (!collectInsertionElements(Piece, ShiftI, Elements, VecEltTy,
1884                                     isBigEndian))
1885         return false;
1886     }
1887     return true;
1888   }
1889 
1890   if (!V->hasOneUse()) return false;
1891 
1892   Instruction *I = dyn_cast<Instruction>(V);
1893   if (!I) return false;
1894   switch (I->getOpcode()) {
1895   default: return false; // Unhandled case.
1896   case Instruction::BitCast:
1897     return collectInsertionElements(I->getOperand(0), Shift, Elements, VecEltTy,
1898                                     isBigEndian);
1899   case Instruction::ZExt:
1900     if (!isMultipleOfTypeSize(
1901                           I->getOperand(0)->getType()->getPrimitiveSizeInBits(),
1902                               VecEltTy))
1903       return false;
1904     return collectInsertionElements(I->getOperand(0), Shift, Elements, VecEltTy,
1905                                     isBigEndian);
1906   case Instruction::Or:
1907     return collectInsertionElements(I->getOperand(0), Shift, Elements, VecEltTy,
1908                                     isBigEndian) &&
1909            collectInsertionElements(I->getOperand(1), Shift, Elements, VecEltTy,
1910                                     isBigEndian);
1911   case Instruction::Shl: {
1912     // Must be shifting by a constant that is a multiple of the element size.
1913     ConstantInt *CI = dyn_cast<ConstantInt>(I->getOperand(1));
1914     if (!CI) return false;
1915     Shift += CI->getZExtValue();
1916     if (!isMultipleOfTypeSize(Shift, VecEltTy)) return false;
1917     return collectInsertionElements(I->getOperand(0), Shift, Elements, VecEltTy,
1918                                     isBigEndian);
1919   }
1920 
1921   }
1922 }
1923 
1924 
1925 /// If the input is an 'or' instruction, we may be doing shifts and ors to
1926 /// assemble the elements of the vector manually.
1927 /// Try to rip the code out and replace it with insertelements.  This is to
1928 /// optimize code like this:
1929 ///
1930 ///    %tmp37 = bitcast float %inc to i32
1931 ///    %tmp38 = zext i32 %tmp37 to i64
1932 ///    %tmp31 = bitcast float %inc5 to i32
1933 ///    %tmp32 = zext i32 %tmp31 to i64
1934 ///    %tmp33 = shl i64 %tmp32, 32
1935 ///    %ins35 = or i64 %tmp33, %tmp38
1936 ///    %tmp43 = bitcast i64 %ins35 to <2 x float>
1937 ///
1938 /// Into two insertelements that do "buildvector{%inc, %inc5}".
1939 static Value *optimizeIntegerToVectorInsertions(BitCastInst &CI,
1940                                                 InstCombiner &IC) {
1941   VectorType *DestVecTy = cast<VectorType>(CI.getType());
1942   Value *IntInput = CI.getOperand(0);
1943 
1944   SmallVector<Value*, 8> Elements(DestVecTy->getNumElements());
1945   if (!collectInsertionElements(IntInput, 0, Elements,
1946                                 DestVecTy->getElementType(),
1947                                 IC.getDataLayout().isBigEndian()))
1948     return nullptr;
1949 
1950   // If we succeeded, we know that all of the element are specified by Elements
1951   // or are zero if Elements has a null entry.  Recast this as a set of
1952   // insertions.
1953   Value *Result = Constant::getNullValue(CI.getType());
1954   for (unsigned i = 0, e = Elements.size(); i != e; ++i) {
1955     if (!Elements[i]) continue;  // Unset element.
1956 
1957     Result = IC.Builder.CreateInsertElement(Result, Elements[i],
1958                                             IC.Builder.getInt32(i));
1959   }
1960 
1961   return Result;
1962 }
1963 
1964 /// Canonicalize scalar bitcasts of extracted elements into a bitcast of the
1965 /// vector followed by extract element. The backend tends to handle bitcasts of
1966 /// vectors better than bitcasts of scalars because vector registers are
1967 /// usually not type-specific like scalar integer or scalar floating-point.
1968 static Instruction *canonicalizeBitCastExtElt(BitCastInst &BitCast,
1969                                               InstCombiner &IC) {
1970   // TODO: Create and use a pattern matcher for ExtractElementInst.
1971   auto *ExtElt = dyn_cast<ExtractElementInst>(BitCast.getOperand(0));
1972   if (!ExtElt || !ExtElt->hasOneUse())
1973     return nullptr;
1974 
1975   // The bitcast must be to a vectorizable type, otherwise we can't make a new
1976   // type to extract from.
1977   Type *DestType = BitCast.getType();
1978   if (!VectorType::isValidElementType(DestType))
1979     return nullptr;
1980 
1981   unsigned NumElts = ExtElt->getVectorOperandType()->getNumElements();
1982   auto *NewVecType = VectorType::get(DestType, NumElts);
1983   auto *NewBC = IC.Builder.CreateBitCast(ExtElt->getVectorOperand(),
1984                                          NewVecType, "bc");
1985   return ExtractElementInst::Create(NewBC, ExtElt->getIndexOperand());
1986 }
1987 
1988 /// Change the type of a bitwise logic operation if we can eliminate a bitcast.
1989 static Instruction *foldBitCastBitwiseLogic(BitCastInst &BitCast,
1990                                             InstCombiner::BuilderTy &Builder) {
1991   Type *DestTy = BitCast.getType();
1992   BinaryOperator *BO;
1993   if (!DestTy->isIntOrIntVectorTy() ||
1994       !match(BitCast.getOperand(0), m_OneUse(m_BinOp(BO))) ||
1995       !BO->isBitwiseLogicOp())
1996     return nullptr;
1997 
1998   // FIXME: This transform is restricted to vector types to avoid backend
1999   // problems caused by creating potentially illegal operations. If a fix-up is
2000   // added to handle that situation, we can remove this check.
2001   if (!DestTy->isVectorTy() || !BO->getType()->isVectorTy())
2002     return nullptr;
2003 
2004   Value *X;
2005   if (match(BO->getOperand(0), m_OneUse(m_BitCast(m_Value(X)))) &&
2006       X->getType() == DestTy && !isa<Constant>(X)) {
2007     // bitcast(logic(bitcast(X), Y)) --> logic'(X, bitcast(Y))
2008     Value *CastedOp1 = Builder.CreateBitCast(BO->getOperand(1), DestTy);
2009     return BinaryOperator::Create(BO->getOpcode(), X, CastedOp1);
2010   }
2011 
2012   if (match(BO->getOperand(1), m_OneUse(m_BitCast(m_Value(X)))) &&
2013       X->getType() == DestTy && !isa<Constant>(X)) {
2014     // bitcast(logic(Y, bitcast(X))) --> logic'(bitcast(Y), X)
2015     Value *CastedOp0 = Builder.CreateBitCast(BO->getOperand(0), DestTy);
2016     return BinaryOperator::Create(BO->getOpcode(), CastedOp0, X);
2017   }
2018 
2019   // Canonicalize vector bitcasts to come before vector bitwise logic with a
2020   // constant. This eases recognition of special constants for later ops.
2021   // Example:
2022   // icmp u/s (a ^ signmask), (b ^ signmask) --> icmp s/u a, b
2023   Constant *C;
2024   if (match(BO->getOperand(1), m_Constant(C))) {
2025     // bitcast (logic X, C) --> logic (bitcast X, C')
2026     Value *CastedOp0 = Builder.CreateBitCast(BO->getOperand(0), DestTy);
2027     Value *CastedC = ConstantExpr::getBitCast(C, DestTy);
2028     return BinaryOperator::Create(BO->getOpcode(), CastedOp0, CastedC);
2029   }
2030 
2031   return nullptr;
2032 }
2033 
2034 /// Change the type of a select if we can eliminate a bitcast.
2035 static Instruction *foldBitCastSelect(BitCastInst &BitCast,
2036                                       InstCombiner::BuilderTy &Builder) {
2037   Value *Cond, *TVal, *FVal;
2038   if (!match(BitCast.getOperand(0),
2039              m_OneUse(m_Select(m_Value(Cond), m_Value(TVal), m_Value(FVal)))))
2040     return nullptr;
2041 
2042   // A vector select must maintain the same number of elements in its operands.
2043   Type *CondTy = Cond->getType();
2044   Type *DestTy = BitCast.getType();
2045   if (CondTy->isVectorTy()) {
2046     if (!DestTy->isVectorTy())
2047       return nullptr;
2048     if (DestTy->getVectorNumElements() != CondTy->getVectorNumElements())
2049       return nullptr;
2050   }
2051 
2052   // FIXME: This transform is restricted from changing the select between
2053   // scalars and vectors to avoid backend problems caused by creating
2054   // potentially illegal operations. If a fix-up is added to handle that
2055   // situation, we can remove this check.
2056   if (DestTy->isVectorTy() != TVal->getType()->isVectorTy())
2057     return nullptr;
2058 
2059   auto *Sel = cast<Instruction>(BitCast.getOperand(0));
2060   Value *X;
2061   if (match(TVal, m_OneUse(m_BitCast(m_Value(X)))) && X->getType() == DestTy &&
2062       !isa<Constant>(X)) {
2063     // bitcast(select(Cond, bitcast(X), Y)) --> select'(Cond, X, bitcast(Y))
2064     Value *CastedVal = Builder.CreateBitCast(FVal, DestTy);
2065     return SelectInst::Create(Cond, X, CastedVal, "", nullptr, Sel);
2066   }
2067 
2068   if (match(FVal, m_OneUse(m_BitCast(m_Value(X)))) && X->getType() == DestTy &&
2069       !isa<Constant>(X)) {
2070     // bitcast(select(Cond, Y, bitcast(X))) --> select'(Cond, bitcast(Y), X)
2071     Value *CastedVal = Builder.CreateBitCast(TVal, DestTy);
2072     return SelectInst::Create(Cond, CastedVal, X, "", nullptr, Sel);
2073   }
2074 
2075   return nullptr;
2076 }
2077 
2078 /// Check if all users of CI are StoreInsts.
2079 static bool hasStoreUsersOnly(CastInst &CI) {
2080   for (User *U : CI.users()) {
2081     if (!isa<StoreInst>(U))
2082       return false;
2083   }
2084   return true;
2085 }
2086 
2087 /// This function handles following case
2088 ///
2089 ///     A  ->  B    cast
2090 ///     PHI
2091 ///     B  ->  A    cast
2092 ///
2093 /// All the related PHI nodes can be replaced by new PHI nodes with type A.
2094 /// The uses of \p CI can be changed to the new PHI node corresponding to \p PN.
2095 Instruction *InstCombiner::optimizeBitCastFromPhi(CastInst &CI, PHINode *PN) {
2096   // BitCast used by Store can be handled in InstCombineLoadStoreAlloca.cpp.
2097   if (hasStoreUsersOnly(CI))
2098     return nullptr;
2099 
2100   Value *Src = CI.getOperand(0);
2101   Type *SrcTy = Src->getType();         // Type B
2102   Type *DestTy = CI.getType();          // Type A
2103 
2104   SmallVector<PHINode *, 4> PhiWorklist;
2105   SmallSetVector<PHINode *, 4> OldPhiNodes;
2106 
2107   // Find all of the A->B casts and PHI nodes.
2108   // We need to inpect all related PHI nodes, but PHIs can be cyclic, so
2109   // OldPhiNodes is used to track all known PHI nodes, before adding a new
2110   // PHI to PhiWorklist, it is checked against and added to OldPhiNodes first.
2111   PhiWorklist.push_back(PN);
2112   OldPhiNodes.insert(PN);
2113   while (!PhiWorklist.empty()) {
2114     auto *OldPN = PhiWorklist.pop_back_val();
2115     for (Value *IncValue : OldPN->incoming_values()) {
2116       if (isa<Constant>(IncValue))
2117         continue;
2118 
2119       if (auto *LI = dyn_cast<LoadInst>(IncValue)) {
2120         // If there is a sequence of one or more load instructions, each loaded
2121         // value is used as address of later load instruction, bitcast is
2122         // necessary to change the value type, don't optimize it. For
2123         // simplicity we give up if the load address comes from another load.
2124         Value *Addr = LI->getOperand(0);
2125         if (Addr == &CI || isa<LoadInst>(Addr))
2126           return nullptr;
2127         if (LI->hasOneUse() && LI->isSimple())
2128           continue;
2129         // If a LoadInst has more than one use, changing the type of loaded
2130         // value may create another bitcast.
2131         return nullptr;
2132       }
2133 
2134       if (auto *PNode = dyn_cast<PHINode>(IncValue)) {
2135         if (OldPhiNodes.insert(PNode))
2136           PhiWorklist.push_back(PNode);
2137         continue;
2138       }
2139 
2140       auto *BCI = dyn_cast<BitCastInst>(IncValue);
2141       // We can't handle other instructions.
2142       if (!BCI)
2143         return nullptr;
2144 
2145       // Verify it's a A->B cast.
2146       Type *TyA = BCI->getOperand(0)->getType();
2147       Type *TyB = BCI->getType();
2148       if (TyA != DestTy || TyB != SrcTy)
2149         return nullptr;
2150     }
2151   }
2152 
2153   // For each old PHI node, create a corresponding new PHI node with a type A.
2154   SmallDenseMap<PHINode *, PHINode *> NewPNodes;
2155   for (auto *OldPN : OldPhiNodes) {
2156     Builder.SetInsertPoint(OldPN);
2157     PHINode *NewPN = Builder.CreatePHI(DestTy, OldPN->getNumOperands());
2158     NewPNodes[OldPN] = NewPN;
2159   }
2160 
2161   // Fill in the operands of new PHI nodes.
2162   for (auto *OldPN : OldPhiNodes) {
2163     PHINode *NewPN = NewPNodes[OldPN];
2164     for (unsigned j = 0, e = OldPN->getNumOperands(); j != e; ++j) {
2165       Value *V = OldPN->getOperand(j);
2166       Value *NewV = nullptr;
2167       if (auto *C = dyn_cast<Constant>(V)) {
2168         NewV = ConstantExpr::getBitCast(C, DestTy);
2169       } else if (auto *LI = dyn_cast<LoadInst>(V)) {
2170         Builder.SetInsertPoint(LI->getNextNode());
2171         NewV = Builder.CreateBitCast(LI, DestTy);
2172         Worklist.Add(LI);
2173       } else if (auto *BCI = dyn_cast<BitCastInst>(V)) {
2174         NewV = BCI->getOperand(0);
2175       } else if (auto *PrevPN = dyn_cast<PHINode>(V)) {
2176         NewV = NewPNodes[PrevPN];
2177       }
2178       assert(NewV);
2179       NewPN->addIncoming(NewV, OldPN->getIncomingBlock(j));
2180     }
2181   }
2182 
2183   // If there is a store with type B, change it to type A.
2184   for (User *U : PN->users()) {
2185     auto *SI = dyn_cast<StoreInst>(U);
2186     if (SI && SI->isSimple() && SI->getOperand(0) == PN) {
2187       Builder.SetInsertPoint(SI);
2188       auto *NewBC =
2189           cast<BitCastInst>(Builder.CreateBitCast(NewPNodes[PN], SrcTy));
2190       SI->setOperand(0, NewBC);
2191       Worklist.Add(SI);
2192       assert(hasStoreUsersOnly(*NewBC));
2193     }
2194   }
2195 
2196   return replaceInstUsesWith(CI, NewPNodes[PN]);
2197 }
2198 
2199 Instruction *InstCombiner::visitBitCast(BitCastInst &CI) {
2200   // If the operands are integer typed then apply the integer transforms,
2201   // otherwise just apply the common ones.
2202   Value *Src = CI.getOperand(0);
2203   Type *SrcTy = Src->getType();
2204   Type *DestTy = CI.getType();
2205 
2206   // Get rid of casts from one type to the same type. These are useless and can
2207   // be replaced by the operand.
2208   if (DestTy == Src->getType())
2209     return replaceInstUsesWith(CI, Src);
2210 
2211   if (PointerType *DstPTy = dyn_cast<PointerType>(DestTy)) {
2212     PointerType *SrcPTy = cast<PointerType>(SrcTy);
2213     Type *DstElTy = DstPTy->getElementType();
2214     Type *SrcElTy = SrcPTy->getElementType();
2215 
2216     // If we are casting a alloca to a pointer to a type of the same
2217     // size, rewrite the allocation instruction to allocate the "right" type.
2218     // There is no need to modify malloc calls because it is their bitcast that
2219     // needs to be cleaned up.
2220     if (AllocaInst *AI = dyn_cast<AllocaInst>(Src))
2221       if (Instruction *V = PromoteCastOfAllocation(CI, *AI))
2222         return V;
2223 
2224     // When the type pointed to is not sized the cast cannot be
2225     // turned into a gep.
2226     Type *PointeeType =
2227         cast<PointerType>(Src->getType()->getScalarType())->getElementType();
2228     if (!PointeeType->isSized())
2229       return nullptr;
2230 
2231     // If the source and destination are pointers, and this cast is equivalent
2232     // to a getelementptr X, 0, 0, 0...  turn it into the appropriate gep.
2233     // This can enhance SROA and other transforms that want type-safe pointers.
2234     unsigned NumZeros = 0;
2235     while (SrcElTy != DstElTy &&
2236            isa<CompositeType>(SrcElTy) && !SrcElTy->isPointerTy() &&
2237            SrcElTy->getNumContainedTypes() /* not "{}" */) {
2238       SrcElTy = cast<CompositeType>(SrcElTy)->getTypeAtIndex(0U);
2239       ++NumZeros;
2240     }
2241 
2242     // If we found a path from the src to dest, create the getelementptr now.
2243     if (SrcElTy == DstElTy) {
2244       SmallVector<Value *, 8> Idxs(NumZeros + 1, Builder.getInt32(0));
2245       return GetElementPtrInst::CreateInBounds(Src, Idxs);
2246     }
2247   }
2248 
2249   if (VectorType *DestVTy = dyn_cast<VectorType>(DestTy)) {
2250     if (DestVTy->getNumElements() == 1 && !SrcTy->isVectorTy()) {
2251       Value *Elem = Builder.CreateBitCast(Src, DestVTy->getElementType());
2252       return InsertElementInst::Create(UndefValue::get(DestTy), Elem,
2253                      Constant::getNullValue(Type::getInt32Ty(CI.getContext())));
2254       // FIXME: Canonicalize bitcast(insertelement) -> insertelement(bitcast)
2255     }
2256 
2257     if (isa<IntegerType>(SrcTy)) {
2258       // If this is a cast from an integer to vector, check to see if the input
2259       // is a trunc or zext of a bitcast from vector.  If so, we can replace all
2260       // the casts with a shuffle and (potentially) a bitcast.
2261       if (isa<TruncInst>(Src) || isa<ZExtInst>(Src)) {
2262         CastInst *SrcCast = cast<CastInst>(Src);
2263         if (BitCastInst *BCIn = dyn_cast<BitCastInst>(SrcCast->getOperand(0)))
2264           if (isa<VectorType>(BCIn->getOperand(0)->getType()))
2265             if (Instruction *I = optimizeVectorResize(BCIn->getOperand(0),
2266                                                cast<VectorType>(DestTy), *this))
2267               return I;
2268       }
2269 
2270       // If the input is an 'or' instruction, we may be doing shifts and ors to
2271       // assemble the elements of the vector manually.  Try to rip the code out
2272       // and replace it with insertelements.
2273       if (Value *V = optimizeIntegerToVectorInsertions(CI, *this))
2274         return replaceInstUsesWith(CI, V);
2275     }
2276   }
2277 
2278   if (VectorType *SrcVTy = dyn_cast<VectorType>(SrcTy)) {
2279     if (SrcVTy->getNumElements() == 1) {
2280       // If our destination is not a vector, then make this a straight
2281       // scalar-scalar cast.
2282       if (!DestTy->isVectorTy()) {
2283         Value *Elem =
2284           Builder.CreateExtractElement(Src,
2285                      Constant::getNullValue(Type::getInt32Ty(CI.getContext())));
2286         return CastInst::Create(Instruction::BitCast, Elem, DestTy);
2287       }
2288 
2289       // Otherwise, see if our source is an insert. If so, then use the scalar
2290       // component directly.
2291       if (InsertElementInst *IEI =
2292             dyn_cast<InsertElementInst>(CI.getOperand(0)))
2293         return CastInst::Create(Instruction::BitCast, IEI->getOperand(1),
2294                                 DestTy);
2295     }
2296   }
2297 
2298   if (ShuffleVectorInst *SVI = dyn_cast<ShuffleVectorInst>(Src)) {
2299     // Okay, we have (bitcast (shuffle ..)).  Check to see if this is
2300     // a bitcast to a vector with the same # elts.
2301     if (SVI->hasOneUse() && DestTy->isVectorTy() &&
2302         DestTy->getVectorNumElements() == SVI->getType()->getNumElements() &&
2303         SVI->getType()->getNumElements() ==
2304         SVI->getOperand(0)->getType()->getVectorNumElements()) {
2305       BitCastInst *Tmp;
2306       // If either of the operands is a cast from CI.getType(), then
2307       // evaluating the shuffle in the casted destination's type will allow
2308       // us to eliminate at least one cast.
2309       if (((Tmp = dyn_cast<BitCastInst>(SVI->getOperand(0))) &&
2310            Tmp->getOperand(0)->getType() == DestTy) ||
2311           ((Tmp = dyn_cast<BitCastInst>(SVI->getOperand(1))) &&
2312            Tmp->getOperand(0)->getType() == DestTy)) {
2313         Value *LHS = Builder.CreateBitCast(SVI->getOperand(0), DestTy);
2314         Value *RHS = Builder.CreateBitCast(SVI->getOperand(1), DestTy);
2315         // Return a new shuffle vector.  Use the same element ID's, as we
2316         // know the vector types match #elts.
2317         return new ShuffleVectorInst(LHS, RHS, SVI->getOperand(2));
2318       }
2319     }
2320   }
2321 
2322   // Handle the A->B->A cast, and there is an intervening PHI node.
2323   if (PHINode *PN = dyn_cast<PHINode>(Src))
2324     if (Instruction *I = optimizeBitCastFromPhi(CI, PN))
2325       return I;
2326 
2327   if (Instruction *I = canonicalizeBitCastExtElt(CI, *this))
2328     return I;
2329 
2330   if (Instruction *I = foldBitCastBitwiseLogic(CI, Builder))
2331     return I;
2332 
2333   if (Instruction *I = foldBitCastSelect(CI, Builder))
2334     return I;
2335 
2336   if (SrcTy->isPointerTy())
2337     return commonPointerCastTransforms(CI);
2338   return commonCastTransforms(CI);
2339 }
2340 
2341 Instruction *InstCombiner::visitAddrSpaceCast(AddrSpaceCastInst &CI) {
2342   // If the destination pointer element type is not the same as the source's
2343   // first do a bitcast to the destination type, and then the addrspacecast.
2344   // This allows the cast to be exposed to other transforms.
2345   Value *Src = CI.getOperand(0);
2346   PointerType *SrcTy = cast<PointerType>(Src->getType()->getScalarType());
2347   PointerType *DestTy = cast<PointerType>(CI.getType()->getScalarType());
2348 
2349   Type *DestElemTy = DestTy->getElementType();
2350   if (SrcTy->getElementType() != DestElemTy) {
2351     Type *MidTy = PointerType::get(DestElemTy, SrcTy->getAddressSpace());
2352     if (VectorType *VT = dyn_cast<VectorType>(CI.getType())) {
2353       // Handle vectors of pointers.
2354       MidTy = VectorType::get(MidTy, VT->getNumElements());
2355     }
2356 
2357     Value *NewBitCast = Builder.CreateBitCast(Src, MidTy);
2358     return new AddrSpaceCastInst(NewBitCast, CI.getType());
2359   }
2360 
2361   return commonPointerCastTransforms(CI);
2362 }
2363