xref: /llvm-project/llvm/utils/TableGen/Common/CodeGenDAGPatterns.cpp (revision 0ceffd362bf01769fa310103b11af9cdc12e2f5b)
1 //===- CodeGenDAGPatterns.cpp - Read DAG patterns from .td file -----------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 //
9 // This file implements the CodeGenDAGPatterns class, which is used to read and
10 // represent the patterns present in a .td file for instructions.
11 //
12 //===----------------------------------------------------------------------===//
13 
14 #include "CodeGenDAGPatterns.h"
15 #include "CodeGenInstruction.h"
16 #include "CodeGenRegisters.h"
17 #include "llvm/ADT/DenseSet.h"
18 #include "llvm/ADT/MapVector.h"
19 #include "llvm/ADT/STLExtras.h"
20 #include "llvm/ADT/SmallSet.h"
21 #include "llvm/ADT/SmallString.h"
22 #include "llvm/ADT/StringExtras.h"
23 #include "llvm/ADT/StringMap.h"
24 #include "llvm/ADT/Twine.h"
25 #include "llvm/Support/Debug.h"
26 #include "llvm/Support/ErrorHandling.h"
27 #include "llvm/Support/TypeSize.h"
28 #include "llvm/TableGen/Error.h"
29 #include "llvm/TableGen/Record.h"
30 #include <algorithm>
31 #include <cstdio>
32 #include <iterator>
33 #include <set>
34 using namespace llvm;
35 
36 #define DEBUG_TYPE "dag-patterns"
37 
38 static inline bool isIntegerOrPtr(MVT VT) {
39   return VT.isInteger() || VT == MVT::iPTR;
40 }
41 static inline bool isFloatingPoint(MVT VT) { return VT.isFloatingPoint(); }
42 static inline bool isVector(MVT VT) { return VT.isVector(); }
43 static inline bool isScalar(MVT VT) { return !VT.isVector(); }
44 
45 template <typename Predicate>
46 static bool berase_if(MachineValueTypeSet &S, Predicate P) {
47   bool Erased = false;
48   // It is ok to iterate over MachineValueTypeSet and remove elements from it
49   // at the same time.
50   for (MVT T : S) {
51     if (!P(T))
52       continue;
53     Erased = true;
54     S.erase(T);
55   }
56   return Erased;
57 }
58 
59 void MachineValueTypeSet::writeToStream(raw_ostream &OS) const {
60   SmallVector<MVT, 4> Types(begin(), end());
61   array_pod_sort(Types.begin(), Types.end());
62 
63   OS << '[';
64   ListSeparator LS(" ");
65   for (const MVT &T : Types)
66     OS << LS << ValueTypeByHwMode::getMVTName(T);
67   OS << ']';
68 }
69 
70 // --- TypeSetByHwMode
71 
72 // This is a parameterized type-set class. For each mode there is a list
73 // of types that are currently possible for a given tree node. Type
74 // inference will apply to each mode separately.
75 
76 TypeSetByHwMode::TypeSetByHwMode(ArrayRef<ValueTypeByHwMode> VTList) {
77   // Take the address space from the first type in the list.
78   if (!VTList.empty())
79     AddrSpace = VTList[0].PtrAddrSpace;
80 
81   for (const ValueTypeByHwMode &VVT : VTList)
82     insert(VVT);
83 }
84 
85 bool TypeSetByHwMode::isValueTypeByHwMode(bool AllowEmpty) const {
86   for (const auto &I : *this) {
87     if (I.second.size() > 1)
88       return false;
89     if (!AllowEmpty && I.second.empty())
90       return false;
91   }
92   return true;
93 }
94 
95 ValueTypeByHwMode TypeSetByHwMode::getValueTypeByHwMode() const {
96   assert(isValueTypeByHwMode(true) &&
97          "The type set has multiple types for at least one HW mode");
98   ValueTypeByHwMode VVT;
99   VVT.PtrAddrSpace = AddrSpace;
100 
101   for (const auto &I : *this) {
102     MVT T = I.second.empty() ? MVT::Other : *I.second.begin();
103     VVT.getOrCreateTypeForMode(I.first, T);
104   }
105   return VVT;
106 }
107 
108 bool TypeSetByHwMode::isPossible() const {
109   for (const auto &I : *this)
110     if (!I.second.empty())
111       return true;
112   return false;
113 }
114 
115 bool TypeSetByHwMode::insert(const ValueTypeByHwMode &VVT) {
116   bool Changed = false;
117   bool ContainsDefault = false;
118   MVT DT = MVT::Other;
119 
120   for (const auto &P : VVT) {
121     unsigned M = P.first;
122     // Make sure there exists a set for each specific mode from VVT.
123     Changed |= getOrCreate(M).insert(P.second).second;
124     // Cache VVT's default mode.
125     if (DefaultMode == M) {
126       ContainsDefault = true;
127       DT = P.second;
128     }
129   }
130 
131   // If VVT has a default mode, add the corresponding type to all
132   // modes in "this" that do not exist in VVT.
133   if (ContainsDefault)
134     for (auto &I : *this)
135       if (!VVT.hasMode(I.first))
136         Changed |= I.second.insert(DT).second;
137 
138   return Changed;
139 }
140 
141 // Constrain the type set to be the intersection with VTS.
142 bool TypeSetByHwMode::constrain(const TypeSetByHwMode &VTS) {
143   bool Changed = false;
144   if (hasDefault()) {
145     for (const auto &I : VTS) {
146       unsigned M = I.first;
147       if (M == DefaultMode || hasMode(M))
148         continue;
149       Map.insert({M, Map.at(DefaultMode)});
150       Changed = true;
151     }
152   }
153 
154   for (auto &I : *this) {
155     unsigned M = I.first;
156     SetType &S = I.second;
157     if (VTS.hasMode(M) || VTS.hasDefault()) {
158       Changed |= intersect(I.second, VTS.get(M));
159     } else if (!S.empty()) {
160       S.clear();
161       Changed = true;
162     }
163   }
164   return Changed;
165 }
166 
167 template <typename Predicate> bool TypeSetByHwMode::constrain(Predicate P) {
168   bool Changed = false;
169   for (auto &I : *this)
170     Changed |= berase_if(I.second, [&P](MVT VT) { return !P(VT); });
171   return Changed;
172 }
173 
174 template <typename Predicate>
175 bool TypeSetByHwMode::assign_if(const TypeSetByHwMode &VTS, Predicate P) {
176   assert(empty());
177   for (const auto &I : VTS) {
178     SetType &S = getOrCreate(I.first);
179     for (auto J : I.second)
180       if (P(J))
181         S.insert(J);
182   }
183   return !empty();
184 }
185 
186 void TypeSetByHwMode::writeToStream(raw_ostream &OS) const {
187   SmallVector<unsigned, 4> Modes;
188   Modes.reserve(Map.size());
189 
190   for (const auto &I : *this)
191     Modes.push_back(I.first);
192   if (Modes.empty()) {
193     OS << "{}";
194     return;
195   }
196   array_pod_sort(Modes.begin(), Modes.end());
197 
198   OS << '{';
199   for (unsigned M : Modes) {
200     OS << ' ' << getModeName(M) << ':';
201     get(M).writeToStream(OS);
202   }
203   OS << " }";
204 }
205 
206 bool TypeSetByHwMode::operator==(const TypeSetByHwMode &VTS) const {
207   // The isSimple call is much quicker than hasDefault - check this first.
208   bool IsSimple = isSimple();
209   bool VTSIsSimple = VTS.isSimple();
210   if (IsSimple && VTSIsSimple)
211     return getSimple() == VTS.getSimple();
212 
213   // Speedup: We have a default if the set is simple.
214   bool HaveDefault = IsSimple || hasDefault();
215   bool VTSHaveDefault = VTSIsSimple || VTS.hasDefault();
216   if (HaveDefault != VTSHaveDefault)
217     return false;
218 
219   SmallSet<unsigned, 4> Modes;
220   for (auto &I : *this)
221     Modes.insert(I.first);
222   for (const auto &I : VTS)
223     Modes.insert(I.first);
224 
225   if (HaveDefault) {
226     // Both sets have default mode.
227     for (unsigned M : Modes) {
228       if (get(M) != VTS.get(M))
229         return false;
230     }
231   } else {
232     // Neither set has default mode.
233     for (unsigned M : Modes) {
234       // If there is no default mode, an empty set is equivalent to not having
235       // the corresponding mode.
236       bool NoModeThis = !hasMode(M) || get(M).empty();
237       bool NoModeVTS = !VTS.hasMode(M) || VTS.get(M).empty();
238       if (NoModeThis != NoModeVTS)
239         return false;
240       if (!NoModeThis)
241         if (get(M) != VTS.get(M))
242           return false;
243     }
244   }
245 
246   return true;
247 }
248 
249 namespace llvm {
250 raw_ostream &operator<<(raw_ostream &OS, const MachineValueTypeSet &T) {
251   T.writeToStream(OS);
252   return OS;
253 }
254 raw_ostream &operator<<(raw_ostream &OS, const TypeSetByHwMode &T) {
255   T.writeToStream(OS);
256   return OS;
257 }
258 } // namespace llvm
259 
260 LLVM_DUMP_METHOD
261 void TypeSetByHwMode::dump() const { dbgs() << *this << '\n'; }
262 
263 bool TypeSetByHwMode::intersect(SetType &Out, const SetType &In) {
264   auto IntersectP = [&](std::optional<MVT> WildVT, function_ref<bool(MVT)> P) {
265     // Complement of In within this partition.
266     auto CompIn = [&](MVT T) -> bool { return !In.count(T) && P(T); };
267 
268     if (!WildVT)
269       return berase_if(Out, CompIn);
270 
271     bool OutW = Out.count(*WildVT), InW = In.count(*WildVT);
272     if (OutW == InW)
273       return berase_if(Out, CompIn);
274 
275     // Compute the intersection of scalars separately to account for only one
276     // set containing WildVT.
277     // The intersection of WildVT with a set of corresponding types that does
278     // not include WildVT will result in the most specific type:
279     // - WildVT is more specific than any set with two elements or more
280     // - WildVT is less specific than any single type.
281     // For example, for iPTR and scalar integer types
282     // { iPTR } * { i32 }     -> { i32 }
283     // { iPTR } * { i32 i64 } -> { iPTR }
284     // and
285     // { iPTR i32 } * { i32 }          -> { i32 }
286     // { iPTR i32 } * { i32 i64 }      -> { i32 i64 }
287     // { iPTR i32 } * { i32 i64 i128 } -> { iPTR i32 }
288 
289     // Looking at just this partition, let In' = elements only in In,
290     // Out' = elements only in Out, and IO = elements common to both. Normally
291     // IO would be returned as the result of the intersection, but we need to
292     // account for WildVT being a "wildcard" of sorts. Since elements in IO are
293     // those that match both sets exactly, they will all belong to the output.
294     // If any of the "leftovers" (i.e. In' or Out') contain WildVT, it means
295     // that the other set doesn't have it, but it could have (1) a more
296     // specific type, or (2) a set of types that is less specific. The
297     // "leftovers" from the other set is what we want to examine more closely.
298 
299     auto Leftovers = [&](const SetType &A, const SetType &B) {
300       SetType Diff = A;
301       berase_if(Diff, [&](MVT T) { return B.count(T) || !P(T); });
302       return Diff;
303     };
304 
305     if (InW) {
306       SetType OutLeftovers = Leftovers(Out, In);
307       if (OutLeftovers.size() < 2) {
308         // WildVT not added to Out. Keep the possible single leftover.
309         return false;
310       }
311       // WildVT replaces the leftovers.
312       berase_if(Out, CompIn);
313       Out.insert(*WildVT);
314       return true;
315     }
316 
317     // OutW == true
318     SetType InLeftovers = Leftovers(In, Out);
319     unsigned SizeOut = Out.size();
320     berase_if(Out, CompIn); // This will remove at least the WildVT.
321     if (InLeftovers.size() < 2) {
322       // WildVT deleted from Out. Add back the possible single leftover.
323       Out.insert(InLeftovers);
324       return true;
325     }
326 
327     // Keep the WildVT in Out.
328     Out.insert(*WildVT);
329     // If WildVT was the only element initially removed from Out, then Out
330     // has not changed.
331     return SizeOut != Out.size();
332   };
333 
334   // Note: must be non-overlapping
335   using WildPartT = std::pair<MVT, std::function<bool(MVT)>>;
336   static const WildPartT WildParts[] = {
337       {MVT::iPTR, [](MVT T) { return T.isScalarInteger() || T == MVT::iPTR; }},
338   };
339 
340   bool Changed = false;
341   for (const auto &I : WildParts)
342     Changed |= IntersectP(I.first, I.second);
343 
344   Changed |= IntersectP(std::nullopt, [&](MVT T) {
345     return !any_of(WildParts, [=](const WildPartT &I) { return I.second(T); });
346   });
347 
348   return Changed;
349 }
350 
351 bool TypeSetByHwMode::validate() const {
352   if (empty())
353     return true;
354   bool AllEmpty = true;
355   for (const auto &I : *this)
356     AllEmpty &= I.second.empty();
357   return !AllEmpty;
358 }
359 
360 // --- TypeInfer
361 
362 bool TypeInfer::MergeInTypeInfo(TypeSetByHwMode &Out,
363                                 const TypeSetByHwMode &In) const {
364   ValidateOnExit _1(Out, *this);
365   In.validate();
366   if (In.empty() || Out == In || TP.hasError())
367     return false;
368   if (Out.empty()) {
369     Out = In;
370     return true;
371   }
372 
373   bool Changed = Out.constrain(In);
374   if (Changed && Out.empty())
375     TP.error("Type contradiction");
376 
377   return Changed;
378 }
379 
380 bool TypeInfer::forceArbitrary(TypeSetByHwMode &Out) {
381   ValidateOnExit _1(Out, *this);
382   if (TP.hasError())
383     return false;
384   assert(!Out.empty() && "cannot pick from an empty set");
385 
386   bool Changed = false;
387   for (auto &I : Out) {
388     TypeSetByHwMode::SetType &S = I.second;
389     if (S.size() <= 1)
390       continue;
391     MVT T = *S.begin(); // Pick the first element.
392     S.clear();
393     S.insert(T);
394     Changed = true;
395   }
396   return Changed;
397 }
398 
399 bool TypeInfer::EnforceInteger(TypeSetByHwMode &Out) {
400   ValidateOnExit _1(Out, *this);
401   if (TP.hasError())
402     return false;
403   if (!Out.empty())
404     return Out.constrain(isIntegerOrPtr);
405 
406   return Out.assign_if(getLegalTypes(), isIntegerOrPtr);
407 }
408 
409 bool TypeInfer::EnforceFloatingPoint(TypeSetByHwMode &Out) {
410   ValidateOnExit _1(Out, *this);
411   if (TP.hasError())
412     return false;
413   if (!Out.empty())
414     return Out.constrain(isFloatingPoint);
415 
416   return Out.assign_if(getLegalTypes(), isFloatingPoint);
417 }
418 
419 bool TypeInfer::EnforceScalar(TypeSetByHwMode &Out) {
420   ValidateOnExit _1(Out, *this);
421   if (TP.hasError())
422     return false;
423   if (!Out.empty())
424     return Out.constrain(isScalar);
425 
426   return Out.assign_if(getLegalTypes(), isScalar);
427 }
428 
429 bool TypeInfer::EnforceVector(TypeSetByHwMode &Out) {
430   ValidateOnExit _1(Out, *this);
431   if (TP.hasError())
432     return false;
433   if (!Out.empty())
434     return Out.constrain(isVector);
435 
436   return Out.assign_if(getLegalTypes(), isVector);
437 }
438 
439 bool TypeInfer::EnforceAny(TypeSetByHwMode &Out) {
440   ValidateOnExit _1(Out, *this);
441   if (TP.hasError() || !Out.empty())
442     return false;
443 
444   Out = getLegalTypes();
445   return true;
446 }
447 
448 template <typename Iter, typename Pred, typename Less>
449 static Iter min_if(Iter B, Iter E, Pred P, Less L) {
450   if (B == E)
451     return E;
452   Iter Min = E;
453   for (Iter I = B; I != E; ++I) {
454     if (!P(*I))
455       continue;
456     if (Min == E || L(*I, *Min))
457       Min = I;
458   }
459   return Min;
460 }
461 
462 template <typename Iter, typename Pred, typename Less>
463 static Iter max_if(Iter B, Iter E, Pred P, Less L) {
464   if (B == E)
465     return E;
466   Iter Max = E;
467   for (Iter I = B; I != E; ++I) {
468     if (!P(*I))
469       continue;
470     if (Max == E || L(*Max, *I))
471       Max = I;
472   }
473   return Max;
474 }
475 
476 /// Make sure that for each type in Small, there exists a larger type in Big.
477 bool TypeInfer::EnforceSmallerThan(TypeSetByHwMode &Small, TypeSetByHwMode &Big,
478                                    bool SmallIsVT) {
479   ValidateOnExit _1(Small, *this), _2(Big, *this);
480   if (TP.hasError())
481     return false;
482   bool Changed = false;
483 
484   assert((!SmallIsVT || !Small.empty()) &&
485          "Small should not be empty for SDTCisVTSmallerThanOp");
486 
487   if (Small.empty())
488     Changed |= EnforceAny(Small);
489   if (Big.empty())
490     Changed |= EnforceAny(Big);
491 
492   assert(Small.hasDefault() && Big.hasDefault());
493 
494   SmallVector<unsigned, 4> Modes;
495   union_modes(Small, Big, Modes);
496 
497   // 1. Only allow integer or floating point types and make sure that
498   //    both sides are both integer or both floating point.
499   // 2. Make sure that either both sides have vector types, or neither
500   //    of them does.
501   for (unsigned M : Modes) {
502     TypeSetByHwMode::SetType &S = Small.get(M);
503     TypeSetByHwMode::SetType &B = Big.get(M);
504 
505     assert((!SmallIsVT || !S.empty()) && "Expected non-empty type");
506 
507     if (any_of(S, isIntegerOrPtr) && any_of(B, isIntegerOrPtr)) {
508       auto NotInt = [](MVT VT) { return !isIntegerOrPtr(VT); };
509       Changed |= berase_if(S, NotInt);
510       Changed |= berase_if(B, NotInt);
511     } else if (any_of(S, isFloatingPoint) && any_of(B, isFloatingPoint)) {
512       auto NotFP = [](MVT VT) { return !isFloatingPoint(VT); };
513       Changed |= berase_if(S, NotFP);
514       Changed |= berase_if(B, NotFP);
515     } else if (SmallIsVT && B.empty()) {
516       // B is empty and since S is a specific VT, it will never be empty. Don't
517       // report this as a change, just clear S and continue. This prevents an
518       // infinite loop.
519       S.clear();
520     } else if (S.empty() || B.empty()) {
521       Changed = !S.empty() || !B.empty();
522       S.clear();
523       B.clear();
524     } else {
525       TP.error("Incompatible types");
526       return Changed;
527     }
528 
529     if (none_of(S, isVector) || none_of(B, isVector)) {
530       Changed |= berase_if(S, isVector);
531       Changed |= berase_if(B, isVector);
532     }
533   }
534 
535   auto LT = [](MVT A, MVT B) -> bool {
536     // Always treat non-scalable MVTs as smaller than scalable MVTs for the
537     // purposes of ordering.
538     auto ASize = std::tuple(A.isScalableVector(), A.getScalarSizeInBits(),
539                             A.getSizeInBits().getKnownMinValue());
540     auto BSize = std::tuple(B.isScalableVector(), B.getScalarSizeInBits(),
541                             B.getSizeInBits().getKnownMinValue());
542     return ASize < BSize;
543   };
544   auto SameKindLE = [](MVT A, MVT B) -> bool {
545     // This function is used when removing elements: when a vector is compared
546     // to a non-vector or a scalable vector to any non-scalable MVT, it should
547     // return false (to avoid removal).
548     if (std::tuple(A.isVector(), A.isScalableVector()) !=
549         std::tuple(B.isVector(), B.isScalableVector()))
550       return false;
551 
552     return std::tuple(A.getScalarSizeInBits(),
553                       A.getSizeInBits().getKnownMinValue()) <=
554            std::tuple(B.getScalarSizeInBits(),
555                       B.getSizeInBits().getKnownMinValue());
556   };
557 
558   for (unsigned M : Modes) {
559     TypeSetByHwMode::SetType &S = Small.get(M);
560     TypeSetByHwMode::SetType &B = Big.get(M);
561     // MinS = min scalar in Small, remove all scalars from Big that are
562     // smaller-or-equal than MinS.
563     auto MinS = min_if(S.begin(), S.end(), isScalar, LT);
564     if (MinS != S.end())
565       Changed |=
566           berase_if(B, std::bind(SameKindLE, std::placeholders::_1, *MinS));
567 
568     // MaxS = max scalar in Big, remove all scalars from Small that are
569     // larger than MaxS.
570     auto MaxS = max_if(B.begin(), B.end(), isScalar, LT);
571     if (MaxS != B.end())
572       Changed |=
573           berase_if(S, std::bind(SameKindLE, *MaxS, std::placeholders::_1));
574 
575     // MinV = min vector in Small, remove all vectors from Big that are
576     // smaller-or-equal than MinV.
577     auto MinV = min_if(S.begin(), S.end(), isVector, LT);
578     if (MinV != S.end())
579       Changed |=
580           berase_if(B, std::bind(SameKindLE, std::placeholders::_1, *MinV));
581 
582     // MaxV = max vector in Big, remove all vectors from Small that are
583     // larger than MaxV.
584     auto MaxV = max_if(B.begin(), B.end(), isVector, LT);
585     if (MaxV != B.end())
586       Changed |=
587           berase_if(S, std::bind(SameKindLE, *MaxV, std::placeholders::_1));
588   }
589 
590   return Changed;
591 }
592 
593 /// 1. Ensure that for each type T in Vec, T is a vector type, and that
594 ///    for each type U in Elem, U is a scalar type.
595 /// 2. Ensure that for each (scalar) type U in Elem, there exists a (vector)
596 ///    type T in Vec, such that U is the element type of T.
597 bool TypeInfer::EnforceVectorEltTypeIs(TypeSetByHwMode &Vec,
598                                        TypeSetByHwMode &Elem) {
599   ValidateOnExit _1(Vec, *this), _2(Elem, *this);
600   if (TP.hasError())
601     return false;
602   bool Changed = false;
603 
604   if (Vec.empty())
605     Changed |= EnforceVector(Vec);
606   if (Elem.empty())
607     Changed |= EnforceScalar(Elem);
608 
609   SmallVector<unsigned, 4> Modes;
610   union_modes(Vec, Elem, Modes);
611   for (unsigned M : Modes) {
612     TypeSetByHwMode::SetType &V = Vec.get(M);
613     TypeSetByHwMode::SetType &E = Elem.get(M);
614 
615     Changed |= berase_if(V, isScalar); // Scalar = !vector
616     Changed |= berase_if(E, isVector); // Vector = !scalar
617     assert(!V.empty() && !E.empty());
618 
619     MachineValueTypeSet VT, ST;
620     // Collect element types from the "vector" set.
621     for (MVT T : V)
622       VT.insert(T.getVectorElementType());
623     // Collect scalar types from the "element" set.
624     for (MVT T : E)
625       ST.insert(T);
626 
627     // Remove from V all (vector) types whose element type is not in S.
628     Changed |= berase_if(V, [&ST](MVT T) -> bool {
629       return !ST.count(T.getVectorElementType());
630     });
631     // Remove from E all (scalar) types, for which there is no corresponding
632     // type in V.
633     Changed |= berase_if(E, [&VT](MVT T) -> bool { return !VT.count(T); });
634   }
635 
636   return Changed;
637 }
638 
639 bool TypeInfer::EnforceVectorEltTypeIs(TypeSetByHwMode &Vec,
640                                        const ValueTypeByHwMode &VVT) {
641   TypeSetByHwMode Tmp(VVT);
642   ValidateOnExit _1(Vec, *this), _2(Tmp, *this);
643   return EnforceVectorEltTypeIs(Vec, Tmp);
644 }
645 
646 /// Ensure that for each type T in Sub, T is a vector type, and there
647 /// exists a type U in Vec such that U is a vector type with the same
648 /// element type as T and at least as many elements as T.
649 bool TypeInfer::EnforceVectorSubVectorTypeIs(TypeSetByHwMode &Vec,
650                                              TypeSetByHwMode &Sub) {
651   ValidateOnExit _1(Vec, *this), _2(Sub, *this);
652   if (TP.hasError())
653     return false;
654 
655   /// Return true if B is a suB-vector of P, i.e. P is a suPer-vector of B.
656   auto IsSubVec = [](MVT B, MVT P) -> bool {
657     if (!B.isVector() || !P.isVector())
658       return false;
659     // Logically a <4 x i32> is a valid subvector of <n x 4 x i32>
660     // but until there are obvious use-cases for this, keep the
661     // types separate.
662     if (B.isScalableVector() != P.isScalableVector())
663       return false;
664     if (B.getVectorElementType() != P.getVectorElementType())
665       return false;
666     return B.getVectorMinNumElements() < P.getVectorMinNumElements();
667   };
668 
669   /// Return true if S has no element (vector type) that T is a sub-vector of,
670   /// i.e. has the same element type as T and more elements.
671   auto NoSubV = [&IsSubVec](const TypeSetByHwMode::SetType &S, MVT T) -> bool {
672     for (auto I : S)
673       if (IsSubVec(T, I))
674         return false;
675     return true;
676   };
677 
678   /// Return true if S has no element (vector type) that T is a super-vector
679   /// of, i.e. has the same element type as T and fewer elements.
680   auto NoSupV = [&IsSubVec](const TypeSetByHwMode::SetType &S, MVT T) -> bool {
681     for (auto I : S)
682       if (IsSubVec(I, T))
683         return false;
684     return true;
685   };
686 
687   bool Changed = false;
688 
689   if (Vec.empty())
690     Changed |= EnforceVector(Vec);
691   if (Sub.empty())
692     Changed |= EnforceVector(Sub);
693 
694   SmallVector<unsigned, 4> Modes;
695   union_modes(Vec, Sub, Modes);
696   for (unsigned M : Modes) {
697     TypeSetByHwMode::SetType &S = Sub.get(M);
698     TypeSetByHwMode::SetType &V = Vec.get(M);
699 
700     Changed |= berase_if(S, isScalar);
701 
702     // Erase all types from S that are not sub-vectors of a type in V.
703     Changed |= berase_if(S, std::bind(NoSubV, V, std::placeholders::_1));
704 
705     // Erase all types from V that are not super-vectors of a type in S.
706     Changed |= berase_if(V, std::bind(NoSupV, S, std::placeholders::_1));
707   }
708 
709   return Changed;
710 }
711 
712 /// 1. Ensure that V has a scalar type iff W has a scalar type.
713 /// 2. Ensure that for each vector type T in V, there exists a vector
714 ///    type U in W, such that T and U have the same number of elements.
715 /// 3. Ensure that for each vector type U in W, there exists a vector
716 ///    type T in V, such that T and U have the same number of elements
717 ///    (reverse of 2).
718 bool TypeInfer::EnforceSameNumElts(TypeSetByHwMode &V, TypeSetByHwMode &W) {
719   ValidateOnExit _1(V, *this), _2(W, *this);
720   if (TP.hasError())
721     return false;
722 
723   bool Changed = false;
724   if (V.empty())
725     Changed |= EnforceAny(V);
726   if (W.empty())
727     Changed |= EnforceAny(W);
728 
729   // An actual vector type cannot have 0 elements, so we can treat scalars
730   // as zero-length vectors. This way both vectors and scalars can be
731   // processed identically.
732   auto NoLength = [](const SmallDenseSet<ElementCount> &Lengths,
733                      MVT T) -> bool {
734     return !Lengths.count(T.isVector() ? T.getVectorElementCount()
735                                        : ElementCount());
736   };
737 
738   SmallVector<unsigned, 4> Modes;
739   union_modes(V, W, Modes);
740   for (unsigned M : Modes) {
741     TypeSetByHwMode::SetType &VS = V.get(M);
742     TypeSetByHwMode::SetType &WS = W.get(M);
743 
744     SmallDenseSet<ElementCount> VN, WN;
745     for (MVT T : VS)
746       VN.insert(T.isVector() ? T.getVectorElementCount() : ElementCount());
747     for (MVT T : WS)
748       WN.insert(T.isVector() ? T.getVectorElementCount() : ElementCount());
749 
750     Changed |= berase_if(VS, std::bind(NoLength, WN, std::placeholders::_1));
751     Changed |= berase_if(WS, std::bind(NoLength, VN, std::placeholders::_1));
752   }
753   return Changed;
754 }
755 
756 namespace {
757 struct TypeSizeComparator {
758   bool operator()(const TypeSize &LHS, const TypeSize &RHS) const {
759     return std::tuple(LHS.isScalable(), LHS.getKnownMinValue()) <
760            std::tuple(RHS.isScalable(), RHS.getKnownMinValue());
761   }
762 };
763 } // end anonymous namespace
764 
765 /// 1. Ensure that for each type T in A, there exists a type U in B,
766 ///    such that T and U have equal size in bits.
767 /// 2. Ensure that for each type U in B, there exists a type T in A
768 ///    such that T and U have equal size in bits (reverse of 1).
769 bool TypeInfer::EnforceSameSize(TypeSetByHwMode &A, TypeSetByHwMode &B) {
770   ValidateOnExit _1(A, *this), _2(B, *this);
771   if (TP.hasError())
772     return false;
773   bool Changed = false;
774   if (A.empty())
775     Changed |= EnforceAny(A);
776   if (B.empty())
777     Changed |= EnforceAny(B);
778 
779   typedef SmallSet<TypeSize, 2, TypeSizeComparator> TypeSizeSet;
780 
781   auto NoSize = [](const TypeSizeSet &Sizes, MVT T) -> bool {
782     return !Sizes.count(T.getSizeInBits());
783   };
784 
785   SmallVector<unsigned, 4> Modes;
786   union_modes(A, B, Modes);
787   for (unsigned M : Modes) {
788     TypeSetByHwMode::SetType &AS = A.get(M);
789     TypeSetByHwMode::SetType &BS = B.get(M);
790     TypeSizeSet AN, BN;
791 
792     for (MVT T : AS)
793       AN.insert(T.getSizeInBits());
794     for (MVT T : BS)
795       BN.insert(T.getSizeInBits());
796 
797     Changed |= berase_if(AS, std::bind(NoSize, BN, std::placeholders::_1));
798     Changed |= berase_if(BS, std::bind(NoSize, AN, std::placeholders::_1));
799   }
800 
801   return Changed;
802 }
803 
804 void TypeInfer::expandOverloads(TypeSetByHwMode &VTS) const {
805   ValidateOnExit _1(VTS, *this);
806   const TypeSetByHwMode &Legal = getLegalTypes();
807   assert(Legal.isSimple() && "Default-mode only expected");
808   const TypeSetByHwMode::SetType &LegalTypes = Legal.getSimple();
809 
810   for (auto &I : VTS)
811     expandOverloads(I.second, LegalTypes);
812 }
813 
814 void TypeInfer::expandOverloads(TypeSetByHwMode::SetType &Out,
815                                 const TypeSetByHwMode::SetType &Legal) const {
816   if (Out.count(MVT::iPTRAny)) {
817     Out.erase(MVT::iPTRAny);
818     Out.insert(MVT::iPTR);
819   } else if (Out.count(MVT::iAny)) {
820     Out.erase(MVT::iAny);
821     for (MVT T : MVT::integer_valuetypes())
822       if (Legal.count(T))
823         Out.insert(T);
824     for (MVT T : MVT::integer_fixedlen_vector_valuetypes())
825       if (Legal.count(T))
826         Out.insert(T);
827     for (MVT T : MVT::integer_scalable_vector_valuetypes())
828       if (Legal.count(T))
829         Out.insert(T);
830   } else if (Out.count(MVT::fAny)) {
831     Out.erase(MVT::fAny);
832     for (MVT T : MVT::fp_valuetypes())
833       if (Legal.count(T))
834         Out.insert(T);
835     for (MVT T : MVT::fp_fixedlen_vector_valuetypes())
836       if (Legal.count(T))
837         Out.insert(T);
838     for (MVT T : MVT::fp_scalable_vector_valuetypes())
839       if (Legal.count(T))
840         Out.insert(T);
841   } else if (Out.count(MVT::vAny)) {
842     Out.erase(MVT::vAny);
843     for (MVT T : MVT::vector_valuetypes())
844       if (Legal.count(T))
845         Out.insert(T);
846   } else if (Out.count(MVT::Any)) {
847     Out.erase(MVT::Any);
848     for (MVT T : MVT::all_valuetypes())
849       if (Legal.count(T))
850         Out.insert(T);
851   }
852 }
853 
854 const TypeSetByHwMode &TypeInfer::getLegalTypes() const {
855   if (!LegalTypesCached) {
856     TypeSetByHwMode::SetType &LegalTypes = LegalCache.getOrCreate(DefaultMode);
857     // Stuff all types from all modes into the default mode.
858     const TypeSetByHwMode &LTS = TP.getDAGPatterns().getLegalTypes();
859     for (const auto &I : LTS)
860       LegalTypes.insert(I.second);
861     LegalTypesCached = true;
862   }
863   assert(LegalCache.isSimple() && "Default-mode only expected");
864   return LegalCache;
865 }
866 
867 TypeInfer::ValidateOnExit::~ValidateOnExit() {
868   if (Infer.Validate && !VTS.validate()) {
869 #if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP)
870     errs() << "Type set is empty for each HW mode:\n"
871               "possible type contradiction in the pattern below "
872               "(use -print-records with llvm-tblgen to see all "
873               "expanded records).\n";
874     Infer.TP.dump();
875     errs() << "Generated from record:\n";
876     Infer.TP.getRecord()->dump();
877 #endif
878     PrintFatalError(Infer.TP.getRecord()->getLoc(),
879                     "Type set is empty for each HW mode in '" +
880                         Infer.TP.getRecord()->getName() + "'");
881   }
882 }
883 
884 //===----------------------------------------------------------------------===//
885 // ScopedName Implementation
886 //===----------------------------------------------------------------------===//
887 
888 bool ScopedName::operator==(const ScopedName &o) const {
889   return Scope == o.Scope && Identifier == o.Identifier;
890 }
891 
892 bool ScopedName::operator!=(const ScopedName &o) const { return !(*this == o); }
893 
894 //===----------------------------------------------------------------------===//
895 // TreePredicateFn Implementation
896 //===----------------------------------------------------------------------===//
897 
898 /// TreePredicateFn constructor.  Here 'N' is a subclass of PatFrag.
899 TreePredicateFn::TreePredicateFn(TreePattern *N) : PatFragRec(N) {
900   assert(
901       (!hasPredCode() || !hasImmCode()) &&
902       ".td file corrupt: can't have a node predicate *and* an imm predicate");
903 }
904 
905 bool TreePredicateFn::hasPredCode() const {
906   return isLoad() || isStore() || isAtomic() || hasNoUse() || hasOneUse() ||
907          !PatFragRec->getRecord()->getValueAsString("PredicateCode").empty();
908 }
909 
910 std::string TreePredicateFn::getPredCode() const {
911   std::string Code;
912 
913   if (!isLoad() && !isStore() && !isAtomic()) {
914     Record *MemoryVT = getMemoryVT();
915 
916     if (MemoryVT)
917       PrintFatalError(getOrigPatFragRecord()->getRecord()->getLoc(),
918                       "MemoryVT requires IsLoad or IsStore");
919   }
920 
921   if (!isLoad() && !isStore()) {
922     if (isUnindexed())
923       PrintFatalError(getOrigPatFragRecord()->getRecord()->getLoc(),
924                       "IsUnindexed requires IsLoad or IsStore");
925 
926     Record *ScalarMemoryVT = getScalarMemoryVT();
927 
928     if (ScalarMemoryVT)
929       PrintFatalError(getOrigPatFragRecord()->getRecord()->getLoc(),
930                       "ScalarMemoryVT requires IsLoad or IsStore");
931   }
932 
933   if (isLoad() + isStore() + isAtomic() > 1)
934     PrintFatalError(getOrigPatFragRecord()->getRecord()->getLoc(),
935                     "IsLoad, IsStore, and IsAtomic are mutually exclusive");
936 
937   if (isLoad()) {
938     if (!isUnindexed() && !isNonExtLoad() && !isAnyExtLoad() &&
939         !isSignExtLoad() && !isZeroExtLoad() && getMemoryVT() == nullptr &&
940         getScalarMemoryVT() == nullptr && getAddressSpaces() == nullptr &&
941         getMinAlignment() < 1)
942       PrintFatalError(getOrigPatFragRecord()->getRecord()->getLoc(),
943                       "IsLoad cannot be used by itself");
944   } else {
945     if (isNonExtLoad())
946       PrintFatalError(getOrigPatFragRecord()->getRecord()->getLoc(),
947                       "IsNonExtLoad requires IsLoad");
948     if (isAnyExtLoad())
949       PrintFatalError(getOrigPatFragRecord()->getRecord()->getLoc(),
950                       "IsAnyExtLoad requires IsLoad");
951 
952     if (!isAtomic()) {
953       if (isSignExtLoad())
954         PrintFatalError(getOrigPatFragRecord()->getRecord()->getLoc(),
955                         "IsSignExtLoad requires IsLoad or IsAtomic");
956       if (isZeroExtLoad())
957         PrintFatalError(getOrigPatFragRecord()->getRecord()->getLoc(),
958                         "IsZeroExtLoad requires IsLoad or IsAtomic");
959     }
960   }
961 
962   if (isStore()) {
963     if (!isUnindexed() && !isTruncStore() && !isNonTruncStore() &&
964         getMemoryVT() == nullptr && getScalarMemoryVT() == nullptr &&
965         getAddressSpaces() == nullptr && getMinAlignment() < 1)
966       PrintFatalError(getOrigPatFragRecord()->getRecord()->getLoc(),
967                       "IsStore cannot be used by itself");
968   } else {
969     if (isNonTruncStore())
970       PrintFatalError(getOrigPatFragRecord()->getRecord()->getLoc(),
971                       "IsNonTruncStore requires IsStore");
972     if (isTruncStore())
973       PrintFatalError(getOrigPatFragRecord()->getRecord()->getLoc(),
974                       "IsTruncStore requires IsStore");
975   }
976 
977   if (isAtomic()) {
978     if (getMemoryVT() == nullptr && !isAtomicOrderingMonotonic() &&
979         getAddressSpaces() == nullptr &&
980         // FIXME: Should atomic loads be IsLoad, IsAtomic, or both?
981         !isZeroExtLoad() && !isSignExtLoad() && !isAtomicOrderingAcquire() &&
982         !isAtomicOrderingRelease() && !isAtomicOrderingAcquireRelease() &&
983         !isAtomicOrderingSequentiallyConsistent() &&
984         !isAtomicOrderingAcquireOrStronger() &&
985         !isAtomicOrderingReleaseOrStronger() &&
986         !isAtomicOrderingWeakerThanAcquire() &&
987         !isAtomicOrderingWeakerThanRelease())
988       PrintFatalError(getOrigPatFragRecord()->getRecord()->getLoc(),
989                       "IsAtomic cannot be used by itself");
990   } else {
991     if (isAtomicOrderingMonotonic())
992       PrintFatalError(getOrigPatFragRecord()->getRecord()->getLoc(),
993                       "IsAtomicOrderingMonotonic requires IsAtomic");
994     if (isAtomicOrderingAcquire())
995       PrintFatalError(getOrigPatFragRecord()->getRecord()->getLoc(),
996                       "IsAtomicOrderingAcquire requires IsAtomic");
997     if (isAtomicOrderingRelease())
998       PrintFatalError(getOrigPatFragRecord()->getRecord()->getLoc(),
999                       "IsAtomicOrderingRelease requires IsAtomic");
1000     if (isAtomicOrderingAcquireRelease())
1001       PrintFatalError(getOrigPatFragRecord()->getRecord()->getLoc(),
1002                       "IsAtomicOrderingAcquireRelease requires IsAtomic");
1003     if (isAtomicOrderingSequentiallyConsistent())
1004       PrintFatalError(
1005           getOrigPatFragRecord()->getRecord()->getLoc(),
1006           "IsAtomicOrderingSequentiallyConsistent requires IsAtomic");
1007     if (isAtomicOrderingAcquireOrStronger())
1008       PrintFatalError(getOrigPatFragRecord()->getRecord()->getLoc(),
1009                       "IsAtomicOrderingAcquireOrStronger requires IsAtomic");
1010     if (isAtomicOrderingReleaseOrStronger())
1011       PrintFatalError(getOrigPatFragRecord()->getRecord()->getLoc(),
1012                       "IsAtomicOrderingReleaseOrStronger requires IsAtomic");
1013     if (isAtomicOrderingWeakerThanAcquire())
1014       PrintFatalError(getOrigPatFragRecord()->getRecord()->getLoc(),
1015                       "IsAtomicOrderingWeakerThanAcquire requires IsAtomic");
1016   }
1017 
1018   if (isLoad() || isStore() || isAtomic()) {
1019     if (ListInit *AddressSpaces = getAddressSpaces()) {
1020       Code += "unsigned AddrSpace = cast<MemSDNode>(N)->getAddressSpace();\n"
1021               " if (";
1022 
1023       ListSeparator LS(" && ");
1024       for (Init *Val : AddressSpaces->getValues()) {
1025         Code += LS;
1026 
1027         IntInit *IntVal = dyn_cast<IntInit>(Val);
1028         if (!IntVal) {
1029           PrintFatalError(getOrigPatFragRecord()->getRecord()->getLoc(),
1030                           "AddressSpaces element must be integer");
1031         }
1032 
1033         Code += "AddrSpace != " + utostr(IntVal->getValue());
1034       }
1035 
1036       Code += ")\nreturn false;\n";
1037     }
1038 
1039     int64_t MinAlign = getMinAlignment();
1040     if (MinAlign > 0) {
1041       Code += "if (cast<MemSDNode>(N)->getAlign() < Align(";
1042       Code += utostr(MinAlign);
1043       Code += "))\nreturn false;\n";
1044     }
1045 
1046     Record *MemoryVT = getMemoryVT();
1047 
1048     if (MemoryVT)
1049       Code += ("if (cast<MemSDNode>(N)->getMemoryVT() != MVT::" +
1050                MemoryVT->getName() + ") return false;\n")
1051                   .str();
1052   }
1053 
1054   if (isAtomic() && isAtomicOrderingMonotonic())
1055     Code += "if (cast<AtomicSDNode>(N)->getMergedOrdering() != "
1056             "AtomicOrdering::Monotonic) return false;\n";
1057   if (isAtomic() && isAtomicOrderingAcquire())
1058     Code += "if (cast<AtomicSDNode>(N)->getMergedOrdering() != "
1059             "AtomicOrdering::Acquire) return false;\n";
1060   if (isAtomic() && isAtomicOrderingRelease())
1061     Code += "if (cast<AtomicSDNode>(N)->getMergedOrdering() != "
1062             "AtomicOrdering::Release) return false;\n";
1063   if (isAtomic() && isAtomicOrderingAcquireRelease())
1064     Code += "if (cast<AtomicSDNode>(N)->getMergedOrdering() != "
1065             "AtomicOrdering::AcquireRelease) return false;\n";
1066   if (isAtomic() && isAtomicOrderingSequentiallyConsistent())
1067     Code += "if (cast<AtomicSDNode>(N)->getMergedOrdering() != "
1068             "AtomicOrdering::SequentiallyConsistent) return false;\n";
1069 
1070   if (isAtomic() && isAtomicOrderingAcquireOrStronger())
1071     Code +=
1072         "if (!isAcquireOrStronger(cast<AtomicSDNode>(N)->getMergedOrdering())) "
1073         "return false;\n";
1074   if (isAtomic() && isAtomicOrderingWeakerThanAcquire())
1075     Code +=
1076         "if (isAcquireOrStronger(cast<AtomicSDNode>(N)->getMergedOrdering())) "
1077         "return false;\n";
1078 
1079   if (isAtomic() && isAtomicOrderingReleaseOrStronger())
1080     Code +=
1081         "if (!isReleaseOrStronger(cast<AtomicSDNode>(N)->getMergedOrdering())) "
1082         "return false;\n";
1083   if (isAtomic() && isAtomicOrderingWeakerThanRelease())
1084     Code +=
1085         "if (isReleaseOrStronger(cast<AtomicSDNode>(N)->getMergedOrdering())) "
1086         "return false;\n";
1087 
1088   // TODO: Handle atomic sextload/zextload normally when ATOMIC_LOAD is removed.
1089   if (isAtomic() && (isZeroExtLoad() || isSignExtLoad()))
1090     Code += "return false;\n";
1091 
1092   if (isLoad() || isStore()) {
1093     StringRef SDNodeName = isLoad() ? "LoadSDNode" : "StoreSDNode";
1094 
1095     if (isUnindexed())
1096       Code += ("if (cast<" + SDNodeName +
1097                ">(N)->getAddressingMode() != ISD::UNINDEXED) "
1098                "return false;\n")
1099                   .str();
1100 
1101     if (isLoad()) {
1102       if ((isNonExtLoad() + isAnyExtLoad() + isSignExtLoad() +
1103            isZeroExtLoad()) > 1)
1104         PrintFatalError(getOrigPatFragRecord()->getRecord()->getLoc(),
1105                         "IsNonExtLoad, IsAnyExtLoad, IsSignExtLoad, and "
1106                         "IsZeroExtLoad are mutually exclusive");
1107       if (isNonExtLoad())
1108         Code += "if (cast<LoadSDNode>(N)->getExtensionType() != "
1109                 "ISD::NON_EXTLOAD) return false;\n";
1110       if (isAnyExtLoad())
1111         Code += "if (cast<LoadSDNode>(N)->getExtensionType() != ISD::EXTLOAD) "
1112                 "return false;\n";
1113       if (isSignExtLoad())
1114         Code += "if (cast<LoadSDNode>(N)->getExtensionType() != ISD::SEXTLOAD) "
1115                 "return false;\n";
1116       if (isZeroExtLoad())
1117         Code += "if (cast<LoadSDNode>(N)->getExtensionType() != ISD::ZEXTLOAD) "
1118                 "return false;\n";
1119     } else {
1120       if ((isNonTruncStore() + isTruncStore()) > 1)
1121         PrintFatalError(
1122             getOrigPatFragRecord()->getRecord()->getLoc(),
1123             "IsNonTruncStore, and IsTruncStore are mutually exclusive");
1124       if (isNonTruncStore())
1125         Code +=
1126             " if (cast<StoreSDNode>(N)->isTruncatingStore()) return false;\n";
1127       if (isTruncStore())
1128         Code +=
1129             " if (!cast<StoreSDNode>(N)->isTruncatingStore()) return false;\n";
1130     }
1131 
1132     Record *ScalarMemoryVT = getScalarMemoryVT();
1133 
1134     if (ScalarMemoryVT)
1135       Code += ("if (cast<" + SDNodeName +
1136                ">(N)->getMemoryVT().getScalarType() != MVT::" +
1137                ScalarMemoryVT->getName() + ") return false;\n")
1138                   .str();
1139   }
1140 
1141   if (hasNoUse())
1142     Code += "if (!SDValue(N, 0).use_empty()) return false;\n";
1143   if (hasOneUse())
1144     Code += "if (!SDValue(N, 0).hasOneUse()) return false;\n";
1145 
1146   std::string PredicateCode =
1147       std::string(PatFragRec->getRecord()->getValueAsString("PredicateCode"));
1148 
1149   Code += PredicateCode;
1150 
1151   if (PredicateCode.empty() && !Code.empty())
1152     Code += "return true;\n";
1153 
1154   return Code;
1155 }
1156 
1157 bool TreePredicateFn::hasImmCode() const {
1158   return !PatFragRec->getRecord()->getValueAsString("ImmediateCode").empty();
1159 }
1160 
1161 std::string TreePredicateFn::getImmCode() const {
1162   return std::string(
1163       PatFragRec->getRecord()->getValueAsString("ImmediateCode"));
1164 }
1165 
1166 bool TreePredicateFn::immCodeUsesAPInt() const {
1167   return getOrigPatFragRecord()->getRecord()->getValueAsBit("IsAPInt");
1168 }
1169 
1170 bool TreePredicateFn::immCodeUsesAPFloat() const {
1171   bool Unset;
1172   // The return value will be false when IsAPFloat is unset.
1173   return getOrigPatFragRecord()->getRecord()->getValueAsBitOrUnset("IsAPFloat",
1174                                                                    Unset);
1175 }
1176 
1177 bool TreePredicateFn::isPredefinedPredicateEqualTo(StringRef Field,
1178                                                    bool Value) const {
1179   bool Unset;
1180   bool Result =
1181       getOrigPatFragRecord()->getRecord()->getValueAsBitOrUnset(Field, Unset);
1182   if (Unset)
1183     return false;
1184   return Result == Value;
1185 }
1186 bool TreePredicateFn::usesOperands() const {
1187   return isPredefinedPredicateEqualTo("PredicateCodeUsesOperands", true);
1188 }
1189 bool TreePredicateFn::hasNoUse() const {
1190   return isPredefinedPredicateEqualTo("HasNoUse", true);
1191 }
1192 bool TreePredicateFn::hasOneUse() const {
1193   return isPredefinedPredicateEqualTo("HasOneUse", true);
1194 }
1195 bool TreePredicateFn::isLoad() const {
1196   return isPredefinedPredicateEqualTo("IsLoad", true);
1197 }
1198 bool TreePredicateFn::isStore() const {
1199   return isPredefinedPredicateEqualTo("IsStore", true);
1200 }
1201 bool TreePredicateFn::isAtomic() const {
1202   return isPredefinedPredicateEqualTo("IsAtomic", true);
1203 }
1204 bool TreePredicateFn::isUnindexed() const {
1205   return isPredefinedPredicateEqualTo("IsUnindexed", true);
1206 }
1207 bool TreePredicateFn::isNonExtLoad() const {
1208   return isPredefinedPredicateEqualTo("IsNonExtLoad", true);
1209 }
1210 bool TreePredicateFn::isAnyExtLoad() const {
1211   return isPredefinedPredicateEqualTo("IsAnyExtLoad", true);
1212 }
1213 bool TreePredicateFn::isSignExtLoad() const {
1214   return isPredefinedPredicateEqualTo("IsSignExtLoad", true);
1215 }
1216 bool TreePredicateFn::isZeroExtLoad() const {
1217   return isPredefinedPredicateEqualTo("IsZeroExtLoad", true);
1218 }
1219 bool TreePredicateFn::isNonTruncStore() const {
1220   return isPredefinedPredicateEqualTo("IsTruncStore", false);
1221 }
1222 bool TreePredicateFn::isTruncStore() const {
1223   return isPredefinedPredicateEqualTo("IsTruncStore", true);
1224 }
1225 bool TreePredicateFn::isAtomicOrderingMonotonic() const {
1226   return isPredefinedPredicateEqualTo("IsAtomicOrderingMonotonic", true);
1227 }
1228 bool TreePredicateFn::isAtomicOrderingAcquire() const {
1229   return isPredefinedPredicateEqualTo("IsAtomicOrderingAcquire", true);
1230 }
1231 bool TreePredicateFn::isAtomicOrderingRelease() const {
1232   return isPredefinedPredicateEqualTo("IsAtomicOrderingRelease", true);
1233 }
1234 bool TreePredicateFn::isAtomicOrderingAcquireRelease() const {
1235   return isPredefinedPredicateEqualTo("IsAtomicOrderingAcquireRelease", true);
1236 }
1237 bool TreePredicateFn::isAtomicOrderingSequentiallyConsistent() const {
1238   return isPredefinedPredicateEqualTo("IsAtomicOrderingSequentiallyConsistent",
1239                                       true);
1240 }
1241 bool TreePredicateFn::isAtomicOrderingAcquireOrStronger() const {
1242   return isPredefinedPredicateEqualTo("IsAtomicOrderingAcquireOrStronger",
1243                                       true);
1244 }
1245 bool TreePredicateFn::isAtomicOrderingWeakerThanAcquire() const {
1246   return isPredefinedPredicateEqualTo("IsAtomicOrderingAcquireOrStronger",
1247                                       false);
1248 }
1249 bool TreePredicateFn::isAtomicOrderingReleaseOrStronger() const {
1250   return isPredefinedPredicateEqualTo("IsAtomicOrderingReleaseOrStronger",
1251                                       true);
1252 }
1253 bool TreePredicateFn::isAtomicOrderingWeakerThanRelease() const {
1254   return isPredefinedPredicateEqualTo("IsAtomicOrderingReleaseOrStronger",
1255                                       false);
1256 }
1257 Record *TreePredicateFn::getMemoryVT() const {
1258   Record *R = getOrigPatFragRecord()->getRecord();
1259   if (R->isValueUnset("MemoryVT"))
1260     return nullptr;
1261   return R->getValueAsDef("MemoryVT");
1262 }
1263 
1264 ListInit *TreePredicateFn::getAddressSpaces() const {
1265   Record *R = getOrigPatFragRecord()->getRecord();
1266   if (R->isValueUnset("AddressSpaces"))
1267     return nullptr;
1268   return R->getValueAsListInit("AddressSpaces");
1269 }
1270 
1271 int64_t TreePredicateFn::getMinAlignment() const {
1272   Record *R = getOrigPatFragRecord()->getRecord();
1273   if (R->isValueUnset("MinAlignment"))
1274     return 0;
1275   return R->getValueAsInt("MinAlignment");
1276 }
1277 
1278 Record *TreePredicateFn::getScalarMemoryVT() const {
1279   Record *R = getOrigPatFragRecord()->getRecord();
1280   if (R->isValueUnset("ScalarMemoryVT"))
1281     return nullptr;
1282   return R->getValueAsDef("ScalarMemoryVT");
1283 }
1284 bool TreePredicateFn::hasGISelPredicateCode() const {
1285   return !PatFragRec->getRecord()
1286               ->getValueAsString("GISelPredicateCode")
1287               .empty();
1288 }
1289 std::string TreePredicateFn::getGISelPredicateCode() const {
1290   return std::string(
1291       PatFragRec->getRecord()->getValueAsString("GISelPredicateCode"));
1292 }
1293 
1294 StringRef TreePredicateFn::getImmType() const {
1295   if (immCodeUsesAPInt())
1296     return "const APInt &";
1297   if (immCodeUsesAPFloat())
1298     return "const APFloat &";
1299   return "int64_t";
1300 }
1301 
1302 StringRef TreePredicateFn::getImmTypeIdentifier() const {
1303   if (immCodeUsesAPInt())
1304     return "APInt";
1305   if (immCodeUsesAPFloat())
1306     return "APFloat";
1307   return "I64";
1308 }
1309 
1310 /// isAlwaysTrue - Return true if this is a noop predicate.
1311 bool TreePredicateFn::isAlwaysTrue() const {
1312   return !hasPredCode() && !hasImmCode();
1313 }
1314 
1315 /// Return the name to use in the generated code to reference this, this is
1316 /// "Predicate_foo" if from a pattern fragment "foo".
1317 std::string TreePredicateFn::getFnName() const {
1318   return "Predicate_" + PatFragRec->getRecord()->getName().str();
1319 }
1320 
1321 /// getCodeToRunOnSDNode - Return the code for the function body that
1322 /// evaluates this predicate.  The argument is expected to be in "Node",
1323 /// not N.  This handles casting and conversion to a concrete node type as
1324 /// appropriate.
1325 std::string TreePredicateFn::getCodeToRunOnSDNode() const {
1326   // Handle immediate predicates first.
1327   std::string ImmCode = getImmCode();
1328   if (!ImmCode.empty()) {
1329     if (isLoad())
1330       PrintFatalError(getOrigPatFragRecord()->getRecord()->getLoc(),
1331                       "IsLoad cannot be used with ImmLeaf or its subclasses");
1332     if (isStore())
1333       PrintFatalError(getOrigPatFragRecord()->getRecord()->getLoc(),
1334                       "IsStore cannot be used with ImmLeaf or its subclasses");
1335     if (isUnindexed())
1336       PrintFatalError(
1337           getOrigPatFragRecord()->getRecord()->getLoc(),
1338           "IsUnindexed cannot be used with ImmLeaf or its subclasses");
1339     if (isNonExtLoad())
1340       PrintFatalError(
1341           getOrigPatFragRecord()->getRecord()->getLoc(),
1342           "IsNonExtLoad cannot be used with ImmLeaf or its subclasses");
1343     if (isAnyExtLoad())
1344       PrintFatalError(
1345           getOrigPatFragRecord()->getRecord()->getLoc(),
1346           "IsAnyExtLoad cannot be used with ImmLeaf or its subclasses");
1347     if (isSignExtLoad())
1348       PrintFatalError(
1349           getOrigPatFragRecord()->getRecord()->getLoc(),
1350           "IsSignExtLoad cannot be used with ImmLeaf or its subclasses");
1351     if (isZeroExtLoad())
1352       PrintFatalError(
1353           getOrigPatFragRecord()->getRecord()->getLoc(),
1354           "IsZeroExtLoad cannot be used with ImmLeaf or its subclasses");
1355     if (isNonTruncStore())
1356       PrintFatalError(
1357           getOrigPatFragRecord()->getRecord()->getLoc(),
1358           "IsNonTruncStore cannot be used with ImmLeaf or its subclasses");
1359     if (isTruncStore())
1360       PrintFatalError(
1361           getOrigPatFragRecord()->getRecord()->getLoc(),
1362           "IsTruncStore cannot be used with ImmLeaf or its subclasses");
1363     if (getMemoryVT())
1364       PrintFatalError(getOrigPatFragRecord()->getRecord()->getLoc(),
1365                       "MemoryVT cannot be used with ImmLeaf or its subclasses");
1366     if (getScalarMemoryVT())
1367       PrintFatalError(
1368           getOrigPatFragRecord()->getRecord()->getLoc(),
1369           "ScalarMemoryVT cannot be used with ImmLeaf or its subclasses");
1370 
1371     std::string Result = ("    " + getImmType() + " Imm = ").str();
1372     if (immCodeUsesAPFloat())
1373       Result += "cast<ConstantFPSDNode>(Node)->getValueAPF();\n";
1374     else if (immCodeUsesAPInt())
1375       Result += "Node->getAsAPIntVal();\n";
1376     else
1377       Result += "cast<ConstantSDNode>(Node)->getSExtValue();\n";
1378     return Result + ImmCode;
1379   }
1380 
1381   // Handle arbitrary node predicates.
1382   assert(hasPredCode() && "Don't have any predicate code!");
1383 
1384   // If this is using PatFrags, there are multiple trees to search. They should
1385   // all have the same class.  FIXME: Is there a way to find a common
1386   // superclass?
1387   StringRef ClassName;
1388   for (const auto &Tree : PatFragRec->getTrees()) {
1389     StringRef TreeClassName;
1390     if (Tree->isLeaf())
1391       TreeClassName = "SDNode";
1392     else {
1393       Record *Op = Tree->getOperator();
1394       const SDNodeInfo &Info = PatFragRec->getDAGPatterns().getSDNodeInfo(Op);
1395       TreeClassName = Info.getSDClassName();
1396     }
1397 
1398     if (ClassName.empty())
1399       ClassName = TreeClassName;
1400     else if (ClassName != TreeClassName) {
1401       PrintFatalError(getOrigPatFragRecord()->getRecord()->getLoc(),
1402                       "PatFrags trees do not have consistent class");
1403     }
1404   }
1405 
1406   std::string Result;
1407   if (ClassName == "SDNode")
1408     Result = "    SDNode *N = Node;\n";
1409   else
1410     Result = "    auto *N = cast<" + ClassName.str() + ">(Node);\n";
1411 
1412   return (Twine(Result) + "    (void)N;\n" + getPredCode()).str();
1413 }
1414 
1415 //===----------------------------------------------------------------------===//
1416 // PatternToMatch implementation
1417 //
1418 
1419 static bool isImmAllOnesAllZerosMatch(const TreePatternNode &P) {
1420   if (!P.isLeaf())
1421     return false;
1422   DefInit *DI = dyn_cast<DefInit>(P.getLeafValue());
1423   if (!DI)
1424     return false;
1425 
1426   Record *R = DI->getDef();
1427   return R->getName() == "immAllOnesV" || R->getName() == "immAllZerosV";
1428 }
1429 
1430 /// getPatternSize - Return the 'size' of this pattern.  We want to match large
1431 /// patterns before small ones.  This is used to determine the size of a
1432 /// pattern.
1433 static unsigned getPatternSize(const TreePatternNode &P,
1434                                const CodeGenDAGPatterns &CGP) {
1435   unsigned Size = 3; // The node itself.
1436   // If the root node is a ConstantSDNode, increases its size.
1437   // e.g. (set R32:$dst, 0).
1438   if (P.isLeaf() && isa<IntInit>(P.getLeafValue()))
1439     Size += 2;
1440 
1441   if (const ComplexPattern *AM = P.getComplexPatternInfo(CGP)) {
1442     Size += AM->getComplexity();
1443     // We don't want to count any children twice, so return early.
1444     return Size;
1445   }
1446 
1447   // If this node has some predicate function that must match, it adds to the
1448   // complexity of this node.
1449   if (!P.getPredicateCalls().empty())
1450     ++Size;
1451 
1452   // Count children in the count if they are also nodes.
1453   for (unsigned i = 0, e = P.getNumChildren(); i != e; ++i) {
1454     const TreePatternNode &Child = P.getChild(i);
1455     if (!Child.isLeaf() && Child.getNumTypes()) {
1456       const TypeSetByHwMode &T0 = Child.getExtType(0);
1457       // At this point, all variable type sets should be simple, i.e. only
1458       // have a default mode.
1459       if (T0.getMachineValueType() != MVT::Other) {
1460         Size += getPatternSize(Child, CGP);
1461         continue;
1462       }
1463     }
1464     if (Child.isLeaf()) {
1465       if (isa<IntInit>(Child.getLeafValue()))
1466         Size += 5; // Matches a ConstantSDNode (+3) and a specific value (+2).
1467       else if (Child.getComplexPatternInfo(CGP))
1468         Size += getPatternSize(Child, CGP);
1469       else if (isImmAllOnesAllZerosMatch(Child))
1470         Size += 4; // Matches a build_vector(+3) and a predicate (+1).
1471       else if (!Child.getPredicateCalls().empty())
1472         ++Size;
1473     }
1474   }
1475 
1476   return Size;
1477 }
1478 
1479 /// Compute the complexity metric for the input pattern.  This roughly
1480 /// corresponds to the number of nodes that are covered.
1481 int PatternToMatch::getPatternComplexity(const CodeGenDAGPatterns &CGP) const {
1482   return getPatternSize(getSrcPattern(), CGP) + getAddedComplexity();
1483 }
1484 
1485 void PatternToMatch::getPredicateRecords(
1486     SmallVectorImpl<Record *> &PredicateRecs) const {
1487   for (Init *I : Predicates->getValues()) {
1488     if (DefInit *Pred = dyn_cast<DefInit>(I)) {
1489       Record *Def = Pred->getDef();
1490       if (!Def->isSubClassOf("Predicate")) {
1491 #ifndef NDEBUG
1492         Def->dump();
1493 #endif
1494         llvm_unreachable("Unknown predicate type!");
1495       }
1496       PredicateRecs.push_back(Def);
1497     }
1498   }
1499   // Sort so that different orders get canonicalized to the same string.
1500   llvm::sort(PredicateRecs, LessRecord());
1501   // Remove duplicate predicates.
1502   PredicateRecs.erase(llvm::unique(PredicateRecs), PredicateRecs.end());
1503 }
1504 
1505 /// getPredicateCheck - Return a single string containing all of this
1506 /// pattern's predicates concatenated with "&&" operators.
1507 ///
1508 std::string PatternToMatch::getPredicateCheck() const {
1509   SmallVector<Record *, 4> PredicateRecs;
1510   getPredicateRecords(PredicateRecs);
1511 
1512   SmallString<128> PredicateCheck;
1513   raw_svector_ostream OS(PredicateCheck);
1514   ListSeparator LS(" && ");
1515   for (Record *Pred : PredicateRecs) {
1516     StringRef CondString = Pred->getValueAsString("CondString");
1517     if (CondString.empty())
1518       continue;
1519     OS << LS << '(' << CondString << ')';
1520   }
1521 
1522   if (!HwModeFeatures.empty())
1523     OS << LS << HwModeFeatures;
1524 
1525   return std::string(PredicateCheck);
1526 }
1527 
1528 //===----------------------------------------------------------------------===//
1529 // SDTypeConstraint implementation
1530 //
1531 
1532 SDTypeConstraint::SDTypeConstraint(Record *R, const CodeGenHwModes &CGH) {
1533   OperandNo = R->getValueAsInt("OperandNum");
1534 
1535   if (R->isSubClassOf("SDTCisVT")) {
1536     ConstraintType = SDTCisVT;
1537     VVT = getValueTypeByHwMode(R->getValueAsDef("VT"), CGH);
1538     for (const auto &P : VVT)
1539       if (P.second == MVT::isVoid)
1540         PrintFatalError(R->getLoc(), "Cannot use 'Void' as type to SDTCisVT");
1541   } else if (R->isSubClassOf("SDTCisPtrTy")) {
1542     ConstraintType = SDTCisPtrTy;
1543   } else if (R->isSubClassOf("SDTCisInt")) {
1544     ConstraintType = SDTCisInt;
1545   } else if (R->isSubClassOf("SDTCisFP")) {
1546     ConstraintType = SDTCisFP;
1547   } else if (R->isSubClassOf("SDTCisVec")) {
1548     ConstraintType = SDTCisVec;
1549   } else if (R->isSubClassOf("SDTCisSameAs")) {
1550     ConstraintType = SDTCisSameAs;
1551     x.SDTCisSameAs_Info.OtherOperandNum = R->getValueAsInt("OtherOperandNum");
1552   } else if (R->isSubClassOf("SDTCisVTSmallerThanOp")) {
1553     ConstraintType = SDTCisVTSmallerThanOp;
1554     x.SDTCisVTSmallerThanOp_Info.OtherOperandNum =
1555         R->getValueAsInt("OtherOperandNum");
1556   } else if (R->isSubClassOf("SDTCisOpSmallerThanOp")) {
1557     ConstraintType = SDTCisOpSmallerThanOp;
1558     x.SDTCisOpSmallerThanOp_Info.BigOperandNum =
1559         R->getValueAsInt("BigOperandNum");
1560   } else if (R->isSubClassOf("SDTCisEltOfVec")) {
1561     ConstraintType = SDTCisEltOfVec;
1562     x.SDTCisEltOfVec_Info.OtherOperandNum = R->getValueAsInt("OtherOpNum");
1563   } else if (R->isSubClassOf("SDTCisSubVecOfVec")) {
1564     ConstraintType = SDTCisSubVecOfVec;
1565     x.SDTCisSubVecOfVec_Info.OtherOperandNum = R->getValueAsInt("OtherOpNum");
1566   } else if (R->isSubClassOf("SDTCVecEltisVT")) {
1567     ConstraintType = SDTCVecEltisVT;
1568     VVT = getValueTypeByHwMode(R->getValueAsDef("VT"), CGH);
1569     for (const auto &P : VVT) {
1570       MVT T = P.second;
1571       if (T.isVector())
1572         PrintFatalError(R->getLoc(),
1573                         "Cannot use vector type as SDTCVecEltisVT");
1574       if (!T.isInteger() && !T.isFloatingPoint())
1575         PrintFatalError(R->getLoc(), "Must use integer or floating point type "
1576                                      "as SDTCVecEltisVT");
1577     }
1578   } else if (R->isSubClassOf("SDTCisSameNumEltsAs")) {
1579     ConstraintType = SDTCisSameNumEltsAs;
1580     x.SDTCisSameNumEltsAs_Info.OtherOperandNum =
1581         R->getValueAsInt("OtherOperandNum");
1582   } else if (R->isSubClassOf("SDTCisSameSizeAs")) {
1583     ConstraintType = SDTCisSameSizeAs;
1584     x.SDTCisSameSizeAs_Info.OtherOperandNum =
1585         R->getValueAsInt("OtherOperandNum");
1586   } else {
1587     PrintFatalError(R->getLoc(),
1588                     "Unrecognized SDTypeConstraint '" + R->getName() + "'!\n");
1589   }
1590 }
1591 
1592 /// getOperandNum - Return the node corresponding to operand #OpNo in tree
1593 /// N, and the result number in ResNo.
1594 static TreePatternNode &getOperandNum(unsigned OpNo, TreePatternNode &N,
1595                                       const SDNodeInfo &NodeInfo,
1596                                       unsigned &ResNo) {
1597   unsigned NumResults = NodeInfo.getNumResults();
1598   if (OpNo < NumResults) {
1599     ResNo = OpNo;
1600     return N;
1601   }
1602 
1603   OpNo -= NumResults;
1604 
1605   if (OpNo >= N.getNumChildren()) {
1606     PrintFatalError([&N, OpNo, NumResults](raw_ostream &OS) {
1607       OS << "Invalid operand number in type constraint " << (OpNo + NumResults);
1608       N.print(OS);
1609     });
1610   }
1611   return N.getChild(OpNo);
1612 }
1613 
1614 /// ApplyTypeConstraint - Given a node in a pattern, apply this type
1615 /// constraint to the nodes operands.  This returns true if it makes a
1616 /// change, false otherwise.  If a type contradiction is found, flag an error.
1617 bool SDTypeConstraint::ApplyTypeConstraint(TreePatternNode &N,
1618                                            const SDNodeInfo &NodeInfo,
1619                                            TreePattern &TP) const {
1620   if (TP.hasError())
1621     return false;
1622 
1623   unsigned ResNo = 0; // The result number being referenced.
1624   TreePatternNode &NodeToApply = getOperandNum(OperandNo, N, NodeInfo, ResNo);
1625   TypeInfer &TI = TP.getInfer();
1626 
1627   switch (ConstraintType) {
1628   case SDTCisVT:
1629     // Operand must be a particular type.
1630     return NodeToApply.UpdateNodeType(ResNo, VVT, TP);
1631   case SDTCisPtrTy:
1632     // Operand must be same as target pointer type.
1633     return NodeToApply.UpdateNodeType(ResNo, MVT::iPTR, TP);
1634   case SDTCisInt:
1635     // Require it to be one of the legal integer VTs.
1636     return TI.EnforceInteger(NodeToApply.getExtType(ResNo));
1637   case SDTCisFP:
1638     // Require it to be one of the legal fp VTs.
1639     return TI.EnforceFloatingPoint(NodeToApply.getExtType(ResNo));
1640   case SDTCisVec:
1641     // Require it to be one of the legal vector VTs.
1642     return TI.EnforceVector(NodeToApply.getExtType(ResNo));
1643   case SDTCisSameAs: {
1644     unsigned OResNo = 0;
1645     TreePatternNode &OtherNode =
1646         getOperandNum(x.SDTCisSameAs_Info.OtherOperandNum, N, NodeInfo, OResNo);
1647     return (int)NodeToApply.UpdateNodeType(ResNo, OtherNode.getExtType(OResNo),
1648                                            TP) |
1649            (int)OtherNode.UpdateNodeType(OResNo, NodeToApply.getExtType(ResNo),
1650                                          TP);
1651   }
1652   case SDTCisVTSmallerThanOp: {
1653     // The NodeToApply must be a leaf node that is a VT.  OtherOperandNum must
1654     // have an integer type that is smaller than the VT.
1655     if (!NodeToApply.isLeaf() || !isa<DefInit>(NodeToApply.getLeafValue()) ||
1656         !cast<DefInit>(NodeToApply.getLeafValue())
1657              ->getDef()
1658              ->isSubClassOf("ValueType")) {
1659       TP.error(N.getOperator()->getName() + " expects a VT operand!");
1660       return false;
1661     }
1662     DefInit *DI = cast<DefInit>(NodeToApply.getLeafValue());
1663     const CodeGenTarget &T = TP.getDAGPatterns().getTargetInfo();
1664     auto VVT = getValueTypeByHwMode(DI->getDef(), T.getHwModes());
1665     TypeSetByHwMode TypeListTmp(VVT);
1666 
1667     unsigned OResNo = 0;
1668     TreePatternNode &OtherNode = getOperandNum(
1669         x.SDTCisVTSmallerThanOp_Info.OtherOperandNum, N, NodeInfo, OResNo);
1670 
1671     return TI.EnforceSmallerThan(TypeListTmp, OtherNode.getExtType(OResNo),
1672                                  /*SmallIsVT*/ true);
1673   }
1674   case SDTCisOpSmallerThanOp: {
1675     unsigned BResNo = 0;
1676     TreePatternNode &BigOperand = getOperandNum(
1677         x.SDTCisOpSmallerThanOp_Info.BigOperandNum, N, NodeInfo, BResNo);
1678     return TI.EnforceSmallerThan(NodeToApply.getExtType(ResNo),
1679                                  BigOperand.getExtType(BResNo));
1680   }
1681   case SDTCisEltOfVec: {
1682     unsigned VResNo = 0;
1683     TreePatternNode &VecOperand = getOperandNum(
1684         x.SDTCisEltOfVec_Info.OtherOperandNum, N, NodeInfo, VResNo);
1685     // Filter vector types out of VecOperand that don't have the right element
1686     // type.
1687     return TI.EnforceVectorEltTypeIs(VecOperand.getExtType(VResNo),
1688                                      NodeToApply.getExtType(ResNo));
1689   }
1690   case SDTCisSubVecOfVec: {
1691     unsigned VResNo = 0;
1692     TreePatternNode &BigVecOperand = getOperandNum(
1693         x.SDTCisSubVecOfVec_Info.OtherOperandNum, N, NodeInfo, VResNo);
1694 
1695     // Filter vector types out of BigVecOperand that don't have the
1696     // right subvector type.
1697     return TI.EnforceVectorSubVectorTypeIs(BigVecOperand.getExtType(VResNo),
1698                                            NodeToApply.getExtType(ResNo));
1699   }
1700   case SDTCVecEltisVT: {
1701     return TI.EnforceVectorEltTypeIs(NodeToApply.getExtType(ResNo), VVT);
1702   }
1703   case SDTCisSameNumEltsAs: {
1704     unsigned OResNo = 0;
1705     TreePatternNode &OtherNode = getOperandNum(
1706         x.SDTCisSameNumEltsAs_Info.OtherOperandNum, N, NodeInfo, OResNo);
1707     return TI.EnforceSameNumElts(OtherNode.getExtType(OResNo),
1708                                  NodeToApply.getExtType(ResNo));
1709   }
1710   case SDTCisSameSizeAs: {
1711     unsigned OResNo = 0;
1712     TreePatternNode &OtherNode = getOperandNum(
1713         x.SDTCisSameSizeAs_Info.OtherOperandNum, N, NodeInfo, OResNo);
1714     return TI.EnforceSameSize(OtherNode.getExtType(OResNo),
1715                               NodeToApply.getExtType(ResNo));
1716   }
1717   }
1718   llvm_unreachable("Invalid ConstraintType!");
1719 }
1720 
1721 // Update the node type to match an instruction operand or result as specified
1722 // in the ins or outs lists on the instruction definition. Return true if the
1723 // type was actually changed.
1724 bool TreePatternNode::UpdateNodeTypeFromInst(unsigned ResNo, Record *Operand,
1725                                              TreePattern &TP) {
1726   // The 'unknown' operand indicates that types should be inferred from the
1727   // context.
1728   if (Operand->isSubClassOf("unknown_class"))
1729     return false;
1730 
1731   // The Operand class specifies a type directly.
1732   if (Operand->isSubClassOf("Operand")) {
1733     Record *R = Operand->getValueAsDef("Type");
1734     const CodeGenTarget &T = TP.getDAGPatterns().getTargetInfo();
1735     return UpdateNodeType(ResNo, getValueTypeByHwMode(R, T.getHwModes()), TP);
1736   }
1737 
1738   // PointerLikeRegClass has a type that is determined at runtime.
1739   if (Operand->isSubClassOf("PointerLikeRegClass"))
1740     return UpdateNodeType(ResNo, MVT::iPTR, TP);
1741 
1742   // Both RegisterClass and RegisterOperand operands derive their types from a
1743   // register class def.
1744   Record *RC = nullptr;
1745   if (Operand->isSubClassOf("RegisterClass"))
1746     RC = Operand;
1747   else if (Operand->isSubClassOf("RegisterOperand"))
1748     RC = Operand->getValueAsDef("RegClass");
1749 
1750   assert(RC && "Unknown operand type");
1751   CodeGenTarget &Tgt = TP.getDAGPatterns().getTargetInfo();
1752   return UpdateNodeType(ResNo, Tgt.getRegisterClass(RC).getValueTypes(), TP);
1753 }
1754 
1755 bool TreePatternNode::ContainsUnresolvedType(TreePattern &TP) const {
1756   for (unsigned i = 0, e = Types.size(); i != e; ++i)
1757     if (!TP.getInfer().isConcrete(Types[i], true))
1758       return true;
1759   for (unsigned i = 0, e = getNumChildren(); i != e; ++i)
1760     if (getChild(i).ContainsUnresolvedType(TP))
1761       return true;
1762   return false;
1763 }
1764 
1765 bool TreePatternNode::hasProperTypeByHwMode() const {
1766   for (const TypeSetByHwMode &S : Types)
1767     if (!S.isSimple())
1768       return true;
1769   for (const TreePatternNodePtr &C : Children)
1770     if (C->hasProperTypeByHwMode())
1771       return true;
1772   return false;
1773 }
1774 
1775 bool TreePatternNode::hasPossibleType() const {
1776   for (const TypeSetByHwMode &S : Types)
1777     if (!S.isPossible())
1778       return false;
1779   for (const TreePatternNodePtr &C : Children)
1780     if (!C->hasPossibleType())
1781       return false;
1782   return true;
1783 }
1784 
1785 bool TreePatternNode::setDefaultMode(unsigned Mode) {
1786   for (TypeSetByHwMode &S : Types) {
1787     S.makeSimple(Mode);
1788     // Check if the selected mode had a type conflict.
1789     if (S.get(DefaultMode).empty())
1790       return false;
1791   }
1792   for (const TreePatternNodePtr &C : Children)
1793     if (!C->setDefaultMode(Mode))
1794       return false;
1795   return true;
1796 }
1797 
1798 //===----------------------------------------------------------------------===//
1799 // SDNodeInfo implementation
1800 //
1801 SDNodeInfo::SDNodeInfo(Record *R, const CodeGenHwModes &CGH) : Def(R) {
1802   EnumName = R->getValueAsString("Opcode");
1803   SDClassName = R->getValueAsString("SDClass");
1804   Record *TypeProfile = R->getValueAsDef("TypeProfile");
1805   NumResults = TypeProfile->getValueAsInt("NumResults");
1806   NumOperands = TypeProfile->getValueAsInt("NumOperands");
1807 
1808   // Parse the properties.
1809   Properties = parseSDPatternOperatorProperties(R);
1810 
1811   // Parse the type constraints.
1812   std::vector<Record *> ConstraintList =
1813       TypeProfile->getValueAsListOfDefs("Constraints");
1814   for (Record *R : ConstraintList)
1815     TypeConstraints.emplace_back(R, CGH);
1816 }
1817 
1818 /// getKnownType - If the type constraints on this node imply a fixed type
1819 /// (e.g. all stores return void, etc), then return it as an
1820 /// MVT::SimpleValueType.  Otherwise, return EEVT::Other.
1821 MVT::SimpleValueType SDNodeInfo::getKnownType(unsigned ResNo) const {
1822   unsigned NumResults = getNumResults();
1823   assert(NumResults <= 1 &&
1824          "We only work with nodes with zero or one result so far!");
1825   assert(ResNo == 0 && "Only handles single result nodes so far");
1826 
1827   for (const SDTypeConstraint &Constraint : TypeConstraints) {
1828     // Make sure that this applies to the correct node result.
1829     if (Constraint.OperandNo >= NumResults) // FIXME: need value #
1830       continue;
1831 
1832     switch (Constraint.ConstraintType) {
1833     default:
1834       break;
1835     case SDTypeConstraint::SDTCisVT:
1836       if (Constraint.VVT.isSimple())
1837         return Constraint.VVT.getSimple().SimpleTy;
1838       break;
1839     case SDTypeConstraint::SDTCisPtrTy:
1840       return MVT::iPTR;
1841     }
1842   }
1843   return MVT::Other;
1844 }
1845 
1846 //===----------------------------------------------------------------------===//
1847 // TreePatternNode implementation
1848 //
1849 
1850 static unsigned GetNumNodeResults(Record *Operator, CodeGenDAGPatterns &CDP) {
1851   if (Operator->getName() == "set" || Operator->getName() == "implicit")
1852     return 0; // All return nothing.
1853 
1854   if (Operator->isSubClassOf("Intrinsic"))
1855     return CDP.getIntrinsic(Operator).IS.RetTys.size();
1856 
1857   if (Operator->isSubClassOf("SDNode"))
1858     return CDP.getSDNodeInfo(Operator).getNumResults();
1859 
1860   if (Operator->isSubClassOf("PatFrags")) {
1861     // If we've already parsed this pattern fragment, get it.  Otherwise, handle
1862     // the forward reference case where one pattern fragment references another
1863     // before it is processed.
1864     if (TreePattern *PFRec = CDP.getPatternFragmentIfRead(Operator)) {
1865       // The number of results of a fragment with alternative records is the
1866       // maximum number of results across all alternatives.
1867       unsigned NumResults = 0;
1868       for (const auto &T : PFRec->getTrees())
1869         NumResults = std::max(NumResults, T->getNumTypes());
1870       return NumResults;
1871     }
1872 
1873     ListInit *LI = Operator->getValueAsListInit("Fragments");
1874     assert(LI && "Invalid Fragment");
1875     unsigned NumResults = 0;
1876     for (Init *I : LI->getValues()) {
1877       Record *Op = nullptr;
1878       if (DagInit *Dag = dyn_cast<DagInit>(I))
1879         if (DefInit *DI = dyn_cast<DefInit>(Dag->getOperator()))
1880           Op = DI->getDef();
1881       assert(Op && "Invalid Fragment");
1882       NumResults = std::max(NumResults, GetNumNodeResults(Op, CDP));
1883     }
1884     return NumResults;
1885   }
1886 
1887   if (Operator->isSubClassOf("Instruction")) {
1888     CodeGenInstruction &InstInfo = CDP.getTargetInfo().getInstruction(Operator);
1889 
1890     unsigned NumDefsToAdd = InstInfo.Operands.NumDefs;
1891 
1892     // Subtract any defaulted outputs.
1893     for (unsigned i = 0; i != InstInfo.Operands.NumDefs; ++i) {
1894       Record *OperandNode = InstInfo.Operands[i].Rec;
1895 
1896       if (OperandNode->isSubClassOf("OperandWithDefaultOps") &&
1897           !CDP.getDefaultOperand(OperandNode).DefaultOps.empty())
1898         --NumDefsToAdd;
1899     }
1900 
1901     // Add on one implicit def if it has a resolvable type.
1902     if (InstInfo.HasOneImplicitDefWithKnownVT(CDP.getTargetInfo()) !=
1903         MVT::Other)
1904       ++NumDefsToAdd;
1905     return NumDefsToAdd;
1906   }
1907 
1908   if (Operator->isSubClassOf("SDNodeXForm"))
1909     return 1; // FIXME: Generalize SDNodeXForm
1910 
1911   if (Operator->isSubClassOf("ValueType"))
1912     return 1; // A type-cast of one result.
1913 
1914   if (Operator->isSubClassOf("ComplexPattern"))
1915     return 1;
1916 
1917   errs() << *Operator;
1918   PrintFatalError("Unhandled node in GetNumNodeResults");
1919 }
1920 
1921 void TreePatternNode::print(raw_ostream &OS) const {
1922   if (isLeaf())
1923     OS << *getLeafValue();
1924   else
1925     OS << '(' << getOperator()->getName();
1926 
1927   for (unsigned i = 0, e = Types.size(); i != e; ++i) {
1928     OS << ':';
1929     getExtType(i).writeToStream(OS);
1930   }
1931 
1932   if (!isLeaf()) {
1933     if (getNumChildren() != 0) {
1934       OS << " ";
1935       ListSeparator LS;
1936       for (unsigned i = 0, e = getNumChildren(); i != e; ++i) {
1937         OS << LS;
1938         getChild(i).print(OS);
1939       }
1940     }
1941     OS << ")";
1942   }
1943 
1944   for (const TreePredicateCall &Pred : PredicateCalls) {
1945     OS << "<<P:";
1946     if (Pred.Scope)
1947       OS << Pred.Scope << ":";
1948     OS << Pred.Fn.getFnName() << ">>";
1949   }
1950   if (TransformFn)
1951     OS << "<<X:" << TransformFn->getName() << ">>";
1952   if (!getName().empty())
1953     OS << ":$" << getName();
1954 
1955   for (const ScopedName &Name : NamesAsPredicateArg)
1956     OS << ":$pred:" << Name.getScope() << ":" << Name.getIdentifier();
1957 }
1958 void TreePatternNode::dump() const { print(errs()); }
1959 
1960 /// isIsomorphicTo - Return true if this node is recursively
1961 /// isomorphic to the specified node.  For this comparison, the node's
1962 /// entire state is considered. The assigned name is ignored, since
1963 /// nodes with differing names are considered isomorphic. However, if
1964 /// the assigned name is present in the dependent variable set, then
1965 /// the assigned name is considered significant and the node is
1966 /// isomorphic if the names match.
1967 bool TreePatternNode::isIsomorphicTo(const TreePatternNode &N,
1968                                      const MultipleUseVarSet &DepVars) const {
1969   if (&N == this)
1970     return true;
1971   if (N.isLeaf() != isLeaf())
1972     return false;
1973 
1974   // Check operator of non-leaves early since it can be cheaper than checking
1975   // types.
1976   if (!isLeaf())
1977     if (N.getOperator() != getOperator() ||
1978         N.getNumChildren() != getNumChildren())
1979       return false;
1980 
1981   if (getExtTypes() != N.getExtTypes() ||
1982       getPredicateCalls() != N.getPredicateCalls() ||
1983       getTransformFn() != N.getTransformFn())
1984     return false;
1985 
1986   if (isLeaf()) {
1987     if (DefInit *DI = dyn_cast<DefInit>(getLeafValue())) {
1988       if (DefInit *NDI = dyn_cast<DefInit>(N.getLeafValue())) {
1989         return ((DI->getDef() == NDI->getDef()) &&
1990                 (!DepVars.contains(getName()) || getName() == N.getName()));
1991       }
1992     }
1993     return getLeafValue() == N.getLeafValue();
1994   }
1995 
1996   for (unsigned i = 0, e = getNumChildren(); i != e; ++i)
1997     if (!getChild(i).isIsomorphicTo(N.getChild(i), DepVars))
1998       return false;
1999   return true;
2000 }
2001 
2002 /// clone - Make a copy of this tree and all of its children.
2003 ///
2004 TreePatternNodePtr TreePatternNode::clone() const {
2005   TreePatternNodePtr New;
2006   if (isLeaf()) {
2007     New = makeIntrusiveRefCnt<TreePatternNode>(getLeafValue(), getNumTypes());
2008   } else {
2009     std::vector<TreePatternNodePtr> CChildren;
2010     CChildren.reserve(Children.size());
2011     for (unsigned i = 0, e = getNumChildren(); i != e; ++i)
2012       CChildren.push_back(getChild(i).clone());
2013     New = makeIntrusiveRefCnt<TreePatternNode>(
2014         getOperator(), std::move(CChildren), getNumTypes());
2015   }
2016   New->setName(getName());
2017   New->setNamesAsPredicateArg(getNamesAsPredicateArg());
2018   New->Types = Types;
2019   New->setPredicateCalls(getPredicateCalls());
2020   New->setGISelFlagsRecord(getGISelFlagsRecord());
2021   New->setTransformFn(getTransformFn());
2022   return New;
2023 }
2024 
2025 /// RemoveAllTypes - Recursively strip all the types of this tree.
2026 void TreePatternNode::RemoveAllTypes() {
2027   // Reset to unknown type.
2028   std::fill(Types.begin(), Types.end(), TypeSetByHwMode());
2029   if (isLeaf())
2030     return;
2031   for (unsigned i = 0, e = getNumChildren(); i != e; ++i)
2032     getChild(i).RemoveAllTypes();
2033 }
2034 
2035 /// SubstituteFormalArguments - Replace the formal arguments in this tree
2036 /// with actual values specified by ArgMap.
2037 void TreePatternNode::SubstituteFormalArguments(
2038     std::map<std::string, TreePatternNodePtr> &ArgMap) {
2039   if (isLeaf())
2040     return;
2041 
2042   for (unsigned i = 0, e = getNumChildren(); i != e; ++i) {
2043     TreePatternNode &Child = getChild(i);
2044     if (Child.isLeaf()) {
2045       Init *Val = Child.getLeafValue();
2046       // Note that, when substituting into an output pattern, Val might be an
2047       // UnsetInit.
2048       if (isa<UnsetInit>(Val) ||
2049           (isa<DefInit>(Val) &&
2050            cast<DefInit>(Val)->getDef()->getName() == "node")) {
2051         // We found a use of a formal argument, replace it with its value.
2052         TreePatternNodePtr NewChild = ArgMap[Child.getName()];
2053         assert(NewChild && "Couldn't find formal argument!");
2054         assert((Child.getPredicateCalls().empty() ||
2055                 NewChild->getPredicateCalls() == Child.getPredicateCalls()) &&
2056                "Non-empty child predicate clobbered!");
2057         setChild(i, std::move(NewChild));
2058       }
2059     } else {
2060       getChild(i).SubstituteFormalArguments(ArgMap);
2061     }
2062   }
2063 }
2064 
2065 /// InlinePatternFragments - If this pattern refers to any pattern
2066 /// fragments, return the set of inlined versions (this can be more than
2067 /// one if a PatFrags record has multiple alternatives).
2068 void TreePatternNode::InlinePatternFragments(
2069     TreePattern &TP, std::vector<TreePatternNodePtr> &OutAlternatives) {
2070 
2071   if (TP.hasError())
2072     return;
2073 
2074   if (isLeaf()) {
2075     OutAlternatives.push_back(this); // nothing to do.
2076     return;
2077   }
2078 
2079   Record *Op = getOperator();
2080 
2081   if (!Op->isSubClassOf("PatFrags")) {
2082     if (getNumChildren() == 0) {
2083       OutAlternatives.push_back(this);
2084       return;
2085     }
2086 
2087     // Recursively inline children nodes.
2088     std::vector<std::vector<TreePatternNodePtr>> ChildAlternatives(
2089         getNumChildren());
2090     for (unsigned i = 0, e = getNumChildren(); i != e; ++i) {
2091       TreePatternNodePtr Child = getChildShared(i);
2092       Child->InlinePatternFragments(TP, ChildAlternatives[i]);
2093       // If there are no alternatives for any child, there are no
2094       // alternatives for this expression as whole.
2095       if (ChildAlternatives[i].empty())
2096         return;
2097 
2098       assert((Child->getPredicateCalls().empty() ||
2099               llvm::all_of(ChildAlternatives[i],
2100                            [&](const TreePatternNodePtr &NewChild) {
2101                              return NewChild->getPredicateCalls() ==
2102                                     Child->getPredicateCalls();
2103                            })) &&
2104              "Non-empty child predicate clobbered!");
2105     }
2106 
2107     // The end result is an all-pairs construction of the resultant pattern.
2108     std::vector<unsigned> Idxs(ChildAlternatives.size());
2109     bool NotDone;
2110     do {
2111       // Create the variant and add it to the output list.
2112       std::vector<TreePatternNodePtr> NewChildren;
2113       NewChildren.reserve(ChildAlternatives.size());
2114       for (unsigned i = 0, e = ChildAlternatives.size(); i != e; ++i)
2115         NewChildren.push_back(ChildAlternatives[i][Idxs[i]]);
2116       TreePatternNodePtr R = makeIntrusiveRefCnt<TreePatternNode>(
2117           getOperator(), std::move(NewChildren), getNumTypes());
2118 
2119       // Copy over properties.
2120       R->setName(getName());
2121       R->setNamesAsPredicateArg(getNamesAsPredicateArg());
2122       R->setPredicateCalls(getPredicateCalls());
2123       R->setGISelFlagsRecord(getGISelFlagsRecord());
2124       R->setTransformFn(getTransformFn());
2125       for (unsigned i = 0, e = getNumTypes(); i != e; ++i)
2126         R->setType(i, getExtType(i));
2127       for (unsigned i = 0, e = getNumResults(); i != e; ++i)
2128         R->setResultIndex(i, getResultIndex(i));
2129 
2130       // Register alternative.
2131       OutAlternatives.push_back(R);
2132 
2133       // Increment indices to the next permutation by incrementing the
2134       // indices from last index backward, e.g., generate the sequence
2135       // [0, 0], [0, 1], [1, 0], [1, 1].
2136       int IdxsIdx;
2137       for (IdxsIdx = Idxs.size() - 1; IdxsIdx >= 0; --IdxsIdx) {
2138         if (++Idxs[IdxsIdx] == ChildAlternatives[IdxsIdx].size())
2139           Idxs[IdxsIdx] = 0;
2140         else
2141           break;
2142       }
2143       NotDone = (IdxsIdx >= 0);
2144     } while (NotDone);
2145 
2146     return;
2147   }
2148 
2149   // Otherwise, we found a reference to a fragment.  First, look up its
2150   // TreePattern record.
2151   TreePattern *Frag = TP.getDAGPatterns().getPatternFragment(Op);
2152 
2153   // Verify that we are passing the right number of operands.
2154   if (Frag->getNumArgs() != getNumChildren()) {
2155     TP.error("'" + Op->getName() + "' fragment requires " +
2156              Twine(Frag->getNumArgs()) + " operands!");
2157     return;
2158   }
2159 
2160   TreePredicateFn PredFn(Frag);
2161   unsigned Scope = 0;
2162   if (TreePredicateFn(Frag).usesOperands())
2163     Scope = TP.getDAGPatterns().allocateScope();
2164 
2165   // Compute the map of formal to actual arguments.
2166   std::map<std::string, TreePatternNodePtr> ArgMap;
2167   for (unsigned i = 0, e = Frag->getNumArgs(); i != e; ++i) {
2168     TreePatternNodePtr Child = getChildShared(i);
2169     if (Scope != 0) {
2170       Child = Child->clone();
2171       Child->addNameAsPredicateArg(ScopedName(Scope, Frag->getArgName(i)));
2172     }
2173     ArgMap[Frag->getArgName(i)] = Child;
2174   }
2175 
2176   // Loop over all fragment alternatives.
2177   for (const auto &Alternative : Frag->getTrees()) {
2178     TreePatternNodePtr FragTree = Alternative->clone();
2179 
2180     if (!PredFn.isAlwaysTrue())
2181       FragTree->addPredicateCall(PredFn, Scope);
2182 
2183     // Resolve formal arguments to their actual value.
2184     if (Frag->getNumArgs())
2185       FragTree->SubstituteFormalArguments(ArgMap);
2186 
2187     // Transfer types.  Note that the resolved alternative may have fewer
2188     // (but not more) results than the PatFrags node.
2189     FragTree->setName(getName());
2190     for (unsigned i = 0, e = FragTree->getNumTypes(); i != e; ++i)
2191       FragTree->UpdateNodeType(i, getExtType(i), TP);
2192 
2193     if (Op->isSubClassOf("GISelFlags"))
2194       FragTree->setGISelFlagsRecord(Op);
2195 
2196     // Transfer in the old predicates.
2197     for (const TreePredicateCall &Pred : getPredicateCalls())
2198       FragTree->addPredicateCall(Pred);
2199 
2200     // The fragment we inlined could have recursive inlining that is needed. See
2201     // if there are any pattern fragments in it and inline them as needed.
2202     FragTree->InlinePatternFragments(TP, OutAlternatives);
2203   }
2204 }
2205 
2206 /// getImplicitType - Check to see if the specified record has an implicit
2207 /// type which should be applied to it.  This will infer the type of register
2208 /// references from the register file information, for example.
2209 ///
2210 /// When Unnamed is set, return the type of a DAG operand with no name, such as
2211 /// the F8RC register class argument in:
2212 ///
2213 ///   (COPY_TO_REGCLASS GPR:$src, F8RC)
2214 ///
2215 /// When Unnamed is false, return the type of a named DAG operand such as the
2216 /// GPR:$src operand above.
2217 ///
2218 static TypeSetByHwMode getImplicitType(Record *R, unsigned ResNo,
2219                                        bool NotRegisters, bool Unnamed,
2220                                        TreePattern &TP) {
2221   CodeGenDAGPatterns &CDP = TP.getDAGPatterns();
2222 
2223   // Check to see if this is a register operand.
2224   if (R->isSubClassOf("RegisterOperand")) {
2225     assert(ResNo == 0 && "Regoperand ref only has one result!");
2226     if (NotRegisters)
2227       return TypeSetByHwMode(); // Unknown.
2228     Record *RegClass = R->getValueAsDef("RegClass");
2229     const CodeGenTarget &T = TP.getDAGPatterns().getTargetInfo();
2230     return TypeSetByHwMode(T.getRegisterClass(RegClass).getValueTypes());
2231   }
2232 
2233   // Check to see if this is a register or a register class.
2234   if (R->isSubClassOf("RegisterClass")) {
2235     assert(ResNo == 0 && "Regclass ref only has one result!");
2236     // An unnamed register class represents itself as an i32 immediate, for
2237     // example on a COPY_TO_REGCLASS instruction.
2238     if (Unnamed)
2239       return TypeSetByHwMode(MVT::i32);
2240 
2241     // In a named operand, the register class provides the possible set of
2242     // types.
2243     if (NotRegisters)
2244       return TypeSetByHwMode(); // Unknown.
2245     const CodeGenTarget &T = TP.getDAGPatterns().getTargetInfo();
2246     return TypeSetByHwMode(T.getRegisterClass(R).getValueTypes());
2247   }
2248 
2249   if (R->isSubClassOf("PatFrags")) {
2250     assert(ResNo == 0 && "FIXME: PatFrag with multiple results?");
2251     // Pattern fragment types will be resolved when they are inlined.
2252     return TypeSetByHwMode(); // Unknown.
2253   }
2254 
2255   if (R->isSubClassOf("Register")) {
2256     assert(ResNo == 0 && "Registers only produce one result!");
2257     if (NotRegisters)
2258       return TypeSetByHwMode(); // Unknown.
2259     const CodeGenTarget &T = TP.getDAGPatterns().getTargetInfo();
2260     return TypeSetByHwMode(T.getRegisterVTs(R));
2261   }
2262 
2263   if (R->isSubClassOf("SubRegIndex")) {
2264     assert(ResNo == 0 && "SubRegisterIndices only produce one result!");
2265     return TypeSetByHwMode(MVT::i32);
2266   }
2267 
2268   if (R->isSubClassOf("ValueType")) {
2269     assert(ResNo == 0 && "This node only has one result!");
2270     // An unnamed VTSDNode represents itself as an MVT::Other immediate.
2271     //
2272     //   (sext_inreg GPR:$src, i16)
2273     //                         ~~~
2274     if (Unnamed)
2275       return TypeSetByHwMode(MVT::Other);
2276     // With a name, the ValueType simply provides the type of the named
2277     // variable.
2278     //
2279     //   (sext_inreg i32:$src, i16)
2280     //               ~~~~~~~~
2281     if (NotRegisters)
2282       return TypeSetByHwMode(); // Unknown.
2283     const CodeGenHwModes &CGH = CDP.getTargetInfo().getHwModes();
2284     return TypeSetByHwMode(getValueTypeByHwMode(R, CGH));
2285   }
2286 
2287   if (R->isSubClassOf("CondCode")) {
2288     assert(ResNo == 0 && "This node only has one result!");
2289     // Using a CondCodeSDNode.
2290     return TypeSetByHwMode(MVT::Other);
2291   }
2292 
2293   if (R->isSubClassOf("ComplexPattern")) {
2294     assert(ResNo == 0 && "FIXME: ComplexPattern with multiple results?");
2295     if (NotRegisters)
2296       return TypeSetByHwMode(); // Unknown.
2297     Record *T = CDP.getComplexPattern(R).getValueType();
2298     const CodeGenHwModes &CGH = CDP.getTargetInfo().getHwModes();
2299     return TypeSetByHwMode(getValueTypeByHwMode(T, CGH));
2300   }
2301   if (R->isSubClassOf("PointerLikeRegClass")) {
2302     assert(ResNo == 0 && "Regclass can only have one result!");
2303     TypeSetByHwMode VTS(MVT::iPTR);
2304     TP.getInfer().expandOverloads(VTS);
2305     return VTS;
2306   }
2307 
2308   if (R->getName() == "node" || R->getName() == "srcvalue" ||
2309       R->getName() == "zero_reg" || R->getName() == "immAllOnesV" ||
2310       R->getName() == "immAllZerosV" || R->getName() == "undef_tied_input") {
2311     // Placeholder.
2312     return TypeSetByHwMode(); // Unknown.
2313   }
2314 
2315   if (R->isSubClassOf("Operand")) {
2316     const CodeGenHwModes &CGH = CDP.getTargetInfo().getHwModes();
2317     Record *T = R->getValueAsDef("Type");
2318     return TypeSetByHwMode(getValueTypeByHwMode(T, CGH));
2319   }
2320 
2321   TP.error("Unknown node flavor used in pattern: " + R->getName());
2322   return TypeSetByHwMode(MVT::Other);
2323 }
2324 
2325 /// getIntrinsicInfo - If this node corresponds to an intrinsic, return the
2326 /// CodeGenIntrinsic information for it, otherwise return a null pointer.
2327 const CodeGenIntrinsic *
2328 TreePatternNode::getIntrinsicInfo(const CodeGenDAGPatterns &CDP) const {
2329   if (getOperator() != CDP.get_intrinsic_void_sdnode() &&
2330       getOperator() != CDP.get_intrinsic_w_chain_sdnode() &&
2331       getOperator() != CDP.get_intrinsic_wo_chain_sdnode())
2332     return nullptr;
2333 
2334   unsigned IID = cast<IntInit>(getChild(0).getLeafValue())->getValue();
2335   return &CDP.getIntrinsicInfo(IID);
2336 }
2337 
2338 /// getComplexPatternInfo - If this node corresponds to a ComplexPattern,
2339 /// return the ComplexPattern information, otherwise return null.
2340 const ComplexPattern *
2341 TreePatternNode::getComplexPatternInfo(const CodeGenDAGPatterns &CGP) const {
2342   Record *Rec;
2343   if (isLeaf()) {
2344     DefInit *DI = dyn_cast<DefInit>(getLeafValue());
2345     if (!DI)
2346       return nullptr;
2347     Rec = DI->getDef();
2348   } else
2349     Rec = getOperator();
2350 
2351   if (!Rec->isSubClassOf("ComplexPattern"))
2352     return nullptr;
2353   return &CGP.getComplexPattern(Rec);
2354 }
2355 
2356 unsigned TreePatternNode::getNumMIResults(const CodeGenDAGPatterns &CGP) const {
2357   // A ComplexPattern specifically declares how many results it fills in.
2358   if (const ComplexPattern *CP = getComplexPatternInfo(CGP))
2359     return CP->getNumOperands();
2360 
2361   // If MIOperandInfo is specified, that gives the count.
2362   if (isLeaf()) {
2363     DefInit *DI = dyn_cast<DefInit>(getLeafValue());
2364     if (DI && DI->getDef()->isSubClassOf("Operand")) {
2365       DagInit *MIOps = DI->getDef()->getValueAsDag("MIOperandInfo");
2366       if (MIOps->getNumArgs())
2367         return MIOps->getNumArgs();
2368     }
2369   }
2370 
2371   // Otherwise there is just one result.
2372   return 1;
2373 }
2374 
2375 /// NodeHasProperty - Return true if this node has the specified property.
2376 bool TreePatternNode::NodeHasProperty(SDNP Property,
2377                                       const CodeGenDAGPatterns &CGP) const {
2378   if (isLeaf()) {
2379     if (const ComplexPattern *CP = getComplexPatternInfo(CGP))
2380       return CP->hasProperty(Property);
2381 
2382     return false;
2383   }
2384 
2385   if (Property != SDNPHasChain) {
2386     // The chain proprety is already present on the different intrinsic node
2387     // types (intrinsic_w_chain, intrinsic_void), and is not explicitly listed
2388     // on the intrinsic. Anything else is specific to the individual intrinsic.
2389     if (const CodeGenIntrinsic *Int = getIntrinsicInfo(CGP))
2390       return Int->hasProperty(Property);
2391   }
2392 
2393   if (!getOperator()->isSubClassOf("SDPatternOperator"))
2394     return false;
2395 
2396   return CGP.getSDNodeInfo(getOperator()).hasProperty(Property);
2397 }
2398 
2399 /// TreeHasProperty - Return true if any node in this tree has the specified
2400 /// property.
2401 bool TreePatternNode::TreeHasProperty(SDNP Property,
2402                                       const CodeGenDAGPatterns &CGP) const {
2403   if (NodeHasProperty(Property, CGP))
2404     return true;
2405   for (unsigned i = 0, e = getNumChildren(); i != e; ++i)
2406     if (getChild(i).TreeHasProperty(Property, CGP))
2407       return true;
2408   return false;
2409 }
2410 
2411 /// isCommutativeIntrinsic - Return true if the node corresponds to a
2412 /// commutative intrinsic.
2413 bool TreePatternNode::isCommutativeIntrinsic(
2414     const CodeGenDAGPatterns &CDP) const {
2415   if (const CodeGenIntrinsic *Int = getIntrinsicInfo(CDP))
2416     return Int->isCommutative;
2417   return false;
2418 }
2419 
2420 static bool isOperandClass(const TreePatternNode &N, StringRef Class) {
2421   if (!N.isLeaf())
2422     return N.getOperator()->isSubClassOf(Class);
2423 
2424   DefInit *DI = dyn_cast<DefInit>(N.getLeafValue());
2425   if (DI && DI->getDef()->isSubClassOf(Class))
2426     return true;
2427 
2428   return false;
2429 }
2430 
2431 static void emitTooManyOperandsError(TreePattern &TP, StringRef InstName,
2432                                      unsigned Expected, unsigned Actual) {
2433   TP.error("Instruction '" + InstName + "' was provided " + Twine(Actual) +
2434            " operands but expected only " + Twine(Expected) + "!");
2435 }
2436 
2437 static void emitTooFewOperandsError(TreePattern &TP, StringRef InstName,
2438                                     unsigned Actual) {
2439   TP.error("Instruction '" + InstName + "' expects more than the provided " +
2440            Twine(Actual) + " operands!");
2441 }
2442 
2443 /// ApplyTypeConstraints - Apply all of the type constraints relevant to
2444 /// this node and its children in the tree.  This returns true if it makes a
2445 /// change, false otherwise.  If a type contradiction is found, flag an error.
2446 bool TreePatternNode::ApplyTypeConstraints(TreePattern &TP, bool NotRegisters) {
2447   if (TP.hasError())
2448     return false;
2449 
2450   CodeGenDAGPatterns &CDP = TP.getDAGPatterns();
2451   if (isLeaf()) {
2452     if (DefInit *DI = dyn_cast<DefInit>(getLeafValue())) {
2453       // If it's a regclass or something else known, include the type.
2454       bool MadeChange = false;
2455       for (unsigned i = 0, e = Types.size(); i != e; ++i)
2456         MadeChange |= UpdateNodeType(
2457             i, getImplicitType(DI->getDef(), i, NotRegisters, !hasName(), TP),
2458             TP);
2459       return MadeChange;
2460     }
2461 
2462     if (IntInit *II = dyn_cast<IntInit>(getLeafValue())) {
2463       assert(Types.size() == 1 && "Invalid IntInit");
2464 
2465       // Int inits are always integers. :)
2466       bool MadeChange = TP.getInfer().EnforceInteger(Types[0]);
2467 
2468       if (!TP.getInfer().isConcrete(Types[0], false))
2469         return MadeChange;
2470 
2471       ValueTypeByHwMode VVT = TP.getInfer().getConcrete(Types[0], false);
2472       for (auto &P : VVT) {
2473         MVT::SimpleValueType VT = P.second.SimpleTy;
2474         if (VT == MVT::iPTR || VT == MVT::iPTRAny)
2475           continue;
2476         unsigned Size = MVT(VT).getFixedSizeInBits();
2477         // Make sure that the value is representable for this type.
2478         if (Size >= 32)
2479           continue;
2480         // Check that the value doesn't use more bits than we have. It must
2481         // either be a sign- or zero-extended equivalent of the original.
2482         int64_t SignBitAndAbove = II->getValue() >> (Size - 1);
2483         if (SignBitAndAbove == -1 || SignBitAndAbove == 0 ||
2484             SignBitAndAbove == 1)
2485           continue;
2486 
2487         TP.error("Integer value '" + Twine(II->getValue()) +
2488                  "' is out of range for type '" + getEnumName(VT) + "'!");
2489         break;
2490       }
2491       return MadeChange;
2492     }
2493 
2494     return false;
2495   }
2496 
2497   if (const CodeGenIntrinsic *Int = getIntrinsicInfo(CDP)) {
2498     bool MadeChange = false;
2499 
2500     // Apply the result type to the node.
2501     unsigned NumRetVTs = Int->IS.RetTys.size();
2502     unsigned NumParamVTs = Int->IS.ParamTys.size();
2503 
2504     for (unsigned i = 0, e = NumRetVTs; i != e; ++i)
2505       MadeChange |= UpdateNodeType(
2506           i, getValueType(Int->IS.RetTys[i]->getValueAsDef("VT")), TP);
2507 
2508     if (getNumChildren() != NumParamVTs + 1) {
2509       TP.error("Intrinsic '" + Int->Name + "' expects " + Twine(NumParamVTs) +
2510                " operands, not " + Twine(getNumChildren() - 1) + " operands!");
2511       return false;
2512     }
2513 
2514     // Apply type info to the intrinsic ID.
2515     MadeChange |= getChild(0).UpdateNodeType(0, MVT::iPTR, TP);
2516 
2517     for (unsigned i = 0, e = getNumChildren() - 1; i != e; ++i) {
2518       MadeChange |= getChild(i + 1).ApplyTypeConstraints(TP, NotRegisters);
2519 
2520       MVT::SimpleValueType OpVT =
2521           getValueType(Int->IS.ParamTys[i]->getValueAsDef("VT"));
2522       assert(getChild(i + 1).getNumTypes() == 1 && "Unhandled case");
2523       MadeChange |= getChild(i + 1).UpdateNodeType(0, OpVT, TP);
2524     }
2525     return MadeChange;
2526   }
2527 
2528   if (getOperator()->isSubClassOf("SDNode")) {
2529     const SDNodeInfo &NI = CDP.getSDNodeInfo(getOperator());
2530 
2531     // Check that the number of operands is sane.  Negative operands -> varargs.
2532     if (NI.getNumOperands() >= 0 &&
2533         getNumChildren() != (unsigned)NI.getNumOperands()) {
2534       TP.error(getOperator()->getName() + " node requires exactly " +
2535                Twine(NI.getNumOperands()) + " operands!");
2536       return false;
2537     }
2538 
2539     bool MadeChange = false;
2540     for (unsigned i = 0, e = getNumChildren(); i != e; ++i)
2541       MadeChange |= getChild(i).ApplyTypeConstraints(TP, NotRegisters);
2542     MadeChange |= NI.ApplyTypeConstraints(*this, TP);
2543     return MadeChange;
2544   }
2545 
2546   if (getOperator()->isSubClassOf("Instruction")) {
2547     const DAGInstruction &Inst = CDP.getInstruction(getOperator());
2548     CodeGenInstruction &InstInfo =
2549         CDP.getTargetInfo().getInstruction(getOperator());
2550 
2551     bool MadeChange = false;
2552 
2553     // Apply the result types to the node, these come from the things in the
2554     // (outs) list of the instruction.
2555     unsigned NumResultsToAdd =
2556         std::min(InstInfo.Operands.NumDefs, Inst.getNumResults());
2557     for (unsigned ResNo = 0; ResNo != NumResultsToAdd; ++ResNo)
2558       MadeChange |= UpdateNodeTypeFromInst(ResNo, Inst.getResult(ResNo), TP);
2559 
2560     // If the instruction has implicit defs, we apply the first one as a result.
2561     // FIXME: This sucks, it should apply all implicit defs.
2562     if (!InstInfo.ImplicitDefs.empty()) {
2563       unsigned ResNo = NumResultsToAdd;
2564 
2565       // FIXME: Generalize to multiple possible types and multiple possible
2566       // ImplicitDefs.
2567       MVT::SimpleValueType VT =
2568           InstInfo.HasOneImplicitDefWithKnownVT(CDP.getTargetInfo());
2569 
2570       if (VT != MVT::Other)
2571         MadeChange |= UpdateNodeType(ResNo, VT, TP);
2572     }
2573 
2574     // If this is an INSERT_SUBREG, constrain the source and destination VTs to
2575     // be the same.
2576     if (getOperator()->getName() == "INSERT_SUBREG") {
2577       assert(getChild(0).getNumTypes() == 1 && "FIXME: Unhandled");
2578       MadeChange |= UpdateNodeType(0, getChild(0).getExtType(0), TP);
2579       MadeChange |= getChild(0).UpdateNodeType(0, getExtType(0), TP);
2580     } else if (getOperator()->getName() == "REG_SEQUENCE") {
2581       // We need to do extra, custom typechecking for REG_SEQUENCE since it is
2582       // variadic.
2583 
2584       unsigned NChild = getNumChildren();
2585       if (NChild < 3) {
2586         TP.error("REG_SEQUENCE requires at least 3 operands!");
2587         return false;
2588       }
2589 
2590       if (NChild % 2 == 0) {
2591         TP.error("REG_SEQUENCE requires an odd number of operands!");
2592         return false;
2593       }
2594 
2595       if (!isOperandClass(getChild(0), "RegisterClass")) {
2596         TP.error("REG_SEQUENCE requires a RegisterClass for first operand!");
2597         return false;
2598       }
2599 
2600       for (unsigned I = 1; I < NChild; I += 2) {
2601         TreePatternNode &SubIdxChild = getChild(I + 1);
2602         if (!isOperandClass(SubIdxChild, "SubRegIndex")) {
2603           TP.error("REG_SEQUENCE requires a SubRegIndex for operand " +
2604                    Twine(I + 1) + "!");
2605           return false;
2606         }
2607       }
2608     }
2609 
2610     unsigned NumResults = Inst.getNumResults();
2611     unsigned NumFixedOperands = InstInfo.Operands.size();
2612 
2613     // If one or more operands with a default value appear at the end of the
2614     // formal operand list for an instruction, we allow them to be overridden
2615     // by optional operands provided in the pattern.
2616     //
2617     // But if an operand B without a default appears at any point after an
2618     // operand A with a default, then we don't allow A to be overridden,
2619     // because there would be no way to specify whether the next operand in
2620     // the pattern was intended to override A or skip it.
2621     unsigned NonOverridableOperands = NumFixedOperands;
2622     while (NonOverridableOperands > NumResults &&
2623            CDP.operandHasDefault(
2624                InstInfo.Operands[NonOverridableOperands - 1].Rec))
2625       --NonOverridableOperands;
2626 
2627     unsigned ChildNo = 0;
2628     assert(NumResults <= NumFixedOperands);
2629     for (unsigned i = NumResults, e = NumFixedOperands; i != e; ++i) {
2630       Record *OperandNode = InstInfo.Operands[i].Rec;
2631 
2632       // If the operand has a default value, do we use it? We must use the
2633       // default if we've run out of children of the pattern DAG to consume,
2634       // or if the operand is followed by a non-defaulted one.
2635       if (CDP.operandHasDefault(OperandNode) &&
2636           (i < NonOverridableOperands || ChildNo >= getNumChildren()))
2637         continue;
2638 
2639       // If we have run out of child nodes and there _isn't_ a default
2640       // value we can use for the next operand, give an error.
2641       if (ChildNo >= getNumChildren()) {
2642         emitTooFewOperandsError(TP, getOperator()->getName(), getNumChildren());
2643         return false;
2644       }
2645 
2646       TreePatternNode *Child = &getChild(ChildNo++);
2647       unsigned ChildResNo = 0; // Instructions always use res #0 of their op.
2648 
2649       // If the operand has sub-operands, they may be provided by distinct
2650       // child patterns, so attempt to match each sub-operand separately.
2651       if (OperandNode->isSubClassOf("Operand")) {
2652         DagInit *MIOpInfo = OperandNode->getValueAsDag("MIOperandInfo");
2653         if (unsigned NumArgs = MIOpInfo->getNumArgs()) {
2654           // But don't do that if the whole operand is being provided by
2655           // a single ComplexPattern-related Operand.
2656 
2657           if (Child->getNumMIResults(CDP) < NumArgs) {
2658             // Match first sub-operand against the child we already have.
2659             Record *SubRec = cast<DefInit>(MIOpInfo->getArg(0))->getDef();
2660             MadeChange |= Child->UpdateNodeTypeFromInst(ChildResNo, SubRec, TP);
2661 
2662             // And the remaining sub-operands against subsequent children.
2663             for (unsigned Arg = 1; Arg < NumArgs; ++Arg) {
2664               if (ChildNo >= getNumChildren()) {
2665                 emitTooFewOperandsError(TP, getOperator()->getName(),
2666                                         getNumChildren());
2667                 return false;
2668               }
2669               Child = &getChild(ChildNo++);
2670 
2671               SubRec = cast<DefInit>(MIOpInfo->getArg(Arg))->getDef();
2672               MadeChange |=
2673                   Child->UpdateNodeTypeFromInst(ChildResNo, SubRec, TP);
2674             }
2675             continue;
2676           }
2677         }
2678       }
2679 
2680       // If we didn't match by pieces above, attempt to match the whole
2681       // operand now.
2682       MadeChange |= Child->UpdateNodeTypeFromInst(ChildResNo, OperandNode, TP);
2683     }
2684 
2685     if (!InstInfo.Operands.isVariadic && ChildNo != getNumChildren()) {
2686       emitTooManyOperandsError(TP, getOperator()->getName(), ChildNo,
2687                                getNumChildren());
2688       return false;
2689     }
2690 
2691     for (unsigned i = 0, e = getNumChildren(); i != e; ++i)
2692       MadeChange |= getChild(i).ApplyTypeConstraints(TP, NotRegisters);
2693     return MadeChange;
2694   }
2695 
2696   if (getOperator()->isSubClassOf("ComplexPattern")) {
2697     bool MadeChange = false;
2698 
2699     if (!NotRegisters) {
2700       assert(Types.size() == 1 && "ComplexPatterns only produce one result!");
2701       Record *T = CDP.getComplexPattern(getOperator()).getValueType();
2702       const CodeGenHwModes &CGH = CDP.getTargetInfo().getHwModes();
2703       const ValueTypeByHwMode VVT = getValueTypeByHwMode(T, CGH);
2704       // TODO: AArch64 and AMDGPU use ComplexPattern<untyped, ...> and then
2705       // exclusively use those as non-leaf nodes with explicit type casts, so
2706       // for backwards compatibility we do no inference in that case. This is
2707       // not supported when the ComplexPattern is used as a leaf value,
2708       // however; this inconsistency should be resolved, either by adding this
2709       // case there or by altering the backends to not do this (e.g. using Any
2710       // instead may work).
2711       if (!VVT.isSimple() || VVT.getSimple() != MVT::Untyped)
2712         MadeChange |= UpdateNodeType(0, VVT, TP);
2713     }
2714 
2715     for (unsigned i = 0; i < getNumChildren(); ++i)
2716       MadeChange |= getChild(i).ApplyTypeConstraints(TP, NotRegisters);
2717 
2718     return MadeChange;
2719   }
2720 
2721   assert(getOperator()->isSubClassOf("SDNodeXForm") && "Unknown node type!");
2722 
2723   // Node transforms always take one operand.
2724   if (getNumChildren() != 1) {
2725     TP.error("Node transform '" + getOperator()->getName() +
2726              "' requires one operand!");
2727     return false;
2728   }
2729 
2730   bool MadeChange = getChild(0).ApplyTypeConstraints(TP, NotRegisters);
2731   return MadeChange;
2732 }
2733 
2734 /// OnlyOnRHSOfCommutative - Return true if this value is only allowed on the
2735 /// RHS of a commutative operation, not the on LHS.
2736 static bool OnlyOnRHSOfCommutative(TreePatternNode &N) {
2737   if (!N.isLeaf() && N.getOperator()->getName() == "imm")
2738     return true;
2739   if (N.isLeaf() && isa<IntInit>(N.getLeafValue()))
2740     return true;
2741   if (isImmAllOnesAllZerosMatch(N))
2742     return true;
2743   return false;
2744 }
2745 
2746 /// canPatternMatch - If it is impossible for this pattern to match on this
2747 /// target, fill in Reason and return false.  Otherwise, return true.  This is
2748 /// used as a sanity check for .td files (to prevent people from writing stuff
2749 /// that can never possibly work), and to prevent the pattern permuter from
2750 /// generating stuff that is useless.
2751 bool TreePatternNode::canPatternMatch(std::string &Reason,
2752                                       const CodeGenDAGPatterns &CDP) {
2753   if (isLeaf())
2754     return true;
2755 
2756   for (unsigned i = 0, e = getNumChildren(); i != e; ++i)
2757     if (!getChild(i).canPatternMatch(Reason, CDP))
2758       return false;
2759 
2760   // If this is an intrinsic, handle cases that would make it not match.  For
2761   // example, if an operand is required to be an immediate.
2762   if (getOperator()->isSubClassOf("Intrinsic")) {
2763     // TODO:
2764     return true;
2765   }
2766 
2767   if (getOperator()->isSubClassOf("ComplexPattern"))
2768     return true;
2769 
2770   // If this node is a commutative operator, check that the LHS isn't an
2771   // immediate.
2772   const SDNodeInfo &NodeInfo = CDP.getSDNodeInfo(getOperator());
2773   bool isCommIntrinsic = isCommutativeIntrinsic(CDP);
2774   if (NodeInfo.hasProperty(SDNPCommutative) || isCommIntrinsic) {
2775     // Scan all of the operands of the node and make sure that only the last one
2776     // is a constant node, unless the RHS also is.
2777     if (!OnlyOnRHSOfCommutative(getChild(getNumChildren() - 1))) {
2778       unsigned Skip = isCommIntrinsic ? 1 : 0; // First operand is intrinsic id.
2779       for (unsigned i = Skip, e = getNumChildren() - 1; i != e; ++i)
2780         if (OnlyOnRHSOfCommutative(getChild(i))) {
2781           Reason =
2782               "Immediate value must be on the RHS of commutative operators!";
2783           return false;
2784         }
2785     }
2786   }
2787 
2788   return true;
2789 }
2790 
2791 //===----------------------------------------------------------------------===//
2792 // TreePattern implementation
2793 //
2794 
2795 TreePattern::TreePattern(Record *TheRec, ListInit *RawPat, bool isInput,
2796                          CodeGenDAGPatterns &cdp)
2797     : TheRecord(TheRec), CDP(cdp), isInputPattern(isInput), HasError(false),
2798       Infer(*this) {
2799   for (Init *I : RawPat->getValues())
2800     Trees.push_back(ParseTreePattern(I, ""));
2801 }
2802 
2803 TreePattern::TreePattern(Record *TheRec, DagInit *Pat, bool isInput,
2804                          CodeGenDAGPatterns &cdp)
2805     : TheRecord(TheRec), CDP(cdp), isInputPattern(isInput), HasError(false),
2806       Infer(*this) {
2807   Trees.push_back(ParseTreePattern(Pat, ""));
2808 }
2809 
2810 TreePattern::TreePattern(Record *TheRec, TreePatternNodePtr Pat, bool isInput,
2811                          CodeGenDAGPatterns &cdp)
2812     : TheRecord(TheRec), CDP(cdp), isInputPattern(isInput), HasError(false),
2813       Infer(*this) {
2814   Trees.push_back(Pat);
2815 }
2816 
2817 void TreePattern::error(const Twine &Msg) {
2818   if (HasError)
2819     return;
2820   dump();
2821   PrintError(TheRecord->getLoc(), "In " + TheRecord->getName() + ": " + Msg);
2822   HasError = true;
2823 }
2824 
2825 void TreePattern::ComputeNamedNodes() {
2826   for (TreePatternNodePtr &Tree : Trees)
2827     ComputeNamedNodes(*Tree);
2828 }
2829 
2830 void TreePattern::ComputeNamedNodes(TreePatternNode &N) {
2831   if (!N.getName().empty())
2832     NamedNodes[N.getName()].push_back(&N);
2833 
2834   for (unsigned i = 0, e = N.getNumChildren(); i != e; ++i)
2835     ComputeNamedNodes(N.getChild(i));
2836 }
2837 
2838 TreePatternNodePtr TreePattern::ParseTreePattern(Init *TheInit,
2839                                                  StringRef OpName) {
2840   RecordKeeper &RK = TheInit->getRecordKeeper();
2841   if (DefInit *DI = dyn_cast<DefInit>(TheInit)) {
2842     Record *R = DI->getDef();
2843 
2844     // Direct reference to a leaf DagNode or PatFrag?  Turn it into a
2845     // TreePatternNode of its own.  For example:
2846     ///   (foo GPR, imm) -> (foo GPR, (imm))
2847     if (R->isSubClassOf("SDNode") || R->isSubClassOf("PatFrags"))
2848       return ParseTreePattern(
2849           DagInit::get(DI, nullptr,
2850                        std::vector<std::pair<Init *, StringInit *>>()),
2851           OpName);
2852 
2853     // Input argument?
2854     TreePatternNodePtr Res = makeIntrusiveRefCnt<TreePatternNode>(DI, 1);
2855     if (R->getName() == "node" && !OpName.empty()) {
2856       if (OpName.empty())
2857         error("'node' argument requires a name to match with operand list");
2858       Args.push_back(std::string(OpName));
2859     }
2860 
2861     Res->setName(OpName);
2862     return Res;
2863   }
2864 
2865   // ?:$name or just $name.
2866   if (isa<UnsetInit>(TheInit)) {
2867     if (OpName.empty())
2868       error("'?' argument requires a name to match with operand list");
2869     TreePatternNodePtr Res = makeIntrusiveRefCnt<TreePatternNode>(TheInit, 1);
2870     Args.push_back(std::string(OpName));
2871     Res->setName(OpName);
2872     return Res;
2873   }
2874 
2875   if (isa<IntInit>(TheInit) || isa<BitInit>(TheInit)) {
2876     if (!OpName.empty())
2877       error("Constant int or bit argument should not have a name!");
2878     if (isa<BitInit>(TheInit))
2879       TheInit = TheInit->convertInitializerTo(IntRecTy::get(RK));
2880     return makeIntrusiveRefCnt<TreePatternNode>(TheInit, 1);
2881   }
2882 
2883   if (BitsInit *BI = dyn_cast<BitsInit>(TheInit)) {
2884     // Turn this into an IntInit.
2885     Init *II = BI->convertInitializerTo(IntRecTy::get(RK));
2886     if (!II || !isa<IntInit>(II))
2887       error("Bits value must be constants!");
2888     return II ? ParseTreePattern(II, OpName) : nullptr;
2889   }
2890 
2891   DagInit *Dag = dyn_cast<DagInit>(TheInit);
2892   if (!Dag) {
2893     TheInit->print(errs());
2894     error("Pattern has unexpected init kind!");
2895     return nullptr;
2896   }
2897   DefInit *OpDef = dyn_cast<DefInit>(Dag->getOperator());
2898   if (!OpDef) {
2899     error("Pattern has unexpected operator type!");
2900     return nullptr;
2901   }
2902   Record *Operator = OpDef->getDef();
2903 
2904   if (Operator->isSubClassOf("ValueType")) {
2905     // If the operator is a ValueType, then this must be "type cast" of a leaf
2906     // node.
2907     if (Dag->getNumArgs() != 1)
2908       error("Type cast only takes one operand!");
2909 
2910     TreePatternNodePtr New =
2911         ParseTreePattern(Dag->getArg(0), Dag->getArgNameStr(0));
2912 
2913     // Apply the type cast.
2914     if (New->getNumTypes() != 1)
2915       error("Type cast can only have one type!");
2916     const CodeGenHwModes &CGH = getDAGPatterns().getTargetInfo().getHwModes();
2917     New->UpdateNodeType(0, getValueTypeByHwMode(Operator, CGH), *this);
2918 
2919     if (!OpName.empty())
2920       error("ValueType cast should not have a name!");
2921     return New;
2922   }
2923 
2924   // Verify that this is something that makes sense for an operator.
2925   if (!Operator->isSubClassOf("PatFrags") &&
2926       !Operator->isSubClassOf("SDNode") &&
2927       !Operator->isSubClassOf("Instruction") &&
2928       !Operator->isSubClassOf("SDNodeXForm") &&
2929       !Operator->isSubClassOf("Intrinsic") &&
2930       !Operator->isSubClassOf("ComplexPattern") &&
2931       Operator->getName() != "set" && Operator->getName() != "implicit")
2932     error("Unrecognized node '" + Operator->getName() + "'!");
2933 
2934   //  Check to see if this is something that is illegal in an input pattern.
2935   if (isInputPattern) {
2936     if (Operator->isSubClassOf("Instruction") ||
2937         Operator->isSubClassOf("SDNodeXForm"))
2938       error("Cannot use '" + Operator->getName() + "' in an input pattern!");
2939   } else {
2940     if (Operator->isSubClassOf("Intrinsic"))
2941       error("Cannot use '" + Operator->getName() + "' in an output pattern!");
2942 
2943     if (Operator->isSubClassOf("SDNode") && Operator->getName() != "imm" &&
2944         Operator->getName() != "timm" && Operator->getName() != "fpimm" &&
2945         Operator->getName() != "tglobaltlsaddr" &&
2946         Operator->getName() != "tconstpool" &&
2947         Operator->getName() != "tjumptable" &&
2948         Operator->getName() != "tframeindex" &&
2949         Operator->getName() != "texternalsym" &&
2950         Operator->getName() != "tblockaddress" &&
2951         Operator->getName() != "tglobaladdr" && Operator->getName() != "bb" &&
2952         Operator->getName() != "vt" && Operator->getName() != "mcsym")
2953       error("Cannot use '" + Operator->getName() + "' in an output pattern!");
2954   }
2955 
2956   std::vector<TreePatternNodePtr> Children;
2957 
2958   // Parse all the operands.
2959   for (unsigned i = 0, e = Dag->getNumArgs(); i != e; ++i)
2960     Children.push_back(ParseTreePattern(Dag->getArg(i), Dag->getArgNameStr(i)));
2961 
2962   // Get the actual number of results before Operator is converted to an
2963   // intrinsic node (which is hard-coded to have either zero or one result).
2964   unsigned NumResults = GetNumNodeResults(Operator, CDP);
2965 
2966   // If the operator is an intrinsic, then this is just syntactic sugar for
2967   // (intrinsic_* <number>, ..children..).  Pick the right intrinsic node, and
2968   // convert the intrinsic name to a number.
2969   if (Operator->isSubClassOf("Intrinsic")) {
2970     const CodeGenIntrinsic &Int = getDAGPatterns().getIntrinsic(Operator);
2971     unsigned IID = getDAGPatterns().getIntrinsicID(Operator) + 1;
2972 
2973     // If this intrinsic returns void, it must have side-effects and thus a
2974     // chain.
2975     if (Int.IS.RetTys.empty())
2976       Operator = getDAGPatterns().get_intrinsic_void_sdnode();
2977     else if (!Int.ME.doesNotAccessMemory() || Int.hasSideEffects)
2978       // Has side-effects, requires chain.
2979       Operator = getDAGPatterns().get_intrinsic_w_chain_sdnode();
2980     else // Otherwise, no chain.
2981       Operator = getDAGPatterns().get_intrinsic_wo_chain_sdnode();
2982 
2983     Children.insert(Children.begin(), makeIntrusiveRefCnt<TreePatternNode>(
2984                                           IntInit::get(RK, IID), 1));
2985   }
2986 
2987   if (Operator->isSubClassOf("ComplexPattern")) {
2988     for (unsigned i = 0; i < Children.size(); ++i) {
2989       TreePatternNodePtr Child = Children[i];
2990 
2991       if (Child->getName().empty())
2992         error("All arguments to a ComplexPattern must be named");
2993 
2994       // Check that the ComplexPattern uses are consistent: "(MY_PAT $a, $b)"
2995       // and "(MY_PAT $b, $a)" should not be allowed in the same pattern;
2996       // neither should "(MY_PAT_1 $a, $b)" and "(MY_PAT_2 $a, $b)".
2997       auto OperandId = std::pair(Operator, i);
2998       auto PrevOp = ComplexPatternOperands.find(Child->getName());
2999       if (PrevOp != ComplexPatternOperands.end()) {
3000         if (PrevOp->getValue() != OperandId)
3001           error("All ComplexPattern operands must appear consistently: "
3002                 "in the same order in just one ComplexPattern instance.");
3003       } else
3004         ComplexPatternOperands[Child->getName()] = OperandId;
3005     }
3006   }
3007 
3008   TreePatternNodePtr Result = makeIntrusiveRefCnt<TreePatternNode>(
3009       Operator, std::move(Children), NumResults);
3010   Result->setName(OpName);
3011 
3012   if (Dag->getName()) {
3013     assert(Result->getName().empty());
3014     Result->setName(Dag->getNameStr());
3015   }
3016   return Result;
3017 }
3018 
3019 /// SimplifyTree - See if we can simplify this tree to eliminate something that
3020 /// will never match in favor of something obvious that will.  This is here
3021 /// strictly as a convenience to target authors because it allows them to write
3022 /// more type generic things and have useless type casts fold away.
3023 ///
3024 /// This returns true if any change is made.
3025 static bool SimplifyTree(TreePatternNodePtr &N) {
3026   if (N->isLeaf())
3027     return false;
3028 
3029   // If we have a bitconvert with a resolved type and if the source and
3030   // destination types are the same, then the bitconvert is useless, remove it.
3031   //
3032   // We make an exception if the types are completely empty. This can come up
3033   // when the pattern being simplified is in the Fragments list of a PatFrags,
3034   // so that the operand is just an untyped "node". In that situation we leave
3035   // bitconverts unsimplified, and simplify them later once the fragment is
3036   // expanded into its true context.
3037   if (N->getOperator()->getName() == "bitconvert" &&
3038       N->getExtType(0).isValueTypeByHwMode(false) &&
3039       !N->getExtType(0).empty() &&
3040       N->getExtType(0) == N->getChild(0).getExtType(0) &&
3041       N->getName().empty()) {
3042     N = N->getChildShared(0);
3043     SimplifyTree(N);
3044     return true;
3045   }
3046 
3047   // Walk all children.
3048   bool MadeChange = false;
3049   for (unsigned i = 0, e = N->getNumChildren(); i != e; ++i)
3050     MadeChange |= SimplifyTree(N->getChildSharedPtr(i));
3051 
3052   return MadeChange;
3053 }
3054 
3055 /// InferAllTypes - Infer/propagate as many types throughout the expression
3056 /// patterns as possible.  Return true if all types are inferred, false
3057 /// otherwise.  Flags an error if a type contradiction is found.
3058 bool TreePattern::InferAllTypes(
3059     const StringMap<SmallVector<TreePatternNode *, 1>> *InNamedTypes) {
3060   if (NamedNodes.empty())
3061     ComputeNamedNodes();
3062 
3063   bool MadeChange = true;
3064   while (MadeChange) {
3065     MadeChange = false;
3066     for (TreePatternNodePtr &Tree : Trees) {
3067       MadeChange |= Tree->ApplyTypeConstraints(*this, false);
3068       MadeChange |= SimplifyTree(Tree);
3069     }
3070 
3071     // If there are constraints on our named nodes, apply them.
3072     for (auto &Entry : NamedNodes) {
3073       SmallVectorImpl<TreePatternNode *> &Nodes = Entry.second;
3074 
3075       // If we have input named node types, propagate their types to the named
3076       // values here.
3077       if (InNamedTypes) {
3078         if (!InNamedTypes->count(Entry.getKey())) {
3079           error("Node '" + std::string(Entry.getKey()) +
3080                 "' in output pattern but not input pattern");
3081           return true;
3082         }
3083 
3084         const SmallVectorImpl<TreePatternNode *> &InNodes =
3085             InNamedTypes->find(Entry.getKey())->second;
3086 
3087         // The input types should be fully resolved by now.
3088         for (TreePatternNode *Node : Nodes) {
3089           // If this node is a register class, and it is the root of the pattern
3090           // then we're mapping something onto an input register.  We allow
3091           // changing the type of the input register in this case.  This allows
3092           // us to match things like:
3093           //  def : Pat<(v1i64 (bitconvert(v2i32 DPR:$src))), (v1i64 DPR:$src)>;
3094           if (Node == Trees[0].get() && Node->isLeaf()) {
3095             DefInit *DI = dyn_cast<DefInit>(Node->getLeafValue());
3096             if (DI && (DI->getDef()->isSubClassOf("RegisterClass") ||
3097                        DI->getDef()->isSubClassOf("RegisterOperand")))
3098               continue;
3099           }
3100 
3101           assert(Node->getNumTypes() == 1 && InNodes[0]->getNumTypes() == 1 &&
3102                  "FIXME: cannot name multiple result nodes yet");
3103           MadeChange |=
3104               Node->UpdateNodeType(0, InNodes[0]->getExtType(0), *this);
3105         }
3106       }
3107 
3108       // If there are multiple nodes with the same name, they must all have the
3109       // same type.
3110       if (Entry.second.size() > 1) {
3111         for (unsigned i = 0, e = Nodes.size() - 1; i != e; ++i) {
3112           TreePatternNode *N1 = Nodes[i], *N2 = Nodes[i + 1];
3113           assert(N1->getNumTypes() == 1 && N2->getNumTypes() == 1 &&
3114                  "FIXME: cannot name multiple result nodes yet");
3115 
3116           MadeChange |= N1->UpdateNodeType(0, N2->getExtType(0), *this);
3117           MadeChange |= N2->UpdateNodeType(0, N1->getExtType(0), *this);
3118         }
3119       }
3120     }
3121   }
3122 
3123   bool HasUnresolvedTypes = false;
3124   for (const TreePatternNodePtr &Tree : Trees)
3125     HasUnresolvedTypes |= Tree->ContainsUnresolvedType(*this);
3126   return !HasUnresolvedTypes;
3127 }
3128 
3129 void TreePattern::print(raw_ostream &OS) const {
3130   OS << getRecord()->getName();
3131   if (!Args.empty()) {
3132     OS << "(";
3133     ListSeparator LS;
3134     for (const std::string &Arg : Args)
3135       OS << LS << Arg;
3136     OS << ")";
3137   }
3138   OS << ": ";
3139 
3140   if (Trees.size() > 1)
3141     OS << "[\n";
3142   for (const TreePatternNodePtr &Tree : Trees) {
3143     OS << "\t";
3144     Tree->print(OS);
3145     OS << "\n";
3146   }
3147 
3148   if (Trees.size() > 1)
3149     OS << "]\n";
3150 }
3151 
3152 void TreePattern::dump() const { print(errs()); }
3153 
3154 //===----------------------------------------------------------------------===//
3155 // CodeGenDAGPatterns implementation
3156 //
3157 
3158 CodeGenDAGPatterns::CodeGenDAGPatterns(RecordKeeper &R,
3159                                        PatternRewriterFn PatternRewriter)
3160     : Records(R), Target(R), Intrinsics(R),
3161       LegalVTS(Target.getLegalValueTypes()), PatternRewriter(PatternRewriter) {
3162   ParseNodeInfo();
3163   ParseNodeTransforms();
3164   ParseComplexPatterns();
3165   ParsePatternFragments();
3166   ParseDefaultOperands();
3167   ParseInstructions();
3168   ParsePatternFragments(/*OutFrags*/ true);
3169   ParsePatterns();
3170 
3171   // Generate variants.  For example, commutative patterns can match
3172   // multiple ways.  Add them to PatternsToMatch as well.
3173   GenerateVariants();
3174 
3175   // Break patterns with parameterized types into a series of patterns,
3176   // where each one has a fixed type and is predicated on the conditions
3177   // of the associated HW mode.
3178   ExpandHwModeBasedTypes();
3179 
3180   // Infer instruction flags.  For example, we can detect loads,
3181   // stores, and side effects in many cases by examining an
3182   // instruction's pattern.
3183   InferInstructionFlags();
3184 
3185   // Verify that instruction flags match the patterns.
3186   VerifyInstructionFlags();
3187 }
3188 
3189 Record *CodeGenDAGPatterns::getSDNodeNamed(StringRef Name) const {
3190   Record *N = Records.getDef(Name);
3191   if (!N || !N->isSubClassOf("SDNode"))
3192     PrintFatalError("Error getting SDNode '" + Name + "'!");
3193 
3194   return N;
3195 }
3196 
3197 // Parse all of the SDNode definitions for the target, populating SDNodes.
3198 void CodeGenDAGPatterns::ParseNodeInfo() {
3199   std::vector<Record *> Nodes = Records.getAllDerivedDefinitions("SDNode");
3200   const CodeGenHwModes &CGH = getTargetInfo().getHwModes();
3201 
3202   while (!Nodes.empty()) {
3203     Record *R = Nodes.back();
3204     SDNodes.insert(std::pair(R, SDNodeInfo(R, CGH)));
3205     Nodes.pop_back();
3206   }
3207 
3208   // Get the builtin intrinsic nodes.
3209   intrinsic_void_sdnode = getSDNodeNamed("intrinsic_void");
3210   intrinsic_w_chain_sdnode = getSDNodeNamed("intrinsic_w_chain");
3211   intrinsic_wo_chain_sdnode = getSDNodeNamed("intrinsic_wo_chain");
3212 }
3213 
3214 /// ParseNodeTransforms - Parse all SDNodeXForm instances into the SDNodeXForms
3215 /// map, and emit them to the file as functions.
3216 void CodeGenDAGPatterns::ParseNodeTransforms() {
3217   std::vector<Record *> Xforms =
3218       Records.getAllDerivedDefinitions("SDNodeXForm");
3219   while (!Xforms.empty()) {
3220     Record *XFormNode = Xforms.back();
3221     Record *SDNode = XFormNode->getValueAsDef("Opcode");
3222     StringRef Code = XFormNode->getValueAsString("XFormFunction");
3223     SDNodeXForms.insert(
3224         std::pair(XFormNode, NodeXForm(SDNode, std::string(Code))));
3225 
3226     Xforms.pop_back();
3227   }
3228 }
3229 
3230 void CodeGenDAGPatterns::ParseComplexPatterns() {
3231   std::vector<Record *> AMs =
3232       Records.getAllDerivedDefinitions("ComplexPattern");
3233   while (!AMs.empty()) {
3234     ComplexPatterns.insert(std::pair(AMs.back(), AMs.back()));
3235     AMs.pop_back();
3236   }
3237 }
3238 
3239 /// ParsePatternFragments - Parse all of the PatFrag definitions in the .td
3240 /// file, building up the PatternFragments map.  After we've collected them all,
3241 /// inline fragments together as necessary, so that there are no references left
3242 /// inside a pattern fragment to a pattern fragment.
3243 ///
3244 void CodeGenDAGPatterns::ParsePatternFragments(bool OutFrags) {
3245   std::vector<Record *> Fragments =
3246       Records.getAllDerivedDefinitions("PatFrags");
3247 
3248   // First step, parse all of the fragments.
3249   for (Record *Frag : Fragments) {
3250     if (OutFrags != Frag->isSubClassOf("OutPatFrag"))
3251       continue;
3252 
3253     ListInit *LI = Frag->getValueAsListInit("Fragments");
3254     TreePattern *P = (PatternFragments[Frag] = std::make_unique<TreePattern>(
3255                           Frag, LI, !Frag->isSubClassOf("OutPatFrag"), *this))
3256                          .get();
3257 
3258     // Validate the argument list, converting it to set, to discard duplicates.
3259     std::vector<std::string> &Args = P->getArgList();
3260     // Copy the args so we can take StringRefs to them.
3261     auto ArgsCopy = Args;
3262     SmallDenseSet<StringRef, 4> OperandsSet;
3263     OperandsSet.insert(ArgsCopy.begin(), ArgsCopy.end());
3264 
3265     if (OperandsSet.count(""))
3266       P->error("Cannot have unnamed 'node' values in pattern fragment!");
3267 
3268     // Parse the operands list.
3269     DagInit *OpsList = Frag->getValueAsDag("Operands");
3270     DefInit *OpsOp = dyn_cast<DefInit>(OpsList->getOperator());
3271     // Special cases: ops == outs == ins. Different names are used to
3272     // improve readability.
3273     if (!OpsOp || (OpsOp->getDef()->getName() != "ops" &&
3274                    OpsOp->getDef()->getName() != "outs" &&
3275                    OpsOp->getDef()->getName() != "ins"))
3276       P->error("Operands list should start with '(ops ... '!");
3277 
3278     // Copy over the arguments.
3279     Args.clear();
3280     for (unsigned j = 0, e = OpsList->getNumArgs(); j != e; ++j) {
3281       if (!isa<DefInit>(OpsList->getArg(j)) ||
3282           cast<DefInit>(OpsList->getArg(j))->getDef()->getName() != "node")
3283         P->error("Operands list should all be 'node' values.");
3284       if (!OpsList->getArgName(j))
3285         P->error("Operands list should have names for each operand!");
3286       StringRef ArgNameStr = OpsList->getArgNameStr(j);
3287       if (!OperandsSet.count(ArgNameStr))
3288         P->error("'" + ArgNameStr +
3289                  "' does not occur in pattern or was multiply specified!");
3290       OperandsSet.erase(ArgNameStr);
3291       Args.push_back(std::string(ArgNameStr));
3292     }
3293 
3294     if (!OperandsSet.empty())
3295       P->error("Operands list does not contain an entry for operand '" +
3296                *OperandsSet.begin() + "'!");
3297 
3298     // If there is a node transformation corresponding to this, keep track of
3299     // it.
3300     Record *Transform = Frag->getValueAsDef("OperandTransform");
3301     if (!getSDNodeTransform(Transform).second.empty()) // not noop xform?
3302       for (const auto &T : P->getTrees())
3303         T->setTransformFn(Transform);
3304   }
3305 
3306   // Now that we've parsed all of the tree fragments, do a closure on them so
3307   // that there are not references to PatFrags left inside of them.
3308   for (Record *Frag : Fragments) {
3309     if (OutFrags != Frag->isSubClassOf("OutPatFrag"))
3310       continue;
3311 
3312     TreePattern &ThePat = *PatternFragments[Frag];
3313     ThePat.InlinePatternFragments();
3314 
3315     // Infer as many types as possible.  Don't worry about it if we don't infer
3316     // all of them, some may depend on the inputs of the pattern.  Also, don't
3317     // validate type sets; validation may cause spurious failures e.g. if a
3318     // fragment needs floating-point types but the current target does not have
3319     // any (this is only an error if that fragment is ever used!).
3320     {
3321       TypeInfer::SuppressValidation SV(ThePat.getInfer());
3322       ThePat.InferAllTypes();
3323       ThePat.resetError();
3324     }
3325 
3326     // If debugging, print out the pattern fragment result.
3327     LLVM_DEBUG(ThePat.dump());
3328   }
3329 }
3330 
3331 void CodeGenDAGPatterns::ParseDefaultOperands() {
3332   std::vector<Record *> DefaultOps;
3333   DefaultOps = Records.getAllDerivedDefinitions("OperandWithDefaultOps");
3334 
3335   // Find some SDNode.
3336   assert(!SDNodes.empty() && "No SDNodes parsed?");
3337   Init *SomeSDNode = DefInit::get(SDNodes.begin()->first);
3338 
3339   for (unsigned i = 0, e = DefaultOps.size(); i != e; ++i) {
3340     DagInit *DefaultInfo = DefaultOps[i]->getValueAsDag("DefaultOps");
3341 
3342     // Clone the DefaultInfo dag node, changing the operator from 'ops' to
3343     // SomeSDnode so that we can parse this.
3344     std::vector<std::pair<Init *, StringInit *>> Ops;
3345     for (unsigned op = 0, e = DefaultInfo->getNumArgs(); op != e; ++op)
3346       Ops.push_back(
3347           std::pair(DefaultInfo->getArg(op), DefaultInfo->getArgName(op)));
3348     DagInit *DI = DagInit::get(SomeSDNode, nullptr, Ops);
3349 
3350     // Create a TreePattern to parse this.
3351     TreePattern P(DefaultOps[i], DI, false, *this);
3352     assert(P.getNumTrees() == 1 && "This ctor can only produce one tree!");
3353 
3354     // Copy the operands over into a DAGDefaultOperand.
3355     DAGDefaultOperand DefaultOpInfo;
3356 
3357     const TreePatternNodePtr &T = P.getTree(0);
3358     for (unsigned op = 0, e = T->getNumChildren(); op != e; ++op) {
3359       TreePatternNodePtr TPN = T->getChildShared(op);
3360       while (TPN->ApplyTypeConstraints(P, false))
3361         /* Resolve all types */;
3362 
3363       if (TPN->ContainsUnresolvedType(P)) {
3364         PrintFatalError("Value #" + Twine(i) + " of OperandWithDefaultOps '" +
3365                         DefaultOps[i]->getName() +
3366                         "' doesn't have a concrete type!");
3367       }
3368       DefaultOpInfo.DefaultOps.push_back(std::move(TPN));
3369     }
3370 
3371     // Insert it into the DefaultOperands map so we can find it later.
3372     DefaultOperands[DefaultOps[i]] = DefaultOpInfo;
3373   }
3374 }
3375 
3376 /// HandleUse - Given "Pat" a leaf in the pattern, check to see if it is an
3377 /// instruction input.  Return true if this is a real use.
3378 static bool HandleUse(TreePattern &I, TreePatternNodePtr Pat,
3379                       std::map<std::string, TreePatternNodePtr> &InstInputs) {
3380   // No name -> not interesting.
3381   if (Pat->getName().empty()) {
3382     if (Pat->isLeaf()) {
3383       DefInit *DI = dyn_cast<DefInit>(Pat->getLeafValue());
3384       if (DI && (DI->getDef()->isSubClassOf("RegisterClass") ||
3385                  DI->getDef()->isSubClassOf("RegisterOperand")))
3386         I.error("Input " + DI->getDef()->getName() + " must be named!");
3387     }
3388     return false;
3389   }
3390 
3391   Record *Rec;
3392   if (Pat->isLeaf()) {
3393     DefInit *DI = dyn_cast<DefInit>(Pat->getLeafValue());
3394     if (!DI)
3395       I.error("Input $" + Pat->getName() + " must be an identifier!");
3396     Rec = DI->getDef();
3397   } else {
3398     Rec = Pat->getOperator();
3399   }
3400 
3401   // SRCVALUE nodes are ignored.
3402   if (Rec->getName() == "srcvalue")
3403     return false;
3404 
3405   TreePatternNodePtr &Slot = InstInputs[Pat->getName()];
3406   if (!Slot) {
3407     Slot = Pat;
3408     return true;
3409   }
3410   Record *SlotRec;
3411   if (Slot->isLeaf()) {
3412     SlotRec = cast<DefInit>(Slot->getLeafValue())->getDef();
3413   } else {
3414     assert(Slot->getNumChildren() == 0 && "can't be a use with children!");
3415     SlotRec = Slot->getOperator();
3416   }
3417 
3418   // Ensure that the inputs agree if we've already seen this input.
3419   if (Rec != SlotRec)
3420     I.error("All $" + Pat->getName() + " inputs must agree with each other");
3421   // Ensure that the types can agree as well.
3422   Slot->UpdateNodeType(0, Pat->getExtType(0), I);
3423   Pat->UpdateNodeType(0, Slot->getExtType(0), I);
3424   if (Slot->getExtTypes() != Pat->getExtTypes())
3425     I.error("All $" + Pat->getName() + " inputs must agree with each other");
3426   return true;
3427 }
3428 
3429 /// FindPatternInputsAndOutputs - Scan the specified TreePatternNode (which is
3430 /// part of "I", the instruction), computing the set of inputs and outputs of
3431 /// the pattern.  Report errors if we see anything naughty.
3432 void CodeGenDAGPatterns::FindPatternInputsAndOutputs(
3433     TreePattern &I, TreePatternNodePtr Pat,
3434     std::map<std::string, TreePatternNodePtr> &InstInputs,
3435     MapVector<std::string, TreePatternNodePtr, std::map<std::string, unsigned>>
3436         &InstResults,
3437     std::vector<Record *> &InstImpResults) {
3438 
3439   // The instruction pattern still has unresolved fragments.  For *named*
3440   // nodes we must resolve those here.  This may not result in multiple
3441   // alternatives.
3442   if (!Pat->getName().empty()) {
3443     TreePattern SrcPattern(I.getRecord(), Pat, true, *this);
3444     SrcPattern.InlinePatternFragments();
3445     SrcPattern.InferAllTypes();
3446     Pat = SrcPattern.getOnlyTree();
3447   }
3448 
3449   if (Pat->isLeaf()) {
3450     bool isUse = HandleUse(I, Pat, InstInputs);
3451     if (!isUse && Pat->getTransformFn())
3452       I.error("Cannot specify a transform function for a non-input value!");
3453     return;
3454   }
3455 
3456   if (Pat->getOperator()->getName() == "implicit") {
3457     for (unsigned i = 0, e = Pat->getNumChildren(); i != e; ++i) {
3458       TreePatternNode &Dest = Pat->getChild(i);
3459       if (!Dest.isLeaf())
3460         I.error("implicitly defined value should be a register!");
3461 
3462       DefInit *Val = dyn_cast<DefInit>(Dest.getLeafValue());
3463       if (!Val || !Val->getDef()->isSubClassOf("Register"))
3464         I.error("implicitly defined value should be a register!");
3465       if (Val)
3466         InstImpResults.push_back(Val->getDef());
3467     }
3468     return;
3469   }
3470 
3471   if (Pat->getOperator()->getName() != "set") {
3472     // If this is not a set, verify that the children nodes are not void typed,
3473     // and recurse.
3474     for (unsigned i = 0, e = Pat->getNumChildren(); i != e; ++i) {
3475       if (Pat->getChild(i).getNumTypes() == 0)
3476         I.error("Cannot have void nodes inside of patterns!");
3477       FindPatternInputsAndOutputs(I, Pat->getChildShared(i), InstInputs,
3478                                   InstResults, InstImpResults);
3479     }
3480 
3481     // If this is a non-leaf node with no children, treat it basically as if
3482     // it were a leaf.  This handles nodes like (imm).
3483     bool isUse = HandleUse(I, Pat, InstInputs);
3484 
3485     if (!isUse && Pat->getTransformFn())
3486       I.error("Cannot specify a transform function for a non-input value!");
3487     return;
3488   }
3489 
3490   // Otherwise, this is a set, validate and collect instruction results.
3491   if (Pat->getNumChildren() == 0)
3492     I.error("set requires operands!");
3493 
3494   if (Pat->getTransformFn())
3495     I.error("Cannot specify a transform function on a set node!");
3496 
3497   // Check the set destinations.
3498   unsigned NumDests = Pat->getNumChildren() - 1;
3499   for (unsigned i = 0; i != NumDests; ++i) {
3500     TreePatternNodePtr Dest = Pat->getChildShared(i);
3501     // For set destinations we also must resolve fragments here.
3502     TreePattern DestPattern(I.getRecord(), Dest, false, *this);
3503     DestPattern.InlinePatternFragments();
3504     DestPattern.InferAllTypes();
3505     Dest = DestPattern.getOnlyTree();
3506 
3507     if (!Dest->isLeaf())
3508       I.error("set destination should be a register!");
3509 
3510     DefInit *Val = dyn_cast<DefInit>(Dest->getLeafValue());
3511     if (!Val) {
3512       I.error("set destination should be a register!");
3513       continue;
3514     }
3515 
3516     if (Val->getDef()->isSubClassOf("RegisterClass") ||
3517         Val->getDef()->isSubClassOf("ValueType") ||
3518         Val->getDef()->isSubClassOf("RegisterOperand") ||
3519         Val->getDef()->isSubClassOf("PointerLikeRegClass")) {
3520       if (Dest->getName().empty())
3521         I.error("set destination must have a name!");
3522       if (InstResults.count(Dest->getName()))
3523         I.error("cannot set '" + Dest->getName() + "' multiple times");
3524       InstResults[Dest->getName()] = Dest;
3525     } else if (Val->getDef()->isSubClassOf("Register")) {
3526       InstImpResults.push_back(Val->getDef());
3527     } else {
3528       I.error("set destination should be a register!");
3529     }
3530   }
3531 
3532   // Verify and collect info from the computation.
3533   FindPatternInputsAndOutputs(I, Pat->getChildShared(NumDests), InstInputs,
3534                               InstResults, InstImpResults);
3535 }
3536 
3537 //===----------------------------------------------------------------------===//
3538 // Instruction Analysis
3539 //===----------------------------------------------------------------------===//
3540 
3541 class InstAnalyzer {
3542   const CodeGenDAGPatterns &CDP;
3543 
3544 public:
3545   bool hasSideEffects;
3546   bool mayStore;
3547   bool mayLoad;
3548   bool isBitcast;
3549   bool isVariadic;
3550   bool hasChain;
3551 
3552   InstAnalyzer(const CodeGenDAGPatterns &cdp)
3553       : CDP(cdp), hasSideEffects(false), mayStore(false), mayLoad(false),
3554         isBitcast(false), isVariadic(false), hasChain(false) {}
3555 
3556   void Analyze(const PatternToMatch &Pat) {
3557     const TreePatternNode &N = Pat.getSrcPattern();
3558     AnalyzeNode(N);
3559     // These properties are detected only on the root node.
3560     isBitcast = IsNodeBitcast(N);
3561   }
3562 
3563 private:
3564   bool IsNodeBitcast(const TreePatternNode &N) const {
3565     if (hasSideEffects || mayLoad || mayStore || isVariadic)
3566       return false;
3567 
3568     if (N.isLeaf())
3569       return false;
3570     if (N.getNumChildren() != 1 || !N.getChild(0).isLeaf())
3571       return false;
3572 
3573     if (N.getOperator()->isSubClassOf("ComplexPattern"))
3574       return false;
3575 
3576     const SDNodeInfo &OpInfo = CDP.getSDNodeInfo(N.getOperator());
3577     if (OpInfo.getNumResults() != 1 || OpInfo.getNumOperands() != 1)
3578       return false;
3579     return OpInfo.getEnumName() == "ISD::BITCAST";
3580   }
3581 
3582 public:
3583   void AnalyzeNode(const TreePatternNode &N) {
3584     if (N.isLeaf()) {
3585       if (DefInit *DI = dyn_cast<DefInit>(N.getLeafValue())) {
3586         Record *LeafRec = DI->getDef();
3587         // Handle ComplexPattern leaves.
3588         if (LeafRec->isSubClassOf("ComplexPattern")) {
3589           const ComplexPattern &CP = CDP.getComplexPattern(LeafRec);
3590           if (CP.hasProperty(SDNPMayStore))
3591             mayStore = true;
3592           if (CP.hasProperty(SDNPMayLoad))
3593             mayLoad = true;
3594           if (CP.hasProperty(SDNPSideEffect))
3595             hasSideEffects = true;
3596         }
3597       }
3598       return;
3599     }
3600 
3601     // Analyze children.
3602     for (unsigned i = 0, e = N.getNumChildren(); i != e; ++i)
3603       AnalyzeNode(N.getChild(i));
3604 
3605     // Notice properties of the node.
3606     if (N.NodeHasProperty(SDNPMayStore, CDP))
3607       mayStore = true;
3608     if (N.NodeHasProperty(SDNPMayLoad, CDP))
3609       mayLoad = true;
3610     if (N.NodeHasProperty(SDNPSideEffect, CDP))
3611       hasSideEffects = true;
3612     if (N.NodeHasProperty(SDNPVariadic, CDP))
3613       isVariadic = true;
3614     if (N.NodeHasProperty(SDNPHasChain, CDP))
3615       hasChain = true;
3616 
3617     if (const CodeGenIntrinsic *IntInfo = N.getIntrinsicInfo(CDP)) {
3618       ModRefInfo MR = IntInfo->ME.getModRef();
3619       // If this is an intrinsic, analyze it.
3620       if (isRefSet(MR))
3621         mayLoad = true; // These may load memory.
3622 
3623       if (isModSet(MR))
3624         mayStore = true; // Intrinsics that can write to memory are 'mayStore'.
3625 
3626       // Consider intrinsics that don't specify any restrictions on memory
3627       // effects as having a side-effect.
3628       if (IntInfo->ME == MemoryEffects::unknown() || IntInfo->hasSideEffects)
3629         hasSideEffects = true;
3630     }
3631   }
3632 };
3633 
3634 static bool InferFromPattern(CodeGenInstruction &InstInfo,
3635                              const InstAnalyzer &PatInfo, Record *PatDef) {
3636   bool Error = false;
3637 
3638   // Remember where InstInfo got its flags.
3639   if (InstInfo.hasUndefFlags())
3640     InstInfo.InferredFrom = PatDef;
3641 
3642   // Check explicitly set flags for consistency.
3643   if (InstInfo.hasSideEffects != PatInfo.hasSideEffects &&
3644       !InstInfo.hasSideEffects_Unset) {
3645     // Allow explicitly setting hasSideEffects = 1 on instructions, even when
3646     // the pattern has no side effects. That could be useful for div/rem
3647     // instructions that may trap.
3648     if (!InstInfo.hasSideEffects) {
3649       Error = true;
3650       PrintError(PatDef->getLoc(), "Pattern doesn't match hasSideEffects = " +
3651                                        Twine(InstInfo.hasSideEffects));
3652     }
3653   }
3654 
3655   if (InstInfo.mayStore != PatInfo.mayStore && !InstInfo.mayStore_Unset) {
3656     Error = true;
3657     PrintError(PatDef->getLoc(),
3658                "Pattern doesn't match mayStore = " + Twine(InstInfo.mayStore));
3659   }
3660 
3661   if (InstInfo.mayLoad != PatInfo.mayLoad && !InstInfo.mayLoad_Unset) {
3662     // Allow explicitly setting mayLoad = 1, even when the pattern has no loads.
3663     // Some targets translate immediates to loads.
3664     if (!InstInfo.mayLoad) {
3665       Error = true;
3666       PrintError(PatDef->getLoc(),
3667                  "Pattern doesn't match mayLoad = " + Twine(InstInfo.mayLoad));
3668     }
3669   }
3670 
3671   // Transfer inferred flags.
3672   InstInfo.hasSideEffects |= PatInfo.hasSideEffects;
3673   InstInfo.mayStore |= PatInfo.mayStore;
3674   InstInfo.mayLoad |= PatInfo.mayLoad;
3675 
3676   // These flags are silently added without any verification.
3677   // FIXME: To match historical behavior of TableGen, for now add those flags
3678   // only when we're inferring from the primary instruction pattern.
3679   if (PatDef->isSubClassOf("Instruction")) {
3680     InstInfo.isBitcast |= PatInfo.isBitcast;
3681     InstInfo.hasChain |= PatInfo.hasChain;
3682     InstInfo.hasChain_Inferred = true;
3683   }
3684 
3685   // Don't infer isVariadic. This flag means something different on SDNodes and
3686   // instructions. For example, a CALL SDNode is variadic because it has the
3687   // call arguments as operands, but a CALL instruction is not variadic - it
3688   // has argument registers as implicit, not explicit uses.
3689 
3690   return Error;
3691 }
3692 
3693 /// hasNullFragReference - Return true if the DAG has any reference to the
3694 /// null_frag operator.
3695 static bool hasNullFragReference(DagInit *DI) {
3696   DefInit *OpDef = dyn_cast<DefInit>(DI->getOperator());
3697   if (!OpDef)
3698     return false;
3699   Record *Operator = OpDef->getDef();
3700 
3701   // If this is the null fragment, return true.
3702   if (Operator->getName() == "null_frag")
3703     return true;
3704   // If any of the arguments reference the null fragment, return true.
3705   for (unsigned i = 0, e = DI->getNumArgs(); i != e; ++i) {
3706     if (auto Arg = dyn_cast<DefInit>(DI->getArg(i)))
3707       if (Arg->getDef()->getName() == "null_frag")
3708         return true;
3709     DagInit *Arg = dyn_cast<DagInit>(DI->getArg(i));
3710     if (Arg && hasNullFragReference(Arg))
3711       return true;
3712   }
3713 
3714   return false;
3715 }
3716 
3717 /// hasNullFragReference - Return true if any DAG in the list references
3718 /// the null_frag operator.
3719 static bool hasNullFragReference(ListInit *LI) {
3720   for (Init *I : LI->getValues()) {
3721     DagInit *DI = dyn_cast<DagInit>(I);
3722     assert(DI && "non-dag in an instruction Pattern list?!");
3723     if (hasNullFragReference(DI))
3724       return true;
3725   }
3726   return false;
3727 }
3728 
3729 /// Get all the instructions in a tree.
3730 static void getInstructionsInTree(TreePatternNode &Tree,
3731                                   SmallVectorImpl<Record *> &Instrs) {
3732   if (Tree.isLeaf())
3733     return;
3734   if (Tree.getOperator()->isSubClassOf("Instruction"))
3735     Instrs.push_back(Tree.getOperator());
3736   for (unsigned i = 0, e = Tree.getNumChildren(); i != e; ++i)
3737     getInstructionsInTree(Tree.getChild(i), Instrs);
3738 }
3739 
3740 /// Check the class of a pattern leaf node against the instruction operand it
3741 /// represents.
3742 static bool checkOperandClass(CGIOperandList::OperandInfo &OI, Record *Leaf) {
3743   if (OI.Rec == Leaf)
3744     return true;
3745 
3746   // Allow direct value types to be used in instruction set patterns.
3747   // The type will be checked later.
3748   if (Leaf->isSubClassOf("ValueType"))
3749     return true;
3750 
3751   // Patterns can also be ComplexPattern instances.
3752   if (Leaf->isSubClassOf("ComplexPattern"))
3753     return true;
3754 
3755   return false;
3756 }
3757 
3758 void CodeGenDAGPatterns::parseInstructionPattern(CodeGenInstruction &CGI,
3759                                                  ListInit *Pat,
3760                                                  DAGInstMap &DAGInsts) {
3761 
3762   assert(!DAGInsts.count(CGI.TheDef) && "Instruction already parsed!");
3763 
3764   // Parse the instruction.
3765   TreePattern I(CGI.TheDef, Pat, true, *this);
3766 
3767   // InstInputs - Keep track of all of the inputs of the instruction, along
3768   // with the record they are declared as.
3769   std::map<std::string, TreePatternNodePtr> InstInputs;
3770 
3771   // InstResults - Keep track of all the virtual registers that are 'set'
3772   // in the instruction, including what reg class they are.
3773   MapVector<std::string, TreePatternNodePtr, std::map<std::string, unsigned>>
3774       InstResults;
3775 
3776   std::vector<Record *> InstImpResults;
3777 
3778   // Verify that the top-level forms in the instruction are of void type, and
3779   // fill in the InstResults map.
3780   SmallString<32> TypesString;
3781   for (unsigned j = 0, e = I.getNumTrees(); j != e; ++j) {
3782     TypesString.clear();
3783     TreePatternNodePtr Pat = I.getTree(j);
3784     if (Pat->getNumTypes() != 0) {
3785       raw_svector_ostream OS(TypesString);
3786       ListSeparator LS;
3787       for (unsigned k = 0, ke = Pat->getNumTypes(); k != ke; ++k) {
3788         OS << LS;
3789         Pat->getExtType(k).writeToStream(OS);
3790       }
3791       I.error("Top-level forms in instruction pattern should have"
3792               " void types, has types " +
3793               OS.str());
3794     }
3795 
3796     // Find inputs and outputs, and verify the structure of the uses/defs.
3797     FindPatternInputsAndOutputs(I, Pat, InstInputs, InstResults,
3798                                 InstImpResults);
3799   }
3800 
3801   // Now that we have inputs and outputs of the pattern, inspect the operands
3802   // list for the instruction.  This determines the order that operands are
3803   // added to the machine instruction the node corresponds to.
3804   unsigned NumResults = InstResults.size();
3805 
3806   // Parse the operands list from the (ops) list, validating it.
3807   assert(I.getArgList().empty() && "Args list should still be empty here!");
3808 
3809   // Check that all of the results occur first in the list.
3810   std::vector<Record *> Results;
3811   std::vector<unsigned> ResultIndices;
3812   SmallVector<TreePatternNodePtr, 2> ResNodes;
3813   for (unsigned i = 0; i != NumResults; ++i) {
3814     if (i == CGI.Operands.size()) {
3815       const std::string &OpName =
3816           llvm::find_if(
3817               InstResults,
3818               [](const std::pair<std::string, TreePatternNodePtr> &P) {
3819                 return P.second;
3820               })
3821               ->first;
3822 
3823       I.error("'" + OpName + "' set but does not appear in operand list!");
3824     }
3825 
3826     const std::string &OpName = CGI.Operands[i].Name;
3827 
3828     // Check that it exists in InstResults.
3829     auto InstResultIter = InstResults.find(OpName);
3830     if (InstResultIter == InstResults.end() || !InstResultIter->second)
3831       I.error("Operand $" + OpName + " does not exist in operand list!");
3832 
3833     TreePatternNodePtr RNode = InstResultIter->second;
3834     Record *R = cast<DefInit>(RNode->getLeafValue())->getDef();
3835     ResNodes.push_back(std::move(RNode));
3836     if (!R)
3837       I.error("Operand $" + OpName +
3838               " should be a set destination: all "
3839               "outputs must occur before inputs in operand list!");
3840 
3841     if (!checkOperandClass(CGI.Operands[i], R))
3842       I.error("Operand $" + OpName + " class mismatch!");
3843 
3844     // Remember the return type.
3845     Results.push_back(CGI.Operands[i].Rec);
3846 
3847     // Remember the result index.
3848     ResultIndices.push_back(std::distance(InstResults.begin(), InstResultIter));
3849 
3850     // Okay, this one checks out.
3851     InstResultIter->second = nullptr;
3852   }
3853 
3854   // Loop over the inputs next.
3855   std::vector<TreePatternNodePtr> ResultNodeOperands;
3856   std::vector<Record *> Operands;
3857   for (unsigned i = NumResults, e = CGI.Operands.size(); i != e; ++i) {
3858     CGIOperandList::OperandInfo &Op = CGI.Operands[i];
3859     const std::string &OpName = Op.Name;
3860     if (OpName.empty()) {
3861       I.error("Operand #" + Twine(i) + " in operands list has no name!");
3862       continue;
3863     }
3864 
3865     if (!InstInputs.count(OpName)) {
3866       // If this is an operand with a DefaultOps set filled in, we can ignore
3867       // this.  When we codegen it, we will do so as always executed.
3868       if (Op.Rec->isSubClassOf("OperandWithDefaultOps")) {
3869         // Does it have a non-empty DefaultOps field?  If so, ignore this
3870         // operand.
3871         if (!getDefaultOperand(Op.Rec).DefaultOps.empty())
3872           continue;
3873       }
3874       I.error("Operand $" + OpName +
3875               " does not appear in the instruction pattern");
3876       continue;
3877     }
3878     TreePatternNodePtr InVal = InstInputs[OpName];
3879     InstInputs.erase(OpName); // It occurred, remove from map.
3880 
3881     if (InVal->isLeaf() && isa<DefInit>(InVal->getLeafValue())) {
3882       Record *InRec = cast<DefInit>(InVal->getLeafValue())->getDef();
3883       if (!checkOperandClass(Op, InRec)) {
3884         I.error("Operand $" + OpName +
3885                 "'s register class disagrees"
3886                 " between the operand and pattern");
3887         continue;
3888       }
3889     }
3890     Operands.push_back(Op.Rec);
3891 
3892     // Construct the result for the dest-pattern operand list.
3893     TreePatternNodePtr OpNode = InVal->clone();
3894 
3895     // No predicate is useful on the result.
3896     OpNode->clearPredicateCalls();
3897 
3898     // Promote the xform function to be an explicit node if set.
3899     if (Record *Xform = OpNode->getTransformFn()) {
3900       OpNode->setTransformFn(nullptr);
3901       std::vector<TreePatternNodePtr> Children;
3902       Children.push_back(OpNode);
3903       OpNode = makeIntrusiveRefCnt<TreePatternNode>(Xform, std::move(Children),
3904                                                     OpNode->getNumTypes());
3905     }
3906 
3907     ResultNodeOperands.push_back(std::move(OpNode));
3908   }
3909 
3910   if (!InstInputs.empty())
3911     I.error("Input operand $" + InstInputs.begin()->first +
3912             " occurs in pattern but not in operands list!");
3913 
3914   TreePatternNodePtr ResultPattern = makeIntrusiveRefCnt<TreePatternNode>(
3915       I.getRecord(), std::move(ResultNodeOperands),
3916       GetNumNodeResults(I.getRecord(), *this));
3917   // Copy fully inferred output node types to instruction result pattern.
3918   for (unsigned i = 0; i != NumResults; ++i) {
3919     assert(ResNodes[i]->getNumTypes() == 1 && "FIXME: Unhandled");
3920     ResultPattern->setType(i, ResNodes[i]->getExtType(0));
3921     ResultPattern->setResultIndex(i, ResultIndices[i]);
3922   }
3923 
3924   // FIXME: Assume only the first tree is the pattern. The others are clobber
3925   // nodes.
3926   TreePatternNodePtr Pattern = I.getTree(0);
3927   TreePatternNodePtr SrcPattern;
3928   if (Pattern->getOperator()->getName() == "set") {
3929     SrcPattern = Pattern->getChild(Pattern->getNumChildren() - 1).clone();
3930   } else {
3931     // Not a set (store or something?)
3932     SrcPattern = Pattern;
3933   }
3934 
3935   // Create and insert the instruction.
3936   // FIXME: InstImpResults should not be part of DAGInstruction.
3937   Record *R = I.getRecord();
3938   DAGInsts.try_emplace(R, std::move(Results), std::move(Operands),
3939                        std::move(InstImpResults), SrcPattern, ResultPattern);
3940 
3941   LLVM_DEBUG(I.dump());
3942 }
3943 
3944 /// ParseInstructions - Parse all of the instructions, inlining and resolving
3945 /// any fragments involved.  This populates the Instructions list with fully
3946 /// resolved instructions.
3947 void CodeGenDAGPatterns::ParseInstructions() {
3948   std::vector<Record *> Instrs =
3949       Records.getAllDerivedDefinitions("Instruction");
3950 
3951   for (Record *Instr : Instrs) {
3952     ListInit *LI = nullptr;
3953 
3954     if (isa<ListInit>(Instr->getValueInit("Pattern")))
3955       LI = Instr->getValueAsListInit("Pattern");
3956 
3957     // If there is no pattern, only collect minimal information about the
3958     // instruction for its operand list.  We have to assume that there is one
3959     // result, as we have no detailed info. A pattern which references the
3960     // null_frag operator is as-if no pattern were specified. Normally this
3961     // is from a multiclass expansion w/ a SDPatternOperator passed in as
3962     // null_frag.
3963     if (!LI || LI->empty() || hasNullFragReference(LI)) {
3964       std::vector<Record *> Results;
3965       std::vector<Record *> Operands;
3966 
3967       CodeGenInstruction &InstInfo = Target.getInstruction(Instr);
3968 
3969       if (InstInfo.Operands.size() != 0) {
3970         for (unsigned j = 0, e = InstInfo.Operands.NumDefs; j < e; ++j)
3971           Results.push_back(InstInfo.Operands[j].Rec);
3972 
3973         // The rest are inputs.
3974         for (unsigned j = InstInfo.Operands.NumDefs,
3975                       e = InstInfo.Operands.size();
3976              j < e; ++j)
3977           Operands.push_back(InstInfo.Operands[j].Rec);
3978       }
3979 
3980       // Create and insert the instruction.
3981       Instructions.try_emplace(Instr, std::move(Results), std::move(Operands),
3982                                std::vector<Record *>());
3983       continue; // no pattern.
3984     }
3985 
3986     CodeGenInstruction &CGI = Target.getInstruction(Instr);
3987     parseInstructionPattern(CGI, LI, Instructions);
3988   }
3989 
3990   // If we can, convert the instructions to be patterns that are matched!
3991   for (auto &Entry : Instructions) {
3992     Record *Instr = Entry.first;
3993     DAGInstruction &TheInst = Entry.second;
3994     TreePatternNodePtr SrcPattern = TheInst.getSrcPattern();
3995     TreePatternNodePtr ResultPattern = TheInst.getResultPattern();
3996 
3997     if (SrcPattern && ResultPattern) {
3998       TreePattern Pattern(Instr, SrcPattern, true, *this);
3999       TreePattern Result(Instr, ResultPattern, false, *this);
4000       ParseOnePattern(Instr, Pattern, Result, TheInst.getImpResults());
4001     }
4002   }
4003 }
4004 
4005 typedef std::pair<TreePatternNode *, unsigned> NameRecord;
4006 
4007 static void FindNames(TreePatternNode &P,
4008                       std::map<std::string, NameRecord> &Names,
4009                       TreePattern *PatternTop) {
4010   if (!P.getName().empty()) {
4011     NameRecord &Rec = Names[P.getName()];
4012     // If this is the first instance of the name, remember the node.
4013     if (Rec.second++ == 0)
4014       Rec.first = &P;
4015     else if (Rec.first->getExtTypes() != P.getExtTypes())
4016       PatternTop->error("repetition of value: $" + P.getName() +
4017                         " where different uses have different types!");
4018   }
4019 
4020   if (!P.isLeaf()) {
4021     for (unsigned i = 0, e = P.getNumChildren(); i != e; ++i)
4022       FindNames(P.getChild(i), Names, PatternTop);
4023   }
4024 }
4025 
4026 void CodeGenDAGPatterns::AddPatternToMatch(TreePattern *Pattern,
4027                                            PatternToMatch &&PTM) {
4028   // Do some sanity checking on the pattern we're about to match.
4029   std::string Reason;
4030   if (!PTM.getSrcPattern().canPatternMatch(Reason, *this)) {
4031     PrintWarning(Pattern->getRecord()->getLoc(),
4032                  Twine("Pattern can never match: ") + Reason);
4033     return;
4034   }
4035 
4036   // If the source pattern's root is a complex pattern, that complex pattern
4037   // must specify the nodes it can potentially match.
4038   if (const ComplexPattern *CP =
4039           PTM.getSrcPattern().getComplexPatternInfo(*this))
4040     if (CP->getRootNodes().empty())
4041       Pattern->error("ComplexPattern at root must specify list of opcodes it"
4042                      " could match");
4043 
4044   // Find all of the named values in the input and output, ensure they have the
4045   // same type.
4046   std::map<std::string, NameRecord> SrcNames, DstNames;
4047   FindNames(PTM.getSrcPattern(), SrcNames, Pattern);
4048   FindNames(PTM.getDstPattern(), DstNames, Pattern);
4049 
4050   // Scan all of the named values in the destination pattern, rejecting them if
4051   // they don't exist in the input pattern.
4052   for (const auto &Entry : DstNames) {
4053     if (SrcNames[Entry.first].first == nullptr)
4054       Pattern->error("Pattern has input without matching name in output: $" +
4055                      Entry.first);
4056   }
4057 
4058   // Scan all of the named values in the source pattern, rejecting them if the
4059   // name isn't used in the dest, and isn't used to tie two values together.
4060   for (const auto &Entry : SrcNames)
4061     if (DstNames[Entry.first].first == nullptr &&
4062         SrcNames[Entry.first].second == 1)
4063       Pattern->error("Pattern has dead named input: $" + Entry.first);
4064 
4065   PatternsToMatch.push_back(std::move(PTM));
4066 }
4067 
4068 void CodeGenDAGPatterns::InferInstructionFlags() {
4069   ArrayRef<const CodeGenInstruction *> Instructions =
4070       Target.getInstructionsByEnumValue();
4071 
4072   unsigned Errors = 0;
4073 
4074   // Try to infer flags from all patterns in PatternToMatch.  These include
4075   // both the primary instruction patterns (which always come first) and
4076   // patterns defined outside the instruction.
4077   for (const PatternToMatch &PTM : ptms()) {
4078     // We can only infer from single-instruction patterns, otherwise we won't
4079     // know which instruction should get the flags.
4080     SmallVector<Record *, 8> PatInstrs;
4081     getInstructionsInTree(PTM.getDstPattern(), PatInstrs);
4082     if (PatInstrs.size() != 1)
4083       continue;
4084 
4085     // Get the single instruction.
4086     CodeGenInstruction &InstInfo = Target.getInstruction(PatInstrs.front());
4087 
4088     // Only infer properties from the first pattern. We'll verify the others.
4089     if (InstInfo.InferredFrom)
4090       continue;
4091 
4092     InstAnalyzer PatInfo(*this);
4093     PatInfo.Analyze(PTM);
4094     Errors += InferFromPattern(InstInfo, PatInfo, PTM.getSrcRecord());
4095   }
4096 
4097   if (Errors)
4098     PrintFatalError("pattern conflicts");
4099 
4100   // If requested by the target, guess any undefined properties.
4101   if (Target.guessInstructionProperties()) {
4102     for (unsigned i = 0, e = Instructions.size(); i != e; ++i) {
4103       CodeGenInstruction *InstInfo =
4104           const_cast<CodeGenInstruction *>(Instructions[i]);
4105       if (InstInfo->InferredFrom)
4106         continue;
4107       // The mayLoad and mayStore flags default to false.
4108       // Conservatively assume hasSideEffects if it wasn't explicit.
4109       if (InstInfo->hasSideEffects_Unset)
4110         InstInfo->hasSideEffects = true;
4111     }
4112     return;
4113   }
4114 
4115   // Complain about any flags that are still undefined.
4116   for (unsigned i = 0, e = Instructions.size(); i != e; ++i) {
4117     CodeGenInstruction *InstInfo =
4118         const_cast<CodeGenInstruction *>(Instructions[i]);
4119     if (InstInfo->InferredFrom)
4120       continue;
4121     if (InstInfo->hasSideEffects_Unset)
4122       PrintError(InstInfo->TheDef->getLoc(),
4123                  "Can't infer hasSideEffects from patterns");
4124     if (InstInfo->mayStore_Unset)
4125       PrintError(InstInfo->TheDef->getLoc(),
4126                  "Can't infer mayStore from patterns");
4127     if (InstInfo->mayLoad_Unset)
4128       PrintError(InstInfo->TheDef->getLoc(),
4129                  "Can't infer mayLoad from patterns");
4130   }
4131 }
4132 
4133 /// Verify instruction flags against pattern node properties.
4134 void CodeGenDAGPatterns::VerifyInstructionFlags() {
4135   unsigned Errors = 0;
4136   for (const PatternToMatch &PTM : ptms()) {
4137     SmallVector<Record *, 8> Instrs;
4138     getInstructionsInTree(PTM.getDstPattern(), Instrs);
4139     if (Instrs.empty())
4140       continue;
4141 
4142     // Count the number of instructions with each flag set.
4143     unsigned NumSideEffects = 0;
4144     unsigned NumStores = 0;
4145     unsigned NumLoads = 0;
4146     for (const Record *Instr : Instrs) {
4147       const CodeGenInstruction &InstInfo = Target.getInstruction(Instr);
4148       NumSideEffects += InstInfo.hasSideEffects;
4149       NumStores += InstInfo.mayStore;
4150       NumLoads += InstInfo.mayLoad;
4151     }
4152 
4153     // Analyze the source pattern.
4154     InstAnalyzer PatInfo(*this);
4155     PatInfo.Analyze(PTM);
4156 
4157     // Collect error messages.
4158     SmallVector<std::string, 4> Msgs;
4159 
4160     // Check for missing flags in the output.
4161     // Permit extra flags for now at least.
4162     if (PatInfo.hasSideEffects && !NumSideEffects)
4163       Msgs.push_back("pattern has side effects, but hasSideEffects isn't set");
4164 
4165     // Don't verify store flags on instructions with side effects. At least for
4166     // intrinsics, side effects implies mayStore.
4167     if (!PatInfo.hasSideEffects && PatInfo.mayStore && !NumStores)
4168       Msgs.push_back("pattern may store, but mayStore isn't set");
4169 
4170     // Similarly, mayStore implies mayLoad on intrinsics.
4171     if (!PatInfo.mayStore && PatInfo.mayLoad && !NumLoads)
4172       Msgs.push_back("pattern may load, but mayLoad isn't set");
4173 
4174     // Print error messages.
4175     if (Msgs.empty())
4176       continue;
4177     ++Errors;
4178 
4179     for (const std::string &Msg : Msgs)
4180       PrintError(
4181           PTM.getSrcRecord()->getLoc(),
4182           Twine(Msg) + " on the " +
4183               (Instrs.size() == 1 ? "instruction" : "output instructions"));
4184     // Provide the location of the relevant instruction definitions.
4185     for (const Record *Instr : Instrs) {
4186       if (Instr != PTM.getSrcRecord())
4187         PrintError(Instr->getLoc(), "defined here");
4188       const CodeGenInstruction &InstInfo = Target.getInstruction(Instr);
4189       if (InstInfo.InferredFrom && InstInfo.InferredFrom != InstInfo.TheDef &&
4190           InstInfo.InferredFrom != PTM.getSrcRecord())
4191         PrintError(InstInfo.InferredFrom->getLoc(), "inferred from pattern");
4192     }
4193   }
4194   if (Errors)
4195     PrintFatalError("Errors in DAG patterns");
4196 }
4197 
4198 /// Given a pattern result with an unresolved type, see if we can find one
4199 /// instruction with an unresolved result type.  Force this result type to an
4200 /// arbitrary element if it's possible types to converge results.
4201 static bool ForceArbitraryInstResultType(TreePatternNode &N, TreePattern &TP) {
4202   if (N.isLeaf())
4203     return false;
4204 
4205   // Analyze children.
4206   for (unsigned i = 0, e = N.getNumChildren(); i != e; ++i)
4207     if (ForceArbitraryInstResultType(N.getChild(i), TP))
4208       return true;
4209 
4210   if (!N.getOperator()->isSubClassOf("Instruction"))
4211     return false;
4212 
4213   // If this type is already concrete or completely unknown we can't do
4214   // anything.
4215   TypeInfer &TI = TP.getInfer();
4216   for (unsigned i = 0, e = N.getNumTypes(); i != e; ++i) {
4217     if (N.getExtType(i).empty() || TI.isConcrete(N.getExtType(i), false))
4218       continue;
4219 
4220     // Otherwise, force its type to an arbitrary choice.
4221     if (TI.forceArbitrary(N.getExtType(i)))
4222       return true;
4223   }
4224 
4225   return false;
4226 }
4227 
4228 // Promote xform function to be an explicit node wherever set.
4229 static TreePatternNodePtr PromoteXForms(TreePatternNodePtr N) {
4230   if (Record *Xform = N->getTransformFn()) {
4231     N->setTransformFn(nullptr);
4232     std::vector<TreePatternNodePtr> Children;
4233     Children.push_back(PromoteXForms(N));
4234     return makeIntrusiveRefCnt<TreePatternNode>(Xform, std::move(Children),
4235                                                 N->getNumTypes());
4236   }
4237 
4238   if (!N->isLeaf())
4239     for (unsigned i = 0, e = N->getNumChildren(); i != e; ++i) {
4240       TreePatternNodePtr Child = N->getChildShared(i);
4241       N->setChild(i, PromoteXForms(Child));
4242     }
4243   return N;
4244 }
4245 
4246 void CodeGenDAGPatterns::ParseOnePattern(
4247     Record *TheDef, TreePattern &Pattern, TreePattern &Result,
4248     const std::vector<Record *> &InstImpResults, bool ShouldIgnore) {
4249 
4250   // Inline pattern fragments and expand multiple alternatives.
4251   Pattern.InlinePatternFragments();
4252   Result.InlinePatternFragments();
4253 
4254   if (Result.getNumTrees() != 1)
4255     Result.error("Cannot use multi-alternative fragments in result pattern!");
4256 
4257   // Infer types.
4258   bool IterateInference;
4259   bool InferredAllPatternTypes, InferredAllResultTypes;
4260   do {
4261     // Infer as many types as possible.  If we cannot infer all of them, we
4262     // can never do anything with this pattern: report it to the user.
4263     InferredAllPatternTypes =
4264         Pattern.InferAllTypes(&Pattern.getNamedNodesMap());
4265 
4266     // Infer as many types as possible.  If we cannot infer all of them, we
4267     // can never do anything with this pattern: report it to the user.
4268     InferredAllResultTypes = Result.InferAllTypes(&Pattern.getNamedNodesMap());
4269 
4270     IterateInference = false;
4271 
4272     // Apply the type of the result to the source pattern.  This helps us
4273     // resolve cases where the input type is known to be a pointer type (which
4274     // is considered resolved), but the result knows it needs to be 32- or
4275     // 64-bits.  Infer the other way for good measure.
4276     for (const auto &T : Pattern.getTrees())
4277       for (unsigned i = 0, e = std::min(Result.getOnlyTree()->getNumTypes(),
4278                                         T->getNumTypes());
4279            i != e; ++i) {
4280         IterateInference |=
4281             T->UpdateNodeType(i, Result.getOnlyTree()->getExtType(i), Result);
4282         IterateInference |=
4283             Result.getOnlyTree()->UpdateNodeType(i, T->getExtType(i), Result);
4284       }
4285 
4286     // If our iteration has converged and the input pattern's types are fully
4287     // resolved but the result pattern is not fully resolved, we may have a
4288     // situation where we have two instructions in the result pattern and
4289     // the instructions require a common register class, but don't care about
4290     // what actual MVT is used.  This is actually a bug in our modelling:
4291     // output patterns should have register classes, not MVTs.
4292     //
4293     // In any case, to handle this, we just go through and disambiguate some
4294     // arbitrary types to the result pattern's nodes.
4295     if (!IterateInference && InferredAllPatternTypes && !InferredAllResultTypes)
4296       IterateInference =
4297           ForceArbitraryInstResultType(*Result.getTree(0), Result);
4298   } while (IterateInference);
4299 
4300   // Verify that we inferred enough types that we can do something with the
4301   // pattern and result.  If these fire the user has to add type casts.
4302   if (!InferredAllPatternTypes)
4303     Pattern.error("Could not infer all types in pattern!");
4304   if (!InferredAllResultTypes) {
4305     Pattern.dump();
4306     Result.error("Could not infer all types in pattern result!");
4307   }
4308 
4309   // Promote xform function to be an explicit node wherever set.
4310   TreePatternNodePtr DstShared = PromoteXForms(Result.getOnlyTree());
4311 
4312   TreePattern Temp(Result.getRecord(), DstShared, false, *this);
4313   Temp.InferAllTypes();
4314 
4315   ListInit *Preds = TheDef->getValueAsListInit("Predicates");
4316   int Complexity = TheDef->getValueAsInt("AddedComplexity");
4317 
4318   if (PatternRewriter)
4319     PatternRewriter(&Pattern);
4320 
4321   // A pattern may end up with an "impossible" type, i.e. a situation
4322   // where all types have been eliminated for some node in this pattern.
4323   // This could occur for intrinsics that only make sense for a specific
4324   // value type, and use a specific register class. If, for some mode,
4325   // that register class does not accept that type, the type inference
4326   // will lead to a contradiction, which is not an error however, but
4327   // a sign that this pattern will simply never match.
4328   if (Temp.getOnlyTree()->hasPossibleType()) {
4329     for (const auto &T : Pattern.getTrees()) {
4330       if (T->hasPossibleType())
4331         AddPatternToMatch(&Pattern,
4332                           PatternToMatch(TheDef, Preds, T, Temp.getOnlyTree(),
4333                                          InstImpResults, Complexity,
4334                                          TheDef->getID(), ShouldIgnore));
4335     }
4336   } else {
4337     // Show a message about a dropped pattern with some info to make it
4338     // easier to identify it in the .td files.
4339     LLVM_DEBUG({
4340       dbgs() << "Dropping: ";
4341       Pattern.dump();
4342       Temp.getOnlyTree()->dump();
4343       dbgs() << "\n";
4344     });
4345   }
4346 }
4347 
4348 void CodeGenDAGPatterns::ParsePatterns() {
4349   std::vector<Record *> Patterns = Records.getAllDerivedDefinitions("Pattern");
4350 
4351   for (Record *CurPattern : Patterns) {
4352     DagInit *Tree = CurPattern->getValueAsDag("PatternToMatch");
4353 
4354     // If the pattern references the null_frag, there's nothing to do.
4355     if (hasNullFragReference(Tree))
4356       continue;
4357 
4358     TreePattern Pattern(CurPattern, Tree, true, *this);
4359 
4360     ListInit *LI = CurPattern->getValueAsListInit("ResultInstrs");
4361     if (LI->empty())
4362       continue; // no pattern.
4363 
4364     // Parse the instruction.
4365     TreePattern Result(CurPattern, LI, false, *this);
4366 
4367     if (Result.getNumTrees() != 1)
4368       Result.error("Cannot handle instructions producing instructions "
4369                    "with temporaries yet!");
4370 
4371     // Validate that the input pattern is correct.
4372     std::map<std::string, TreePatternNodePtr> InstInputs;
4373     MapVector<std::string, TreePatternNodePtr, std::map<std::string, unsigned>>
4374         InstResults;
4375     std::vector<Record *> InstImpResults;
4376     for (unsigned j = 0, ee = Pattern.getNumTrees(); j != ee; ++j)
4377       FindPatternInputsAndOutputs(Pattern, Pattern.getTree(j), InstInputs,
4378                                   InstResults, InstImpResults);
4379 
4380     ParseOnePattern(CurPattern, Pattern, Result, InstImpResults,
4381                     CurPattern->getValueAsBit("GISelShouldIgnore"));
4382   }
4383 }
4384 
4385 static void collectModes(std::set<unsigned> &Modes, const TreePatternNode &N) {
4386   for (const TypeSetByHwMode &VTS : N.getExtTypes())
4387     for (const auto &I : VTS)
4388       Modes.insert(I.first);
4389 
4390   for (unsigned i = 0, e = N.getNumChildren(); i != e; ++i)
4391     collectModes(Modes, N.getChild(i));
4392 }
4393 
4394 void CodeGenDAGPatterns::ExpandHwModeBasedTypes() {
4395   const CodeGenHwModes &CGH = getTargetInfo().getHwModes();
4396   if (CGH.getNumModeIds() == 1)
4397     return;
4398 
4399   std::vector<PatternToMatch> Copy;
4400   PatternsToMatch.swap(Copy);
4401 
4402   auto AppendPattern = [this](PatternToMatch &P, unsigned Mode,
4403                               StringRef Check) {
4404     TreePatternNodePtr NewSrc = P.getSrcPattern().clone();
4405     TreePatternNodePtr NewDst = P.getDstPattern().clone();
4406     if (!NewSrc->setDefaultMode(Mode) || !NewDst->setDefaultMode(Mode)) {
4407       return;
4408     }
4409 
4410     PatternsToMatch.emplace_back(
4411         P.getSrcRecord(), P.getPredicates(), std::move(NewSrc),
4412         std::move(NewDst), P.getDstRegs(), P.getAddedComplexity(),
4413         Record::getNewUID(Records), P.getGISelShouldIgnore(), Check);
4414   };
4415 
4416   for (PatternToMatch &P : Copy) {
4417     const TreePatternNode *SrcP = nullptr, *DstP = nullptr;
4418     if (P.getSrcPattern().hasProperTypeByHwMode())
4419       SrcP = &P.getSrcPattern();
4420     if (P.getDstPattern().hasProperTypeByHwMode())
4421       DstP = &P.getDstPattern();
4422     if (!SrcP && !DstP) {
4423       PatternsToMatch.push_back(P);
4424       continue;
4425     }
4426 
4427     std::set<unsigned> Modes;
4428     if (SrcP)
4429       collectModes(Modes, *SrcP);
4430     if (DstP)
4431       collectModes(Modes, *DstP);
4432 
4433     // The predicate for the default mode needs to be constructed for each
4434     // pattern separately.
4435     // Since not all modes must be present in each pattern, if a mode m is
4436     // absent, then there is no point in constructing a check for m. If such
4437     // a check was created, it would be equivalent to checking the default
4438     // mode, except not all modes' predicates would be a part of the checking
4439     // code. The subsequently generated check for the default mode would then
4440     // have the exact same patterns, but a different predicate code. To avoid
4441     // duplicated patterns with different predicate checks, construct the
4442     // default check as a negation of all predicates that are actually present
4443     // in the source/destination patterns.
4444     SmallString<128> DefaultCheck;
4445 
4446     for (unsigned M : Modes) {
4447       if (M == DefaultMode)
4448         continue;
4449 
4450       // Fill the map entry for this mode.
4451       const HwMode &HM = CGH.getMode(M);
4452       AppendPattern(P, M, HM.Predicates);
4453 
4454       // Add negations of the HM's predicates to the default predicate.
4455       if (!DefaultCheck.empty())
4456         DefaultCheck += " && ";
4457       DefaultCheck += "!(";
4458       DefaultCheck += HM.Predicates;
4459       DefaultCheck += ")";
4460     }
4461 
4462     bool HasDefault = Modes.count(DefaultMode);
4463     if (HasDefault)
4464       AppendPattern(P, DefaultMode, DefaultCheck);
4465   }
4466 }
4467 
4468 /// Dependent variable map for CodeGenDAGPattern variant generation
4469 typedef StringMap<int> DepVarMap;
4470 
4471 static void FindDepVarsOf(TreePatternNode &N, DepVarMap &DepMap) {
4472   if (N.isLeaf()) {
4473     if (N.hasName() && isa<DefInit>(N.getLeafValue()))
4474       DepMap[N.getName()]++;
4475   } else {
4476     for (size_t i = 0, e = N.getNumChildren(); i != e; ++i)
4477       FindDepVarsOf(N.getChild(i), DepMap);
4478   }
4479 }
4480 
4481 /// Find dependent variables within child patterns
4482 static void FindDepVars(TreePatternNode &N, MultipleUseVarSet &DepVars) {
4483   DepVarMap depcounts;
4484   FindDepVarsOf(N, depcounts);
4485   for (const auto &Pair : depcounts) {
4486     if (Pair.getValue() > 1)
4487       DepVars.insert(Pair.getKey());
4488   }
4489 }
4490 
4491 #ifndef NDEBUG
4492 /// Dump the dependent variable set:
4493 static void DumpDepVars(MultipleUseVarSet &DepVars) {
4494   if (DepVars.empty()) {
4495     LLVM_DEBUG(errs() << "<empty set>");
4496   } else {
4497     LLVM_DEBUG(errs() << "[ ");
4498     for (const auto &DepVar : DepVars) {
4499       LLVM_DEBUG(errs() << DepVar.getKey() << " ");
4500     }
4501     LLVM_DEBUG(errs() << "]");
4502   }
4503 }
4504 #endif
4505 
4506 /// CombineChildVariants - Given a bunch of permutations of each child of the
4507 /// 'operator' node, put them together in all possible ways.
4508 static void CombineChildVariants(
4509     TreePatternNodePtr Orig,
4510     const std::vector<std::vector<TreePatternNodePtr>> &ChildVariants,
4511     std::vector<TreePatternNodePtr> &OutVariants, CodeGenDAGPatterns &CDP,
4512     const MultipleUseVarSet &DepVars) {
4513   // Make sure that each operand has at least one variant to choose from.
4514   for (const auto &Variants : ChildVariants)
4515     if (Variants.empty())
4516       return;
4517 
4518   // The end result is an all-pairs construction of the resultant pattern.
4519   std::vector<unsigned> Idxs(ChildVariants.size());
4520   bool NotDone;
4521   do {
4522 #ifndef NDEBUG
4523     LLVM_DEBUG(if (!Idxs.empty()) {
4524       errs() << Orig->getOperator()->getName() << ": Idxs = [ ";
4525       for (unsigned Idx : Idxs) {
4526         errs() << Idx << " ";
4527       }
4528       errs() << "]\n";
4529     });
4530 #endif
4531     // Create the variant and add it to the output list.
4532     std::vector<TreePatternNodePtr> NewChildren;
4533     NewChildren.reserve(ChildVariants.size());
4534     for (unsigned i = 0, e = ChildVariants.size(); i != e; ++i)
4535       NewChildren.push_back(ChildVariants[i][Idxs[i]]);
4536     TreePatternNodePtr R = makeIntrusiveRefCnt<TreePatternNode>(
4537         Orig->getOperator(), std::move(NewChildren), Orig->getNumTypes());
4538 
4539     // Copy over properties.
4540     R->setName(Orig->getName());
4541     R->setNamesAsPredicateArg(Orig->getNamesAsPredicateArg());
4542     R->setPredicateCalls(Orig->getPredicateCalls());
4543     R->setGISelFlagsRecord(Orig->getGISelFlagsRecord());
4544     R->setTransformFn(Orig->getTransformFn());
4545     for (unsigned i = 0, e = Orig->getNumTypes(); i != e; ++i)
4546       R->setType(i, Orig->getExtType(i));
4547 
4548     // If this pattern cannot match, do not include it as a variant.
4549     std::string ErrString;
4550     // Scan to see if this pattern has already been emitted.  We can get
4551     // duplication due to things like commuting:
4552     //   (and GPRC:$a, GPRC:$b) -> (and GPRC:$b, GPRC:$a)
4553     // which are the same pattern.  Ignore the dups.
4554     if (R->canPatternMatch(ErrString, CDP) &&
4555         none_of(OutVariants, [&](TreePatternNodePtr Variant) {
4556           return R->isIsomorphicTo(*Variant, DepVars);
4557         }))
4558       OutVariants.push_back(R);
4559 
4560     // Increment indices to the next permutation by incrementing the
4561     // indices from last index backward, e.g., generate the sequence
4562     // [0, 0], [0, 1], [1, 0], [1, 1].
4563     int IdxsIdx;
4564     for (IdxsIdx = Idxs.size() - 1; IdxsIdx >= 0; --IdxsIdx) {
4565       if (++Idxs[IdxsIdx] == ChildVariants[IdxsIdx].size())
4566         Idxs[IdxsIdx] = 0;
4567       else
4568         break;
4569     }
4570     NotDone = (IdxsIdx >= 0);
4571   } while (NotDone);
4572 }
4573 
4574 /// CombineChildVariants - A helper function for binary operators.
4575 ///
4576 static void CombineChildVariants(TreePatternNodePtr Orig,
4577                                  const std::vector<TreePatternNodePtr> &LHS,
4578                                  const std::vector<TreePatternNodePtr> &RHS,
4579                                  std::vector<TreePatternNodePtr> &OutVariants,
4580                                  CodeGenDAGPatterns &CDP,
4581                                  const MultipleUseVarSet &DepVars) {
4582   std::vector<std::vector<TreePatternNodePtr>> ChildVariants;
4583   ChildVariants.push_back(LHS);
4584   ChildVariants.push_back(RHS);
4585   CombineChildVariants(Orig, ChildVariants, OutVariants, CDP, DepVars);
4586 }
4587 
4588 static void
4589 GatherChildrenOfAssociativeOpcode(TreePatternNodePtr N,
4590                                   std::vector<TreePatternNodePtr> &Children) {
4591   assert(N->getNumChildren() == 2 &&
4592          "Associative but doesn't have 2 children!");
4593   Record *Operator = N->getOperator();
4594 
4595   // Only permit raw nodes.
4596   if (!N->getName().empty() || !N->getPredicateCalls().empty() ||
4597       N->getTransformFn()) {
4598     Children.push_back(N);
4599     return;
4600   }
4601 
4602   if (N->getChild(0).isLeaf() || N->getChild(0).getOperator() != Operator)
4603     Children.push_back(N->getChildShared(0));
4604   else
4605     GatherChildrenOfAssociativeOpcode(N->getChildShared(0), Children);
4606 
4607   if (N->getChild(1).isLeaf() || N->getChild(1).getOperator() != Operator)
4608     Children.push_back(N->getChildShared(1));
4609   else
4610     GatherChildrenOfAssociativeOpcode(N->getChildShared(1), Children);
4611 }
4612 
4613 /// GenerateVariantsOf - Given a pattern N, generate all permutations we can of
4614 /// the (potentially recursive) pattern by using algebraic laws.
4615 ///
4616 static void GenerateVariantsOf(TreePatternNodePtr N,
4617                                std::vector<TreePatternNodePtr> &OutVariants,
4618                                CodeGenDAGPatterns &CDP,
4619                                const MultipleUseVarSet &DepVars) {
4620   // We cannot permute leaves or ComplexPattern uses.
4621   if (N->isLeaf() || N->getOperator()->isSubClassOf("ComplexPattern")) {
4622     OutVariants.push_back(N);
4623     return;
4624   }
4625 
4626   // Look up interesting info about the node.
4627   const SDNodeInfo &NodeInfo = CDP.getSDNodeInfo(N->getOperator());
4628 
4629   // If this node is associative, re-associate.
4630   if (NodeInfo.hasProperty(SDNPAssociative)) {
4631     // Re-associate by pulling together all of the linked operators
4632     std::vector<TreePatternNodePtr> MaximalChildren;
4633     GatherChildrenOfAssociativeOpcode(N, MaximalChildren);
4634 
4635     // Only handle child sizes of 3.  Otherwise we'll end up trying too many
4636     // permutations.
4637     if (MaximalChildren.size() == 3) {
4638       // Find the variants of all of our maximal children.
4639       std::vector<TreePatternNodePtr> AVariants, BVariants, CVariants;
4640       GenerateVariantsOf(MaximalChildren[0], AVariants, CDP, DepVars);
4641       GenerateVariantsOf(MaximalChildren[1], BVariants, CDP, DepVars);
4642       GenerateVariantsOf(MaximalChildren[2], CVariants, CDP, DepVars);
4643 
4644       // There are only two ways we can permute the tree:
4645       //   (A op B) op C    and    A op (B op C)
4646       // Within these forms, we can also permute A/B/C.
4647 
4648       // Generate legal pair permutations of A/B/C.
4649       std::vector<TreePatternNodePtr> ABVariants;
4650       std::vector<TreePatternNodePtr> BAVariants;
4651       std::vector<TreePatternNodePtr> ACVariants;
4652       std::vector<TreePatternNodePtr> CAVariants;
4653       std::vector<TreePatternNodePtr> BCVariants;
4654       std::vector<TreePatternNodePtr> CBVariants;
4655       CombineChildVariants(N, AVariants, BVariants, ABVariants, CDP, DepVars);
4656       CombineChildVariants(N, BVariants, AVariants, BAVariants, CDP, DepVars);
4657       CombineChildVariants(N, AVariants, CVariants, ACVariants, CDP, DepVars);
4658       CombineChildVariants(N, CVariants, AVariants, CAVariants, CDP, DepVars);
4659       CombineChildVariants(N, BVariants, CVariants, BCVariants, CDP, DepVars);
4660       CombineChildVariants(N, CVariants, BVariants, CBVariants, CDP, DepVars);
4661 
4662       // Combine those into the result: (x op x) op x
4663       CombineChildVariants(N, ABVariants, CVariants, OutVariants, CDP, DepVars);
4664       CombineChildVariants(N, BAVariants, CVariants, OutVariants, CDP, DepVars);
4665       CombineChildVariants(N, ACVariants, BVariants, OutVariants, CDP, DepVars);
4666       CombineChildVariants(N, CAVariants, BVariants, OutVariants, CDP, DepVars);
4667       CombineChildVariants(N, BCVariants, AVariants, OutVariants, CDP, DepVars);
4668       CombineChildVariants(N, CBVariants, AVariants, OutVariants, CDP, DepVars);
4669 
4670       // Combine those into the result: x op (x op x)
4671       CombineChildVariants(N, CVariants, ABVariants, OutVariants, CDP, DepVars);
4672       CombineChildVariants(N, CVariants, BAVariants, OutVariants, CDP, DepVars);
4673       CombineChildVariants(N, BVariants, ACVariants, OutVariants, CDP, DepVars);
4674       CombineChildVariants(N, BVariants, CAVariants, OutVariants, CDP, DepVars);
4675       CombineChildVariants(N, AVariants, BCVariants, OutVariants, CDP, DepVars);
4676       CombineChildVariants(N, AVariants, CBVariants, OutVariants, CDP, DepVars);
4677       return;
4678     }
4679   }
4680 
4681   // Compute permutations of all children.
4682   std::vector<std::vector<TreePatternNodePtr>> ChildVariants(
4683       N->getNumChildren());
4684   for (unsigned i = 0, e = N->getNumChildren(); i != e; ++i)
4685     GenerateVariantsOf(N->getChildShared(i), ChildVariants[i], CDP, DepVars);
4686 
4687   // Build all permutations based on how the children were formed.
4688   CombineChildVariants(N, ChildVariants, OutVariants, CDP, DepVars);
4689 
4690   // If this node is commutative, consider the commuted order.
4691   bool isCommIntrinsic = N->isCommutativeIntrinsic(CDP);
4692   if (NodeInfo.hasProperty(SDNPCommutative) || isCommIntrinsic) {
4693     unsigned Skip = isCommIntrinsic ? 1 : 0; // First operand is intrinsic id.
4694     assert(N->getNumChildren() >= (2 + Skip) &&
4695            "Commutative but doesn't have 2 children!");
4696     // Don't allow commuting children which are actually register references.
4697     bool NoRegisters = true;
4698     unsigned i = 0 + Skip;
4699     unsigned e = 2 + Skip;
4700     for (; i != e; ++i) {
4701       TreePatternNode &Child = N->getChild(i);
4702       if (Child.isLeaf())
4703         if (DefInit *DI = dyn_cast<DefInit>(Child.getLeafValue())) {
4704           Record *RR = DI->getDef();
4705           if (RR->isSubClassOf("Register"))
4706             NoRegisters = false;
4707         }
4708     }
4709     // Consider the commuted order.
4710     if (NoRegisters) {
4711       // Swap the first two operands after the intrinsic id, if present.
4712       unsigned i = isCommIntrinsic ? 1 : 0;
4713       std::swap(ChildVariants[i], ChildVariants[i + 1]);
4714       CombineChildVariants(N, ChildVariants, OutVariants, CDP, DepVars);
4715     }
4716   }
4717 }
4718 
4719 // GenerateVariants - Generate variants.  For example, commutative patterns can
4720 // match multiple ways.  Add them to PatternsToMatch as well.
4721 void CodeGenDAGPatterns::GenerateVariants() {
4722   LLVM_DEBUG(errs() << "Generating instruction variants.\n");
4723 
4724   // Loop over all of the patterns we've collected, checking to see if we can
4725   // generate variants of the instruction, through the exploitation of
4726   // identities.  This permits the target to provide aggressive matching without
4727   // the .td file having to contain tons of variants of instructions.
4728   //
4729   // Note that this loop adds new patterns to the PatternsToMatch list, but we
4730   // intentionally do not reconsider these.  Any variants of added patterns have
4731   // already been added.
4732   //
4733   for (unsigned i = 0, e = PatternsToMatch.size(); i != e; ++i) {
4734     MultipleUseVarSet DepVars;
4735     std::vector<TreePatternNodePtr> Variants;
4736     FindDepVars(PatternsToMatch[i].getSrcPattern(), DepVars);
4737     LLVM_DEBUG(errs() << "Dependent/multiply used variables: ");
4738     LLVM_DEBUG(DumpDepVars(DepVars));
4739     LLVM_DEBUG(errs() << "\n");
4740     GenerateVariantsOf(PatternsToMatch[i].getSrcPatternShared(), Variants,
4741                        *this, DepVars);
4742 
4743     assert(PatternsToMatch[i].getHwModeFeatures().empty() &&
4744            "HwModes should not have been expanded yet!");
4745 
4746     assert(!Variants.empty() && "Must create at least original variant!");
4747     if (Variants.size() == 1) // No additional variants for this pattern.
4748       continue;
4749 
4750     LLVM_DEBUG(errs() << "FOUND VARIANTS OF: ";
4751                PatternsToMatch[i].getSrcPattern().dump(); errs() << "\n");
4752 
4753     for (unsigned v = 0, e = Variants.size(); v != e; ++v) {
4754       TreePatternNodePtr Variant = Variants[v];
4755 
4756       LLVM_DEBUG(errs() << "  VAR#" << v << ": "; Variant->dump();
4757                  errs() << "\n");
4758 
4759       // Scan to see if an instruction or explicit pattern already matches this.
4760       bool AlreadyExists = false;
4761       for (unsigned p = 0, e = PatternsToMatch.size(); p != e; ++p) {
4762         // Skip if the top level predicates do not match.
4763         if ((i != p) && (PatternsToMatch[i].getPredicates() !=
4764                          PatternsToMatch[p].getPredicates()))
4765           continue;
4766         // Check to see if this variant already exists.
4767         if (Variant->isIsomorphicTo(PatternsToMatch[p].getSrcPattern(),
4768                                     DepVars)) {
4769           LLVM_DEBUG(errs() << "  *** ALREADY EXISTS, ignoring variant.\n");
4770           AlreadyExists = true;
4771           break;
4772         }
4773       }
4774       // If we already have it, ignore the variant.
4775       if (AlreadyExists)
4776         continue;
4777 
4778       // Otherwise, add it to the list of patterns we have.
4779       PatternsToMatch.emplace_back(
4780           PatternsToMatch[i].getSrcRecord(), PatternsToMatch[i].getPredicates(),
4781           Variant, PatternsToMatch[i].getDstPatternShared(),
4782           PatternsToMatch[i].getDstRegs(),
4783           PatternsToMatch[i].getAddedComplexity(), Record::getNewUID(Records),
4784           PatternsToMatch[i].getGISelShouldIgnore(),
4785           PatternsToMatch[i].getHwModeFeatures());
4786     }
4787 
4788     LLVM_DEBUG(errs() << "\n");
4789   }
4790 }
4791