blob: d41de68f02ffabd82d1bfd76624bcc2f94d4c405 [file] [log] [blame]
Chris Lattnerab3242f2008-01-06 01:10:31 +00001//===- CodeGenDAGPatterns.h - Read DAG patterns from .td file ---*- C++ -*-===//
Chris Lattner8cab0212008-01-05 22:25:12 +00002//
Chandler Carruth2946cd72019-01-19 08:50:56 +00003// 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
Chris Lattner8cab0212008-01-05 22:25:12 +00006//
7//===----------------------------------------------------------------------===//
8//
Chris Lattnerab3242f2008-01-06 01:10:31 +00009// This file declares the CodeGenDAGPatterns class, which is used to read and
Chris Lattner8cab0212008-01-05 22:25:12 +000010// represent the patterns present in a .td file for instructions.
11//
12//===----------------------------------------------------------------------===//
13
Benjamin Kramera7c40ef2014-08-13 16:26:38 +000014#ifndef LLVM_UTILS_TABLEGEN_CODEGENDAGPATTERNS_H
15#define LLVM_UTILS_TABLEGEN_CODEGENDAGPATTERNS_H
Chris Lattner8cab0212008-01-05 22:25:12 +000016
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +000017#include "CodeGenHwModes.h"
Chris Lattner8cab0212008-01-05 22:25:12 +000018#include "CodeGenIntrinsics.h"
Chandler Carruth91d19d82012-12-04 10:37:14 +000019#include "CodeGenTarget.h"
Matt Arsenault303327d2017-12-20 19:36:28 +000020#include "SDNodeProperties.h"
Craig Topperbd199f82018-12-05 00:47:59 +000021#include "llvm/ADT/MapVector.h"
Chris Lattnercabe0372010-03-15 06:00:16 +000022#include "llvm/ADT/SmallVector.h"
23#include "llvm/ADT/StringMap.h"
Zachary Turner249dc142017-09-20 18:01:40 +000024#include "llvm/ADT/StringSet.h"
Craig Topperc4965bc2012-02-05 07:21:30 +000025#include "llvm/Support/ErrorHandling.h"
Krzysztof Parzyszekaffd2012017-09-19 18:42:34 +000026#include "llvm/Support/MathExtras.h"
Chris Lattner1802b172010-03-19 01:07:44 +000027#include <algorithm>
Krzysztof Parzyszekaffd2012017-09-19 18:42:34 +000028#include <array>
Daniel Sanders7e523672017-11-11 03:23:44 +000029#include <functional>
Chris Lattner1802b172010-03-19 01:07:44 +000030#include <map>
Craig Topperbd199f82018-12-05 00:47:59 +000031#include <numeric>
Chandler Carruth91d19d82012-12-04 10:37:14 +000032#include <set>
33#include <vector>
Chris Lattner8cab0212008-01-05 22:25:12 +000034
35namespace llvm {
Chris Lattner8cab0212008-01-05 22:25:12 +000036
Krzysztof Parzyszekaffd2012017-09-19 18:42:34 +000037class Record;
38class Init;
39class ListInit;
40class DagInit;
41class SDNodeInfo;
42class TreePattern;
43class TreePatternNode;
44class CodeGenDAGPatterns;
45class ComplexPattern;
46
Florian Hahn75e87c32018-05-30 21:00:18 +000047/// Shared pointer for TreePatternNode.
48using TreePatternNodePtr = std::shared_ptr<TreePatternNode>;
49
Krzysztof Parzyszekaffd2012017-09-19 18:42:34 +000050/// This represents a set of MVTs. Since the underlying type for the MVT
51/// is uint8_t, there are at most 256 values. To reduce the number of memory
52/// allocations and deallocations, represent the set as a sequence of bits.
53/// To reduce the allocations even further, make MachineValueTypeSet own
54/// the storage and use std::array as the bit container.
55struct MachineValueTypeSet {
56 static_assert(std::is_same<std::underlying_type<MVT::SimpleValueType>::type,
57 uint8_t>::value,
58 "Change uint8_t here to the SimpleValueType's type");
59 static unsigned constexpr Capacity = std::numeric_limits<uint8_t>::max()+1;
60 using WordType = uint64_t;
Craig Topperd022d252017-09-21 04:55:04 +000061 static unsigned constexpr WordWidth = CHAR_BIT*sizeof(WordType);
Krzysztof Parzyszekaffd2012017-09-19 18:42:34 +000062 static unsigned constexpr NumWords = Capacity/WordWidth;
63 static_assert(NumWords*WordWidth == Capacity,
64 "Capacity should be a multiple of WordWidth");
65
66 LLVM_ATTRIBUTE_ALWAYS_INLINE
67 MachineValueTypeSet() {
68 clear();
69 }
70
71 LLVM_ATTRIBUTE_ALWAYS_INLINE
72 unsigned size() const {
73 unsigned Count = 0;
74 for (WordType W : Words)
75 Count += countPopulation(W);
76 return Count;
77 }
78 LLVM_ATTRIBUTE_ALWAYS_INLINE
79 void clear() {
80 std::memset(Words.data(), 0, NumWords*sizeof(WordType));
81 }
82 LLVM_ATTRIBUTE_ALWAYS_INLINE
83 bool empty() const {
84 for (WordType W : Words)
85 if (W != 0)
86 return false;
87 return true;
88 }
89 LLVM_ATTRIBUTE_ALWAYS_INLINE
90 unsigned count(MVT T) const {
91 return (Words[T.SimpleTy / WordWidth] >> (T.SimpleTy % WordWidth)) & 1;
92 }
93 std::pair<MachineValueTypeSet&,bool> insert(MVT T) {
94 bool V = count(T.SimpleTy);
95 Words[T.SimpleTy / WordWidth] |= WordType(1) << (T.SimpleTy % WordWidth);
96 return {*this, V};
97 }
98 MachineValueTypeSet &insert(const MachineValueTypeSet &S) {
99 for (unsigned i = 0; i != NumWords; ++i)
100 Words[i] |= S.Words[i];
101 return *this;
102 }
103 LLVM_ATTRIBUTE_ALWAYS_INLINE
104 void erase(MVT T) {
105 Words[T.SimpleTy / WordWidth] &= ~(WordType(1) << (T.SimpleTy % WordWidth));
106 }
107
108 struct const_iterator {
109 // Some implementations of the C++ library require these traits to be
110 // defined.
111 using iterator_category = std::forward_iterator_tag;
112 using value_type = MVT;
113 using difference_type = ptrdiff_t;
114 using pointer = const MVT*;
115 using reference = const MVT&;
116
117 LLVM_ATTRIBUTE_ALWAYS_INLINE
118 MVT operator*() const {
119 assert(Pos != Capacity);
120 return MVT::SimpleValueType(Pos);
121 }
122 LLVM_ATTRIBUTE_ALWAYS_INLINE
123 const_iterator(const MachineValueTypeSet *S, bool End) : Set(S) {
124 Pos = End ? Capacity : find_from_pos(0);
125 }
126 LLVM_ATTRIBUTE_ALWAYS_INLINE
127 const_iterator &operator++() {
128 assert(Pos != Capacity);
129 Pos = find_from_pos(Pos+1);
130 return *this;
131 }
132
133 LLVM_ATTRIBUTE_ALWAYS_INLINE
134 bool operator==(const const_iterator &It) const {
135 return Set == It.Set && Pos == It.Pos;
136 }
137 LLVM_ATTRIBUTE_ALWAYS_INLINE
138 bool operator!=(const const_iterator &It) const {
139 return !operator==(It);
140 }
141
142 private:
143 unsigned find_from_pos(unsigned P) const {
144 unsigned SkipWords = P / WordWidth;
145 unsigned SkipBits = P % WordWidth;
146 unsigned Count = SkipWords * WordWidth;
147
148 // If P is in the middle of a word, process it manually here, because
149 // the trailing bits need to be masked off to use findFirstSet.
150 if (SkipBits != 0) {
151 WordType W = Set->Words[SkipWords];
152 W &= maskLeadingOnes<WordType>(WordWidth-SkipBits);
153 if (W != 0)
154 return Count + findFirstSet(W);
155 Count += WordWidth;
156 SkipWords++;
157 }
158
159 for (unsigned i = SkipWords; i != NumWords; ++i) {
160 WordType W = Set->Words[i];
161 if (W != 0)
162 return Count + findFirstSet(W);
163 Count += WordWidth;
164 }
165 return Capacity;
166 }
167
168 const MachineValueTypeSet *Set;
169 unsigned Pos;
170 };
171
172 LLVM_ATTRIBUTE_ALWAYS_INLINE
173 const_iterator begin() const { return const_iterator(this, false); }
174 LLVM_ATTRIBUTE_ALWAYS_INLINE
175 const_iterator end() const { return const_iterator(this, true); }
176
177 LLVM_ATTRIBUTE_ALWAYS_INLINE
178 bool operator==(const MachineValueTypeSet &S) const {
179 return Words == S.Words;
180 }
181 LLVM_ATTRIBUTE_ALWAYS_INLINE
182 bool operator!=(const MachineValueTypeSet &S) const {
183 return !operator==(S);
184 }
185
186private:
187 friend struct const_iterator;
188 std::array<WordType,NumWords> Words;
189};
190
191struct TypeSetByHwMode : public InfoByHwMode<MachineValueTypeSet> {
192 using SetType = MachineValueTypeSet;
Tom Stellard9ad714f2019-02-20 19:43:47 +0000193 std::vector<unsigned> AddrSpaces;
Jim Grosbach50986b52010-12-24 05:06:32 +0000194
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +0000195 TypeSetByHwMode() = default;
196 TypeSetByHwMode(const TypeSetByHwMode &VTS) = default;
197 TypeSetByHwMode(MVT::SimpleValueType VT)
198 : TypeSetByHwMode(ValueTypeByHwMode(VT)) {}
199 TypeSetByHwMode(ValueTypeByHwMode VT)
200 : TypeSetByHwMode(ArrayRef<ValueTypeByHwMode>(&VT, 1)) {}
201 TypeSetByHwMode(ArrayRef<ValueTypeByHwMode> VTList);
Jim Grosbach50986b52010-12-24 05:06:32 +0000202
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +0000203 SetType &getOrCreate(unsigned Mode) {
204 if (hasMode(Mode))
205 return get(Mode);
206 return Map.insert({Mode,SetType()}).first->second;
207 }
Jim Grosbach50986b52010-12-24 05:06:32 +0000208
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +0000209 bool isValueTypeByHwMode(bool AllowEmpty) const;
210 ValueTypeByHwMode getValueTypeByHwMode() const;
Krzysztof Parzyszekaffd2012017-09-19 18:42:34 +0000211
212 LLVM_ATTRIBUTE_ALWAYS_INLINE
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +0000213 bool isMachineValueType() const {
214 return isDefaultOnly() && Map.begin()->second.size() == 1;
215 }
Jim Grosbach50986b52010-12-24 05:06:32 +0000216
Krzysztof Parzyszekaffd2012017-09-19 18:42:34 +0000217 LLVM_ATTRIBUTE_ALWAYS_INLINE
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +0000218 MVT getMachineValueType() const {
219 assert(isMachineValueType());
220 return *Map.begin()->second.begin();
221 }
Jim Grosbach50986b52010-12-24 05:06:32 +0000222
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +0000223 bool isPossible() const;
Krzysztof Parzyszekaffd2012017-09-19 18:42:34 +0000224
225 LLVM_ATTRIBUTE_ALWAYS_INLINE
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +0000226 bool isDefaultOnly() const {
Krzysztof Parzyszekaffd2012017-09-19 18:42:34 +0000227 return Map.size() == 1 && Map.begin()->first == DefaultMode;
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +0000228 }
Jim Grosbach50986b52010-12-24 05:06:32 +0000229
Tom Stellard9ad714f2019-02-20 19:43:47 +0000230 bool isPointer() const {
231 return getValueTypeByHwMode().isPointer();
232 }
233
234 unsigned getPtrAddrSpace() const {
235 assert(isPointer());
236 return getValueTypeByHwMode().PtrAddrSpace;
237 }
238
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +0000239 bool insert(const ValueTypeByHwMode &VVT);
240 bool constrain(const TypeSetByHwMode &VTS);
241 template <typename Predicate> bool constrain(Predicate P);
Zachary Turner249dc142017-09-20 18:01:40 +0000242 template <typename Predicate>
243 bool assign_if(const TypeSetByHwMode &VTS, Predicate P);
Jim Grosbach50986b52010-12-24 05:06:32 +0000244
Zachary Turner249dc142017-09-20 18:01:40 +0000245 void writeToStream(raw_ostream &OS) const;
246 static void writeToStream(const SetType &S, raw_ostream &OS);
Jim Grosbach50986b52010-12-24 05:06:32 +0000247
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +0000248 bool operator==(const TypeSetByHwMode &VTS) const;
249 bool operator!=(const TypeSetByHwMode &VTS) const { return !(*this == VTS); }
Jim Grosbach50986b52010-12-24 05:06:32 +0000250
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +0000251 void dump() const;
Krzysztof Parzyszek2e0f7bd2017-12-21 17:12:43 +0000252 bool validate() const;
Craig Topper74169dc2014-01-28 04:49:01 +0000253
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +0000254private:
Tom Stellard9ad714f2019-02-20 19:43:47 +0000255 unsigned PtrAddrSpace = std::numeric_limits<unsigned>::max();
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +0000256 /// Intersect two sets. Return true if anything has changed.
257 bool intersect(SetType &Out, const SetType &In);
258};
Jim Grosbach50986b52010-12-24 05:06:32 +0000259
Krzysztof Parzyszek7725e492017-09-22 18:29:37 +0000260raw_ostream &operator<<(raw_ostream &OS, const TypeSetByHwMode &T);
261
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +0000262struct TypeInfer {
263 TypeInfer(TreePattern &T) : TP(T), ForceMode(0) {}
Jim Grosbach50986b52010-12-24 05:06:32 +0000264
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +0000265 bool isConcrete(const TypeSetByHwMode &VTS, bool AllowEmpty) const {
266 return VTS.isValueTypeByHwMode(AllowEmpty);
267 }
268 ValueTypeByHwMode getConcrete(const TypeSetByHwMode &VTS,
269 bool AllowEmpty) const {
270 assert(VTS.isValueTypeByHwMode(AllowEmpty));
271 return VTS.getValueTypeByHwMode();
272 }
Duncan Sands13237ac2008-06-06 12:08:01 +0000273
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +0000274 /// The protocol in the following functions (Merge*, force*, Enforce*,
275 /// expand*) is to return "true" if a change has been made, "false"
276 /// otherwise.
Chris Lattner8cab0212008-01-05 22:25:12 +0000277
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +0000278 bool MergeInTypeInfo(TypeSetByHwMode &Out, const TypeSetByHwMode &In);
279 bool MergeInTypeInfo(TypeSetByHwMode &Out, MVT::SimpleValueType InVT) {
280 return MergeInTypeInfo(Out, TypeSetByHwMode(InVT));
281 }
282 bool MergeInTypeInfo(TypeSetByHwMode &Out, ValueTypeByHwMode InVT) {
283 return MergeInTypeInfo(Out, TypeSetByHwMode(InVT));
284 }
Bob Wilsonf7e587f2009-08-12 22:30:59 +0000285
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +0000286 /// Reduce the set \p Out to have at most one element for each mode.
287 bool forceArbitrary(TypeSetByHwMode &Out);
Chris Lattnercabe0372010-03-15 06:00:16 +0000288
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +0000289 /// The following four functions ensure that upon return the set \p Out
290 /// will only contain types of the specified kind: integer, floating-point,
291 /// scalar, or vector.
292 /// If \p Out is empty, all legal types of the specified kind will be added
293 /// to it. Otherwise, all types that are not of the specified kind will be
294 /// removed from \p Out.
295 bool EnforceInteger(TypeSetByHwMode &Out);
296 bool EnforceFloatingPoint(TypeSetByHwMode &Out);
297 bool EnforceScalar(TypeSetByHwMode &Out);
298 bool EnforceVector(TypeSetByHwMode &Out);
Chris Lattnercabe0372010-03-15 06:00:16 +0000299
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +0000300 /// If \p Out is empty, fill it with all legal types. Otherwise, leave it
301 /// unchanged.
302 bool EnforceAny(TypeSetByHwMode &Out);
303 /// Make sure that for each type in \p Small, there exists a larger type
304 /// in \p Big.
305 bool EnforceSmallerThan(TypeSetByHwMode &Small, TypeSetByHwMode &Big);
306 /// 1. Ensure that for each type T in \p Vec, T is a vector type, and that
307 /// for each type U in \p Elem, U is a scalar type.
308 /// 2. Ensure that for each (scalar) type U in \p Elem, there exists a
309 /// (vector) type T in \p Vec, such that U is the element type of T.
310 bool EnforceVectorEltTypeIs(TypeSetByHwMode &Vec, TypeSetByHwMode &Elem);
311 bool EnforceVectorEltTypeIs(TypeSetByHwMode &Vec,
312 const ValueTypeByHwMode &VVT);
313 /// Ensure that for each type T in \p Sub, T is a vector type, and there
314 /// exists a type U in \p Vec such that U is a vector type with the same
315 /// element type as T and at least as many elements as T.
316 bool EnforceVectorSubVectorTypeIs(TypeSetByHwMode &Vec,
317 TypeSetByHwMode &Sub);
318 /// 1. Ensure that \p V has a scalar type iff \p W has a scalar type.
319 /// 2. Ensure that for each vector type T in \p V, there exists a vector
320 /// type U in \p W, such that T and U have the same number of elements.
321 /// 3. Ensure that for each vector type U in \p W, there exists a vector
322 /// type T in \p V, such that T and U have the same number of elements
323 /// (reverse of 2).
324 bool EnforceSameNumElts(TypeSetByHwMode &V, TypeSetByHwMode &W);
325 /// 1. Ensure that for each type T in \p A, there exists a type U in \p B,
326 /// such that T and U have equal size in bits.
327 /// 2. Ensure that for each type U in \p B, there exists a type T in \p A
328 /// such that T and U have equal size in bits (reverse of 1).
329 bool EnforceSameSize(TypeSetByHwMode &A, TypeSetByHwMode &B);
Chris Lattnercabe0372010-03-15 06:00:16 +0000330
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +0000331 /// For each overloaded type (i.e. of form *Any), replace it with the
332 /// corresponding subset of legal, specific types.
333 void expandOverloads(TypeSetByHwMode &VTS);
334 void expandOverloads(TypeSetByHwMode::SetType &Out,
335 const TypeSetByHwMode::SetType &Legal);
Jim Grosbach50986b52010-12-24 05:06:32 +0000336
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +0000337 struct ValidateOnExit {
Krzysztof Parzyszek2e0f7bd2017-12-21 17:12:43 +0000338 ValidateOnExit(TypeSetByHwMode &T, TypeInfer &TI) : Infer(TI), VTS(T) {}
339 #ifndef NDEBUG
340 ~ValidateOnExit();
341 #else
342 ~ValidateOnExit() {} // Empty destructor with NDEBUG.
343 #endif
344 TypeInfer &Infer;
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +0000345 TypeSetByHwMode &VTS;
Chris Lattnercabe0372010-03-15 06:00:16 +0000346 };
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +0000347
Ulrich Weigand22b1af82018-07-13 16:42:15 +0000348 struct SuppressValidation {
349 SuppressValidation(TypeInfer &TI) : Infer(TI), SavedValidate(TI.Validate) {
350 Infer.Validate = false;
351 }
352 ~SuppressValidation() {
353 Infer.Validate = SavedValidate;
354 }
355 TypeInfer &Infer;
356 bool SavedValidate;
357 };
358
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +0000359 TreePattern &TP;
360 unsigned ForceMode; // Mode to use when set.
361 bool CodeGen = false; // Set during generation of matcher code.
Ulrich Weigand22b1af82018-07-13 16:42:15 +0000362 bool Validate = true; // Indicate whether to validate types.
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +0000363
364private:
Simon Pilgrim45e61c52018-08-17 15:54:07 +0000365 const TypeSetByHwMode &getLegalTypes();
Krzysztof Parzyszekaffd2012017-09-19 18:42:34 +0000366
Simon Pilgrim45e61c52018-08-17 15:54:07 +0000367 /// Cached legal types (in default mode).
Krzysztof Parzyszekaffd2012017-09-19 18:42:34 +0000368 bool LegalTypesCached = false;
Simon Pilgrim45e61c52018-08-17 15:54:07 +0000369 TypeSetByHwMode LegalCache;
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +0000370};
Chris Lattner8cab0212008-01-05 22:25:12 +0000371
Scott Michel94420742008-03-05 17:49:05 +0000372/// Set type used to track multiply used variables in patterns
Zachary Turner249dc142017-09-20 18:01:40 +0000373typedef StringSet<> MultipleUseVarSet;
Scott Michel94420742008-03-05 17:49:05 +0000374
Chris Lattner8cab0212008-01-05 22:25:12 +0000375/// SDTypeConstraint - This is a discriminated union of constraints,
376/// corresponding to the SDTypeConstraint tablegen class in Target.td.
377struct SDTypeConstraint {
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +0000378 SDTypeConstraint(Record *R, const CodeGenHwModes &CGH);
Jim Grosbach50986b52010-12-24 05:06:32 +0000379
Chris Lattner8cab0212008-01-05 22:25:12 +0000380 unsigned OperandNo; // The operand # this constraint applies to.
Jim Grosbach50986b52010-12-24 05:06:32 +0000381 enum {
382 SDTCisVT, SDTCisPtrTy, SDTCisInt, SDTCisFP, SDTCisVec, SDTCisSameAs,
David Greene127fd1d2011-01-24 20:53:18 +0000383 SDTCisVTSmallerThanOp, SDTCisOpSmallerThanOp, SDTCisEltOfVec,
Craig Topper9a44b3f2015-11-26 07:02:18 +0000384 SDTCisSubVecOfVec, SDTCVecEltisVT, SDTCisSameNumEltsAs, SDTCisSameSizeAs
Chris Lattner8cab0212008-01-05 22:25:12 +0000385 } ConstraintType;
Jim Grosbach50986b52010-12-24 05:06:32 +0000386
Chris Lattner8cab0212008-01-05 22:25:12 +0000387 union { // The discriminated union.
388 struct {
Chris Lattner8cab0212008-01-05 22:25:12 +0000389 unsigned OtherOperandNum;
390 } SDTCisSameAs_Info;
391 struct {
392 unsigned OtherOperandNum;
393 } SDTCisVTSmallerThanOp_Info;
394 struct {
395 unsigned BigOperandNum;
396 } SDTCisOpSmallerThanOp_Info;
397 struct {
398 unsigned OtherOperandNum;
Nate Begeman17bedbc2008-02-09 01:37:05 +0000399 } SDTCisEltOfVec_Info;
David Greene127fd1d2011-01-24 20:53:18 +0000400 struct {
401 unsigned OtherOperandNum;
402 } SDTCisSubVecOfVec_Info;
Craig Topper0be34582015-03-05 07:11:34 +0000403 struct {
Craig Topper0be34582015-03-05 07:11:34 +0000404 unsigned OtherOperandNum;
405 } SDTCisSameNumEltsAs_Info;
Craig Topper9a44b3f2015-11-26 07:02:18 +0000406 struct {
407 unsigned OtherOperandNum;
408 } SDTCisSameSizeAs_Info;
Chris Lattner8cab0212008-01-05 22:25:12 +0000409 } x;
410
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +0000411 // The VT for SDTCisVT and SDTCVecEltisVT.
412 // Must not be in the union because it has a non-trivial destructor.
413 ValueTypeByHwMode VVT;
414
Chris Lattner8cab0212008-01-05 22:25:12 +0000415 /// ApplyTypeConstraint - Given a node in a pattern, apply this type
416 /// constraint to the nodes operands. This returns true if it makes a
Joerg Sonnenberger635debe2012-10-25 20:33:17 +0000417 /// change, false otherwise. If a type contradiction is found, an error
418 /// is flagged.
Florian Hahn6b1db822018-06-14 20:32:58 +0000419 bool ApplyTypeConstraint(TreePatternNode *N, const SDNodeInfo &NodeInfo,
Chris Lattner8cab0212008-01-05 22:25:12 +0000420 TreePattern &TP) const;
Chris Lattner8cab0212008-01-05 22:25:12 +0000421};
422
Nicolai Haehnle445b0b62018-11-30 14:15:13 +0000423/// ScopedName - A name of a node associated with a "scope" that indicates
424/// the context (e.g. instance of Pattern or PatFrag) in which the name was
425/// used. This enables substitution of pattern fragments while keeping track
426/// of what name(s) were originally given to various nodes in the tree.
427class ScopedName {
428 unsigned Scope;
429 std::string Identifier;
430public:
431 ScopedName(unsigned Scope, StringRef Identifier)
432 : Scope(Scope), Identifier(Identifier) {
433 assert(Scope != 0 &&
434 "Scope == 0 is used to indicate predicates without arguments");
435 }
436
437 unsigned getScope() const { return Scope; }
438 const std::string &getIdentifier() const { return Identifier; }
439
440 std::string getFullName() const;
441
442 bool operator==(const ScopedName &o) const;
443 bool operator!=(const ScopedName &o) const;
444};
445
Chris Lattner8cab0212008-01-05 22:25:12 +0000446/// SDNodeInfo - One of these records is created for each SDNode instance in
447/// the target .td file. This represents the various dag nodes we will be
448/// processing.
449class SDNodeInfo {
450 Record *Def;
Craig Topperbcd3c372017-05-31 21:12:46 +0000451 StringRef EnumName;
452 StringRef SDClassName;
Chris Lattner8cab0212008-01-05 22:25:12 +0000453 unsigned Properties;
454 unsigned NumResults;
455 int NumOperands;
456 std::vector<SDTypeConstraint> TypeConstraints;
457public:
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +0000458 // Parse the specified record.
459 SDNodeInfo(Record *R, const CodeGenHwModes &CGH);
Jim Grosbach50986b52010-12-24 05:06:32 +0000460
Chris Lattner8cab0212008-01-05 22:25:12 +0000461 unsigned getNumResults() const { return NumResults; }
Jim Grosbach50986b52010-12-24 05:06:32 +0000462
Chris Lattner135091b2010-03-28 08:48:47 +0000463 /// getNumOperands - This is the number of operands required or -1 if
464 /// variadic.
Chris Lattner8cab0212008-01-05 22:25:12 +0000465 int getNumOperands() const { return NumOperands; }
466 Record *getRecord() const { return Def; }
Craig Topperbcd3c372017-05-31 21:12:46 +0000467 StringRef getEnumName() const { return EnumName; }
468 StringRef getSDClassName() const { return SDClassName; }
Jim Grosbach50986b52010-12-24 05:06:32 +0000469
Chris Lattner8cab0212008-01-05 22:25:12 +0000470 const std::vector<SDTypeConstraint> &getTypeConstraints() const {
471 return TypeConstraints;
472 }
Jim Grosbach50986b52010-12-24 05:06:32 +0000473
Chris Lattner99e53b32010-02-28 00:22:30 +0000474 /// getKnownType - If the type constraints on this node imply a fixed type
475 /// (e.g. all stores return void, etc), then return it as an
Chris Lattnerda5b4ad2010-03-19 01:14:27 +0000476 /// MVT::SimpleValueType. Otherwise, return MVT::Other.
Chris Lattner6c2d1782010-03-24 00:41:19 +0000477 MVT::SimpleValueType getKnownType(unsigned ResNo) const;
Jim Grosbach50986b52010-12-24 05:06:32 +0000478
Chris Lattner8cab0212008-01-05 22:25:12 +0000479 /// hasProperty - Return true if this node has the specified property.
480 ///
481 bool hasProperty(enum SDNP Prop) const { return Properties & (1 << Prop); }
482
483 /// ApplyTypeConstraints - Given a node in a pattern, apply the type
484 /// constraints for this node to the operands of the node. This returns
485 /// true if it makes a change, false otherwise. If a type contradiction is
Joerg Sonnenberger635debe2012-10-25 20:33:17 +0000486 /// found, an error is flagged.
Florian Hahn6b1db822018-06-14 20:32:58 +0000487 bool ApplyTypeConstraints(TreePatternNode *N, TreePattern &TP) const;
Chris Lattner8cab0212008-01-05 22:25:12 +0000488};
Simon Pilgrim8c4d0612017-09-22 16:57:28 +0000489
Chris Lattner514e2922011-04-17 21:38:24 +0000490/// TreePredicateFn - This is an abstraction that represents the predicates on
491/// a PatFrag node. This is a simple one-word wrapper around a pointer to
492/// provide nice accessors.
493class TreePredicateFn {
494 /// PatFragRec - This is the TreePattern for the PatFrag that we
495 /// originally came from.
496 TreePattern *PatFragRec;
497public:
498 /// TreePredicateFn constructor. Here 'N' is a subclass of PatFrag.
Chris Lattner2ff8c1a2011-04-17 22:05:17 +0000499 TreePredicateFn(TreePattern *N);
Chris Lattner514e2922011-04-17 21:38:24 +0000500
Simon Pilgrim8c4d0612017-09-22 16:57:28 +0000501
Chris Lattner514e2922011-04-17 21:38:24 +0000502 TreePattern *getOrigPatFragRecord() const { return PatFragRec; }
Simon Pilgrim8c4d0612017-09-22 16:57:28 +0000503
Chris Lattner514e2922011-04-17 21:38:24 +0000504 /// isAlwaysTrue - Return true if this is a noop predicate.
505 bool isAlwaysTrue() const;
Simon Pilgrim8c4d0612017-09-22 16:57:28 +0000506
Daniel Sandersadbf58d2017-10-15 19:01:32 +0000507 bool isImmediatePattern() const { return hasImmCode(); }
Simon Pilgrim8c4d0612017-09-22 16:57:28 +0000508
Chris Lattner07add492011-04-18 06:22:33 +0000509 /// getImmediatePredicateCode - Return the code that evaluates this pattern if
510 /// this is an immediate predicate. It is an error to call this on a
511 /// non-immediate pattern.
Daniel Sanders3f267bf2017-10-15 02:06:44 +0000512 std::string getImmediatePredicateCode() const {
513 std::string Result = getImmCode();
Chris Lattner07add492011-04-18 06:22:33 +0000514 assert(!Result.empty() && "Isn't an immediate pattern!");
515 return Result;
516 }
Simon Pilgrim8c4d0612017-09-22 16:57:28 +0000517
Chris Lattner514e2922011-04-17 21:38:24 +0000518 bool operator==(const TreePredicateFn &RHS) const {
519 return PatFragRec == RHS.PatFragRec;
520 }
521
522 bool operator!=(const TreePredicateFn &RHS) const { return !(*this == RHS); }
523
524 /// Return the name to use in the generated code to reference this, this is
525 /// "Predicate_foo" if from a pattern fragment "foo".
526 std::string getFnName() const;
Simon Pilgrim8c4d0612017-09-22 16:57:28 +0000527
Chris Lattner514e2922011-04-17 21:38:24 +0000528 /// getCodeToRunOnSDNode - Return the code for the function body that
529 /// evaluates this predicate. The argument is expected to be in "Node",
530 /// not N. This handles casting and conversion to a concrete node type as
531 /// appropriate.
532 std::string getCodeToRunOnSDNode() const;
Simon Pilgrim8c4d0612017-09-22 16:57:28 +0000533
Daniel Sanders649c5852017-10-13 20:42:18 +0000534 /// Get the data type of the argument to getImmediatePredicateCode().
Simon Pilgrim6ecae9f2017-10-14 21:27:53 +0000535 StringRef getImmType() const;
Daniel Sanders649c5852017-10-13 20:42:18 +0000536
Daniel Sanders11300ce2017-10-13 21:28:03 +0000537 /// Get a string that describes the type returned by getImmType() but is
538 /// usable as part of an identifier.
Simon Pilgrim6ecae9f2017-10-14 21:27:53 +0000539 StringRef getImmTypeIdentifier() const;
Daniel Sanders11300ce2017-10-13 21:28:03 +0000540
Nicolai Haehnle445b0b62018-11-30 14:15:13 +0000541 // Predicate code uses the PatFrag's captured operands.
542 bool usesOperands() const;
543
Daniel Sanders3f267bf2017-10-15 02:06:44 +0000544 // Is the desired predefined predicate for a load?
545 bool isLoad() const;
546 // Is the desired predefined predicate for a store?
547 bool isStore() const;
Daniel Sanders87d196c2017-11-13 22:26:13 +0000548 // Is the desired predefined predicate for an atomic?
549 bool isAtomic() const;
Daniel Sanders3f267bf2017-10-15 02:06:44 +0000550
551 /// Is this predicate the predefined unindexed load predicate?
552 /// Is this predicate the predefined unindexed store predicate?
553 bool isUnindexed() const;
554 /// Is this predicate the predefined non-extending load predicate?
555 bool isNonExtLoad() const;
556 /// Is this predicate the predefined any-extend load predicate?
557 bool isAnyExtLoad() const;
558 /// Is this predicate the predefined sign-extend load predicate?
559 bool isSignExtLoad() const;
560 /// Is this predicate the predefined zero-extend load predicate?
561 bool isZeroExtLoad() const;
562 /// Is this predicate the predefined non-truncating store predicate?
563 bool isNonTruncStore() const;
564 /// Is this predicate the predefined truncating store predicate?
565 bool isTruncStore() const;
566
Daniel Sanders6d9d30a2017-11-13 23:03:47 +0000567 /// Is this predicate the predefined monotonic atomic predicate?
568 bool isAtomicOrderingMonotonic() const;
569 /// Is this predicate the predefined acquire atomic predicate?
570 bool isAtomicOrderingAcquire() const;
571 /// Is this predicate the predefined release atomic predicate?
572 bool isAtomicOrderingRelease() const;
573 /// Is this predicate the predefined acquire-release atomic predicate?
574 bool isAtomicOrderingAcquireRelease() const;
575 /// Is this predicate the predefined sequentially consistent atomic predicate?
576 bool isAtomicOrderingSequentiallyConsistent() const;
577
Daniel Sanders0c43b3a2017-11-30 21:05:59 +0000578 /// Is this predicate the predefined acquire-or-stronger atomic predicate?
579 bool isAtomicOrderingAcquireOrStronger() const;
580 /// Is this predicate the predefined weaker-than-acquire atomic predicate?
581 bool isAtomicOrderingWeakerThanAcquire() const;
582
583 /// Is this predicate the predefined release-or-stronger atomic predicate?
584 bool isAtomicOrderingReleaseOrStronger() const;
585 /// Is this predicate the predefined weaker-than-release atomic predicate?
586 bool isAtomicOrderingWeakerThanRelease() const;
587
Daniel Sanders3f267bf2017-10-15 02:06:44 +0000588 /// If non-null, indicates that this predicate is a predefined memory VT
589 /// predicate for a load/store and returns the ValueType record for the memory VT.
590 Record *getMemoryVT() const;
591 /// If non-null, indicates that this predicate is a predefined memory VT
592 /// predicate (checking only the scalar type) for load/store and returns the
593 /// ValueType record for the memory VT.
594 Record *getScalarMemoryVT() const;
595
Matt Arsenaultd00d8572019-07-15 20:59:42 +0000596 ListInit *getAddressSpaces() const;
597
Daniel Sanders8ead1292018-06-15 23:13:43 +0000598 // If true, indicates that GlobalISel-based C++ code was supplied.
599 bool hasGISelPredicateCode() const;
600 std::string getGISelPredicateCode() const;
601
Chris Lattner514e2922011-04-17 21:38:24 +0000602private:
Daniel Sandersadbf58d2017-10-15 19:01:32 +0000603 bool hasPredCode() const;
604 bool hasImmCode() const;
Daniel Sanders3f267bf2017-10-15 02:06:44 +0000605 std::string getPredCode() const;
606 std::string getImmCode() const;
Daniel Sanders649c5852017-10-13 20:42:18 +0000607 bool immCodeUsesAPInt() const;
608 bool immCodeUsesAPFloat() const;
Daniel Sanders3f267bf2017-10-15 02:06:44 +0000609
610 bool isPredefinedPredicateEqualTo(StringRef Field, bool Value) const;
Chris Lattner514e2922011-04-17 21:38:24 +0000611};
Simon Pilgrim8c4d0612017-09-22 16:57:28 +0000612
Nicolai Haehnle445b0b62018-11-30 14:15:13 +0000613struct TreePredicateCall {
614 TreePredicateFn Fn;
615
616 // Scope -- unique identifier for retrieving named arguments. 0 is used when
617 // the predicate does not use named arguments.
618 unsigned Scope;
619
620 TreePredicateCall(const TreePredicateFn &Fn, unsigned Scope)
621 : Fn(Fn), Scope(Scope) {}
622
623 bool operator==(const TreePredicateCall &o) const {
624 return Fn == o.Fn && Scope == o.Scope;
625 }
626 bool operator!=(const TreePredicateCall &o) const {
627 return !(*this == o);
628 }
629};
Chris Lattner8cab0212008-01-05 22:25:12 +0000630
Chris Lattner8cab0212008-01-05 22:25:12 +0000631class TreePatternNode {
Chris Lattnerf1447252010-03-19 21:37:09 +0000632 /// The type of each node result. Before and during type inference, each
633 /// result may be a set of possible types. After (successful) type inference,
634 /// each is a single concrete type.
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +0000635 std::vector<TypeSetByHwMode> Types;
Jim Grosbach50986b52010-12-24 05:06:32 +0000636
Craig Topperbd199f82018-12-05 00:47:59 +0000637 /// The index of each result in results of the pattern.
638 std::vector<unsigned> ResultPerm;
639
Chris Lattner8cab0212008-01-05 22:25:12 +0000640 /// Operator - The Record for the operator if this is an interior node (not
641 /// a leaf).
642 Record *Operator;
Jim Grosbach50986b52010-12-24 05:06:32 +0000643
Chris Lattner8cab0212008-01-05 22:25:12 +0000644 /// Val - The init value (e.g. the "GPRC" record, or "7") for a leaf.
645 ///
David Greeneaf8ee2c2011-07-29 22:43:06 +0000646 Init *Val;
Jim Grosbach50986b52010-12-24 05:06:32 +0000647
Chris Lattner8cab0212008-01-05 22:25:12 +0000648 /// Name - The name given to this node with the :$foo notation.
649 ///
650 std::string Name;
Jim Grosbach50986b52010-12-24 05:06:32 +0000651
Nicolai Haehnle445b0b62018-11-30 14:15:13 +0000652 std::vector<ScopedName> NamesAsPredicateArg;
653
654 /// PredicateCalls - The predicate functions to execute on this node to check
Dan Gohman6e979022008-10-15 06:17:21 +0000655 /// for a match. If this list is empty, no predicate is involved.
Nicolai Haehnle445b0b62018-11-30 14:15:13 +0000656 std::vector<TreePredicateCall> PredicateCalls;
Jim Grosbach50986b52010-12-24 05:06:32 +0000657
Chris Lattner8cab0212008-01-05 22:25:12 +0000658 /// TransformFn - The transformation function to execute on this node before
659 /// it can be substituted into the resulting instruction on a pattern match.
660 Record *TransformFn;
Jim Grosbach50986b52010-12-24 05:06:32 +0000661
Florian Hahn75e87c32018-05-30 21:00:18 +0000662 std::vector<TreePatternNodePtr> Children;
663
Chris Lattner8cab0212008-01-05 22:25:12 +0000664public:
Craig Topper26fc06352018-07-15 06:52:49 +0000665 TreePatternNode(Record *Op, std::vector<TreePatternNodePtr> Ch,
Jim Grosbach50986b52010-12-24 05:06:32 +0000666 unsigned NumResults)
Craig Topper26fc06352018-07-15 06:52:49 +0000667 : Operator(Op), Val(nullptr), TransformFn(nullptr),
668 Children(std::move(Ch)) {
Chris Lattnerf1447252010-03-19 21:37:09 +0000669 Types.resize(NumResults);
Craig Topperbd199f82018-12-05 00:47:59 +0000670 ResultPerm.resize(NumResults);
671 std::iota(ResultPerm.begin(), ResultPerm.end(), 0);
Chris Lattnerf1447252010-03-19 21:37:09 +0000672 }
David Greeneaf8ee2c2011-07-29 22:43:06 +0000673 TreePatternNode(Init *val, unsigned NumResults) // leaf ctor
Craig Topperada08572014-04-16 04:21:27 +0000674 : Operator(nullptr), Val(val), TransformFn(nullptr) {
Chris Lattnerf1447252010-03-19 21:37:09 +0000675 Types.resize(NumResults);
Craig Topperbd199f82018-12-05 00:47:59 +0000676 ResultPerm.resize(NumResults);
677 std::iota(ResultPerm.begin(), ResultPerm.end(), 0);
Chris Lattner8cab0212008-01-05 22:25:12 +0000678 }
Jim Grosbach50986b52010-12-24 05:06:32 +0000679
Jakob Stoklund Olesenb5b91102013-03-23 18:08:44 +0000680 bool hasName() const { return !Name.empty(); }
Chris Lattner8cab0212008-01-05 22:25:12 +0000681 const std::string &getName() const { return Name; }
Chris Lattneradf7ecf2010-03-28 06:50:34 +0000682 void setName(StringRef N) { Name.assign(N.begin(), N.end()); }
Jim Grosbach50986b52010-12-24 05:06:32 +0000683
Nicolai Haehnle445b0b62018-11-30 14:15:13 +0000684 const std::vector<ScopedName> &getNamesAsPredicateArg() const {
685 return NamesAsPredicateArg;
686 }
687 void setNamesAsPredicateArg(const std::vector<ScopedName>& Names) {
688 NamesAsPredicateArg = Names;
689 }
690 void addNameAsPredicateArg(const ScopedName &N) {
691 NamesAsPredicateArg.push_back(N);
692 }
693
Craig Topperada08572014-04-16 04:21:27 +0000694 bool isLeaf() const { return Val != nullptr; }
Jim Grosbach50986b52010-12-24 05:06:32 +0000695
Chris Lattnercabe0372010-03-15 06:00:16 +0000696 // Type accessors.
Chris Lattnerf1447252010-03-19 21:37:09 +0000697 unsigned getNumTypes() const { return Types.size(); }
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +0000698 ValueTypeByHwMode getType(unsigned ResNo) const {
699 return Types[ResNo].getValueTypeByHwMode();
Chris Lattnerf1447252010-03-19 21:37:09 +0000700 }
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +0000701 const std::vector<TypeSetByHwMode> &getExtTypes() const { return Types; }
702 const TypeSetByHwMode &getExtType(unsigned ResNo) const {
703 return Types[ResNo];
704 }
705 TypeSetByHwMode &getExtType(unsigned ResNo) { return Types[ResNo]; }
706 void setType(unsigned ResNo, const TypeSetByHwMode &T) { Types[ResNo] = T; }
707 MVT::SimpleValueType getSimpleType(unsigned ResNo) const {
708 return Types[ResNo].getMachineValueType().SimpleTy;
709 }
Jim Grosbach50986b52010-12-24 05:06:32 +0000710
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +0000711 bool hasConcreteType(unsigned ResNo) const {
712 return Types[ResNo].isValueTypeByHwMode(false);
Chris Lattnerf1447252010-03-19 21:37:09 +0000713 }
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +0000714 bool isTypeCompletelyUnknown(unsigned ResNo, TreePattern &TP) const {
715 return Types[ResNo].empty();
Chris Lattnerf1447252010-03-19 21:37:09 +0000716 }
Jim Grosbach50986b52010-12-24 05:06:32 +0000717
Craig Topperbd199f82018-12-05 00:47:59 +0000718 unsigned getNumResults() const { return ResultPerm.size(); }
719 unsigned getResultIndex(unsigned ResNo) const { return ResultPerm[ResNo]; }
720 void setResultIndex(unsigned ResNo, unsigned RI) { ResultPerm[ResNo] = RI; }
721
David Greeneaf8ee2c2011-07-29 22:43:06 +0000722 Init *getLeafValue() const { assert(isLeaf()); return Val; }
Chris Lattner8cab0212008-01-05 22:25:12 +0000723 Record *getOperator() const { assert(!isLeaf()); return Operator; }
Jim Grosbach50986b52010-12-24 05:06:32 +0000724
Chris Lattner8cab0212008-01-05 22:25:12 +0000725 unsigned getNumChildren() const { return Children.size(); }
Florian Hahn6b1db822018-06-14 20:32:58 +0000726 TreePatternNode *getChild(unsigned N) const { return Children[N].get(); }
Florian Hahn75e87c32018-05-30 21:00:18 +0000727 const TreePatternNodePtr &getChildShared(unsigned N) const {
728 return Children[N];
Chris Lattner8cab0212008-01-05 22:25:12 +0000729 }
Florian Hahn75e87c32018-05-30 21:00:18 +0000730 void setChild(unsigned i, TreePatternNodePtr N) { Children[i] = N; }
Jim Grosbach50986b52010-12-24 05:06:32 +0000731
Chris Lattneraa7d3e02010-02-16 06:10:58 +0000732 /// hasChild - Return true if N is any of our children.
733 bool hasChild(const TreePatternNode *N) const {
734 for (unsigned i = 0, e = Children.size(); i != e; ++i)
Florian Hahn75e87c32018-05-30 21:00:18 +0000735 if (Children[i].get() == N)
736 return true;
Chris Lattneraa7d3e02010-02-16 06:10:58 +0000737 return false;
738 }
Chris Lattner89c65662008-01-06 05:36:50 +0000739
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +0000740 bool hasProperTypeByHwMode() const;
741 bool hasPossibleType() const;
742 bool setDefaultMode(unsigned Mode);
743
Nicolai Haehnle445b0b62018-11-30 14:15:13 +0000744 bool hasAnyPredicate() const { return !PredicateCalls.empty(); }
Simon Pilgrim8c4d0612017-09-22 16:57:28 +0000745
Nicolai Haehnle445b0b62018-11-30 14:15:13 +0000746 const std::vector<TreePredicateCall> &getPredicateCalls() const {
747 return PredicateCalls;
Chris Lattner514e2922011-04-17 21:38:24 +0000748 }
Nicolai Haehnle445b0b62018-11-30 14:15:13 +0000749 void clearPredicateCalls() { PredicateCalls.clear(); }
750 void setPredicateCalls(const std::vector<TreePredicateCall> &Calls) {
751 assert(PredicateCalls.empty() && "Overwriting non-empty predicate list!");
752 PredicateCalls = Calls;
Dan Gohman6e979022008-10-15 06:17:21 +0000753 }
Nicolai Haehnle445b0b62018-11-30 14:15:13 +0000754 void addPredicateCall(const TreePredicateCall &Call) {
755 assert(!Call.Fn.isAlwaysTrue() && "Empty predicate string!");
756 assert(!is_contained(PredicateCalls, Call) && "predicate applied recursively");
757 PredicateCalls.push_back(Call);
758 }
759 void addPredicateCall(const TreePredicateFn &Fn, unsigned Scope) {
760 assert((Scope != 0) == Fn.usesOperands());
761 addPredicateCall(TreePredicateCall(Fn, Scope));
Dan Gohman6e979022008-10-15 06:17:21 +0000762 }
Chris Lattner8cab0212008-01-05 22:25:12 +0000763
764 Record *getTransformFn() const { return TransformFn; }
765 void setTransformFn(Record *Fn) { TransformFn = Fn; }
Jim Grosbach50986b52010-12-24 05:06:32 +0000766
Chris Lattner89c65662008-01-06 05:36:50 +0000767 /// getIntrinsicInfo - If this node corresponds to an intrinsic, return the
768 /// CodeGenIntrinsic information for it, otherwise return a null pointer.
769 const CodeGenIntrinsic *getIntrinsicInfo(const CodeGenDAGPatterns &CDP) const;
Evan Cheng49bad4c2008-06-16 20:29:38 +0000770
Chris Lattner53c39ba2010-02-14 22:22:58 +0000771 /// getComplexPatternInfo - If this node corresponds to a ComplexPattern,
772 /// return the ComplexPattern information, otherwise return null.
773 const ComplexPattern *
774 getComplexPatternInfo(const CodeGenDAGPatterns &CGP) const;
775
Tim Northoverc807a172014-05-20 11:52:46 +0000776 /// Returns the number of MachineInstr operands that would be produced by this
777 /// node if it mapped directly to an output Instruction's
778 /// operand. ComplexPattern specifies this explicitly; MIOperandInfo gives it
779 /// for Operands; otherwise 1.
780 unsigned getNumMIResults(const CodeGenDAGPatterns &CGP) const;
781
Chris Lattner53c39ba2010-02-14 22:22:58 +0000782 /// NodeHasProperty - Return true if this node has the specified property.
Chris Lattner450d5042010-02-14 22:33:49 +0000783 bool NodeHasProperty(SDNP Property, const CodeGenDAGPatterns &CGP) const;
Jim Grosbach50986b52010-12-24 05:06:32 +0000784
Chris Lattner53c39ba2010-02-14 22:22:58 +0000785 /// TreeHasProperty - Return true if any node in this tree has the specified
786 /// property.
Chris Lattner450d5042010-02-14 22:33:49 +0000787 bool TreeHasProperty(SDNP Property, const CodeGenDAGPatterns &CGP) const;
Jim Grosbach50986b52010-12-24 05:06:32 +0000788
Evan Cheng49bad4c2008-06-16 20:29:38 +0000789 /// isCommutativeIntrinsic - Return true if the node is an intrinsic which is
790 /// marked isCommutative.
791 bool isCommutativeIntrinsic(const CodeGenDAGPatterns &CDP) const;
Jim Grosbach50986b52010-12-24 05:06:32 +0000792
Daniel Dunbar38a22bf2009-07-03 00:10:29 +0000793 void print(raw_ostream &OS) const;
Chris Lattner8cab0212008-01-05 22:25:12 +0000794 void dump() const;
Jim Grosbach50986b52010-12-24 05:06:32 +0000795
Chris Lattner8cab0212008-01-05 22:25:12 +0000796public: // Higher level manipulation routines.
797
798 /// clone - Return a new copy of this tree.
799 ///
Florian Hahn75e87c32018-05-30 21:00:18 +0000800 TreePatternNodePtr clone() const;
Chris Lattner53c39ba2010-02-14 22:22:58 +0000801
802 /// RemoveAllTypes - Recursively strip all the types of this tree.
803 void RemoveAllTypes();
Jim Grosbach50986b52010-12-24 05:06:32 +0000804
Chris Lattner8cab0212008-01-05 22:25:12 +0000805 /// isIsomorphicTo - Return true if this node is recursively isomorphic to
806 /// the specified node. For this comparison, all of the state of the node
807 /// is considered, except for the assigned name. Nodes with differing names
808 /// that are otherwise identical are considered isomorphic.
Florian Hahn6b1db822018-06-14 20:32:58 +0000809 bool isIsomorphicTo(const TreePatternNode *N,
Scott Michel94420742008-03-05 17:49:05 +0000810 const MultipleUseVarSet &DepVars) const;
Jim Grosbach50986b52010-12-24 05:06:32 +0000811
Chris Lattner8cab0212008-01-05 22:25:12 +0000812 /// SubstituteFormalArguments - Replace the formal arguments in this tree
813 /// with actual values specified by ArgMap.
Florian Hahn75e87c32018-05-30 21:00:18 +0000814 void
815 SubstituteFormalArguments(std::map<std::string, TreePatternNodePtr> &ArgMap);
Chris Lattner8cab0212008-01-05 22:25:12 +0000816
817 /// InlinePatternFragments - If this pattern refers to any pattern
Ulrich Weigandc48aefb2018-07-13 13:18:00 +0000818 /// fragments, return the set of inlined versions (this can be more than
819 /// one if a PatFrags record has multiple alternatives).
820 void InlinePatternFragments(TreePatternNodePtr T,
821 TreePattern &TP,
822 std::vector<TreePatternNodePtr> &OutAlternatives);
Jim Grosbach50986b52010-12-24 05:06:32 +0000823
Bob Wilson1b97f3f2009-01-05 17:23:09 +0000824 /// ApplyTypeConstraints - Apply all of the type constraints relevant to
Chris Lattner8cab0212008-01-05 22:25:12 +0000825 /// this node and its children in the tree. This returns true if it makes a
Joerg Sonnenberger635debe2012-10-25 20:33:17 +0000826 /// change, false otherwise. If a type contradiction is found, flag an error.
Chris Lattner8cab0212008-01-05 22:25:12 +0000827 bool ApplyTypeConstraints(TreePattern &TP, bool NotRegisters);
Jim Grosbach50986b52010-12-24 05:06:32 +0000828
Chris Lattner8cab0212008-01-05 22:25:12 +0000829 /// UpdateNodeType - Set the node type of N to VT if VT contains
Joerg Sonnenberger635debe2012-10-25 20:33:17 +0000830 /// information. If N already contains a conflicting type, then flag an
831 /// error. This returns true if any information was updated.
Chris Lattner8cab0212008-01-05 22:25:12 +0000832 ///
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +0000833 bool UpdateNodeType(unsigned ResNo, const TypeSetByHwMode &InTy,
834 TreePattern &TP);
Chris Lattnerf1447252010-03-19 21:37:09 +0000835 bool UpdateNodeType(unsigned ResNo, MVT::SimpleValueType InTy,
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +0000836 TreePattern &TP);
837 bool UpdateNodeType(unsigned ResNo, ValueTypeByHwMode InTy,
838 TreePattern &TP);
Jim Grosbach50986b52010-12-24 05:06:32 +0000839
Jakob Stoklund Olesen57a86502013-03-18 04:08:07 +0000840 // Update node type with types inferred from an instruction operand or result
841 // def from the ins/outs lists.
842 // Return true if the type changed.
843 bool UpdateNodeTypeFromInst(unsigned ResNo, Record *Operand, TreePattern &TP);
844
Chris Lattner8cab0212008-01-05 22:25:12 +0000845 /// ContainsUnresolvedType - Return true if this tree contains any
846 /// unresolved types.
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +0000847 bool ContainsUnresolvedType(TreePattern &TP) const;
Jim Grosbach50986b52010-12-24 05:06:32 +0000848
Chris Lattner8cab0212008-01-05 22:25:12 +0000849 /// canPatternMatch - If it is impossible for this pattern to match on this
850 /// target, fill in Reason and return false. Otherwise, return true.
Dan Gohmanfc4ad7de2008-04-03 00:02:49 +0000851 bool canPatternMatch(std::string &Reason, const CodeGenDAGPatterns &CDP);
Chris Lattner8cab0212008-01-05 22:25:12 +0000852};
853
Chris Lattnerdd2ec582010-02-14 21:10:33 +0000854inline raw_ostream &operator<<(raw_ostream &OS, const TreePatternNode &TPN) {
855 TPN.print(OS);
856 return OS;
857}
Jim Grosbach50986b52010-12-24 05:06:32 +0000858
Chris Lattner8cab0212008-01-05 22:25:12 +0000859
860/// TreePattern - Represent a pattern, used for instructions, pattern
861/// fragments, etc.
862///
863class TreePattern {
864 /// Trees - The list of pattern trees which corresponds to this pattern.
865 /// Note that PatFrag's only have a single tree.
866 ///
Florian Hahn75e87c32018-05-30 21:00:18 +0000867 std::vector<TreePatternNodePtr> Trees;
Jim Grosbach50986b52010-12-24 05:06:32 +0000868
Chris Lattnercabe0372010-03-15 06:00:16 +0000869 /// NamedNodes - This is all of the nodes that have names in the trees in this
870 /// pattern.
Florian Hahn75e87c32018-05-30 21:00:18 +0000871 StringMap<SmallVector<TreePatternNode *, 1>> NamedNodes;
Jim Grosbach50986b52010-12-24 05:06:32 +0000872
Chris Lattner8cab0212008-01-05 22:25:12 +0000873 /// TheRecord - The actual TableGen record corresponding to this pattern.
874 ///
875 Record *TheRecord;
Jim Grosbach50986b52010-12-24 05:06:32 +0000876
Chris Lattner8cab0212008-01-05 22:25:12 +0000877 /// Args - This is a list of all of the arguments to this pattern (for
878 /// PatFrag patterns), which are the 'node' markers in this pattern.
879 std::vector<std::string> Args;
Jim Grosbach50986b52010-12-24 05:06:32 +0000880
Chris Lattner8cab0212008-01-05 22:25:12 +0000881 /// CDP - the top-level object coordinating this madness.
882 ///
Chris Lattnerab3242f2008-01-06 01:10:31 +0000883 CodeGenDAGPatterns &CDP;
Chris Lattner8cab0212008-01-05 22:25:12 +0000884
885 /// isInputPattern - True if this is an input pattern, something to match.
886 /// False if this is an output pattern, something to emit.
887 bool isInputPattern;
Joerg Sonnenberger635debe2012-10-25 20:33:17 +0000888
889 /// hasError - True if the currently processed nodes have unresolvable types
890 /// or other non-fatal errors
891 bool HasError;
Tim Northoverc807a172014-05-20 11:52:46 +0000892
893 /// It's important that the usage of operands in ComplexPatterns is
894 /// consistent: each named operand can be defined by at most one
895 /// ComplexPattern. This records the ComplexPattern instance and the operand
896 /// number for each operand encountered in a ComplexPattern to aid in that
897 /// check.
898 StringMap<std::pair<Record *, unsigned>> ComplexPatternOperands;
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +0000899
900 TypeInfer Infer;
901
Chris Lattner8cab0212008-01-05 22:25:12 +0000902public:
Jim Grosbach50986b52010-12-24 05:06:32 +0000903
Chris Lattner8cab0212008-01-05 22:25:12 +0000904 /// TreePattern constructor - Parse the specified DagInits into the
905 /// current record.
David Greeneaf8ee2c2011-07-29 22:43:06 +0000906 TreePattern(Record *TheRec, ListInit *RawPat, bool isInput,
Chris Lattnerab3242f2008-01-06 01:10:31 +0000907 CodeGenDAGPatterns &ise);
David Greeneaf8ee2c2011-07-29 22:43:06 +0000908 TreePattern(Record *TheRec, DagInit *Pat, bool isInput,
Chris Lattnerab3242f2008-01-06 01:10:31 +0000909 CodeGenDAGPatterns &ise);
Florian Hahn75e87c32018-05-30 21:00:18 +0000910 TreePattern(Record *TheRec, TreePatternNodePtr Pat, bool isInput,
David Blaikiecf195302014-11-17 22:55:41 +0000911 CodeGenDAGPatterns &ise);
Jim Grosbach50986b52010-12-24 05:06:32 +0000912
Chris Lattner8cab0212008-01-05 22:25:12 +0000913 /// getTrees - Return the tree patterns which corresponds to this pattern.
914 ///
Florian Hahn75e87c32018-05-30 21:00:18 +0000915 const std::vector<TreePatternNodePtr> &getTrees() const { return Trees; }
Chris Lattner8cab0212008-01-05 22:25:12 +0000916 unsigned getNumTrees() const { return Trees.size(); }
Florian Hahn75e87c32018-05-30 21:00:18 +0000917 const TreePatternNodePtr &getTree(unsigned i) const { return Trees[i]; }
Florian Hahn53b14db2018-06-10 21:06:24 +0000918 void setTree(unsigned i, TreePatternNodePtr Tree) { Trees[i] = Tree; }
Florian Hahn4dd569c2018-06-13 20:59:53 +0000919 const TreePatternNodePtr &getOnlyTree() const {
Chris Lattner8cab0212008-01-05 22:25:12 +0000920 assert(Trees.size() == 1 && "Doesn't have exactly one pattern!");
921 return Trees[0];
922 }
Jim Grosbach50986b52010-12-24 05:06:32 +0000923
Florian Hahn75e87c32018-05-30 21:00:18 +0000924 const StringMap<SmallVector<TreePatternNode *, 1>> &getNamedNodesMap() {
Chris Lattnercabe0372010-03-15 06:00:16 +0000925 if (NamedNodes.empty())
926 ComputeNamedNodes();
927 return NamedNodes;
928 }
Jim Grosbach50986b52010-12-24 05:06:32 +0000929
Chris Lattner8cab0212008-01-05 22:25:12 +0000930 /// getRecord - Return the actual TableGen record corresponding to this
931 /// pattern.
932 ///
933 Record *getRecord() const { return TheRecord; }
Jim Grosbach50986b52010-12-24 05:06:32 +0000934
Chris Lattner8cab0212008-01-05 22:25:12 +0000935 unsigned getNumArgs() const { return Args.size(); }
936 const std::string &getArgName(unsigned i) const {
937 assert(i < Args.size() && "Argument reference out of range!");
938 return Args[i];
939 }
940 std::vector<std::string> &getArgList() { return Args; }
Jim Grosbach50986b52010-12-24 05:06:32 +0000941
Chris Lattnerab3242f2008-01-06 01:10:31 +0000942 CodeGenDAGPatterns &getDAGPatterns() const { return CDP; }
Chris Lattner8cab0212008-01-05 22:25:12 +0000943
944 /// InlinePatternFragments - If this pattern refers to any pattern
945 /// fragments, inline them into place, giving us a pattern without any
Ulrich Weigandc48aefb2018-07-13 13:18:00 +0000946 /// PatFrags references. This may increase the number of trees in the
947 /// pattern if a PatFrags has multiple alternatives.
Chris Lattner8cab0212008-01-05 22:25:12 +0000948 void InlinePatternFragments() {
Ulrich Weigandc48aefb2018-07-13 13:18:00 +0000949 std::vector<TreePatternNodePtr> Copy = Trees;
950 Trees.clear();
951 for (unsigned i = 0, e = Copy.size(); i != e; ++i)
952 Copy[i]->InlinePatternFragments(Copy[i], *this, Trees);
Chris Lattner8cab0212008-01-05 22:25:12 +0000953 }
Jim Grosbach50986b52010-12-24 05:06:32 +0000954
Chris Lattner8cab0212008-01-05 22:25:12 +0000955 /// InferAllTypes - Infer/propagate as many types throughout the expression
Jim Grosbach975c1cb2009-03-26 16:17:51 +0000956 /// patterns as possible. Return true if all types are inferred, false
Joerg Sonnenberger635debe2012-10-25 20:33:17 +0000957 /// otherwise. Bail out if a type contradiction is found.
Florian Hahn75e87c32018-05-30 21:00:18 +0000958 bool InferAllTypes(
959 const StringMap<SmallVector<TreePatternNode *, 1>> *NamedTypes = nullptr);
Jim Grosbach50986b52010-12-24 05:06:32 +0000960
Joerg Sonnenberger635debe2012-10-25 20:33:17 +0000961 /// error - If this is the first error in the current resolution step,
962 /// print it and set the error flag. Otherwise, continue silently.
Matt Arsenaultea8df3a2014-11-11 23:48:11 +0000963 void error(const Twine &Msg);
Joerg Sonnenberger635debe2012-10-25 20:33:17 +0000964 bool hasError() const {
965 return HasError;
966 }
967 void resetError() {
968 HasError = false;
969 }
Jim Grosbach50986b52010-12-24 05:06:32 +0000970
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +0000971 TypeInfer &getInfer() { return Infer; }
972
Daniel Dunbar38a22bf2009-07-03 00:10:29 +0000973 void print(raw_ostream &OS) const;
Chris Lattner8cab0212008-01-05 22:25:12 +0000974 void dump() const;
Jim Grosbach50986b52010-12-24 05:06:32 +0000975
Chris Lattner8cab0212008-01-05 22:25:12 +0000976private:
Florian Hahn75e87c32018-05-30 21:00:18 +0000977 TreePatternNodePtr ParseTreePattern(Init *DI, StringRef OpName);
Chris Lattnercabe0372010-03-15 06:00:16 +0000978 void ComputeNamedNodes();
Florian Hahn6b1db822018-06-14 20:32:58 +0000979 void ComputeNamedNodes(TreePatternNode *N);
Chris Lattner8cab0212008-01-05 22:25:12 +0000980};
981
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +0000982
983inline bool TreePatternNode::UpdateNodeType(unsigned ResNo,
984 const TypeSetByHwMode &InTy,
985 TreePattern &TP) {
986 TypeSetByHwMode VTS(InTy);
987 TP.getInfer().expandOverloads(VTS);
988 return TP.getInfer().MergeInTypeInfo(Types[ResNo], VTS);
989}
990
991inline bool TreePatternNode::UpdateNodeType(unsigned ResNo,
992 MVT::SimpleValueType InTy,
993 TreePattern &TP) {
994 TypeSetByHwMode VTS(InTy);
995 TP.getInfer().expandOverloads(VTS);
996 return TP.getInfer().MergeInTypeInfo(Types[ResNo], VTS);
997}
998
999inline bool TreePatternNode::UpdateNodeType(unsigned ResNo,
1000 ValueTypeByHwMode InTy,
1001 TreePattern &TP) {
1002 TypeSetByHwMode VTS(InTy);
1003 TP.getInfer().expandOverloads(VTS);
1004 return TP.getInfer().MergeInTypeInfo(Types[ResNo], VTS);
1005}
1006
1007
Tom Stellardb7246a72012-09-06 14:15:52 +00001008/// DAGDefaultOperand - One of these is created for each OperandWithDefaultOps
1009/// that has a set ExecuteAlways / DefaultOps field.
Chris Lattner8cab0212008-01-05 22:25:12 +00001010struct DAGDefaultOperand {
Florian Hahn75e87c32018-05-30 21:00:18 +00001011 std::vector<TreePatternNodePtr> DefaultOps;
Chris Lattner8cab0212008-01-05 22:25:12 +00001012};
1013
1014class DAGInstruction {
Chris Lattner8cab0212008-01-05 22:25:12 +00001015 std::vector<Record*> Results;
1016 std::vector<Record*> Operands;
1017 std::vector<Record*> ImpResults;
Ulrich Weigandc48aefb2018-07-13 13:18:00 +00001018 TreePatternNodePtr SrcPattern;
Florian Hahn75e87c32018-05-30 21:00:18 +00001019 TreePatternNodePtr ResultPattern;
1020
Chris Lattner8cab0212008-01-05 22:25:12 +00001021public:
Ulrich Weigandc48aefb2018-07-13 13:18:00 +00001022 DAGInstruction(const std::vector<Record*> &results,
Chris Lattner8cab0212008-01-05 22:25:12 +00001023 const std::vector<Record*> &operands,
Ulrich Weigandc48aefb2018-07-13 13:18:00 +00001024 const std::vector<Record*> &impresults,
1025 TreePatternNodePtr srcpattern = nullptr,
1026 TreePatternNodePtr resultpattern = nullptr)
1027 : Results(results), Operands(operands), ImpResults(impresults),
1028 SrcPattern(srcpattern), ResultPattern(resultpattern) {}
Chris Lattner8cab0212008-01-05 22:25:12 +00001029
Chris Lattner8cab0212008-01-05 22:25:12 +00001030 unsigned getNumResults() const { return Results.size(); }
1031 unsigned getNumOperands() const { return Operands.size(); }
1032 unsigned getNumImpResults() const { return ImpResults.size(); }
Chris Lattner8cab0212008-01-05 22:25:12 +00001033 const std::vector<Record*>& getImpResults() const { return ImpResults; }
Jim Grosbach50986b52010-12-24 05:06:32 +00001034
Chris Lattner8cab0212008-01-05 22:25:12 +00001035 Record *getResult(unsigned RN) const {
1036 assert(RN < Results.size());
1037 return Results[RN];
1038 }
Jim Grosbach50986b52010-12-24 05:06:32 +00001039
Chris Lattner8cab0212008-01-05 22:25:12 +00001040 Record *getOperand(unsigned ON) const {
1041 assert(ON < Operands.size());
1042 return Operands[ON];
1043 }
1044
1045 Record *getImpResult(unsigned RN) const {
1046 assert(RN < ImpResults.size());
1047 return ImpResults[RN];
1048 }
Jim Grosbach50986b52010-12-24 05:06:32 +00001049
Ulrich Weigandc48aefb2018-07-13 13:18:00 +00001050 TreePatternNodePtr getSrcPattern() const { return SrcPattern; }
Florian Hahn75e87c32018-05-30 21:00:18 +00001051 TreePatternNodePtr getResultPattern() const { return ResultPattern; }
Chris Lattner8cab0212008-01-05 22:25:12 +00001052};
Jim Grosbach50986b52010-12-24 05:06:32 +00001053
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +00001054/// This class represents a condition that has to be satisfied for a pattern
1055/// to be tried. It is a generalization of a class "Pattern" from Target.td:
1056/// in addition to the Target.td's predicates, this class can also represent
1057/// conditions associated with HW modes. Both types will eventually become
1058/// strings containing C++ code to be executed, the difference is in how
1059/// these strings are generated.
1060class Predicate {
1061public:
1062 Predicate(Record *R, bool C = true) : Def(R), IfCond(C), IsHwMode(false) {
1063 assert(R->isSubClassOf("Predicate") &&
1064 "Predicate objects should only be created for records derived"
1065 "from Predicate class");
1066 }
1067 Predicate(StringRef FS, bool C = true) : Def(nullptr), Features(FS.str()),
1068 IfCond(C), IsHwMode(true) {}
1069
1070 /// Return a string which contains the C++ condition code that will serve
1071 /// as a predicate during instruction selection.
1072 std::string getCondString() const {
1073 // The string will excute in a subclass of SelectionDAGISel.
1074 // Cast to std::string explicitly to avoid ambiguity with StringRef.
1075 std::string C = IsHwMode
1076 ? std::string("MF->getSubtarget().checkFeatures(\"" + Features + "\")")
1077 : std::string(Def->getValueAsString("CondString"));
Matt Arsenault57ef94f2019-07-30 15:56:43 +00001078 if (C.empty())
1079 return "";
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +00001080 return IfCond ? C : "!("+C+')';
1081 }
Matt Arsenault57ef94f2019-07-30 15:56:43 +00001082
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +00001083 bool operator==(const Predicate &P) const {
1084 return IfCond == P.IfCond && IsHwMode == P.IsHwMode && Def == P.Def;
1085 }
1086 bool operator<(const Predicate &P) const {
1087 if (IsHwMode != P.IsHwMode)
1088 return IsHwMode < P.IsHwMode;
1089 assert(!Def == !P.Def && "Inconsistency between Def and IsHwMode");
1090 if (IfCond != P.IfCond)
1091 return IfCond < P.IfCond;
1092 if (Def)
1093 return LessRecord()(Def, P.Def);
1094 return Features < P.Features;
1095 }
1096 Record *Def; ///< Predicate definition from .td file, null for
1097 ///< HW modes.
1098 std::string Features; ///< Feature string for HW mode.
1099 bool IfCond; ///< The boolean value that the condition has to
1100 ///< evaluate to for this predicate to be true.
1101 bool IsHwMode; ///< Does this predicate correspond to a HW mode?
1102};
1103
Chris Lattnerab3242f2008-01-06 01:10:31 +00001104/// PatternToMatch - Used by CodeGenDAGPatterns to keep tab of patterns
Chris Lattner8cab0212008-01-05 22:25:12 +00001105/// processed to produce isel.
Chris Lattner7ed81692010-02-18 06:47:49 +00001106class PatternToMatch {
1107public:
Craig Topperd78567f2018-06-10 23:15:48 +00001108 PatternToMatch(Record *srcrecord, std::vector<Predicate> preds,
Florian Hahn75e87c32018-05-30 21:00:18 +00001109 TreePatternNodePtr src, TreePatternNodePtr dst,
Craig Topperd78567f2018-06-10 23:15:48 +00001110 std::vector<Record *> dstregs, int complexity,
Florian Hahn75e87c32018-05-30 21:00:18 +00001111 unsigned uid, unsigned setmode = 0)
1112 : SrcRecord(srcrecord), SrcPattern(src), DstPattern(dst),
Craig Topper73ed2e62018-07-15 01:10:28 +00001113 Predicates(std::move(preds)), Dstregs(std::move(dstregs)),
Florian Hahn75e87c32018-05-30 21:00:18 +00001114 AddedComplexity(complexity), ID(uid), ForceMode(setmode) {}
Chris Lattner8cab0212008-01-05 22:25:12 +00001115
Jim Grosbachfb116ae2010-12-07 23:05:49 +00001116 Record *SrcRecord; // Originating Record for the pattern.
Florian Hahn75e87c32018-05-30 21:00:18 +00001117 TreePatternNodePtr SrcPattern; // Source pattern to match.
1118 TreePatternNodePtr DstPattern; // Resulting pattern.
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +00001119 std::vector<Predicate> Predicates; // Top level predicate conditions
1120 // to match.
Chris Lattner8cab0212008-01-05 22:25:12 +00001121 std::vector<Record*> Dstregs; // Physical register defs being matched.
Tom Stellard6655dd62014-08-01 00:32:36 +00001122 int AddedComplexity; // Add to matching pattern complexity.
Chris Lattnerd39f75b2010-03-01 22:09:11 +00001123 unsigned ID; // Unique ID for the record.
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +00001124 unsigned ForceMode; // Force this mode in type inference when set.
Chris Lattner8cab0212008-01-05 22:25:12 +00001125
Jim Grosbachfb116ae2010-12-07 23:05:49 +00001126 Record *getSrcRecord() const { return SrcRecord; }
Florian Hahn75e87c32018-05-30 21:00:18 +00001127 TreePatternNode *getSrcPattern() const { return SrcPattern.get(); }
1128 TreePatternNodePtr getSrcPatternShared() const { return SrcPattern; }
1129 TreePatternNode *getDstPattern() const { return DstPattern.get(); }
1130 TreePatternNodePtr getDstPatternShared() const { return DstPattern; }
Chris Lattner8cab0212008-01-05 22:25:12 +00001131 const std::vector<Record*> &getDstRegs() const { return Dstregs; }
Tom Stellard6655dd62014-08-01 00:32:36 +00001132 int getAddedComplexity() const { return AddedComplexity; }
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +00001133 const std::vector<Predicate> &getPredicates() const { return Predicates; }
Dan Gohman49e19e92008-08-22 00:20:26 +00001134
1135 std::string getPredicateCheck() const;
Jim Grosbach50986b52010-12-24 05:06:32 +00001136
Chris Lattner05925fe2010-03-29 01:40:38 +00001137 /// Compute the complexity metric for the input pattern. This roughly
1138 /// corresponds to the number of nodes that are covered.
Tom Stellard6655dd62014-08-01 00:32:36 +00001139 int getPatternComplexity(const CodeGenDAGPatterns &CGP) const;
Chris Lattner8cab0212008-01-05 22:25:12 +00001140};
1141
Chris Lattnerab3242f2008-01-06 01:10:31 +00001142class CodeGenDAGPatterns {
Chris Lattner8cab0212008-01-05 22:25:12 +00001143 RecordKeeper &Records;
1144 CodeGenTarget Target;
Justin Bogner92a8c612016-07-15 16:31:37 +00001145 CodeGenIntrinsicTable Intrinsics;
1146 CodeGenIntrinsicTable TgtIntrinsics;
Jim Grosbach50986b52010-12-24 05:06:32 +00001147
Sean Silvaa4e2c5f2012-09-19 01:47:00 +00001148 std::map<Record*, SDNodeInfo, LessRecordByID> SDNodes;
Krzysztof Parzyszek426bf362017-09-12 15:31:26 +00001149 std::map<Record*, std::pair<Record*, std::string>, LessRecordByID>
1150 SDNodeXForms;
Sean Silvaa4e2c5f2012-09-19 01:47:00 +00001151 std::map<Record*, ComplexPattern, LessRecordByID> ComplexPatterns;
David Blaikie3c6ca232014-11-13 21:40:02 +00001152 std::map<Record *, std::unique_ptr<TreePattern>, LessRecordByID>
1153 PatternFragments;
Sean Silvaa4e2c5f2012-09-19 01:47:00 +00001154 std::map<Record*, DAGDefaultOperand, LessRecordByID> DefaultOperands;
1155 std::map<Record*, DAGInstruction, LessRecordByID> Instructions;
Jim Grosbach50986b52010-12-24 05:06:32 +00001156
Chris Lattner8cab0212008-01-05 22:25:12 +00001157 // Specific SDNode definitions:
1158 Record *intrinsic_void_sdnode;
1159 Record *intrinsic_w_chain_sdnode, *intrinsic_wo_chain_sdnode;
Jim Grosbach50986b52010-12-24 05:06:32 +00001160
Chris Lattner8cab0212008-01-05 22:25:12 +00001161 /// PatternsToMatch - All of the things we are matching on the DAG. The first
1162 /// value is the pattern to match, the second pattern is the result to
1163 /// emit.
1164 std::vector<PatternToMatch> PatternsToMatch;
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +00001165
1166 TypeSetByHwMode LegalVTS;
1167
Daniel Sanders7e523672017-11-11 03:23:44 +00001168 using PatternRewriterFn = std::function<void (TreePattern *)>;
1169 PatternRewriterFn PatternRewriter;
1170
Nicolai Haehnle445b0b62018-11-30 14:15:13 +00001171 unsigned NumScopes = 0;
1172
Chris Lattner8cab0212008-01-05 22:25:12 +00001173public:
Daniel Sanders7e523672017-11-11 03:23:44 +00001174 CodeGenDAGPatterns(RecordKeeper &R,
1175 PatternRewriterFn PatternRewriter = nullptr);
Jim Grosbach50986b52010-12-24 05:06:32 +00001176
Dan Gohmanfc4ad7de2008-04-03 00:02:49 +00001177 CodeGenTarget &getTargetInfo() { return Target; }
Chris Lattner8cab0212008-01-05 22:25:12 +00001178 const CodeGenTarget &getTargetInfo() const { return Target; }
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +00001179 const TypeSetByHwMode &getLegalTypes() const { return LegalVTS; }
Jim Grosbach50986b52010-12-24 05:06:32 +00001180
Daniel Sanders9e0ae7b2017-10-13 19:00:01 +00001181 Record *getSDNodeNamed(const std::string &Name) const;
Jim Grosbach50986b52010-12-24 05:06:32 +00001182
Chris Lattner8cab0212008-01-05 22:25:12 +00001183 const SDNodeInfo &getSDNodeInfo(Record *R) const {
Simon Pilgrimb021b132017-10-07 14:34:24 +00001184 auto F = SDNodes.find(R);
1185 assert(F != SDNodes.end() && "Unknown node!");
1186 return F->second;
Chris Lattner8cab0212008-01-05 22:25:12 +00001187 }
Jim Grosbach50986b52010-12-24 05:06:32 +00001188
Chris Lattnercc43e792008-01-05 22:54:53 +00001189 // Node transformation lookups.
1190 typedef std::pair<Record*, std::string> NodeXForm;
1191 const NodeXForm &getSDNodeTransform(Record *R) const {
Simon Pilgrimb021b132017-10-07 14:34:24 +00001192 auto F = SDNodeXForms.find(R);
1193 assert(F != SDNodeXForms.end() && "Invalid transform!");
1194 return F->second;
Chris Lattner8cab0212008-01-05 22:25:12 +00001195 }
Jim Grosbach50986b52010-12-24 05:06:32 +00001196
Sean Silvaa4e2c5f2012-09-19 01:47:00 +00001197 typedef std::map<Record*, NodeXForm, LessRecordByID>::const_iterator
Benjamin Kramerc2dbd5d2009-08-23 10:39:21 +00001198 nx_iterator;
Chris Lattnercc43e792008-01-05 22:54:53 +00001199 nx_iterator nx_begin() const { return SDNodeXForms.begin(); }
1200 nx_iterator nx_end() const { return SDNodeXForms.end(); }
1201
Jim Grosbach50986b52010-12-24 05:06:32 +00001202
Chris Lattner8cab0212008-01-05 22:25:12 +00001203 const ComplexPattern &getComplexPattern(Record *R) const {
Simon Pilgrimb021b132017-10-07 14:34:24 +00001204 auto F = ComplexPatterns.find(R);
1205 assert(F != ComplexPatterns.end() && "Unknown addressing mode!");
1206 return F->second;
Chris Lattner8cab0212008-01-05 22:25:12 +00001207 }
Jim Grosbach50986b52010-12-24 05:06:32 +00001208
Chris Lattner8cab0212008-01-05 22:25:12 +00001209 const CodeGenIntrinsic &getIntrinsic(Record *R) const {
1210 for (unsigned i = 0, e = Intrinsics.size(); i != e; ++i)
1211 if (Intrinsics[i].TheDef == R) return Intrinsics[i];
Dale Johannesenb842d522009-02-05 01:49:45 +00001212 for (unsigned i = 0, e = TgtIntrinsics.size(); i != e; ++i)
1213 if (TgtIntrinsics[i].TheDef == R) return TgtIntrinsics[i];
Craig Topperc4965bc2012-02-05 07:21:30 +00001214 llvm_unreachable("Unknown intrinsic!");
Chris Lattner8cab0212008-01-05 22:25:12 +00001215 }
Jim Grosbach50986b52010-12-24 05:06:32 +00001216
Chris Lattner8cab0212008-01-05 22:25:12 +00001217 const CodeGenIntrinsic &getIntrinsicInfo(unsigned IID) const {
Dale Johannesenb842d522009-02-05 01:49:45 +00001218 if (IID-1 < Intrinsics.size())
1219 return Intrinsics[IID-1];
1220 if (IID-Intrinsics.size()-1 < TgtIntrinsics.size())
1221 return TgtIntrinsics[IID-Intrinsics.size()-1];
Craig Topperc4965bc2012-02-05 07:21:30 +00001222 llvm_unreachable("Bad intrinsic ID!");
Chris Lattner8cab0212008-01-05 22:25:12 +00001223 }
Jim Grosbach50986b52010-12-24 05:06:32 +00001224
Chris Lattner8cab0212008-01-05 22:25:12 +00001225 unsigned getIntrinsicID(Record *R) const {
1226 for (unsigned i = 0, e = Intrinsics.size(); i != e; ++i)
1227 if (Intrinsics[i].TheDef == R) return i;
Dale Johannesenb842d522009-02-05 01:49:45 +00001228 for (unsigned i = 0, e = TgtIntrinsics.size(); i != e; ++i)
1229 if (TgtIntrinsics[i].TheDef == R) return i + Intrinsics.size();
Craig Topperc4965bc2012-02-05 07:21:30 +00001230 llvm_unreachable("Unknown intrinsic!");
Chris Lattner8cab0212008-01-05 22:25:12 +00001231 }
Jim Grosbach50986b52010-12-24 05:06:32 +00001232
Chris Lattner7ed81692010-02-18 06:47:49 +00001233 const DAGDefaultOperand &getDefaultOperand(Record *R) const {
Simon Pilgrimb021b132017-10-07 14:34:24 +00001234 auto F = DefaultOperands.find(R);
1235 assert(F != DefaultOperands.end() &&"Isn't an analyzed default operand!");
1236 return F->second;
Chris Lattner8cab0212008-01-05 22:25:12 +00001237 }
Jim Grosbach50986b52010-12-24 05:06:32 +00001238
Chris Lattner8cab0212008-01-05 22:25:12 +00001239 // Pattern Fragment information.
1240 TreePattern *getPatternFragment(Record *R) const {
Simon Pilgrimb021b132017-10-07 14:34:24 +00001241 auto F = PatternFragments.find(R);
1242 assert(F != PatternFragments.end() && "Invalid pattern fragment request!");
1243 return F->second.get();
Chris Lattner8cab0212008-01-05 22:25:12 +00001244 }
Chris Lattnerf1447252010-03-19 21:37:09 +00001245 TreePattern *getPatternFragmentIfRead(Record *R) const {
Simon Pilgrimb021b132017-10-07 14:34:24 +00001246 auto F = PatternFragments.find(R);
1247 if (F == PatternFragments.end())
David Blaikie3c6ca232014-11-13 21:40:02 +00001248 return nullptr;
Simon Pilgrimb021b132017-10-07 14:34:24 +00001249 return F->second.get();
Chris Lattnerf1447252010-03-19 21:37:09 +00001250 }
Jim Grosbach50986b52010-12-24 05:06:32 +00001251
David Blaikiefcacc742014-11-13 21:56:57 +00001252 typedef std::map<Record *, std::unique_ptr<TreePattern>,
1253 LessRecordByID>::const_iterator pf_iterator;
Chris Lattner8cab0212008-01-05 22:25:12 +00001254 pf_iterator pf_begin() const { return PatternFragments.begin(); }
1255 pf_iterator pf_end() const { return PatternFragments.end(); }
Ahmed Bougacha36f70352016-12-21 23:26:20 +00001256 iterator_range<pf_iterator> ptfs() const { return PatternFragments; }
Chris Lattner8cab0212008-01-05 22:25:12 +00001257
1258 // Patterns to match information.
Chris Lattner9abe77b2008-01-05 22:30:17 +00001259 typedef std::vector<PatternToMatch>::const_iterator ptm_iterator;
1260 ptm_iterator ptm_begin() const { return PatternsToMatch.begin(); }
1261 ptm_iterator ptm_end() const { return PatternsToMatch.end(); }
Ahmed Bougacha36f70352016-12-21 23:26:20 +00001262 iterator_range<ptm_iterator> ptms() const { return PatternsToMatch; }
Jim Grosbach50986b52010-12-24 05:06:32 +00001263
Ahmed Bougacha14107512013-10-28 18:07:21 +00001264 /// Parse the Pattern for an instruction, and insert the result in DAGInsts.
1265 typedef std::map<Record*, DAGInstruction, LessRecordByID> DAGInstMap;
Ulrich Weigandc48aefb2018-07-13 13:18:00 +00001266 void parseInstructionPattern(
Ahmed Bougacha14107512013-10-28 18:07:21 +00001267 CodeGenInstruction &CGI, ListInit *Pattern,
1268 DAGInstMap &DAGInsts);
Jim Grosbach50986b52010-12-24 05:06:32 +00001269
Chris Lattner8cab0212008-01-05 22:25:12 +00001270 const DAGInstruction &getInstruction(Record *R) const {
Simon Pilgrimb021b132017-10-07 14:34:24 +00001271 auto F = Instructions.find(R);
1272 assert(F != Instructions.end() && "Unknown instruction!");
1273 return F->second;
Chris Lattner8cab0212008-01-05 22:25:12 +00001274 }
Jim Grosbach50986b52010-12-24 05:06:32 +00001275
Chris Lattner8cab0212008-01-05 22:25:12 +00001276 Record *get_intrinsic_void_sdnode() const {
1277 return intrinsic_void_sdnode;
1278 }
1279 Record *get_intrinsic_w_chain_sdnode() const {
1280 return intrinsic_w_chain_sdnode;
1281 }
1282 Record *get_intrinsic_wo_chain_sdnode() const {
1283 return intrinsic_wo_chain_sdnode;
1284 }
Jim Grosbach50986b52010-12-24 05:06:32 +00001285
Jakob Stoklund Olesene4197252009-10-15 18:50:03 +00001286 bool hasTargetIntrinsics() { return !TgtIntrinsics.empty(); }
1287
Nicolai Haehnle445b0b62018-11-30 14:15:13 +00001288 unsigned allocateScope() { return ++NumScopes; }
1289
Simon Tathamc74322a2019-07-04 08:43:20 +00001290 bool operandHasDefault(Record *Op) const {
1291 return Op->isSubClassOf("OperandWithDefaultOps") &&
1292 !getDefaultOperand(Op).DefaultOps.empty();
1293 }
1294
Chris Lattner8cab0212008-01-05 22:25:12 +00001295private:
1296 void ParseNodeInfo();
Chris Lattnercc43e792008-01-05 22:54:53 +00001297 void ParseNodeTransforms();
Chris Lattner8cab0212008-01-05 22:25:12 +00001298 void ParseComplexPatterns();
Hal Finkel2756dc12014-02-28 00:26:56 +00001299 void ParsePatternFragments(bool OutFrags = false);
Chris Lattner8cab0212008-01-05 22:25:12 +00001300 void ParseDefaultOperands();
1301 void ParseInstructions();
1302 void ParsePatterns();
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +00001303 void ExpandHwModeBasedTypes();
Dan Gohmanfc4ad7de2008-04-03 00:02:49 +00001304 void InferInstructionFlags();
Chris Lattner8cab0212008-01-05 22:25:12 +00001305 void GenerateVariants();
Jakob Stoklund Olesena9d322a2012-08-28 03:26:49 +00001306 void VerifyInstructionFlags();
Jim Grosbach50986b52010-12-24 05:06:32 +00001307
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +00001308 std::vector<Predicate> makePredList(ListInit *L);
1309
Ulrich Weigandc48aefb2018-07-13 13:18:00 +00001310 void ParseOnePattern(Record *TheDef,
1311 TreePattern &Pattern, TreePattern &Result,
1312 const std::vector<Record *> &InstImpResults);
Craig Topper18e6b572017-06-25 17:33:49 +00001313 void AddPatternToMatch(TreePattern *Pattern, PatternToMatch &&PTM);
Florian Hahn75e87c32018-05-30 21:00:18 +00001314 void FindPatternInputsAndOutputs(
Florian Hahn6b1db822018-06-14 20:32:58 +00001315 TreePattern &I, TreePatternNodePtr Pat,
Florian Hahn75e87c32018-05-30 21:00:18 +00001316 std::map<std::string, TreePatternNodePtr> &InstInputs,
Craig Topperbd199f82018-12-05 00:47:59 +00001317 MapVector<std::string, TreePatternNodePtr,
1318 std::map<std::string, unsigned>> &InstResults,
Florian Hahn75e87c32018-05-30 21:00:18 +00001319 std::vector<Record *> &InstImpResults);
Chris Lattner8cab0212008-01-05 22:25:12 +00001320};
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +00001321
1322
Florian Hahn6b1db822018-06-14 20:32:58 +00001323inline bool SDNodeInfo::ApplyTypeConstraints(TreePatternNode *N,
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +00001324 TreePattern &TP) const {
1325 bool MadeChange = false;
1326 for (unsigned i = 0, e = TypeConstraints.size(); i != e; ++i)
1327 MadeChange |= TypeConstraints[i].ApplyTypeConstraint(N, *this, TP);
1328 return MadeChange;
1329 }
Matt Arsenault303327d2017-12-20 19:36:28 +00001330
Chris Lattner8cab0212008-01-05 22:25:12 +00001331} // end namespace llvm
1332
1333#endif