Use getZExtOrTrunc helper instead of manually doing zext/trunc check. NFC.
[oota-llvm.git] / lib / CodeGen / SelectionDAG / SelectionDAG.cpp
1 //===-- SelectionDAG.cpp - Implement the SelectionDAG data structures -----===//
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 implements the SelectionDAG class.
11 //
12 //===----------------------------------------------------------------------===//
13
14 #include "llvm/CodeGen/SelectionDAG.h"
15 #include "SDNodeDbgValue.h"
16 #include "llvm/ADT/SetVector.h"
17 #include "llvm/ADT/SmallPtrSet.h"
18 #include "llvm/ADT/SmallSet.h"
19 #include "llvm/ADT/SmallVector.h"
20 #include "llvm/ADT/StringExtras.h"
21 #include "llvm/Analysis/ValueTracking.h"
22 #include "llvm/CodeGen/MachineBasicBlock.h"
23 #include "llvm/CodeGen/MachineConstantPool.h"
24 #include "llvm/CodeGen/MachineFrameInfo.h"
25 #include "llvm/CodeGen/MachineModuleInfo.h"
26 #include "llvm/IR/CallingConv.h"
27 #include "llvm/IR/Constants.h"
28 #include "llvm/IR/DataLayout.h"
29 #include "llvm/IR/DebugInfo.h"
30 #include "llvm/IR/DerivedTypes.h"
31 #include "llvm/IR/Function.h"
32 #include "llvm/IR/GlobalAlias.h"
33 #include "llvm/IR/GlobalVariable.h"
34 #include "llvm/IR/Intrinsics.h"
35 #include "llvm/Support/CommandLine.h"
36 #include "llvm/Support/Debug.h"
37 #include "llvm/Support/ErrorHandling.h"
38 #include "llvm/Support/ManagedStatic.h"
39 #include "llvm/Support/MathExtras.h"
40 #include "llvm/Support/Mutex.h"
41 #include "llvm/Support/raw_ostream.h"
42 #include "llvm/Target/TargetInstrInfo.h"
43 #include "llvm/Target/TargetIntrinsicInfo.h"
44 #include "llvm/Target/TargetLowering.h"
45 #include "llvm/Target/TargetMachine.h"
46 #include "llvm/Target/TargetOptions.h"
47 #include "llvm/Target/TargetRegisterInfo.h"
48 #include "llvm/Target/TargetSelectionDAGInfo.h"
49 #include "llvm/Target/TargetSubtargetInfo.h"
50 #include <algorithm>
51 #include <cmath>
52 #include <utility>
53
54 using namespace llvm;
55
56 /// makeVTList - Return an instance of the SDVTList struct initialized with the
57 /// specified members.
58 static SDVTList makeVTList(const EVT *VTs, unsigned NumVTs) {
59   SDVTList Res = {VTs, NumVTs};
60   return Res;
61 }
62
63 // Default null implementations of the callbacks.
64 void SelectionDAG::DAGUpdateListener::NodeDeleted(SDNode*, SDNode*) {}
65 void SelectionDAG::DAGUpdateListener::NodeUpdated(SDNode*) {}
66
67 //===----------------------------------------------------------------------===//
68 //                              ConstantFPSDNode Class
69 //===----------------------------------------------------------------------===//
70
71 /// isExactlyValue - We don't rely on operator== working on double values, as
72 /// it returns true for things that are clearly not equal, like -0.0 and 0.0.
73 /// As such, this method can be used to do an exact bit-for-bit comparison of
74 /// two floating point values.
75 bool ConstantFPSDNode::isExactlyValue(const APFloat& V) const {
76   return getValueAPF().bitwiseIsEqual(V);
77 }
78
79 bool ConstantFPSDNode::isValueValidForType(EVT VT,
80                                            const APFloat& Val) {
81   assert(VT.isFloatingPoint() && "Can only convert between FP types");
82
83   // convert modifies in place, so make a copy.
84   APFloat Val2 = APFloat(Val);
85   bool losesInfo;
86   (void) Val2.convert(SelectionDAG::EVTToAPFloatSemantics(VT),
87                       APFloat::rmNearestTiesToEven,
88                       &losesInfo);
89   return !losesInfo;
90 }
91
92 //===----------------------------------------------------------------------===//
93 //                              ISD Namespace
94 //===----------------------------------------------------------------------===//
95
96 /// isBuildVectorAllOnes - Return true if the specified node is a
97 /// BUILD_VECTOR where all of the elements are ~0 or undef.
98 bool ISD::isBuildVectorAllOnes(const SDNode *N) {
99   // Look through a bit convert.
100   while (N->getOpcode() == ISD::BITCAST)
101     N = N->getOperand(0).getNode();
102
103   if (N->getOpcode() != ISD::BUILD_VECTOR) return false;
104
105   unsigned i = 0, e = N->getNumOperands();
106
107   // Skip over all of the undef values.
108   while (i != e && N->getOperand(i).getOpcode() == ISD::UNDEF)
109     ++i;
110
111   // Do not accept an all-undef vector.
112   if (i == e) return false;
113
114   // Do not accept build_vectors that aren't all constants or which have non-~0
115   // elements. We have to be a bit careful here, as the type of the constant
116   // may not be the same as the type of the vector elements due to type
117   // legalization (the elements are promoted to a legal type for the target and
118   // a vector of a type may be legal when the base element type is not).
119   // We only want to check enough bits to cover the vector elements, because
120   // we care if the resultant vector is all ones, not whether the individual
121   // constants are.
122   SDValue NotZero = N->getOperand(i);
123   unsigned EltSize = N->getValueType(0).getVectorElementType().getSizeInBits();
124   if (ConstantSDNode *CN = dyn_cast<ConstantSDNode>(NotZero)) {
125     if (CN->getAPIntValue().countTrailingOnes() < EltSize)
126       return false;
127   } else if (ConstantFPSDNode *CFPN = dyn_cast<ConstantFPSDNode>(NotZero)) {
128     if (CFPN->getValueAPF().bitcastToAPInt().countTrailingOnes() < EltSize)
129       return false;
130   } else
131     return false;
132
133   // Okay, we have at least one ~0 value, check to see if the rest match or are
134   // undefs. Even with the above element type twiddling, this should be OK, as
135   // the same type legalization should have applied to all the elements.
136   for (++i; i != e; ++i)
137     if (N->getOperand(i) != NotZero &&
138         N->getOperand(i).getOpcode() != ISD::UNDEF)
139       return false;
140   return true;
141 }
142
143
144 /// isBuildVectorAllZeros - Return true if the specified node is a
145 /// BUILD_VECTOR where all of the elements are 0 or undef.
146 bool ISD::isBuildVectorAllZeros(const SDNode *N) {
147   // Look through a bit convert.
148   while (N->getOpcode() == ISD::BITCAST)
149     N = N->getOperand(0).getNode();
150
151   if (N->getOpcode() != ISD::BUILD_VECTOR) return false;
152
153   bool IsAllUndef = true;
154   for (const SDValue &Op : N->op_values()) {
155     if (Op.getOpcode() == ISD::UNDEF)
156       continue;
157     IsAllUndef = false;
158     // Do not accept build_vectors that aren't all constants or which have non-0
159     // elements. We have to be a bit careful here, as the type of the constant
160     // may not be the same as the type of the vector elements due to type
161     // legalization (the elements are promoted to a legal type for the target
162     // and a vector of a type may be legal when the base element type is not).
163     // We only want to check enough bits to cover the vector elements, because
164     // we care if the resultant vector is all zeros, not whether the individual
165     // constants are.
166     unsigned EltSize = N->getValueType(0).getVectorElementType().getSizeInBits();
167     if (ConstantSDNode *CN = dyn_cast<ConstantSDNode>(Op)) {
168       if (CN->getAPIntValue().countTrailingZeros() < EltSize)
169         return false;
170     } else if (ConstantFPSDNode *CFPN = dyn_cast<ConstantFPSDNode>(Op)) {
171       if (CFPN->getValueAPF().bitcastToAPInt().countTrailingZeros() < EltSize)
172         return false;
173     } else
174       return false;
175   }
176
177   // Do not accept an all-undef vector.
178   if (IsAllUndef)
179     return false;
180   return true;
181 }
182
183 /// \brief Return true if the specified node is a BUILD_VECTOR node of
184 /// all ConstantSDNode or undef.
185 bool ISD::isBuildVectorOfConstantSDNodes(const SDNode *N) {
186   if (N->getOpcode() != ISD::BUILD_VECTOR)
187     return false;
188
189   for (const SDValue &Op : N->op_values()) {
190     if (Op.getOpcode() == ISD::UNDEF)
191       continue;
192     if (!isa<ConstantSDNode>(Op))
193       return false;
194   }
195   return true;
196 }
197
198 /// \brief Return true if the specified node is a BUILD_VECTOR node of
199 /// all ConstantFPSDNode or undef.
200 bool ISD::isBuildVectorOfConstantFPSDNodes(const SDNode *N) {
201   if (N->getOpcode() != ISD::BUILD_VECTOR)
202     return false;
203
204   for (const SDValue &Op : N->op_values()) {
205     if (Op.getOpcode() == ISD::UNDEF)
206       continue;
207     if (!isa<ConstantFPSDNode>(Op))
208       return false;
209   }
210   return true;
211 }
212
213 /// isScalarToVector - Return true if the specified node is a
214 /// ISD::SCALAR_TO_VECTOR node or a BUILD_VECTOR node where only the low
215 /// element is not an undef.
216 bool ISD::isScalarToVector(const SDNode *N) {
217   if (N->getOpcode() == ISD::SCALAR_TO_VECTOR)
218     return true;
219
220   if (N->getOpcode() != ISD::BUILD_VECTOR)
221     return false;
222   if (N->getOperand(0).getOpcode() == ISD::UNDEF)
223     return false;
224   unsigned NumElems = N->getNumOperands();
225   if (NumElems == 1)
226     return false;
227   for (unsigned i = 1; i < NumElems; ++i) {
228     SDValue V = N->getOperand(i);
229     if (V.getOpcode() != ISD::UNDEF)
230       return false;
231   }
232   return true;
233 }
234
235 /// allOperandsUndef - Return true if the node has at least one operand
236 /// and all operands of the specified node are ISD::UNDEF.
237 bool ISD::allOperandsUndef(const SDNode *N) {
238   // Return false if the node has no operands.
239   // This is "logically inconsistent" with the definition of "all" but
240   // is probably the desired behavior.
241   if (N->getNumOperands() == 0)
242     return false;
243
244   for (const SDValue &Op : N->op_values())
245     if (Op.getOpcode() != ISD::UNDEF)
246       return false;
247
248   return true;
249 }
250
251 ISD::NodeType ISD::getExtForLoadExtType(bool IsFP, ISD::LoadExtType ExtType) {
252   switch (ExtType) {
253   case ISD::EXTLOAD:
254     return IsFP ? ISD::FP_EXTEND : ISD::ANY_EXTEND;
255   case ISD::SEXTLOAD:
256     return ISD::SIGN_EXTEND;
257   case ISD::ZEXTLOAD:
258     return ISD::ZERO_EXTEND;
259   default:
260     break;
261   }
262
263   llvm_unreachable("Invalid LoadExtType");
264 }
265
266 /// getSetCCSwappedOperands - Return the operation corresponding to (Y op X)
267 /// when given the operation for (X op Y).
268 ISD::CondCode ISD::getSetCCSwappedOperands(ISD::CondCode Operation) {
269   // To perform this operation, we just need to swap the L and G bits of the
270   // operation.
271   unsigned OldL = (Operation >> 2) & 1;
272   unsigned OldG = (Operation >> 1) & 1;
273   return ISD::CondCode((Operation & ~6) |  // Keep the N, U, E bits
274                        (OldL << 1) |       // New G bit
275                        (OldG << 2));       // New L bit.
276 }
277
278 /// getSetCCInverse - Return the operation corresponding to !(X op Y), where
279 /// 'op' is a valid SetCC operation.
280 ISD::CondCode ISD::getSetCCInverse(ISD::CondCode Op, bool isInteger) {
281   unsigned Operation = Op;
282   if (isInteger)
283     Operation ^= 7;   // Flip L, G, E bits, but not U.
284   else
285     Operation ^= 15;  // Flip all of the condition bits.
286
287   if (Operation > ISD::SETTRUE2)
288     Operation &= ~8;  // Don't let N and U bits get set.
289
290   return ISD::CondCode(Operation);
291 }
292
293
294 /// isSignedOp - For an integer comparison, return 1 if the comparison is a
295 /// signed operation and 2 if the result is an unsigned comparison.  Return zero
296 /// if the operation does not depend on the sign of the input (setne and seteq).
297 static int isSignedOp(ISD::CondCode Opcode) {
298   switch (Opcode) {
299   default: llvm_unreachable("Illegal integer setcc operation!");
300   case ISD::SETEQ:
301   case ISD::SETNE: return 0;
302   case ISD::SETLT:
303   case ISD::SETLE:
304   case ISD::SETGT:
305   case ISD::SETGE: return 1;
306   case ISD::SETULT:
307   case ISD::SETULE:
308   case ISD::SETUGT:
309   case ISD::SETUGE: return 2;
310   }
311 }
312
313 /// getSetCCOrOperation - Return the result of a logical OR between different
314 /// comparisons of identical values: ((X op1 Y) | (X op2 Y)).  This function
315 /// returns SETCC_INVALID if it is not possible to represent the resultant
316 /// comparison.
317 ISD::CondCode ISD::getSetCCOrOperation(ISD::CondCode Op1, ISD::CondCode Op2,
318                                        bool isInteger) {
319   if (isInteger && (isSignedOp(Op1) | isSignedOp(Op2)) == 3)
320     // Cannot fold a signed integer setcc with an unsigned integer setcc.
321     return ISD::SETCC_INVALID;
322
323   unsigned Op = Op1 | Op2;  // Combine all of the condition bits.
324
325   // If the N and U bits get set then the resultant comparison DOES suddenly
326   // care about orderedness, and is true when ordered.
327   if (Op > ISD::SETTRUE2)
328     Op &= ~16;     // Clear the U bit if the N bit is set.
329
330   // Canonicalize illegal integer setcc's.
331   if (isInteger && Op == ISD::SETUNE)  // e.g. SETUGT | SETULT
332     Op = ISD::SETNE;
333
334   return ISD::CondCode(Op);
335 }
336
337 /// getSetCCAndOperation - Return the result of a logical AND between different
338 /// comparisons of identical values: ((X op1 Y) & (X op2 Y)).  This
339 /// function returns zero if it is not possible to represent the resultant
340 /// comparison.
341 ISD::CondCode ISD::getSetCCAndOperation(ISD::CondCode Op1, ISD::CondCode Op2,
342                                         bool isInteger) {
343   if (isInteger && (isSignedOp(Op1) | isSignedOp(Op2)) == 3)
344     // Cannot fold a signed setcc with an unsigned setcc.
345     return ISD::SETCC_INVALID;
346
347   // Combine all of the condition bits.
348   ISD::CondCode Result = ISD::CondCode(Op1 & Op2);
349
350   // Canonicalize illegal integer setcc's.
351   if (isInteger) {
352     switch (Result) {
353     default: break;
354     case ISD::SETUO : Result = ISD::SETFALSE; break;  // SETUGT & SETULT
355     case ISD::SETOEQ:                                 // SETEQ  & SETU[LG]E
356     case ISD::SETUEQ: Result = ISD::SETEQ   ; break;  // SETUGE & SETULE
357     case ISD::SETOLT: Result = ISD::SETULT  ; break;  // SETULT & SETNE
358     case ISD::SETOGT: Result = ISD::SETUGT  ; break;  // SETUGT & SETNE
359     }
360   }
361
362   return Result;
363 }
364
365 //===----------------------------------------------------------------------===//
366 //                           SDNode Profile Support
367 //===----------------------------------------------------------------------===//
368
369 /// AddNodeIDOpcode - Add the node opcode to the NodeID data.
370 ///
371 static void AddNodeIDOpcode(FoldingSetNodeID &ID, unsigned OpC)  {
372   ID.AddInteger(OpC);
373 }
374
375 /// AddNodeIDValueTypes - Value type lists are intern'd so we can represent them
376 /// solely with their pointer.
377 static void AddNodeIDValueTypes(FoldingSetNodeID &ID, SDVTList VTList) {
378   ID.AddPointer(VTList.VTs);
379 }
380
381 /// AddNodeIDOperands - Various routines for adding operands to the NodeID data.
382 ///
383 static void AddNodeIDOperands(FoldingSetNodeID &ID,
384                               ArrayRef<SDValue> Ops) {
385   for (auto& Op : Ops) {
386     ID.AddPointer(Op.getNode());
387     ID.AddInteger(Op.getResNo());
388   }
389 }
390
391 /// AddNodeIDOperands - Various routines for adding operands to the NodeID data.
392 ///
393 static void AddNodeIDOperands(FoldingSetNodeID &ID,
394                               ArrayRef<SDUse> Ops) {
395   for (auto& Op : Ops) {
396     ID.AddPointer(Op.getNode());
397     ID.AddInteger(Op.getResNo());
398   }
399 }
400 /// Add logical or fast math flag values to FoldingSetNodeID value.
401 static void AddNodeIDFlags(FoldingSetNodeID &ID, unsigned Opcode,
402                            const SDNodeFlags *Flags) {
403   if (!Flags || !isBinOpWithFlags(Opcode))
404     return;
405
406   unsigned RawFlags = Flags->getRawFlags();
407   // If no flags are set, do not alter the ID. We must match the ID of nodes
408   // that were created without explicitly specifying flags. This also saves time
409   // and allows a gradual increase in API usage of the optional optimization
410   // flags.
411   if (RawFlags != 0)
412     ID.AddInteger(RawFlags);
413 }
414
415 static void AddNodeIDFlags(FoldingSetNodeID &ID, const SDNode *N) {
416   if (auto *Node = dyn_cast<BinaryWithFlagsSDNode>(N))
417     AddNodeIDFlags(ID, Node->getOpcode(), &Node->Flags);
418 }
419
420 static void AddNodeIDNode(FoldingSetNodeID &ID, unsigned short OpC,
421                           SDVTList VTList, ArrayRef<SDValue> OpList) {
422   AddNodeIDOpcode(ID, OpC);
423   AddNodeIDValueTypes(ID, VTList);
424   AddNodeIDOperands(ID, OpList);
425 }
426
427 /// If this is an SDNode with special info, add this info to the NodeID data.
428 static void AddNodeIDCustom(FoldingSetNodeID &ID, const SDNode *N) {
429   switch (N->getOpcode()) {
430   case ISD::TargetExternalSymbol:
431   case ISD::ExternalSymbol:
432   case ISD::MCSymbol:
433     llvm_unreachable("Should only be used on nodes with operands");
434   default: break;  // Normal nodes don't need extra info.
435   case ISD::TargetConstant:
436   case ISD::Constant: {
437     const ConstantSDNode *C = cast<ConstantSDNode>(N);
438     ID.AddPointer(C->getConstantIntValue());
439     ID.AddBoolean(C->isOpaque());
440     break;
441   }
442   case ISD::TargetConstantFP:
443   case ISD::ConstantFP: {
444     ID.AddPointer(cast<ConstantFPSDNode>(N)->getConstantFPValue());
445     break;
446   }
447   case ISD::TargetGlobalAddress:
448   case ISD::GlobalAddress:
449   case ISD::TargetGlobalTLSAddress:
450   case ISD::GlobalTLSAddress: {
451     const GlobalAddressSDNode *GA = cast<GlobalAddressSDNode>(N);
452     ID.AddPointer(GA->getGlobal());
453     ID.AddInteger(GA->getOffset());
454     ID.AddInteger(GA->getTargetFlags());
455     ID.AddInteger(GA->getAddressSpace());
456     break;
457   }
458   case ISD::BasicBlock:
459     ID.AddPointer(cast<BasicBlockSDNode>(N)->getBasicBlock());
460     break;
461   case ISD::Register:
462     ID.AddInteger(cast<RegisterSDNode>(N)->getReg());
463     break;
464   case ISD::RegisterMask:
465     ID.AddPointer(cast<RegisterMaskSDNode>(N)->getRegMask());
466     break;
467   case ISD::SRCVALUE:
468     ID.AddPointer(cast<SrcValueSDNode>(N)->getValue());
469     break;
470   case ISD::FrameIndex:
471   case ISD::TargetFrameIndex:
472     ID.AddInteger(cast<FrameIndexSDNode>(N)->getIndex());
473     break;
474   case ISD::JumpTable:
475   case ISD::TargetJumpTable:
476     ID.AddInteger(cast<JumpTableSDNode>(N)->getIndex());
477     ID.AddInteger(cast<JumpTableSDNode>(N)->getTargetFlags());
478     break;
479   case ISD::ConstantPool:
480   case ISD::TargetConstantPool: {
481     const ConstantPoolSDNode *CP = cast<ConstantPoolSDNode>(N);
482     ID.AddInteger(CP->getAlignment());
483     ID.AddInteger(CP->getOffset());
484     if (CP->isMachineConstantPoolEntry())
485       CP->getMachineCPVal()->addSelectionDAGCSEId(ID);
486     else
487       ID.AddPointer(CP->getConstVal());
488     ID.AddInteger(CP->getTargetFlags());
489     break;
490   }
491   case ISD::TargetIndex: {
492     const TargetIndexSDNode *TI = cast<TargetIndexSDNode>(N);
493     ID.AddInteger(TI->getIndex());
494     ID.AddInteger(TI->getOffset());
495     ID.AddInteger(TI->getTargetFlags());
496     break;
497   }
498   case ISD::LOAD: {
499     const LoadSDNode *LD = cast<LoadSDNode>(N);
500     ID.AddInteger(LD->getMemoryVT().getRawBits());
501     ID.AddInteger(LD->getRawSubclassData());
502     ID.AddInteger(LD->getPointerInfo().getAddrSpace());
503     break;
504   }
505   case ISD::STORE: {
506     const StoreSDNode *ST = cast<StoreSDNode>(N);
507     ID.AddInteger(ST->getMemoryVT().getRawBits());
508     ID.AddInteger(ST->getRawSubclassData());
509     ID.AddInteger(ST->getPointerInfo().getAddrSpace());
510     break;
511   }
512   case ISD::ATOMIC_CMP_SWAP:
513   case ISD::ATOMIC_CMP_SWAP_WITH_SUCCESS:
514   case ISD::ATOMIC_SWAP:
515   case ISD::ATOMIC_LOAD_ADD:
516   case ISD::ATOMIC_LOAD_SUB:
517   case ISD::ATOMIC_LOAD_AND:
518   case ISD::ATOMIC_LOAD_OR:
519   case ISD::ATOMIC_LOAD_XOR:
520   case ISD::ATOMIC_LOAD_NAND:
521   case ISD::ATOMIC_LOAD_MIN:
522   case ISD::ATOMIC_LOAD_MAX:
523   case ISD::ATOMIC_LOAD_UMIN:
524   case ISD::ATOMIC_LOAD_UMAX:
525   case ISD::ATOMIC_LOAD:
526   case ISD::ATOMIC_STORE: {
527     const AtomicSDNode *AT = cast<AtomicSDNode>(N);
528     ID.AddInteger(AT->getMemoryVT().getRawBits());
529     ID.AddInteger(AT->getRawSubclassData());
530     ID.AddInteger(AT->getPointerInfo().getAddrSpace());
531     break;
532   }
533   case ISD::PREFETCH: {
534     const MemSDNode *PF = cast<MemSDNode>(N);
535     ID.AddInteger(PF->getPointerInfo().getAddrSpace());
536     break;
537   }
538   case ISD::VECTOR_SHUFFLE: {
539     const ShuffleVectorSDNode *SVN = cast<ShuffleVectorSDNode>(N);
540     for (unsigned i = 0, e = N->getValueType(0).getVectorNumElements();
541          i != e; ++i)
542       ID.AddInteger(SVN->getMaskElt(i));
543     break;
544   }
545   case ISD::TargetBlockAddress:
546   case ISD::BlockAddress: {
547     const BlockAddressSDNode *BA = cast<BlockAddressSDNode>(N);
548     ID.AddPointer(BA->getBlockAddress());
549     ID.AddInteger(BA->getOffset());
550     ID.AddInteger(BA->getTargetFlags());
551     break;
552   }
553   } // end switch (N->getOpcode())
554
555   AddNodeIDFlags(ID, N);
556
557   // Target specific memory nodes could also have address spaces to check.
558   if (N->isTargetMemoryOpcode())
559     ID.AddInteger(cast<MemSDNode>(N)->getPointerInfo().getAddrSpace());
560 }
561
562 /// AddNodeIDNode - Generic routine for adding a nodes info to the NodeID
563 /// data.
564 static void AddNodeIDNode(FoldingSetNodeID &ID, const SDNode *N) {
565   AddNodeIDOpcode(ID, N->getOpcode());
566   // Add the return value info.
567   AddNodeIDValueTypes(ID, N->getVTList());
568   // Add the operand info.
569   AddNodeIDOperands(ID, N->ops());
570
571   // Handle SDNode leafs with special info.
572   AddNodeIDCustom(ID, N);
573 }
574
575 /// encodeMemSDNodeFlags - Generic routine for computing a value for use in
576 /// the CSE map that carries volatility, temporalness, indexing mode, and
577 /// extension/truncation information.
578 ///
579 static inline unsigned
580 encodeMemSDNodeFlags(int ConvType, ISD::MemIndexedMode AM, bool isVolatile,
581                      bool isNonTemporal, bool isInvariant) {
582   assert((ConvType & 3) == ConvType &&
583          "ConvType may not require more than 2 bits!");
584   assert((AM & 7) == AM &&
585          "AM may not require more than 3 bits!");
586   return ConvType |
587          (AM << 2) |
588          (isVolatile << 5) |
589          (isNonTemporal << 6) |
590          (isInvariant << 7);
591 }
592
593 //===----------------------------------------------------------------------===//
594 //                              SelectionDAG Class
595 //===----------------------------------------------------------------------===//
596
597 /// doNotCSE - Return true if CSE should not be performed for this node.
598 static bool doNotCSE(SDNode *N) {
599   if (N->getValueType(0) == MVT::Glue)
600     return true; // Never CSE anything that produces a flag.
601
602   switch (N->getOpcode()) {
603   default: break;
604   case ISD::HANDLENODE:
605   case ISD::EH_LABEL:
606     return true;   // Never CSE these nodes.
607   }
608
609   // Check that remaining values produced are not flags.
610   for (unsigned i = 1, e = N->getNumValues(); i != e; ++i)
611     if (N->getValueType(i) == MVT::Glue)
612       return true; // Never CSE anything that produces a flag.
613
614   return false;
615 }
616
617 /// RemoveDeadNodes - This method deletes all unreachable nodes in the
618 /// SelectionDAG.
619 void SelectionDAG::RemoveDeadNodes() {
620   // Create a dummy node (which is not added to allnodes), that adds a reference
621   // to the root node, preventing it from being deleted.
622   HandleSDNode Dummy(getRoot());
623
624   SmallVector<SDNode*, 128> DeadNodes;
625
626   // Add all obviously-dead nodes to the DeadNodes worklist.
627   for (SDNode &Node : allnodes())
628     if (Node.use_empty())
629       DeadNodes.push_back(&Node);
630
631   RemoveDeadNodes(DeadNodes);
632
633   // If the root changed (e.g. it was a dead load, update the root).
634   setRoot(Dummy.getValue());
635 }
636
637 /// RemoveDeadNodes - This method deletes the unreachable nodes in the
638 /// given list, and any nodes that become unreachable as a result.
639 void SelectionDAG::RemoveDeadNodes(SmallVectorImpl<SDNode *> &DeadNodes) {
640
641   // Process the worklist, deleting the nodes and adding their uses to the
642   // worklist.
643   while (!DeadNodes.empty()) {
644     SDNode *N = DeadNodes.pop_back_val();
645
646     for (DAGUpdateListener *DUL = UpdateListeners; DUL; DUL = DUL->Next)
647       DUL->NodeDeleted(N, nullptr);
648
649     // Take the node out of the appropriate CSE map.
650     RemoveNodeFromCSEMaps(N);
651
652     // Next, brutally remove the operand list.  This is safe to do, as there are
653     // no cycles in the graph.
654     for (SDNode::op_iterator I = N->op_begin(), E = N->op_end(); I != E; ) {
655       SDUse &Use = *I++;
656       SDNode *Operand = Use.getNode();
657       Use.set(SDValue());
658
659       // Now that we removed this operand, see if there are no uses of it left.
660       if (Operand->use_empty())
661         DeadNodes.push_back(Operand);
662     }
663
664     DeallocateNode(N);
665   }
666 }
667
668 void SelectionDAG::RemoveDeadNode(SDNode *N){
669   SmallVector<SDNode*, 16> DeadNodes(1, N);
670
671   // Create a dummy node that adds a reference to the root node, preventing
672   // it from being deleted.  (This matters if the root is an operand of the
673   // dead node.)
674   HandleSDNode Dummy(getRoot());
675
676   RemoveDeadNodes(DeadNodes);
677 }
678
679 void SelectionDAG::DeleteNode(SDNode *N) {
680   // First take this out of the appropriate CSE map.
681   RemoveNodeFromCSEMaps(N);
682
683   // Finally, remove uses due to operands of this node, remove from the
684   // AllNodes list, and delete the node.
685   DeleteNodeNotInCSEMaps(N);
686 }
687
688 void SelectionDAG::DeleteNodeNotInCSEMaps(SDNode *N) {
689   assert(N != AllNodes.begin() && "Cannot delete the entry node!");
690   assert(N->use_empty() && "Cannot delete a node that is not dead!");
691
692   // Drop all of the operands and decrement used node's use counts.
693   N->DropOperands();
694
695   DeallocateNode(N);
696 }
697
698 void SDDbgInfo::erase(const SDNode *Node) {
699   DbgValMapType::iterator I = DbgValMap.find(Node);
700   if (I == DbgValMap.end())
701     return;
702   for (auto &Val: I->second)
703     Val->setIsInvalidated();
704   DbgValMap.erase(I);
705 }
706
707 void SelectionDAG::DeallocateNode(SDNode *N) {
708   if (N->OperandsNeedDelete)
709     delete[] N->OperandList;
710
711   // Set the opcode to DELETED_NODE to help catch bugs when node
712   // memory is reallocated.
713   N->NodeType = ISD::DELETED_NODE;
714
715   NodeAllocator.Deallocate(AllNodes.remove(N));
716
717   // If any of the SDDbgValue nodes refer to this SDNode, invalidate
718   // them and forget about that node.
719   DbgInfo->erase(N);
720 }
721
722 #ifndef NDEBUG
723 /// VerifySDNode - Sanity check the given SDNode.  Aborts if it is invalid.
724 static void VerifySDNode(SDNode *N) {
725   switch (N->getOpcode()) {
726   default:
727     break;
728   case ISD::BUILD_PAIR: {
729     EVT VT = N->getValueType(0);
730     assert(N->getNumValues() == 1 && "Too many results!");
731     assert(!VT.isVector() && (VT.isInteger() || VT.isFloatingPoint()) &&
732            "Wrong return type!");
733     assert(N->getNumOperands() == 2 && "Wrong number of operands!");
734     assert(N->getOperand(0).getValueType() == N->getOperand(1).getValueType() &&
735            "Mismatched operand types!");
736     assert(N->getOperand(0).getValueType().isInteger() == VT.isInteger() &&
737            "Wrong operand type!");
738     assert(VT.getSizeInBits() == 2 * N->getOperand(0).getValueSizeInBits() &&
739            "Wrong return type size");
740     break;
741   }
742   case ISD::BUILD_VECTOR: {
743     assert(N->getNumValues() == 1 && "Too many results!");
744     assert(N->getValueType(0).isVector() && "Wrong return type!");
745     assert(N->getNumOperands() == N->getValueType(0).getVectorNumElements() &&
746            "Wrong number of operands!");
747     EVT EltVT = N->getValueType(0).getVectorElementType();
748     for (SDNode::op_iterator I = N->op_begin(), E = N->op_end(); I != E; ++I) {
749       assert((I->getValueType() == EltVT ||
750              (EltVT.isInteger() && I->getValueType().isInteger() &&
751               EltVT.bitsLE(I->getValueType()))) &&
752             "Wrong operand type!");
753       assert(I->getValueType() == N->getOperand(0).getValueType() &&
754              "Operands must all have the same type");
755     }
756     break;
757   }
758   }
759 }
760 #endif // NDEBUG
761
762 /// \brief Insert a newly allocated node into the DAG.
763 ///
764 /// Handles insertion into the all nodes list and CSE map, as well as
765 /// verification and other common operations when a new node is allocated.
766 void SelectionDAG::InsertNode(SDNode *N) {
767   AllNodes.push_back(N);
768 #ifndef NDEBUG
769   VerifySDNode(N);
770 #endif
771 }
772
773 /// RemoveNodeFromCSEMaps - Take the specified node out of the CSE map that
774 /// correspond to it.  This is useful when we're about to delete or repurpose
775 /// the node.  We don't want future request for structurally identical nodes
776 /// to return N anymore.
777 bool SelectionDAG::RemoveNodeFromCSEMaps(SDNode *N) {
778   bool Erased = false;
779   switch (N->getOpcode()) {
780   case ISD::HANDLENODE: return false;  // noop.
781   case ISD::CONDCODE:
782     assert(CondCodeNodes[cast<CondCodeSDNode>(N)->get()] &&
783            "Cond code doesn't exist!");
784     Erased = CondCodeNodes[cast<CondCodeSDNode>(N)->get()] != nullptr;
785     CondCodeNodes[cast<CondCodeSDNode>(N)->get()] = nullptr;
786     break;
787   case ISD::ExternalSymbol:
788     Erased = ExternalSymbols.erase(cast<ExternalSymbolSDNode>(N)->getSymbol());
789     break;
790   case ISD::TargetExternalSymbol: {
791     ExternalSymbolSDNode *ESN = cast<ExternalSymbolSDNode>(N);
792     Erased = TargetExternalSymbols.erase(
793                std::pair<std::string,unsigned char>(ESN->getSymbol(),
794                                                     ESN->getTargetFlags()));
795     break;
796   }
797   case ISD::MCSymbol: {
798     auto *MCSN = cast<MCSymbolSDNode>(N);
799     Erased = MCSymbols.erase(MCSN->getMCSymbol());
800     break;
801   }
802   case ISD::VALUETYPE: {
803     EVT VT = cast<VTSDNode>(N)->getVT();
804     if (VT.isExtended()) {
805       Erased = ExtendedValueTypeNodes.erase(VT);
806     } else {
807       Erased = ValueTypeNodes[VT.getSimpleVT().SimpleTy] != nullptr;
808       ValueTypeNodes[VT.getSimpleVT().SimpleTy] = nullptr;
809     }
810     break;
811   }
812   default:
813     // Remove it from the CSE Map.
814     assert(N->getOpcode() != ISD::DELETED_NODE && "DELETED_NODE in CSEMap!");
815     assert(N->getOpcode() != ISD::EntryToken && "EntryToken in CSEMap!");
816     Erased = CSEMap.RemoveNode(N);
817     break;
818   }
819 #ifndef NDEBUG
820   // Verify that the node was actually in one of the CSE maps, unless it has a
821   // flag result (which cannot be CSE'd) or is one of the special cases that are
822   // not subject to CSE.
823   if (!Erased && N->getValueType(N->getNumValues()-1) != MVT::Glue &&
824       !N->isMachineOpcode() && !doNotCSE(N)) {
825     N->dump(this);
826     dbgs() << "\n";
827     llvm_unreachable("Node is not in map!");
828   }
829 #endif
830   return Erased;
831 }
832
833 /// AddModifiedNodeToCSEMaps - The specified node has been removed from the CSE
834 /// maps and modified in place. Add it back to the CSE maps, unless an identical
835 /// node already exists, in which case transfer all its users to the existing
836 /// node. This transfer can potentially trigger recursive merging.
837 ///
838 void
839 SelectionDAG::AddModifiedNodeToCSEMaps(SDNode *N) {
840   // For node types that aren't CSE'd, just act as if no identical node
841   // already exists.
842   if (!doNotCSE(N)) {
843     SDNode *Existing = CSEMap.GetOrInsertNode(N);
844     if (Existing != N) {
845       // If there was already an existing matching node, use ReplaceAllUsesWith
846       // to replace the dead one with the existing one.  This can cause
847       // recursive merging of other unrelated nodes down the line.
848       ReplaceAllUsesWith(N, Existing);
849
850       // N is now dead. Inform the listeners and delete it.
851       for (DAGUpdateListener *DUL = UpdateListeners; DUL; DUL = DUL->Next)
852         DUL->NodeDeleted(N, Existing);
853       DeleteNodeNotInCSEMaps(N);
854       return;
855     }
856   }
857
858   // If the node doesn't already exist, we updated it.  Inform listeners.
859   for (DAGUpdateListener *DUL = UpdateListeners; DUL; DUL = DUL->Next)
860     DUL->NodeUpdated(N);
861 }
862
863 /// FindModifiedNodeSlot - Find a slot for the specified node if its operands
864 /// were replaced with those specified.  If this node is never memoized,
865 /// return null, otherwise return a pointer to the slot it would take.  If a
866 /// node already exists with these operands, the slot will be non-null.
867 SDNode *SelectionDAG::FindModifiedNodeSlot(SDNode *N, SDValue Op,
868                                            void *&InsertPos) {
869   if (doNotCSE(N))
870     return nullptr;
871
872   SDValue Ops[] = { Op };
873   FoldingSetNodeID ID;
874   AddNodeIDNode(ID, N->getOpcode(), N->getVTList(), Ops);
875   AddNodeIDCustom(ID, N);
876   SDNode *Node = FindNodeOrInsertPos(ID, N->getDebugLoc(), InsertPos);
877   return Node;
878 }
879
880 /// FindModifiedNodeSlot - Find a slot for the specified node if its operands
881 /// were replaced with those specified.  If this node is never memoized,
882 /// return null, otherwise return a pointer to the slot it would take.  If a
883 /// node already exists with these operands, the slot will be non-null.
884 SDNode *SelectionDAG::FindModifiedNodeSlot(SDNode *N,
885                                            SDValue Op1, SDValue Op2,
886                                            void *&InsertPos) {
887   if (doNotCSE(N))
888     return nullptr;
889
890   SDValue Ops[] = { Op1, Op2 };
891   FoldingSetNodeID ID;
892   AddNodeIDNode(ID, N->getOpcode(), N->getVTList(), Ops);
893   AddNodeIDCustom(ID, N);
894   SDNode *Node = FindNodeOrInsertPos(ID, N->getDebugLoc(), InsertPos);
895   return Node;
896 }
897
898
899 /// FindModifiedNodeSlot - Find a slot for the specified node if its operands
900 /// were replaced with those specified.  If this node is never memoized,
901 /// return null, otherwise return a pointer to the slot it would take.  If a
902 /// node already exists with these operands, the slot will be non-null.
903 SDNode *SelectionDAG::FindModifiedNodeSlot(SDNode *N, ArrayRef<SDValue> Ops,
904                                            void *&InsertPos) {
905   if (doNotCSE(N))
906     return nullptr;
907
908   FoldingSetNodeID ID;
909   AddNodeIDNode(ID, N->getOpcode(), N->getVTList(), Ops);
910   AddNodeIDCustom(ID, N);
911   SDNode *Node = FindNodeOrInsertPos(ID, N->getDebugLoc(), InsertPos);
912   return Node;
913 }
914
915 /// getEVTAlignment - Compute the default alignment value for the
916 /// given type.
917 ///
918 unsigned SelectionDAG::getEVTAlignment(EVT VT) const {
919   Type *Ty = VT == MVT::iPTR ?
920                    PointerType::get(Type::getInt8Ty(*getContext()), 0) :
921                    VT.getTypeForEVT(*getContext());
922
923   return getDataLayout().getABITypeAlignment(Ty);
924 }
925
926 // EntryNode could meaningfully have debug info if we can find it...
927 SelectionDAG::SelectionDAG(const TargetMachine &tm, CodeGenOpt::Level OL)
928     : TM(tm), TSI(nullptr), TLI(nullptr), OptLevel(OL),
929       EntryNode(ISD::EntryToken, 0, DebugLoc(), getVTList(MVT::Other)),
930       Root(getEntryNode()), NewNodesMustHaveLegalTypes(false),
931       UpdateListeners(nullptr) {
932   AllNodes.push_back(&EntryNode);
933   DbgInfo = new SDDbgInfo();
934 }
935
936 void SelectionDAG::init(MachineFunction &mf) {
937   MF = &mf;
938   TLI = getSubtarget().getTargetLowering();
939   TSI = getSubtarget().getSelectionDAGInfo();
940   Context = &mf.getFunction()->getContext();
941 }
942
943 SelectionDAG::~SelectionDAG() {
944   assert(!UpdateListeners && "Dangling registered DAGUpdateListeners");
945   allnodes_clear();
946   delete DbgInfo;
947 }
948
949 void SelectionDAG::allnodes_clear() {
950   assert(&*AllNodes.begin() == &EntryNode);
951   AllNodes.remove(AllNodes.begin());
952   while (!AllNodes.empty())
953     DeallocateNode(AllNodes.begin());
954 }
955
956 BinarySDNode *SelectionDAG::GetBinarySDNode(unsigned Opcode, SDLoc DL,
957                                             SDVTList VTs, SDValue N1,
958                                             SDValue N2,
959                                             const SDNodeFlags *Flags) {
960   if (isBinOpWithFlags(Opcode)) {
961     // If no flags were passed in, use a default flags object.
962     SDNodeFlags F;
963     if (Flags == nullptr)
964       Flags = &F;
965
966     BinaryWithFlagsSDNode *FN = new (NodeAllocator) BinaryWithFlagsSDNode(
967         Opcode, DL.getIROrder(), DL.getDebugLoc(), VTs, N1, N2, *Flags);
968
969     return FN;
970   }
971
972   BinarySDNode *N = new (NodeAllocator)
973       BinarySDNode(Opcode, DL.getIROrder(), DL.getDebugLoc(), VTs, N1, N2);
974   return N;
975 }
976
977 SDNode *SelectionDAG::FindNodeOrInsertPos(const FoldingSetNodeID &ID,
978                                           void *&InsertPos) {
979   SDNode *N = CSEMap.FindNodeOrInsertPos(ID, InsertPos);
980   if (N) {
981     switch (N->getOpcode()) {
982     default: break;
983     case ISD::Constant:
984     case ISD::ConstantFP:
985       llvm_unreachable("Querying for Constant and ConstantFP nodes requires "
986                        "debug location.  Use another overload.");
987     }
988   }
989   return N;
990 }
991
992 SDNode *SelectionDAG::FindNodeOrInsertPos(const FoldingSetNodeID &ID,
993                                           DebugLoc DL, void *&InsertPos) {
994   SDNode *N = CSEMap.FindNodeOrInsertPos(ID, InsertPos);
995   if (N) {
996     switch (N->getOpcode()) {
997     default: break; // Process only regular (non-target) constant nodes.
998     case ISD::Constant:
999     case ISD::ConstantFP:
1000       // Erase debug location from the node if the node is used at several
1001       // different places to do not propagate one location to all uses as it
1002       // leads to incorrect debug info.
1003       if (N->getDebugLoc() != DL)
1004         N->setDebugLoc(DebugLoc());
1005       break;
1006     }
1007   }
1008   return N;
1009 }
1010
1011 void SelectionDAG::clear() {
1012   allnodes_clear();
1013   OperandAllocator.Reset();
1014   CSEMap.clear();
1015
1016   ExtendedValueTypeNodes.clear();
1017   ExternalSymbols.clear();
1018   TargetExternalSymbols.clear();
1019   MCSymbols.clear();
1020   std::fill(CondCodeNodes.begin(), CondCodeNodes.end(),
1021             static_cast<CondCodeSDNode*>(nullptr));
1022   std::fill(ValueTypeNodes.begin(), ValueTypeNodes.end(),
1023             static_cast<SDNode*>(nullptr));
1024
1025   EntryNode.UseList = nullptr;
1026   AllNodes.push_back(&EntryNode);
1027   Root = getEntryNode();
1028   DbgInfo->clear();
1029 }
1030
1031 SDValue SelectionDAG::getAnyExtOrTrunc(SDValue Op, SDLoc DL, EVT VT) {
1032   return VT.bitsGT(Op.getValueType()) ?
1033     getNode(ISD::ANY_EXTEND, DL, VT, Op) :
1034     getNode(ISD::TRUNCATE, DL, VT, Op);
1035 }
1036
1037 SDValue SelectionDAG::getSExtOrTrunc(SDValue Op, SDLoc DL, EVT VT) {
1038   return VT.bitsGT(Op.getValueType()) ?
1039     getNode(ISD::SIGN_EXTEND, DL, VT, Op) :
1040     getNode(ISD::TRUNCATE, DL, VT, Op);
1041 }
1042
1043 SDValue SelectionDAG::getZExtOrTrunc(SDValue Op, SDLoc DL, EVT VT) {
1044   return VT.bitsGT(Op.getValueType()) ?
1045     getNode(ISD::ZERO_EXTEND, DL, VT, Op) :
1046     getNode(ISD::TRUNCATE, DL, VT, Op);
1047 }
1048
1049 SDValue SelectionDAG::getBoolExtOrTrunc(SDValue Op, SDLoc SL, EVT VT,
1050                                         EVT OpVT) {
1051   if (VT.bitsLE(Op.getValueType()))
1052     return getNode(ISD::TRUNCATE, SL, VT, Op);
1053
1054   TargetLowering::BooleanContent BType = TLI->getBooleanContents(OpVT);
1055   return getNode(TLI->getExtendForContent(BType), SL, VT, Op);
1056 }
1057
1058 SDValue SelectionDAG::getZeroExtendInReg(SDValue Op, SDLoc DL, EVT VT) {
1059   assert(!VT.isVector() &&
1060          "getZeroExtendInReg should use the vector element type instead of "
1061          "the vector type!");
1062   if (Op.getValueType() == VT) return Op;
1063   unsigned BitWidth = Op.getValueType().getScalarType().getSizeInBits();
1064   APInt Imm = APInt::getLowBitsSet(BitWidth,
1065                                    VT.getSizeInBits());
1066   return getNode(ISD::AND, DL, Op.getValueType(), Op,
1067                  getConstant(Imm, DL, Op.getValueType()));
1068 }
1069
1070 SDValue SelectionDAG::getAnyExtendVectorInReg(SDValue Op, SDLoc DL, EVT VT) {
1071   assert(VT.isVector() && "This DAG node is restricted to vector types.");
1072   assert(VT.getSizeInBits() == Op.getValueType().getSizeInBits() &&
1073          "The sizes of the input and result must match in order to perform the "
1074          "extend in-register.");
1075   assert(VT.getVectorNumElements() < Op.getValueType().getVectorNumElements() &&
1076          "The destination vector type must have fewer lanes than the input.");
1077   return getNode(ISD::ANY_EXTEND_VECTOR_INREG, DL, VT, Op);
1078 }
1079
1080 SDValue SelectionDAG::getSignExtendVectorInReg(SDValue Op, SDLoc DL, EVT VT) {
1081   assert(VT.isVector() && "This DAG node is restricted to vector types.");
1082   assert(VT.getSizeInBits() == Op.getValueType().getSizeInBits() &&
1083          "The sizes of the input and result must match in order to perform the "
1084          "extend in-register.");
1085   assert(VT.getVectorNumElements() < Op.getValueType().getVectorNumElements() &&
1086          "The destination vector type must have fewer lanes than the input.");
1087   return getNode(ISD::SIGN_EXTEND_VECTOR_INREG, DL, VT, Op);
1088 }
1089
1090 SDValue SelectionDAG::getZeroExtendVectorInReg(SDValue Op, SDLoc DL, EVT VT) {
1091   assert(VT.isVector() && "This DAG node is restricted to vector types.");
1092   assert(VT.getSizeInBits() == Op.getValueType().getSizeInBits() &&
1093          "The sizes of the input and result must match in order to perform the "
1094          "extend in-register.");
1095   assert(VT.getVectorNumElements() < Op.getValueType().getVectorNumElements() &&
1096          "The destination vector type must have fewer lanes than the input.");
1097   return getNode(ISD::ZERO_EXTEND_VECTOR_INREG, DL, VT, Op);
1098 }
1099
1100 /// getNOT - Create a bitwise NOT operation as (XOR Val, -1).
1101 ///
1102 SDValue SelectionDAG::getNOT(SDLoc DL, SDValue Val, EVT VT) {
1103   EVT EltVT = VT.getScalarType();
1104   SDValue NegOne =
1105     getConstant(APInt::getAllOnesValue(EltVT.getSizeInBits()), DL, VT);
1106   return getNode(ISD::XOR, DL, VT, Val, NegOne);
1107 }
1108
1109 SDValue SelectionDAG::getLogicalNOT(SDLoc DL, SDValue Val, EVT VT) {
1110   EVT EltVT = VT.getScalarType();
1111   SDValue TrueValue;
1112   switch (TLI->getBooleanContents(VT)) {
1113     case TargetLowering::ZeroOrOneBooleanContent:
1114     case TargetLowering::UndefinedBooleanContent:
1115       TrueValue = getConstant(1, DL, VT);
1116       break;
1117     case TargetLowering::ZeroOrNegativeOneBooleanContent:
1118       TrueValue = getConstant(APInt::getAllOnesValue(EltVT.getSizeInBits()), DL,
1119                               VT);
1120       break;
1121   }
1122   return getNode(ISD::XOR, DL, VT, Val, TrueValue);
1123 }
1124
1125 SDValue SelectionDAG::getConstant(uint64_t Val, SDLoc DL, EVT VT, bool isT,
1126                                   bool isO) {
1127   EVT EltVT = VT.getScalarType();
1128   assert((EltVT.getSizeInBits() >= 64 ||
1129          (uint64_t)((int64_t)Val >> EltVT.getSizeInBits()) + 1 < 2) &&
1130          "getConstant with a uint64_t value that doesn't fit in the type!");
1131   return getConstant(APInt(EltVT.getSizeInBits(), Val), DL, VT, isT, isO);
1132 }
1133
1134 SDValue SelectionDAG::getConstant(const APInt &Val, SDLoc DL, EVT VT, bool isT,
1135                                   bool isO)
1136 {
1137   return getConstant(*ConstantInt::get(*Context, Val), DL, VT, isT, isO);
1138 }
1139
1140 SDValue SelectionDAG::getConstant(const ConstantInt &Val, SDLoc DL, EVT VT,
1141                                   bool isT, bool isO) {
1142   assert(VT.isInteger() && "Cannot create FP integer constant!");
1143
1144   EVT EltVT = VT.getScalarType();
1145   const ConstantInt *Elt = &Val;
1146
1147   // In some cases the vector type is legal but the element type is illegal and
1148   // needs to be promoted, for example v8i8 on ARM.  In this case, promote the
1149   // inserted value (the type does not need to match the vector element type).
1150   // Any extra bits introduced will be truncated away.
1151   if (VT.isVector() && TLI->getTypeAction(*getContext(), EltVT) ==
1152       TargetLowering::TypePromoteInteger) {
1153    EltVT = TLI->getTypeToTransformTo(*getContext(), EltVT);
1154    APInt NewVal = Elt->getValue().zext(EltVT.getSizeInBits());
1155    Elt = ConstantInt::get(*getContext(), NewVal);
1156   }
1157   // In other cases the element type is illegal and needs to be expanded, for
1158   // example v2i64 on MIPS32. In this case, find the nearest legal type, split
1159   // the value into n parts and use a vector type with n-times the elements.
1160   // Then bitcast to the type requested.
1161   // Legalizing constants too early makes the DAGCombiner's job harder so we
1162   // only legalize if the DAG tells us we must produce legal types.
1163   else if (NewNodesMustHaveLegalTypes && VT.isVector() &&
1164            TLI->getTypeAction(*getContext(), EltVT) ==
1165            TargetLowering::TypeExpandInteger) {
1166     APInt NewVal = Elt->getValue();
1167     EVT ViaEltVT = TLI->getTypeToTransformTo(*getContext(), EltVT);
1168     unsigned ViaEltSizeInBits = ViaEltVT.getSizeInBits();
1169     unsigned ViaVecNumElts = VT.getSizeInBits() / ViaEltSizeInBits;
1170     EVT ViaVecVT = EVT::getVectorVT(*getContext(), ViaEltVT, ViaVecNumElts);
1171
1172     // Check the temporary vector is the correct size. If this fails then
1173     // getTypeToTransformTo() probably returned a type whose size (in bits)
1174     // isn't a power-of-2 factor of the requested type size.
1175     assert(ViaVecVT.getSizeInBits() == VT.getSizeInBits());
1176
1177     SmallVector<SDValue, 2> EltParts;
1178     for (unsigned i = 0; i < ViaVecNumElts / VT.getVectorNumElements(); ++i) {
1179       EltParts.push_back(getConstant(NewVal.lshr(i * ViaEltSizeInBits)
1180                                            .trunc(ViaEltSizeInBits), DL,
1181                                      ViaEltVT, isT, isO));
1182     }
1183
1184     // EltParts is currently in little endian order. If we actually want
1185     // big-endian order then reverse it now.
1186     if (getDataLayout().isBigEndian())
1187       std::reverse(EltParts.begin(), EltParts.end());
1188
1189     // The elements must be reversed when the element order is different
1190     // to the endianness of the elements (because the BITCAST is itself a
1191     // vector shuffle in this situation). However, we do not need any code to
1192     // perform this reversal because getConstant() is producing a vector
1193     // splat.
1194     // This situation occurs in MIPS MSA.
1195
1196     SmallVector<SDValue, 8> Ops;
1197     for (unsigned i = 0; i < VT.getVectorNumElements(); ++i)
1198       Ops.insert(Ops.end(), EltParts.begin(), EltParts.end());
1199
1200     SDValue Result = getNode(ISD::BITCAST, SDLoc(), VT,
1201                              getNode(ISD::BUILD_VECTOR, SDLoc(), ViaVecVT,
1202                                      Ops));
1203     return Result;
1204   }
1205
1206   assert(Elt->getBitWidth() == EltVT.getSizeInBits() &&
1207          "APInt size does not match type size!");
1208   unsigned Opc = isT ? ISD::TargetConstant : ISD::Constant;
1209   FoldingSetNodeID ID;
1210   AddNodeIDNode(ID, Opc, getVTList(EltVT), None);
1211   ID.AddPointer(Elt);
1212   ID.AddBoolean(isO);
1213   void *IP = nullptr;
1214   SDNode *N = nullptr;
1215   if ((N = FindNodeOrInsertPos(ID, DL.getDebugLoc(), IP)))
1216     if (!VT.isVector())
1217       return SDValue(N, 0);
1218
1219   if (!N) {
1220     N = new (NodeAllocator) ConstantSDNode(isT, isO, Elt, DL.getDebugLoc(),
1221                                            EltVT);
1222     CSEMap.InsertNode(N, IP);
1223     InsertNode(N);
1224   }
1225
1226   SDValue Result(N, 0);
1227   if (VT.isVector()) {
1228     SmallVector<SDValue, 8> Ops;
1229     Ops.assign(VT.getVectorNumElements(), Result);
1230     Result = getNode(ISD::BUILD_VECTOR, SDLoc(), VT, Ops);
1231   }
1232   return Result;
1233 }
1234
1235 SDValue SelectionDAG::getIntPtrConstant(uint64_t Val, SDLoc DL, bool isTarget) {
1236   return getConstant(Val, DL, TLI->getPointerTy(getDataLayout()), isTarget);
1237 }
1238
1239 SDValue SelectionDAG::getConstantFP(const APFloat& V, SDLoc DL, EVT VT,
1240                                     bool isTarget) {
1241   return getConstantFP(*ConstantFP::get(*getContext(), V), DL, VT, isTarget);
1242 }
1243
1244 SDValue SelectionDAG::getConstantFP(const ConstantFP& V, SDLoc DL, EVT VT,
1245                                     bool isTarget){
1246   assert(VT.isFloatingPoint() && "Cannot create integer FP constant!");
1247
1248   EVT EltVT = VT.getScalarType();
1249
1250   // Do the map lookup using the actual bit pattern for the floating point
1251   // value, so that we don't have problems with 0.0 comparing equal to -0.0, and
1252   // we don't have issues with SNANs.
1253   unsigned Opc = isTarget ? ISD::TargetConstantFP : ISD::ConstantFP;
1254   FoldingSetNodeID ID;
1255   AddNodeIDNode(ID, Opc, getVTList(EltVT), None);
1256   ID.AddPointer(&V);
1257   void *IP = nullptr;
1258   SDNode *N = nullptr;
1259   if ((N = FindNodeOrInsertPos(ID, DL.getDebugLoc(), IP)))
1260     if (!VT.isVector())
1261       return SDValue(N, 0);
1262
1263   if (!N) {
1264     N = new (NodeAllocator) ConstantFPSDNode(isTarget, &V, DL.getDebugLoc(),
1265                                              EltVT);
1266     CSEMap.InsertNode(N, IP);
1267     InsertNode(N);
1268   }
1269
1270   SDValue Result(N, 0);
1271   if (VT.isVector()) {
1272     SmallVector<SDValue, 8> Ops;
1273     Ops.assign(VT.getVectorNumElements(), Result);
1274     Result = getNode(ISD::BUILD_VECTOR, SDLoc(), VT, Ops);
1275   }
1276   return Result;
1277 }
1278
1279 SDValue SelectionDAG::getConstantFP(double Val, SDLoc DL, EVT VT,
1280                                     bool isTarget) {
1281   EVT EltVT = VT.getScalarType();
1282   if (EltVT==MVT::f32)
1283     return getConstantFP(APFloat((float)Val), DL, VT, isTarget);
1284   else if (EltVT==MVT::f64)
1285     return getConstantFP(APFloat(Val), DL, VT, isTarget);
1286   else if (EltVT==MVT::f80 || EltVT==MVT::f128 || EltVT==MVT::ppcf128 ||
1287            EltVT==MVT::f16) {
1288     bool ignored;
1289     APFloat apf = APFloat(Val);
1290     apf.convert(EVTToAPFloatSemantics(EltVT), APFloat::rmNearestTiesToEven,
1291                 &ignored);
1292     return getConstantFP(apf, DL, VT, isTarget);
1293   } else
1294     llvm_unreachable("Unsupported type in getConstantFP");
1295 }
1296
1297 SDValue SelectionDAG::getGlobalAddress(const GlobalValue *GV, SDLoc DL,
1298                                        EVT VT, int64_t Offset,
1299                                        bool isTargetGA,
1300                                        unsigned char TargetFlags) {
1301   assert((TargetFlags == 0 || isTargetGA) &&
1302          "Cannot set target flags on target-independent globals");
1303
1304   // Truncate (with sign-extension) the offset value to the pointer size.
1305   unsigned BitWidth = getDataLayout().getPointerTypeSizeInBits(GV->getType());
1306   if (BitWidth < 64)
1307     Offset = SignExtend64(Offset, BitWidth);
1308
1309   unsigned Opc;
1310   if (GV->isThreadLocal())
1311     Opc = isTargetGA ? ISD::TargetGlobalTLSAddress : ISD::GlobalTLSAddress;
1312   else
1313     Opc = isTargetGA ? ISD::TargetGlobalAddress : ISD::GlobalAddress;
1314
1315   FoldingSetNodeID ID;
1316   AddNodeIDNode(ID, Opc, getVTList(VT), None);
1317   ID.AddPointer(GV);
1318   ID.AddInteger(Offset);
1319   ID.AddInteger(TargetFlags);
1320   ID.AddInteger(GV->getType()->getAddressSpace());
1321   void *IP = nullptr;
1322   if (SDNode *E = FindNodeOrInsertPos(ID, DL.getDebugLoc(), IP))
1323     return SDValue(E, 0);
1324
1325   SDNode *N = new (NodeAllocator) GlobalAddressSDNode(Opc, DL.getIROrder(),
1326                                                       DL.getDebugLoc(), GV, VT,
1327                                                       Offset, TargetFlags);
1328   CSEMap.InsertNode(N, IP);
1329     InsertNode(N);
1330   return SDValue(N, 0);
1331 }
1332
1333 SDValue SelectionDAG::getFrameIndex(int FI, EVT VT, bool isTarget) {
1334   unsigned Opc = isTarget ? ISD::TargetFrameIndex : ISD::FrameIndex;
1335   FoldingSetNodeID ID;
1336   AddNodeIDNode(ID, Opc, getVTList(VT), None);
1337   ID.AddInteger(FI);
1338   void *IP = nullptr;
1339   if (SDNode *E = FindNodeOrInsertPos(ID, IP))
1340     return SDValue(E, 0);
1341
1342   SDNode *N = new (NodeAllocator) FrameIndexSDNode(FI, VT, isTarget);
1343   CSEMap.InsertNode(N, IP);
1344   InsertNode(N);
1345   return SDValue(N, 0);
1346 }
1347
1348 SDValue SelectionDAG::getJumpTable(int JTI, EVT VT, bool isTarget,
1349                                    unsigned char TargetFlags) {
1350   assert((TargetFlags == 0 || isTarget) &&
1351          "Cannot set target flags on target-independent jump tables");
1352   unsigned Opc = isTarget ? ISD::TargetJumpTable : ISD::JumpTable;
1353   FoldingSetNodeID ID;
1354   AddNodeIDNode(ID, Opc, getVTList(VT), None);
1355   ID.AddInteger(JTI);
1356   ID.AddInteger(TargetFlags);
1357   void *IP = nullptr;
1358   if (SDNode *E = FindNodeOrInsertPos(ID, IP))
1359     return SDValue(E, 0);
1360
1361   SDNode *N = new (NodeAllocator) JumpTableSDNode(JTI, VT, isTarget,
1362                                                   TargetFlags);
1363   CSEMap.InsertNode(N, IP);
1364   InsertNode(N);
1365   return SDValue(N, 0);
1366 }
1367
1368 SDValue SelectionDAG::getConstantPool(const Constant *C, EVT VT,
1369                                       unsigned Alignment, int Offset,
1370                                       bool isTarget,
1371                                       unsigned char TargetFlags) {
1372   assert((TargetFlags == 0 || isTarget) &&
1373          "Cannot set target flags on target-independent globals");
1374   if (Alignment == 0)
1375     Alignment = getDataLayout().getPrefTypeAlignment(C->getType());
1376   unsigned Opc = isTarget ? ISD::TargetConstantPool : ISD::ConstantPool;
1377   FoldingSetNodeID ID;
1378   AddNodeIDNode(ID, Opc, getVTList(VT), None);
1379   ID.AddInteger(Alignment);
1380   ID.AddInteger(Offset);
1381   ID.AddPointer(C);
1382   ID.AddInteger(TargetFlags);
1383   void *IP = nullptr;
1384   if (SDNode *E = FindNodeOrInsertPos(ID, IP))
1385     return SDValue(E, 0);
1386
1387   SDNode *N = new (NodeAllocator) ConstantPoolSDNode(isTarget, C, VT, Offset,
1388                                                      Alignment, TargetFlags);
1389   CSEMap.InsertNode(N, IP);
1390   InsertNode(N);
1391   return SDValue(N, 0);
1392 }
1393
1394
1395 SDValue SelectionDAG::getConstantPool(MachineConstantPoolValue *C, EVT VT,
1396                                       unsigned Alignment, int Offset,
1397                                       bool isTarget,
1398                                       unsigned char TargetFlags) {
1399   assert((TargetFlags == 0 || isTarget) &&
1400          "Cannot set target flags on target-independent globals");
1401   if (Alignment == 0)
1402     Alignment = getDataLayout().getPrefTypeAlignment(C->getType());
1403   unsigned Opc = isTarget ? ISD::TargetConstantPool : ISD::ConstantPool;
1404   FoldingSetNodeID ID;
1405   AddNodeIDNode(ID, Opc, getVTList(VT), None);
1406   ID.AddInteger(Alignment);
1407   ID.AddInteger(Offset);
1408   C->addSelectionDAGCSEId(ID);
1409   ID.AddInteger(TargetFlags);
1410   void *IP = nullptr;
1411   if (SDNode *E = FindNodeOrInsertPos(ID, IP))
1412     return SDValue(E, 0);
1413
1414   SDNode *N = new (NodeAllocator) ConstantPoolSDNode(isTarget, C, VT, Offset,
1415                                                      Alignment, TargetFlags);
1416   CSEMap.InsertNode(N, IP);
1417   InsertNode(N);
1418   return SDValue(N, 0);
1419 }
1420
1421 SDValue SelectionDAG::getTargetIndex(int Index, EVT VT, int64_t Offset,
1422                                      unsigned char TargetFlags) {
1423   FoldingSetNodeID ID;
1424   AddNodeIDNode(ID, ISD::TargetIndex, getVTList(VT), None);
1425   ID.AddInteger(Index);
1426   ID.AddInteger(Offset);
1427   ID.AddInteger(TargetFlags);
1428   void *IP = nullptr;
1429   if (SDNode *E = FindNodeOrInsertPos(ID, IP))
1430     return SDValue(E, 0);
1431
1432   SDNode *N = new (NodeAllocator) TargetIndexSDNode(Index, VT, Offset,
1433                                                     TargetFlags);
1434   CSEMap.InsertNode(N, IP);
1435   InsertNode(N);
1436   return SDValue(N, 0);
1437 }
1438
1439 SDValue SelectionDAG::getBasicBlock(MachineBasicBlock *MBB) {
1440   FoldingSetNodeID ID;
1441   AddNodeIDNode(ID, ISD::BasicBlock, getVTList(MVT::Other), None);
1442   ID.AddPointer(MBB);
1443   void *IP = nullptr;
1444   if (SDNode *E = FindNodeOrInsertPos(ID, IP))
1445     return SDValue(E, 0);
1446
1447   SDNode *N = new (NodeAllocator) BasicBlockSDNode(MBB);
1448   CSEMap.InsertNode(N, IP);
1449   InsertNode(N);
1450   return SDValue(N, 0);
1451 }
1452
1453 SDValue SelectionDAG::getValueType(EVT VT) {
1454   if (VT.isSimple() && (unsigned)VT.getSimpleVT().SimpleTy >=
1455       ValueTypeNodes.size())
1456     ValueTypeNodes.resize(VT.getSimpleVT().SimpleTy+1);
1457
1458   SDNode *&N = VT.isExtended() ?
1459     ExtendedValueTypeNodes[VT] : ValueTypeNodes[VT.getSimpleVT().SimpleTy];
1460
1461   if (N) return SDValue(N, 0);
1462   N = new (NodeAllocator) VTSDNode(VT);
1463   InsertNode(N);
1464   return SDValue(N, 0);
1465 }
1466
1467 SDValue SelectionDAG::getExternalSymbol(const char *Sym, EVT VT) {
1468   SDNode *&N = ExternalSymbols[Sym];
1469   if (N) return SDValue(N, 0);
1470   N = new (NodeAllocator) ExternalSymbolSDNode(false, Sym, 0, VT);
1471   InsertNode(N);
1472   return SDValue(N, 0);
1473 }
1474
1475 SDValue SelectionDAG::getMCSymbol(MCSymbol *Sym, EVT VT) {
1476   SDNode *&N = MCSymbols[Sym];
1477   if (N)
1478     return SDValue(N, 0);
1479   N = new (NodeAllocator) MCSymbolSDNode(Sym, VT);
1480   InsertNode(N);
1481   return SDValue(N, 0);
1482 }
1483
1484 SDValue SelectionDAG::getTargetExternalSymbol(const char *Sym, EVT VT,
1485                                               unsigned char TargetFlags) {
1486   SDNode *&N =
1487     TargetExternalSymbols[std::pair<std::string,unsigned char>(Sym,
1488                                                                TargetFlags)];
1489   if (N) return SDValue(N, 0);
1490   N = new (NodeAllocator) ExternalSymbolSDNode(true, Sym, TargetFlags, VT);
1491   InsertNode(N);
1492   return SDValue(N, 0);
1493 }
1494
1495 SDValue SelectionDAG::getCondCode(ISD::CondCode Cond) {
1496   if ((unsigned)Cond >= CondCodeNodes.size())
1497     CondCodeNodes.resize(Cond+1);
1498
1499   if (!CondCodeNodes[Cond]) {
1500     CondCodeSDNode *N = new (NodeAllocator) CondCodeSDNode(Cond);
1501     CondCodeNodes[Cond] = N;
1502     InsertNode(N);
1503   }
1504
1505   return SDValue(CondCodeNodes[Cond], 0);
1506 }
1507
1508 // commuteShuffle - swaps the values of N1 and N2, and swaps all indices in
1509 // the shuffle mask M that point at N1 to point at N2, and indices that point
1510 // N2 to point at N1.
1511 static void commuteShuffle(SDValue &N1, SDValue &N2, SmallVectorImpl<int> &M) {
1512   std::swap(N1, N2);
1513   ShuffleVectorSDNode::commuteMask(M);
1514 }
1515
1516 SDValue SelectionDAG::getVectorShuffle(EVT VT, SDLoc dl, SDValue N1,
1517                                        SDValue N2, const int *Mask) {
1518   assert(VT == N1.getValueType() && VT == N2.getValueType() &&
1519          "Invalid VECTOR_SHUFFLE");
1520
1521   // Canonicalize shuffle undef, undef -> undef
1522   if (N1.getOpcode() == ISD::UNDEF && N2.getOpcode() == ISD::UNDEF)
1523     return getUNDEF(VT);
1524
1525   // Validate that all indices in Mask are within the range of the elements
1526   // input to the shuffle.
1527   unsigned NElts = VT.getVectorNumElements();
1528   SmallVector<int, 8> MaskVec;
1529   for (unsigned i = 0; i != NElts; ++i) {
1530     assert(Mask[i] < (int)(NElts * 2) && "Index out of range");
1531     MaskVec.push_back(Mask[i]);
1532   }
1533
1534   // Canonicalize shuffle v, v -> v, undef
1535   if (N1 == N2) {
1536     N2 = getUNDEF(VT);
1537     for (unsigned i = 0; i != NElts; ++i)
1538       if (MaskVec[i] >= (int)NElts) MaskVec[i] -= NElts;
1539   }
1540
1541   // Canonicalize shuffle undef, v -> v, undef.  Commute the shuffle mask.
1542   if (N1.getOpcode() == ISD::UNDEF)
1543     commuteShuffle(N1, N2, MaskVec);
1544
1545   // If shuffling a splat, try to blend the splat instead. We do this here so
1546   // that even when this arises during lowering we don't have to re-handle it.
1547   auto BlendSplat = [&](BuildVectorSDNode *BV, int Offset) {
1548     BitVector UndefElements;
1549     SDValue Splat = BV->getSplatValue(&UndefElements);
1550     if (!Splat)
1551       return;
1552
1553     for (int i = 0; i < (int)NElts; ++i) {
1554       if (MaskVec[i] < Offset || MaskVec[i] >= (Offset + (int)NElts))
1555         continue;
1556
1557       // If this input comes from undef, mark it as such.
1558       if (UndefElements[MaskVec[i] - Offset]) {
1559         MaskVec[i] = -1;
1560         continue;
1561       }
1562
1563       // If we can blend a non-undef lane, use that instead.
1564       if (!UndefElements[i])
1565         MaskVec[i] = i + Offset;
1566     }
1567   };
1568   if (auto *N1BV = dyn_cast<BuildVectorSDNode>(N1))
1569     BlendSplat(N1BV, 0);
1570   if (auto *N2BV = dyn_cast<BuildVectorSDNode>(N2))
1571     BlendSplat(N2BV, NElts);
1572
1573   // Canonicalize all index into lhs, -> shuffle lhs, undef
1574   // Canonicalize all index into rhs, -> shuffle rhs, undef
1575   bool AllLHS = true, AllRHS = true;
1576   bool N2Undef = N2.getOpcode() == ISD::UNDEF;
1577   for (unsigned i = 0; i != NElts; ++i) {
1578     if (MaskVec[i] >= (int)NElts) {
1579       if (N2Undef)
1580         MaskVec[i] = -1;
1581       else
1582         AllLHS = false;
1583     } else if (MaskVec[i] >= 0) {
1584       AllRHS = false;
1585     }
1586   }
1587   if (AllLHS && AllRHS)
1588     return getUNDEF(VT);
1589   if (AllLHS && !N2Undef)
1590     N2 = getUNDEF(VT);
1591   if (AllRHS) {
1592     N1 = getUNDEF(VT);
1593     commuteShuffle(N1, N2, MaskVec);
1594   }
1595   // Reset our undef status after accounting for the mask.
1596   N2Undef = N2.getOpcode() == ISD::UNDEF;
1597   // Re-check whether both sides ended up undef.
1598   if (N1.getOpcode() == ISD::UNDEF && N2Undef)
1599     return getUNDEF(VT);
1600
1601   // If Identity shuffle return that node.
1602   bool Identity = true, AllSame = true;
1603   for (unsigned i = 0; i != NElts; ++i) {
1604     if (MaskVec[i] >= 0 && MaskVec[i] != (int)i) Identity = false;
1605     if (MaskVec[i] != MaskVec[0]) AllSame = false;
1606   }
1607   if (Identity && NElts)
1608     return N1;
1609
1610   // Shuffling a constant splat doesn't change the result.
1611   if (N2Undef) {
1612     SDValue V = N1;
1613
1614     // Look through any bitcasts. We check that these don't change the number
1615     // (and size) of elements and just changes their types.
1616     while (V.getOpcode() == ISD::BITCAST)
1617       V = V->getOperand(0);
1618
1619     // A splat should always show up as a build vector node.
1620     if (auto *BV = dyn_cast<BuildVectorSDNode>(V)) {
1621       BitVector UndefElements;
1622       SDValue Splat = BV->getSplatValue(&UndefElements);
1623       // If this is a splat of an undef, shuffling it is also undef.
1624       if (Splat && Splat.getOpcode() == ISD::UNDEF)
1625         return getUNDEF(VT);
1626
1627       bool SameNumElts =
1628           V.getValueType().getVectorNumElements() == VT.getVectorNumElements();
1629
1630       // We only have a splat which can skip shuffles if there is a splatted
1631       // value and no undef lanes rearranged by the shuffle.
1632       if (Splat && UndefElements.none()) {
1633         // Splat of <x, x, ..., x>, return <x, x, ..., x>, provided that the
1634         // number of elements match or the value splatted is a zero constant.
1635         if (SameNumElts)
1636           return N1;
1637         if (auto *C = dyn_cast<ConstantSDNode>(Splat))
1638           if (C->isNullValue())
1639             return N1;
1640       }
1641
1642       // If the shuffle itself creates a splat, build the vector directly.
1643       if (AllSame && SameNumElts) {
1644         const SDValue &Splatted = BV->getOperand(MaskVec[0]);
1645         SmallVector<SDValue, 8> Ops(NElts, Splatted);
1646
1647         EVT BuildVT = BV->getValueType(0);
1648         SDValue NewBV = getNode(ISD::BUILD_VECTOR, dl, BuildVT, Ops);
1649
1650         // We may have jumped through bitcasts, so the type of the
1651         // BUILD_VECTOR may not match the type of the shuffle.
1652         if (BuildVT != VT)
1653           NewBV = getNode(ISD::BITCAST, dl, VT, NewBV);
1654         return NewBV;
1655       }
1656     }
1657   }
1658
1659   FoldingSetNodeID ID;
1660   SDValue Ops[2] = { N1, N2 };
1661   AddNodeIDNode(ID, ISD::VECTOR_SHUFFLE, getVTList(VT), Ops);
1662   for (unsigned i = 0; i != NElts; ++i)
1663     ID.AddInteger(MaskVec[i]);
1664
1665   void* IP = nullptr;
1666   if (SDNode *E = FindNodeOrInsertPos(ID, dl.getDebugLoc(), IP))
1667     return SDValue(E, 0);
1668
1669   // Allocate the mask array for the node out of the BumpPtrAllocator, since
1670   // SDNode doesn't have access to it.  This memory will be "leaked" when
1671   // the node is deallocated, but recovered when the NodeAllocator is released.
1672   int *MaskAlloc = OperandAllocator.Allocate<int>(NElts);
1673   memcpy(MaskAlloc, &MaskVec[0], NElts * sizeof(int));
1674
1675   ShuffleVectorSDNode *N =
1676     new (NodeAllocator) ShuffleVectorSDNode(VT, dl.getIROrder(),
1677                                             dl.getDebugLoc(), N1, N2,
1678                                             MaskAlloc);
1679   CSEMap.InsertNode(N, IP);
1680   InsertNode(N);
1681   return SDValue(N, 0);
1682 }
1683
1684 SDValue SelectionDAG::getCommutedVectorShuffle(const ShuffleVectorSDNode &SV) {
1685   MVT VT = SV.getSimpleValueType(0);
1686   SmallVector<int, 8> MaskVec(SV.getMask().begin(), SV.getMask().end());
1687   ShuffleVectorSDNode::commuteMask(MaskVec);
1688
1689   SDValue Op0 = SV.getOperand(0);
1690   SDValue Op1 = SV.getOperand(1);
1691   return getVectorShuffle(VT, SDLoc(&SV), Op1, Op0, &MaskVec[0]);
1692 }
1693
1694 SDValue SelectionDAG::getConvertRndSat(EVT VT, SDLoc dl,
1695                                        SDValue Val, SDValue DTy,
1696                                        SDValue STy, SDValue Rnd, SDValue Sat,
1697                                        ISD::CvtCode Code) {
1698   // If the src and dest types are the same and the conversion is between
1699   // integer types of the same sign or two floats, no conversion is necessary.
1700   if (DTy == STy &&
1701       (Code == ISD::CVT_UU || Code == ISD::CVT_SS || Code == ISD::CVT_FF))
1702     return Val;
1703
1704   FoldingSetNodeID ID;
1705   SDValue Ops[] = { Val, DTy, STy, Rnd, Sat };
1706   AddNodeIDNode(ID, ISD::CONVERT_RNDSAT, getVTList(VT), Ops);
1707   void* IP = nullptr;
1708   if (SDNode *E = FindNodeOrInsertPos(ID, dl.getDebugLoc(), IP))
1709     return SDValue(E, 0);
1710
1711   CvtRndSatSDNode *N = new (NodeAllocator) CvtRndSatSDNode(VT, dl.getIROrder(),
1712                                                            dl.getDebugLoc(),
1713                                                            Ops, Code);
1714   CSEMap.InsertNode(N, IP);
1715   InsertNode(N);
1716   return SDValue(N, 0);
1717 }
1718
1719 SDValue SelectionDAG::getRegister(unsigned RegNo, EVT VT) {
1720   FoldingSetNodeID ID;
1721   AddNodeIDNode(ID, ISD::Register, getVTList(VT), None);
1722   ID.AddInteger(RegNo);
1723   void *IP = nullptr;
1724   if (SDNode *E = FindNodeOrInsertPos(ID, IP))
1725     return SDValue(E, 0);
1726
1727   SDNode *N = new (NodeAllocator) RegisterSDNode(RegNo, VT);
1728   CSEMap.InsertNode(N, IP);
1729   InsertNode(N);
1730   return SDValue(N, 0);
1731 }
1732
1733 SDValue SelectionDAG::getRegisterMask(const uint32_t *RegMask) {
1734   FoldingSetNodeID ID;
1735   AddNodeIDNode(ID, ISD::RegisterMask, getVTList(MVT::Untyped), None);
1736   ID.AddPointer(RegMask);
1737   void *IP = nullptr;
1738   if (SDNode *E = FindNodeOrInsertPos(ID, IP))
1739     return SDValue(E, 0);
1740
1741   SDNode *N = new (NodeAllocator) RegisterMaskSDNode(RegMask);
1742   CSEMap.InsertNode(N, IP);
1743   InsertNode(N);
1744   return SDValue(N, 0);
1745 }
1746
1747 SDValue SelectionDAG::getEHLabel(SDLoc dl, SDValue Root, MCSymbol *Label) {
1748   FoldingSetNodeID ID;
1749   SDValue Ops[] = { Root };
1750   AddNodeIDNode(ID, ISD::EH_LABEL, getVTList(MVT::Other), Ops);
1751   ID.AddPointer(Label);
1752   void *IP = nullptr;
1753   if (SDNode *E = FindNodeOrInsertPos(ID, IP))
1754     return SDValue(E, 0);
1755
1756   SDNode *N = new (NodeAllocator) EHLabelSDNode(dl.getIROrder(),
1757                                                 dl.getDebugLoc(), Root, Label);
1758   CSEMap.InsertNode(N, IP);
1759   InsertNode(N);
1760   return SDValue(N, 0);
1761 }
1762
1763
1764 SDValue SelectionDAG::getBlockAddress(const BlockAddress *BA, EVT VT,
1765                                       int64_t Offset,
1766                                       bool isTarget,
1767                                       unsigned char TargetFlags) {
1768   unsigned Opc = isTarget ? ISD::TargetBlockAddress : ISD::BlockAddress;
1769
1770   FoldingSetNodeID ID;
1771   AddNodeIDNode(ID, Opc, getVTList(VT), None);
1772   ID.AddPointer(BA);
1773   ID.AddInteger(Offset);
1774   ID.AddInteger(TargetFlags);
1775   void *IP = nullptr;
1776   if (SDNode *E = FindNodeOrInsertPos(ID, IP))
1777     return SDValue(E, 0);
1778
1779   SDNode *N = new (NodeAllocator) BlockAddressSDNode(Opc, VT, BA, Offset,
1780                                                      TargetFlags);
1781   CSEMap.InsertNode(N, IP);
1782   InsertNode(N);
1783   return SDValue(N, 0);
1784 }
1785
1786 SDValue SelectionDAG::getSrcValue(const Value *V) {
1787   assert((!V || V->getType()->isPointerTy()) &&
1788          "SrcValue is not a pointer?");
1789
1790   FoldingSetNodeID ID;
1791   AddNodeIDNode(ID, ISD::SRCVALUE, getVTList(MVT::Other), None);
1792   ID.AddPointer(V);
1793
1794   void *IP = nullptr;
1795   if (SDNode *E = FindNodeOrInsertPos(ID, IP))
1796     return SDValue(E, 0);
1797
1798   SDNode *N = new (NodeAllocator) SrcValueSDNode(V);
1799   CSEMap.InsertNode(N, IP);
1800   InsertNode(N);
1801   return SDValue(N, 0);
1802 }
1803
1804 /// getMDNode - Return an MDNodeSDNode which holds an MDNode.
1805 SDValue SelectionDAG::getMDNode(const MDNode *MD) {
1806   FoldingSetNodeID ID;
1807   AddNodeIDNode(ID, ISD::MDNODE_SDNODE, getVTList(MVT::Other), None);
1808   ID.AddPointer(MD);
1809
1810   void *IP = nullptr;
1811   if (SDNode *E = FindNodeOrInsertPos(ID, IP))
1812     return SDValue(E, 0);
1813
1814   SDNode *N = new (NodeAllocator) MDNodeSDNode(MD);
1815   CSEMap.InsertNode(N, IP);
1816   InsertNode(N);
1817   return SDValue(N, 0);
1818 }
1819
1820 SDValue SelectionDAG::getBitcast(EVT VT, SDValue V) {
1821   if (VT == V.getValueType())
1822     return V;
1823
1824   return getNode(ISD::BITCAST, SDLoc(V), VT, V);
1825 }
1826
1827 /// getAddrSpaceCast - Return an AddrSpaceCastSDNode.
1828 SDValue SelectionDAG::getAddrSpaceCast(SDLoc dl, EVT VT, SDValue Ptr,
1829                                        unsigned SrcAS, unsigned DestAS) {
1830   SDValue Ops[] = {Ptr};
1831   FoldingSetNodeID ID;
1832   AddNodeIDNode(ID, ISD::ADDRSPACECAST, getVTList(VT), Ops);
1833   ID.AddInteger(SrcAS);
1834   ID.AddInteger(DestAS);
1835
1836   void *IP = nullptr;
1837   if (SDNode *E = FindNodeOrInsertPos(ID, dl.getDebugLoc(), IP))
1838     return SDValue(E, 0);
1839
1840   SDNode *N = new (NodeAllocator) AddrSpaceCastSDNode(dl.getIROrder(),
1841                                                       dl.getDebugLoc(),
1842                                                       VT, Ptr, SrcAS, DestAS);
1843   CSEMap.InsertNode(N, IP);
1844   InsertNode(N);
1845   return SDValue(N, 0);
1846 }
1847
1848 /// getShiftAmountOperand - Return the specified value casted to
1849 /// the target's desired shift amount type.
1850 SDValue SelectionDAG::getShiftAmountOperand(EVT LHSTy, SDValue Op) {
1851   EVT OpTy = Op.getValueType();
1852   EVT ShTy = TLI->getShiftAmountTy(LHSTy, getDataLayout());
1853   if (OpTy == ShTy || OpTy.isVector()) return Op;
1854
1855   return getZExtOrTrunc(Op, SDLoc(Op), ShTy);
1856 }
1857
1858 /// CreateStackTemporary - Create a stack temporary, suitable for holding the
1859 /// specified value type.
1860 SDValue SelectionDAG::CreateStackTemporary(EVT VT, unsigned minAlign) {
1861   MachineFrameInfo *FrameInfo = getMachineFunction().getFrameInfo();
1862   unsigned ByteSize = VT.getStoreSize();
1863   Type *Ty = VT.getTypeForEVT(*getContext());
1864   unsigned StackAlign =
1865       std::max((unsigned)getDataLayout().getPrefTypeAlignment(Ty), minAlign);
1866
1867   int FrameIdx = FrameInfo->CreateStackObject(ByteSize, StackAlign, false);
1868   return getFrameIndex(FrameIdx, TLI->getPointerTy(getDataLayout()));
1869 }
1870
1871 /// CreateStackTemporary - Create a stack temporary suitable for holding
1872 /// either of the specified value types.
1873 SDValue SelectionDAG::CreateStackTemporary(EVT VT1, EVT VT2) {
1874   unsigned Bytes = std::max(VT1.getStoreSize(), VT2.getStoreSize());
1875   Type *Ty1 = VT1.getTypeForEVT(*getContext());
1876   Type *Ty2 = VT2.getTypeForEVT(*getContext());
1877   const DataLayout &DL = getDataLayout();
1878   unsigned Align =
1879       std::max(DL.getPrefTypeAlignment(Ty1), DL.getPrefTypeAlignment(Ty2));
1880
1881   MachineFrameInfo *FrameInfo = getMachineFunction().getFrameInfo();
1882   int FrameIdx = FrameInfo->CreateStackObject(Bytes, Align, false);
1883   return getFrameIndex(FrameIdx, TLI->getPointerTy(getDataLayout()));
1884 }
1885
1886 SDValue SelectionDAG::FoldSetCC(EVT VT, SDValue N1,
1887                                 SDValue N2, ISD::CondCode Cond, SDLoc dl) {
1888   // These setcc operations always fold.
1889   switch (Cond) {
1890   default: break;
1891   case ISD::SETFALSE:
1892   case ISD::SETFALSE2: return getConstant(0, dl, VT);
1893   case ISD::SETTRUE:
1894   case ISD::SETTRUE2: {
1895     TargetLowering::BooleanContent Cnt =
1896         TLI->getBooleanContents(N1->getValueType(0));
1897     return getConstant(
1898         Cnt == TargetLowering::ZeroOrNegativeOneBooleanContent ? -1ULL : 1, dl,
1899         VT);
1900   }
1901
1902   case ISD::SETOEQ:
1903   case ISD::SETOGT:
1904   case ISD::SETOGE:
1905   case ISD::SETOLT:
1906   case ISD::SETOLE:
1907   case ISD::SETONE:
1908   case ISD::SETO:
1909   case ISD::SETUO:
1910   case ISD::SETUEQ:
1911   case ISD::SETUNE:
1912     assert(!N1.getValueType().isInteger() && "Illegal setcc for integer!");
1913     break;
1914   }
1915
1916   if (ConstantSDNode *N2C = dyn_cast<ConstantSDNode>(N2)) {
1917     const APInt &C2 = N2C->getAPIntValue();
1918     if (ConstantSDNode *N1C = dyn_cast<ConstantSDNode>(N1)) {
1919       const APInt &C1 = N1C->getAPIntValue();
1920
1921       switch (Cond) {
1922       default: llvm_unreachable("Unknown integer setcc!");
1923       case ISD::SETEQ:  return getConstant(C1 == C2, dl, VT);
1924       case ISD::SETNE:  return getConstant(C1 != C2, dl, VT);
1925       case ISD::SETULT: return getConstant(C1.ult(C2), dl, VT);
1926       case ISD::SETUGT: return getConstant(C1.ugt(C2), dl, VT);
1927       case ISD::SETULE: return getConstant(C1.ule(C2), dl, VT);
1928       case ISD::SETUGE: return getConstant(C1.uge(C2), dl, VT);
1929       case ISD::SETLT:  return getConstant(C1.slt(C2), dl, VT);
1930       case ISD::SETGT:  return getConstant(C1.sgt(C2), dl, VT);
1931       case ISD::SETLE:  return getConstant(C1.sle(C2), dl, VT);
1932       case ISD::SETGE:  return getConstant(C1.sge(C2), dl, VT);
1933       }
1934     }
1935   }
1936   if (ConstantFPSDNode *N1C = dyn_cast<ConstantFPSDNode>(N1)) {
1937     if (ConstantFPSDNode *N2C = dyn_cast<ConstantFPSDNode>(N2)) {
1938       APFloat::cmpResult R = N1C->getValueAPF().compare(N2C->getValueAPF());
1939       switch (Cond) {
1940       default: break;
1941       case ISD::SETEQ:  if (R==APFloat::cmpUnordered)
1942                           return getUNDEF(VT);
1943                         // fall through
1944       case ISD::SETOEQ: return getConstant(R==APFloat::cmpEqual, dl, VT);
1945       case ISD::SETNE:  if (R==APFloat::cmpUnordered)
1946                           return getUNDEF(VT);
1947                         // fall through
1948       case ISD::SETONE: return getConstant(R==APFloat::cmpGreaterThan ||
1949                                            R==APFloat::cmpLessThan, dl, VT);
1950       case ISD::SETLT:  if (R==APFloat::cmpUnordered)
1951                           return getUNDEF(VT);
1952                         // fall through
1953       case ISD::SETOLT: return getConstant(R==APFloat::cmpLessThan, dl, VT);
1954       case ISD::SETGT:  if (R==APFloat::cmpUnordered)
1955                           return getUNDEF(VT);
1956                         // fall through
1957       case ISD::SETOGT: return getConstant(R==APFloat::cmpGreaterThan, dl, VT);
1958       case ISD::SETLE:  if (R==APFloat::cmpUnordered)
1959                           return getUNDEF(VT);
1960                         // fall through
1961       case ISD::SETOLE: return getConstant(R==APFloat::cmpLessThan ||
1962                                            R==APFloat::cmpEqual, dl, VT);
1963       case ISD::SETGE:  if (R==APFloat::cmpUnordered)
1964                           return getUNDEF(VT);
1965                         // fall through
1966       case ISD::SETOGE: return getConstant(R==APFloat::cmpGreaterThan ||
1967                                            R==APFloat::cmpEqual, dl, VT);
1968       case ISD::SETO:   return getConstant(R!=APFloat::cmpUnordered, dl, VT);
1969       case ISD::SETUO:  return getConstant(R==APFloat::cmpUnordered, dl, VT);
1970       case ISD::SETUEQ: return getConstant(R==APFloat::cmpUnordered ||
1971                                            R==APFloat::cmpEqual, dl, VT);
1972       case ISD::SETUNE: return getConstant(R!=APFloat::cmpEqual, dl, VT);
1973       case ISD::SETULT: return getConstant(R==APFloat::cmpUnordered ||
1974                                            R==APFloat::cmpLessThan, dl, VT);
1975       case ISD::SETUGT: return getConstant(R==APFloat::cmpGreaterThan ||
1976                                            R==APFloat::cmpUnordered, dl, VT);
1977       case ISD::SETULE: return getConstant(R!=APFloat::cmpGreaterThan, dl, VT);
1978       case ISD::SETUGE: return getConstant(R!=APFloat::cmpLessThan, dl, VT);
1979       }
1980     } else {
1981       // Ensure that the constant occurs on the RHS.
1982       ISD::CondCode SwappedCond = ISD::getSetCCSwappedOperands(Cond);
1983       MVT CompVT = N1.getValueType().getSimpleVT();
1984       if (!TLI->isCondCodeLegal(SwappedCond, CompVT))
1985         return SDValue();
1986
1987       return getSetCC(dl, VT, N2, N1, SwappedCond);
1988     }
1989   }
1990
1991   // Could not fold it.
1992   return SDValue();
1993 }
1994
1995 /// SignBitIsZero - Return true if the sign bit of Op is known to be zero.  We
1996 /// use this predicate to simplify operations downstream.
1997 bool SelectionDAG::SignBitIsZero(SDValue Op, unsigned Depth) const {
1998   // This predicate is not safe for vector operations.
1999   if (Op.getValueType().isVector())
2000     return false;
2001
2002   unsigned BitWidth = Op.getValueType().getScalarType().getSizeInBits();
2003   return MaskedValueIsZero(Op, APInt::getSignBit(BitWidth), Depth);
2004 }
2005
2006 /// MaskedValueIsZero - Return true if 'V & Mask' is known to be zero.  We use
2007 /// this predicate to simplify operations downstream.  Mask is known to be zero
2008 /// for bits that V cannot have.
2009 bool SelectionDAG::MaskedValueIsZero(SDValue Op, const APInt &Mask,
2010                                      unsigned Depth) const {
2011   APInt KnownZero, KnownOne;
2012   computeKnownBits(Op, KnownZero, KnownOne, Depth);
2013   return (KnownZero & Mask) == Mask;
2014 }
2015
2016 /// Determine which bits of Op are known to be either zero or one and return
2017 /// them in the KnownZero/KnownOne bitsets.
2018 void SelectionDAG::computeKnownBits(SDValue Op, APInt &KnownZero,
2019                                     APInt &KnownOne, unsigned Depth) const {
2020   unsigned BitWidth = Op.getValueType().getScalarType().getSizeInBits();
2021
2022   KnownZero = KnownOne = APInt(BitWidth, 0);   // Don't know anything.
2023   if (Depth == 6)
2024     return;  // Limit search depth.
2025
2026   APInt KnownZero2, KnownOne2;
2027
2028   switch (Op.getOpcode()) {
2029   case ISD::Constant:
2030     // We know all of the bits for a constant!
2031     KnownOne = cast<ConstantSDNode>(Op)->getAPIntValue();
2032     KnownZero = ~KnownOne;
2033     break;
2034   case ISD::AND:
2035     // If either the LHS or the RHS are Zero, the result is zero.
2036     computeKnownBits(Op.getOperand(1), KnownZero, KnownOne, Depth+1);
2037     computeKnownBits(Op.getOperand(0), KnownZero2, KnownOne2, Depth+1);
2038
2039     // Output known-1 bits are only known if set in both the LHS & RHS.
2040     KnownOne &= KnownOne2;
2041     // Output known-0 are known to be clear if zero in either the LHS | RHS.
2042     KnownZero |= KnownZero2;
2043     break;
2044   case ISD::OR:
2045     computeKnownBits(Op.getOperand(1), KnownZero, KnownOne, Depth+1);
2046     computeKnownBits(Op.getOperand(0), KnownZero2, KnownOne2, Depth+1);
2047
2048     // Output known-0 bits are only known if clear in both the LHS & RHS.
2049     KnownZero &= KnownZero2;
2050     // Output known-1 are known to be set if set in either the LHS | RHS.
2051     KnownOne |= KnownOne2;
2052     break;
2053   case ISD::XOR: {
2054     computeKnownBits(Op.getOperand(1), KnownZero, KnownOne, Depth+1);
2055     computeKnownBits(Op.getOperand(0), KnownZero2, KnownOne2, Depth+1);
2056
2057     // Output known-0 bits are known if clear or set in both the LHS & RHS.
2058     APInt KnownZeroOut = (KnownZero & KnownZero2) | (KnownOne & KnownOne2);
2059     // Output known-1 are known to be set if set in only one of the LHS, RHS.
2060     KnownOne = (KnownZero & KnownOne2) | (KnownOne & KnownZero2);
2061     KnownZero = KnownZeroOut;
2062     break;
2063   }
2064   case ISD::MUL: {
2065     computeKnownBits(Op.getOperand(1), KnownZero, KnownOne, Depth+1);
2066     computeKnownBits(Op.getOperand(0), KnownZero2, KnownOne2, Depth+1);
2067
2068     // If low bits are zero in either operand, output low known-0 bits.
2069     // Also compute a conserative estimate for high known-0 bits.
2070     // More trickiness is possible, but this is sufficient for the
2071     // interesting case of alignment computation.
2072     KnownOne.clearAllBits();
2073     unsigned TrailZ = KnownZero.countTrailingOnes() +
2074                       KnownZero2.countTrailingOnes();
2075     unsigned LeadZ =  std::max(KnownZero.countLeadingOnes() +
2076                                KnownZero2.countLeadingOnes(),
2077                                BitWidth) - BitWidth;
2078
2079     TrailZ = std::min(TrailZ, BitWidth);
2080     LeadZ = std::min(LeadZ, BitWidth);
2081     KnownZero = APInt::getLowBitsSet(BitWidth, TrailZ) |
2082                 APInt::getHighBitsSet(BitWidth, LeadZ);
2083     break;
2084   }
2085   case ISD::UDIV: {
2086     // For the purposes of computing leading zeros we can conservatively
2087     // treat a udiv as a logical right shift by the power of 2 known to
2088     // be less than the denominator.
2089     computeKnownBits(Op.getOperand(0), KnownZero2, KnownOne2, Depth+1);
2090     unsigned LeadZ = KnownZero2.countLeadingOnes();
2091
2092     KnownOne2.clearAllBits();
2093     KnownZero2.clearAllBits();
2094     computeKnownBits(Op.getOperand(1), KnownZero2, KnownOne2, Depth+1);
2095     unsigned RHSUnknownLeadingOnes = KnownOne2.countLeadingZeros();
2096     if (RHSUnknownLeadingOnes != BitWidth)
2097       LeadZ = std::min(BitWidth,
2098                        LeadZ + BitWidth - RHSUnknownLeadingOnes - 1);
2099
2100     KnownZero = APInt::getHighBitsSet(BitWidth, LeadZ);
2101     break;
2102   }
2103   case ISD::SELECT:
2104     computeKnownBits(Op.getOperand(2), KnownZero, KnownOne, Depth+1);
2105     computeKnownBits(Op.getOperand(1), KnownZero2, KnownOne2, Depth+1);
2106
2107     // Only known if known in both the LHS and RHS.
2108     KnownOne &= KnownOne2;
2109     KnownZero &= KnownZero2;
2110     break;
2111   case ISD::SELECT_CC:
2112     computeKnownBits(Op.getOperand(3), KnownZero, KnownOne, Depth+1);
2113     computeKnownBits(Op.getOperand(2), KnownZero2, KnownOne2, Depth+1);
2114
2115     // Only known if known in both the LHS and RHS.
2116     KnownOne &= KnownOne2;
2117     KnownZero &= KnownZero2;
2118     break;
2119   case ISD::SADDO:
2120   case ISD::UADDO:
2121   case ISD::SSUBO:
2122   case ISD::USUBO:
2123   case ISD::SMULO:
2124   case ISD::UMULO:
2125     if (Op.getResNo() != 1)
2126       break;
2127     // The boolean result conforms to getBooleanContents.
2128     // If we know the result of a setcc has the top bits zero, use this info.
2129     // We know that we have an integer-based boolean since these operations
2130     // are only available for integer.
2131     if (TLI->getBooleanContents(Op.getValueType().isVector(), false) ==
2132             TargetLowering::ZeroOrOneBooleanContent &&
2133         BitWidth > 1)
2134       KnownZero |= APInt::getHighBitsSet(BitWidth, BitWidth - 1);
2135     break;
2136   case ISD::SETCC:
2137     // If we know the result of a setcc has the top bits zero, use this info.
2138     if (TLI->getBooleanContents(Op.getOperand(0).getValueType()) ==
2139             TargetLowering::ZeroOrOneBooleanContent &&
2140         BitWidth > 1)
2141       KnownZero |= APInt::getHighBitsSet(BitWidth, BitWidth - 1);
2142     break;
2143   case ISD::SHL:
2144     // (shl X, C1) & C2 == 0   iff   (X & C2 >>u C1) == 0
2145     if (ConstantSDNode *SA = dyn_cast<ConstantSDNode>(Op.getOperand(1))) {
2146       unsigned ShAmt = SA->getZExtValue();
2147
2148       // If the shift count is an invalid immediate, don't do anything.
2149       if (ShAmt >= BitWidth)
2150         break;
2151
2152       computeKnownBits(Op.getOperand(0), KnownZero, KnownOne, Depth+1);
2153       KnownZero <<= ShAmt;
2154       KnownOne  <<= ShAmt;
2155       // low bits known zero.
2156       KnownZero |= APInt::getLowBitsSet(BitWidth, ShAmt);
2157     }
2158     break;
2159   case ISD::SRL:
2160     // (ushr X, C1) & C2 == 0   iff  (-1 >> C1) & C2 == 0
2161     if (ConstantSDNode *SA = dyn_cast<ConstantSDNode>(Op.getOperand(1))) {
2162       unsigned ShAmt = SA->getZExtValue();
2163
2164       // If the shift count is an invalid immediate, don't do anything.
2165       if (ShAmt >= BitWidth)
2166         break;
2167
2168       computeKnownBits(Op.getOperand(0), KnownZero, KnownOne, Depth+1);
2169       KnownZero = KnownZero.lshr(ShAmt);
2170       KnownOne  = KnownOne.lshr(ShAmt);
2171
2172       APInt HighBits = APInt::getHighBitsSet(BitWidth, ShAmt);
2173       KnownZero |= HighBits;  // High bits known zero.
2174     }
2175     break;
2176   case ISD::SRA:
2177     if (ConstantSDNode *SA = dyn_cast<ConstantSDNode>(Op.getOperand(1))) {
2178       unsigned ShAmt = SA->getZExtValue();
2179
2180       // If the shift count is an invalid immediate, don't do anything.
2181       if (ShAmt >= BitWidth)
2182         break;
2183
2184       // If any of the demanded bits are produced by the sign extension, we also
2185       // demand the input sign bit.
2186       APInt HighBits = APInt::getHighBitsSet(BitWidth, ShAmt);
2187
2188       computeKnownBits(Op.getOperand(0), KnownZero, KnownOne, Depth+1);
2189       KnownZero = KnownZero.lshr(ShAmt);
2190       KnownOne  = KnownOne.lshr(ShAmt);
2191
2192       // Handle the sign bits.
2193       APInt SignBit = APInt::getSignBit(BitWidth);
2194       SignBit = SignBit.lshr(ShAmt);  // Adjust to where it is now in the mask.
2195
2196       if (KnownZero.intersects(SignBit)) {
2197         KnownZero |= HighBits;  // New bits are known zero.
2198       } else if (KnownOne.intersects(SignBit)) {
2199         KnownOne  |= HighBits;  // New bits are known one.
2200       }
2201     }
2202     break;
2203   case ISD::SIGN_EXTEND_INREG: {
2204     EVT EVT = cast<VTSDNode>(Op.getOperand(1))->getVT();
2205     unsigned EBits = EVT.getScalarType().getSizeInBits();
2206
2207     // Sign extension.  Compute the demanded bits in the result that are not
2208     // present in the input.
2209     APInt NewBits = APInt::getHighBitsSet(BitWidth, BitWidth - EBits);
2210
2211     APInt InSignBit = APInt::getSignBit(EBits);
2212     APInt InputDemandedBits = APInt::getLowBitsSet(BitWidth, EBits);
2213
2214     // If the sign extended bits are demanded, we know that the sign
2215     // bit is demanded.
2216     InSignBit = InSignBit.zext(BitWidth);
2217     if (NewBits.getBoolValue())
2218       InputDemandedBits |= InSignBit;
2219
2220     computeKnownBits(Op.getOperand(0), KnownZero, KnownOne, Depth+1);
2221     KnownOne &= InputDemandedBits;
2222     KnownZero &= InputDemandedBits;
2223
2224     // If the sign bit of the input is known set or clear, then we know the
2225     // top bits of the result.
2226     if (KnownZero.intersects(InSignBit)) {         // Input sign bit known clear
2227       KnownZero |= NewBits;
2228       KnownOne  &= ~NewBits;
2229     } else if (KnownOne.intersects(InSignBit)) {   // Input sign bit known set
2230       KnownOne  |= NewBits;
2231       KnownZero &= ~NewBits;
2232     } else {                              // Input sign bit unknown
2233       KnownZero &= ~NewBits;
2234       KnownOne  &= ~NewBits;
2235     }
2236     break;
2237   }
2238   case ISD::CTTZ:
2239   case ISD::CTTZ_ZERO_UNDEF:
2240   case ISD::CTLZ:
2241   case ISD::CTLZ_ZERO_UNDEF:
2242   case ISD::CTPOP: {
2243     unsigned LowBits = Log2_32(BitWidth)+1;
2244     KnownZero = APInt::getHighBitsSet(BitWidth, BitWidth - LowBits);
2245     KnownOne.clearAllBits();
2246     break;
2247   }
2248   case ISD::LOAD: {
2249     LoadSDNode *LD = cast<LoadSDNode>(Op);
2250     // If this is a ZEXTLoad and we are looking at the loaded value.
2251     if (ISD::isZEXTLoad(Op.getNode()) && Op.getResNo() == 0) {
2252       EVT VT = LD->getMemoryVT();
2253       unsigned MemBits = VT.getScalarType().getSizeInBits();
2254       KnownZero |= APInt::getHighBitsSet(BitWidth, BitWidth - MemBits);
2255     } else if (const MDNode *Ranges = LD->getRanges()) {
2256       computeKnownBitsFromRangeMetadata(*Ranges, KnownZero);
2257     }
2258     break;
2259   }
2260   case ISD::ZERO_EXTEND: {
2261     EVT InVT = Op.getOperand(0).getValueType();
2262     unsigned InBits = InVT.getScalarType().getSizeInBits();
2263     APInt NewBits   = APInt::getHighBitsSet(BitWidth, BitWidth - InBits);
2264     KnownZero = KnownZero.trunc(InBits);
2265     KnownOne = KnownOne.trunc(InBits);
2266     computeKnownBits(Op.getOperand(0), KnownZero, KnownOne, Depth+1);
2267     KnownZero = KnownZero.zext(BitWidth);
2268     KnownOne = KnownOne.zext(BitWidth);
2269     KnownZero |= NewBits;
2270     break;
2271   }
2272   case ISD::SIGN_EXTEND: {
2273     EVT InVT = Op.getOperand(0).getValueType();
2274     unsigned InBits = InVT.getScalarType().getSizeInBits();
2275     APInt NewBits   = APInt::getHighBitsSet(BitWidth, BitWidth - InBits);
2276
2277     KnownZero = KnownZero.trunc(InBits);
2278     KnownOne = KnownOne.trunc(InBits);
2279     computeKnownBits(Op.getOperand(0), KnownZero, KnownOne, Depth+1);
2280
2281     // Note if the sign bit is known to be zero or one.
2282     bool SignBitKnownZero = KnownZero.isNegative();
2283     bool SignBitKnownOne  = KnownOne.isNegative();
2284
2285     KnownZero = KnownZero.zext(BitWidth);
2286     KnownOne = KnownOne.zext(BitWidth);
2287
2288     // If the sign bit is known zero or one, the top bits match.
2289     if (SignBitKnownZero)
2290       KnownZero |= NewBits;
2291     else if (SignBitKnownOne)
2292       KnownOne  |= NewBits;
2293     break;
2294   }
2295   case ISD::ANY_EXTEND: {
2296     EVT InVT = Op.getOperand(0).getValueType();
2297     unsigned InBits = InVT.getScalarType().getSizeInBits();
2298     KnownZero = KnownZero.trunc(InBits);
2299     KnownOne = KnownOne.trunc(InBits);
2300     computeKnownBits(Op.getOperand(0), KnownZero, KnownOne, Depth+1);
2301     KnownZero = KnownZero.zext(BitWidth);
2302     KnownOne = KnownOne.zext(BitWidth);
2303     break;
2304   }
2305   case ISD::TRUNCATE: {
2306     EVT InVT = Op.getOperand(0).getValueType();
2307     unsigned InBits = InVT.getScalarType().getSizeInBits();
2308     KnownZero = KnownZero.zext(InBits);
2309     KnownOne = KnownOne.zext(InBits);
2310     computeKnownBits(Op.getOperand(0), KnownZero, KnownOne, Depth+1);
2311     KnownZero = KnownZero.trunc(BitWidth);
2312     KnownOne = KnownOne.trunc(BitWidth);
2313     break;
2314   }
2315   case ISD::AssertZext: {
2316     EVT VT = cast<VTSDNode>(Op.getOperand(1))->getVT();
2317     APInt InMask = APInt::getLowBitsSet(BitWidth, VT.getSizeInBits());
2318     computeKnownBits(Op.getOperand(0), KnownZero, KnownOne, Depth+1);
2319     KnownZero |= (~InMask);
2320     KnownOne  &= (~KnownZero);
2321     break;
2322   }
2323   case ISD::FGETSIGN:
2324     // All bits are zero except the low bit.
2325     KnownZero = APInt::getHighBitsSet(BitWidth, BitWidth - 1);
2326     break;
2327
2328   case ISD::SUB: {
2329     if (ConstantSDNode *CLHS = dyn_cast<ConstantSDNode>(Op.getOperand(0))) {
2330       // We know that the top bits of C-X are clear if X contains less bits
2331       // than C (i.e. no wrap-around can happen).  For example, 20-X is
2332       // positive if we can prove that X is >= 0 and < 16.
2333       if (CLHS->getAPIntValue().isNonNegative()) {
2334         unsigned NLZ = (CLHS->getAPIntValue()+1).countLeadingZeros();
2335         // NLZ can't be BitWidth with no sign bit
2336         APInt MaskV = APInt::getHighBitsSet(BitWidth, NLZ+1);
2337         computeKnownBits(Op.getOperand(1), KnownZero2, KnownOne2, Depth+1);
2338
2339         // If all of the MaskV bits are known to be zero, then we know the
2340         // output top bits are zero, because we now know that the output is
2341         // from [0-C].
2342         if ((KnownZero2 & MaskV) == MaskV) {
2343           unsigned NLZ2 = CLHS->getAPIntValue().countLeadingZeros();
2344           // Top bits known zero.
2345           KnownZero = APInt::getHighBitsSet(BitWidth, NLZ2);
2346         }
2347       }
2348     }
2349   }
2350   // fall through
2351   case ISD::ADD:
2352   case ISD::ADDE: {
2353     // Output known-0 bits are known if clear or set in both the low clear bits
2354     // common to both LHS & RHS.  For example, 8+(X<<3) is known to have the
2355     // low 3 bits clear.
2356     // Output known-0 bits are also known if the top bits of each input are
2357     // known to be clear. For example, if one input has the top 10 bits clear
2358     // and the other has the top 8 bits clear, we know the top 7 bits of the
2359     // output must be clear.
2360     computeKnownBits(Op.getOperand(0), KnownZero2, KnownOne2, Depth+1);
2361     unsigned KnownZeroHigh = KnownZero2.countLeadingOnes();
2362     unsigned KnownZeroLow = KnownZero2.countTrailingOnes();
2363
2364     computeKnownBits(Op.getOperand(1), KnownZero2, KnownOne2, Depth+1);
2365     KnownZeroHigh = std::min(KnownZeroHigh,
2366                              KnownZero2.countLeadingOnes());
2367     KnownZeroLow = std::min(KnownZeroLow,
2368                             KnownZero2.countTrailingOnes());
2369
2370     if (Op.getOpcode() == ISD::ADD) {
2371       KnownZero |= APInt::getLowBitsSet(BitWidth, KnownZeroLow);
2372       if (KnownZeroHigh > 1)
2373         KnownZero |= APInt::getHighBitsSet(BitWidth, KnownZeroHigh - 1);
2374       break;
2375     }
2376
2377     // With ADDE, a carry bit may be added in, so we can only use this
2378     // information if we know (at least) that the low two bits are clear.  We
2379     // then return to the caller that the low bit is unknown but that other bits
2380     // are known zero.
2381     if (KnownZeroLow >= 2) // ADDE
2382       KnownZero |= APInt::getBitsSet(BitWidth, 1, KnownZeroLow);
2383     break;
2384   }
2385   case ISD::SREM:
2386     if (ConstantSDNode *Rem = dyn_cast<ConstantSDNode>(Op.getOperand(1))) {
2387       const APInt &RA = Rem->getAPIntValue().abs();
2388       if (RA.isPowerOf2()) {
2389         APInt LowBits = RA - 1;
2390         computeKnownBits(Op.getOperand(0), KnownZero2,KnownOne2,Depth+1);
2391
2392         // The low bits of the first operand are unchanged by the srem.
2393         KnownZero = KnownZero2 & LowBits;
2394         KnownOne = KnownOne2 & LowBits;
2395
2396         // If the first operand is non-negative or has all low bits zero, then
2397         // the upper bits are all zero.
2398         if (KnownZero2[BitWidth-1] || ((KnownZero2 & LowBits) == LowBits))
2399           KnownZero |= ~LowBits;
2400
2401         // If the first operand is negative and not all low bits are zero, then
2402         // the upper bits are all one.
2403         if (KnownOne2[BitWidth-1] && ((KnownOne2 & LowBits) != 0))
2404           KnownOne |= ~LowBits;
2405         assert((KnownZero & KnownOne) == 0&&"Bits known to be one AND zero?");
2406       }
2407     }
2408     break;
2409   case ISD::UREM: {
2410     if (ConstantSDNode *Rem = dyn_cast<ConstantSDNode>(Op.getOperand(1))) {
2411       const APInt &RA = Rem->getAPIntValue();
2412       if (RA.isPowerOf2()) {
2413         APInt LowBits = (RA - 1);
2414         computeKnownBits(Op.getOperand(0), KnownZero2, KnownOne2, Depth + 1);
2415
2416         // The upper bits are all zero, the lower ones are unchanged.
2417         KnownZero = KnownZero2 | ~LowBits;
2418         KnownOne = KnownOne2 & LowBits;
2419         break;
2420       }
2421     }
2422
2423     // Since the result is less than or equal to either operand, any leading
2424     // zero bits in either operand must also exist in the result.
2425     computeKnownBits(Op.getOperand(0), KnownZero, KnownOne, Depth+1);
2426     computeKnownBits(Op.getOperand(1), KnownZero2, KnownOne2, Depth+1);
2427
2428     uint32_t Leaders = std::max(KnownZero.countLeadingOnes(),
2429                                 KnownZero2.countLeadingOnes());
2430     KnownOne.clearAllBits();
2431     KnownZero = APInt::getHighBitsSet(BitWidth, Leaders);
2432     break;
2433   }
2434   case ISD::EXTRACT_ELEMENT: {
2435     computeKnownBits(Op.getOperand(0), KnownZero, KnownOne, Depth+1);
2436     const unsigned Index =
2437       cast<ConstantSDNode>(Op.getOperand(1))->getZExtValue();
2438     const unsigned BitWidth = Op.getValueType().getSizeInBits();
2439
2440     // Remove low part of known bits mask
2441     KnownZero = KnownZero.getHiBits(KnownZero.getBitWidth() - Index * BitWidth);
2442     KnownOne = KnownOne.getHiBits(KnownOne.getBitWidth() - Index * BitWidth);
2443
2444     // Remove high part of known bit mask
2445     KnownZero = KnownZero.trunc(BitWidth);
2446     KnownOne = KnownOne.trunc(BitWidth);
2447     break;
2448   }
2449   case ISD::SMIN:
2450   case ISD::SMAX:
2451   case ISD::UMIN:
2452   case ISD::UMAX: {
2453     APInt Op0Zero, Op0One;
2454     APInt Op1Zero, Op1One;
2455     computeKnownBits(Op.getOperand(0), Op0Zero, Op0One, Depth);
2456     computeKnownBits(Op.getOperand(1), Op1Zero, Op1One, Depth);
2457
2458     KnownZero = Op0Zero & Op1Zero;
2459     KnownOne = Op0One & Op1One;
2460     break;
2461   }
2462   case ISD::FrameIndex:
2463   case ISD::TargetFrameIndex:
2464     if (unsigned Align = InferPtrAlignment(Op)) {
2465       // The low bits are known zero if the pointer is aligned.
2466       KnownZero = APInt::getLowBitsSet(BitWidth, Log2_32(Align));
2467       break;
2468     }
2469     break;
2470
2471   default:
2472     if (Op.getOpcode() < ISD::BUILTIN_OP_END)
2473       break;
2474     // Fallthrough
2475   case ISD::INTRINSIC_WO_CHAIN:
2476   case ISD::INTRINSIC_W_CHAIN:
2477   case ISD::INTRINSIC_VOID:
2478     // Allow the target to implement this method for its nodes.
2479     TLI->computeKnownBitsForTargetNode(Op, KnownZero, KnownOne, *this, Depth);
2480     break;
2481   }
2482
2483   assert((KnownZero & KnownOne) == 0 && "Bits known to be one AND zero?");
2484 }
2485
2486 /// ComputeNumSignBits - Return the number of times the sign bit of the
2487 /// register is replicated into the other bits.  We know that at least 1 bit
2488 /// is always equal to the sign bit (itself), but other cases can give us
2489 /// information.  For example, immediately after an "SRA X, 2", we know that
2490 /// the top 3 bits are all equal to each other, so we return 3.
2491 unsigned SelectionDAG::ComputeNumSignBits(SDValue Op, unsigned Depth) const{
2492   EVT VT = Op.getValueType();
2493   assert(VT.isInteger() && "Invalid VT!");
2494   unsigned VTBits = VT.getScalarType().getSizeInBits();
2495   unsigned Tmp, Tmp2;
2496   unsigned FirstAnswer = 1;
2497
2498   if (Depth == 6)
2499     return 1;  // Limit search depth.
2500
2501   switch (Op.getOpcode()) {
2502   default: break;
2503   case ISD::AssertSext:
2504     Tmp = cast<VTSDNode>(Op.getOperand(1))->getVT().getSizeInBits();
2505     return VTBits-Tmp+1;
2506   case ISD::AssertZext:
2507     Tmp = cast<VTSDNode>(Op.getOperand(1))->getVT().getSizeInBits();
2508     return VTBits-Tmp;
2509
2510   case ISD::Constant: {
2511     const APInt &Val = cast<ConstantSDNode>(Op)->getAPIntValue();
2512     return Val.getNumSignBits();
2513   }
2514
2515   case ISD::SIGN_EXTEND:
2516     Tmp =
2517         VTBits-Op.getOperand(0).getValueType().getScalarType().getSizeInBits();
2518     return ComputeNumSignBits(Op.getOperand(0), Depth+1) + Tmp;
2519
2520   case ISD::SIGN_EXTEND_INREG:
2521     // Max of the input and what this extends.
2522     Tmp =
2523       cast<VTSDNode>(Op.getOperand(1))->getVT().getScalarType().getSizeInBits();
2524     Tmp = VTBits-Tmp+1;
2525
2526     Tmp2 = ComputeNumSignBits(Op.getOperand(0), Depth+1);
2527     return std::max(Tmp, Tmp2);
2528
2529   case ISD::SRA:
2530     Tmp = ComputeNumSignBits(Op.getOperand(0), Depth+1);
2531     // SRA X, C   -> adds C sign bits.
2532     if (ConstantSDNode *C = dyn_cast<ConstantSDNode>(Op.getOperand(1))) {
2533       Tmp += C->getZExtValue();
2534       if (Tmp > VTBits) Tmp = VTBits;
2535     }
2536     return Tmp;
2537   case ISD::SHL:
2538     if (ConstantSDNode *C = dyn_cast<ConstantSDNode>(Op.getOperand(1))) {
2539       // shl destroys sign bits.
2540       Tmp = ComputeNumSignBits(Op.getOperand(0), Depth+1);
2541       if (C->getZExtValue() >= VTBits ||      // Bad shift.
2542           C->getZExtValue() >= Tmp) break;    // Shifted all sign bits out.
2543       return Tmp - C->getZExtValue();
2544     }
2545     break;
2546   case ISD::AND:
2547   case ISD::OR:
2548   case ISD::XOR:    // NOT is handled here.
2549     // Logical binary ops preserve the number of sign bits at the worst.
2550     Tmp = ComputeNumSignBits(Op.getOperand(0), Depth+1);
2551     if (Tmp != 1) {
2552       Tmp2 = ComputeNumSignBits(Op.getOperand(1), Depth+1);
2553       FirstAnswer = std::min(Tmp, Tmp2);
2554       // We computed what we know about the sign bits as our first
2555       // answer. Now proceed to the generic code that uses
2556       // computeKnownBits, and pick whichever answer is better.
2557     }
2558     break;
2559
2560   case ISD::SELECT:
2561     Tmp = ComputeNumSignBits(Op.getOperand(1), Depth+1);
2562     if (Tmp == 1) return 1;  // Early out.
2563     Tmp2 = ComputeNumSignBits(Op.getOperand(2), Depth+1);
2564     return std::min(Tmp, Tmp2);
2565   case ISD::SMIN:
2566   case ISD::SMAX:
2567   case ISD::UMIN:
2568   case ISD::UMAX:
2569     Tmp = ComputeNumSignBits(Op.getOperand(0), Depth + 1);
2570     if (Tmp == 1)
2571       return 1;  // Early out.
2572     Tmp2 = ComputeNumSignBits(Op.getOperand(1), Depth + 1);
2573     return std::min(Tmp, Tmp2);
2574   case ISD::SADDO:
2575   case ISD::UADDO:
2576   case ISD::SSUBO:
2577   case ISD::USUBO:
2578   case ISD::SMULO:
2579   case ISD::UMULO:
2580     if (Op.getResNo() != 1)
2581       break;
2582     // The boolean result conforms to getBooleanContents.  Fall through.
2583     // If setcc returns 0/-1, all bits are sign bits.
2584     // We know that we have an integer-based boolean since these operations
2585     // are only available for integer.
2586     if (TLI->getBooleanContents(Op.getValueType().isVector(), false) ==
2587         TargetLowering::ZeroOrNegativeOneBooleanContent)
2588       return VTBits;
2589     break;
2590   case ISD::SETCC:
2591     // If setcc returns 0/-1, all bits are sign bits.
2592     if (TLI->getBooleanContents(Op.getOperand(0).getValueType()) ==
2593         TargetLowering::ZeroOrNegativeOneBooleanContent)
2594       return VTBits;
2595     break;
2596   case ISD::ROTL:
2597   case ISD::ROTR:
2598     if (ConstantSDNode *C = dyn_cast<ConstantSDNode>(Op.getOperand(1))) {
2599       unsigned RotAmt = C->getZExtValue() & (VTBits-1);
2600
2601       // Handle rotate right by N like a rotate left by 32-N.
2602       if (Op.getOpcode() == ISD::ROTR)
2603         RotAmt = (VTBits-RotAmt) & (VTBits-1);
2604
2605       // If we aren't rotating out all of the known-in sign bits, return the
2606       // number that are left.  This handles rotl(sext(x), 1) for example.
2607       Tmp = ComputeNumSignBits(Op.getOperand(0), Depth+1);
2608       if (Tmp > RotAmt+1) return Tmp-RotAmt;
2609     }
2610     break;
2611   case ISD::ADD:
2612     // Add can have at most one carry bit.  Thus we know that the output
2613     // is, at worst, one more bit than the inputs.
2614     Tmp = ComputeNumSignBits(Op.getOperand(0), Depth+1);
2615     if (Tmp == 1) return 1;  // Early out.
2616
2617     // Special case decrementing a value (ADD X, -1):
2618     if (ConstantSDNode *CRHS = dyn_cast<ConstantSDNode>(Op.getOperand(1)))
2619       if (CRHS->isAllOnesValue()) {
2620         APInt KnownZero, KnownOne;
2621         computeKnownBits(Op.getOperand(0), KnownZero, KnownOne, Depth+1);
2622
2623         // If the input is known to be 0 or 1, the output is 0/-1, which is all
2624         // sign bits set.
2625         if ((KnownZero | APInt(VTBits, 1)).isAllOnesValue())
2626           return VTBits;
2627
2628         // If we are subtracting one from a positive number, there is no carry
2629         // out of the result.
2630         if (KnownZero.isNegative())
2631           return Tmp;
2632       }
2633
2634     Tmp2 = ComputeNumSignBits(Op.getOperand(1), Depth+1);
2635     if (Tmp2 == 1) return 1;
2636     return std::min(Tmp, Tmp2)-1;
2637
2638   case ISD::SUB:
2639     Tmp2 = ComputeNumSignBits(Op.getOperand(1), Depth+1);
2640     if (Tmp2 == 1) return 1;
2641
2642     // Handle NEG.
2643     if (ConstantSDNode *CLHS = dyn_cast<ConstantSDNode>(Op.getOperand(0)))
2644       if (CLHS->isNullValue()) {
2645         APInt KnownZero, KnownOne;
2646         computeKnownBits(Op.getOperand(1), KnownZero, KnownOne, Depth+1);
2647         // If the input is known to be 0 or 1, the output is 0/-1, which is all
2648         // sign bits set.
2649         if ((KnownZero | APInt(VTBits, 1)).isAllOnesValue())
2650           return VTBits;
2651
2652         // If the input is known to be positive (the sign bit is known clear),
2653         // the output of the NEG has the same number of sign bits as the input.
2654         if (KnownZero.isNegative())
2655           return Tmp2;
2656
2657         // Otherwise, we treat this like a SUB.
2658       }
2659
2660     // Sub can have at most one carry bit.  Thus we know that the output
2661     // is, at worst, one more bit than the inputs.
2662     Tmp = ComputeNumSignBits(Op.getOperand(0), Depth+1);
2663     if (Tmp == 1) return 1;  // Early out.
2664     return std::min(Tmp, Tmp2)-1;
2665   case ISD::TRUNCATE:
2666     // FIXME: it's tricky to do anything useful for this, but it is an important
2667     // case for targets like X86.
2668     break;
2669   case ISD::EXTRACT_ELEMENT: {
2670     const int KnownSign = ComputeNumSignBits(Op.getOperand(0), Depth+1);
2671     const int BitWidth = Op.getValueType().getSizeInBits();
2672     const int Items =
2673       Op.getOperand(0).getValueType().getSizeInBits() / BitWidth;
2674
2675     // Get reverse index (starting from 1), Op1 value indexes elements from
2676     // little end. Sign starts at big end.
2677     const int rIndex = Items - 1 -
2678       cast<ConstantSDNode>(Op.getOperand(1))->getZExtValue();
2679
2680     // If the sign portion ends in our element the substraction gives correct
2681     // result. Otherwise it gives either negative or > bitwidth result
2682     return std::max(std::min(KnownSign - rIndex * BitWidth, BitWidth), 0);
2683   }
2684   }
2685
2686   // If we are looking at the loaded value of the SDNode.
2687   if (Op.getResNo() == 0) {
2688     // Handle LOADX separately here. EXTLOAD case will fallthrough.
2689     if (LoadSDNode *LD = dyn_cast<LoadSDNode>(Op)) {
2690       unsigned ExtType = LD->getExtensionType();
2691       switch (ExtType) {
2692         default: break;
2693         case ISD::SEXTLOAD:    // '17' bits known
2694           Tmp = LD->getMemoryVT().getScalarType().getSizeInBits();
2695           return VTBits-Tmp+1;
2696         case ISD::ZEXTLOAD:    // '16' bits known
2697           Tmp = LD->getMemoryVT().getScalarType().getSizeInBits();
2698           return VTBits-Tmp;
2699       }
2700     }
2701   }
2702
2703   // Allow the target to implement this method for its nodes.
2704   if (Op.getOpcode() >= ISD::BUILTIN_OP_END ||
2705       Op.getOpcode() == ISD::INTRINSIC_WO_CHAIN ||
2706       Op.getOpcode() == ISD::INTRINSIC_W_CHAIN ||
2707       Op.getOpcode() == ISD::INTRINSIC_VOID) {
2708     unsigned NumBits = TLI->ComputeNumSignBitsForTargetNode(Op, *this, Depth);
2709     if (NumBits > 1) FirstAnswer = std::max(FirstAnswer, NumBits);
2710   }
2711
2712   // Finally, if we can prove that the top bits of the result are 0's or 1's,
2713   // use this information.
2714   APInt KnownZero, KnownOne;
2715   computeKnownBits(Op, KnownZero, KnownOne, Depth);
2716
2717   APInt Mask;
2718   if (KnownZero.isNegative()) {        // sign bit is 0
2719     Mask = KnownZero;
2720   } else if (KnownOne.isNegative()) {  // sign bit is 1;
2721     Mask = KnownOne;
2722   } else {
2723     // Nothing known.
2724     return FirstAnswer;
2725   }
2726
2727   // Okay, we know that the sign bit in Mask is set.  Use CLZ to determine
2728   // the number of identical bits in the top of the input value.
2729   Mask = ~Mask;
2730   Mask <<= Mask.getBitWidth()-VTBits;
2731   // Return # leading zeros.  We use 'min' here in case Val was zero before
2732   // shifting.  We don't want to return '64' as for an i32 "0".
2733   return std::max(FirstAnswer, std::min(VTBits, Mask.countLeadingZeros()));
2734 }
2735
2736 /// isBaseWithConstantOffset - Return true if the specified operand is an
2737 /// ISD::ADD with a ConstantSDNode on the right-hand side, or if it is an
2738 /// ISD::OR with a ConstantSDNode that is guaranteed to have the same
2739 /// semantics as an ADD.  This handles the equivalence:
2740 ///     X|Cst == X+Cst iff X&Cst = 0.
2741 bool SelectionDAG::isBaseWithConstantOffset(SDValue Op) const {
2742   if ((Op.getOpcode() != ISD::ADD && Op.getOpcode() != ISD::OR) ||
2743       !isa<ConstantSDNode>(Op.getOperand(1)))
2744     return false;
2745
2746   if (Op.getOpcode() == ISD::OR &&
2747       !MaskedValueIsZero(Op.getOperand(0),
2748                      cast<ConstantSDNode>(Op.getOperand(1))->getAPIntValue()))
2749     return false;
2750
2751   return true;
2752 }
2753
2754
2755 bool SelectionDAG::isKnownNeverNaN(SDValue Op) const {
2756   // If we're told that NaNs won't happen, assume they won't.
2757   if (getTarget().Options.NoNaNsFPMath)
2758     return true;
2759
2760   // If the value is a constant, we can obviously see if it is a NaN or not.
2761   if (const ConstantFPSDNode *C = dyn_cast<ConstantFPSDNode>(Op))
2762     return !C->getValueAPF().isNaN();
2763
2764   // TODO: Recognize more cases here.
2765
2766   return false;
2767 }
2768
2769 bool SelectionDAG::isKnownNeverZero(SDValue Op) const {
2770   // If the value is a constant, we can obviously see if it is a zero or not.
2771   if (const ConstantFPSDNode *C = dyn_cast<ConstantFPSDNode>(Op))
2772     return !C->isZero();
2773
2774   // TODO: Recognize more cases here.
2775   switch (Op.getOpcode()) {
2776   default: break;
2777   case ISD::OR:
2778     if (const ConstantSDNode *C = dyn_cast<ConstantSDNode>(Op.getOperand(1)))
2779       return !C->isNullValue();
2780     break;
2781   }
2782
2783   return false;
2784 }
2785
2786 bool SelectionDAG::isEqualTo(SDValue A, SDValue B) const {
2787   // Check the obvious case.
2788   if (A == B) return true;
2789
2790   // For for negative and positive zero.
2791   if (const ConstantFPSDNode *CA = dyn_cast<ConstantFPSDNode>(A))
2792     if (const ConstantFPSDNode *CB = dyn_cast<ConstantFPSDNode>(B))
2793       if (CA->isZero() && CB->isZero()) return true;
2794
2795   // Otherwise they may not be equal.
2796   return false;
2797 }
2798
2799 /// getNode - Gets or creates the specified node.
2800 ///
2801 SDValue SelectionDAG::getNode(unsigned Opcode, SDLoc DL, EVT VT) {
2802   FoldingSetNodeID ID;
2803   AddNodeIDNode(ID, Opcode, getVTList(VT), None);
2804   void *IP = nullptr;
2805   if (SDNode *E = FindNodeOrInsertPos(ID, DL.getDebugLoc(), IP))
2806     return SDValue(E, 0);
2807
2808   SDNode *N = new (NodeAllocator) SDNode(Opcode, DL.getIROrder(),
2809                                          DL.getDebugLoc(), getVTList(VT));
2810   CSEMap.InsertNode(N, IP);
2811
2812   InsertNode(N);
2813   return SDValue(N, 0);
2814 }
2815
2816 SDValue SelectionDAG::getNode(unsigned Opcode, SDLoc DL,
2817                               EVT VT, SDValue Operand) {
2818   // Constant fold unary operations with an integer constant operand. Even
2819   // opaque constant will be folded, because the folding of unary operations
2820   // doesn't create new constants with different values. Nevertheless, the
2821   // opaque flag is preserved during folding to prevent future folding with
2822   // other constants.
2823   if (ConstantSDNode *C = dyn_cast<ConstantSDNode>(Operand)) {
2824     const APInt &Val = C->getAPIntValue();
2825     switch (Opcode) {
2826     default: break;
2827     case ISD::SIGN_EXTEND:
2828       return getConstant(Val.sextOrTrunc(VT.getSizeInBits()), DL, VT,
2829                          C->isTargetOpcode(), C->isOpaque());
2830     case ISD::ANY_EXTEND:
2831     case ISD::ZERO_EXTEND:
2832     case ISD::TRUNCATE:
2833       return getConstant(Val.zextOrTrunc(VT.getSizeInBits()), DL, VT,
2834                          C->isTargetOpcode(), C->isOpaque());
2835     case ISD::UINT_TO_FP:
2836     case ISD::SINT_TO_FP: {
2837       APFloat apf(EVTToAPFloatSemantics(VT),
2838                   APInt::getNullValue(VT.getSizeInBits()));
2839       (void)apf.convertFromAPInt(Val,
2840                                  Opcode==ISD::SINT_TO_FP,
2841                                  APFloat::rmNearestTiesToEven);
2842       return getConstantFP(apf, DL, VT);
2843     }
2844     case ISD::BITCAST:
2845       if (VT == MVT::f16 && C->getValueType(0) == MVT::i16)
2846         return getConstantFP(APFloat(APFloat::IEEEhalf, Val), DL, VT);
2847       if (VT == MVT::f32 && C->getValueType(0) == MVT::i32)
2848         return getConstantFP(APFloat(APFloat::IEEEsingle, Val), DL, VT);
2849       else if (VT == MVT::f64 && C->getValueType(0) == MVT::i64)
2850         return getConstantFP(APFloat(APFloat::IEEEdouble, Val), DL, VT);
2851       break;
2852     case ISD::BSWAP:
2853       return getConstant(Val.byteSwap(), DL, VT, C->isTargetOpcode(),
2854                          C->isOpaque());
2855     case ISD::CTPOP:
2856       return getConstant(Val.countPopulation(), DL, VT, C->isTargetOpcode(),
2857                          C->isOpaque());
2858     case ISD::CTLZ:
2859     case ISD::CTLZ_ZERO_UNDEF:
2860       return getConstant(Val.countLeadingZeros(), DL, VT, C->isTargetOpcode(),
2861                          C->isOpaque());
2862     case ISD::CTTZ:
2863     case ISD::CTTZ_ZERO_UNDEF:
2864       return getConstant(Val.countTrailingZeros(), DL, VT, C->isTargetOpcode(),
2865                          C->isOpaque());
2866     }
2867   }
2868
2869   // Constant fold unary operations with a floating point constant operand.
2870   if (ConstantFPSDNode *C = dyn_cast<ConstantFPSDNode>(Operand)) {
2871     APFloat V = C->getValueAPF();    // make copy
2872     switch (Opcode) {
2873     case ISD::FNEG:
2874       V.changeSign();
2875       return getConstantFP(V, DL, VT);
2876     case ISD::FABS:
2877       V.clearSign();
2878       return getConstantFP(V, DL, VT);
2879     case ISD::FCEIL: {
2880       APFloat::opStatus fs = V.roundToIntegral(APFloat::rmTowardPositive);
2881       if (fs == APFloat::opOK || fs == APFloat::opInexact)
2882         return getConstantFP(V, DL, VT);
2883       break;
2884     }
2885     case ISD::FTRUNC: {
2886       APFloat::opStatus fs = V.roundToIntegral(APFloat::rmTowardZero);
2887       if (fs == APFloat::opOK || fs == APFloat::opInexact)
2888         return getConstantFP(V, DL, VT);
2889       break;
2890     }
2891     case ISD::FFLOOR: {
2892       APFloat::opStatus fs = V.roundToIntegral(APFloat::rmTowardNegative);
2893       if (fs == APFloat::opOK || fs == APFloat::opInexact)
2894         return getConstantFP(V, DL, VT);
2895       break;
2896     }
2897     case ISD::FP_EXTEND: {
2898       bool ignored;
2899       // This can return overflow, underflow, or inexact; we don't care.
2900       // FIXME need to be more flexible about rounding mode.
2901       (void)V.convert(EVTToAPFloatSemantics(VT),
2902                       APFloat::rmNearestTiesToEven, &ignored);
2903       return getConstantFP(V, DL, VT);
2904     }
2905     case ISD::FP_TO_SINT:
2906     case ISD::FP_TO_UINT: {
2907       integerPart x[2];
2908       bool ignored;
2909       static_assert(integerPartWidth >= 64, "APFloat parts too small!");
2910       // FIXME need to be more flexible about rounding mode.
2911       APFloat::opStatus s = V.convertToInteger(x, VT.getSizeInBits(),
2912                             Opcode==ISD::FP_TO_SINT,
2913                             APFloat::rmTowardZero, &ignored);
2914       if (s==APFloat::opInvalidOp)     // inexact is OK, in fact usual
2915         break;
2916       APInt api(VT.getSizeInBits(), x);
2917       return getConstant(api, DL, VT);
2918     }
2919     case ISD::BITCAST:
2920       if (VT == MVT::i16 && C->getValueType(0) == MVT::f16)
2921         return getConstant((uint16_t)V.bitcastToAPInt().getZExtValue(), DL, VT);
2922       else if (VT == MVT::i32 && C->getValueType(0) == MVT::f32)
2923         return getConstant((uint32_t)V.bitcastToAPInt().getZExtValue(), DL, VT);
2924       else if (VT == MVT::i64 && C->getValueType(0) == MVT::f64)
2925         return getConstant(V.bitcastToAPInt().getZExtValue(), DL, VT);
2926       break;
2927     }
2928   }
2929
2930   // Constant fold unary operations with a vector integer or float operand.
2931   if (BuildVectorSDNode *BV = dyn_cast<BuildVectorSDNode>(Operand)) {
2932     if (BV->isConstant()) {
2933       switch (Opcode) {
2934       default:
2935         // FIXME: Entirely reasonable to perform folding of other unary
2936         // operations here as the need arises.
2937         break;
2938       case ISD::FNEG:
2939       case ISD::FABS:
2940       case ISD::FCEIL:
2941       case ISD::FTRUNC:
2942       case ISD::FFLOOR:
2943       case ISD::FP_EXTEND:
2944       case ISD::FP_TO_SINT:
2945       case ISD::FP_TO_UINT:
2946       case ISD::TRUNCATE:
2947       case ISD::UINT_TO_FP:
2948       case ISD::SINT_TO_FP:
2949       case ISD::BSWAP:
2950       case ISD::CTLZ:
2951       case ISD::CTLZ_ZERO_UNDEF:
2952       case ISD::CTTZ:
2953       case ISD::CTTZ_ZERO_UNDEF:
2954       case ISD::CTPOP: {
2955         EVT SVT = VT.getScalarType();
2956         EVT InVT = BV->getValueType(0);
2957         EVT InSVT = InVT.getScalarType();
2958
2959         // Find legal integer scalar type for constant promotion and
2960         // ensure that its scalar size is at least as large as source.
2961         EVT LegalSVT = SVT;
2962         if (SVT.isInteger()) {
2963           LegalSVT = TLI->getTypeToTransformTo(*getContext(), SVT);
2964           if (LegalSVT.bitsLT(SVT)) break;
2965         }
2966
2967         // Let the above scalar folding handle the folding of each element.
2968         SmallVector<SDValue, 8> Ops;
2969         for (int i = 0, e = VT.getVectorNumElements(); i != e; ++i) {
2970           SDValue OpN = BV->getOperand(i);
2971           EVT OpVT = OpN.getValueType();
2972
2973           // Build vector (integer) scalar operands may need implicit
2974           // truncation - do this before constant folding.
2975           if (OpVT.isInteger() && OpVT.bitsGT(InSVT))
2976             OpN = getNode(ISD::TRUNCATE, DL, InSVT, OpN);
2977
2978           OpN = getNode(Opcode, DL, SVT, OpN);
2979
2980           // Legalize the (integer) scalar constant if necessary.
2981           if (LegalSVT != SVT)
2982             OpN = getNode(ISD::ANY_EXTEND, DL, LegalSVT, OpN);
2983
2984           if (OpN.getOpcode() != ISD::UNDEF &&
2985               OpN.getOpcode() != ISD::Constant &&
2986               OpN.getOpcode() != ISD::ConstantFP)
2987             break;
2988           Ops.push_back(OpN);
2989         }
2990         if (Ops.size() == VT.getVectorNumElements())
2991           return getNode(ISD::BUILD_VECTOR, DL, VT, Ops);
2992         break;
2993       }
2994       }
2995     }
2996   }
2997
2998   unsigned OpOpcode = Operand.getNode()->getOpcode();
2999   switch (Opcode) {
3000   case ISD::TokenFactor:
3001   case ISD::MERGE_VALUES:
3002   case ISD::CONCAT_VECTORS:
3003     return Operand;         // Factor, merge or concat of one node?  No need.
3004   case ISD::FP_ROUND: llvm_unreachable("Invalid method to make FP_ROUND node");
3005   case ISD::FP_EXTEND:
3006     assert(VT.isFloatingPoint() &&
3007            Operand.getValueType().isFloatingPoint() && "Invalid FP cast!");
3008     if (Operand.getValueType() == VT) return Operand;  // noop conversion.
3009     assert((!VT.isVector() ||
3010             VT.getVectorNumElements() ==
3011             Operand.getValueType().getVectorNumElements()) &&
3012            "Vector element count mismatch!");
3013     if (Operand.getOpcode() == ISD::UNDEF)
3014       return getUNDEF(VT);
3015     break;
3016   case ISD::SIGN_EXTEND:
3017     assert(VT.isInteger() && Operand.getValueType().isInteger() &&
3018            "Invalid SIGN_EXTEND!");
3019     if (Operand.getValueType() == VT) return Operand;   // noop extension
3020     assert(Operand.getValueType().getScalarType().bitsLT(VT.getScalarType()) &&
3021            "Invalid sext node, dst < src!");
3022     assert((!VT.isVector() ||
3023             VT.getVectorNumElements() ==
3024             Operand.getValueType().getVectorNumElements()) &&
3025            "Vector element count mismatch!");
3026     if (OpOpcode == ISD::SIGN_EXTEND || OpOpcode == ISD::ZERO_EXTEND)
3027       return getNode(OpOpcode, DL, VT, Operand.getNode()->getOperand(0));
3028     else if (OpOpcode == ISD::UNDEF)
3029       // sext(undef) = 0, because the top bits will all be the same.
3030       return getConstant(0, DL, VT);
3031     break;
3032   case ISD::ZERO_EXTEND:
3033     assert(VT.isInteger() && Operand.getValueType().isInteger() &&
3034            "Invalid ZERO_EXTEND!");
3035     if (Operand.getValueType() == VT) return Operand;   // noop extension
3036     assert(Operand.getValueType().getScalarType().bitsLT(VT.getScalarType()) &&
3037            "Invalid zext node, dst < src!");
3038     assert((!VT.isVector() ||
3039             VT.getVectorNumElements() ==
3040             Operand.getValueType().getVectorNumElements()) &&
3041            "Vector element count mismatch!");
3042     if (OpOpcode == ISD::ZERO_EXTEND)   // (zext (zext x)) -> (zext x)
3043       return getNode(ISD::ZERO_EXTEND, DL, VT,
3044                      Operand.getNode()->getOperand(0));
3045     else if (OpOpcode == ISD::UNDEF)
3046       // zext(undef) = 0, because the top bits will be zero.
3047       return getConstant(0, DL, VT);
3048     break;
3049   case ISD::ANY_EXTEND:
3050     assert(VT.isInteger() && Operand.getValueType().isInteger() &&
3051            "Invalid ANY_EXTEND!");
3052     if (Operand.getValueType() == VT) return Operand;   // noop extension
3053     assert(Operand.getValueType().getScalarType().bitsLT(VT.getScalarType()) &&
3054            "Invalid anyext node, dst < src!");
3055     assert((!VT.isVector() ||
3056             VT.getVectorNumElements() ==
3057             Operand.getValueType().getVectorNumElements()) &&
3058            "Vector element count mismatch!");
3059
3060     if (OpOpcode == ISD::ZERO_EXTEND || OpOpcode == ISD::SIGN_EXTEND ||
3061         OpOpcode == ISD::ANY_EXTEND)
3062       // (ext (zext x)) -> (zext x)  and  (ext (sext x)) -> (sext x)
3063       return getNode(OpOpcode, DL, VT, Operand.getNode()->getOperand(0));
3064     else if (OpOpcode == ISD::UNDEF)
3065       return getUNDEF(VT);
3066
3067     // (ext (trunx x)) -> x
3068     if (OpOpcode == ISD::TRUNCATE) {
3069       SDValue OpOp = Operand.getNode()->getOperand(0);
3070       if (OpOp.getValueType() == VT)
3071         return OpOp;
3072     }
3073     break;
3074   case ISD::TRUNCATE:
3075     assert(VT.isInteger() && Operand.getValueType().isInteger() &&
3076            "Invalid TRUNCATE!");
3077     if (Operand.getValueType() == VT) return Operand;   // noop truncate
3078     assert(Operand.getValueType().getScalarType().bitsGT(VT.getScalarType()) &&
3079            "Invalid truncate node, src < dst!");
3080     assert((!VT.isVector() ||
3081             VT.getVectorNumElements() ==
3082             Operand.getValueType().getVectorNumElements()) &&
3083            "Vector element count mismatch!");
3084     if (OpOpcode == ISD::TRUNCATE)
3085       return getNode(ISD::TRUNCATE, DL, VT, Operand.getNode()->getOperand(0));
3086     if (OpOpcode == ISD::ZERO_EXTEND || OpOpcode == ISD::SIGN_EXTEND ||
3087         OpOpcode == ISD::ANY_EXTEND) {
3088       // If the source is smaller than the dest, we still need an extend.
3089       if (Operand.getNode()->getOperand(0).getValueType().getScalarType()
3090             .bitsLT(VT.getScalarType()))
3091         return getNode(OpOpcode, DL, VT, Operand.getNode()->getOperand(0));
3092       if (Operand.getNode()->getOperand(0).getValueType().bitsGT(VT))
3093         return getNode(ISD::TRUNCATE, DL, VT, Operand.getNode()->getOperand(0));
3094       return Operand.getNode()->getOperand(0);
3095     }
3096     if (OpOpcode == ISD::UNDEF)
3097       return getUNDEF(VT);
3098     break;
3099   case ISD::BSWAP:
3100     assert(VT.isInteger() && VT == Operand.getValueType() &&
3101            "Invalid BSWAP!");
3102     assert((VT.getScalarSizeInBits() % 16 == 0) &&
3103            "BSWAP types must be a multiple of 16 bits!");
3104     if (OpOpcode == ISD::UNDEF)
3105       return getUNDEF(VT);
3106     break;
3107   case ISD::BITCAST:
3108     // Basic sanity checking.
3109     assert(VT.getSizeInBits() == Operand.getValueType().getSizeInBits()
3110            && "Cannot BITCAST between types of different sizes!");
3111     if (VT == Operand.getValueType()) return Operand;  // noop conversion.
3112     if (OpOpcode == ISD::BITCAST)  // bitconv(bitconv(x)) -> bitconv(x)
3113       return getNode(ISD::BITCAST, DL, VT, Operand.getOperand(0));
3114     if (OpOpcode == ISD::UNDEF)
3115       return getUNDEF(VT);
3116     break;
3117   case ISD::SCALAR_TO_VECTOR:
3118     assert(VT.isVector() && !Operand.getValueType().isVector() &&
3119            (VT.getVectorElementType() == Operand.getValueType() ||
3120             (VT.getVectorElementType().isInteger() &&
3121              Operand.getValueType().isInteger() &&
3122              VT.getVectorElementType().bitsLE(Operand.getValueType()))) &&
3123            "Illegal SCALAR_TO_VECTOR node!");
3124     if (OpOpcode == ISD::UNDEF)
3125       return getUNDEF(VT);
3126     // scalar_to_vector(extract_vector_elt V, 0) -> V, top bits are undefined.
3127     if (OpOpcode == ISD::EXTRACT_VECTOR_ELT &&
3128         isa<ConstantSDNode>(Operand.getOperand(1)) &&
3129         Operand.getConstantOperandVal(1) == 0 &&
3130         Operand.getOperand(0).getValueType() == VT)
3131       return Operand.getOperand(0);
3132     break;
3133   case ISD::FNEG:
3134     // -(X-Y) -> (Y-X) is unsafe because when X==Y, -0.0 != +0.0
3135     if (getTarget().Options.UnsafeFPMath && OpOpcode == ISD::FSUB)
3136       return getNode(ISD::FSUB, DL, VT, Operand.getNode()->getOperand(1),
3137                      Operand.getNode()->getOperand(0));
3138     if (OpOpcode == ISD::FNEG)  // --X -> X
3139       return Operand.getNode()->getOperand(0);
3140     break;
3141   case ISD::FABS:
3142     if (OpOpcode == ISD::FNEG)  // abs(-X) -> abs(X)
3143       return getNode(ISD::FABS, DL, VT, Operand.getNode()->getOperand(0));
3144     break;
3145   }
3146
3147   SDNode *N;
3148   SDVTList VTs = getVTList(VT);
3149   if (VT != MVT::Glue) { // Don't CSE flag producing nodes
3150     FoldingSetNodeID ID;
3151     SDValue Ops[1] = { Operand };
3152     AddNodeIDNode(ID, Opcode, VTs, Ops);
3153     void *IP = nullptr;
3154     if (SDNode *E = FindNodeOrInsertPos(ID, DL.getDebugLoc(), IP))
3155       return SDValue(E, 0);
3156
3157     N = new (NodeAllocator) UnarySDNode(Opcode, DL.getIROrder(),
3158                                         DL.getDebugLoc(), VTs, Operand);
3159     CSEMap.InsertNode(N, IP);
3160   } else {
3161     N = new (NodeAllocator) UnarySDNode(Opcode, DL.getIROrder(),
3162                                         DL.getDebugLoc(), VTs, Operand);
3163   }
3164
3165   InsertNode(N);
3166   return SDValue(N, 0);
3167 }
3168
3169 static std::pair<APInt, bool> FoldValue(unsigned Opcode, const APInt &C1,
3170                                         const APInt &C2) {
3171   switch (Opcode) {
3172   case ISD::ADD:  return std::make_pair(C1 + C2, true);
3173   case ISD::SUB:  return std::make_pair(C1 - C2, true);
3174   case ISD::MUL:  return std::make_pair(C1 * C2, true);
3175   case ISD::AND:  return std::make_pair(C1 & C2, true);
3176   case ISD::OR:   return std::make_pair(C1 | C2, true);
3177   case ISD::XOR:  return std::make_pair(C1 ^ C2, true);
3178   case ISD::SHL:  return std::make_pair(C1 << C2, true);
3179   case ISD::SRL:  return std::make_pair(C1.lshr(C2), true);
3180   case ISD::SRA:  return std::make_pair(C1.ashr(C2), true);
3181   case ISD::ROTL: return std::make_pair(C1.rotl(C2), true);
3182   case ISD::ROTR: return std::make_pair(C1.rotr(C2), true);
3183   case ISD::UDIV:
3184     if (!C2.getBoolValue())
3185       break;
3186     return std::make_pair(C1.udiv(C2), true);
3187   case ISD::UREM:
3188     if (!C2.getBoolValue())
3189       break;
3190     return std::make_pair(C1.urem(C2), true);
3191   case ISD::SDIV:
3192     if (!C2.getBoolValue())
3193       break;
3194     return std::make_pair(C1.sdiv(C2), true);
3195   case ISD::SREM:
3196     if (!C2.getBoolValue())
3197       break;
3198     return std::make_pair(C1.srem(C2), true);
3199   }
3200   return std::make_pair(APInt(1, 0), false);
3201 }
3202
3203 SDValue SelectionDAG::FoldConstantArithmetic(unsigned Opcode, SDLoc DL, EVT VT,
3204                                              const ConstantSDNode *Cst1,
3205                                              const ConstantSDNode *Cst2) {
3206   if (Cst1->isOpaque() || Cst2->isOpaque())
3207     return SDValue();
3208
3209   std::pair<APInt, bool> Folded = FoldValue(Opcode, Cst1->getAPIntValue(),
3210                                             Cst2->getAPIntValue());
3211   if (!Folded.second)
3212     return SDValue();
3213   return getConstant(Folded.first, DL, VT);
3214 }
3215
3216 SDValue SelectionDAG::FoldConstantArithmetic(unsigned Opcode, SDLoc DL, EVT VT,
3217                                              SDNode *Cst1, SDNode *Cst2) {
3218   // If the opcode is a target-specific ISD node, there's nothing we can
3219   // do here and the operand rules may not line up with the below, so
3220   // bail early.
3221   if (Opcode >= ISD::BUILTIN_OP_END)
3222     return SDValue();
3223
3224   // Handle the case of two scalars.
3225   if (const ConstantSDNode *Scalar1 = dyn_cast<ConstantSDNode>(Cst1)) {
3226     if (const ConstantSDNode *Scalar2 = dyn_cast<ConstantSDNode>(Cst2)) {
3227       if (SDValue Folded =
3228           FoldConstantArithmetic(Opcode, DL, VT, Scalar1, Scalar2)) {
3229         if (!VT.isVector())
3230           return Folded;
3231         SmallVector<SDValue, 4> Outputs;
3232         // We may have a vector type but a scalar result. Create a splat.
3233         Outputs.resize(VT.getVectorNumElements(), Outputs.back());
3234         // Build a big vector out of the scalar elements we generated.
3235         return getNode(ISD::BUILD_VECTOR, SDLoc(), VT, Outputs);
3236       } else {
3237         return SDValue();
3238       }
3239     }
3240   }
3241
3242   // For vectors extract each constant element into Inputs so we can constant
3243   // fold them individually.
3244   BuildVectorSDNode *BV1 = dyn_cast<BuildVectorSDNode>(Cst1);
3245   BuildVectorSDNode *BV2 = dyn_cast<BuildVectorSDNode>(Cst2);
3246   if (!BV1 || !BV2)
3247     return SDValue();
3248
3249   assert(BV1->getNumOperands() == BV2->getNumOperands() && "Out of sync!");
3250
3251   EVT SVT = VT.getScalarType();
3252   SmallVector<SDValue, 4> Outputs;
3253   for (unsigned I = 0, E = BV1->getNumOperands(); I != E; ++I) {
3254     ConstantSDNode *V1 = dyn_cast<ConstantSDNode>(BV1->getOperand(I));
3255     ConstantSDNode *V2 = dyn_cast<ConstantSDNode>(BV2->getOperand(I));
3256     if (!V1 || !V2) // Not a constant, bail.
3257       return SDValue();
3258
3259     if (V1->isOpaque() || V2->isOpaque())
3260       return SDValue();
3261
3262     // Avoid BUILD_VECTOR nodes that perform implicit truncation.
3263     // FIXME: This is valid and could be handled by truncating the APInts.
3264     if (V1->getValueType(0) != SVT || V2->getValueType(0) != SVT)
3265       return SDValue();
3266
3267     // Fold one vector element.
3268     std::pair<APInt, bool> Folded = FoldValue(Opcode, V1->getAPIntValue(),
3269                                               V2->getAPIntValue());
3270     if (!Folded.second)
3271       return SDValue();
3272     Outputs.push_back(getConstant(Folded.first, DL, SVT));
3273   }
3274
3275   assert(VT.getVectorNumElements() == Outputs.size() &&
3276          "Vector size mismatch!");
3277
3278   // We may have a vector type but a scalar result. Create a splat.
3279   Outputs.resize(VT.getVectorNumElements(), Outputs.back());
3280
3281   // Build a big vector out of the scalar elements we generated.
3282   return getNode(ISD::BUILD_VECTOR, SDLoc(), VT, Outputs);
3283 }
3284
3285 SDValue SelectionDAG::getNode(unsigned Opcode, SDLoc DL, EVT VT, SDValue N1,
3286                               SDValue N2, const SDNodeFlags *Flags) {
3287   ConstantSDNode *N1C = dyn_cast<ConstantSDNode>(N1);
3288   ConstantSDNode *N2C = dyn_cast<ConstantSDNode>(N2);
3289   switch (Opcode) {
3290   default: break;
3291   case ISD::TokenFactor:
3292     assert(VT == MVT::Other && N1.getValueType() == MVT::Other &&
3293            N2.getValueType() == MVT::Other && "Invalid token factor!");
3294     // Fold trivial token factors.
3295     if (N1.getOpcode() == ISD::EntryToken) return N2;
3296     if (N2.getOpcode() == ISD::EntryToken) return N1;
3297     if (N1 == N2) return N1;
3298     break;
3299   case ISD::CONCAT_VECTORS:
3300     // Concat of UNDEFs is UNDEF.
3301     if (N1.getOpcode() == ISD::UNDEF &&
3302         N2.getOpcode() == ISD::UNDEF)
3303       return getUNDEF(VT);
3304
3305     // A CONCAT_VECTOR with all operands BUILD_VECTOR can be simplified to
3306     // one big BUILD_VECTOR.
3307     if (N1.getOpcode() == ISD::BUILD_VECTOR &&
3308         N2.getOpcode() == ISD::BUILD_VECTOR) {
3309       SmallVector<SDValue, 16> Elts(N1.getNode()->op_begin(),
3310                                     N1.getNode()->op_end());
3311       Elts.append(N2.getNode()->op_begin(), N2.getNode()->op_end());
3312
3313       // BUILD_VECTOR requires all inputs to be of the same type, find the
3314       // maximum type and extend them all.
3315       EVT SVT = VT.getScalarType();
3316       for (SDValue Op : Elts)
3317         SVT = (SVT.bitsLT(Op.getValueType()) ? Op.getValueType() : SVT);
3318       if (SVT.bitsGT(VT.getScalarType()))
3319         for (SDValue &Op : Elts)
3320           Op = TLI->isZExtFree(Op.getValueType(), SVT)
3321              ? getZExtOrTrunc(Op, DL, SVT)
3322              : getSExtOrTrunc(Op, DL, SVT);
3323
3324       return getNode(ISD::BUILD_VECTOR, DL, VT, Elts);
3325     }
3326     break;
3327   case ISD::AND:
3328     assert(VT.isInteger() && "This operator does not apply to FP types!");
3329     assert(N1.getValueType() == N2.getValueType() &&
3330            N1.getValueType() == VT && "Binary operator types must match!");
3331     // (X & 0) -> 0.  This commonly occurs when legalizing i64 values, so it's
3332     // worth handling here.
3333     if (N2C && N2C->isNullValue())
3334       return N2;
3335     if (N2C && N2C->isAllOnesValue())  // X & -1 -> X
3336       return N1;
3337     break;
3338   case ISD::OR:
3339   case ISD::XOR:
3340   case ISD::ADD:
3341   case ISD::SUB:
3342     assert(VT.isInteger() && "This operator does not apply to FP types!");
3343     assert(N1.getValueType() == N2.getValueType() &&
3344            N1.getValueType() == VT && "Binary operator types must match!");
3345     // (X ^|+- 0) -> X.  This commonly occurs when legalizing i64 values, so
3346     // it's worth handling here.
3347     if (N2C && N2C->isNullValue())
3348       return N1;
3349     break;
3350   case ISD::UDIV:
3351   case ISD::UREM:
3352   case ISD::MULHU:
3353   case ISD::MULHS:
3354   case ISD::MUL:
3355   case ISD::SDIV:
3356   case ISD::SREM:
3357     assert(VT.isInteger() && "This operator does not apply to FP types!");
3358     assert(N1.getValueType() == N2.getValueType() &&
3359            N1.getValueType() == VT && "Binary operator types must match!");
3360     break;
3361   case ISD::FADD:
3362   case ISD::FSUB:
3363   case ISD::FMUL:
3364   case ISD::FDIV:
3365   case ISD::FREM:
3366     if (getTarget().Options.UnsafeFPMath) {
3367       if (Opcode == ISD::FADD) {
3368         // 0+x --> x
3369         if (ConstantFPSDNode *CFP = dyn_cast<ConstantFPSDNode>(N1))
3370           if (CFP->getValueAPF().isZero())
3371             return N2;
3372         // x+0 --> x
3373         if (ConstantFPSDNode *CFP = dyn_cast<ConstantFPSDNode>(N2))
3374           if (CFP->getValueAPF().isZero())
3375             return N1;
3376       } else if (Opcode == ISD::FSUB) {
3377         // x-0 --> x
3378         if (ConstantFPSDNode *CFP = dyn_cast<ConstantFPSDNode>(N2))
3379           if (CFP->getValueAPF().isZero())
3380             return N1;
3381       } else if (Opcode == ISD::FMUL) {
3382         ConstantFPSDNode *CFP = dyn_cast<ConstantFPSDNode>(N1);
3383         SDValue V = N2;
3384
3385         // If the first operand isn't the constant, try the second
3386         if (!CFP) {
3387           CFP = dyn_cast<ConstantFPSDNode>(N2);
3388           V = N1;
3389         }
3390
3391         if (CFP) {
3392           // 0*x --> 0
3393           if (CFP->isZero())
3394             return SDValue(CFP,0);
3395           // 1*x --> x
3396           if (CFP->isExactlyValue(1.0))
3397             return V;
3398         }
3399       }
3400     }
3401     assert(VT.isFloatingPoint() && "This operator only applies to FP types!");
3402     assert(N1.getValueType() == N2.getValueType() &&
3403            N1.getValueType() == VT && "Binary operator types must match!");
3404     break;
3405   case ISD::FCOPYSIGN:   // N1 and result must match.  N1/N2 need not match.
3406     assert(N1.getValueType() == VT &&
3407            N1.getValueType().isFloatingPoint() &&
3408            N2.getValueType().isFloatingPoint() &&
3409            "Invalid FCOPYSIGN!");
3410     break;
3411   case ISD::SHL:
3412   case ISD::SRA:
3413   case ISD::SRL:
3414   case ISD::ROTL:
3415   case ISD::ROTR:
3416     assert(VT == N1.getValueType() &&
3417            "Shift operators return type must be the same as their first arg");
3418     assert(VT.isInteger() && N2.getValueType().isInteger() &&
3419            "Shifts only work on integers");
3420     assert((!VT.isVector() || VT == N2.getValueType()) &&
3421            "Vector shift amounts must be in the same as their first arg");
3422     // Verify that the shift amount VT is bit enough to hold valid shift
3423     // amounts.  This catches things like trying to shift an i1024 value by an
3424     // i8, which is easy to fall into in generic code that uses
3425     // TLI.getShiftAmount().
3426     assert(N2.getValueType().getSizeInBits() >=
3427                    Log2_32_Ceil(N1.getValueType().getSizeInBits()) &&
3428            "Invalid use of small shift amount with oversized value!");
3429
3430     // Always fold shifts of i1 values so the code generator doesn't need to
3431     // handle them.  Since we know the size of the shift has to be less than the
3432     // size of the value, the shift/rotate count is guaranteed to be zero.
3433     if (VT == MVT::i1)
3434       return N1;
3435     if (N2C && N2C->isNullValue())
3436       return N1;
3437     break;
3438   case ISD::FP_ROUND_INREG: {
3439     EVT EVT = cast<VTSDNode>(N2)->getVT();
3440     assert(VT == N1.getValueType() && "Not an inreg round!");
3441     assert(VT.isFloatingPoint() && EVT.isFloatingPoint() &&
3442            "Cannot FP_ROUND_INREG integer types");
3443     assert(EVT.isVector() == VT.isVector() &&
3444            "FP_ROUND_INREG type should be vector iff the operand "
3445            "type is vector!");
3446     assert((!EVT.isVector() ||
3447             EVT.getVectorNumElements() == VT.getVectorNumElements()) &&
3448            "Vector element counts must match in FP_ROUND_INREG");
3449     assert(EVT.bitsLE(VT) && "Not rounding down!");
3450     (void)EVT;
3451     if (cast<VTSDNode>(N2)->getVT() == VT) return N1;  // Not actually rounding.
3452     break;
3453   }
3454   case ISD::FP_ROUND:
3455     assert(VT.isFloatingPoint() &&
3456            N1.getValueType().isFloatingPoint() &&
3457            VT.bitsLE(N1.getValueType()) &&
3458            isa<ConstantSDNode>(N2) && "Invalid FP_ROUND!");
3459     if (N1.getValueType() == VT) return N1;  // noop conversion.
3460     break;
3461   case ISD::AssertSext:
3462   case ISD::AssertZext: {
3463     EVT EVT = cast<VTSDNode>(N2)->getVT();
3464     assert(VT == N1.getValueType() && "Not an inreg extend!");
3465     assert(VT.isInteger() && EVT.isInteger() &&
3466            "Cannot *_EXTEND_INREG FP types");
3467     assert(!EVT.isVector() &&
3468            "AssertSExt/AssertZExt type should be the vector element type "
3469            "rather than the vector type!");
3470     assert(EVT.bitsLE(VT) && "Not extending!");
3471     if (VT == EVT) return N1; // noop assertion.
3472     break;
3473   }
3474   case ISD::SIGN_EXTEND_INREG: {
3475     EVT EVT = cast<VTSDNode>(N2)->getVT();
3476     assert(VT == N1.getValueType() && "Not an inreg extend!");
3477     assert(VT.isInteger() && EVT.isInteger() &&
3478            "Cannot *_EXTEND_INREG FP types");
3479     assert(EVT.isVector() == VT.isVector() &&
3480            "SIGN_EXTEND_INREG type should be vector iff the operand "
3481            "type is vector!");
3482     assert((!EVT.isVector() ||
3483             EVT.getVectorNumElements() == VT.getVectorNumElements()) &&
3484            "Vector element counts must match in SIGN_EXTEND_INREG");
3485     assert(EVT.bitsLE(VT) && "Not extending!");
3486     if (EVT == VT) return N1;  // Not actually extending
3487
3488     auto SignExtendInReg = [&](APInt Val) {
3489       unsigned FromBits = EVT.getScalarType().getSizeInBits();
3490       Val <<= Val.getBitWidth() - FromBits;
3491       Val = Val.ashr(Val.getBitWidth() - FromBits);
3492       return getConstant(Val, DL, VT.getScalarType());
3493     };
3494
3495     if (N1C) {
3496       APInt Val = N1C->getAPIntValue();
3497       return SignExtendInReg(Val);
3498     }
3499     if (ISD::isBuildVectorOfConstantSDNodes(N1.getNode())) {
3500       SmallVector<SDValue, 8> Ops;
3501       for (int i = 0, e = VT.getVectorNumElements(); i != e; ++i) {
3502         SDValue Op = N1.getOperand(i);
3503         if (Op.getValueType() != VT.getScalarType()) break;
3504         if (Op.getOpcode() == ISD::UNDEF) {
3505           Ops.push_back(Op);
3506           continue;
3507         }
3508         if (ConstantSDNode *C = dyn_cast<ConstantSDNode>(Op)) {
3509           APInt Val = C->getAPIntValue();
3510           Ops.push_back(SignExtendInReg(Val));
3511           continue;
3512         }
3513         break;
3514       }
3515       if (Ops.size() == VT.getVectorNumElements())
3516         return getNode(ISD::BUILD_VECTOR, DL, VT, Ops);
3517     }
3518     break;
3519   }
3520   case ISD::EXTRACT_VECTOR_ELT:
3521     // EXTRACT_VECTOR_ELT of an UNDEF is an UNDEF.
3522     if (N1.getOpcode() == ISD::UNDEF)
3523       return getUNDEF(VT);
3524
3525     // EXTRACT_VECTOR_ELT of out-of-bounds element is an UNDEF
3526     if (N2C && N2C->getZExtValue() >= N1.getValueType().getVectorNumElements())
3527       return getUNDEF(VT);
3528
3529     // EXTRACT_VECTOR_ELT of CONCAT_VECTORS is often formed while lowering is
3530     // expanding copies of large vectors from registers.
3531     if (N2C &&
3532         N1.getOpcode() == ISD::CONCAT_VECTORS &&
3533         N1.getNumOperands() > 0) {
3534       unsigned Factor =
3535         N1.getOperand(0).getValueType().getVectorNumElements();
3536       return getNode(ISD::EXTRACT_VECTOR_ELT, DL, VT,
3537                      N1.getOperand(N2C->getZExtValue() / Factor),
3538                      getConstant(N2C->getZExtValue() % Factor, DL,
3539                                  N2.getValueType()));
3540     }
3541
3542     // EXTRACT_VECTOR_ELT of BUILD_VECTOR is often formed while lowering is
3543     // expanding large vector constants.
3544     if (N2C && N1.getOpcode() == ISD::BUILD_VECTOR) {
3545       SDValue Elt = N1.getOperand(N2C->getZExtValue());
3546
3547       if (VT != Elt.getValueType())
3548         // If the vector element type is not legal, the BUILD_VECTOR operands
3549         // are promoted and implicitly truncated, and the result implicitly
3550         // extended. Make that explicit here.
3551         Elt = getAnyExtOrTrunc(Elt, DL, VT);
3552
3553       return Elt;
3554     }
3555
3556     // EXTRACT_VECTOR_ELT of INSERT_VECTOR_ELT is often formed when vector
3557     // operations are lowered to scalars.
3558     if (N1.getOpcode() == ISD::INSERT_VECTOR_ELT) {
3559       // If the indices are the same, return the inserted element else
3560       // if the indices are known different, extract the element from
3561       // the original vector.
3562       SDValue N1Op2 = N1.getOperand(2);
3563       ConstantSDNode *N1Op2C = dyn_cast<ConstantSDNode>(N1Op2);
3564
3565       if (N1Op2C && N2C) {
3566         if (N1Op2C->getZExtValue() == N2C->getZExtValue()) {
3567           if (VT == N1.getOperand(1).getValueType())
3568             return N1.getOperand(1);
3569           else
3570             return getSExtOrTrunc(N1.getOperand(1), DL, VT);
3571         }
3572
3573         return getNode(ISD::EXTRACT_VECTOR_ELT, DL, VT, N1.getOperand(0), N2);
3574       }
3575     }
3576     break;
3577   case ISD::EXTRACT_ELEMENT:
3578     assert(N2C && (unsigned)N2C->getZExtValue() < 2 && "Bad EXTRACT_ELEMENT!");
3579     assert(!N1.getValueType().isVector() && !VT.isVector() &&
3580            (N1.getValueType().isInteger() == VT.isInteger()) &&
3581            N1.getValueType() != VT &&
3582            "Wrong types for EXTRACT_ELEMENT!");
3583
3584     // EXTRACT_ELEMENT of BUILD_PAIR is often formed while legalize is expanding
3585     // 64-bit integers into 32-bit parts.  Instead of building the extract of
3586     // the BUILD_PAIR, only to have legalize rip it apart, just do it now.
3587     if (N1.getOpcode() == ISD::BUILD_PAIR)
3588       return N1.getOperand(N2C->getZExtValue());
3589
3590     // EXTRACT_ELEMENT of a constant int is also very common.
3591     if (ConstantSDNode *C = dyn_cast<ConstantSDNode>(N1)) {
3592       unsigned ElementSize = VT.getSizeInBits();
3593       unsigned Shift = ElementSize * N2C->getZExtValue();
3594       APInt ShiftedVal = C->getAPIntValue().lshr(Shift);
3595       return getConstant(ShiftedVal.trunc(ElementSize), DL, VT);
3596     }
3597     break;
3598   case ISD::EXTRACT_SUBVECTOR: {
3599     SDValue Index = N2;
3600     if (VT.isSimple() && N1.getValueType().isSimple()) {
3601       assert(VT.isVector() && N1.getValueType().isVector() &&
3602              "Extract subvector VTs must be a vectors!");
3603       assert(VT.getVectorElementType() ==
3604              N1.getValueType().getVectorElementType() &&
3605              "Extract subvector VTs must have the same element type!");
3606       assert(VT.getSimpleVT() <= N1.getSimpleValueType() &&
3607              "Extract subvector must be from larger vector to smaller vector!");
3608
3609       if (isa<ConstantSDNode>(Index)) {
3610         assert((VT.getVectorNumElements() +
3611                 cast<ConstantSDNode>(Index)->getZExtValue()
3612                 <= N1.getValueType().getVectorNumElements())
3613                && "Extract subvector overflow!");
3614       }
3615
3616       // Trivial extraction.
3617       if (VT.getSimpleVT() == N1.getSimpleValueType())
3618         return N1;
3619     }
3620     break;
3621   }
3622   }
3623
3624   // Perform trivial constant folding.
3625   if (SDValue SV =
3626           FoldConstantArithmetic(Opcode, DL, VT, N1.getNode(), N2.getNode()))
3627     return SV;
3628
3629   // Canonicalize constant to RHS if commutative.
3630   if (N1C && !N2C && isCommutativeBinOp(Opcode)) {
3631     std::swap(N1C, N2C);
3632     std::swap(N1, N2);
3633   }
3634
3635   // Constant fold FP operations.
3636   bool HasFPExceptions = TLI->hasFloatingPointExceptions();
3637   ConstantFPSDNode *N1CFP = dyn_cast<ConstantFPSDNode>(N1);
3638   ConstantFPSDNode *N2CFP = dyn_cast<ConstantFPSDNode>(N2);
3639   if (N1CFP) {
3640     if (!N2CFP && isCommutativeBinOp(Opcode)) {
3641       // Canonicalize constant to RHS if commutative.
3642       std::swap(N1CFP, N2CFP);
3643       std::swap(N1, N2);
3644     } else if (N2CFP) {
3645       APFloat V1 = N1CFP->getValueAPF(), V2 = N2CFP->getValueAPF();
3646       APFloat::opStatus s;
3647       switch (Opcode) {
3648       case ISD::FADD:
3649         s = V1.add(V2, APFloat::rmNearestTiesToEven);
3650         if (!HasFPExceptions || s != APFloat::opInvalidOp)
3651           return getConstantFP(V1, DL, VT);
3652         break;
3653       case ISD::FSUB:
3654         s = V1.subtract(V2, APFloat::rmNearestTiesToEven);
3655         if (!HasFPExceptions || s!=APFloat::opInvalidOp)
3656           return getConstantFP(V1, DL, VT);
3657         break;
3658       case ISD::FMUL:
3659         s = V1.multiply(V2, APFloat::rmNearestTiesToEven);
3660         if (!HasFPExceptions || s!=APFloat::opInvalidOp)
3661           return getConstantFP(V1, DL, VT);
3662         break;
3663       case ISD::FDIV:
3664         s = V1.divide(V2, APFloat::rmNearestTiesToEven);
3665         if (!HasFPExceptions || (s!=APFloat::opInvalidOp &&
3666                                  s!=APFloat::opDivByZero)) {
3667           return getConstantFP(V1, DL, VT);
3668         }
3669         break;
3670       case ISD::FREM :
3671         s = V1.mod(V2, APFloat::rmNearestTiesToEven);
3672         if (!HasFPExceptions || (s!=APFloat::opInvalidOp &&
3673                                  s!=APFloat::opDivByZero)) {
3674           return getConstantFP(V1, DL, VT);
3675         }
3676         break;
3677       case ISD::FCOPYSIGN:
3678         V1.copySign(V2);
3679         return getConstantFP(V1, DL, VT);
3680       default: break;
3681       }
3682     }
3683
3684     if (Opcode == ISD::FP_ROUND) {
3685       APFloat V = N1CFP->getValueAPF();    // make copy
3686       bool ignored;
3687       // This can return overflow, underflow, or inexact; we don't care.
3688       // FIXME need to be more flexible about rounding mode.
3689       (void)V.convert(EVTToAPFloatSemantics(VT),
3690                       APFloat::rmNearestTiesToEven, &ignored);
3691       return getConstantFP(V, DL, VT);
3692     }
3693   }
3694
3695   // Canonicalize an UNDEF to the RHS, even over a constant.
3696   if (N1.getOpcode() == ISD::UNDEF) {
3697     if (isCommutativeBinOp(Opcode)) {
3698       std::swap(N1, N2);
3699     } else {
3700       switch (Opcode) {
3701       case ISD::FP_ROUND_INREG:
3702       case ISD::SIGN_EXTEND_INREG:
3703       case ISD::SUB:
3704       case ISD::FSUB:
3705       case ISD::FDIV:
3706       case ISD::FREM:
3707       case ISD::SRA:
3708         return N1;     // fold op(undef, arg2) -> undef
3709       case ISD::UDIV:
3710       case ISD::SDIV:
3711       case ISD::UREM:
3712       case ISD::SREM:
3713       case ISD::SRL:
3714       case ISD::SHL:
3715         if (!VT.isVector())
3716           return getConstant(0, DL, VT);    // fold op(undef, arg2) -> 0
3717         // For vectors, we can't easily build an all zero vector, just return
3718         // the LHS.
3719         return N2;
3720       }
3721     }
3722   }
3723
3724   // Fold a bunch of operators when the RHS is undef.
3725   if (N2.getOpcode() == ISD::UNDEF) {
3726     switch (Opcode) {
3727     case ISD::XOR:
3728       if (N1.getOpcode() == ISD::UNDEF)
3729         // Handle undef ^ undef -> 0 special case. This is a common
3730         // idiom (misuse).
3731         return getConstant(0, DL, VT);
3732       // fallthrough
3733     case ISD::ADD:
3734     case ISD::ADDC:
3735     case ISD::ADDE:
3736     case ISD::SUB:
3737     case ISD::UDIV:
3738     case ISD::SDIV:
3739     case ISD::UREM:
3740     case ISD::SREM:
3741       return N2;       // fold op(arg1, undef) -> undef
3742     case ISD::FADD:
3743     case ISD::FSUB:
3744     case ISD::FMUL:
3745     case ISD::FDIV:
3746     case ISD::FREM:
3747       if (getTarget().Options.UnsafeFPMath)
3748         return N2;
3749       break;
3750     case ISD::MUL:
3751     case ISD::AND:
3752     case ISD::SRL:
3753     case ISD::SHL:
3754       if (!VT.isVector())
3755         return getConstant(0, DL, VT);  // fold op(arg1, undef) -> 0
3756       // For vectors, we can't easily build an all zero vector, just return
3757       // the LHS.
3758       return N1;
3759     case ISD::OR:
3760       if (!VT.isVector())
3761         return getConstant(APInt::getAllOnesValue(VT.getSizeInBits()), DL, VT);
3762       // For vectors, we can't easily build an all one vector, just return
3763       // the LHS.
3764       return N1;
3765     case ISD::SRA:
3766       return N1;
3767     }
3768   }
3769
3770   // Memoize this node if possible.
3771   BinarySDNode *N;
3772   SDVTList VTs = getVTList(VT);
3773   if (VT != MVT::Glue) {
3774     SDValue Ops[] = {N1, N2};
3775     FoldingSetNodeID ID;
3776     AddNodeIDNode(ID, Opcode, VTs, Ops);
3777     AddNodeIDFlags(ID, Opcode, Flags);
3778     void *IP = nullptr;
3779     if (SDNode *E = FindNodeOrInsertPos(ID, DL.getDebugLoc(), IP))
3780       return SDValue(E, 0);
3781
3782     N = GetBinarySDNode(Opcode, DL, VTs, N1, N2, Flags);
3783
3784     CSEMap.InsertNode(N, IP);
3785   } else {
3786     N = GetBinarySDNode(Opcode, DL, VTs, N1, N2, Flags);
3787   }
3788
3789   InsertNode(N);
3790   return SDValue(N, 0);
3791 }
3792
3793 SDValue SelectionDAG::getNode(unsigned Opcode, SDLoc DL, EVT VT,
3794                               SDValue N1, SDValue N2, SDValue N3) {
3795   // Perform various simplifications.
3796   ConstantSDNode *N1C = dyn_cast<ConstantSDNode>(N1);
3797   switch (Opcode) {
3798   case ISD::FMA: {
3799     ConstantFPSDNode *N1CFP = dyn_cast<ConstantFPSDNode>(N1);
3800     ConstantFPSDNode *N2CFP = dyn_cast<ConstantFPSDNode>(N2);
3801     ConstantFPSDNode *N3CFP = dyn_cast<ConstantFPSDNode>(N3);
3802     if (N1CFP && N2CFP && N3CFP) {
3803       APFloat  V1 = N1CFP->getValueAPF();
3804       const APFloat &V2 = N2CFP->getValueAPF();
3805       const APFloat &V3 = N3CFP->getValueAPF();
3806       APFloat::opStatus s =
3807         V1.fusedMultiplyAdd(V2, V3, APFloat::rmNearestTiesToEven);
3808       if (!TLI->hasFloatingPointExceptions() || s != APFloat::opInvalidOp)
3809         return getConstantFP(V1, DL, VT);
3810     }
3811     break;
3812   }
3813   case ISD::CONCAT_VECTORS:
3814     // A CONCAT_VECTOR with all operands BUILD_VECTOR can be simplified to
3815     // one big BUILD_VECTOR.
3816     if (N1.getOpcode() == ISD::BUILD_VECTOR &&
3817         N2.getOpcode() == ISD::BUILD_VECTOR &&
3818         N3.getOpcode() == ISD::BUILD_VECTOR) {
3819       SmallVector<SDValue, 16> Elts(N1.getNode()->op_begin(),
3820                                     N1.getNode()->op_end());
3821       Elts.append(N2.getNode()->op_begin(), N2.getNode()->op_end());
3822       Elts.append(N3.getNode()->op_begin(), N3.getNode()->op_end());
3823       return getNode(ISD::BUILD_VECTOR, DL, VT, Elts);
3824     }
3825     break;
3826   case ISD::SETCC: {
3827     // Use FoldSetCC to simplify SETCC's.
3828     SDValue Simp = FoldSetCC(VT, N1, N2, cast<CondCodeSDNode>(N3)->get(), DL);
3829     if (Simp.getNode()) return Simp;
3830     break;
3831   }
3832   case ISD::SELECT:
3833     if (N1C) {
3834      if (N1C->getZExtValue())
3835        return N2;             // select true, X, Y -> X
3836      return N3;             // select false, X, Y -> Y
3837     }
3838
3839     if (N2 == N3) return N2;   // select C, X, X -> X
3840     break;
3841   case ISD::VECTOR_SHUFFLE:
3842     llvm_unreachable("should use getVectorShuffle constructor!");
3843   case ISD::INSERT_SUBVECTOR: {
3844     SDValue Index = N3;
3845     if (VT.isSimple() && N1.getValueType().isSimple()
3846         && N2.getValueType().isSimple()) {
3847       assert(VT.isVector() && N1.getValueType().isVector() &&
3848              N2.getValueType().isVector() &&
3849              "Insert subvector VTs must be a vectors");
3850       assert(VT == N1.getValueType() &&
3851              "Dest and insert subvector source types must match!");
3852       assert(N2.getSimpleValueType() <= N1.getSimpleValueType() &&
3853              "Insert subvector must be from smaller vector to larger vector!");
3854       if (isa<ConstantSDNode>(Index)) {
3855         assert((N2.getValueType().getVectorNumElements() +
3856                 cast<ConstantSDNode>(Index)->getZExtValue()
3857                 <= VT.getVectorNumElements())
3858                && "Insert subvector overflow!");
3859       }
3860
3861       // Trivial insertion.
3862       if (VT.getSimpleVT() == N2.getSimpleValueType())
3863         return N2;
3864     }
3865     break;
3866   }
3867   case ISD::BITCAST:
3868     // Fold bit_convert nodes from a type to themselves.
3869     if (N1.getValueType() == VT)
3870       return N1;
3871     break;
3872   }
3873
3874   // Memoize node if it doesn't produce a flag.
3875   SDNode *N;
3876   SDVTList VTs = getVTList(VT);
3877   if (VT != MVT::Glue) {
3878     SDValue Ops[] = { N1, N2, N3 };
3879     FoldingSetNodeID ID;
3880     AddNodeIDNode(ID, Opcode, VTs, Ops);
3881     void *IP = nullptr;
3882     if (SDNode *E = FindNodeOrInsertPos(ID, DL.getDebugLoc(), IP))
3883       return SDValue(E, 0);
3884
3885     N = new (NodeAllocator) TernarySDNode(Opcode, DL.getIROrder(),
3886                                           DL.getDebugLoc(), VTs, N1, N2, N3);
3887     CSEMap.InsertNode(N, IP);
3888   } else {
3889     N = new (NodeAllocator) TernarySDNode(Opcode, DL.getIROrder(),
3890                                           DL.getDebugLoc(), VTs, N1, N2, N3);
3891   }
3892
3893   InsertNode(N);
3894   return SDValue(N, 0);
3895 }
3896
3897 SDValue SelectionDAG::getNode(unsigned Opcode, SDLoc DL, EVT VT,
3898                               SDValue N1, SDValue N2, SDValue N3,
3899                               SDValue N4) {
3900   SDValue Ops[] = { N1, N2, N3, N4 };
3901   return getNode(Opcode, DL, VT, Ops);
3902 }
3903
3904 SDValue SelectionDAG::getNode(unsigned Opcode, SDLoc DL, EVT VT,
3905                               SDValue N1, SDValue N2, SDValue N3,
3906                               SDValue N4, SDValue N5) {
3907   SDValue Ops[] = { N1, N2, N3, N4, N5 };
3908   return getNode(Opcode, DL, VT, Ops);
3909 }
3910
3911 /// getStackArgumentTokenFactor - Compute a TokenFactor to force all
3912 /// the incoming stack arguments to be loaded from the stack.
3913 SDValue SelectionDAG::getStackArgumentTokenFactor(SDValue Chain) {
3914   SmallVector<SDValue, 8> ArgChains;
3915
3916   // Include the original chain at the beginning of the list. When this is
3917   // used by target LowerCall hooks, this helps legalize find the
3918   // CALLSEQ_BEGIN node.
3919   ArgChains.push_back(Chain);
3920
3921   // Add a chain value for each stack argument.
3922   for (SDNode::use_iterator U = getEntryNode().getNode()->use_begin(),
3923        UE = getEntryNode().getNode()->use_end(); U != UE; ++U)
3924     if (LoadSDNode *L = dyn_cast<LoadSDNode>(*U))
3925       if (FrameIndexSDNode *FI = dyn_cast<FrameIndexSDNode>(L->getBasePtr()))
3926         if (FI->getIndex() < 0)
3927           ArgChains.push_back(SDValue(L, 1));
3928
3929   // Build a tokenfactor for all the chains.
3930   return getNode(ISD::TokenFactor, SDLoc(Chain), MVT::Other, ArgChains);
3931 }
3932
3933 /// getMemsetValue - Vectorized representation of the memset value
3934 /// operand.
3935 static SDValue getMemsetValue(SDValue Value, EVT VT, SelectionDAG &DAG,
3936                               SDLoc dl) {
3937   assert(Value.getOpcode() != ISD::UNDEF);
3938
3939   unsigned NumBits = VT.getScalarType().getSizeInBits();
3940   if (ConstantSDNode *C = dyn_cast<ConstantSDNode>(Value)) {
3941     assert(C->getAPIntValue().getBitWidth() == 8);
3942     APInt Val = APInt::getSplat(NumBits, C->getAPIntValue());
3943     if (VT.isInteger())
3944       return DAG.getConstant(Val, dl, VT);
3945     return DAG.getConstantFP(APFloat(DAG.EVTToAPFloatSemantics(VT), Val), dl,
3946                              VT);
3947   }
3948
3949   assert(Value.getValueType() == MVT::i8 && "memset with non-byte fill value?");
3950   EVT IntVT = VT.getScalarType();
3951   if (!IntVT.isInteger())
3952     IntVT = EVT::getIntegerVT(*DAG.getContext(), IntVT.getSizeInBits());
3953
3954   Value = DAG.getNode(ISD::ZERO_EXTEND, dl, IntVT, Value);
3955   if (NumBits > 8) {
3956     // Use a multiplication with 0x010101... to extend the input to the
3957     // required length.
3958     APInt Magic = APInt::getSplat(NumBits, APInt(8, 0x01));
3959     Value = DAG.getNode(ISD::MUL, dl, IntVT, Value,
3960                         DAG.getConstant(Magic, dl, IntVT));
3961   }
3962
3963   if (VT != Value.getValueType() && !VT.isInteger())
3964     Value = DAG.getNode(ISD::BITCAST, dl, VT.getScalarType(), Value);
3965   if (VT != Value.getValueType()) {
3966     assert(VT.getVectorElementType() == Value.getValueType() &&
3967            "value type should be one vector element here");
3968     SmallVector<SDValue, 8> BVOps(VT.getVectorNumElements(), Value);
3969     Value = DAG.getNode(ISD::BUILD_VECTOR, dl, VT, BVOps);
3970   }
3971
3972   return Value;
3973 }
3974
3975 /// getMemsetStringVal - Similar to getMemsetValue. Except this is only
3976 /// used when a memcpy is turned into a memset when the source is a constant
3977 /// string ptr.
3978 static SDValue getMemsetStringVal(EVT VT, SDLoc dl, SelectionDAG &DAG,
3979                                   const TargetLowering &TLI, StringRef Str) {
3980   // Handle vector with all elements zero.
3981   if (Str.empty()) {
3982     if (VT.isInteger())
3983       return DAG.getConstant(0, dl, VT);
3984     else if (VT == MVT::f32 || VT == MVT::f64 || VT == MVT::f128)
3985       return DAG.getConstantFP(0.0, dl, VT);
3986     else if (VT.isVector()) {
3987       unsigned NumElts = VT.getVectorNumElements();
3988       MVT EltVT = (VT.getVectorElementType() == MVT::f32) ? MVT::i32 : MVT::i64;
3989       return DAG.getNode(ISD::BITCAST, dl, VT,
3990                          DAG.getConstant(0, dl,
3991                                          EVT::getVectorVT(*DAG.getContext(),
3992                                                           EltVT, NumElts)));
3993     } else
3994       llvm_unreachable("Expected type!");
3995   }
3996
3997   assert(!VT.isVector() && "Can't handle vector type here!");
3998   unsigned NumVTBits = VT.getSizeInBits();
3999   unsigned NumVTBytes = NumVTBits / 8;
4000   unsigned NumBytes = std::min(NumVTBytes, unsigned(Str.size()));
4001
4002   APInt Val(NumVTBits, 0);
4003   if (DAG.getDataLayout().isLittleEndian()) {
4004     for (unsigned i = 0; i != NumBytes; ++i)
4005       Val |= (uint64_t)(unsigned char)Str[i] << i*8;
4006   } else {
4007     for (unsigned i = 0; i != NumBytes; ++i)
4008       Val |= (uint64_t)(unsigned char)Str[i] << (NumVTBytes-i-1)*8;
4009   }
4010
4011   // If the "cost" of materializing the integer immediate is less than the cost
4012   // of a load, then it is cost effective to turn the load into the immediate.
4013   Type *Ty = VT.getTypeForEVT(*DAG.getContext());
4014   if (TLI.shouldConvertConstantLoadToIntImm(Val, Ty))
4015     return DAG.getConstant(Val, dl, VT);
4016   return SDValue(nullptr, 0);
4017 }
4018
4019 /// getMemBasePlusOffset - Returns base and offset node for the
4020 ///
4021 static SDValue getMemBasePlusOffset(SDValue Base, unsigned Offset, SDLoc dl,
4022                                       SelectionDAG &DAG) {
4023   EVT VT = Base.getValueType();
4024   return DAG.getNode(ISD::ADD, dl,
4025                      VT, Base, DAG.getConstant(Offset, dl, VT));
4026 }
4027
4028 /// isMemSrcFromString - Returns true if memcpy source is a string constant.
4029 ///
4030 static bool isMemSrcFromString(SDValue Src, StringRef &Str) {
4031   unsigned SrcDelta = 0;
4032   GlobalAddressSDNode *G = nullptr;
4033   if (Src.getOpcode() == ISD::GlobalAddress)
4034     G = cast<GlobalAddressSDNode>(Src);
4035   else if (Src.getOpcode() == ISD::ADD &&
4036            Src.getOperand(0).getOpcode() == ISD::GlobalAddress &&
4037            Src.getOperand(1).getOpcode() == ISD::Constant) {
4038     G = cast<GlobalAddressSDNode>(Src.getOperand(0));
4039     SrcDelta = cast<ConstantSDNode>(Src.getOperand(1))->getZExtValue();
4040   }
4041   if (!G)
4042     return false;
4043
4044   return getConstantStringInfo(G->getGlobal(), Str, SrcDelta, false);
4045 }
4046
4047 /// Determines the optimal series of memory ops to replace the memset / memcpy.
4048 /// Return true if the number of memory ops is below the threshold (Limit).
4049 /// It returns the types of the sequence of memory ops to perform
4050 /// memset / memcpy by reference.
4051 static bool FindOptimalMemOpLowering(std::vector<EVT> &MemOps,
4052                                      unsigned Limit, uint64_t Size,
4053                                      unsigned DstAlign, unsigned SrcAlign,
4054                                      bool IsMemset,
4055                                      bool ZeroMemset,
4056                                      bool MemcpyStrSrc,
4057                                      bool AllowOverlap,
4058                                      SelectionDAG &DAG,
4059                                      const TargetLowering &TLI) {
4060   assert((SrcAlign == 0 || SrcAlign >= DstAlign) &&
4061          "Expecting memcpy / memset source to meet alignment requirement!");
4062   // If 'SrcAlign' is zero, that means the memory operation does not need to
4063   // load the value, i.e. memset or memcpy from constant string. Otherwise,
4064   // it's the inferred alignment of the source. 'DstAlign', on the other hand,
4065   // is the specified alignment of the memory operation. If it is zero, that
4066   // means it's possible to change the alignment of the destination.
4067   // 'MemcpyStrSrc' indicates whether the memcpy source is constant so it does
4068   // not need to be loaded.
4069   EVT VT = TLI.getOptimalMemOpType(Size, DstAlign, SrcAlign,
4070                                    IsMemset, ZeroMemset, MemcpyStrSrc,
4071                                    DAG.getMachineFunction());
4072
4073   if (VT == MVT::Other) {
4074     unsigned AS = 0;
4075     if (DstAlign >= DAG.getDataLayout().getPointerPrefAlignment(AS) ||
4076         TLI.allowsMisalignedMemoryAccesses(VT, AS, DstAlign)) {
4077       VT = TLI.getPointerTy(DAG.getDataLayout());
4078     } else {
4079       switch (DstAlign & 7) {
4080       case 0:  VT = MVT::i64; break;
4081       case 4:  VT = MVT::i32; break;
4082       case 2:  VT = MVT::i16; break;
4083       default: VT = MVT::i8;  break;
4084       }
4085     }
4086
4087     MVT LVT = MVT::i64;
4088     while (!TLI.isTypeLegal(LVT))
4089       LVT = (MVT::SimpleValueType)(LVT.SimpleTy - 1);
4090     assert(LVT.isInteger());
4091
4092     if (VT.bitsGT(LVT))
4093       VT = LVT;
4094   }
4095
4096   unsigned NumMemOps = 0;
4097   while (Size != 0) {
4098     unsigned VTSize = VT.getSizeInBits() / 8;
4099     while (VTSize > Size) {
4100       // For now, only use non-vector load / store's for the left-over pieces.
4101       EVT NewVT = VT;
4102       unsigned NewVTSize;
4103
4104       bool Found = false;
4105       if (VT.isVector() || VT.isFloatingPoint()) {
4106         NewVT = (VT.getSizeInBits() > 64) ? MVT::i64 : MVT::i32;
4107         if (TLI.isOperationLegalOrCustom(ISD::STORE, NewVT) &&
4108             TLI.isSafeMemOpType(NewVT.getSimpleVT()))
4109           Found = true;
4110         else if (NewVT == MVT::i64 &&
4111                  TLI.isOperationLegalOrCustom(ISD::STORE, MVT::f64) &&
4112                  TLI.isSafeMemOpType(MVT::f64)) {
4113           // i64 is usually not legal on 32-bit targets, but f64 may be.
4114           NewVT = MVT::f64;
4115           Found = true;
4116         }
4117       }
4118
4119       if (!Found) {
4120         do {
4121           NewVT = (MVT::SimpleValueType)(NewVT.getSimpleVT().SimpleTy - 1);
4122           if (NewVT == MVT::i8)
4123             break;
4124         } while (!TLI.isSafeMemOpType(NewVT.getSimpleVT()));
4125       }
4126       NewVTSize = NewVT.getSizeInBits() / 8;
4127
4128       // If the new VT cannot cover all of the remaining bits, then consider
4129       // issuing a (or a pair of) unaligned and overlapping load / store.
4130       // FIXME: Only does this for 64-bit or more since we don't have proper
4131       // cost model for unaligned load / store.
4132       bool Fast;
4133       unsigned AS = 0;
4134       if (NumMemOps && AllowOverlap &&
4135           VTSize >= 8 && NewVTSize < Size &&
4136           TLI.allowsMisalignedMemoryAccesses(VT, AS, DstAlign, &Fast) && Fast)
4137         VTSize = Size;
4138       else {
4139         VT = NewVT;
4140         VTSize = NewVTSize;
4141       }
4142     }
4143
4144     if (++NumMemOps > Limit)
4145       return false;
4146
4147     MemOps.push_back(VT);
4148     Size -= VTSize;
4149   }
4150
4151   return true;
4152 }
4153
4154 static SDValue getMemcpyLoadsAndStores(SelectionDAG &DAG, SDLoc dl,
4155                                        SDValue Chain, SDValue Dst,
4156                                        SDValue Src, uint64_t Size,
4157                                        unsigned Align, bool isVol,
4158                                        bool AlwaysInline,
4159                                        MachinePointerInfo DstPtrInfo,
4160                                        MachinePointerInfo SrcPtrInfo) {
4161   // Turn a memcpy of undef to nop.
4162   if (Src.getOpcode() == ISD::UNDEF)
4163     return Chain;
4164
4165   // Expand memcpy to a series of load and store ops if the size operand falls
4166   // below a certain threshold.
4167   // TODO: In the AlwaysInline case, if the size is big then generate a loop
4168   // rather than maybe a humongous number of loads and stores.
4169   const TargetLowering &TLI = DAG.getTargetLoweringInfo();
4170   std::vector<EVT> MemOps;
4171   bool DstAlignCanChange = false;
4172   MachineFunction &MF = DAG.getMachineFunction();
4173   MachineFrameInfo *MFI = MF.getFrameInfo();
4174   bool OptSize = MF.getFunction()->hasFnAttribute(Attribute::OptimizeForSize);
4175   FrameIndexSDNode *FI = dyn_cast<FrameIndexSDNode>(Dst);
4176   if (FI && !MFI->isFixedObjectIndex(FI->getIndex()))
4177     DstAlignCanChange = true;
4178   unsigned SrcAlign = DAG.InferPtrAlignment(Src);
4179   if (Align > SrcAlign)
4180     SrcAlign = Align;
4181   StringRef Str;
4182   bool CopyFromStr = isMemSrcFromString(Src, Str);
4183   bool isZeroStr = CopyFromStr && Str.empty();
4184   unsigned Limit = AlwaysInline ? ~0U : TLI.getMaxStoresPerMemcpy(OptSize);
4185
4186   if (!FindOptimalMemOpLowering(MemOps, Limit, Size,
4187                                 (DstAlignCanChange ? 0 : Align),
4188                                 (isZeroStr ? 0 : SrcAlign),
4189                                 false, false, CopyFromStr, true, DAG, TLI))
4190     return SDValue();
4191
4192   if (DstAlignCanChange) {
4193     Type *Ty = MemOps[0].getTypeForEVT(*DAG.getContext());
4194     unsigned NewAlign = (unsigned)DAG.getDataLayout().getABITypeAlignment(Ty);
4195
4196     // Don't promote to an alignment that would require dynamic stack
4197     // realignment.
4198     const TargetRegisterInfo *TRI = MF.getSubtarget().getRegisterInfo();
4199     if (!TRI->needsStackRealignment(MF))
4200       while (NewAlign > Align &&
4201              DAG.getDataLayout().exceedsNaturalStackAlignment(NewAlign))
4202           NewAlign /= 2;
4203
4204     if (NewAlign > Align) {
4205       // Give the stack frame object a larger alignment if needed.
4206       if (MFI->getObjectAlignment(FI->getIndex()) < NewAlign)
4207         MFI->setObjectAlignment(FI->getIndex(), NewAlign);
4208       Align = NewAlign;
4209     }
4210   }
4211
4212   SmallVector<SDValue, 8> OutChains;
4213   unsigned NumMemOps = MemOps.size();
4214   uint64_t SrcOff = 0, DstOff = 0;
4215   for (unsigned i = 0; i != NumMemOps; ++i) {
4216     EVT VT = MemOps[i];
4217     unsigned VTSize = VT.getSizeInBits() / 8;
4218     SDValue Value, Store;
4219
4220     if (VTSize > Size) {
4221       // Issuing an unaligned load / store pair  that overlaps with the previous
4222       // pair. Adjust the offset accordingly.
4223       assert(i == NumMemOps-1 && i != 0);
4224       SrcOff -= VTSize - Size;
4225       DstOff -= VTSize - Size;
4226     }
4227
4228     if (CopyFromStr &&
4229         (isZeroStr || (VT.isInteger() && !VT.isVector()))) {
4230       // It's unlikely a store of a vector immediate can be done in a single
4231       // instruction. It would require a load from a constantpool first.
4232       // We only handle zero vectors here.
4233       // FIXME: Handle other cases where store of vector immediate is done in
4234       // a single instruction.
4235       Value = getMemsetStringVal(VT, dl, DAG, TLI, Str.substr(SrcOff));
4236       if (Value.getNode())
4237         Store = DAG.getStore(Chain, dl, Value,
4238                              getMemBasePlusOffset(Dst, DstOff, dl, DAG),
4239                              DstPtrInfo.getWithOffset(DstOff), isVol,
4240                              false, Align);
4241     }
4242
4243     if (!Store.getNode()) {
4244       // The type might not be legal for the target.  This should only happen
4245       // if the type is smaller than a legal type, as on PPC, so the right
4246       // thing to do is generate a LoadExt/StoreTrunc pair.  These simplify
4247       // to Load/Store if NVT==VT.
4248       // FIXME does the case above also need this?
4249       EVT NVT = TLI.getTypeToTransformTo(*DAG.getContext(), VT);
4250       assert(NVT.bitsGE(VT));
4251       Value = DAG.getExtLoad(ISD::EXTLOAD, dl, NVT, Chain,
4252                              getMemBasePlusOffset(Src, SrcOff, dl, DAG),
4253                              SrcPtrInfo.getWithOffset(SrcOff), VT, isVol, false,
4254                              false, MinAlign(SrcAlign, SrcOff));
4255       Store = DAG.getTruncStore(Chain, dl, Value,
4256                                 getMemBasePlusOffset(Dst, DstOff, dl, DAG),
4257                                 DstPtrInfo.getWithOffset(DstOff), VT, isVol,
4258                                 false, Align);
4259     }
4260     OutChains.push_back(Store);
4261     SrcOff += VTSize;
4262     DstOff += VTSize;
4263     Size -= VTSize;
4264   }
4265
4266   return DAG.getNode(ISD::TokenFactor, dl, MVT::Other, OutChains);
4267 }
4268
4269 static SDValue getMemmoveLoadsAndStores(SelectionDAG &DAG, SDLoc dl,
4270                                         SDValue Chain, SDValue Dst,
4271                                         SDValue Src, uint64_t Size,
4272                                         unsigned Align,  bool isVol,
4273                                         bool AlwaysInline,
4274                                         MachinePointerInfo DstPtrInfo,
4275                                         MachinePointerInfo SrcPtrInfo) {
4276   // Turn a memmove of undef to nop.
4277   if (Src.getOpcode() == ISD::UNDEF)
4278     return Chain;
4279
4280   // Expand memmove to a series of load and store ops if the size operand falls
4281   // below a certain threshold.
4282   const TargetLowering &TLI = DAG.getTargetLoweringInfo();
4283   std::vector<EVT> MemOps;
4284   bool DstAlignCanChange = false;
4285   MachineFunction &MF = DAG.getMachineFunction();
4286   MachineFrameInfo *MFI = MF.getFrameInfo();
4287   bool OptSize = MF.getFunction()->hasFnAttribute(Attribute::OptimizeForSize);
4288   FrameIndexSDNode *FI = dyn_cast<FrameIndexSDNode>(Dst);
4289   if (FI && !MFI->isFixedObjectIndex(FI->getIndex()))
4290     DstAlignCanChange = true;
4291   unsigned SrcAlign = DAG.InferPtrAlignment(Src);
4292   if (Align > SrcAlign)
4293     SrcAlign = Align;
4294   unsigned Limit = AlwaysInline ? ~0U : TLI.getMaxStoresPerMemmove(OptSize);
4295
4296   if (!FindOptimalMemOpLowering(MemOps, Limit, Size,
4297                                 (DstAlignCanChange ? 0 : Align), SrcAlign,
4298                                 false, false, false, false, DAG, TLI))
4299     return SDValue();
4300
4301   if (DstAlignCanChange) {
4302     Type *Ty = MemOps[0].getTypeForEVT(*DAG.getContext());
4303     unsigned NewAlign = (unsigned)DAG.getDataLayout().getABITypeAlignment(Ty);
4304     if (NewAlign > Align) {
4305       // Give the stack frame object a larger alignment if needed.
4306       if (MFI->getObjectAlignment(FI->getIndex()) < NewAlign)
4307         MFI->setObjectAlignment(FI->getIndex(), NewAlign);
4308       Align = NewAlign;
4309     }
4310   }
4311
4312   uint64_t SrcOff = 0, DstOff = 0;
4313   SmallVector<SDValue, 8> LoadValues;
4314   SmallVector<SDValue, 8> LoadChains;
4315   SmallVector<SDValue, 8> OutChains;
4316   unsigned NumMemOps = MemOps.size();
4317   for (unsigned i = 0; i < NumMemOps; i++) {
4318     EVT VT = MemOps[i];
4319     unsigned VTSize = VT.getSizeInBits() / 8;
4320     SDValue Value;
4321
4322     Value = DAG.getLoad(VT, dl, Chain,
4323                         getMemBasePlusOffset(Src, SrcOff, dl, DAG),
4324                         SrcPtrInfo.getWithOffset(SrcOff), isVol,
4325                         false, false, SrcAlign);
4326     LoadValues.push_back(Value);
4327     LoadChains.push_back(Value.getValue(1));
4328     SrcOff += VTSize;
4329   }
4330   Chain = DAG.getNode(ISD::TokenFactor, dl, MVT::Other, LoadChains);
4331   OutChains.clear();
4332   for (unsigned i = 0; i < NumMemOps; i++) {
4333     EVT VT = MemOps[i];
4334     unsigned VTSize = VT.getSizeInBits() / 8;
4335     SDValue Store;
4336
4337     Store = DAG.getStore(Chain, dl, LoadValues[i],
4338                          getMemBasePlusOffset(Dst, DstOff, dl, DAG),
4339                          DstPtrInfo.getWithOffset(DstOff), isVol, false, Align);
4340     OutChains.push_back(Store);
4341     DstOff += VTSize;
4342   }
4343
4344   return DAG.getNode(ISD::TokenFactor, dl, MVT::Other, OutChains);
4345 }
4346
4347 /// \brief Lower the call to 'memset' intrinsic function into a series of store
4348 /// operations.
4349 ///
4350 /// \param DAG Selection DAG where lowered code is placed.
4351 /// \param dl Link to corresponding IR location.
4352 /// \param Chain Control flow dependency.
4353 /// \param Dst Pointer to destination memory location.
4354 /// \param Src Value of byte to write into the memory.
4355 /// \param Size Number of bytes to write.
4356 /// \param Align Alignment of the destination in bytes.
4357 /// \param isVol True if destination is volatile.
4358 /// \param DstPtrInfo IR information on the memory pointer.
4359 /// \returns New head in the control flow, if lowering was successful, empty
4360 /// SDValue otherwise.
4361 ///
4362 /// The function tries to replace 'llvm.memset' intrinsic with several store
4363 /// operations and value calculation code. This is usually profitable for small
4364 /// memory size.
4365 static SDValue getMemsetStores(SelectionDAG &DAG, SDLoc dl,
4366                                SDValue Chain, SDValue Dst,
4367                                SDValue Src, uint64_t Size,
4368                                unsigned Align, bool isVol,
4369                                MachinePointerInfo DstPtrInfo) {
4370   // Turn a memset of undef to nop.
4371   if (Src.getOpcode() == ISD::UNDEF)
4372     return Chain;
4373
4374   // Expand memset to a series of load/store ops if the size operand
4375   // falls below a certain threshold.
4376   const TargetLowering &TLI = DAG.getTargetLoweringInfo();
4377   std::vector<EVT> MemOps;
4378   bool DstAlignCanChange = false;
4379   MachineFunction &MF = DAG.getMachineFunction();
4380   MachineFrameInfo *MFI = MF.getFrameInfo();
4381   bool OptSize = MF.getFunction()->hasFnAttribute(Attribute::OptimizeForSize);
4382   FrameIndexSDNode *FI = dyn_cast<FrameIndexSDNode>(Dst);
4383   if (FI && !MFI->isFixedObjectIndex(FI->getIndex()))
4384     DstAlignCanChange = true;
4385   bool IsZeroVal =
4386     isa<ConstantSDNode>(Src) && cast<ConstantSDNode>(Src)->isNullValue();
4387   if (!FindOptimalMemOpLowering(MemOps, TLI.getMaxStoresPerMemset(OptSize),
4388                                 Size, (DstAlignCanChange ? 0 : Align), 0,
4389                                 true, IsZeroVal, false, true, DAG, TLI))
4390     return SDValue();
4391
4392   if (DstAlignCanChange) {
4393     Type *Ty = MemOps[0].getTypeForEVT(*DAG.getContext());
4394     unsigned NewAlign = (unsigned)DAG.getDataLayout().getABITypeAlignment(Ty);
4395     if (NewAlign > Align) {
4396       // Give the stack frame object a larger alignment if needed.
4397       if (MFI->getObjectAlignment(FI->getIndex()) < NewAlign)
4398         MFI->setObjectAlignment(FI->getIndex(), NewAlign);
4399       Align = NewAlign;
4400     }
4401   }
4402
4403   SmallVector<SDValue, 8> OutChains;
4404   uint64_t DstOff = 0;
4405   unsigned NumMemOps = MemOps.size();
4406
4407   // Find the largest store and generate the bit pattern for it.
4408   EVT LargestVT = MemOps[0];
4409   for (unsigned i = 1; i < NumMemOps; i++)
4410     if (MemOps[i].bitsGT(LargestVT))
4411       LargestVT = MemOps[i];
4412   SDValue MemSetValue = getMemsetValue(Src, LargestVT, DAG, dl);
4413
4414   for (unsigned i = 0; i < NumMemOps; i++) {
4415     EVT VT = MemOps[i];
4416     unsigned VTSize = VT.getSizeInBits() / 8;
4417     if (VTSize > Size) {
4418       // Issuing an unaligned load / store pair  that overlaps with the previous
4419       // pair. Adjust the offset accordingly.
4420       assert(i == NumMemOps-1 && i != 0);
4421       DstOff -= VTSize - Size;
4422     }
4423
4424     // If this store is smaller than the largest store see whether we can get
4425     // the smaller value for free with a truncate.
4426     SDValue Value = MemSetValue;
4427     if (VT.bitsLT(LargestVT)) {
4428       if (!LargestVT.isVector() && !VT.isVector() &&
4429           TLI.isTruncateFree(LargestVT, VT))
4430         Value = DAG.getNode(ISD::TRUNCATE, dl, VT, MemSetValue);
4431       else
4432         Value = getMemsetValue(Src, VT, DAG, dl);
4433     }
4434     assert(Value.getValueType() == VT && "Value with wrong type.");
4435     SDValue Store = DAG.getStore(Chain, dl, Value,
4436                                  getMemBasePlusOffset(Dst, DstOff, dl, DAG),
4437                                  DstPtrInfo.getWithOffset(DstOff),
4438                                  isVol, false, Align);
4439     OutChains.push_back(Store);
4440     DstOff += VT.getSizeInBits() / 8;
4441     Size -= VTSize;
4442   }
4443
4444   return DAG.getNode(ISD::TokenFactor, dl, MVT::Other, OutChains);
4445 }
4446
4447 SDValue SelectionDAG::getMemcpy(SDValue Chain, SDLoc dl, SDValue Dst,
4448                                 SDValue Src, SDValue Size,
4449                                 unsigned Align, bool isVol, bool AlwaysInline,
4450                                 bool isTailCall, MachinePointerInfo DstPtrInfo,
4451                                 MachinePointerInfo SrcPtrInfo) {
4452   assert(Align && "The SDAG layer expects explicit alignment and reserves 0");
4453
4454   // Check to see if we should lower the memcpy to loads and stores first.
4455   // For cases within the target-specified limits, this is the best choice.
4456   ConstantSDNode *ConstantSize = dyn_cast<ConstantSDNode>(Size);
4457   if (ConstantSize) {
4458     // Memcpy with size zero? Just return the original chain.
4459     if (ConstantSize->isNullValue())
4460       return Chain;
4461
4462     SDValue Result = getMemcpyLoadsAndStores(*this, dl, Chain, Dst, Src,
4463                                              ConstantSize->getZExtValue(),Align,
4464                                 isVol, false, DstPtrInfo, SrcPtrInfo);
4465     if (Result.getNode())
4466       return Result;
4467   }
4468
4469   // Then check to see if we should lower the memcpy with target-specific
4470   // code. If the target chooses to do this, this is the next best.
4471   if (TSI) {
4472     SDValue Result = TSI->EmitTargetCodeForMemcpy(
4473         *this, dl, Chain, Dst, Src, Size, Align, isVol, AlwaysInline,
4474         DstPtrInfo, SrcPtrInfo);
4475     if (Result.getNode())
4476       return Result;
4477   }
4478
4479   // If we really need inline code and the target declined to provide it,
4480   // use a (potentially long) sequence of loads and stores.
4481   if (AlwaysInline) {
4482     assert(ConstantSize && "AlwaysInline requires a constant size!");
4483     return getMemcpyLoadsAndStores(*this, dl, Chain, Dst, Src,
4484                                    ConstantSize->getZExtValue(), Align, isVol,
4485                                    true, DstPtrInfo, SrcPtrInfo);
4486   }
4487
4488   // FIXME: If the memcpy is volatile (isVol), lowering it to a plain libc
4489   // memcpy is not guaranteed to be safe. libc memcpys aren't required to
4490   // respect volatile, so they may do things like read or write memory
4491   // beyond the given memory regions. But fixing this isn't easy, and most
4492   // people don't care.
4493
4494   // Emit a library call.
4495   TargetLowering::ArgListTy Args;
4496   TargetLowering::ArgListEntry Entry;
4497   Entry.Ty = getDataLayout().getIntPtrType(*getContext());
4498   Entry.Node = Dst; Args.push_back(Entry);
4499   Entry.Node = Src; Args.push_back(Entry);
4500   Entry.Node = Size; Args.push_back(Entry);
4501   // FIXME: pass in SDLoc
4502   TargetLowering::CallLoweringInfo CLI(*this);
4503   CLI.setDebugLoc(dl)
4504       .setChain(Chain)
4505       .setCallee(TLI->getLibcallCallingConv(RTLIB::MEMCPY),
4506                  Type::getVoidTy(*getContext()),
4507                  getExternalSymbol(TLI->getLibcallName(RTLIB::MEMCPY),
4508                                    TLI->getPointerTy(getDataLayout())),
4509                  std::move(Args), 0)
4510       .setDiscardResult()
4511       .setTailCall(isTailCall);
4512
4513   std::pair<SDValue,SDValue> CallResult = TLI->LowerCallTo(CLI);
4514   return CallResult.second;
4515 }
4516
4517 SDValue SelectionDAG::getMemmove(SDValue Chain, SDLoc dl, SDValue Dst,
4518                                  SDValue Src, SDValue Size,
4519                                  unsigned Align, bool isVol, bool isTailCall,
4520                                  MachinePointerInfo DstPtrInfo,
4521                                  MachinePointerInfo SrcPtrInfo) {
4522   assert(Align && "The SDAG layer expects explicit alignment and reserves 0");
4523
4524   // Check to see if we should lower the memmove to loads and stores first.
4525   // For cases within the target-specified limits, this is the best choice.
4526   ConstantSDNode *ConstantSize = dyn_cast<ConstantSDNode>(Size);
4527   if (ConstantSize) {
4528     // Memmove with size zero? Just return the original chain.
4529     if (ConstantSize->isNullValue())
4530       return Chain;
4531
4532     SDValue Result =
4533       getMemmoveLoadsAndStores(*this, dl, Chain, Dst, Src,
4534                                ConstantSize->getZExtValue(), Align, isVol,
4535                                false, DstPtrInfo, SrcPtrInfo);
4536     if (Result.getNode())
4537       return Result;
4538   }
4539
4540   // Then check to see if we should lower the memmove with target-specific
4541   // code. If the target chooses to do this, this is the next best.
4542   if (TSI) {
4543     SDValue Result = TSI->EmitTargetCodeForMemmove(
4544         *this, dl, Chain, Dst, Src, Size, Align, isVol, DstPtrInfo, SrcPtrInfo);
4545     if (Result.getNode())
4546       return Result;
4547   }
4548
4549   // FIXME: If the memmove is volatile, lowering it to plain libc memmove may
4550   // not be safe.  See memcpy above for more details.
4551
4552   // Emit a library call.
4553   TargetLowering::ArgListTy Args;
4554   TargetLowering::ArgListEntry Entry;
4555   Entry.Ty = getDataLayout().getIntPtrType(*getContext());
4556   Entry.Node = Dst; Args.push_back(Entry);
4557   Entry.Node = Src; Args.push_back(Entry);
4558   Entry.Node = Size; Args.push_back(Entry);
4559   // FIXME:  pass in SDLoc
4560   TargetLowering::CallLoweringInfo CLI(*this);
4561   CLI.setDebugLoc(dl)
4562       .setChain(Chain)
4563       .setCallee(TLI->getLibcallCallingConv(RTLIB::MEMMOVE),
4564                  Type::getVoidTy(*getContext()),
4565                  getExternalSymbol(TLI->getLibcallName(RTLIB::MEMMOVE),
4566                                    TLI->getPointerTy(getDataLayout())),
4567                  std::move(Args), 0)
4568       .setDiscardResult()
4569       .setTailCall(isTailCall);
4570
4571   std::pair<SDValue,SDValue> CallResult = TLI->LowerCallTo(CLI);
4572   return CallResult.second;
4573 }
4574
4575 SDValue SelectionDAG::getMemset(SDValue Chain, SDLoc dl, SDValue Dst,
4576                                 SDValue Src, SDValue Size,
4577                                 unsigned Align, bool isVol, bool isTailCall,
4578                                 MachinePointerInfo DstPtrInfo) {
4579   assert(Align && "The SDAG layer expects explicit alignment and reserves 0");
4580
4581   // Check to see if we should lower the memset to stores first.
4582   // For cases within the target-specified limits, this is the best choice.
4583   ConstantSDNode *ConstantSize = dyn_cast<ConstantSDNode>(Size);
4584   if (ConstantSize) {
4585     // Memset with size zero? Just return the original chain.
4586     if (ConstantSize->isNullValue())
4587       return Chain;
4588
4589     SDValue Result =
4590       getMemsetStores(*this, dl, Chain, Dst, Src, ConstantSize->getZExtValue(),
4591                       Align, isVol, DstPtrInfo);
4592
4593     if (Result.getNode())
4594       return Result;
4595   }
4596
4597   // Then check to see if we should lower the memset with target-specific
4598   // code. If the target chooses to do this, this is the next best.
4599   if (TSI) {
4600     SDValue Result = TSI->EmitTargetCodeForMemset(
4601         *this, dl, Chain, Dst, Src, Size, Align, isVol, DstPtrInfo);
4602     if (Result.getNode())
4603       return Result;
4604   }
4605
4606   // Emit a library call.
4607   Type *IntPtrTy = getDataLayout().getIntPtrType(*getContext());
4608   TargetLowering::ArgListTy Args;
4609   TargetLowering::ArgListEntry Entry;
4610   Entry.Node = Dst; Entry.Ty = IntPtrTy;
4611   Args.push_back(Entry);
4612   Entry.Node = Src;
4613   Entry.Ty = Src.getValueType().getTypeForEVT(*getContext());
4614   Args.push_back(Entry);
4615   Entry.Node = Size;
4616   Entry.Ty = IntPtrTy;
4617   Args.push_back(Entry);
4618
4619   // FIXME: pass in SDLoc
4620   TargetLowering::CallLoweringInfo CLI(*this);
4621   CLI.setDebugLoc(dl)
4622       .setChain(Chain)
4623       .setCallee(TLI->getLibcallCallingConv(RTLIB::MEMSET),
4624                  Type::getVoidTy(*getContext()),
4625                  getExternalSymbol(TLI->getLibcallName(RTLIB::MEMSET),
4626                                    TLI->getPointerTy(getDataLayout())),
4627                  std::move(Args), 0)
4628       .setDiscardResult()
4629       .setTailCall(isTailCall);
4630
4631   std::pair<SDValue,SDValue> CallResult = TLI->LowerCallTo(CLI);
4632   return CallResult.second;
4633 }
4634
4635 SDValue SelectionDAG::getAtomic(unsigned Opcode, SDLoc dl, EVT MemVT,
4636                                 SDVTList VTList, ArrayRef<SDValue> Ops,
4637                                 MachineMemOperand *MMO,
4638                                 AtomicOrdering SuccessOrdering,
4639                                 AtomicOrdering FailureOrdering,
4640                                 SynchronizationScope SynchScope) {
4641   FoldingSetNodeID ID;
4642   ID.AddInteger(MemVT.getRawBits());
4643   AddNodeIDNode(ID, Opcode, VTList, Ops);
4644   ID.AddInteger(MMO->getPointerInfo().getAddrSpace());
4645   void* IP = nullptr;
4646   if (SDNode *E = FindNodeOrInsertPos(ID, dl.getDebugLoc(), IP)) {
4647     cast<AtomicSDNode>(E)->refineAlignment(MMO);
4648     return SDValue(E, 0);
4649   }
4650
4651   // Allocate the operands array for the node out of the BumpPtrAllocator, since
4652   // SDNode doesn't have access to it.  This memory will be "leaked" when
4653   // the node is deallocated, but recovered when the allocator is released.
4654   // If the number of operands is less than 5 we use AtomicSDNode's internal
4655   // storage.
4656   unsigned NumOps = Ops.size();
4657   SDUse *DynOps = NumOps > 4 ? OperandAllocator.Allocate<SDUse>(NumOps)
4658                              : nullptr;
4659
4660   SDNode *N = new (NodeAllocator) AtomicSDNode(Opcode, dl.getIROrder(),
4661                                                dl.getDebugLoc(), VTList, MemVT,
4662                                                Ops.data(), DynOps, NumOps, MMO,
4663                                                SuccessOrdering, FailureOrdering,
4664                                                SynchScope);
4665   CSEMap.InsertNode(N, IP);
4666   InsertNode(N);
4667   return SDValue(N, 0);
4668 }
4669
4670 SDValue SelectionDAG::getAtomic(unsigned Opcode, SDLoc dl, EVT MemVT,
4671                                 SDVTList VTList, ArrayRef<SDValue> Ops,
4672                                 MachineMemOperand *MMO,
4673                                 AtomicOrdering Ordering,
4674                                 SynchronizationScope SynchScope) {
4675   return getAtomic(Opcode, dl, MemVT, VTList, Ops, MMO, Ordering,
4676                    Ordering, SynchScope);
4677 }
4678
4679 SDValue SelectionDAG::getAtomicCmpSwap(
4680     unsigned Opcode, SDLoc dl, EVT MemVT, SDVTList VTs, SDValue Chain,
4681     SDValue Ptr, SDValue Cmp, SDValue Swp, MachinePointerInfo PtrInfo,
4682     unsigned Alignment, AtomicOrdering SuccessOrdering,
4683     AtomicOrdering FailureOrdering, SynchronizationScope SynchScope) {
4684   assert(Opcode == ISD::ATOMIC_CMP_SWAP ||
4685          Opcode == ISD::ATOMIC_CMP_SWAP_WITH_SUCCESS);
4686   assert(Cmp.getValueType() == Swp.getValueType() && "Invalid Atomic Op Types");
4687
4688   if (Alignment == 0)  // Ensure that codegen never sees alignment 0
4689     Alignment = getEVTAlignment(MemVT);
4690
4691   MachineFunction &MF = getMachineFunction();
4692
4693   // FIXME: Volatile isn't really correct; we should keep track of atomic
4694   // orderings in the memoperand.
4695   unsigned Flags = MachineMemOperand::MOVolatile;
4696   Flags |= MachineMemOperand::MOLoad;
4697   Flags |= MachineMemOperand::MOStore;
4698
4699   MachineMemOperand *MMO =
4700     MF.getMachineMemOperand(PtrInfo, Flags, MemVT.getStoreSize(), Alignment);
4701
4702   return getAtomicCmpSwap(Opcode, dl, MemVT, VTs, Chain, Ptr, Cmp, Swp, MMO,
4703                           SuccessOrdering, FailureOrdering, SynchScope);
4704 }
4705
4706 SDValue SelectionDAG::getAtomicCmpSwap(unsigned Opcode, SDLoc dl, EVT MemVT,
4707                                        SDVTList VTs, SDValue Chain, SDValue Ptr,
4708                                        SDValue Cmp, SDValue Swp,
4709                                        MachineMemOperand *MMO,
4710                                        AtomicOrdering SuccessOrdering,
4711                                        AtomicOrdering FailureOrdering,
4712                                        SynchronizationScope SynchScope) {
4713   assert(Opcode == ISD::ATOMIC_CMP_SWAP ||
4714          Opcode == ISD::ATOMIC_CMP_SWAP_WITH_SUCCESS);
4715   assert(Cmp.getValueType() == Swp.getValueType() && "Invalid Atomic Op Types");
4716
4717   SDValue Ops[] = {Chain, Ptr, Cmp, Swp};
4718   return getAtomic(Opcode, dl, MemVT, VTs, Ops, MMO,
4719                    SuccessOrdering, FailureOrdering, SynchScope);
4720 }
4721
4722 SDValue SelectionDAG::getAtomic(unsigned Opcode, SDLoc dl, EVT MemVT,
4723                                 SDValue Chain,
4724                                 SDValue Ptr, SDValue Val,
4725                                 const Value* PtrVal,
4726                                 unsigned Alignment,
4727                                 AtomicOrdering Ordering,
4728                                 SynchronizationScope SynchScope) {
4729   if (Alignment == 0)  // Ensure that codegen never sees alignment 0
4730     Alignment = getEVTAlignment(MemVT);
4731
4732   MachineFunction &MF = getMachineFunction();
4733   // An atomic store does not load. An atomic load does not store.
4734   // (An atomicrmw obviously both loads and stores.)
4735   // For now, atomics are considered to be volatile always, and they are
4736   // chained as such.
4737   // FIXME: Volatile isn't really correct; we should keep track of atomic
4738   // orderings in the memoperand.
4739   unsigned Flags = MachineMemOperand::MOVolatile;
4740   if (Opcode != ISD::ATOMIC_STORE)
4741     Flags |= MachineMemOperand::MOLoad;
4742   if (Opcode != ISD::ATOMIC_LOAD)
4743     Flags |= MachineMemOperand::MOStore;
4744
4745   MachineMemOperand *MMO =
4746     MF.getMachineMemOperand(MachinePointerInfo(PtrVal), Flags,
4747                             MemVT.getStoreSize(), Alignment);
4748
4749   return getAtomic(Opcode, dl, MemVT, Chain, Ptr, Val, MMO,
4750                    Ordering, SynchScope);
4751 }
4752
4753 SDValue SelectionDAG::getAtomic(unsigned Opcode, SDLoc dl, EVT MemVT,
4754                                 SDValue Chain,
4755                                 SDValue Ptr, SDValue Val,
4756                                 MachineMemOperand *MMO,
4757                                 AtomicOrdering Ordering,
4758                                 SynchronizationScope SynchScope) {
4759   assert((Opcode == ISD::ATOMIC_LOAD_ADD ||
4760           Opcode == ISD::ATOMIC_LOAD_SUB ||
4761           Opcode == ISD::ATOMIC_LOAD_AND ||
4762           Opcode == ISD::ATOMIC_LOAD_OR ||
4763           Opcode == ISD::ATOMIC_LOAD_XOR ||
4764           Opcode == ISD::ATOMIC_LOAD_NAND ||
4765           Opcode == ISD::ATOMIC_LOAD_MIN ||
4766           Opcode == ISD::ATOMIC_LOAD_MAX ||
4767           Opcode == ISD::ATOMIC_LOAD_UMIN ||
4768           Opcode == ISD::ATOMIC_LOAD_UMAX ||
4769           Opcode == ISD::ATOMIC_SWAP ||
4770           Opcode == ISD::ATOMIC_STORE) &&
4771          "Invalid Atomic Op");
4772
4773   EVT VT = Val.getValueType();
4774
4775   SDVTList VTs = Opcode == ISD::ATOMIC_STORE ? getVTList(MVT::Other) :
4776                                                getVTList(VT, MVT::Other);
4777   SDValue Ops[] = {Chain, Ptr, Val};
4778   return getAtomic(Opcode, dl, MemVT, VTs, Ops, MMO, Ordering, SynchScope);
4779 }
4780
4781 SDValue SelectionDAG::getAtomic(unsigned Opcode, SDLoc dl, EVT MemVT,
4782                                 EVT VT, SDValue Chain,
4783                                 SDValue Ptr,
4784                                 MachineMemOperand *MMO,
4785                                 AtomicOrdering Ordering,
4786                                 SynchronizationScope SynchScope) {
4787   assert(Opcode == ISD::ATOMIC_LOAD && "Invalid Atomic Op");
4788
4789   SDVTList VTs = getVTList(VT, MVT::Other);
4790   SDValue Ops[] = {Chain, Ptr};
4791   return getAtomic(Opcode, dl, MemVT, VTs, Ops, MMO, Ordering, SynchScope);
4792 }
4793
4794 /// getMergeValues - Create a MERGE_VALUES node from the given operands.
4795 SDValue SelectionDAG::getMergeValues(ArrayRef<SDValue> Ops, SDLoc dl) {
4796   if (Ops.size() == 1)
4797     return Ops[0];
4798
4799   SmallVector<EVT, 4> VTs;
4800   VTs.reserve(Ops.size());
4801   for (unsigned i = 0; i < Ops.size(); ++i)
4802     VTs.push_back(Ops[i].getValueType());
4803   return getNode(ISD::MERGE_VALUES, dl, getVTList(VTs), Ops);
4804 }
4805
4806 SDValue
4807 SelectionDAG::getMemIntrinsicNode(unsigned Opcode, SDLoc dl, SDVTList VTList,
4808                                   ArrayRef<SDValue> Ops,
4809                                   EVT MemVT, MachinePointerInfo PtrInfo,
4810                                   unsigned Align, bool Vol,
4811                                   bool ReadMem, bool WriteMem, unsigned Size) {
4812   if (Align == 0)  // Ensure that codegen never sees alignment 0
4813     Align = getEVTAlignment(MemVT);
4814
4815   MachineFunction &MF = getMachineFunction();
4816   unsigned Flags = 0;
4817   if (WriteMem)
4818     Flags |= MachineMemOperand::MOStore;
4819   if (ReadMem)
4820     Flags |= MachineMemOperand::MOLoad;
4821   if (Vol)
4822     Flags |= MachineMemOperand::MOVolatile;
4823   if (!Size)
4824     Size = MemVT.getStoreSize();
4825   MachineMemOperand *MMO =
4826     MF.getMachineMemOperand(PtrInfo, Flags, Size, Align);
4827
4828   return getMemIntrinsicNode(Opcode, dl, VTList, Ops, MemVT, MMO);
4829 }
4830
4831 SDValue
4832 SelectionDAG::getMemIntrinsicNode(unsigned Opcode, SDLoc dl, SDVTList VTList,
4833                                   ArrayRef<SDValue> Ops, EVT MemVT,
4834                                   MachineMemOperand *MMO) {
4835   assert((Opcode == ISD::INTRINSIC_VOID ||
4836           Opcode == ISD::INTRINSIC_W_CHAIN ||
4837           Opcode == ISD::PREFETCH ||
4838           Opcode == ISD::LIFETIME_START ||
4839           Opcode == ISD::LIFETIME_END ||
4840           (Opcode <= INT_MAX &&
4841            (int)Opcode >= ISD::FIRST_TARGET_MEMORY_OPCODE)) &&
4842          "Opcode is not a memory-accessing opcode!");
4843
4844   // Memoize the node unless it returns a flag.
4845   MemIntrinsicSDNode *N;
4846   if (VTList.VTs[VTList.NumVTs-1] != MVT::Glue) {
4847     FoldingSetNodeID ID;
4848     AddNodeIDNode(ID, Opcode, VTList, Ops);
4849     ID.AddInteger(MMO->getPointerInfo().getAddrSpace());
4850     void *IP = nullptr;
4851     if (SDNode *E = FindNodeOrInsertPos(ID, dl.getDebugLoc(), IP)) {
4852       cast<MemIntrinsicSDNode>(E)->refineAlignment(MMO);
4853       return SDValue(E, 0);
4854     }
4855
4856     N = new (NodeAllocator) MemIntrinsicSDNode(Opcode, dl.getIROrder(),
4857                                                dl.getDebugLoc(), VTList, Ops,
4858                                                MemVT, MMO);
4859     CSEMap.InsertNode(N, IP);
4860   } else {
4861     N = new (NodeAllocator) MemIntrinsicSDNode(Opcode, dl.getIROrder(),
4862                                                dl.getDebugLoc(), VTList, Ops,
4863                                                MemVT, MMO);
4864   }
4865   InsertNode(N);
4866   return SDValue(N, 0);
4867 }
4868
4869 /// InferPointerInfo - If the specified ptr/offset is a frame index, infer a
4870 /// MachinePointerInfo record from it.  This is particularly useful because the
4871 /// code generator has many cases where it doesn't bother passing in a
4872 /// MachinePointerInfo to getLoad or getStore when it has "FI+Cst".
4873 static MachinePointerInfo InferPointerInfo(SDValue Ptr, int64_t Offset = 0) {
4874   // If this is FI+Offset, we can model it.
4875   if (const FrameIndexSDNode *FI = dyn_cast<FrameIndexSDNode>(Ptr))
4876     return MachinePointerInfo::getFixedStack(FI->getIndex(), Offset);
4877
4878   // If this is (FI+Offset1)+Offset2, we can model it.
4879   if (Ptr.getOpcode() != ISD::ADD ||
4880       !isa<ConstantSDNode>(Ptr.getOperand(1)) ||
4881       !isa<FrameIndexSDNode>(Ptr.getOperand(0)))
4882     return MachinePointerInfo();
4883
4884   int FI = cast<FrameIndexSDNode>(Ptr.getOperand(0))->getIndex();
4885   return MachinePointerInfo::getFixedStack(FI, Offset+
4886                        cast<ConstantSDNode>(Ptr.getOperand(1))->getSExtValue());
4887 }
4888
4889 /// InferPointerInfo - If the specified ptr/offset is a frame index, infer a
4890 /// MachinePointerInfo record from it.  This is particularly useful because the
4891 /// code generator has many cases where it doesn't bother passing in a
4892 /// MachinePointerInfo to getLoad or getStore when it has "FI+Cst".
4893 static MachinePointerInfo InferPointerInfo(SDValue Ptr, SDValue OffsetOp) {
4894   // If the 'Offset' value isn't a constant, we can't handle this.
4895   if (ConstantSDNode *OffsetNode = dyn_cast<ConstantSDNode>(OffsetOp))
4896     return InferPointerInfo(Ptr, OffsetNode->getSExtValue());
4897   if (OffsetOp.getOpcode() == ISD::UNDEF)
4898     return InferPointerInfo(Ptr);
4899   return MachinePointerInfo();
4900 }
4901
4902
4903 SDValue
4904 SelectionDAG::getLoad(ISD::MemIndexedMode AM, ISD::LoadExtType ExtType,
4905                       EVT VT, SDLoc dl, SDValue Chain,
4906                       SDValue Ptr, SDValue Offset,
4907                       MachinePointerInfo PtrInfo, EVT MemVT,
4908                       bool isVolatile, bool isNonTemporal, bool isInvariant,
4909                       unsigned Alignment, const AAMDNodes &AAInfo,
4910                       const MDNode *Ranges) {
4911   assert(Chain.getValueType() == MVT::Other &&
4912         "Invalid chain type");
4913   if (Alignment == 0)  // Ensure that codegen never sees alignment 0
4914     Alignment = getEVTAlignment(VT);
4915
4916   unsigned Flags = MachineMemOperand::MOLoad;
4917   if (isVolatile)
4918     Flags |= MachineMemOperand::MOVolatile;
4919   if (isNonTemporal)
4920     Flags |= MachineMemOperand::MONonTemporal;
4921   if (isInvariant)
4922     Flags |= MachineMemOperand::MOInvariant;
4923
4924   // If we don't have a PtrInfo, infer the trivial frame index case to simplify
4925   // clients.
4926   if (PtrInfo.V.isNull())
4927     PtrInfo = InferPointerInfo(Ptr, Offset);
4928
4929   MachineFunction &MF = getMachineFunction();
4930   MachineMemOperand *MMO =
4931     MF.getMachineMemOperand(PtrInfo, Flags, MemVT.getStoreSize(), Alignment,
4932                             AAInfo, Ranges);
4933   return getLoad(AM, ExtType, VT, dl, Chain, Ptr, Offset, MemVT, MMO);
4934 }
4935
4936 SDValue
4937 SelectionDAG::getLoad(ISD::MemIndexedMode AM, ISD::LoadExtType ExtType,
4938                       EVT VT, SDLoc dl, SDValue Chain,
4939                       SDValue Ptr, SDValue Offset, EVT MemVT,
4940                       MachineMemOperand *MMO) {
4941   if (VT == MemVT) {
4942     ExtType = ISD::NON_EXTLOAD;
4943   } else if (ExtType == ISD::NON_EXTLOAD) {
4944     assert(VT == MemVT && "Non-extending load from different memory type!");
4945   } else {
4946     // Extending load.
4947     assert(MemVT.getScalarType().bitsLT(VT.getScalarType()) &&
4948            "Should only be an extending load, not truncating!");
4949     assert(VT.isInteger() == MemVT.isInteger() &&
4950            "Cannot convert from FP to Int or Int -> FP!");
4951     assert(VT.isVector() == MemVT.isVector() &&
4952            "Cannot use an ext load to convert to or from a vector!");
4953     assert((!VT.isVector() ||
4954             VT.getVectorNumElements() == MemVT.getVectorNumElements()) &&
4955            "Cannot use an ext load to change the number of vector elements!");
4956   }
4957
4958   bool Indexed = AM != ISD::UNINDEXED;
4959   assert((Indexed || Offset.getOpcode() == ISD::UNDEF) &&
4960          "Unindexed load with an offset!");
4961
4962   SDVTList VTs = Indexed ?
4963     getVTList(VT, Ptr.getValueType(), MVT::Other) : getVTList(VT, MVT::Other);
4964   SDValue Ops[] = { Chain, Ptr, Offset };
4965   FoldingSetNodeID ID;
4966   AddNodeIDNode(ID, ISD::LOAD, VTs, Ops);
4967   ID.AddInteger(MemVT.getRawBits());
4968   ID.AddInteger(encodeMemSDNodeFlags(ExtType, AM, MMO->isVolatile(),
4969                                      MMO->isNonTemporal(),
4970                                      MMO->isInvariant()));
4971   ID.AddInteger(MMO->getPointerInfo().getAddrSpace());
4972   void *IP = nullptr;
4973   if (SDNode *E = FindNodeOrInsertPos(ID, dl.getDebugLoc(), IP)) {
4974     cast<LoadSDNode>(E)->refineAlignment(MMO);
4975     return SDValue(E, 0);
4976   }
4977   SDNode *N = new (NodeAllocator) LoadSDNode(Ops, dl.getIROrder(),
4978                                              dl.getDebugLoc(), VTs, AM, ExtType,
4979                                              MemVT, MMO);
4980   CSEMap.InsertNode(N, IP);
4981   InsertNode(N);
4982   return SDValue(N, 0);
4983 }
4984
4985 SDValue SelectionDAG::getLoad(EVT VT, SDLoc dl,
4986                               SDValue Chain, SDValue Ptr,
4987                               MachinePointerInfo PtrInfo,
4988                               bool isVolatile, bool isNonTemporal,
4989                               bool isInvariant, unsigned Alignment,
4990                               const AAMDNodes &AAInfo,
4991                               const MDNode *Ranges) {
4992   SDValue Undef = getUNDEF(Ptr.getValueType());
4993   return getLoad(ISD::UNINDEXED, ISD::NON_EXTLOAD, VT, dl, Chain, Ptr, Undef,
4994                  PtrInfo, VT, isVolatile, isNonTemporal, isInvariant, Alignment,
4995                  AAInfo, Ranges);
4996 }
4997
4998 SDValue SelectionDAG::getLoad(EVT VT, SDLoc dl,
4999                               SDValue Chain, SDValue Ptr,
5000                               MachineMemOperand *MMO) {
5001   SDValue Undef = getUNDEF(Ptr.getValueType());
5002   return getLoad(ISD::UNINDEXED, ISD::NON_EXTLOAD, VT, dl, Chain, Ptr, Undef,
5003                  VT, MMO);
5004 }
5005
5006 SDValue SelectionDAG::getExtLoad(ISD::LoadExtType ExtType, SDLoc dl, EVT VT,
5007                                  SDValue Chain, SDValue Ptr,
5008                                  MachinePointerInfo PtrInfo, EVT MemVT,
5009                                  bool isVolatile, bool isNonTemporal,
5010                                  bool isInvariant, unsigned Alignment,
5011                                  const AAMDNodes &AAInfo) {
5012   SDValue Undef = getUNDEF(Ptr.getValueType());
5013   return getLoad(ISD::UNINDEXED, ExtType, VT, dl, Chain, Ptr, Undef,
5014                  PtrInfo, MemVT, isVolatile, isNonTemporal, isInvariant,
5015                  Alignment, AAInfo);
5016 }
5017
5018
5019 SDValue SelectionDAG::getExtLoad(ISD::LoadExtType ExtType, SDLoc dl, EVT VT,
5020                                  SDValue Chain, SDValue Ptr, EVT MemVT,
5021                                  MachineMemOperand *MMO) {
5022   SDValue Undef = getUNDEF(Ptr.getValueType());
5023   return getLoad(ISD::UNINDEXED, ExtType, VT, dl, Chain, Ptr, Undef,
5024                  MemVT, MMO);
5025 }
5026
5027 SDValue
5028 SelectionDAG::getIndexedLoad(SDValue OrigLoad, SDLoc dl, SDValue Base,
5029                              SDValue Offset, ISD::MemIndexedMode AM) {
5030   LoadSDNode *LD = cast<LoadSDNode>(OrigLoad);
5031   assert(LD->getOffset().getOpcode() == ISD::UNDEF &&
5032          "Load is already a indexed load!");
5033   return getLoad(AM, LD->getExtensionType(), OrigLoad.getValueType(), dl,
5034                  LD->getChain(), Base, Offset, LD->getPointerInfo(),
5035                  LD->getMemoryVT(), LD->isVolatile(), LD->isNonTemporal(),
5036                  false, LD->getAlignment());
5037 }
5038
5039 SDValue SelectionDAG::getStore(SDValue Chain, SDLoc dl, SDValue Val,
5040                                SDValue Ptr, MachinePointerInfo PtrInfo,
5041                                bool isVolatile, bool isNonTemporal,
5042                                unsigned Alignment, const AAMDNodes &AAInfo) {
5043   assert(Chain.getValueType() == MVT::Other &&
5044         "Invalid chain type");
5045   if (Alignment == 0)  // Ensure that codegen never sees alignment 0
5046     Alignment = getEVTAlignment(Val.getValueType());
5047
5048   unsigned Flags = MachineMemOperand::MOStore;
5049   if (isVolatile)
5050     Flags |= MachineMemOperand::MOVolatile;
5051   if (isNonTemporal)
5052     Flags |= MachineMemOperand::MONonTemporal;
5053
5054   if (PtrInfo.V.isNull())
5055     PtrInfo = InferPointerInfo(Ptr);
5056
5057   MachineFunction &MF = getMachineFunction();
5058   MachineMemOperand *MMO =
5059     MF.getMachineMemOperand(PtrInfo, Flags,
5060                             Val.getValueType().getStoreSize(), Alignment,
5061                             AAInfo);
5062
5063   return getStore(Chain, dl, Val, Ptr, MMO);
5064 }
5065
5066 SDValue SelectionDAG::getStore(SDValue Chain, SDLoc dl, SDValue Val,
5067                                SDValue Ptr, MachineMemOperand *MMO) {
5068   assert(Chain.getValueType() == MVT::Other &&
5069         "Invalid chain type");
5070   EVT VT = Val.getValueType();
5071   SDVTList VTs = getVTList(MVT::Other);
5072   SDValue Undef = getUNDEF(Ptr.getValueType());
5073   SDValue Ops[] = { Chain, Val, Ptr, Undef };
5074   FoldingSetNodeID ID;
5075   AddNodeIDNode(ID, ISD::STORE, VTs, Ops);
5076   ID.AddInteger(VT.getRawBits());
5077   ID.AddInteger(encodeMemSDNodeFlags(false, ISD::UNINDEXED, MMO->isVolatile(),
5078                                      MMO->isNonTemporal(), MMO->isInvariant()));
5079   ID.AddInteger(MMO->getPointerInfo().getAddrSpace());
5080   void *IP = nullptr;
5081   if (SDNode *E = FindNodeOrInsertPos(ID, dl.getDebugLoc(), IP)) {
5082     cast<StoreSDNode>(E)->refineAlignment(MMO);
5083     return SDValue(E, 0);
5084   }
5085   SDNode *N = new (NodeAllocator) StoreSDNode(Ops, dl.getIROrder(),
5086                                               dl.getDebugLoc(), VTs,
5087                                               ISD::UNINDEXED, false, VT, MMO);
5088   CSEMap.InsertNode(N, IP);
5089   InsertNode(N);
5090   return SDValue(N, 0);
5091 }
5092
5093 SDValue SelectionDAG::getTruncStore(SDValue Chain, SDLoc dl, SDValue Val,
5094                                     SDValue Ptr, MachinePointerInfo PtrInfo,
5095                                     EVT SVT,bool isVolatile, bool isNonTemporal,
5096                                     unsigned Alignment,
5097                                     const AAMDNodes &AAInfo) {
5098   assert(Chain.getValueType() == MVT::Other &&
5099         "Invalid chain type");
5100   if (Alignment == 0)  // Ensure that codegen never sees alignment 0
5101     Alignment = getEVTAlignment(SVT);
5102
5103   unsigned Flags = MachineMemOperand::MOStore;
5104   if (isVolatile)
5105     Flags |= MachineMemOperand::MOVolatile;
5106   if (isNonTemporal)
5107     Flags |= MachineMemOperand::MONonTemporal;
5108
5109   if (PtrInfo.V.isNull())
5110     PtrInfo = InferPointerInfo(Ptr);
5111
5112   MachineFunction &MF = getMachineFunction();
5113   MachineMemOperand *MMO =
5114     MF.getMachineMemOperand(PtrInfo, Flags, SVT.getStoreSize(), Alignment,
5115                             AAInfo);
5116
5117   return getTruncStore(Chain, dl, Val, Ptr, SVT, MMO);
5118 }
5119
5120 SDValue SelectionDAG::getTruncStore(SDValue Chain, SDLoc dl, SDValue Val,
5121                                     SDValue Ptr, EVT SVT,
5122                                     MachineMemOperand *MMO) {
5123   EVT VT = Val.getValueType();
5124
5125   assert(Chain.getValueType() == MVT::Other &&
5126         "Invalid chain type");
5127   if (VT == SVT)
5128     return getStore(Chain, dl, Val, Ptr, MMO);
5129
5130   assert(SVT.getScalarType().bitsLT(VT.getScalarType()) &&
5131          "Should only be a truncating store, not extending!");
5132   assert(VT.isInteger() == SVT.isInteger() &&
5133          "Can't do FP-INT conversion!");
5134   assert(VT.isVector() == SVT.isVector() &&
5135          "Cannot use trunc store to convert to or from a vector!");
5136   assert((!VT.isVector() ||
5137           VT.getVectorNumElements() == SVT.getVectorNumElements()) &&
5138          "Cannot use trunc store to change the number of vector elements!");
5139
5140   SDVTList VTs = getVTList(MVT::Other);
5141   SDValue Undef = getUNDEF(Ptr.getValueType());
5142   SDValue Ops[] = { Chain, Val, Ptr, Undef };
5143   FoldingSetNodeID ID;
5144   AddNodeIDNode(ID, ISD::STORE, VTs, Ops);
5145   ID.AddInteger(SVT.getRawBits());
5146   ID.AddInteger(encodeMemSDNodeFlags(true, ISD::UNINDEXED, MMO->isVolatile(),
5147                                      MMO->isNonTemporal(), MMO->isInvariant()));
5148   ID.AddInteger(MMO->getPointerInfo().getAddrSpace());
5149   void *IP = nullptr;
5150   if (SDNode *E = FindNodeOrInsertPos(ID, dl.getDebugLoc(), IP)) {
5151     cast<StoreSDNode>(E)->refineAlignment(MMO);
5152     return SDValue(E, 0);
5153   }
5154   SDNode *N = new (NodeAllocator) StoreSDNode(Ops, dl.getIROrder(),
5155                                               dl.getDebugLoc(), VTs,
5156                                               ISD::UNINDEXED, true, SVT, MMO);
5157   CSEMap.InsertNode(N, IP);
5158   InsertNode(N);
5159   return SDValue(N, 0);
5160 }
5161
5162 SDValue
5163 SelectionDAG::getIndexedStore(SDValue OrigStore, SDLoc dl, SDValue Base,
5164                               SDValue Offset, ISD::MemIndexedMode AM) {
5165   StoreSDNode *ST = cast<StoreSDNode>(OrigStore);
5166   assert(ST->getOffset().getOpcode() == ISD::UNDEF &&
5167          "Store is already a indexed store!");
5168   SDVTList VTs = getVTList(Base.getValueType(), MVT::Other);
5169   SDValue Ops[] = { ST->getChain(), ST->getValue(), Base, Offset };
5170   FoldingSetNodeID ID;
5171   AddNodeIDNode(ID, ISD::STORE, VTs, Ops);
5172   ID.AddInteger(ST->getMemoryVT().getRawBits());
5173   ID.AddInteger(ST->getRawSubclassData());
5174   ID.AddInteger(ST->getPointerInfo().getAddrSpace());
5175   void *IP = nullptr;
5176   if (SDNode *E = FindNodeOrInsertPos(ID, dl.getDebugLoc(), IP))
5177     return SDValue(E, 0);
5178
5179   SDNode *N = new (NodeAllocator) StoreSDNode(Ops, dl.getIROrder(),
5180                                               dl.getDebugLoc(), VTs, AM,
5181                                               ST->isTruncatingStore(),
5182                                               ST->getMemoryVT(),
5183                                               ST->getMemOperand());
5184   CSEMap.InsertNode(N, IP);
5185   InsertNode(N);
5186   return SDValue(N, 0);
5187 }
5188
5189 SDValue
5190 SelectionDAG::getMaskedLoad(EVT VT, SDLoc dl, SDValue Chain,
5191                             SDValue Ptr, SDValue Mask, SDValue Src0, EVT MemVT,
5192                             MachineMemOperand *MMO, ISD::LoadExtType ExtTy) {
5193
5194   SDVTList VTs = getVTList(VT, MVT::Other);
5195   SDValue Ops[] = { Chain, Ptr, Mask, Src0 };
5196   FoldingSetNodeID ID;
5197   AddNodeIDNode(ID, ISD::MLOAD, VTs, Ops);
5198   ID.AddInteger(VT.getRawBits());
5199   ID.AddInteger(encodeMemSDNodeFlags(ExtTy, ISD::UNINDEXED,
5200                                      MMO->isVolatile(),
5201                                      MMO->isNonTemporal(),
5202                                      MMO->isInvariant()));
5203   ID.AddInteger(MMO->getPointerInfo().getAddrSpace());
5204   void *IP = nullptr;
5205   if (SDNode *E = FindNodeOrInsertPos(ID, dl.getDebugLoc(), IP)) {
5206     cast<MaskedLoadSDNode>(E)->refineAlignment(MMO);
5207     return SDValue(E, 0);
5208   }
5209   SDNode *N = new (NodeAllocator) MaskedLoadSDNode(dl.getIROrder(),
5210                                              dl.getDebugLoc(), Ops, 4, VTs,
5211                                              ExtTy, MemVT, MMO);
5212   CSEMap.InsertNode(N, IP);
5213   InsertNode(N);
5214   return SDValue(N, 0);
5215 }
5216
5217 SDValue SelectionDAG::getMaskedStore(SDValue Chain, SDLoc dl, SDValue Val,
5218                                      SDValue Ptr, SDValue Mask, EVT MemVT,
5219                                      MachineMemOperand *MMO, bool isTrunc) {
5220   assert(Chain.getValueType() == MVT::Other &&
5221         "Invalid chain type");
5222   EVT VT = Val.getValueType();
5223   SDVTList VTs = getVTList(MVT::Other);
5224   SDValue Ops[] = { Chain, Ptr, Mask, Val };
5225   FoldingSetNodeID ID;
5226   AddNodeIDNode(ID, ISD::MSTORE, VTs, Ops);
5227   ID.AddInteger(VT.getRawBits());
5228   ID.AddInteger(encodeMemSDNodeFlags(false, ISD::UNINDEXED, MMO->isVolatile(),
5229                                      MMO->isNonTemporal(), MMO->isInvariant()));
5230   ID.AddInteger(MMO->getPointerInfo().getAddrSpace());
5231   void *IP = nullptr;
5232   if (SDNode *E = FindNodeOrInsertPos(ID, dl.getDebugLoc(), IP)) {
5233     cast<MaskedStoreSDNode>(E)->refineAlignment(MMO);
5234     return SDValue(E, 0);
5235   }
5236   SDNode *N = new (NodeAllocator) MaskedStoreSDNode(dl.getIROrder(),
5237                                                     dl.getDebugLoc(), Ops, 4,
5238                                                     VTs, isTrunc, MemVT, MMO);
5239   CSEMap.InsertNode(N, IP);
5240   InsertNode(N);
5241   return SDValue(N, 0);
5242 }
5243
5244 SDValue
5245 SelectionDAG::getMaskedGather(SDVTList VTs, EVT VT, SDLoc dl,
5246                               ArrayRef<SDValue> Ops,
5247                               MachineMemOperand *MMO) {
5248
5249   FoldingSetNodeID ID;
5250   AddNodeIDNode(ID, ISD::MGATHER, VTs, Ops);
5251   ID.AddInteger(VT.getRawBits());
5252   ID.AddInteger(encodeMemSDNodeFlags(ISD::NON_EXTLOAD, ISD::UNINDEXED,
5253                                      MMO->isVolatile(),
5254                                      MMO->isNonTemporal(),
5255                                      MMO->isInvariant()));
5256   ID.AddInteger(MMO->getPointerInfo().getAddrSpace());
5257   void *IP = nullptr;
5258   if (SDNode *E = FindNodeOrInsertPos(ID, dl.getDebugLoc(), IP)) {
5259     cast<MaskedGatherSDNode>(E)->refineAlignment(MMO);
5260     return SDValue(E, 0);
5261   }
5262   MaskedGatherSDNode *N = 
5263     new (NodeAllocator) MaskedGatherSDNode(dl.getIROrder(), dl.getDebugLoc(),
5264                                            Ops, VTs, VT, MMO);
5265   CSEMap.InsertNode(N, IP);
5266   InsertNode(N);
5267   return SDValue(N, 0);
5268 }
5269
5270 SDValue SelectionDAG::getMaskedScatter(SDVTList VTs, EVT VT, SDLoc dl,
5271                                        ArrayRef<SDValue> Ops,
5272                                        MachineMemOperand *MMO) {
5273   FoldingSetNodeID ID;
5274   AddNodeIDNode(ID, ISD::MSCATTER, VTs, Ops);
5275   ID.AddInteger(VT.getRawBits());
5276   ID.AddInteger(encodeMemSDNodeFlags(false, ISD::UNINDEXED, MMO->isVolatile(),
5277                                      MMO->isNonTemporal(),
5278                                      MMO->isInvariant()));
5279   ID.AddInteger(MMO->getPointerInfo().getAddrSpace());
5280   void *IP = nullptr;
5281   if (SDNode *E = FindNodeOrInsertPos(ID, dl.getDebugLoc(), IP)) {
5282     cast<MaskedScatterSDNode>(E)->refineAlignment(MMO);
5283     return SDValue(E, 0);
5284   }
5285   SDNode *N =
5286     new (NodeAllocator) MaskedScatterSDNode(dl.getIROrder(), dl.getDebugLoc(),
5287                                             Ops, VTs, VT, MMO);
5288   CSEMap.InsertNode(N, IP);
5289   InsertNode(N);
5290   return SDValue(N, 0);
5291 }
5292
5293 SDValue SelectionDAG::getVAArg(EVT VT, SDLoc dl,
5294                                SDValue Chain, SDValue Ptr,
5295                                SDValue SV,
5296                                unsigned Align) {
5297   SDValue Ops[] = { Chain, Ptr, SV, getTargetConstant(Align, dl, MVT::i32) };
5298   return getNode(ISD::VAARG, dl, getVTList(VT, MVT::Other), Ops);
5299 }
5300
5301 SDValue SelectionDAG::getNode(unsigned Opcode, SDLoc DL, EVT VT,
5302                               ArrayRef<SDUse> Ops) {
5303   switch (Ops.size()) {
5304   case 0: return getNode(Opcode, DL, VT);
5305   case 1: return getNode(Opcode, DL, VT, static_cast<const SDValue>(Ops[0]));
5306   case 2: return getNode(Opcode, DL, VT, Ops[0], Ops[1]);
5307   case 3: return getNode(Opcode, DL, VT, Ops[0], Ops[1], Ops[2]);
5308   default: break;
5309   }
5310
5311   // Copy from an SDUse array into an SDValue array for use with
5312   // the regular getNode logic.
5313   SmallVector<SDValue, 8> NewOps(Ops.begin(), Ops.end());
5314   return getNode(Opcode, DL, VT, NewOps);
5315 }
5316
5317 SDValue SelectionDAG::getNode(unsigned Opcode, SDLoc DL, EVT VT,
5318                               ArrayRef<SDValue> Ops) {
5319   unsigned NumOps = Ops.size();
5320   switch (NumOps) {
5321   case 0: return getNode(Opcode, DL, VT);
5322   case 1: return getNode(Opcode, DL, VT, Ops[0]);
5323   case 2: return getNode(Opcode, DL, VT, Ops[0], Ops[1]);
5324   case 3: return getNode(Opcode, DL, VT, Ops[0], Ops[1], Ops[2]);
5325   default: break;
5326   }
5327
5328   switch (Opcode) {
5329   default: break;
5330   case ISD::SELECT_CC: {
5331     assert(NumOps == 5 && "SELECT_CC takes 5 operands!");
5332     assert(Ops[0].getValueType() == Ops[1].getValueType() &&
5333            "LHS and RHS of condition must have same type!");
5334     assert(Ops[2].getValueType() == Ops[3].getValueType() &&
5335            "True and False arms of SelectCC must have same type!");
5336     assert(Ops[2].getValueType() == VT &&
5337            "select_cc node must be of same type as true and false value!");
5338     break;
5339   }
5340   case ISD::BR_CC: {
5341     assert(NumOps == 5 && "BR_CC takes 5 operands!");
5342     assert(Ops[2].getValueType() == Ops[3].getValueType() &&
5343            "LHS/RHS of comparison should match types!");
5344     break;
5345   }
5346   }
5347
5348   // Memoize nodes.
5349   SDNode *N;
5350   SDVTList VTs = getVTList(VT);
5351
5352   if (VT != MVT::Glue) {
5353     FoldingSetNodeID ID;
5354     AddNodeIDNode(ID, Opcode, VTs, Ops);
5355     void *IP = nullptr;
5356
5357     if (SDNode *E = FindNodeOrInsertPos(ID, DL.getDebugLoc(), IP))
5358       return SDValue(E, 0);
5359
5360     N = new (NodeAllocator) SDNode(Opcode, DL.getIROrder(), DL.getDebugLoc(),
5361                                    VTs, Ops);
5362     CSEMap.InsertNode(N, IP);
5363   } else {
5364     N = new (NodeAllocator) SDNode(Opcode, DL.getIROrder(), DL.getDebugLoc(),
5365                                    VTs, Ops);
5366   }
5367
5368   InsertNode(N);
5369   return SDValue(N, 0);
5370 }
5371
5372 SDValue SelectionDAG::getNode(unsigned Opcode, SDLoc DL,
5373                               ArrayRef<EVT> ResultTys, ArrayRef<SDValue> Ops) {
5374   return getNode(Opcode, DL, getVTList(ResultTys), Ops);
5375 }
5376
5377 SDValue SelectionDAG::getNode(unsigned Opcode, SDLoc DL, SDVTList VTList,
5378                               ArrayRef<SDValue> Ops) {
5379   if (VTList.NumVTs == 1)
5380     return getNode(Opcode, DL, VTList.VTs[0], Ops);
5381
5382 #if 0
5383   switch (Opcode) {
5384   // FIXME: figure out how to safely handle things like
5385   // int foo(int x) { return 1 << (x & 255); }
5386   // int bar() { return foo(256); }
5387   case ISD::SRA_PARTS:
5388   case ISD::SRL_PARTS:
5389   case ISD::SHL_PARTS:
5390     if (N3.getOpcode() == ISD::SIGN_EXTEND_INREG &&
5391         cast<VTSDNode>(N3.getOperand(1))->getVT() != MVT::i1)
5392       return getNode(Opcode, DL, VT, N1, N2, N3.getOperand(0));
5393     else if (N3.getOpcode() == ISD::AND)
5394       if (ConstantSDNode *AndRHS = dyn_cast<ConstantSDNode>(N3.getOperand(1))) {
5395         // If the and is only masking out bits that cannot effect the shift,
5396         // eliminate the and.
5397         unsigned NumBits = VT.getScalarType().getSizeInBits()*2;
5398         if ((AndRHS->getValue() & (NumBits-1)) == NumBits-1)
5399           return getNode(Opcode, DL, VT, N1, N2, N3.getOperand(0));
5400       }
5401     break;
5402   }
5403 #endif
5404
5405   // Memoize the node unless it returns a flag.
5406   SDNode *N;
5407   unsigned NumOps = Ops.size();
5408   if (VTList.VTs[VTList.NumVTs-1] != MVT::Glue) {
5409     FoldingSetNodeID ID;
5410     AddNodeIDNode(ID, Opcode, VTList, Ops);
5411     void *IP = nullptr;
5412     if (SDNode *E = FindNodeOrInsertPos(ID, DL.getDebugLoc(), IP))
5413       return SDValue(E, 0);
5414
5415     if (NumOps == 1) {
5416       N = new (NodeAllocator) UnarySDNode(Opcode, DL.getIROrder(),
5417                                           DL.getDebugLoc(), VTList, Ops[0]);
5418     } else if (NumOps == 2) {
5419       N = new (NodeAllocator) BinarySDNode(Opcode, DL.getIROrder(),
5420                                            DL.getDebugLoc(), VTList, Ops[0],
5421                                            Ops[1]);
5422     } else if (NumOps == 3) {
5423       N = new (NodeAllocator) TernarySDNode(Opcode, DL.getIROrder(),
5424                                             DL.getDebugLoc(), VTList, Ops[0],
5425                                             Ops[1], Ops[2]);
5426     } else {
5427       N = new (NodeAllocator) SDNode(Opcode, DL.getIROrder(), DL.getDebugLoc(),
5428                                      VTList, Ops);
5429     }
5430     CSEMap.InsertNode(N, IP);
5431   } else {
5432     if (NumOps == 1) {
5433       N = new (NodeAllocator) UnarySDNode(Opcode, DL.getIROrder(),
5434                                           DL.getDebugLoc(), VTList, Ops[0]);
5435     } else if (NumOps == 2) {
5436       N = new (NodeAllocator) BinarySDNode(Opcode, DL.getIROrder(),
5437                                            DL.getDebugLoc(), VTList, Ops[0],
5438                                            Ops[1]);
5439     } else if (NumOps == 3) {
5440       N = new (NodeAllocator) TernarySDNode(Opcode, DL.getIROrder(),
5441                                             DL.getDebugLoc(), VTList, Ops[0],
5442                                             Ops[1], Ops[2]);
5443     } else {
5444       N = new (NodeAllocator) SDNode(Opcode, DL.getIROrder(), DL.getDebugLoc(),
5445                                      VTList, Ops);
5446     }
5447   }
5448   InsertNode(N);
5449   return SDValue(N, 0);
5450 }
5451
5452 SDValue SelectionDAG::getNode(unsigned Opcode, SDLoc DL, SDVTList VTList) {
5453   return getNode(Opcode, DL, VTList, None);
5454 }
5455
5456 SDValue SelectionDAG::getNode(unsigned Opcode, SDLoc DL, SDVTList VTList,
5457                               SDValue N1) {
5458   SDValue Ops[] = { N1 };
5459   return getNode(Opcode, DL, VTList, Ops);
5460 }
5461
5462 SDValue SelectionDAG::getNode(unsigned Opcode, SDLoc DL, SDVTList VTList,
5463                               SDValue N1, SDValue N2) {
5464   SDValue Ops[] = { N1, N2 };
5465   return getNode(Opcode, DL, VTList, Ops);
5466 }
5467
5468 SDValue SelectionDAG::getNode(unsigned Opcode, SDLoc DL, SDVTList VTList,
5469                               SDValue N1, SDValue N2, SDValue N3) {
5470   SDValue Ops[] = { N1, N2, N3 };
5471   return getNode(Opcode, DL, VTList, Ops);
5472 }
5473
5474 SDValue SelectionDAG::getNode(unsigned Opcode, SDLoc DL, SDVTList VTList,
5475                               SDValue N1, SDValue N2, SDValue N3,
5476                               SDValue N4) {
5477   SDValue Ops[] = { N1, N2, N3, N4 };
5478   return getNode(Opcode, DL, VTList, Ops);
5479 }
5480
5481 SDValue SelectionDAG::getNode(unsigned Opcode, SDLoc DL, SDVTList VTList,
5482                               SDValue N1, SDValue N2, SDValue N3,
5483                               SDValue N4, SDValue N5) {
5484   SDValue Ops[] = { N1, N2, N3, N4, N5 };
5485   return getNode(Opcode, DL, VTList, Ops);
5486 }
5487
5488 SDVTList SelectionDAG::getVTList(EVT VT) {
5489   return makeVTList(SDNode::getValueTypeList(VT), 1);
5490 }
5491
5492 SDVTList SelectionDAG::getVTList(EVT VT1, EVT VT2) {
5493   FoldingSetNodeID ID;
5494   ID.AddInteger(2U);
5495   ID.AddInteger(VT1.getRawBits());
5496   ID.AddInteger(VT2.getRawBits());
5497
5498   void *IP = nullptr;
5499   SDVTListNode *Result = VTListMap.FindNodeOrInsertPos(ID, IP);
5500   if (!Result) {
5501     EVT *Array = Allocator.Allocate<EVT>(2);
5502     Array[0] = VT1;
5503     Array[1] = VT2;
5504     Result = new (Allocator) SDVTListNode(ID.Intern(Allocator), Array, 2);
5505     VTListMap.InsertNode(Result, IP);
5506   }
5507   return Result->getSDVTList();
5508 }
5509
5510 SDVTList SelectionDAG::getVTList(EVT VT1, EVT VT2, EVT VT3) {
5511   FoldingSetNodeID ID;
5512   ID.AddInteger(3U);
5513   ID.AddInteger(VT1.getRawBits());
5514   ID.AddInteger(VT2.getRawBits());
5515   ID.AddInteger(VT3.getRawBits());
5516
5517   void *IP = nullptr;
5518   SDVTListNode *Result = VTListMap.FindNodeOrInsertPos(ID, IP);
5519   if (!Result) {
5520     EVT *Array = Allocator.Allocate<EVT>(3);
5521     Array[0] = VT1;
5522     Array[1] = VT2;
5523     Array[2] = VT3;
5524     Result = new (Allocator) SDVTListNode(ID.Intern(Allocator), Array, 3);
5525     VTListMap.InsertNode(Result, IP);
5526   }
5527   return Result->getSDVTList();
5528 }
5529
5530 SDVTList SelectionDAG::getVTList(EVT VT1, EVT VT2, EVT VT3, EVT VT4) {
5531   FoldingSetNodeID ID;
5532   ID.AddInteger(4U);
5533   ID.AddInteger(VT1.getRawBits());
5534   ID.AddInteger(VT2.getRawBits());
5535   ID.AddInteger(VT3.getRawBits());
5536   ID.AddInteger(VT4.getRawBits());
5537
5538   void *IP = nullptr;
5539   SDVTListNode *Result = VTListMap.FindNodeOrInsertPos(ID, IP);
5540   if (!Result) {
5541     EVT *Array = Allocator.Allocate<EVT>(4);
5542     Array[0] = VT1;
5543     Array[1] = VT2;
5544     Array[2] = VT3;
5545     Array[3] = VT4;
5546     Result = new (Allocator) SDVTListNode(ID.Intern(Allocator), Array, 4);
5547     VTListMap.InsertNode(Result, IP);
5548   }
5549   return Result->getSDVTList();
5550 }
5551
5552 SDVTList SelectionDAG::getVTList(ArrayRef<EVT> VTs) {
5553   unsigned NumVTs = VTs.size();
5554   FoldingSetNodeID ID;
5555   ID.AddInteger(NumVTs);
5556   for (unsigned index = 0; index < NumVTs; index++) {
5557     ID.AddInteger(VTs[index].getRawBits());
5558   }
5559
5560   void *IP = nullptr;
5561   SDVTListNode *Result = VTListMap.FindNodeOrInsertPos(ID, IP);
5562   if (!Result) {
5563     EVT *Array = Allocator.Allocate<EVT>(NumVTs);
5564     std::copy(VTs.begin(), VTs.end(), Array);
5565     Result = new (Allocator) SDVTListNode(ID.Intern(Allocator), Array, NumVTs);
5566     VTListMap.InsertNode(Result, IP);
5567   }
5568   return Result->getSDVTList();
5569 }
5570
5571
5572 /// UpdateNodeOperands - *Mutate* the specified node in-place to have the
5573 /// specified operands.  If the resultant node already exists in the DAG,
5574 /// this does not modify the specified node, instead it returns the node that
5575 /// already exists.  If the resultant node does not exist in the DAG, the
5576 /// input node is returned.  As a degenerate case, if you specify the same
5577 /// input operands as the node already has, the input node is returned.
5578 SDNode *SelectionDAG::UpdateNodeOperands(SDNode *N, SDValue Op) {
5579   assert(N->getNumOperands() == 1 && "Update with wrong number of operands");
5580
5581   // Check to see if there is no change.
5582   if (Op == N->getOperand(0)) return N;
5583
5584   // See if the modified node already exists.
5585   void *InsertPos = nullptr;
5586   if (SDNode *Existing = FindModifiedNodeSlot(N, Op, InsertPos))
5587     return Existing;
5588
5589   // Nope it doesn't.  Remove the node from its current place in the maps.
5590   if (InsertPos)
5591     if (!RemoveNodeFromCSEMaps(N))
5592       InsertPos = nullptr;
5593
5594   // Now we update the operands.
5595   N->OperandList[0].set(Op);
5596
5597   // If this gets put into a CSE map, add it.
5598   if (InsertPos) CSEMap.InsertNode(N, InsertPos);
5599   return N;
5600 }
5601
5602 SDNode *SelectionDAG::UpdateNodeOperands(SDNode *N, SDValue Op1, SDValue Op2) {
5603   assert(N->getNumOperands() == 2 && "Update with wrong number of operands");
5604
5605   // Check to see if there is no change.
5606   if (Op1 == N->getOperand(0) && Op2 == N->getOperand(1))
5607     return N;   // No operands changed, just return the input node.
5608
5609   // See if the modified node already exists.
5610   void *InsertPos = nullptr;
5611   if (SDNode *Existing = FindModifiedNodeSlot(N, Op1, Op2, InsertPos))
5612     return Existing;
5613
5614   // Nope it doesn't.  Remove the node from its current place in the maps.
5615   if (InsertPos)
5616     if (!RemoveNodeFromCSEMaps(N))
5617       InsertPos = nullptr;
5618
5619   // Now we update the operands.
5620   if (N->OperandList[0] != Op1)
5621     N->OperandList[0].set(Op1);
5622   if (N->OperandList[1] != Op2)
5623     N->OperandList[1].set(Op2);
5624
5625   // If this gets put into a CSE map, add it.
5626   if (InsertPos) CSEMap.InsertNode(N, InsertPos);
5627   return N;
5628 }
5629
5630 SDNode *SelectionDAG::
5631 UpdateNodeOperands(SDNode *N, SDValue Op1, SDValue Op2, SDValue Op3) {
5632   SDValue Ops[] = { Op1, Op2, Op3 };
5633   return UpdateNodeOperands(N, Ops);
5634 }
5635
5636 SDNode *SelectionDAG::
5637 UpdateNodeOperands(SDNode *N, SDValue Op1, SDValue Op2,
5638                    SDValue Op3, SDValue Op4) {
5639   SDValue Ops[] = { Op1, Op2, Op3, Op4 };
5640   return UpdateNodeOperands(N, Ops);
5641 }
5642
5643 SDNode *SelectionDAG::
5644 UpdateNodeOperands(SDNode *N, SDValue Op1, SDValue Op2,
5645                    SDValue Op3, SDValue Op4, SDValue Op5) {
5646   SDValue Ops[] = { Op1, Op2, Op3, Op4, Op5 };
5647   return UpdateNodeOperands(N, Ops);
5648 }
5649
5650 SDNode *SelectionDAG::
5651 UpdateNodeOperands(SDNode *N, ArrayRef<SDValue> Ops) {
5652   unsigned NumOps = Ops.size();
5653   assert(N->getNumOperands() == NumOps &&
5654          "Update with wrong number of operands");
5655
5656   // If no operands changed just return the input node.
5657   if (Ops.empty() || std::equal(Ops.begin(), Ops.end(), N->op_begin()))
5658     return N;
5659
5660   // See if the modified node already exists.
5661   void *InsertPos = nullptr;
5662   if (SDNode *Existing = FindModifiedNodeSlot(N, Ops, InsertPos))
5663     return Existing;
5664
5665   // Nope it doesn't.  Remove the node from its current place in the maps.
5666   if (InsertPos)
5667     if (!RemoveNodeFromCSEMaps(N))
5668       InsertPos = nullptr;
5669
5670   // Now we update the operands.
5671   for (unsigned i = 0; i != NumOps; ++i)
5672     if (N->OperandList[i] != Ops[i])
5673       N->OperandList[i].set(Ops[i]);
5674
5675   // If this gets put into a CSE map, add it.
5676   if (InsertPos) CSEMap.InsertNode(N, InsertPos);
5677   return N;
5678 }
5679
5680 /// DropOperands - Release the operands and set this node to have
5681 /// zero operands.
5682 void SDNode::DropOperands() {
5683   // Unlike the code in MorphNodeTo that does this, we don't need to
5684   // watch for dead nodes here.
5685   for (op_iterator I = op_begin(), E = op_end(); I != E; ) {
5686     SDUse &Use = *I++;
5687     Use.set(SDValue());
5688   }
5689 }
5690
5691 /// SelectNodeTo - These are wrappers around MorphNodeTo that accept a
5692 /// machine opcode.
5693 ///
5694 SDNode *SelectionDAG::SelectNodeTo(SDNode *N, unsigned MachineOpc,
5695                                    EVT VT) {
5696   SDVTList VTs = getVTList(VT);
5697   return SelectNodeTo(N, MachineOpc, VTs, None);
5698 }
5699
5700 SDNode *SelectionDAG::SelectNodeTo(SDNode *N, unsigned MachineOpc,
5701                                    EVT VT, SDValue Op1) {
5702   SDVTList VTs = getVTList(VT);
5703   SDValue Ops[] = { Op1 };
5704   return SelectNodeTo(N, MachineOpc, VTs, Ops);
5705 }
5706
5707 SDNode *SelectionDAG::SelectNodeTo(SDNode *N, unsigned MachineOpc,
5708                                    EVT VT, SDValue Op1,
5709                                    SDValue Op2) {
5710   SDVTList VTs = getVTList(VT);
5711   SDValue Ops[] = { Op1, Op2 };
5712   return SelectNodeTo(N, MachineOpc, VTs, Ops);
5713 }
5714
5715 SDNode *SelectionDAG::SelectNodeTo(SDNode *N, unsigned MachineOpc,
5716                                    EVT VT, SDValue Op1,
5717                                    SDValue Op2, SDValue Op3) {
5718   SDVTList VTs = getVTList(VT);
5719   SDValue Ops[] = { Op1, Op2, Op3 };
5720   return SelectNodeTo(N, MachineOpc, VTs, Ops);
5721 }
5722
5723 SDNode *SelectionDAG::SelectNodeTo(SDNode *N, unsigned MachineOpc,
5724                                    EVT VT, ArrayRef<SDValue> Ops) {
5725   SDVTList VTs = getVTList(VT);
5726   return SelectNodeTo(N, MachineOpc, VTs, Ops);
5727 }
5728
5729 SDNode *SelectionDAG::SelectNodeTo(SDNode *N, unsigned MachineOpc,
5730                                    EVT VT1, EVT VT2, ArrayRef<SDValue> Ops) {
5731   SDVTList VTs = getVTList(VT1, VT2);
5732   return SelectNodeTo(N, MachineOpc, VTs, Ops);
5733 }
5734
5735 SDNode *SelectionDAG::SelectNodeTo(SDNode *N, unsigned MachineOpc,
5736                                    EVT VT1, EVT VT2) {
5737   SDVTList VTs = getVTList(VT1, VT2);
5738   return SelectNodeTo(N, MachineOpc, VTs, None);
5739 }
5740
5741 SDNode *SelectionDAG::SelectNodeTo(SDNode *N, unsigned MachineOpc,
5742                                    EVT VT1, EVT VT2, EVT VT3,
5743                                    ArrayRef<SDValue> Ops) {
5744   SDVTList VTs = getVTList(VT1, VT2, VT3);
5745   return SelectNodeTo(N, MachineOpc, VTs, Ops);
5746 }
5747
5748 SDNode *SelectionDAG::SelectNodeTo(SDNode *N, unsigned MachineOpc,
5749                                    EVT VT1, EVT VT2, EVT VT3, EVT VT4,
5750                                    ArrayRef<SDValue> Ops) {
5751   SDVTList VTs = getVTList(VT1, VT2, VT3, VT4);
5752   return SelectNodeTo(N, MachineOpc, VTs, Ops);
5753 }
5754
5755 SDNode *SelectionDAG::SelectNodeTo(SDNode *N, unsigned MachineOpc,
5756                                    EVT VT1, EVT VT2,
5757                                    SDValue Op1) {
5758   SDVTList VTs = getVTList(VT1, VT2);
5759   SDValue Ops[] = { Op1 };
5760   return SelectNodeTo(N, MachineOpc, VTs, Ops);
5761 }
5762
5763 SDNode *SelectionDAG::SelectNodeTo(SDNode *N, unsigned MachineOpc,
5764                                    EVT VT1, EVT VT2,
5765                                    SDValue Op1, SDValue Op2) {
5766   SDVTList VTs = getVTList(VT1, VT2);
5767   SDValue Ops[] = { Op1, Op2 };
5768   return SelectNodeTo(N, MachineOpc, VTs, Ops);
5769 }
5770
5771 SDNode *SelectionDAG::SelectNodeTo(SDNode *N, unsigned MachineOpc,
5772                                    EVT VT1, EVT VT2,
5773                                    SDValue Op1, SDValue Op2,
5774                                    SDValue Op3) {
5775   SDVTList VTs = getVTList(VT1, VT2);
5776   SDValue Ops[] = { Op1, Op2, Op3 };
5777   return SelectNodeTo(N, MachineOpc, VTs, Ops);
5778 }
5779
5780 SDNode *SelectionDAG::SelectNodeTo(SDNode *N, unsigned MachineOpc,
5781                                    EVT VT1, EVT VT2, EVT VT3,
5782                                    SDValue Op1, SDValue Op2,
5783                                    SDValue Op3) {
5784   SDVTList VTs = getVTList(VT1, VT2, VT3);
5785   SDValue Ops[] = { Op1, Op2, Op3 };
5786   return SelectNodeTo(N, MachineOpc, VTs, Ops);
5787 }
5788
5789 SDNode *SelectionDAG::SelectNodeTo(SDNode *N, unsigned MachineOpc,
5790                                    SDVTList VTs,ArrayRef<SDValue> Ops) {
5791   N = MorphNodeTo(N, ~MachineOpc, VTs, Ops);
5792   // Reset the NodeID to -1.
5793   N->setNodeId(-1);
5794   return N;
5795 }
5796
5797 /// UpdadeSDLocOnMergedSDNode - If the opt level is -O0 then it throws away
5798 /// the line number information on the merged node since it is not possible to
5799 /// preserve the information that operation is associated with multiple lines.
5800 /// This will make the debugger working better at -O0, were there is a higher
5801 /// probability having other instructions associated with that line.
5802 ///
5803 /// For IROrder, we keep the smaller of the two
5804 SDNode *SelectionDAG::UpdadeSDLocOnMergedSDNode(SDNode *N, SDLoc OLoc) {
5805   DebugLoc NLoc = N->getDebugLoc();
5806   if (NLoc && OptLevel == CodeGenOpt::None && OLoc.getDebugLoc() != NLoc) {
5807     N->setDebugLoc(DebugLoc());
5808   }
5809   unsigned Order = std::min(N->getIROrder(), OLoc.getIROrder());
5810   N->setIROrder(Order);
5811   return N;
5812 }
5813
5814 /// MorphNodeTo - This *mutates* the specified node to have the specified
5815 /// return type, opcode, and operands.
5816 ///
5817 /// Note that MorphNodeTo returns the resultant node.  If there is already a
5818 /// node of the specified opcode and operands, it returns that node instead of
5819 /// the current one.  Note that the SDLoc need not be the same.
5820 ///
5821 /// Using MorphNodeTo is faster than creating a new node and swapping it in
5822 /// with ReplaceAllUsesWith both because it often avoids allocating a new
5823 /// node, and because it doesn't require CSE recalculation for any of
5824 /// the node's users.
5825 ///
5826 /// However, note that MorphNodeTo recursively deletes dead nodes from the DAG.
5827 /// As a consequence it isn't appropriate to use from within the DAG combiner or
5828 /// the legalizer which maintain worklists that would need to be updated when
5829 /// deleting things.
5830 SDNode *SelectionDAG::MorphNodeTo(SDNode *N, unsigned Opc,
5831                                   SDVTList VTs, ArrayRef<SDValue> Ops) {
5832   unsigned NumOps = Ops.size();
5833   // If an identical node already exists, use it.
5834   void *IP = nullptr;
5835   if (VTs.VTs[VTs.NumVTs-1] != MVT::Glue) {
5836     FoldingSetNodeID ID;
5837     AddNodeIDNode(ID, Opc, VTs, Ops);
5838     if (SDNode *ON = FindNodeOrInsertPos(ID, N->getDebugLoc(), IP))
5839       return UpdadeSDLocOnMergedSDNode(ON, SDLoc(N));
5840   }
5841
5842   if (!RemoveNodeFromCSEMaps(N))
5843     IP = nullptr;
5844
5845   // Start the morphing.
5846   N->NodeType = Opc;
5847   N->ValueList = VTs.VTs;
5848   N->NumValues = VTs.NumVTs;
5849
5850   // Clear the operands list, updating used nodes to remove this from their
5851   // use list.  Keep track of any operands that become dead as a result.
5852   SmallPtrSet<SDNode*, 16> DeadNodeSet;
5853   for (SDNode::op_iterator I = N->op_begin(), E = N->op_end(); I != E; ) {
5854     SDUse &Use = *I++;
5855     SDNode *Used = Use.getNode();
5856     Use.set(SDValue());
5857     if (Used->use_empty())
5858       DeadNodeSet.insert(Used);
5859   }
5860
5861   if (MachineSDNode *MN = dyn_cast<MachineSDNode>(N)) {
5862     // Initialize the memory references information.
5863     MN->setMemRefs(nullptr, nullptr);
5864     // If NumOps is larger than the # of operands we can have in a
5865     // MachineSDNode, reallocate the operand list.
5866     if (NumOps > MN->NumOperands || !MN->OperandsNeedDelete) {
5867       if (MN->OperandsNeedDelete)
5868         delete[] MN->OperandList;
5869       if (NumOps > array_lengthof(MN->LocalOperands))
5870         // We're creating a final node that will live unmorphed for the
5871         // remainder of the current SelectionDAG iteration, so we can allocate
5872         // the operands directly out of a pool with no recycling metadata.
5873         MN->InitOperands(OperandAllocator.Allocate<SDUse>(NumOps),
5874                          Ops.data(), NumOps);
5875       else
5876         MN->InitOperands(MN->LocalOperands, Ops.data(), NumOps);
5877       MN->OperandsNeedDelete = false;
5878     } else
5879       MN->InitOperands(MN->OperandList, Ops.data(), NumOps);
5880   } else {
5881     // If NumOps is larger than the # of operands we currently have, reallocate
5882     // the operand list.
5883     if (NumOps > N->NumOperands) {
5884       if (N->OperandsNeedDelete)
5885         delete[] N->OperandList;
5886       N->InitOperands(new SDUse[NumOps], Ops.data(), NumOps);
5887       N->OperandsNeedDelete = true;
5888     } else
5889       N->InitOperands(N->OperandList, Ops.data(), NumOps);
5890   }
5891
5892   // Delete any nodes that are still dead after adding the uses for the
5893   // new operands.
5894   if (!DeadNodeSet.empty()) {
5895     SmallVector<SDNode *, 16> DeadNodes;
5896     for (SDNode *N : DeadNodeSet)
5897       if (N->use_empty())
5898         DeadNodes.push_back(N);
5899     RemoveDeadNodes(DeadNodes);
5900   }
5901
5902   if (IP)
5903     CSEMap.InsertNode(N, IP);   // Memoize the new node.
5904   return N;
5905 }
5906
5907
5908 /// getMachineNode - These are used for target selectors to create a new node
5909 /// with specified return type(s), MachineInstr opcode, and operands.
5910 ///
5911 /// Note that getMachineNode returns the resultant node.  If there is already a
5912 /// node of the specified opcode and operands, it returns that node instead of
5913 /// the current one.
5914 MachineSDNode *
5915 SelectionDAG::getMachineNode(unsigned Opcode, SDLoc dl, EVT VT) {
5916   SDVTList VTs = getVTList(VT);
5917   return getMachineNode(Opcode, dl, VTs, None);
5918 }
5919
5920 MachineSDNode *
5921 SelectionDAG::getMachineNode(unsigned Opcode, SDLoc dl, EVT VT, SDValue Op1) {
5922   SDVTList VTs = getVTList(VT);
5923   SDValue Ops[] = { Op1 };
5924   return getMachineNode(Opcode, dl, VTs, Ops);
5925 }
5926
5927 MachineSDNode *
5928 SelectionDAG::getMachineNode(unsigned Opcode, SDLoc dl, EVT VT,
5929                              SDValue Op1, SDValue Op2) {
5930   SDVTList VTs = getVTList(VT);
5931   SDValue Ops[] = { Op1, Op2 };
5932   return getMachineNode(Opcode, dl, VTs, Ops);
5933 }
5934
5935 MachineSDNode *
5936 SelectionDAG::getMachineNode(unsigned Opcode, SDLoc dl, EVT VT,
5937                              SDValue Op1, SDValue Op2, SDValue Op3) {
5938   SDVTList VTs = getVTList(VT);
5939   SDValue Ops[] = { Op1, Op2, Op3 };
5940   return getMachineNode(Opcode, dl, VTs, Ops);
5941 }
5942
5943 MachineSDNode *
5944 SelectionDAG::getMachineNode(unsigned Opcode, SDLoc dl, EVT VT,
5945                              ArrayRef<SDValue> Ops) {
5946   SDVTList VTs = getVTList(VT);
5947   return getMachineNode(Opcode, dl, VTs, Ops);
5948 }
5949
5950 MachineSDNode *
5951 SelectionDAG::getMachineNode(unsigned Opcode, SDLoc dl, EVT VT1, EVT VT2) {
5952   SDVTList VTs = getVTList(VT1, VT2);
5953   return getMachineNode(Opcode, dl, VTs, None);
5954 }
5955
5956 MachineSDNode *
5957 SelectionDAG::getMachineNode(unsigned Opcode, SDLoc dl,
5958                              EVT VT1, EVT VT2, SDValue Op1) {
5959   SDVTList VTs = getVTList(VT1, VT2);
5960   SDValue Ops[] = { Op1 };
5961   return getMachineNode(Opcode, dl, VTs, Ops);
5962 }
5963
5964 MachineSDNode *
5965 SelectionDAG::getMachineNode(unsigned Opcode, SDLoc dl,
5966                              EVT VT1, EVT VT2, SDValue Op1, SDValue Op2) {
5967   SDVTList VTs = getVTList(VT1, VT2);
5968   SDValue Ops[] = { Op1, Op2 };
5969   return getMachineNode(Opcode, dl, VTs, Ops);
5970 }
5971
5972 MachineSDNode *
5973 SelectionDAG::getMachineNode(unsigned Opcode, SDLoc dl,
5974                              EVT VT1, EVT VT2, SDValue Op1,
5975                              SDValue Op2, SDValue Op3) {
5976   SDVTList VTs = getVTList(VT1, VT2);
5977   SDValue Ops[] = { Op1, Op2, Op3 };
5978   return getMachineNode(Opcode, dl, VTs, Ops);
5979 }
5980
5981 MachineSDNode *
5982 SelectionDAG::getMachineNode(unsigned Opcode, SDLoc dl,
5983                              EVT VT1, EVT VT2,
5984                              ArrayRef<SDValue> Ops) {
5985   SDVTList VTs = getVTList(VT1, VT2);
5986   return getMachineNode(Opcode, dl, VTs, Ops);
5987 }
5988
5989 MachineSDNode *
5990 SelectionDAG::getMachineNode(unsigned Opcode, SDLoc dl,
5991                              EVT VT1, EVT VT2, EVT VT3,
5992                              SDValue Op1, SDValue Op2) {
5993   SDVTList VTs = getVTList(VT1, VT2, VT3);
5994   SDValue Ops[] = { Op1, Op2 };
5995   return getMachineNode(Opcode, dl, VTs, Ops);
5996 }
5997
5998 MachineSDNode *
5999 SelectionDAG::getMachineNode(unsigned Opcode, SDLoc dl,
6000                              EVT VT1, EVT VT2, EVT VT3,
6001                              SDValue Op1, SDValue Op2, SDValue Op3) {
6002   SDVTList VTs = getVTList(VT1, VT2, VT3);
6003   SDValue Ops[] = { Op1, Op2, Op3 };
6004   return getMachineNode(Opcode, dl, VTs, Ops);
6005 }
6006
6007 MachineSDNode *
6008 SelectionDAG::getMachineNode(unsigned Opcode, SDLoc dl,
6009                              EVT VT1, EVT VT2, EVT VT3,
6010                              ArrayRef<SDValue> Ops) {
6011   SDVTList VTs = getVTList(VT1, VT2, VT3);
6012   return getMachineNode(Opcode, dl, VTs, Ops);
6013 }
6014
6015 MachineSDNode *
6016 SelectionDAG::getMachineNode(unsigned Opcode, SDLoc dl, EVT VT1,
6017                              EVT VT2, EVT VT3, EVT VT4,
6018                              ArrayRef<SDValue> Ops) {
6019   SDVTList VTs = getVTList(VT1, VT2, VT3, VT4);
6020   return getMachineNode(Opcode, dl, VTs, Ops);
6021 }
6022
6023 MachineSDNode *
6024 SelectionDAG::getMachineNode(unsigned Opcode, SDLoc dl,
6025                              ArrayRef<EVT> ResultTys,
6026                              ArrayRef<SDValue> Ops) {
6027   SDVTList VTs = getVTList(ResultTys);
6028   return getMachineNode(Opcode, dl, VTs, Ops);
6029 }
6030
6031 MachineSDNode *
6032 SelectionDAG::getMachineNode(unsigned Opcode, SDLoc DL, SDVTList VTs,
6033                              ArrayRef<SDValue> OpsArray) {
6034   bool DoCSE = VTs.VTs[VTs.NumVTs-1] != MVT::Glue;
6035   MachineSDNode *N;
6036   void *IP = nullptr;
6037   const SDValue *Ops = OpsArray.data();
6038   unsigned NumOps = OpsArray.size();
6039
6040   if (DoCSE) {
6041     FoldingSetNodeID ID;
6042     AddNodeIDNode(ID, ~Opcode, VTs, OpsArray);
6043     IP = nullptr;
6044     if (SDNode *E = FindNodeOrInsertPos(ID, DL.getDebugLoc(), IP)) {
6045       return cast<MachineSDNode>(UpdadeSDLocOnMergedSDNode(E, DL));
6046     }
6047   }
6048
6049   // Allocate a new MachineSDNode.
6050   N = new (NodeAllocator) MachineSDNode(~Opcode, DL.getIROrder(),
6051                                         DL.getDebugLoc(), VTs);
6052
6053   // Initialize the operands list.
6054   if (NumOps > array_lengthof(N->LocalOperands))
6055     // We're creating a final node that will live unmorphed for the
6056     // remainder of the current SelectionDAG iteration, so we can allocate
6057     // the operands directly out of a pool with no recycling metadata.
6058     N->InitOperands(OperandAllocator.Allocate<SDUse>(NumOps),
6059                     Ops, NumOps);
6060   else
6061     N->InitOperands(N->LocalOperands, Ops, NumOps);
6062   N->OperandsNeedDelete = false;
6063
6064   if (DoCSE)
6065     CSEMap.InsertNode(N, IP);
6066
6067   InsertNode(N);
6068   return N;
6069 }
6070
6071 /// getTargetExtractSubreg - A convenience function for creating
6072 /// TargetOpcode::EXTRACT_SUBREG nodes.
6073 SDValue
6074 SelectionDAG::getTargetExtractSubreg(int SRIdx, SDLoc DL, EVT VT,
6075                                      SDValue Operand) {
6076   SDValue SRIdxVal = getTargetConstant(SRIdx, DL, MVT::i32);
6077   SDNode *Subreg = getMachineNode(TargetOpcode::EXTRACT_SUBREG, DL,
6078                                   VT, Operand, SRIdxVal);
6079   return SDValue(Subreg, 0);
6080 }
6081
6082 /// getTargetInsertSubreg - A convenience function for creating
6083 /// TargetOpcode::INSERT_SUBREG nodes.
6084 SDValue
6085 SelectionDAG::getTargetInsertSubreg(int SRIdx, SDLoc DL, EVT VT,
6086                                     SDValue Operand, SDValue Subreg) {
6087   SDValue SRIdxVal = getTargetConstant(SRIdx, DL, MVT::i32);
6088   SDNode *Result = getMachineNode(TargetOpcode::INSERT_SUBREG, DL,
6089                                   VT, Operand, Subreg, SRIdxVal);
6090   return SDValue(Result, 0);
6091 }
6092
6093 /// getNodeIfExists - Get the specified node if it's already available, or
6094 /// else return NULL.
6095 SDNode *SelectionDAG::getNodeIfExists(unsigned Opcode, SDVTList VTList,
6096                                       ArrayRef<SDValue> Ops,
6097                                       const SDNodeFlags *Flags) {
6098   if (VTList.VTs[VTList.NumVTs - 1] != MVT::Glue) {
6099     FoldingSetNodeID ID;
6100     AddNodeIDNode(ID, Opcode, VTList, Ops);
6101     AddNodeIDFlags(ID, Opcode, Flags);
6102     void *IP = nullptr;
6103     if (SDNode *E = FindNodeOrInsertPos(ID, DebugLoc(), IP))
6104       return E;
6105   }
6106   return nullptr;
6107 }
6108
6109 /// getDbgValue - Creates a SDDbgValue node.
6110 ///
6111 /// SDNode
6112 SDDbgValue *SelectionDAG::getDbgValue(MDNode *Var, MDNode *Expr, SDNode *N,
6113                                       unsigned R, bool IsIndirect, uint64_t Off,
6114                                       DebugLoc DL, unsigned O) {
6115   assert(cast<DILocalVariable>(Var)->isValidLocationForIntrinsic(DL) &&
6116          "Expected inlined-at fields to agree");
6117   return new (DbgInfo->getAlloc())
6118       SDDbgValue(Var, Expr, N, R, IsIndirect, Off, DL, O);
6119 }
6120
6121 /// Constant
6122 SDDbgValue *SelectionDAG::getConstantDbgValue(MDNode *Var, MDNode *Expr,
6123                                               const Value *C, uint64_t Off,
6124                                               DebugLoc DL, unsigned O) {
6125   assert(cast<DILocalVariable>(Var)->isValidLocationForIntrinsic(DL) &&
6126          "Expected inlined-at fields to agree");
6127   return new (DbgInfo->getAlloc()) SDDbgValue(Var, Expr, C, Off, DL, O);
6128 }
6129
6130 /// FrameIndex
6131 SDDbgValue *SelectionDAG::getFrameIndexDbgValue(MDNode *Var, MDNode *Expr,
6132                                                 unsigned FI, uint64_t Off,
6133                                                 DebugLoc DL, unsigned O) {
6134   assert(cast<DILocalVariable>(Var)->isValidLocationForIntrinsic(DL) &&
6135          "Expected inlined-at fields to agree");
6136   return new (DbgInfo->getAlloc()) SDDbgValue(Var, Expr, FI, Off, DL, O);
6137 }
6138
6139 namespace {
6140
6141 /// RAUWUpdateListener - Helper for ReplaceAllUsesWith - When the node
6142 /// pointed to by a use iterator is deleted, increment the use iterator
6143 /// so that it doesn't dangle.
6144 ///
6145 class RAUWUpdateListener : public SelectionDAG::DAGUpdateListener {
6146   SDNode::use_iterator &UI;
6147   SDNode::use_iterator &UE;
6148
6149   void NodeDeleted(SDNode *N, SDNode *E) override {
6150     // Increment the iterator as needed.
6151     while (UI != UE && N == *UI)
6152       ++UI;
6153   }
6154
6155 public:
6156   RAUWUpdateListener(SelectionDAG &d,
6157                      SDNode::use_iterator &ui,
6158                      SDNode::use_iterator &ue)
6159     : SelectionDAG::DAGUpdateListener(d), UI(ui), UE(ue) {}
6160 };
6161
6162 }
6163
6164 /// ReplaceAllUsesWith - Modify anything using 'From' to use 'To' instead.
6165 /// This can cause recursive merging of nodes in the DAG.
6166 ///
6167 /// This version assumes From has a single result value.
6168 ///
6169 void SelectionDAG::ReplaceAllUsesWith(SDValue FromN, SDValue To) {
6170   SDNode *From = FromN.getNode();
6171   assert(From->getNumValues() == 1 && FromN.getResNo() == 0 &&
6172          "Cannot replace with this method!");
6173   assert(From != To.getNode() && "Cannot replace uses of with self");
6174
6175   // Iterate over all the existing uses of From. New uses will be added
6176   // to the beginning of the use list, which we avoid visiting.
6177   // This specifically avoids visiting uses of From that arise while the
6178   // replacement is happening, because any such uses would be the result
6179   // of CSE: If an existing node looks like From after one of its operands
6180   // is replaced by To, we don't want to replace of all its users with To
6181   // too. See PR3018 for more info.
6182   SDNode::use_iterator UI = From->use_begin(), UE = From->use_end();
6183   RAUWUpdateListener Listener(*this, UI, UE);
6184   while (UI != UE) {
6185     SDNode *User = *UI;
6186
6187     // This node is about to morph, remove its old self from the CSE maps.
6188     RemoveNodeFromCSEMaps(User);
6189
6190     // A user can appear in a use list multiple times, and when this
6191     // happens the uses are usually next to each other in the list.
6192     // To help reduce the number of CSE recomputations, process all
6193     // the uses of this user that we can find this way.
6194     do {
6195       SDUse &Use = UI.getUse();
6196       ++UI;
6197       Use.set(To);
6198     } while (UI != UE && *UI == User);
6199
6200     // Now that we have modified User, add it back to the CSE maps.  If it
6201     // already exists there, recursively merge the results together.
6202     AddModifiedNodeToCSEMaps(User);
6203   }
6204
6205   // If we just RAUW'd the root, take note.
6206   if (FromN == getRoot())
6207     setRoot(To);
6208 }
6209
6210 /// ReplaceAllUsesWith - Modify anything using 'From' to use 'To' instead.
6211 /// This can cause recursive merging of nodes in the DAG.
6212 ///
6213 /// This version assumes that for each value of From, there is a
6214 /// corresponding value in To in the same position with the same type.
6215 ///
6216 void SelectionDAG::ReplaceAllUsesWith(SDNode *From, SDNode *To) {
6217 #ifndef NDEBUG
6218   for (unsigned i = 0, e = From->getNumValues(); i != e; ++i)
6219     assert((!From->hasAnyUseOfValue(i) ||
6220             From->getValueType(i) == To->getValueType(i)) &&
6221            "Cannot use this version of ReplaceAllUsesWith!");
6222 #endif
6223
6224   // Handle the trivial case.
6225   if (From == To)
6226     return;
6227
6228   // Iterate over just the existing users of From. See the comments in
6229   // the ReplaceAllUsesWith above.
6230   SDNode::use_iterator UI = From->use_begin(), UE = From->use_end();
6231   RAUWUpdateListener Listener(*this, UI, UE);
6232   while (UI != UE) {
6233     SDNode *User = *UI;
6234
6235     // This node is about to morph, remove its old self from the CSE maps.
6236     RemoveNodeFromCSEMaps(User);
6237
6238     // A user can appear in a use list multiple times, and when this
6239     // happens the uses are usually next to each other in the list.
6240     // To help reduce the number of CSE recomputations, process all
6241     // the uses of this user that we can find this way.
6242     do {
6243       SDUse &Use = UI.getUse();
6244       ++UI;
6245       Use.setNode(To);
6246     } while (UI != UE && *UI == User);
6247
6248     // Now that we have modified User, add it back to the CSE maps.  If it
6249     // already exists there, recursively merge the results together.
6250     AddModifiedNodeToCSEMaps(User);
6251   }
6252
6253   // If we just RAUW'd the root, take note.
6254   if (From == getRoot().getNode())
6255     setRoot(SDValue(To, getRoot().getResNo()));
6256 }
6257
6258 /// ReplaceAllUsesWith - Modify anything using 'From' to use 'To' instead.
6259 /// This can cause recursive merging of nodes in the DAG.
6260 ///
6261 /// This version can replace From with any result values.  To must match the
6262 /// number and types of values returned by From.
6263 void SelectionDAG::ReplaceAllUsesWith(SDNode *From, const SDValue *To) {
6264   if (From->getNumValues() == 1)  // Handle the simple case efficiently.
6265     return ReplaceAllUsesWith(SDValue(From, 0), To[0]);
6266
6267   // Iterate over just the existing users of From. See the comments in
6268   // the ReplaceAllUsesWith above.
6269   SDNode::use_iterator UI = From->use_begin(), UE = From->use_end();
6270   RAUWUpdateListener Listener(*this, UI, UE);
6271   while (UI != UE) {
6272     SDNode *User = *UI;
6273
6274     // This node is about to morph, remove its old self from the CSE maps.
6275     RemoveNodeFromCSEMaps(User);
6276
6277     // A user can appear in a use list multiple times, and when this
6278     // happens the uses are usually next to each other in the list.
6279     // To help reduce the number of CSE recomputations, process all
6280     // the uses of this user that we can find this way.
6281     do {
6282       SDUse &Use = UI.getUse();
6283       const SDValue &ToOp = To[Use.getResNo()];
6284       ++UI;
6285       Use.set(ToOp);
6286     } while (UI != UE && *UI == User);
6287
6288     // Now that we have modified User, add it back to the CSE maps.  If it
6289     // already exists there, recursively merge the results together.
6290     AddModifiedNodeToCSEMaps(User);
6291   }
6292
6293   // If we just RAUW'd the root, take note.
6294   if (From == getRoot().getNode())
6295     setRoot(SDValue(To[getRoot().getResNo()]));
6296 }
6297
6298 /// ReplaceAllUsesOfValueWith - Replace any uses of From with To, leaving
6299 /// uses of other values produced by From.getNode() alone.  The Deleted
6300 /// vector is handled the same way as for ReplaceAllUsesWith.
6301 void SelectionDAG::ReplaceAllUsesOfValueWith(SDValue From, SDValue To){
6302   // Handle the really simple, really trivial case efficiently.
6303   if (From == To) return;
6304
6305   // Handle the simple, trivial, case efficiently.
6306   if (From.getNode()->getNumValues() == 1) {
6307     ReplaceAllUsesWith(From, To);
6308     return;
6309   }
6310
6311   // Iterate over just the existing users of From. See the comments in
6312   // the ReplaceAllUsesWith above.
6313   SDNode::use_iterator UI = From.getNode()->use_begin(),
6314                        UE = From.getNode()->use_end();
6315   RAUWUpdateListener Listener(*this, UI, UE);
6316   while (UI != UE) {
6317     SDNode *User = *UI;
6318     bool UserRemovedFromCSEMaps = false;
6319
6320     // A user can appear in a use list multiple times, and when this
6321     // happens the uses are usually next to each other in the list.
6322     // To help reduce the number of CSE recomputations, process all
6323     // the uses of this user that we can find this way.
6324     do {
6325       SDUse &Use = UI.getUse();
6326
6327       // Skip uses of different values from the same node.
6328       if (Use.getResNo() != From.getResNo()) {
6329         ++UI;
6330         continue;
6331       }
6332
6333       // If this node hasn't been modified yet, it's still in the CSE maps,
6334       // so remove its old self from the CSE maps.
6335       if (!UserRemovedFromCSEMaps) {
6336         RemoveNodeFromCSEMaps(User);
6337         UserRemovedFromCSEMaps = true;
6338       }
6339
6340       ++UI;
6341       Use.set(To);
6342     } while (UI != UE && *UI == User);
6343
6344     // We are iterating over all uses of the From node, so if a use
6345     // doesn't use the specific value, no changes are made.
6346     if (!UserRemovedFromCSEMaps)
6347       continue;
6348
6349     // Now that we have modified User, add it back to the CSE maps.  If it
6350     // already exists there, recursively merge the results together.
6351     AddModifiedNodeToCSEMaps(User);
6352   }
6353
6354   // If we just RAUW'd the root, take note.
6355   if (From == getRoot())
6356     setRoot(To);
6357 }
6358
6359 namespace {
6360   /// UseMemo - This class is used by SelectionDAG::ReplaceAllUsesOfValuesWith
6361   /// to record information about a use.
6362   struct UseMemo {
6363     SDNode *User;
6364     unsigned Index;
6365     SDUse *Use;
6366   };
6367
6368   /// operator< - Sort Memos by User.
6369   bool operator<(const UseMemo &L, const UseMemo &R) {
6370     return (intptr_t)L.User < (intptr_t)R.User;
6371   }
6372 }
6373
6374 /// ReplaceAllUsesOfValuesWith - Replace any uses of From with To, leaving
6375 /// uses of other values produced by From.getNode() alone.  The same value
6376 /// may appear in both the From and To list.  The Deleted vector is
6377 /// handled the same way as for ReplaceAllUsesWith.
6378 void SelectionDAG::ReplaceAllUsesOfValuesWith(const SDValue *From,
6379                                               const SDValue *To,
6380                                               unsigned Num){
6381   // Handle the simple, trivial case efficiently.
6382   if (Num == 1)
6383     return ReplaceAllUsesOfValueWith(*From, *To);
6384
6385   // Read up all the uses and make records of them. This helps
6386   // processing new uses that are introduced during the
6387   // replacement process.
6388   SmallVector<UseMemo, 4> Uses;
6389   for (unsigned i = 0; i != Num; ++i) {
6390     unsigned FromResNo = From[i].getResNo();
6391     SDNode *FromNode = From[i].getNode();
6392     for (SDNode::use_iterator UI = FromNode->use_begin(),
6393          E = FromNode->use_end(); UI != E; ++UI) {
6394       SDUse &Use = UI.getUse();
6395       if (Use.getResNo() == FromResNo) {
6396         UseMemo Memo = { *UI, i, &Use };
6397         Uses.push_back(Memo);
6398       }
6399     }
6400   }
6401
6402   // Sort the uses, so that all the uses from a given User are together.
6403   std::sort(Uses.begin(), Uses.end());
6404
6405   for (unsigned UseIndex = 0, UseIndexEnd = Uses.size();
6406        UseIndex != UseIndexEnd; ) {
6407     // We know that this user uses some value of From.  If it is the right
6408     // value, update it.
6409     SDNode *User = Uses[UseIndex].User;
6410
6411     // This node is about to morph, remove its old self from the CSE maps.
6412     RemoveNodeFromCSEMaps(User);
6413
6414     // The Uses array is sorted, so all the uses for a given User
6415     // are next to each other in the list.
6416     // To help reduce the number of CSE recomputations, process all
6417     // the uses of this user that we can find this way.
6418     do {
6419       unsigned i = Uses[UseIndex].Index;
6420       SDUse &Use = *Uses[UseIndex].Use;
6421       ++UseIndex;
6422
6423       Use.set(To[i]);
6424     } while (UseIndex != UseIndexEnd && Uses[UseIndex].User == User);
6425
6426     // Now that we have modified User, add it back to the CSE maps.  If it
6427     // already exists there, recursively merge the results together.
6428     AddModifiedNodeToCSEMaps(User);
6429   }
6430 }
6431
6432 /// AssignTopologicalOrder - Assign a unique node id for each node in the DAG
6433 /// based on their topological order. It returns the maximum id and a vector
6434 /// of the SDNodes* in assigned order by reference.
6435 unsigned SelectionDAG::AssignTopologicalOrder() {
6436
6437   unsigned DAGSize = 0;
6438
6439   // SortedPos tracks the progress of the algorithm. Nodes before it are
6440   // sorted, nodes after it are unsorted. When the algorithm completes
6441   // it is at the end of the list.
6442   allnodes_iterator SortedPos = allnodes_begin();
6443
6444   // Visit all the nodes. Move nodes with no operands to the front of
6445   // the list immediately. Annotate nodes that do have operands with their
6446   // operand count. Before we do this, the Node Id fields of the nodes
6447   // may contain arbitrary values. After, the Node Id fields for nodes
6448   // before SortedPos will contain the topological sort index, and the
6449   // Node Id fields for nodes At SortedPos and after will contain the
6450   // count of outstanding operands.
6451   for (allnodes_iterator I = allnodes_begin(),E = allnodes_end(); I != E; ) {
6452     SDNode *N = I++;
6453     checkForCycles(N, this);
6454     unsigned Degree = N->getNumOperands();
6455     if (Degree == 0) {
6456       // A node with no uses, add it to the result array immediately.
6457       N->setNodeId(DAGSize++);
6458       allnodes_iterator Q = N;
6459       if (Q != SortedPos)
6460         SortedPos = AllNodes.insert(SortedPos, AllNodes.remove(Q));
6461       assert(SortedPos != AllNodes.end() && "Overran node list");
6462       ++SortedPos;
6463     } else {
6464       // Temporarily use the Node Id as scratch space for the degree count.
6465       N->setNodeId(Degree);
6466     }
6467   }
6468
6469   // Visit all the nodes. As we iterate, move nodes into sorted order,
6470   // such that by the time the end is reached all nodes will be sorted.
6471   for (SDNode &Node : allnodes()) {
6472     SDNode *N = &Node;
6473     checkForCycles(N, this);
6474     // N is in sorted position, so all its uses have one less operand
6475     // that needs to be sorted.
6476     for (SDNode::use_iterator UI = N->use_begin(), UE = N->use_end();
6477          UI != UE; ++UI) {
6478       SDNode *P = *UI;
6479       unsigned Degree = P->getNodeId();
6480       assert(Degree != 0 && "Invalid node degree");
6481       --Degree;
6482       if (Degree == 0) {
6483         // All of P's operands are sorted, so P may sorted now.
6484         P->setNodeId(DAGSize++);
6485         if (P != SortedPos)
6486           SortedPos = AllNodes.insert(SortedPos, AllNodes.remove(P));
6487         assert(SortedPos != AllNodes.end() && "Overran node list");
6488         ++SortedPos;
6489       } else {
6490         // Update P's outstanding operand count.
6491         P->setNodeId(Degree);
6492       }
6493     }
6494     if (&Node == SortedPos) {
6495 #ifndef NDEBUG
6496       allnodes_iterator I = N;
6497       SDNode *S = ++I;
6498       dbgs() << "Overran sorted position:\n";
6499       S->dumprFull(this); dbgs() << "\n";
6500       dbgs() << "Checking if this is due to cycles\n";
6501       checkForCycles(this, true);
6502 #endif
6503       llvm_unreachable(nullptr);
6504     }
6505   }
6506
6507   assert(SortedPos == AllNodes.end() &&
6508          "Topological sort incomplete!");
6509   assert(AllNodes.front().getOpcode() == ISD::EntryToken &&
6510          "First node in topological sort is not the entry token!");
6511   assert(AllNodes.front().getNodeId() == 0 &&
6512          "First node in topological sort has non-zero id!");
6513   assert(AllNodes.front().getNumOperands() == 0 &&
6514          "First node in topological sort has operands!");
6515   assert(AllNodes.back().getNodeId() == (int)DAGSize-1 &&
6516          "Last node in topologic sort has unexpected id!");
6517   assert(AllNodes.back().use_empty() &&
6518          "Last node in topologic sort has users!");
6519   assert(DAGSize == allnodes_size() && "Node count mismatch!");
6520   return DAGSize;
6521 }
6522
6523 /// AddDbgValue - Add a dbg_value SDNode. If SD is non-null that means the
6524 /// value is produced by SD.
6525 void SelectionDAG::AddDbgValue(SDDbgValue *DB, SDNode *SD, bool isParameter) {
6526   if (SD) {
6527     assert(DbgInfo->getSDDbgValues(SD).empty() || SD->getHasDebugValue());
6528     SD->setHasDebugValue(true);
6529   }
6530   DbgInfo->add(DB, SD, isParameter);
6531 }
6532
6533 /// TransferDbgValues - Transfer SDDbgValues.
6534 void SelectionDAG::TransferDbgValues(SDValue From, SDValue To) {
6535   if (From == To || !From.getNode()->getHasDebugValue())
6536     return;
6537   SDNode *FromNode = From.getNode();
6538   SDNode *ToNode = To.getNode();
6539   ArrayRef<SDDbgValue *> DVs = GetDbgValues(FromNode);
6540   SmallVector<SDDbgValue *, 2> ClonedDVs;
6541   for (ArrayRef<SDDbgValue *>::iterator I = DVs.begin(), E = DVs.end();
6542        I != E; ++I) {
6543     SDDbgValue *Dbg = *I;
6544     if (Dbg->getKind() == SDDbgValue::SDNODE) {
6545       SDDbgValue *Clone =
6546           getDbgValue(Dbg->getVariable(), Dbg->getExpression(), ToNode,
6547                       To.getResNo(), Dbg->isIndirect(), Dbg->getOffset(),
6548                       Dbg->getDebugLoc(), Dbg->getOrder());
6549       ClonedDVs.push_back(Clone);
6550     }
6551   }
6552   for (SmallVectorImpl<SDDbgValue *>::iterator I = ClonedDVs.begin(),
6553          E = ClonedDVs.end(); I != E; ++I)
6554     AddDbgValue(*I, ToNode, false);
6555 }
6556
6557 //===----------------------------------------------------------------------===//
6558 //                              SDNode Class
6559 //===----------------------------------------------------------------------===//
6560
6561 HandleSDNode::~HandleSDNode() {
6562   DropOperands();
6563 }
6564
6565 GlobalAddressSDNode::GlobalAddressSDNode(unsigned Opc, unsigned Order,
6566                                          DebugLoc DL, const GlobalValue *GA,
6567                                          EVT VT, int64_t o, unsigned char TF)
6568   : SDNode(Opc, Order, DL, getSDVTList(VT)), Offset(o), TargetFlags(TF) {
6569   TheGlobal = GA;
6570 }
6571
6572 AddrSpaceCastSDNode::AddrSpaceCastSDNode(unsigned Order, DebugLoc dl, EVT VT,
6573                                          SDValue X, unsigned SrcAS,
6574                                          unsigned DestAS)
6575  : UnarySDNode(ISD::ADDRSPACECAST, Order, dl, getSDVTList(VT), X),
6576    SrcAddrSpace(SrcAS), DestAddrSpace(DestAS) {}
6577
6578 MemSDNode::MemSDNode(unsigned Opc, unsigned Order, DebugLoc dl, SDVTList VTs,
6579                      EVT memvt, MachineMemOperand *mmo)
6580  : SDNode(Opc, Order, dl, VTs), MemoryVT(memvt), MMO(mmo) {
6581   SubclassData = encodeMemSDNodeFlags(0, ISD::UNINDEXED, MMO->isVolatile(),
6582                                       MMO->isNonTemporal(), MMO->isInvariant());
6583   assert(isVolatile() == MMO->isVolatile() && "Volatile encoding error!");
6584   assert(isNonTemporal() == MMO->isNonTemporal() &&
6585          "Non-temporal encoding error!");
6586   // We check here that the size of the memory operand fits within the size of
6587   // the MMO. This is because the MMO might indicate only a possible address
6588   // range instead of specifying the affected memory addresses precisely.
6589   assert(memvt.getStoreSize() <= MMO->getSize() && "Size mismatch!");
6590 }
6591
6592 MemSDNode::MemSDNode(unsigned Opc, unsigned Order, DebugLoc dl, SDVTList VTs,
6593                      ArrayRef<SDValue> Ops, EVT memvt, MachineMemOperand *mmo)
6594    : SDNode(Opc, Order, dl, VTs, Ops),
6595      MemoryVT(memvt), MMO(mmo) {
6596   SubclassData = encodeMemSDNodeFlags(0, ISD::UNINDEXED, MMO->isVolatile(),
6597                                       MMO->isNonTemporal(), MMO->isInvariant());
6598   assert(isVolatile() == MMO->isVolatile() && "Volatile encoding error!");
6599   assert(memvt.getStoreSize() <= MMO->getSize() && "Size mismatch!");
6600 }
6601
6602 /// Profile - Gather unique data for the node.
6603 ///
6604 void SDNode::Profile(FoldingSetNodeID &ID) const {
6605   AddNodeIDNode(ID, this);
6606 }
6607
6608 namespace {
6609   struct EVTArray {
6610     std::vector<EVT> VTs;
6611
6612     EVTArray() {
6613       VTs.reserve(MVT::LAST_VALUETYPE);
6614       for (unsigned i = 0; i < MVT::LAST_VALUETYPE; ++i)
6615         VTs.push_back(MVT((MVT::SimpleValueType)i));
6616     }
6617   };
6618 }
6619
6620 static ManagedStatic<std::set<EVT, EVT::compareRawBits> > EVTs;
6621 static ManagedStatic<EVTArray> SimpleVTArray;
6622 static ManagedStatic<sys::SmartMutex<true> > VTMutex;
6623
6624 /// getValueTypeList - Return a pointer to the specified value type.
6625 ///
6626 const EVT *SDNode::getValueTypeList(EVT VT) {
6627   if (VT.isExtended()) {
6628     sys::SmartScopedLock<true> Lock(*VTMutex);
6629     return &(*EVTs->insert(VT).first);
6630   } else {
6631     assert(VT.getSimpleVT() < MVT::LAST_VALUETYPE &&
6632            "Value type out of range!");
6633     return &SimpleVTArray->VTs[VT.getSimpleVT().SimpleTy];
6634   }
6635 }
6636
6637 /// hasNUsesOfValue - Return true if there are exactly NUSES uses of the
6638 /// indicated value.  This method ignores uses of other values defined by this
6639 /// operation.
6640 bool SDNode::hasNUsesOfValue(unsigned NUses, unsigned Value) const {
6641   assert(Value < getNumValues() && "Bad value!");
6642
6643   // TODO: Only iterate over uses of a given value of the node
6644   for (SDNode::use_iterator UI = use_begin(), E = use_end(); UI != E; ++UI) {
6645     if (UI.getUse().getResNo() == Value) {
6646       if (NUses == 0)
6647         return false;
6648       --NUses;
6649     }
6650   }
6651
6652   // Found exactly the right number of uses?
6653   return NUses == 0;
6654 }
6655
6656
6657 /// hasAnyUseOfValue - Return true if there are any use of the indicated
6658 /// value. This method ignores uses of other values defined by this operation.
6659 bool SDNode::hasAnyUseOfValue(unsigned Value) const {
6660   assert(Value < getNumValues() && "Bad value!");
6661
6662   for (SDNode::use_iterator UI = use_begin(), E = use_end(); UI != E; ++UI)
6663     if (UI.getUse().getResNo() == Value)
6664       return true;
6665
6666   return false;
6667 }
6668
6669
6670 /// isOnlyUserOf - Return true if this node is the only use of N.
6671 ///
6672 bool SDNode::isOnlyUserOf(const SDNode *N) const {
6673   bool Seen = false;
6674   for (SDNode::use_iterator I = N->use_begin(), E = N->use_end(); I != E; ++I) {
6675     SDNode *User = *I;
6676     if (User == this)
6677       Seen = true;
6678     else
6679       return false;
6680   }
6681
6682   return Seen;
6683 }
6684
6685 /// isOperand - Return true if this node is an operand of N.
6686 ///
6687 bool SDValue::isOperandOf(const SDNode *N) const {
6688   for (const SDValue &Op : N->op_values())
6689     if (*this == Op)
6690       return true;
6691   return false;
6692 }
6693
6694 bool SDNode::isOperandOf(const SDNode *N) const {
6695   for (const SDValue &Op : N->op_values())
6696     if (this == Op.getNode())
6697       return true;
6698   return false;
6699 }
6700
6701 /// reachesChainWithoutSideEffects - Return true if this operand (which must
6702 /// be a chain) reaches the specified operand without crossing any
6703 /// side-effecting instructions on any chain path.  In practice, this looks
6704 /// through token factors and non-volatile loads.  In order to remain efficient,
6705 /// this only looks a couple of nodes in, it does not do an exhaustive search.
6706 bool SDValue::reachesChainWithoutSideEffects(SDValue Dest,
6707                                                unsigned Depth) const {
6708   if (*this == Dest) return true;
6709
6710   // Don't search too deeply, we just want to be able to see through
6711   // TokenFactor's etc.
6712   if (Depth == 0) return false;
6713
6714   // If this is a token factor, all inputs to the TF happen in parallel.  If any
6715   // of the operands of the TF does not reach dest, then we cannot do the xform.
6716   if (getOpcode() == ISD::TokenFactor) {
6717     for (unsigned i = 0, e = getNumOperands(); i != e; ++i)
6718       if (!getOperand(i).reachesChainWithoutSideEffects(Dest, Depth-1))
6719         return false;
6720     return true;
6721   }
6722
6723   // Loads don't have side effects, look through them.
6724   if (LoadSDNode *Ld = dyn_cast<LoadSDNode>(*this)) {
6725     if (!Ld->isVolatile())
6726       return Ld->getChain().reachesChainWithoutSideEffects(Dest, Depth-1);
6727   }
6728   return false;
6729 }
6730
6731 /// hasPredecessor - Return true if N is a predecessor of this node.
6732 /// N is either an operand of this node, or can be reached by recursively
6733 /// traversing up the operands.
6734 /// NOTE: This is an expensive method. Use it carefully.
6735 bool SDNode::hasPredecessor(const SDNode *N) const {
6736   SmallPtrSet<const SDNode *, 32> Visited;
6737   SmallVector<const SDNode *, 16> Worklist;
6738   return hasPredecessorHelper(N, Visited, Worklist);
6739 }
6740
6741 bool
6742 SDNode::hasPredecessorHelper(const SDNode *N,
6743                              SmallPtrSetImpl<const SDNode *> &Visited,
6744                              SmallVectorImpl<const SDNode *> &Worklist) const {
6745   if (Visited.empty()) {
6746     Worklist.push_back(this);
6747   } else {
6748     // Take a look in the visited set. If we've already encountered this node
6749     // we needn't search further.
6750     if (Visited.count(N))
6751       return true;
6752   }
6753
6754   // Haven't visited N yet. Continue the search.
6755   while (!Worklist.empty()) {
6756     const SDNode *M = Worklist.pop_back_val();
6757     for (const SDValue &OpV : M->op_values()) {
6758       SDNode *Op = OpV.getNode();
6759       if (Visited.insert(Op).second)
6760         Worklist.push_back(Op);
6761       if (Op == N)
6762         return true;
6763     }
6764   }
6765
6766   return false;
6767 }
6768
6769 uint64_t SDNode::getConstantOperandVal(unsigned Num) const {
6770   assert(Num < NumOperands && "Invalid child # of SDNode!");
6771   return cast<ConstantSDNode>(OperandList[Num])->getZExtValue();
6772 }
6773
6774 SDValue SelectionDAG::UnrollVectorOp(SDNode *N, unsigned ResNE) {
6775   assert(N->getNumValues() == 1 &&
6776          "Can't unroll a vector with multiple results!");
6777
6778   EVT VT = N->getValueType(0);
6779   unsigned NE = VT.getVectorNumElements();
6780   EVT EltVT = VT.getVectorElementType();
6781   SDLoc dl(N);
6782
6783   SmallVector<SDValue, 8> Scalars;
6784   SmallVector<SDValue, 4> Operands(N->getNumOperands());
6785
6786   // If ResNE is 0, fully unroll the vector op.
6787   if (ResNE == 0)
6788     ResNE = NE;
6789   else if (NE > ResNE)
6790     NE = ResNE;
6791
6792   unsigned i;
6793   for (i= 0; i != NE; ++i) {
6794     for (unsigned j = 0, e = N->getNumOperands(); j != e; ++j) {
6795       SDValue Operand = N->getOperand(j);
6796       EVT OperandVT = Operand.getValueType();
6797       if (OperandVT.isVector()) {
6798         // A vector operand; extract a single element.
6799         EVT OperandEltVT = OperandVT.getVectorElementType();
6800         Operands[j] =
6801             getNode(ISD::EXTRACT_VECTOR_ELT, dl, OperandEltVT, Operand,
6802                     getConstant(i, dl, TLI->getVectorIdxTy(getDataLayout())));
6803       } else {
6804         // A scalar operand; just use it as is.
6805         Operands[j] = Operand;
6806       }
6807     }
6808
6809     switch (N->getOpcode()) {
6810     default:
6811       Scalars.push_back(getNode(N->getOpcode(), dl, EltVT, Operands));
6812       break;
6813     case ISD::VSELECT:
6814       Scalars.push_back(getNode(ISD::SELECT, dl, EltVT, Operands));
6815       break;
6816     case ISD::SHL:
6817     case ISD::SRA:
6818     case ISD::SRL:
6819     case ISD::ROTL:
6820     case ISD::ROTR:
6821       Scalars.push_back(getNode(N->getOpcode(), dl, EltVT, Operands[0],
6822                                getShiftAmountOperand(Operands[0].getValueType(),
6823                                                      Operands[1])));
6824       break;
6825     case ISD::SIGN_EXTEND_INREG:
6826     case ISD::FP_ROUND_INREG: {
6827       EVT ExtVT = cast<VTSDNode>(Operands[1])->getVT().getVectorElementType();
6828       Scalars.push_back(getNode(N->getOpcode(), dl, EltVT,
6829                                 Operands[0],
6830                                 getValueType(ExtVT)));
6831     }
6832     }
6833   }
6834
6835   for (; i < ResNE; ++i)
6836     Scalars.push_back(getUNDEF(EltVT));
6837
6838   return getNode(ISD::BUILD_VECTOR, dl,
6839                  EVT::getVectorVT(*getContext(), EltVT, ResNE), Scalars);
6840 }
6841
6842
6843 /// isConsecutiveLoad - Return true if LD is loading 'Bytes' bytes from a
6844 /// location that is 'Dist' units away from the location that the 'Base' load
6845 /// is loading from.
6846 bool SelectionDAG::isConsecutiveLoad(LoadSDNode *LD, LoadSDNode *Base,
6847                                      unsigned Bytes, int Dist) const {
6848   if (LD->getChain() != Base->getChain())
6849     return false;
6850   EVT VT = LD->getValueType(0);
6851   if (VT.getSizeInBits() / 8 != Bytes)
6852     return false;
6853
6854   SDValue Loc = LD->getOperand(1);
6855   SDValue BaseLoc = Base->getOperand(1);
6856   if (Loc.getOpcode() == ISD::FrameIndex) {
6857     if (BaseLoc.getOpcode() != ISD::FrameIndex)
6858       return false;
6859     const MachineFrameInfo *MFI = getMachineFunction().getFrameInfo();
6860     int FI  = cast<FrameIndexSDNode>(Loc)->getIndex();
6861     int BFI = cast<FrameIndexSDNode>(BaseLoc)->getIndex();
6862     int FS  = MFI->getObjectSize(FI);
6863     int BFS = MFI->getObjectSize(BFI);
6864     if (FS != BFS || FS != (int)Bytes) return false;
6865     return MFI->getObjectOffset(FI) == (MFI->getObjectOffset(BFI) + Dist*Bytes);
6866   }
6867
6868   // Handle X + C.
6869   if (isBaseWithConstantOffset(Loc)) {
6870     int64_t LocOffset = cast<ConstantSDNode>(Loc.getOperand(1))->getSExtValue();
6871     if (Loc.getOperand(0) == BaseLoc) {
6872       // If the base location is a simple address with no offset itself, then
6873       // the second load's first add operand should be the base address.
6874       if (LocOffset == Dist * (int)Bytes)
6875         return true;
6876     } else if (isBaseWithConstantOffset(BaseLoc)) {
6877       // The base location itself has an offset, so subtract that value from the
6878       // second load's offset before comparing to distance * size.
6879       int64_t BOffset =
6880         cast<ConstantSDNode>(BaseLoc.getOperand(1))->getSExtValue();
6881       if (Loc.getOperand(0) == BaseLoc.getOperand(0)) {
6882         if ((LocOffset - BOffset) == Dist * (int)Bytes)
6883           return true;
6884       }
6885     }
6886   }
6887   const GlobalValue *GV1 = nullptr;
6888   const GlobalValue *GV2 = nullptr;
6889   int64_t Offset1 = 0;
6890   int64_t Offset2 = 0;
6891   bool isGA1 = TLI->isGAPlusOffset(Loc.getNode(), GV1, Offset1);
6892   bool isGA2 = TLI->isGAPlusOffset(BaseLoc.getNode(), GV2, Offset2);
6893   if (isGA1 && isGA2 && GV1 == GV2)
6894     return Offset1 == (Offset2 + Dist*Bytes);
6895   return false;
6896 }
6897
6898
6899 /// InferPtrAlignment - Infer alignment of a load / store address. Return 0 if
6900 /// it cannot be inferred.
6901 unsigned SelectionDAG::InferPtrAlignment(SDValue Ptr) const {
6902   // If this is a GlobalAddress + cst, return the alignment.
6903   const GlobalValue *GV;
6904   int64_t GVOffset = 0;
6905   if (TLI->isGAPlusOffset(Ptr.getNode(), GV, GVOffset)) {
6906     unsigned PtrWidth = getDataLayout().getPointerTypeSizeInBits(GV->getType());
6907     APInt KnownZero(PtrWidth, 0), KnownOne(PtrWidth, 0);
6908     llvm::computeKnownBits(const_cast<GlobalValue *>(GV), KnownZero, KnownOne,
6909                            getDataLayout());
6910     unsigned AlignBits = KnownZero.countTrailingOnes();
6911     unsigned Align = AlignBits ? 1 << std::min(31U, AlignBits) : 0;
6912     if (Align)
6913       return MinAlign(Align, GVOffset);
6914   }
6915
6916   // If this is a direct reference to a stack slot, use information about the
6917   // stack slot's alignment.
6918   int FrameIdx = 1 << 31;
6919   int64_t FrameOffset = 0;
6920   if (FrameIndexSDNode *FI = dyn_cast<FrameIndexSDNode>(Ptr)) {
6921     FrameIdx = FI->getIndex();
6922   } else if (isBaseWithConstantOffset(Ptr) &&
6923              isa<FrameIndexSDNode>(Ptr.getOperand(0))) {
6924     // Handle FI+Cst
6925     FrameIdx = cast<FrameIndexSDNode>(Ptr.getOperand(0))->getIndex();
6926     FrameOffset = Ptr.getConstantOperandVal(1);
6927   }
6928
6929   if (FrameIdx != (1 << 31)) {
6930     const MachineFrameInfo &MFI = *getMachineFunction().getFrameInfo();
6931     unsigned FIInfoAlign = MinAlign(MFI.getObjectAlignment(FrameIdx),
6932                                     FrameOffset);
6933     return FIInfoAlign;
6934   }
6935
6936   return 0;
6937 }
6938
6939 /// GetSplitDestVTs - Compute the VTs needed for the low/hi parts of a type
6940 /// which is split (or expanded) into two not necessarily identical pieces.
6941 std::pair<EVT, EVT> SelectionDAG::GetSplitDestVTs(const EVT &VT) const {
6942   // Currently all types are split in half.
6943   EVT LoVT, HiVT;
6944   if (!VT.isVector()) {
6945     LoVT = HiVT = TLI->getTypeToTransformTo(*getContext(), VT);
6946   } else {
6947     unsigned NumElements = VT.getVectorNumElements();
6948     assert(!(NumElements & 1) && "Splitting vector, but not in half!");
6949     LoVT = HiVT = EVT::getVectorVT(*getContext(), VT.getVectorElementType(),
6950                                    NumElements/2);
6951   }
6952   return std::make_pair(LoVT, HiVT);
6953 }
6954
6955 /// SplitVector - Split the vector with EXTRACT_SUBVECTOR and return the
6956 /// low/high part.
6957 std::pair<SDValue, SDValue>
6958 SelectionDAG::SplitVector(const SDValue &N, const SDLoc &DL, const EVT &LoVT,
6959                           const EVT &HiVT) {
6960   assert(LoVT.getVectorNumElements() + HiVT.getVectorNumElements() <=
6961          N.getValueType().getVectorNumElements() &&
6962          "More vector elements requested than available!");
6963   SDValue Lo, Hi;
6964   Lo = getNode(ISD::EXTRACT_SUBVECTOR, DL, LoVT, N,
6965                getConstant(0, DL, TLI->getVectorIdxTy(getDataLayout())));
6966   Hi = getNode(ISD::EXTRACT_SUBVECTOR, DL, HiVT, N,
6967                getConstant(LoVT.getVectorNumElements(), DL,
6968                            TLI->getVectorIdxTy(getDataLayout())));
6969   return std::make_pair(Lo, Hi);
6970 }
6971
6972 void SelectionDAG::ExtractVectorElements(SDValue Op,
6973                                          SmallVectorImpl<SDValue> &Args,
6974                                          unsigned Start, unsigned Count) {
6975   EVT VT = Op.getValueType();
6976   if (Count == 0)
6977     Count = VT.getVectorNumElements();
6978
6979   EVT EltVT = VT.getVectorElementType();
6980   EVT IdxTy = TLI->getVectorIdxTy(getDataLayout());
6981   SDLoc SL(Op);
6982   for (unsigned i = Start, e = Start + Count; i != e; ++i) {
6983     Args.push_back(getNode(ISD::EXTRACT_VECTOR_ELT, SL, EltVT,
6984                            Op, getConstant(i, SL, IdxTy)));
6985   }
6986 }
6987
6988 // getAddressSpace - Return the address space this GlobalAddress belongs to.
6989 unsigned GlobalAddressSDNode::getAddressSpace() const {
6990   return getGlobal()->getType()->getAddressSpace();
6991 }
6992
6993
6994 Type *ConstantPoolSDNode::getType() const {
6995   if (isMachineConstantPoolEntry())
6996     return Val.MachineCPVal->getType();
6997   return Val.ConstVal->getType();
6998 }
6999
7000 bool BuildVectorSDNode::isConstantSplat(APInt &SplatValue,
7001                                         APInt &SplatUndef,
7002                                         unsigned &SplatBitSize,
7003                                         bool &HasAnyUndefs,
7004                                         unsigned MinSplatBits,
7005                                         bool isBigEndian) const {
7006   EVT VT = getValueType(0);
7007   assert(VT.isVector() && "Expected a vector type");
7008   unsigned sz = VT.getSizeInBits();
7009   if (MinSplatBits > sz)
7010     return false;
7011
7012   SplatValue = APInt(sz, 0);
7013   SplatUndef = APInt(sz, 0);
7014
7015   // Get the bits.  Bits with undefined values (when the corresponding element
7016   // of the vector is an ISD::UNDEF value) are set in SplatUndef and cleared
7017   // in SplatValue.  If any of the values are not constant, give up and return
7018   // false.
7019   unsigned int nOps = getNumOperands();
7020   assert(nOps > 0 && "isConstantSplat has 0-size build vector");
7021   unsigned EltBitSize = VT.getVectorElementType().getSizeInBits();
7022
7023   for (unsigned j = 0; j < nOps; ++j) {
7024     unsigned i = isBigEndian ? nOps-1-j : j;
7025     SDValue OpVal = getOperand(i);
7026     unsigned BitPos = j * EltBitSize;
7027
7028     if (OpVal.getOpcode() == ISD::UNDEF)
7029       SplatUndef |= APInt::getBitsSet(sz, BitPos, BitPos + EltBitSize);
7030     else if (ConstantSDNode *CN = dyn_cast<ConstantSDNode>(OpVal))
7031       SplatValue |= CN->getAPIntValue().zextOrTrunc(EltBitSize).
7032                     zextOrTrunc(sz) << BitPos;
7033     else if (ConstantFPSDNode *CN = dyn_cast<ConstantFPSDNode>(OpVal))
7034       SplatValue |= CN->getValueAPF().bitcastToAPInt().zextOrTrunc(sz) <<BitPos;
7035      else
7036       return false;
7037   }
7038
7039   // The build_vector is all constants or undefs.  Find the smallest element
7040   // size that splats the vector.
7041
7042   HasAnyUndefs = (SplatUndef != 0);
7043   while (sz > 8) {
7044
7045     unsigned HalfSize = sz / 2;
7046     APInt HighValue = SplatValue.lshr(HalfSize).trunc(HalfSize);
7047     APInt LowValue = SplatValue.trunc(HalfSize);
7048     APInt HighUndef = SplatUndef.lshr(HalfSize).trunc(HalfSize);
7049     APInt LowUndef = SplatUndef.trunc(HalfSize);
7050
7051     // If the two halves do not match (ignoring undef bits), stop here.
7052     if ((HighValue & ~LowUndef) != (LowValue & ~HighUndef) ||
7053         MinSplatBits > HalfSize)
7054       break;
7055
7056     SplatValue = HighValue | LowValue;
7057     SplatUndef = HighUndef & LowUndef;
7058
7059     sz = HalfSize;
7060   }
7061
7062   SplatBitSize = sz;
7063   return true;
7064 }
7065
7066 SDValue BuildVectorSDNode::getSplatValue(BitVector *UndefElements) const {
7067   if (UndefElements) {
7068     UndefElements->clear();
7069     UndefElements->resize(getNumOperands());
7070   }
7071   SDValue Splatted;
7072   for (unsigned i = 0, e = getNumOperands(); i != e; ++i) {
7073     SDValue Op = getOperand(i);
7074     if (Op.getOpcode() == ISD::UNDEF) {
7075       if (UndefElements)
7076         (*UndefElements)[i] = true;
7077     } else if (!Splatted) {
7078       Splatted = Op;
7079     } else if (Splatted != Op) {
7080       return SDValue();
7081     }
7082   }
7083
7084   if (!Splatted) {
7085     assert(getOperand(0).getOpcode() == ISD::UNDEF &&
7086            "Can only have a splat without a constant for all undefs.");
7087     return getOperand(0);
7088   }
7089
7090   return Splatted;
7091 }
7092
7093 ConstantSDNode *
7094 BuildVectorSDNode::getConstantSplatNode(BitVector *UndefElements) const {
7095   return dyn_cast_or_null<ConstantSDNode>(getSplatValue(UndefElements));
7096 }
7097
7098 ConstantFPSDNode *
7099 BuildVectorSDNode::getConstantFPSplatNode(BitVector *UndefElements) const {
7100   return dyn_cast_or_null<ConstantFPSDNode>(getSplatValue(UndefElements));
7101 }
7102
7103 bool BuildVectorSDNode::isConstant() const {
7104   for (const SDValue &Op : op_values()) {
7105     unsigned Opc = Op.getOpcode();
7106     if (Opc != ISD::UNDEF && Opc != ISD::Constant && Opc != ISD::ConstantFP)
7107       return false;
7108   }
7109   return true;
7110 }
7111
7112 bool ShuffleVectorSDNode::isSplatMask(const int *Mask, EVT VT) {
7113   // Find the first non-undef value in the shuffle mask.
7114   unsigned i, e;
7115   for (i = 0, e = VT.getVectorNumElements(); i != e && Mask[i] < 0; ++i)
7116     /* search */;
7117
7118   assert(i != e && "VECTOR_SHUFFLE node with all undef indices!");
7119
7120   // Make sure all remaining elements are either undef or the same as the first
7121   // non-undef value.
7122   for (int Idx = Mask[i]; i != e; ++i)
7123     if (Mask[i] >= 0 && Mask[i] != Idx)
7124       return false;
7125   return true;
7126 }
7127
7128 #ifndef NDEBUG
7129 static void checkForCyclesHelper(const SDNode *N,
7130                                  SmallPtrSetImpl<const SDNode*> &Visited,
7131                                  SmallPtrSetImpl<const SDNode*> &Checked,
7132                                  const llvm::SelectionDAG *DAG) {
7133   // If this node has already been checked, don't check it again.
7134   if (Checked.count(N))
7135     return;
7136
7137   // If a node has already been visited on this depth-first walk, reject it as
7138   // a cycle.
7139   if (!Visited.insert(N).second) {
7140     errs() << "Detected cycle in SelectionDAG\n";
7141     dbgs() << "Offending node:\n";
7142     N->dumprFull(DAG); dbgs() << "\n";
7143     abort();
7144   }
7145
7146   for (const SDValue &Op : N->op_values())
7147     checkForCyclesHelper(Op.getNode(), Visited, Checked, DAG);
7148
7149   Checked.insert(N);
7150   Visited.erase(N);
7151 }
7152 #endif
7153
7154 void llvm::checkForCycles(const llvm::SDNode *N,
7155                           const llvm::SelectionDAG *DAG,
7156                           bool force) {
7157 #ifndef NDEBUG
7158   bool check = force;
7159 #ifdef XDEBUG
7160   check = true;
7161 #endif  // XDEBUG
7162   if (check) {
7163     assert(N && "Checking nonexistent SDNode");
7164     SmallPtrSet<const SDNode*, 32> visited;
7165     SmallPtrSet<const SDNode*, 32> checked;
7166     checkForCyclesHelper(N, visited, checked, DAG);
7167   }
7168 #endif  // !NDEBUG
7169 }
7170
7171 void llvm::checkForCycles(const llvm::SelectionDAG *DAG, bool force) {
7172   checkForCycles(DAG->getRoot().getNode(), DAG, force);
7173 }