blob: 062fe59d82032750d1c672e50bc2c0d62fd2fdd1 [file] [log] [blame]
Ahmed Bougacha36f70352016-12-21 23:26:20 +00001//===- GlobalISelEmitter.cpp - Generate an instruction selector -----------===//
2//
3// The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9//
10/// \file
11/// This tablegen backend emits code for use by the GlobalISel instruction
12/// selector. See include/llvm/CodeGen/TargetGlobalISel.td.
13///
14/// This file analyzes the patterns recognized by the SelectionDAGISel tablegen
15/// backend, filters out the ones that are unsupported, maps
16/// SelectionDAG-specific constructs to their GlobalISel counterpart
17/// (when applicable: MVT to LLT; SDNode to generic Instruction).
18///
19/// Not all patterns are supported: pass the tablegen invocation
20/// "-warn-on-skipped-patterns" to emit a warning when a pattern is skipped,
21/// as well as why.
22///
23/// The generated file defines a single method:
24/// bool <Target>InstructionSelector::selectImpl(MachineInstr &I) const;
25/// intended to be used in InstructionSelector::select as the first-step
26/// selector for the patterns that don't require complex C++.
27///
28/// FIXME: We'll probably want to eventually define a base
29/// "TargetGenInstructionSelector" class.
30///
31//===----------------------------------------------------------------------===//
32
33#include "CodeGenDAGPatterns.h"
Daniel Sanderse7b0d662017-04-21 15:59:56 +000034#include "SubtargetFeatureInfo.h"
Ahmed Bougacha36f70352016-12-21 23:26:20 +000035#include "llvm/ADT/Optional.h"
Daniel Sanders0ed28822017-04-12 08:23:08 +000036#include "llvm/ADT/SmallSet.h"
Ahmed Bougacha36f70352016-12-21 23:26:20 +000037#include "llvm/ADT/Statistic.h"
38#include "llvm/CodeGen/MachineValueType.h"
Daniel Sandersf76f3152017-11-16 00:46:35 +000039#include "llvm/Support/CodeGenCoverage.h"
Ahmed Bougacha36f70352016-12-21 23:26:20 +000040#include "llvm/Support/CommandLine.h"
Ahmed Bougacha982c5eb2017-02-10 04:00:17 +000041#include "llvm/Support/Error.h"
Daniel Sanders52b4ce72017-03-07 23:20:35 +000042#include "llvm/Support/LowLevelTypeImpl.h"
Pavel Labath52a82e22017-02-21 09:19:41 +000043#include "llvm/Support/ScopedPrinter.h"
Ahmed Bougacha36f70352016-12-21 23:26:20 +000044#include "llvm/TableGen/Error.h"
45#include "llvm/TableGen/Record.h"
46#include "llvm/TableGen/TableGenBackend.h"
Daniel Sanders8a4bae92017-03-14 21:32:08 +000047#include <numeric>
Daniel Sandersf76f3152017-11-16 00:46:35 +000048#include <string>
Ahmed Bougacha36f70352016-12-21 23:26:20 +000049using namespace llvm;
50
51#define DEBUG_TYPE "gisel-emitter"
52
53STATISTIC(NumPatternTotal, "Total number of patterns");
Daniel Sandersb41ce2b2017-02-20 14:31:27 +000054STATISTIC(NumPatternImported, "Number of patterns imported from SelectionDAG");
55STATISTIC(NumPatternImportsSkipped, "Number of SelectionDAG imports skipped");
Daniel Sandersf76f3152017-11-16 00:46:35 +000056STATISTIC(NumPatternsTested, "Number of patterns executed according to coverage information");
Ahmed Bougacha36f70352016-12-21 23:26:20 +000057STATISTIC(NumPatternEmitted, "Number of patterns emitted");
58
Daniel Sanders0848b232017-03-27 13:15:13 +000059cl::OptionCategory GlobalISelEmitterCat("Options for -gen-global-isel");
60
Ahmed Bougacha36f70352016-12-21 23:26:20 +000061static cl::opt<bool> WarnOnSkippedPatterns(
62 "warn-on-skipped-patterns",
63 cl::desc("Explain why a pattern was skipped for inclusion "
64 "in the GlobalISel selector"),
Daniel Sanders0848b232017-03-27 13:15:13 +000065 cl::init(false), cl::cat(GlobalISelEmitterCat));
Ahmed Bougacha36f70352016-12-21 23:26:20 +000066
Daniel Sandersf76f3152017-11-16 00:46:35 +000067static cl::opt<bool> GenerateCoverage(
68 "instrument-gisel-coverage",
69 cl::desc("Generate coverage instrumentation for GlobalISel"),
70 cl::init(false), cl::cat(GlobalISelEmitterCat));
71
72static cl::opt<std::string> UseCoverageFile(
73 "gisel-coverage-file", cl::init(""),
74 cl::desc("Specify file to retrieve coverage information from"),
75 cl::cat(GlobalISelEmitterCat));
76
Daniel Sandersbdfebb82017-03-15 20:18:38 +000077namespace {
Ahmed Bougacha36f70352016-12-21 23:26:20 +000078//===- Helper functions ---------------------------------------------------===//
79
Daniel Sanders11300ce2017-10-13 21:28:03 +000080
81/// Get the name of the enum value used to number the predicate function.
82std::string getEnumNameForPredicate(const TreePredicateFn &Predicate) {
Simon Pilgrim6ecae9f2017-10-14 21:27:53 +000083 return "GIPFP_" + Predicate.getImmTypeIdentifier().str() + "_" +
Daniel Sanders11300ce2017-10-13 21:28:03 +000084 Predicate.getFnName();
85}
86
87/// Get the opcode used to check this predicate.
88std::string getMatchOpcodeForPredicate(const TreePredicateFn &Predicate) {
Simon Pilgrim6ecae9f2017-10-14 21:27:53 +000089 return "GIM_Check" + Predicate.getImmTypeIdentifier().str() + "ImmPredicate";
Daniel Sanders11300ce2017-10-13 21:28:03 +000090}
91
Daniel Sanders52b4ce72017-03-07 23:20:35 +000092/// This class stands in for LLT wherever we want to tablegen-erate an
93/// equivalent at compiler run-time.
94class LLTCodeGen {
95private:
96 LLT Ty;
97
98public:
99 LLTCodeGen(const LLT &Ty) : Ty(Ty) {}
100
Daniel Sanders7aac7cc2017-07-20 09:25:44 +0000101 std::string getCxxEnumValue() const {
102 std::string Str;
103 raw_string_ostream OS(Str);
104
105 emitCxxEnumValue(OS);
106 return OS.str();
107 }
108
Daniel Sanders6ab0daa2017-07-04 14:35:06 +0000109 void emitCxxEnumValue(raw_ostream &OS) const {
110 if (Ty.isScalar()) {
111 OS << "GILLT_s" << Ty.getSizeInBits();
112 return;
113 }
114 if (Ty.isVector()) {
115 OS << "GILLT_v" << Ty.getNumElements() << "s" << Ty.getScalarSizeInBits();
116 return;
117 }
Daniel Sandersa71f4542017-10-16 00:56:30 +0000118 if (Ty.isPointer()) {
119 OS << "GILLT_p" << Ty.getAddressSpace();
120 if (Ty.getSizeInBits() > 0)
121 OS << "s" << Ty.getSizeInBits();
122 return;
123 }
Daniel Sanders6ab0daa2017-07-04 14:35:06 +0000124 llvm_unreachable("Unhandled LLT");
125 }
126
Daniel Sanders52b4ce72017-03-07 23:20:35 +0000127 void emitCxxConstructorCall(raw_ostream &OS) const {
128 if (Ty.isScalar()) {
129 OS << "LLT::scalar(" << Ty.getSizeInBits() << ")";
130 return;
131 }
132 if (Ty.isVector()) {
Daniel Sanders32291982017-06-28 13:50:04 +0000133 OS << "LLT::vector(" << Ty.getNumElements() << ", "
134 << Ty.getScalarSizeInBits() << ")";
Daniel Sanders52b4ce72017-03-07 23:20:35 +0000135 return;
136 }
Daniel Sandersa71f4542017-10-16 00:56:30 +0000137 if (Ty.isPointer() && Ty.getSizeInBits() > 0) {
138 OS << "LLT::pointer(" << Ty.getAddressSpace() << ", "
139 << Ty.getSizeInBits() << ")";
140 return;
141 }
Daniel Sanders52b4ce72017-03-07 23:20:35 +0000142 llvm_unreachable("Unhandled LLT");
143 }
Daniel Sanders8a4bae92017-03-14 21:32:08 +0000144
145 const LLT &get() const { return Ty; }
Daniel Sanders6ab0daa2017-07-04 14:35:06 +0000146
147 /// This ordering is used for std::unique() and std::sort(). There's no
Daniel Sanders032e7f22017-08-17 13:18:35 +0000148 /// particular logic behind the order but either A < B or B < A must be
149 /// true if A != B.
Daniel Sanders6ab0daa2017-07-04 14:35:06 +0000150 bool operator<(const LLTCodeGen &Other) const {
Daniel Sanders032e7f22017-08-17 13:18:35 +0000151 if (Ty.isValid() != Other.Ty.isValid())
152 return Ty.isValid() < Other.Ty.isValid();
Daniel Sanders6ab0daa2017-07-04 14:35:06 +0000153 if (!Ty.isValid())
Daniel Sanders6ab0daa2017-07-04 14:35:06 +0000154 return false;
Daniel Sanders032e7f22017-08-17 13:18:35 +0000155
156 if (Ty.isVector() != Other.Ty.isVector())
157 return Ty.isVector() < Other.Ty.isVector();
158 if (Ty.isScalar() != Other.Ty.isScalar())
159 return Ty.isScalar() < Other.Ty.isScalar();
160 if (Ty.isPointer() != Other.Ty.isPointer())
161 return Ty.isPointer() < Other.Ty.isPointer();
162
163 if (Ty.isPointer() && Ty.getAddressSpace() != Other.Ty.getAddressSpace())
164 return Ty.getAddressSpace() < Other.Ty.getAddressSpace();
165
166 if (Ty.isVector() && Ty.getNumElements() != Other.Ty.getNumElements())
167 return Ty.getNumElements() < Other.Ty.getNumElements();
168
169 return Ty.getSizeInBits() < Other.Ty.getSizeInBits();
Daniel Sanders6ab0daa2017-07-04 14:35:06 +0000170 }
Daniel Sanders8a4bae92017-03-14 21:32:08 +0000171};
172
173class InstructionMatcher;
Ahmed Bougacha36f70352016-12-21 23:26:20 +0000174/// Convert an MVT to an equivalent LLT if possible, or the invalid LLT() for
175/// MVTs that don't map cleanly to an LLT (e.g., iPTR, *any, ...).
Daniel Sanders52b4ce72017-03-07 23:20:35 +0000176static Optional<LLTCodeGen> MVTToLLT(MVT::SimpleValueType SVT) {
Ahmed Bougacha36f70352016-12-21 23:26:20 +0000177 MVT VT(SVT);
Daniel Sandersa71f4542017-10-16 00:56:30 +0000178
Daniel Sanders52b4ce72017-03-07 23:20:35 +0000179 if (VT.isVector() && VT.getVectorNumElements() != 1)
Daniel Sanders32291982017-06-28 13:50:04 +0000180 return LLTCodeGen(
181 LLT::vector(VT.getVectorNumElements(), VT.getScalarSizeInBits()));
Daniel Sandersa71f4542017-10-16 00:56:30 +0000182
Daniel Sanders52b4ce72017-03-07 23:20:35 +0000183 if (VT.isInteger() || VT.isFloatingPoint())
184 return LLTCodeGen(LLT::scalar(VT.getSizeInBits()));
185 return None;
Ahmed Bougacha36f70352016-12-21 23:26:20 +0000186}
187
Daniel Sandersd0656a32017-04-13 09:45:37 +0000188static std::string explainPredicates(const TreePatternNode *N) {
189 std::string Explanation = "";
190 StringRef Separator = "";
191 for (const auto &P : N->getPredicateFns()) {
192 Explanation +=
193 (Separator + P.getOrigPatFragRecord()->getRecord()->getName()).str();
Daniel Sanders76664652017-11-28 22:07:05 +0000194 Separator = ", ";
195
Daniel Sandersd0656a32017-04-13 09:45:37 +0000196 if (P.isAlwaysTrue())
197 Explanation += " always-true";
198 if (P.isImmediatePattern())
199 Explanation += " immediate";
Daniel Sanders3f267bf2017-10-15 02:06:44 +0000200
201 if (P.isUnindexed())
202 Explanation += " unindexed";
203
204 if (P.isNonExtLoad())
205 Explanation += " non-extload";
206 if (P.isAnyExtLoad())
207 Explanation += " extload";
208 if (P.isSignExtLoad())
209 Explanation += " sextload";
210 if (P.isZeroExtLoad())
211 Explanation += " zextload";
212
213 if (P.isNonTruncStore())
214 Explanation += " non-truncstore";
215 if (P.isTruncStore())
216 Explanation += " truncstore";
217
218 if (Record *VT = P.getMemoryVT())
219 Explanation += (" MemVT=" + VT->getName()).str();
220 if (Record *VT = P.getScalarMemoryVT())
221 Explanation += (" ScalarVT(MemVT)=" + VT->getName()).str();
Daniel Sanders76664652017-11-28 22:07:05 +0000222
223 if (P.isAtomicOrderingMonotonic())
224 Explanation += " monotonic";
225 if (P.isAtomicOrderingAcquire())
226 Explanation += " acquire";
227 if (P.isAtomicOrderingRelease())
228 Explanation += " release";
229 if (P.isAtomicOrderingAcquireRelease())
230 Explanation += " acq_rel";
231 if (P.isAtomicOrderingSequentiallyConsistent())
232 Explanation += " seq_cst";
Daniel Sanders0c43b3a2017-11-30 21:05:59 +0000233 if (P.isAtomicOrderingAcquireOrStronger())
234 Explanation += " >=acquire";
235 if (P.isAtomicOrderingWeakerThanAcquire())
236 Explanation += " <acquire";
237 if (P.isAtomicOrderingReleaseOrStronger())
238 Explanation += " >=release";
239 if (P.isAtomicOrderingWeakerThanRelease())
240 Explanation += " <release";
Daniel Sandersd0656a32017-04-13 09:45:37 +0000241 }
242 return Explanation;
243}
244
Daniel Sandersd0656a32017-04-13 09:45:37 +0000245std::string explainOperator(Record *Operator) {
246 if (Operator->isSubClassOf("SDNode"))
Craig Topper2b8419a2017-05-31 19:01:11 +0000247 return (" (" + Operator->getValueAsString("Opcode") + ")").str();
Daniel Sandersd0656a32017-04-13 09:45:37 +0000248
249 if (Operator->isSubClassOf("Intrinsic"))
250 return (" (Operator is an Intrinsic, " + Operator->getName() + ")").str();
251
Daniel Sandersdf39cba2017-10-15 18:22:54 +0000252 if (Operator->isSubClassOf("ComplexPattern"))
253 return (" (Operator is an unmapped ComplexPattern, " + Operator->getName() +
254 ")")
255 .str();
256
257 return (" (Operator " + Operator->getName() + " not understood)").str();
Daniel Sandersd0656a32017-04-13 09:45:37 +0000258}
259
260/// Helper function to let the emitter report skip reason error messages.
261static Error failedImport(const Twine &Reason) {
262 return make_error<StringError>(Reason, inconvertibleErrorCode());
263}
264
265static Error isTrivialOperatorNode(const TreePatternNode *N) {
266 std::string Explanation = "";
267 std::string Separator = "";
Daniel Sanders2c269f62017-08-24 09:11:20 +0000268
269 bool HasUnsupportedPredicate = false;
270 for (const auto &Predicate : N->getPredicateFns()) {
271 if (Predicate.isAlwaysTrue())
272 continue;
273
274 if (Predicate.isImmediatePattern())
275 continue;
276
Daniel Sandersa71f4542017-10-16 00:56:30 +0000277 if (Predicate.isNonExtLoad())
278 continue;
Daniel Sandersd66e0902017-10-23 18:19:24 +0000279
Daniel Sanders76664652017-11-28 22:07:05 +0000280 if (Predicate.isNonTruncStore())
Daniel Sandersd66e0902017-10-23 18:19:24 +0000281 continue;
282
Daniel Sanders76664652017-11-28 22:07:05 +0000283 if (Predicate.isLoad() || Predicate.isStore()) {
284 if (Predicate.isUnindexed())
285 continue;
286 }
287
288 if (Predicate.isAtomic() && Predicate.getMemoryVT())
289 continue;
290
291 if (Predicate.isAtomic() &&
292 (Predicate.isAtomicOrderingMonotonic() ||
293 Predicate.isAtomicOrderingAcquire() ||
294 Predicate.isAtomicOrderingRelease() ||
295 Predicate.isAtomicOrderingAcquireRelease() ||
Daniel Sanders0c43b3a2017-11-30 21:05:59 +0000296 Predicate.isAtomicOrderingSequentiallyConsistent() ||
297 Predicate.isAtomicOrderingAcquireOrStronger() ||
298 Predicate.isAtomicOrderingWeakerThanAcquire() ||
299 Predicate.isAtomicOrderingReleaseOrStronger() ||
300 Predicate.isAtomicOrderingWeakerThanRelease()))
Daniel Sandersd66e0902017-10-23 18:19:24 +0000301 continue;
302
Daniel Sanders2c269f62017-08-24 09:11:20 +0000303 HasUnsupportedPredicate = true;
Daniel Sandersd0656a32017-04-13 09:45:37 +0000304 Explanation = Separator + "Has a predicate (" + explainPredicates(N) + ")";
305 Separator = ", ";
Daniel Sanders3f267bf2017-10-15 02:06:44 +0000306 Explanation += (Separator + "first-failing:" +
307 Predicate.getOrigPatFragRecord()->getRecord()->getName())
308 .str();
Daniel Sanders2c269f62017-08-24 09:11:20 +0000309 break;
Daniel Sandersd0656a32017-04-13 09:45:37 +0000310 }
311
312 if (N->getTransformFn()) {
313 Explanation += Separator + "Has a transform function";
314 Separator = ", ";
315 }
316
Daniel Sanders2c269f62017-08-24 09:11:20 +0000317 if (!HasUnsupportedPredicate && !N->getTransformFn())
Daniel Sandersd0656a32017-04-13 09:45:37 +0000318 return Error::success();
319
320 return failedImport(Explanation);
Ahmed Bougacha36f70352016-12-21 23:26:20 +0000321}
322
Daniel Sandersa6e2ceb2017-06-20 12:36:34 +0000323static Record *getInitValueAsRegClass(Init *V) {
324 if (DefInit *VDefInit = dyn_cast<DefInit>(V)) {
325 if (VDefInit->getDef()->isSubClassOf("RegisterOperand"))
326 return VDefInit->getDef()->getValueAsDef("RegClass");
327 if (VDefInit->getDef()->isSubClassOf("RegisterClass"))
328 return VDefInit->getDef();
329 }
330 return nullptr;
331}
332
Daniel Sanders6ab0daa2017-07-04 14:35:06 +0000333std::string
334getNameForFeatureBitset(const std::vector<Record *> &FeatureBitset) {
335 std::string Name = "GIFBS";
336 for (const auto &Feature : FeatureBitset)
337 Name += ("_" + Feature->getName()).str();
338 return Name;
339}
Daniel Sanders7aac7cc2017-07-20 09:25:44 +0000340
341//===- MatchTable Helpers -------------------------------------------------===//
342
343class MatchTable;
344
345/// A record to be stored in a MatchTable.
346///
347/// This class represents any and all output that may be required to emit the
348/// MatchTable. Instances are most often configured to represent an opcode or
349/// value that will be emitted to the table with some formatting but it can also
350/// represent commas, comments, and other formatting instructions.
351struct MatchTableRecord {
352 enum RecordFlagsBits {
353 MTRF_None = 0x0,
354 /// Causes EmitStr to be formatted as comment when emitted.
355 MTRF_Comment = 0x1,
356 /// Causes the record value to be followed by a comma when emitted.
357 MTRF_CommaFollows = 0x2,
358 /// Causes the record value to be followed by a line break when emitted.
359 MTRF_LineBreakFollows = 0x4,
360 /// Indicates that the record defines a label and causes an additional
361 /// comment to be emitted containing the index of the label.
362 MTRF_Label = 0x8,
363 /// Causes the record to be emitted as the index of the label specified by
364 /// LabelID along with a comment indicating where that label is.
365 MTRF_JumpTarget = 0x10,
366 /// Causes the formatter to add a level of indentation before emitting the
367 /// record.
368 MTRF_Indent = 0x20,
369 /// Causes the formatter to remove a level of indentation after emitting the
370 /// record.
371 MTRF_Outdent = 0x40,
372 };
373
374 /// When MTRF_Label or MTRF_JumpTarget is used, indicates a label id to
375 /// reference or define.
376 unsigned LabelID;
377 /// The string to emit. Depending on the MTRF_* flags it may be a comment, a
378 /// value, a label name.
379 std::string EmitStr;
380
381private:
382 /// The number of MatchTable elements described by this record. Comments are 0
383 /// while values are typically 1. Values >1 may occur when we need to emit
384 /// values that exceed the size of a MatchTable element.
385 unsigned NumElements;
386
387public:
388 /// A bitfield of RecordFlagsBits flags.
389 unsigned Flags;
390
391 MatchTableRecord(Optional<unsigned> LabelID_, StringRef EmitStr,
392 unsigned NumElements, unsigned Flags)
393 : LabelID(LabelID_.hasValue() ? LabelID_.getValue() : ~0u),
394 EmitStr(EmitStr), NumElements(NumElements), Flags(Flags) {
395 assert((!LabelID_.hasValue() || LabelID != ~0u) &&
396 "This value is reserved for non-labels");
397 }
398
399 void emit(raw_ostream &OS, bool LineBreakNextAfterThis,
400 const MatchTable &Table) const;
401 unsigned size() const { return NumElements; }
402};
403
404/// Holds the contents of a generated MatchTable to enable formatting and the
405/// necessary index tracking needed to support GIM_Try.
406class MatchTable {
407 /// An unique identifier for the table. The generated table will be named
408 /// MatchTable${ID}.
409 unsigned ID;
410 /// The records that make up the table. Also includes comments describing the
411 /// values being emitted and line breaks to format it.
412 std::vector<MatchTableRecord> Contents;
413 /// The currently defined labels.
414 DenseMap<unsigned, unsigned> LabelMap;
415 /// Tracks the sum of MatchTableRecord::NumElements as the table is built.
416 unsigned CurrentSize;
417
Daniel Sanders8e82af22017-07-27 11:03:45 +0000418 /// A unique identifier for a MatchTable label.
419 static unsigned CurrentLabelID;
420
Daniel Sanders7aac7cc2017-07-20 09:25:44 +0000421public:
422 static MatchTableRecord LineBreak;
423 static MatchTableRecord Comment(StringRef Comment) {
424 return MatchTableRecord(None, Comment, 0, MatchTableRecord::MTRF_Comment);
425 }
426 static MatchTableRecord Opcode(StringRef Opcode, int IndentAdjust = 0) {
427 unsigned ExtraFlags = 0;
428 if (IndentAdjust > 0)
429 ExtraFlags |= MatchTableRecord::MTRF_Indent;
430 if (IndentAdjust < 0)
431 ExtraFlags |= MatchTableRecord::MTRF_Outdent;
432
433 return MatchTableRecord(None, Opcode, 1,
434 MatchTableRecord::MTRF_CommaFollows | ExtraFlags);
435 }
436 static MatchTableRecord NamedValue(StringRef NamedValue) {
437 return MatchTableRecord(None, NamedValue, 1,
438 MatchTableRecord::MTRF_CommaFollows);
439 }
440 static MatchTableRecord NamedValue(StringRef Namespace,
441 StringRef NamedValue) {
442 return MatchTableRecord(None, (Namespace + "::" + NamedValue).str(), 1,
443 MatchTableRecord::MTRF_CommaFollows);
444 }
445 static MatchTableRecord IntValue(int64_t IntValue) {
446 return MatchTableRecord(None, llvm::to_string(IntValue), 1,
447 MatchTableRecord::MTRF_CommaFollows);
448 }
449 static MatchTableRecord Label(unsigned LabelID) {
450 return MatchTableRecord(LabelID, "Label " + llvm::to_string(LabelID), 0,
451 MatchTableRecord::MTRF_Label |
452 MatchTableRecord::MTRF_Comment |
453 MatchTableRecord::MTRF_LineBreakFollows);
454 }
455 static MatchTableRecord JumpTarget(unsigned LabelID) {
Daniel Sanders8e82af22017-07-27 11:03:45 +0000456 return MatchTableRecord(LabelID, "Label " + llvm::to_string(LabelID), 1,
Daniel Sanders7aac7cc2017-07-20 09:25:44 +0000457 MatchTableRecord::MTRF_JumpTarget |
458 MatchTableRecord::MTRF_Comment |
459 MatchTableRecord::MTRF_CommaFollows);
460 }
461
462 MatchTable(unsigned ID) : ID(ID), CurrentSize(0) {}
463
464 void push_back(const MatchTableRecord &Value) {
465 if (Value.Flags & MatchTableRecord::MTRF_Label)
466 defineLabel(Value.LabelID);
467 Contents.push_back(Value);
468 CurrentSize += Value.size();
469 }
470
Daniel Sanders8e82af22017-07-27 11:03:45 +0000471 unsigned allocateLabelID() const { return CurrentLabelID++; }
472
Daniel Sanders7aac7cc2017-07-20 09:25:44 +0000473 void defineLabel(unsigned LabelID) {
Daniel Sanders8e82af22017-07-27 11:03:45 +0000474 LabelMap.insert(std::make_pair(LabelID, CurrentSize));
Daniel Sanders7aac7cc2017-07-20 09:25:44 +0000475 }
476
477 unsigned getLabelIndex(unsigned LabelID) const {
478 const auto I = LabelMap.find(LabelID);
479 assert(I != LabelMap.end() && "Use of undeclared label");
480 return I->second;
481 }
482
Daniel Sanders8e82af22017-07-27 11:03:45 +0000483 void emitUse(raw_ostream &OS) const { OS << "MatchTable" << ID; }
484
485 void emitDeclaration(raw_ostream &OS) const {
Daniel Sanders7aac7cc2017-07-20 09:25:44 +0000486 unsigned Indentation = 4;
Daniel Sanderscbbbfe42017-07-27 12:47:31 +0000487 OS << " constexpr static int64_t MatchTable" << ID << "[] = {";
Daniel Sanders7aac7cc2017-07-20 09:25:44 +0000488 LineBreak.emit(OS, true, *this);
489 OS << std::string(Indentation, ' ');
490
491 for (auto I = Contents.begin(), E = Contents.end(); I != E;
492 ++I) {
493 bool LineBreakIsNext = false;
494 const auto &NextI = std::next(I);
495
496 if (NextI != E) {
497 if (NextI->EmitStr == "" &&
498 NextI->Flags == MatchTableRecord::MTRF_LineBreakFollows)
499 LineBreakIsNext = true;
500 }
501
502 if (I->Flags & MatchTableRecord::MTRF_Indent)
503 Indentation += 2;
504
505 I->emit(OS, LineBreakIsNext, *this);
506 if (I->Flags & MatchTableRecord::MTRF_LineBreakFollows)
507 OS << std::string(Indentation, ' ');
508
509 if (I->Flags & MatchTableRecord::MTRF_Outdent)
510 Indentation -= 2;
511 }
512 OS << "};\n";
513 }
514};
515
Daniel Sanders8e82af22017-07-27 11:03:45 +0000516unsigned MatchTable::CurrentLabelID = 0;
517
Daniel Sanders7aac7cc2017-07-20 09:25:44 +0000518MatchTableRecord MatchTable::LineBreak = {
519 None, "" /* Emit String */, 0 /* Elements */,
520 MatchTableRecord::MTRF_LineBreakFollows};
521
522void MatchTableRecord::emit(raw_ostream &OS, bool LineBreakIsNextAfterThis,
523 const MatchTable &Table) const {
524 bool UseLineComment =
525 LineBreakIsNextAfterThis | (Flags & MTRF_LineBreakFollows);
526 if (Flags & (MTRF_JumpTarget | MTRF_CommaFollows))
527 UseLineComment = false;
528
529 if (Flags & MTRF_Comment)
530 OS << (UseLineComment ? "// " : "/*");
531
532 OS << EmitStr;
533 if (Flags & MTRF_Label)
534 OS << ": @" << Table.getLabelIndex(LabelID);
535
536 if (Flags & MTRF_Comment && !UseLineComment)
537 OS << "*/";
538
539 if (Flags & MTRF_JumpTarget) {
540 if (Flags & MTRF_Comment)
541 OS << " ";
542 OS << Table.getLabelIndex(LabelID);
543 }
544
545 if (Flags & MTRF_CommaFollows) {
546 OS << ",";
547 if (!LineBreakIsNextAfterThis && !(Flags & MTRF_LineBreakFollows))
548 OS << " ";
549 }
550
551 if (Flags & MTRF_LineBreakFollows)
552 OS << "\n";
553}
554
555MatchTable &operator<<(MatchTable &Table, const MatchTableRecord &Value) {
556 Table.push_back(Value);
557 return Table;
558}
559
Ahmed Bougacha36f70352016-12-21 23:26:20 +0000560//===- Matchers -----------------------------------------------------------===//
561
Daniel Sandersbee57392017-04-04 13:25:23 +0000562class OperandMatcher;
Daniel Sandersbdfebb82017-03-15 20:18:38 +0000563class MatchAction;
564
565/// Generates code to check that a match rule matches.
566class RuleMatcher {
Daniel Sanders7438b262017-10-31 23:03:18 +0000567public:
568 using ActionVec = std::vector<std::unique_ptr<MatchAction>>;
569 using action_iterator = ActionVec::iterator;
570
571protected:
Daniel Sandersbdfebb82017-03-15 20:18:38 +0000572 /// A list of matchers that all need to succeed for the current rule to match.
573 /// FIXME: This currently supports a single match position but could be
574 /// extended to support multiple positions to support div/rem fusion or
575 /// load-multiple instructions.
576 std::vector<std::unique_ptr<InstructionMatcher>> Matchers;
577
578 /// A list of actions that need to be taken when all predicates in this rule
579 /// have succeeded.
Daniel Sanders7438b262017-10-31 23:03:18 +0000580 ActionVec Actions;
Daniel Sandersbdfebb82017-03-15 20:18:38 +0000581
Daniel Sandersa7b75262017-10-31 18:50:24 +0000582 using DefinedInsnVariablesMap =
583 std::map<const InstructionMatcher *, unsigned>;
584
Daniel Sandersb96f40d2017-03-20 15:20:42 +0000585 /// A map of instruction matchers to the local variables created by
Daniel Sanders9d662d22017-07-06 10:06:12 +0000586 /// emitCaptureOpcodes().
Daniel Sanders078572b2017-08-02 11:03:36 +0000587 DefinedInsnVariablesMap InsnVariableIDs;
Daniel Sandersb96f40d2017-03-20 15:20:42 +0000588
Daniel Sandersa7b75262017-10-31 18:50:24 +0000589 using MutatableInsnSet = SmallPtrSet<const InstructionMatcher *, 4>;
590
591 // The set of instruction matchers that have not yet been claimed for mutation
592 // by a BuildMI.
593 MutatableInsnSet MutatableInsns;
594
Daniel Sandersbfa9e2c2017-10-14 00:31:58 +0000595 /// A map of named operands defined by the matchers that may be referenced by
596 /// the renderers.
597 StringMap<OperandMatcher *> DefinedOperands;
598
Daniel Sandersb96f40d2017-03-20 15:20:42 +0000599 /// ID for the next instruction variable defined with defineInsnVar()
600 unsigned NextInsnVarID;
601
Daniel Sanders198447a2017-11-01 00:29:47 +0000602 /// ID for the next output instruction allocated with allocateOutputInsnID()
603 unsigned NextOutputInsnID;
604
Daniel Sanders9cbe7c72017-11-01 19:57:57 +0000605 /// ID for the next temporary register ID allocated with allocateTempRegID()
606 unsigned NextTempRegID;
607
Daniel Sanderse7b0d662017-04-21 15:59:56 +0000608 std::vector<Record *> RequiredFeatures;
609
Daniel Sandersbfa9e2c2017-10-14 00:31:58 +0000610 ArrayRef<SMLoc> SrcLoc;
611
Daniel Sandersdf39cba2017-10-15 18:22:54 +0000612 typedef std::tuple<Record *, unsigned, unsigned>
613 DefinedComplexPatternSubOperand;
614 typedef StringMap<DefinedComplexPatternSubOperand>
615 DefinedComplexPatternSubOperandMap;
616 /// A map of Symbolic Names to ComplexPattern sub-operands.
617 DefinedComplexPatternSubOperandMap ComplexSubOperands;
618
Daniel Sandersf76f3152017-11-16 00:46:35 +0000619 uint64_t RuleID;
620 static uint64_t NextRuleID;
621
Daniel Sandersbdfebb82017-03-15 20:18:38 +0000622public:
Daniel Sandersbfa9e2c2017-10-14 00:31:58 +0000623 RuleMatcher(ArrayRef<SMLoc> SrcLoc)
Daniel Sandersa7b75262017-10-31 18:50:24 +0000624 : Matchers(), Actions(), InsnVariableIDs(), MutatableInsns(),
Daniel Sanders198447a2017-11-01 00:29:47 +0000625 DefinedOperands(), NextInsnVarID(0), NextOutputInsnID(0),
Daniel Sandersf76f3152017-11-16 00:46:35 +0000626 NextTempRegID(0), SrcLoc(SrcLoc), ComplexSubOperands(),
627 RuleID(NextRuleID++) {}
Zachary Turnerb7dbd872017-03-20 19:56:52 +0000628 RuleMatcher(RuleMatcher &&Other) = default;
629 RuleMatcher &operator=(RuleMatcher &&Other) = default;
Daniel Sandersbdfebb82017-03-15 20:18:38 +0000630
Daniel Sandersf76f3152017-11-16 00:46:35 +0000631 uint64_t getRuleID() const { return RuleID; }
632
Daniel Sanders05540042017-08-08 10:44:31 +0000633 InstructionMatcher &addInstructionMatcher(StringRef SymbolicName);
Daniel Sanderse7b0d662017-04-21 15:59:56 +0000634 void addRequiredFeature(Record *Feature);
Daniel Sanders6ab0daa2017-07-04 14:35:06 +0000635 const std::vector<Record *> &getRequiredFeatures() const;
Daniel Sandersbdfebb82017-03-15 20:18:38 +0000636
637 template <class Kind, class... Args> Kind &addAction(Args &&... args);
Daniel Sanders7438b262017-10-31 23:03:18 +0000638 template <class Kind, class... Args>
639 action_iterator insertAction(action_iterator InsertPt, Args &&... args);
Daniel Sandersbdfebb82017-03-15 20:18:38 +0000640
Daniel Sanders6ab0daa2017-07-04 14:35:06 +0000641 /// Define an instruction without emitting any code to do so.
642 /// This is used for the root of the match.
643 unsigned implicitlyDefineInsnVar(const InstructionMatcher &Matcher);
644 /// Define an instruction and emit corresponding state-machine opcodes.
Daniel Sanders7aac7cc2017-07-20 09:25:44 +0000645 unsigned defineInsnVar(MatchTable &Table, const InstructionMatcher &Matcher,
Daniel Sanders6ab0daa2017-07-04 14:35:06 +0000646 unsigned InsnVarID, unsigned OpIdx);
647 unsigned getInsnVarID(const InstructionMatcher &InsnMatcher) const;
Daniel Sanders078572b2017-08-02 11:03:36 +0000648 DefinedInsnVariablesMap::const_iterator defined_insn_vars_begin() const {
649 return InsnVariableIDs.begin();
650 }
651 DefinedInsnVariablesMap::const_iterator defined_insn_vars_end() const {
652 return InsnVariableIDs.end();
653 }
654 iterator_range<typename DefinedInsnVariablesMap::const_iterator>
655 defined_insn_vars() const {
656 return make_range(defined_insn_vars_begin(), defined_insn_vars_end());
657 }
Daniel Sandersb96f40d2017-03-20 15:20:42 +0000658
Daniel Sandersa7b75262017-10-31 18:50:24 +0000659 MutatableInsnSet::const_iterator mutatable_insns_begin() const {
660 return MutatableInsns.begin();
661 }
662 MutatableInsnSet::const_iterator mutatable_insns_end() const {
663 return MutatableInsns.end();
664 }
665 iterator_range<typename MutatableInsnSet::const_iterator>
666 mutatable_insns() const {
667 return make_range(mutatable_insns_begin(), mutatable_insns_end());
668 }
669 void reserveInsnMatcherForMutation(const InstructionMatcher *InsnMatcher) {
670 bool R = MutatableInsns.erase(InsnMatcher);
671 assert(R && "Reserving a mutatable insn that isn't available");
672 (void)R;
673 }
674
Daniel Sanders7438b262017-10-31 23:03:18 +0000675 action_iterator actions_begin() { return Actions.begin(); }
676 action_iterator actions_end() { return Actions.end(); }
677 iterator_range<action_iterator> actions() {
678 return make_range(actions_begin(), actions_end());
679 }
680
Daniel Sandersbfa9e2c2017-10-14 00:31:58 +0000681 void defineOperand(StringRef SymbolicName, OperandMatcher &OM);
682
Daniel Sandersdf39cba2017-10-15 18:22:54 +0000683 void defineComplexSubOperand(StringRef SymbolicName, Record *ComplexPattern,
684 unsigned RendererID, unsigned SubOperandID) {
685 assert(ComplexSubOperands.count(SymbolicName) == 0 && "Already defined");
686 ComplexSubOperands[SymbolicName] =
687 std::make_tuple(ComplexPattern, RendererID, SubOperandID);
688 }
689 Optional<DefinedComplexPatternSubOperand>
690 getComplexSubOperand(StringRef SymbolicName) const {
691 const auto &I = ComplexSubOperands.find(SymbolicName);
692 if (I == ComplexSubOperands.end())
693 return None;
694 return I->second;
695 }
696
Daniel Sanders05540042017-08-08 10:44:31 +0000697 const InstructionMatcher &getInstructionMatcher(StringRef SymbolicName) const;
Daniel Sandersbfa9e2c2017-10-14 00:31:58 +0000698 const OperandMatcher &getOperandMatcher(StringRef Name) const;
Daniel Sanders05540042017-08-08 10:44:31 +0000699
Daniel Sanders7aac7cc2017-07-20 09:25:44 +0000700 void emitCaptureOpcodes(MatchTable &Table);
Daniel Sandersb96f40d2017-03-20 15:20:42 +0000701
Daniel Sanders8e82af22017-07-27 11:03:45 +0000702 void emit(MatchTable &Table);
Daniel Sandersbdfebb82017-03-15 20:18:38 +0000703
Daniel Sanders6ab0daa2017-07-04 14:35:06 +0000704 /// Compare the priority of this object and B.
705 ///
706 /// Returns true if this object is more important than B.
707 bool isHigherPriorityThan(const RuleMatcher &B) const;
Daniel Sandersbdfebb82017-03-15 20:18:38 +0000708
Daniel Sanders6ab0daa2017-07-04 14:35:06 +0000709 /// Report the maximum number of temporary operands needed by the rule
710 /// matcher.
711 unsigned countRendererFns() const;
Daniel Sanders2deea182017-04-22 15:11:04 +0000712
Daniel Sanders6ab0daa2017-07-04 14:35:06 +0000713 // FIXME: Remove this as soon as possible
714 InstructionMatcher &insnmatcher_front() const { return *Matchers.front(); }
Daniel Sanders198447a2017-11-01 00:29:47 +0000715
716 unsigned allocateOutputInsnID() { return NextOutputInsnID++; }
Daniel Sanders9cbe7c72017-11-01 19:57:57 +0000717 unsigned allocateTempRegID() { return NextTempRegID++; }
Daniel Sandersbdfebb82017-03-15 20:18:38 +0000718};
719
Daniel Sandersf76f3152017-11-16 00:46:35 +0000720uint64_t RuleMatcher::NextRuleID = 0;
721
Daniel Sanders7438b262017-10-31 23:03:18 +0000722using action_iterator = RuleMatcher::action_iterator;
723
Daniel Sandersdc662ff2017-01-26 11:10:14 +0000724template <class PredicateTy> class PredicateListMatcher {
725private:
726 typedef std::vector<std::unique_ptr<PredicateTy>> PredicateVec;
727 PredicateVec Predicates;
Ahmed Bougacha36f70352016-12-21 23:26:20 +0000728
Daniel Sanders2c269f62017-08-24 09:11:20 +0000729 /// Template instantiations should specialize this to return a string to use
730 /// for the comment emitted when there are no predicates.
731 std::string getNoPredicateComment() const;
732
Daniel Sandersdc662ff2017-01-26 11:10:14 +0000733public:
734 /// Construct a new operand predicate and add it to the matcher.
735 template <class Kind, class... Args>
Daniel Sandersbfa9e2c2017-10-14 00:31:58 +0000736 Optional<Kind *> addPredicate(Args&&... args) {
Ahmed Bougachab67a3ce2017-01-26 22:07:37 +0000737 Predicates.emplace_back(
738 llvm::make_unique<Kind>(std::forward<Args>(args)...));
Daniel Sandersbfa9e2c2017-10-14 00:31:58 +0000739 return static_cast<Kind *>(Predicates.back().get());
Ahmed Bougacha36f70352016-12-21 23:26:20 +0000740 }
Daniel Sandersdc662ff2017-01-26 11:10:14 +0000741
Daniel Sanders32291982017-06-28 13:50:04 +0000742 typename PredicateVec::const_iterator predicates_begin() const {
743 return Predicates.begin();
744 }
745 typename PredicateVec::const_iterator predicates_end() const {
746 return Predicates.end();
747 }
Daniel Sandersdc662ff2017-01-26 11:10:14 +0000748 iterator_range<typename PredicateVec::const_iterator> predicates() const {
749 return make_range(predicates_begin(), predicates_end());
750 }
Daniel Sanders32291982017-06-28 13:50:04 +0000751 typename PredicateVec::size_type predicates_size() const {
752 return Predicates.size();
753 }
Daniel Sandersdc662ff2017-01-26 11:10:14 +0000754
Daniel Sanders9d662d22017-07-06 10:06:12 +0000755 /// Emit MatchTable opcodes that tests whether all the predicates are met.
Ahmed Bougachab67a3ce2017-01-26 22:07:37 +0000756 template <class... Args>
Daniel Sanders7aac7cc2017-07-20 09:25:44 +0000757 void emitPredicateListOpcodes(MatchTable &Table, Args &&... args) const {
Daniel Sandersdc662ff2017-01-26 11:10:14 +0000758 if (Predicates.empty()) {
Daniel Sanders2c269f62017-08-24 09:11:20 +0000759 Table << MatchTable::Comment(getNoPredicateComment())
760 << MatchTable::LineBreak;
Daniel Sandersdc662ff2017-01-26 11:10:14 +0000761 return;
762 }
763
Daniel Sanders6ab0daa2017-07-04 14:35:06 +0000764 for (const auto &Predicate : predicates())
Daniel Sanders7aac7cc2017-07-20 09:25:44 +0000765 Predicate->emitPredicateOpcodes(Table, std::forward<Args>(args)...);
Daniel Sandersdc662ff2017-01-26 11:10:14 +0000766 }
Ahmed Bougacha36f70352016-12-21 23:26:20 +0000767};
768
Daniel Sandersdc662ff2017-01-26 11:10:14 +0000769/// Generates code to check a predicate of an operand.
770///
771/// Typical predicates include:
772/// * Operand is a particular register.
773/// * Operand is assigned a particular register bank.
774/// * Operand is an MBB.
775class OperandPredicateMatcher {
776public:
Daniel Sanders759ff412017-02-24 13:58:11 +0000777 /// This enum is used for RTTI and also defines the priority that is given to
778 /// the predicate when generating the matcher code. Kinds with higher priority
779 /// must be tested first.
780 ///
Daniel Sanders066ebbf2017-02-24 15:43:30 +0000781 /// The relative priority of OPM_LLT, OPM_RegBank, and OPM_MBB do not matter
782 /// but OPM_Int must have priority over OPM_RegBank since constant integers
783 /// are represented by a virtual register defined by a G_CONSTANT instruction.
Daniel Sanders759ff412017-02-24 13:58:11 +0000784 enum PredicateKind {
Daniel Sanders1e4569f2017-10-20 20:55:29 +0000785 OPM_SameOperand,
Daniel Sanders8a4bae92017-03-14 21:32:08 +0000786 OPM_ComplexPattern,
Daniel Sandersfe12c0f2017-07-11 08:57:29 +0000787 OPM_IntrinsicID,
Daniel Sanders05540042017-08-08 10:44:31 +0000788 OPM_Instruction,
Daniel Sanders066ebbf2017-02-24 15:43:30 +0000789 OPM_Int,
Daniel Sanders452c8ae2017-05-23 19:33:16 +0000790 OPM_LiteralInt,
Daniel Sanders759ff412017-02-24 13:58:11 +0000791 OPM_LLT,
Daniel Sandersa71f4542017-10-16 00:56:30 +0000792 OPM_PointerToAny,
Daniel Sanders759ff412017-02-24 13:58:11 +0000793 OPM_RegBank,
794 OPM_MBB,
795 };
796
797protected:
798 PredicateKind Kind;
799
800public:
801 OperandPredicateMatcher(PredicateKind Kind) : Kind(Kind) {}
Daniel Sandersdc662ff2017-01-26 11:10:14 +0000802 virtual ~OperandPredicateMatcher() {}
803
Daniel Sanders759ff412017-02-24 13:58:11 +0000804 PredicateKind getKind() const { return Kind; }
805
Daniel Sanders9d662d22017-07-06 10:06:12 +0000806 /// Emit MatchTable opcodes to capture instructions into the MIs table.
Daniel Sandersbee57392017-04-04 13:25:23 +0000807 ///
Daniel Sanders9d662d22017-07-06 10:06:12 +0000808 /// Only InstructionOperandMatcher needs to do anything for this method the
809 /// rest just walk the tree.
Daniel Sanders7aac7cc2017-07-20 09:25:44 +0000810 virtual void emitCaptureOpcodes(MatchTable &Table, RuleMatcher &Rule,
Daniel Sanders9d662d22017-07-06 10:06:12 +0000811 unsigned InsnVarID, unsigned OpIdx) const {}
Daniel Sandersbee57392017-04-04 13:25:23 +0000812
Daniel Sanders9d662d22017-07-06 10:06:12 +0000813 /// Emit MatchTable opcodes that check the predicate for the given operand.
Daniel Sanders7aac7cc2017-07-20 09:25:44 +0000814 virtual void emitPredicateOpcodes(MatchTable &Table, RuleMatcher &Rule,
Daniel Sanders6ab0daa2017-07-04 14:35:06 +0000815 unsigned InsnVarID,
816 unsigned OpIdx) const = 0;
Daniel Sanders759ff412017-02-24 13:58:11 +0000817
818 /// Compare the priority of this object and B.
819 ///
820 /// Returns true if this object is more important than B.
Daniel Sanders05540042017-08-08 10:44:31 +0000821 virtual bool isHigherPriorityThan(const OperandPredicateMatcher &B) const;
Daniel Sanders8a4bae92017-03-14 21:32:08 +0000822
823 /// Report the maximum number of temporary operands needed by the predicate
824 /// matcher.
Daniel Sanders2deea182017-04-22 15:11:04 +0000825 virtual unsigned countRendererFns() const { return 0; }
Daniel Sandersdc662ff2017-01-26 11:10:14 +0000826};
827
Daniel Sanders2c269f62017-08-24 09:11:20 +0000828template <>
829std::string
830PredicateListMatcher<OperandPredicateMatcher>::getNoPredicateComment() const {
831 return "No operand predicates";
832}
833
Daniel Sandersbfa9e2c2017-10-14 00:31:58 +0000834/// Generates code to check that a register operand is defined by the same exact
835/// one as another.
836class SameOperandMatcher : public OperandPredicateMatcher {
Daniel Sanders1e4569f2017-10-20 20:55:29 +0000837 std::string MatchingName;
Daniel Sandersbfa9e2c2017-10-14 00:31:58 +0000838
839public:
Daniel Sanders1e4569f2017-10-20 20:55:29 +0000840 SameOperandMatcher(StringRef MatchingName)
841 : OperandPredicateMatcher(OPM_SameOperand), MatchingName(MatchingName) {}
Daniel Sandersbfa9e2c2017-10-14 00:31:58 +0000842
843 static bool classof(const OperandPredicateMatcher *P) {
Daniel Sanders1e4569f2017-10-20 20:55:29 +0000844 return P->getKind() == OPM_SameOperand;
Daniel Sandersbfa9e2c2017-10-14 00:31:58 +0000845 }
846
847 void emitPredicateOpcodes(MatchTable &Table, RuleMatcher &Rule,
848 unsigned InsnVarID, unsigned OpIdx) const override;
849};
850
Daniel Sandersdc662ff2017-01-26 11:10:14 +0000851/// Generates code to check that an operand is a particular LLT.
852class LLTOperandMatcher : public OperandPredicateMatcher {
853protected:
Daniel Sanders52b4ce72017-03-07 23:20:35 +0000854 LLTCodeGen Ty;
Ahmed Bougacha36f70352016-12-21 23:26:20 +0000855
Daniel Sandersdc662ff2017-01-26 11:10:14 +0000856public:
Daniel Sanders032e7f22017-08-17 13:18:35 +0000857 static std::set<LLTCodeGen> KnownTypes;
858
Daniel Sanders52b4ce72017-03-07 23:20:35 +0000859 LLTOperandMatcher(const LLTCodeGen &Ty)
Daniel Sanders032e7f22017-08-17 13:18:35 +0000860 : OperandPredicateMatcher(OPM_LLT), Ty(Ty) {
861 KnownTypes.insert(Ty);
862 }
Daniel Sanders759ff412017-02-24 13:58:11 +0000863
864 static bool classof(const OperandPredicateMatcher *P) {
865 return P->getKind() == OPM_LLT;
866 }
Daniel Sandersdc662ff2017-01-26 11:10:14 +0000867
Daniel Sanders7aac7cc2017-07-20 09:25:44 +0000868 void emitPredicateOpcodes(MatchTable &Table, RuleMatcher &Rule,
Daniel Sanders6ab0daa2017-07-04 14:35:06 +0000869 unsigned InsnVarID, unsigned OpIdx) const override {
Daniel Sanders7aac7cc2017-07-20 09:25:44 +0000870 Table << MatchTable::Opcode("GIM_CheckType") << MatchTable::Comment("MI")
871 << MatchTable::IntValue(InsnVarID) << MatchTable::Comment("Op")
872 << MatchTable::IntValue(OpIdx) << MatchTable::Comment("Type")
873 << MatchTable::NamedValue(Ty.getCxxEnumValue())
874 << MatchTable::LineBreak;
Ahmed Bougacha36f70352016-12-21 23:26:20 +0000875 }
876};
877
Daniel Sanders032e7f22017-08-17 13:18:35 +0000878std::set<LLTCodeGen> LLTOperandMatcher::KnownTypes;
879
Daniel Sandersa71f4542017-10-16 00:56:30 +0000880/// Generates code to check that an operand is a pointer to any address space.
881///
882/// In SelectionDAG, the types did not describe pointers or address spaces. As a
883/// result, iN is used to describe a pointer of N bits to any address space and
884/// PatFrag predicates are typically used to constrain the address space. There's
885/// no reliable means to derive the missing type information from the pattern so
886/// imported rules must test the components of a pointer separately.
887///
Daniel Sandersea8711b2017-10-16 03:36:29 +0000888/// If SizeInBits is zero, then the pointer size will be obtained from the
889/// subtarget.
Daniel Sandersa71f4542017-10-16 00:56:30 +0000890class PointerToAnyOperandMatcher : public OperandPredicateMatcher {
891protected:
892 unsigned SizeInBits;
893
894public:
895 PointerToAnyOperandMatcher(unsigned SizeInBits)
896 : OperandPredicateMatcher(OPM_PointerToAny), SizeInBits(SizeInBits) {}
897
898 static bool classof(const OperandPredicateMatcher *P) {
899 return P->getKind() == OPM_PointerToAny;
900 }
901
902 void emitPredicateOpcodes(MatchTable &Table, RuleMatcher &Rule,
903 unsigned InsnVarID, unsigned OpIdx) const override {
904 Table << MatchTable::Opcode("GIM_CheckPointerToAny") << MatchTable::Comment("MI")
905 << MatchTable::IntValue(InsnVarID) << MatchTable::Comment("Op")
906 << MatchTable::IntValue(OpIdx) << MatchTable::Comment("SizeInBits")
907 << MatchTable::IntValue(SizeInBits) << MatchTable::LineBreak;
908 }
909};
910
Daniel Sanders8a4bae92017-03-14 21:32:08 +0000911/// Generates code to check that an operand is a particular target constant.
912class ComplexPatternOperandMatcher : public OperandPredicateMatcher {
913protected:
Daniel Sanders4f3eb242017-04-05 13:14:03 +0000914 const OperandMatcher &Operand;
Daniel Sanders8a4bae92017-03-14 21:32:08 +0000915 const Record &TheDef;
Daniel Sanders8a4bae92017-03-14 21:32:08 +0000916
Daniel Sanders4f3eb242017-04-05 13:14:03 +0000917 unsigned getAllocatedTemporariesBaseID() const;
918
Daniel Sanders8a4bae92017-03-14 21:32:08 +0000919public:
Daniel Sanders4f3eb242017-04-05 13:14:03 +0000920 ComplexPatternOperandMatcher(const OperandMatcher &Operand,
921 const Record &TheDef)
922 : OperandPredicateMatcher(OPM_ComplexPattern), Operand(Operand),
923 TheDef(TheDef) {}
924
925 static bool classof(const OperandPredicateMatcher *P) {
926 return P->getKind() == OPM_ComplexPattern;
927 }
Daniel Sanders8a4bae92017-03-14 21:32:08 +0000928
Daniel Sanders7aac7cc2017-07-20 09:25:44 +0000929 void emitPredicateOpcodes(MatchTable &Table, RuleMatcher &Rule,
Daniel Sanders6ab0daa2017-07-04 14:35:06 +0000930 unsigned InsnVarID, unsigned OpIdx) const override {
Daniel Sanders2deea182017-04-22 15:11:04 +0000931 unsigned ID = getAllocatedTemporariesBaseID();
Daniel Sanders7aac7cc2017-07-20 09:25:44 +0000932 Table << MatchTable::Opcode("GIM_CheckComplexPattern")
933 << MatchTable::Comment("MI") << MatchTable::IntValue(InsnVarID)
934 << MatchTable::Comment("Op") << MatchTable::IntValue(OpIdx)
935 << MatchTable::Comment("Renderer") << MatchTable::IntValue(ID)
936 << MatchTable::NamedValue(("GICP_" + TheDef.getName()).str())
937 << MatchTable::LineBreak;
Daniel Sanders8a4bae92017-03-14 21:32:08 +0000938 }
939
Daniel Sanders2deea182017-04-22 15:11:04 +0000940 unsigned countRendererFns() const override {
941 return 1;
Daniel Sanders8a4bae92017-03-14 21:32:08 +0000942 }
943};
944
Daniel Sandersdc662ff2017-01-26 11:10:14 +0000945/// Generates code to check that an operand is in a particular register bank.
946class RegisterBankOperandMatcher : public OperandPredicateMatcher {
947protected:
Ahmed Bougacha36f70352016-12-21 23:26:20 +0000948 const CodeGenRegisterClass &RC;
949
Daniel Sandersdc662ff2017-01-26 11:10:14 +0000950public:
Daniel Sanders759ff412017-02-24 13:58:11 +0000951 RegisterBankOperandMatcher(const CodeGenRegisterClass &RC)
952 : OperandPredicateMatcher(OPM_RegBank), RC(RC) {}
953
954 static bool classof(const OperandPredicateMatcher *P) {
955 return P->getKind() == OPM_RegBank;
956 }
Daniel Sandersdc662ff2017-01-26 11:10:14 +0000957
Daniel Sanders7aac7cc2017-07-20 09:25:44 +0000958 void emitPredicateOpcodes(MatchTable &Table, RuleMatcher &Rule,
Daniel Sanders6ab0daa2017-07-04 14:35:06 +0000959 unsigned InsnVarID, unsigned OpIdx) const override {
Daniel Sanders7aac7cc2017-07-20 09:25:44 +0000960 Table << MatchTable::Opcode("GIM_CheckRegBankForClass")
961 << MatchTable::Comment("MI") << MatchTable::IntValue(InsnVarID)
962 << MatchTable::Comment("Op") << MatchTable::IntValue(OpIdx)
963 << MatchTable::Comment("RC")
964 << MatchTable::NamedValue(RC.getQualifiedName() + "RegClassID")
965 << MatchTable::LineBreak;
Ahmed Bougacha36f70352016-12-21 23:26:20 +0000966 }
967};
968
Daniel Sandersdc662ff2017-01-26 11:10:14 +0000969/// Generates code to check that an operand is a basic block.
970class MBBOperandMatcher : public OperandPredicateMatcher {
971public:
Daniel Sanders759ff412017-02-24 13:58:11 +0000972 MBBOperandMatcher() : OperandPredicateMatcher(OPM_MBB) {}
973
974 static bool classof(const OperandPredicateMatcher *P) {
975 return P->getKind() == OPM_MBB;
976 }
977
Daniel Sanders7aac7cc2017-07-20 09:25:44 +0000978 void emitPredicateOpcodes(MatchTable &Table, RuleMatcher &Rule,
Daniel Sanders6ab0daa2017-07-04 14:35:06 +0000979 unsigned InsnVarID, unsigned OpIdx) const override {
Daniel Sanders7aac7cc2017-07-20 09:25:44 +0000980 Table << MatchTable::Opcode("GIM_CheckIsMBB") << MatchTable::Comment("MI")
981 << MatchTable::IntValue(InsnVarID) << MatchTable::Comment("Op")
982 << MatchTable::IntValue(OpIdx) << MatchTable::LineBreak;
Daniel Sandersdc662ff2017-01-26 11:10:14 +0000983 }
984};
985
Daniel Sanders452c8ae2017-05-23 19:33:16 +0000986/// Generates code to check that an operand is a G_CONSTANT with a particular
987/// int.
988class ConstantIntOperandMatcher : public OperandPredicateMatcher {
Daniel Sanders066ebbf2017-02-24 15:43:30 +0000989protected:
990 int64_t Value;
991
992public:
Daniel Sanders452c8ae2017-05-23 19:33:16 +0000993 ConstantIntOperandMatcher(int64_t Value)
Daniel Sanders066ebbf2017-02-24 15:43:30 +0000994 : OperandPredicateMatcher(OPM_Int), Value(Value) {}
995
996 static bool classof(const OperandPredicateMatcher *P) {
997 return P->getKind() == OPM_Int;
998 }
999
Daniel Sanders7aac7cc2017-07-20 09:25:44 +00001000 void emitPredicateOpcodes(MatchTable &Table, RuleMatcher &Rule,
Daniel Sanders6ab0daa2017-07-04 14:35:06 +00001001 unsigned InsnVarID, unsigned OpIdx) const override {
Daniel Sanders7aac7cc2017-07-20 09:25:44 +00001002 Table << MatchTable::Opcode("GIM_CheckConstantInt")
1003 << MatchTable::Comment("MI") << MatchTable::IntValue(InsnVarID)
1004 << MatchTable::Comment("Op") << MatchTable::IntValue(OpIdx)
1005 << MatchTable::IntValue(Value) << MatchTable::LineBreak;
Daniel Sanders066ebbf2017-02-24 15:43:30 +00001006 }
1007};
1008
Daniel Sanders452c8ae2017-05-23 19:33:16 +00001009/// Generates code to check that an operand is a raw int (where MO.isImm() or
1010/// MO.isCImm() is true).
1011class LiteralIntOperandMatcher : public OperandPredicateMatcher {
1012protected:
1013 int64_t Value;
1014
1015public:
1016 LiteralIntOperandMatcher(int64_t Value)
1017 : OperandPredicateMatcher(OPM_LiteralInt), Value(Value) {}
1018
1019 static bool classof(const OperandPredicateMatcher *P) {
1020 return P->getKind() == OPM_LiteralInt;
1021 }
1022
Daniel Sanders7aac7cc2017-07-20 09:25:44 +00001023 void emitPredicateOpcodes(MatchTable &Table, RuleMatcher &Rule,
Daniel Sanders6ab0daa2017-07-04 14:35:06 +00001024 unsigned InsnVarID, unsigned OpIdx) const override {
Daniel Sanders7aac7cc2017-07-20 09:25:44 +00001025 Table << MatchTable::Opcode("GIM_CheckLiteralInt")
1026 << MatchTable::Comment("MI") << MatchTable::IntValue(InsnVarID)
1027 << MatchTable::Comment("Op") << MatchTable::IntValue(OpIdx)
1028 << MatchTable::IntValue(Value) << MatchTable::LineBreak;
Daniel Sanders452c8ae2017-05-23 19:33:16 +00001029 }
1030};
1031
Daniel Sandersfe12c0f2017-07-11 08:57:29 +00001032/// Generates code to check that an operand is an intrinsic ID.
1033class IntrinsicIDOperandMatcher : public OperandPredicateMatcher {
1034protected:
1035 const CodeGenIntrinsic *II;
1036
1037public:
1038 IntrinsicIDOperandMatcher(const CodeGenIntrinsic *II)
1039 : OperandPredicateMatcher(OPM_IntrinsicID), II(II) {}
1040
1041 static bool classof(const OperandPredicateMatcher *P) {
1042 return P->getKind() == OPM_IntrinsicID;
1043 }
1044
Daniel Sanders7aac7cc2017-07-20 09:25:44 +00001045 void emitPredicateOpcodes(MatchTable &Table, RuleMatcher &Rule,
Daniel Sandersfe12c0f2017-07-11 08:57:29 +00001046 unsigned InsnVarID, unsigned OpIdx) const override {
Daniel Sanders7aac7cc2017-07-20 09:25:44 +00001047 Table << MatchTable::Opcode("GIM_CheckIntrinsicID")
1048 << MatchTable::Comment("MI") << MatchTable::IntValue(InsnVarID)
1049 << MatchTable::Comment("Op") << MatchTable::IntValue(OpIdx)
1050 << MatchTable::NamedValue("Intrinsic::" + II->EnumName)
1051 << MatchTable::LineBreak;
Daniel Sandersfe12c0f2017-07-11 08:57:29 +00001052 }
1053};
1054
Daniel Sandersdc662ff2017-01-26 11:10:14 +00001055/// Generates code to check that a set of predicates match for a particular
1056/// operand.
1057class OperandMatcher : public PredicateListMatcher<OperandPredicateMatcher> {
1058protected:
Daniel Sandersb96f40d2017-03-20 15:20:42 +00001059 InstructionMatcher &Insn;
Ahmed Bougacha36f70352016-12-21 23:26:20 +00001060 unsigned OpIdx;
Daniel Sanders066ebbf2017-02-24 15:43:30 +00001061 std::string SymbolicName;
Ahmed Bougacha36f70352016-12-21 23:26:20 +00001062
Daniel Sanders4f3eb242017-04-05 13:14:03 +00001063 /// The index of the first temporary variable allocated to this operand. The
1064 /// number of allocated temporaries can be found with
Daniel Sanders2deea182017-04-22 15:11:04 +00001065 /// countRendererFns().
Daniel Sanders4f3eb242017-04-05 13:14:03 +00001066 unsigned AllocatedTemporariesBaseID;
1067
Daniel Sandersdc662ff2017-01-26 11:10:14 +00001068public:
Daniel Sandersb96f40d2017-03-20 15:20:42 +00001069 OperandMatcher(InstructionMatcher &Insn, unsigned OpIdx,
Daniel Sanders4f3eb242017-04-05 13:14:03 +00001070 const std::string &SymbolicName,
1071 unsigned AllocatedTemporariesBaseID)
1072 : Insn(Insn), OpIdx(OpIdx), SymbolicName(SymbolicName),
1073 AllocatedTemporariesBaseID(AllocatedTemporariesBaseID) {}
Daniel Sanders066ebbf2017-02-24 15:43:30 +00001074
1075 bool hasSymbolicName() const { return !SymbolicName.empty(); }
1076 const StringRef getSymbolicName() const { return SymbolicName; }
Daniel Sandersffc7d582017-03-29 15:37:18 +00001077 void setSymbolicName(StringRef Name) {
1078 assert(SymbolicName.empty() && "Operand already has a symbolic name");
1079 SymbolicName = Name;
1080 }
Daniel Sanders066ebbf2017-02-24 15:43:30 +00001081 unsigned getOperandIndex() const { return OpIdx; }
1082
Daniel Sanders6ab0daa2017-07-04 14:35:06 +00001083 std::string getOperandExpr(unsigned InsnVarID) const {
1084 return "State.MIs[" + llvm::to_string(InsnVarID) + "]->getOperand(" +
1085 llvm::to_string(OpIdx) + ")";
Daniel Sanderse604ef52017-02-20 15:30:43 +00001086 }
Daniel Sandersdc662ff2017-01-26 11:10:14 +00001087
Daniel Sandersb96f40d2017-03-20 15:20:42 +00001088 InstructionMatcher &getInstructionMatcher() const { return Insn; }
1089
Daniel Sandersa71f4542017-10-16 00:56:30 +00001090 Error addTypeCheckPredicate(const TypeSetByHwMode &VTy,
Reid Klecknercfdd4a22017-10-16 20:31:16 +00001091 bool OperandIsAPointer);
Daniel Sandersa71f4542017-10-16 00:56:30 +00001092
Daniel Sanders9d662d22017-07-06 10:06:12 +00001093 /// Emit MatchTable opcodes to capture instructions into the MIs table.
Daniel Sanders7aac7cc2017-07-20 09:25:44 +00001094 void emitCaptureOpcodes(MatchTable &Table, RuleMatcher &Rule,
Daniel Sanders9d662d22017-07-06 10:06:12 +00001095 unsigned InsnVarID) const {
Daniel Sandersbee57392017-04-04 13:25:23 +00001096 for (const auto &Predicate : predicates())
Daniel Sanders7aac7cc2017-07-20 09:25:44 +00001097 Predicate->emitCaptureOpcodes(Table, Rule, InsnVarID, OpIdx);
Daniel Sandersbee57392017-04-04 13:25:23 +00001098 }
1099
Daniel Sanders9d662d22017-07-06 10:06:12 +00001100 /// Emit MatchTable opcodes that test whether the instruction named in
Daniel Sanders6ab0daa2017-07-04 14:35:06 +00001101 /// InsnVarID matches all the predicates and all the operands.
Daniel Sanders7aac7cc2017-07-20 09:25:44 +00001102 void emitPredicateOpcodes(MatchTable &Table, RuleMatcher &Rule,
Daniel Sanders6ab0daa2017-07-04 14:35:06 +00001103 unsigned InsnVarID) const {
Daniel Sanders7aac7cc2017-07-20 09:25:44 +00001104 std::string Comment;
1105 raw_string_ostream CommentOS(Comment);
1106 CommentOS << "MIs[" << InsnVarID << "] ";
Daniel Sanders066ebbf2017-02-24 15:43:30 +00001107 if (SymbolicName.empty())
Daniel Sanders7aac7cc2017-07-20 09:25:44 +00001108 CommentOS << "Operand " << OpIdx;
Daniel Sanders066ebbf2017-02-24 15:43:30 +00001109 else
Daniel Sanders7aac7cc2017-07-20 09:25:44 +00001110 CommentOS << SymbolicName;
1111 Table << MatchTable::Comment(CommentOS.str()) << MatchTable::LineBreak;
1112
1113 emitPredicateListOpcodes(Table, Rule, InsnVarID, OpIdx);
Daniel Sandersdc662ff2017-01-26 11:10:14 +00001114 }
Daniel Sanders759ff412017-02-24 13:58:11 +00001115
1116 /// Compare the priority of this object and B.
1117 ///
1118 /// Returns true if this object is more important than B.
1119 bool isHigherPriorityThan(const OperandMatcher &B) const {
1120 // Operand matchers involving more predicates have higher priority.
1121 if (predicates_size() > B.predicates_size())
1122 return true;
1123 if (predicates_size() < B.predicates_size())
1124 return false;
1125
1126 // This assumes that predicates are added in a consistent order.
1127 for (const auto &Predicate : zip(predicates(), B.predicates())) {
1128 if (std::get<0>(Predicate)->isHigherPriorityThan(*std::get<1>(Predicate)))
1129 return true;
1130 if (std::get<1>(Predicate)->isHigherPriorityThan(*std::get<0>(Predicate)))
1131 return false;
1132 }
1133
1134 return false;
1135 };
Daniel Sanders8a4bae92017-03-14 21:32:08 +00001136
1137 /// Report the maximum number of temporary operands needed by the operand
1138 /// matcher.
Daniel Sanders2deea182017-04-22 15:11:04 +00001139 unsigned countRendererFns() const {
Daniel Sanders8a4bae92017-03-14 21:32:08 +00001140 return std::accumulate(
1141 predicates().begin(), predicates().end(), 0,
1142 [](unsigned A,
1143 const std::unique_ptr<OperandPredicateMatcher> &Predicate) {
Daniel Sanders2deea182017-04-22 15:11:04 +00001144 return A + Predicate->countRendererFns();
Daniel Sanders8a4bae92017-03-14 21:32:08 +00001145 });
1146 }
Daniel Sanders4f3eb242017-04-05 13:14:03 +00001147
1148 unsigned getAllocatedTemporariesBaseID() const {
1149 return AllocatedTemporariesBaseID;
1150 }
Daniel Sandersbfa9e2c2017-10-14 00:31:58 +00001151
1152 bool isSameAsAnotherOperand() const {
1153 for (const auto &Predicate : predicates())
1154 if (isa<SameOperandMatcher>(Predicate))
1155 return true;
1156 return false;
1157 }
Daniel Sandersdc662ff2017-01-26 11:10:14 +00001158};
1159
Reid Klecknercfdd4a22017-10-16 20:31:16 +00001160// Specialize OperandMatcher::addPredicate() to refrain from adding redundant
1161// predicates.
1162template <>
1163template <class Kind, class... Args>
1164Optional<Kind *>
1165PredicateListMatcher<OperandPredicateMatcher>::addPredicate(Args &&... args) {
1166 if (static_cast<OperandMatcher *>(this)->isSameAsAnotherOperand())
1167 return None;
1168 Predicates.emplace_back(llvm::make_unique<Kind>(std::forward<Args>(args)...));
1169 return static_cast<Kind *>(Predicates.back().get());
1170}
1171
1172Error OperandMatcher::addTypeCheckPredicate(const TypeSetByHwMode &VTy,
1173 bool OperandIsAPointer) {
1174 if (!VTy.isMachineValueType())
1175 return failedImport("unsupported typeset");
1176
1177 if (VTy.getMachineValueType() == MVT::iPTR && OperandIsAPointer) {
1178 addPredicate<PointerToAnyOperandMatcher>(0);
1179 return Error::success();
1180 }
1181
1182 auto OpTyOrNone = MVTToLLT(VTy.getMachineValueType().SimpleTy);
1183 if (!OpTyOrNone)
1184 return failedImport("unsupported type");
1185
1186 if (OperandIsAPointer)
1187 addPredicate<PointerToAnyOperandMatcher>(OpTyOrNone->get().getSizeInBits());
1188 else
1189 addPredicate<LLTOperandMatcher>(*OpTyOrNone);
1190 return Error::success();
1191}
1192
Daniel Sanders4f3eb242017-04-05 13:14:03 +00001193unsigned ComplexPatternOperandMatcher::getAllocatedTemporariesBaseID() const {
1194 return Operand.getAllocatedTemporariesBaseID();
1195}
1196
Daniel Sandersdc662ff2017-01-26 11:10:14 +00001197/// Generates code to check a predicate on an instruction.
1198///
1199/// Typical predicates include:
1200/// * The opcode of the instruction is a particular value.
1201/// * The nsw/nuw flag is/isn't set.
1202class InstructionPredicateMatcher {
Daniel Sanders759ff412017-02-24 13:58:11 +00001203protected:
1204 /// This enum is used for RTTI and also defines the priority that is given to
1205 /// the predicate when generating the matcher code. Kinds with higher priority
1206 /// must be tested first.
1207 enum PredicateKind {
1208 IPM_Opcode,
Daniel Sanders2c269f62017-08-24 09:11:20 +00001209 IPM_ImmPredicate,
Daniel Sanders76664652017-11-28 22:07:05 +00001210 IPM_AtomicOrderingMMO,
Daniel Sanders759ff412017-02-24 13:58:11 +00001211 };
1212
1213 PredicateKind Kind;
1214
Daniel Sandersdc662ff2017-01-26 11:10:14 +00001215public:
Daniel Sanders8d4d72f2017-02-24 14:53:35 +00001216 InstructionPredicateMatcher(PredicateKind Kind) : Kind(Kind) {}
Daniel Sandersdc662ff2017-01-26 11:10:14 +00001217 virtual ~InstructionPredicateMatcher() {}
1218
Daniel Sanders759ff412017-02-24 13:58:11 +00001219 PredicateKind getKind() const { return Kind; }
1220
Daniel Sanders9d662d22017-07-06 10:06:12 +00001221 /// Emit MatchTable opcodes that test whether the instruction named in
Daniel Sanders6ab0daa2017-07-04 14:35:06 +00001222 /// InsnVarID matches the predicate.
Daniel Sanders7aac7cc2017-07-20 09:25:44 +00001223 virtual void emitPredicateOpcodes(MatchTable &Table, RuleMatcher &Rule,
Daniel Sanders6ab0daa2017-07-04 14:35:06 +00001224 unsigned InsnVarID) const = 0;
Daniel Sanders759ff412017-02-24 13:58:11 +00001225
1226 /// Compare the priority of this object and B.
1227 ///
1228 /// Returns true if this object is more important than B.
Daniel Sanders32291982017-06-28 13:50:04 +00001229 virtual bool
1230 isHigherPriorityThan(const InstructionPredicateMatcher &B) const {
Daniel Sanders759ff412017-02-24 13:58:11 +00001231 return Kind < B.Kind;
1232 };
Daniel Sanders8a4bae92017-03-14 21:32:08 +00001233
1234 /// Report the maximum number of temporary operands needed by the predicate
1235 /// matcher.
Daniel Sanders2deea182017-04-22 15:11:04 +00001236 virtual unsigned countRendererFns() const { return 0; }
Daniel Sandersdc662ff2017-01-26 11:10:14 +00001237};
1238
Daniel Sanders2c269f62017-08-24 09:11:20 +00001239template <>
1240std::string
1241PredicateListMatcher<InstructionPredicateMatcher>::getNoPredicateComment() const {
1242 return "No instruction predicates";
1243}
1244
Daniel Sandersdc662ff2017-01-26 11:10:14 +00001245/// Generates code to check the opcode of an instruction.
1246class InstructionOpcodeMatcher : public InstructionPredicateMatcher {
1247protected:
1248 const CodeGenInstruction *I;
1249
1250public:
Daniel Sanders8d4d72f2017-02-24 14:53:35 +00001251 InstructionOpcodeMatcher(const CodeGenInstruction *I)
1252 : InstructionPredicateMatcher(IPM_Opcode), I(I) {}
Daniel Sandersdc662ff2017-01-26 11:10:14 +00001253
Daniel Sanders759ff412017-02-24 13:58:11 +00001254 static bool classof(const InstructionPredicateMatcher *P) {
1255 return P->getKind() == IPM_Opcode;
1256 }
1257
Daniel Sanders7aac7cc2017-07-20 09:25:44 +00001258 void emitPredicateOpcodes(MatchTable &Table, RuleMatcher &Rule,
Daniel Sanders6ab0daa2017-07-04 14:35:06 +00001259 unsigned InsnVarID) const override {
Daniel Sanders7aac7cc2017-07-20 09:25:44 +00001260 Table << MatchTable::Opcode("GIM_CheckOpcode") << MatchTable::Comment("MI")
1261 << MatchTable::IntValue(InsnVarID)
1262 << MatchTable::NamedValue(I->Namespace, I->TheDef->getName())
1263 << MatchTable::LineBreak;
Daniel Sandersdc662ff2017-01-26 11:10:14 +00001264 }
Daniel Sanders759ff412017-02-24 13:58:11 +00001265
1266 /// Compare the priority of this object and B.
1267 ///
Daniel Sanders066ebbf2017-02-24 15:43:30 +00001268 /// Returns true if this object is more important than B.
Daniel Sanders32291982017-06-28 13:50:04 +00001269 bool
1270 isHigherPriorityThan(const InstructionPredicateMatcher &B) const override {
Daniel Sanders759ff412017-02-24 13:58:11 +00001271 if (InstructionPredicateMatcher::isHigherPriorityThan(B))
1272 return true;
1273 if (B.InstructionPredicateMatcher::isHigherPriorityThan(*this))
1274 return false;
1275
1276 // Prioritize opcodes for cosmetic reasons in the generated source. Although
1277 // this is cosmetic at the moment, we may want to drive a similar ordering
1278 // using instruction frequency information to improve compile time.
1279 if (const InstructionOpcodeMatcher *BO =
1280 dyn_cast<InstructionOpcodeMatcher>(&B))
1281 return I->TheDef->getName() < BO->I->TheDef->getName();
1282
1283 return false;
1284 };
Daniel Sanders05540042017-08-08 10:44:31 +00001285
1286 bool isConstantInstruction() const {
1287 return I->TheDef->getName() == "G_CONSTANT";
1288 }
Daniel Sandersdc662ff2017-01-26 11:10:14 +00001289};
1290
Daniel Sanders2c269f62017-08-24 09:11:20 +00001291/// Generates code to check that this instruction is a constant whose value
1292/// meets an immediate predicate.
1293///
1294/// Immediates are slightly odd since they are typically used like an operand
1295/// but are represented as an operator internally. We typically write simm8:$src
1296/// in a tablegen pattern, but this is just syntactic sugar for
1297/// (imm:i32)<<P:Predicate_simm8>>:$imm which more directly describes the nodes
1298/// that will be matched and the predicate (which is attached to the imm
1299/// operator) that will be tested. In SelectionDAG this describes a
1300/// ConstantSDNode whose internal value will be tested using the simm8 predicate.
1301///
1302/// The corresponding GlobalISel representation is %1 = G_CONSTANT iN Value. In
1303/// this representation, the immediate could be tested with an
1304/// InstructionMatcher, InstructionOpcodeMatcher, OperandMatcher, and a
1305/// OperandPredicateMatcher-subclass to check the Value meets the predicate but
1306/// there are two implementation issues with producing that matcher
1307/// configuration from the SelectionDAG pattern:
1308/// * ImmLeaf is a PatFrag whose root is an InstructionMatcher. This means that
1309/// were we to sink the immediate predicate to the operand we would have to
1310/// have two partial implementations of PatFrag support, one for immediates
1311/// and one for non-immediates.
1312/// * At the point we handle the predicate, the OperandMatcher hasn't been
1313/// created yet. If we were to sink the predicate to the OperandMatcher we
1314/// would also have to complicate (or duplicate) the code that descends and
1315/// creates matchers for the subtree.
1316/// Overall, it's simpler to handle it in the place it was found.
1317class InstructionImmPredicateMatcher : public InstructionPredicateMatcher {
1318protected:
1319 TreePredicateFn Predicate;
1320
1321public:
1322 InstructionImmPredicateMatcher(const TreePredicateFn &Predicate)
1323 : InstructionPredicateMatcher(IPM_ImmPredicate), Predicate(Predicate) {}
1324
1325 static bool classof(const InstructionPredicateMatcher *P) {
1326 return P->getKind() == IPM_ImmPredicate;
1327 }
1328
1329 void emitPredicateOpcodes(MatchTable &Table, RuleMatcher &Rule,
1330 unsigned InsnVarID) const override {
Daniel Sanders11300ce2017-10-13 21:28:03 +00001331 Table << MatchTable::Opcode(getMatchOpcodeForPredicate(Predicate))
Daniel Sanders2c269f62017-08-24 09:11:20 +00001332 << MatchTable::Comment("MI") << MatchTable::IntValue(InsnVarID)
1333 << MatchTable::Comment("Predicate")
Daniel Sanders11300ce2017-10-13 21:28:03 +00001334 << MatchTable::NamedValue(getEnumNameForPredicate(Predicate))
Daniel Sanders2c269f62017-08-24 09:11:20 +00001335 << MatchTable::LineBreak;
1336 }
1337};
1338
Daniel Sanders76664652017-11-28 22:07:05 +00001339/// Generates code to check that a memory instruction has a atomic ordering
1340/// MachineMemoryOperand.
1341class AtomicOrderingMMOPredicateMatcher : public InstructionPredicateMatcher {
Daniel Sanders0c43b3a2017-11-30 21:05:59 +00001342public:
1343 enum AOComparator {
1344 AO_Exactly,
1345 AO_OrStronger,
1346 AO_WeakerThan,
1347 };
1348
1349protected:
Daniel Sanders76664652017-11-28 22:07:05 +00001350 StringRef Order;
Daniel Sanders0c43b3a2017-11-30 21:05:59 +00001351 AOComparator Comparator;
Daniel Sanders76664652017-11-28 22:07:05 +00001352
Daniel Sanders39690bd2017-10-15 02:41:12 +00001353public:
Daniel Sanders0c43b3a2017-11-30 21:05:59 +00001354 AtomicOrderingMMOPredicateMatcher(StringRef Order,
1355 AOComparator Comparator = AO_Exactly)
1356 : InstructionPredicateMatcher(IPM_AtomicOrderingMMO), Order(Order),
1357 Comparator(Comparator) {}
Daniel Sanders39690bd2017-10-15 02:41:12 +00001358
1359 static bool classof(const InstructionPredicateMatcher *P) {
Daniel Sanders76664652017-11-28 22:07:05 +00001360 return P->getKind() == IPM_AtomicOrderingMMO;
Daniel Sanders39690bd2017-10-15 02:41:12 +00001361 }
1362
1363 void emitPredicateOpcodes(MatchTable &Table, RuleMatcher &Rule,
1364 unsigned InsnVarID) const override {
Daniel Sanders0c43b3a2017-11-30 21:05:59 +00001365 StringRef Opcode = "GIM_CheckAtomicOrdering";
1366
1367 if (Comparator == AO_OrStronger)
1368 Opcode = "GIM_CheckAtomicOrderingOrStrongerThan";
1369 if (Comparator == AO_WeakerThan)
1370 Opcode = "GIM_CheckAtomicOrderingWeakerThan";
1371
1372 Table << MatchTable::Opcode(Opcode) << MatchTable::Comment("MI")
1373 << MatchTable::IntValue(InsnVarID) << MatchTable::Comment("Order")
Daniel Sanders76664652017-11-28 22:07:05 +00001374 << MatchTable::NamedValue(("(int64_t)AtomicOrdering::" + Order).str())
Daniel Sanders39690bd2017-10-15 02:41:12 +00001375 << MatchTable::LineBreak;
1376 }
1377};
1378
Daniel Sandersdc662ff2017-01-26 11:10:14 +00001379/// Generates code to check that a set of predicates and operands match for a
1380/// particular instruction.
1381///
1382/// Typical predicates include:
1383/// * Has a specific opcode.
1384/// * Has an nsw/nuw flag or doesn't.
1385class InstructionMatcher
1386 : public PredicateListMatcher<InstructionPredicateMatcher> {
1387protected:
Daniel Sanders4f3eb242017-04-05 13:14:03 +00001388 typedef std::vector<std::unique_ptr<OperandMatcher>> OperandVec;
Daniel Sanders066ebbf2017-02-24 15:43:30 +00001389
Daniel Sandersbfa9e2c2017-10-14 00:31:58 +00001390 RuleMatcher &Rule;
1391
Daniel Sanders066ebbf2017-02-24 15:43:30 +00001392 /// The operands to match. All rendered operands must be present even if the
1393 /// condition is always true.
1394 OperandVec Operands;
Daniel Sandersdc662ff2017-01-26 11:10:14 +00001395
Daniel Sanders05540042017-08-08 10:44:31 +00001396 std::string SymbolicName;
1397
Daniel Sandersdc662ff2017-01-26 11:10:14 +00001398public:
Daniel Sandersbfa9e2c2017-10-14 00:31:58 +00001399 InstructionMatcher(RuleMatcher &Rule, StringRef SymbolicName)
1400 : Rule(Rule), SymbolicName(SymbolicName) {}
1401
1402 RuleMatcher &getRuleMatcher() const { return Rule; }
Daniel Sanders05540042017-08-08 10:44:31 +00001403
Daniel Sandersdc662ff2017-01-26 11:10:14 +00001404 /// Add an operand to the matcher.
Daniel Sanders4f3eb242017-04-05 13:14:03 +00001405 OperandMatcher &addOperand(unsigned OpIdx, const std::string &SymbolicName,
1406 unsigned AllocatedTemporariesBaseID) {
1407 Operands.emplace_back(new OperandMatcher(*this, OpIdx, SymbolicName,
1408 AllocatedTemporariesBaseID));
Daniel Sandersbfa9e2c2017-10-14 00:31:58 +00001409 if (!SymbolicName.empty())
1410 Rule.defineOperand(SymbolicName, *Operands.back());
1411
Daniel Sanders4f3eb242017-04-05 13:14:03 +00001412 return *Operands.back();
Daniel Sandersdc662ff2017-01-26 11:10:14 +00001413 }
1414
Daniel Sandersffc7d582017-03-29 15:37:18 +00001415 OperandMatcher &getOperand(unsigned OpIdx) {
1416 auto I = std::find_if(Operands.begin(), Operands.end(),
Daniel Sanders4f3eb242017-04-05 13:14:03 +00001417 [&OpIdx](const std::unique_ptr<OperandMatcher> &X) {
1418 return X->getOperandIndex() == OpIdx;
Daniel Sandersffc7d582017-03-29 15:37:18 +00001419 });
1420 if (I != Operands.end())
Daniel Sanders4f3eb242017-04-05 13:14:03 +00001421 return **I;
Daniel Sandersffc7d582017-03-29 15:37:18 +00001422 llvm_unreachable("Failed to lookup operand");
1423 }
1424
Daniel Sanders05540042017-08-08 10:44:31 +00001425 StringRef getSymbolicName() const { return SymbolicName; }
Daniel Sanders066ebbf2017-02-24 15:43:30 +00001426 unsigned getNumOperands() const { return Operands.size(); }
Daniel Sandersbee57392017-04-04 13:25:23 +00001427 OperandVec::iterator operands_begin() { return Operands.begin(); }
1428 OperandVec::iterator operands_end() { return Operands.end(); }
1429 iterator_range<OperandVec::iterator> operands() {
1430 return make_range(operands_begin(), operands_end());
1431 }
Daniel Sandersffc7d582017-03-29 15:37:18 +00001432 OperandVec::const_iterator operands_begin() const { return Operands.begin(); }
1433 OperandVec::const_iterator operands_end() const { return Operands.end(); }
Daniel Sanders066ebbf2017-02-24 15:43:30 +00001434 iterator_range<OperandVec::const_iterator> operands() const {
1435 return make_range(operands_begin(), operands_end());
1436 }
1437
Daniel Sanders9d662d22017-07-06 10:06:12 +00001438 /// Emit MatchTable opcodes to check the shape of the match and capture
1439 /// instructions into the MIs table.
Daniel Sanders7aac7cc2017-07-20 09:25:44 +00001440 void emitCaptureOpcodes(MatchTable &Table, RuleMatcher &Rule,
Daniel Sanders9d662d22017-07-06 10:06:12 +00001441 unsigned InsnID) {
Daniel Sanders7aac7cc2017-07-20 09:25:44 +00001442 Table << MatchTable::Opcode("GIM_CheckNumOperands")
1443 << MatchTable::Comment("MI") << MatchTable::IntValue(InsnID)
1444 << MatchTable::Comment("Expected")
1445 << MatchTable::IntValue(getNumOperands()) << MatchTable::LineBreak;
Daniel Sanders6ab0daa2017-07-04 14:35:06 +00001446 for (const auto &Operand : Operands)
Daniel Sanders7aac7cc2017-07-20 09:25:44 +00001447 Operand->emitCaptureOpcodes(Table, Rule, InsnID);
Daniel Sandersb96f40d2017-03-20 15:20:42 +00001448 }
1449
Daniel Sanders9d662d22017-07-06 10:06:12 +00001450 /// Emit MatchTable opcodes that test whether the instruction named in
Daniel Sandersdc662ff2017-01-26 11:10:14 +00001451 /// InsnVarName matches all the predicates and all the operands.
Daniel Sanders7aac7cc2017-07-20 09:25:44 +00001452 void emitPredicateOpcodes(MatchTable &Table, RuleMatcher &Rule,
Daniel Sanders6ab0daa2017-07-04 14:35:06 +00001453 unsigned InsnVarID) const {
Daniel Sanders7aac7cc2017-07-20 09:25:44 +00001454 emitPredicateListOpcodes(Table, Rule, InsnVarID);
Daniel Sanders6ab0daa2017-07-04 14:35:06 +00001455 for (const auto &Operand : Operands)
Daniel Sanders7aac7cc2017-07-20 09:25:44 +00001456 Operand->emitPredicateOpcodes(Table, Rule, InsnVarID);
Ahmed Bougacha36f70352016-12-21 23:26:20 +00001457 }
Daniel Sanders759ff412017-02-24 13:58:11 +00001458
1459 /// Compare the priority of this object and B.
1460 ///
1461 /// Returns true if this object is more important than B.
1462 bool isHigherPriorityThan(const InstructionMatcher &B) const {
1463 // Instruction matchers involving more operands have higher priority.
1464 if (Operands.size() > B.Operands.size())
1465 return true;
1466 if (Operands.size() < B.Operands.size())
1467 return false;
1468
1469 for (const auto &Predicate : zip(predicates(), B.predicates())) {
1470 if (std::get<0>(Predicate)->isHigherPriorityThan(*std::get<1>(Predicate)))
1471 return true;
1472 if (std::get<1>(Predicate)->isHigherPriorityThan(*std::get<0>(Predicate)))
1473 return false;
1474 }
1475
1476 for (const auto &Operand : zip(Operands, B.Operands)) {
Daniel Sanders4f3eb242017-04-05 13:14:03 +00001477 if (std::get<0>(Operand)->isHigherPriorityThan(*std::get<1>(Operand)))
Daniel Sanders759ff412017-02-24 13:58:11 +00001478 return true;
Daniel Sanders4f3eb242017-04-05 13:14:03 +00001479 if (std::get<1>(Operand)->isHigherPriorityThan(*std::get<0>(Operand)))
Daniel Sanders759ff412017-02-24 13:58:11 +00001480 return false;
1481 }
1482
1483 return false;
1484 };
Daniel Sanders8a4bae92017-03-14 21:32:08 +00001485
1486 /// Report the maximum number of temporary operands needed by the instruction
1487 /// matcher.
Daniel Sanders2deea182017-04-22 15:11:04 +00001488 unsigned countRendererFns() const {
Daniel Sanders8a4bae92017-03-14 21:32:08 +00001489 return std::accumulate(predicates().begin(), predicates().end(), 0,
1490 [](unsigned A,
1491 const std::unique_ptr<InstructionPredicateMatcher>
1492 &Predicate) {
Daniel Sanders2deea182017-04-22 15:11:04 +00001493 return A + Predicate->countRendererFns();
Daniel Sanders8a4bae92017-03-14 21:32:08 +00001494 }) +
Daniel Sanders4f3eb242017-04-05 13:14:03 +00001495 std::accumulate(
1496 Operands.begin(), Operands.end(), 0,
1497 [](unsigned A, const std::unique_ptr<OperandMatcher> &Operand) {
Daniel Sanders2deea182017-04-22 15:11:04 +00001498 return A + Operand->countRendererFns();
Daniel Sanders4f3eb242017-04-05 13:14:03 +00001499 });
Daniel Sanders8a4bae92017-03-14 21:32:08 +00001500 }
Daniel Sanders05540042017-08-08 10:44:31 +00001501
1502 bool isConstantInstruction() const {
1503 for (const auto &P : predicates())
1504 if (const InstructionOpcodeMatcher *Opcode =
1505 dyn_cast<InstructionOpcodeMatcher>(P.get()))
1506 return Opcode->isConstantInstruction();
1507 return false;
1508 }
Ahmed Bougacha36f70352016-12-21 23:26:20 +00001509};
1510
Daniel Sandersbee57392017-04-04 13:25:23 +00001511/// Generates code to check that the operand is a register defined by an
1512/// instruction that matches the given instruction matcher.
1513///
1514/// For example, the pattern:
1515/// (set $dst, (G_MUL (G_ADD $src1, $src2), $src3))
1516/// would use an InstructionOperandMatcher for operand 1 of the G_MUL to match
1517/// the:
1518/// (G_ADD $src1, $src2)
1519/// subpattern.
1520class InstructionOperandMatcher : public OperandPredicateMatcher {
1521protected:
1522 std::unique_ptr<InstructionMatcher> InsnMatcher;
1523
1524public:
Daniel Sandersbfa9e2c2017-10-14 00:31:58 +00001525 InstructionOperandMatcher(RuleMatcher &Rule, StringRef SymbolicName)
Daniel Sandersbee57392017-04-04 13:25:23 +00001526 : OperandPredicateMatcher(OPM_Instruction),
Daniel Sandersbfa9e2c2017-10-14 00:31:58 +00001527 InsnMatcher(new InstructionMatcher(Rule, SymbolicName)) {}
Daniel Sandersbee57392017-04-04 13:25:23 +00001528
1529 static bool classof(const OperandPredicateMatcher *P) {
1530 return P->getKind() == OPM_Instruction;
1531 }
1532
1533 InstructionMatcher &getInsnMatcher() const { return *InsnMatcher; }
1534
Daniel Sanders7aac7cc2017-07-20 09:25:44 +00001535 void emitCaptureOpcodes(MatchTable &Table, RuleMatcher &Rule,
Daniel Sanders9d662d22017-07-06 10:06:12 +00001536 unsigned InsnID, unsigned OpIdx) const override {
Daniel Sanders7aac7cc2017-07-20 09:25:44 +00001537 unsigned InsnVarID = Rule.defineInsnVar(Table, *InsnMatcher, InsnID, OpIdx);
1538 InsnMatcher->emitCaptureOpcodes(Table, Rule, InsnVarID);
Daniel Sandersbee57392017-04-04 13:25:23 +00001539 }
1540
Daniel Sanders7aac7cc2017-07-20 09:25:44 +00001541 void emitPredicateOpcodes(MatchTable &Table, RuleMatcher &Rule,
Daniel Sanders6ab0daa2017-07-04 14:35:06 +00001542 unsigned InsnVarID_,
1543 unsigned OpIdx_) const override {
1544 unsigned InsnVarID = Rule.getInsnVarID(*InsnMatcher);
Daniel Sanders7aac7cc2017-07-20 09:25:44 +00001545 InsnMatcher->emitPredicateOpcodes(Table, Rule, InsnVarID);
Daniel Sandersbee57392017-04-04 13:25:23 +00001546 }
1547};
1548
Daniel Sanders43c882c2017-02-01 10:53:10 +00001549//===- Actions ------------------------------------------------------------===//
Daniel Sanders066ebbf2017-02-24 15:43:30 +00001550class OperandRenderer {
1551public:
Daniel Sanderscc36dbf2017-06-27 10:11:39 +00001552 enum RendererKind {
1553 OR_Copy,
Daniel Sandersd66e0902017-10-23 18:19:24 +00001554 OR_CopyOrAddZeroReg,
Daniel Sanderscc36dbf2017-06-27 10:11:39 +00001555 OR_CopySubReg,
Daniel Sanders05540042017-08-08 10:44:31 +00001556 OR_CopyConstantAsImm,
Daniel Sanders11300ce2017-10-13 21:28:03 +00001557 OR_CopyFConstantAsFPImm,
Daniel Sanderscc36dbf2017-06-27 10:11:39 +00001558 OR_Imm,
1559 OR_Register,
Daniel Sanders9cbe7c72017-11-01 19:57:57 +00001560 OR_TempRegister,
Daniel Sanderscc36dbf2017-06-27 10:11:39 +00001561 OR_ComplexPattern
1562 };
Daniel Sanders066ebbf2017-02-24 15:43:30 +00001563
1564protected:
1565 RendererKind Kind;
1566
1567public:
1568 OperandRenderer(RendererKind Kind) : Kind(Kind) {}
1569 virtual ~OperandRenderer() {}
1570
1571 RendererKind getKind() const { return Kind; }
1572
Daniel Sanders7aac7cc2017-07-20 09:25:44 +00001573 virtual void emitRenderOpcodes(MatchTable &Table,
1574 RuleMatcher &Rule) const = 0;
Daniel Sanders066ebbf2017-02-24 15:43:30 +00001575};
1576
1577/// A CopyRenderer emits code to copy a single operand from an existing
1578/// instruction to the one being built.
1579class CopyRenderer : public OperandRenderer {
1580protected:
Daniel Sandersd93a35a2017-07-05 09:39:33 +00001581 unsigned NewInsnID;
Daniel Sanders066ebbf2017-02-24 15:43:30 +00001582 /// The name of the operand.
1583 const StringRef SymbolicName;
1584
1585public:
Daniel Sandersbd83ad42017-10-24 01:48:34 +00001586 CopyRenderer(unsigned NewInsnID, StringRef SymbolicName)
1587 : OperandRenderer(OR_Copy), NewInsnID(NewInsnID),
Daniel Sanders05540042017-08-08 10:44:31 +00001588 SymbolicName(SymbolicName) {
1589 assert(!SymbolicName.empty() && "Cannot copy from an unspecified source");
1590 }
Daniel Sanders066ebbf2017-02-24 15:43:30 +00001591
1592 static bool classof(const OperandRenderer *R) {
1593 return R->getKind() == OR_Copy;
1594 }
1595
1596 const StringRef getSymbolicName() const { return SymbolicName; }
1597
Daniel Sanders7aac7cc2017-07-20 09:25:44 +00001598 void emitRenderOpcodes(MatchTable &Table, RuleMatcher &Rule) const override {
Daniel Sandersbfa9e2c2017-10-14 00:31:58 +00001599 const OperandMatcher &Operand = Rule.getOperandMatcher(SymbolicName);
Daniel Sandersd93a35a2017-07-05 09:39:33 +00001600 unsigned OldInsnVarID = Rule.getInsnVarID(Operand.getInstructionMatcher());
Daniel Sanders7aac7cc2017-07-20 09:25:44 +00001601 Table << MatchTable::Opcode("GIR_Copy") << MatchTable::Comment("NewInsnID")
1602 << MatchTable::IntValue(NewInsnID) << MatchTable::Comment("OldInsnID")
1603 << MatchTable::IntValue(OldInsnVarID) << MatchTable::Comment("OpIdx")
1604 << MatchTable::IntValue(Operand.getOperandIndex())
1605 << MatchTable::Comment(SymbolicName) << MatchTable::LineBreak;
Daniel Sanders066ebbf2017-02-24 15:43:30 +00001606 }
1607};
1608
Daniel Sandersd66e0902017-10-23 18:19:24 +00001609/// A CopyOrAddZeroRegRenderer emits code to copy a single operand from an
1610/// existing instruction to the one being built. If the operand turns out to be
1611/// a 'G_CONSTANT 0' then it replaces the operand with a zero register.
1612class CopyOrAddZeroRegRenderer : public OperandRenderer {
1613protected:
1614 unsigned NewInsnID;
1615 /// The name of the operand.
1616 const StringRef SymbolicName;
1617 const Record *ZeroRegisterDef;
1618
1619public:
1620 CopyOrAddZeroRegRenderer(unsigned NewInsnID,
Daniel Sandersd66e0902017-10-23 18:19:24 +00001621 StringRef SymbolicName, Record *ZeroRegisterDef)
1622 : OperandRenderer(OR_CopyOrAddZeroReg), NewInsnID(NewInsnID),
1623 SymbolicName(SymbolicName), ZeroRegisterDef(ZeroRegisterDef) {
1624 assert(!SymbolicName.empty() && "Cannot copy from an unspecified source");
1625 }
1626
1627 static bool classof(const OperandRenderer *R) {
1628 return R->getKind() == OR_CopyOrAddZeroReg;
1629 }
1630
1631 const StringRef getSymbolicName() const { return SymbolicName; }
1632
1633 void emitRenderOpcodes(MatchTable &Table, RuleMatcher &Rule) const override {
1634 const OperandMatcher &Operand = Rule.getOperandMatcher(SymbolicName);
1635 unsigned OldInsnVarID = Rule.getInsnVarID(Operand.getInstructionMatcher());
1636 Table << MatchTable::Opcode("GIR_CopyOrAddZeroReg")
1637 << MatchTable::Comment("NewInsnID") << MatchTable::IntValue(NewInsnID)
1638 << MatchTable::Comment("OldInsnID")
1639 << MatchTable::IntValue(OldInsnVarID) << MatchTable::Comment("OpIdx")
1640 << MatchTable::IntValue(Operand.getOperandIndex())
1641 << MatchTable::NamedValue(
1642 (ZeroRegisterDef->getValue("Namespace")
1643 ? ZeroRegisterDef->getValueAsString("Namespace")
1644 : ""),
1645 ZeroRegisterDef->getName())
1646 << MatchTable::Comment(SymbolicName) << MatchTable::LineBreak;
1647 }
1648};
1649
Daniel Sanders05540042017-08-08 10:44:31 +00001650/// A CopyConstantAsImmRenderer emits code to render a G_CONSTANT instruction to
1651/// an extended immediate operand.
1652class CopyConstantAsImmRenderer : public OperandRenderer {
1653protected:
1654 unsigned NewInsnID;
1655 /// The name of the operand.
1656 const std::string SymbolicName;
1657 bool Signed;
1658
1659public:
1660 CopyConstantAsImmRenderer(unsigned NewInsnID, StringRef SymbolicName)
1661 : OperandRenderer(OR_CopyConstantAsImm), NewInsnID(NewInsnID),
1662 SymbolicName(SymbolicName), Signed(true) {}
1663
1664 static bool classof(const OperandRenderer *R) {
1665 return R->getKind() == OR_CopyConstantAsImm;
1666 }
1667
1668 const StringRef getSymbolicName() const { return SymbolicName; }
1669
1670 void emitRenderOpcodes(MatchTable &Table, RuleMatcher &Rule) const override {
1671 const InstructionMatcher &InsnMatcher = Rule.getInstructionMatcher(SymbolicName);
1672 unsigned OldInsnVarID = Rule.getInsnVarID(InsnMatcher);
1673 Table << MatchTable::Opcode(Signed ? "GIR_CopyConstantAsSImm"
1674 : "GIR_CopyConstantAsUImm")
1675 << MatchTable::Comment("NewInsnID") << MatchTable::IntValue(NewInsnID)
1676 << MatchTable::Comment("OldInsnID")
1677 << MatchTable::IntValue(OldInsnVarID)
1678 << MatchTable::Comment(SymbolicName) << MatchTable::LineBreak;
1679 }
1680};
1681
Daniel Sanders11300ce2017-10-13 21:28:03 +00001682/// A CopyFConstantAsFPImmRenderer emits code to render a G_FCONSTANT
1683/// instruction to an extended immediate operand.
1684class CopyFConstantAsFPImmRenderer : public OperandRenderer {
1685protected:
1686 unsigned NewInsnID;
1687 /// The name of the operand.
1688 const std::string SymbolicName;
1689
1690public:
1691 CopyFConstantAsFPImmRenderer(unsigned NewInsnID, StringRef SymbolicName)
1692 : OperandRenderer(OR_CopyFConstantAsFPImm), NewInsnID(NewInsnID),
1693 SymbolicName(SymbolicName) {}
1694
1695 static bool classof(const OperandRenderer *R) {
1696 return R->getKind() == OR_CopyFConstantAsFPImm;
1697 }
1698
1699 const StringRef getSymbolicName() const { return SymbolicName; }
1700
1701 void emitRenderOpcodes(MatchTable &Table, RuleMatcher &Rule) const override {
1702 const InstructionMatcher &InsnMatcher = Rule.getInstructionMatcher(SymbolicName);
1703 unsigned OldInsnVarID = Rule.getInsnVarID(InsnMatcher);
1704 Table << MatchTable::Opcode("GIR_CopyFConstantAsFPImm")
1705 << MatchTable::Comment("NewInsnID") << MatchTable::IntValue(NewInsnID)
1706 << MatchTable::Comment("OldInsnID")
1707 << MatchTable::IntValue(OldInsnVarID)
1708 << MatchTable::Comment(SymbolicName) << MatchTable::LineBreak;
1709 }
1710};
1711
Daniel Sanderscc36dbf2017-06-27 10:11:39 +00001712/// A CopySubRegRenderer emits code to copy a single register operand from an
1713/// existing instruction to the one being built and indicate that only a
1714/// subregister should be copied.
1715class CopySubRegRenderer : public OperandRenderer {
1716protected:
Daniel Sandersd93a35a2017-07-05 09:39:33 +00001717 unsigned NewInsnID;
Daniel Sanderscc36dbf2017-06-27 10:11:39 +00001718 /// The name of the operand.
1719 const StringRef SymbolicName;
1720 /// The subregister to extract.
1721 const CodeGenSubRegIndex *SubReg;
1722
1723public:
Daniel Sandersbd83ad42017-10-24 01:48:34 +00001724 CopySubRegRenderer(unsigned NewInsnID, StringRef SymbolicName,
1725 const CodeGenSubRegIndex *SubReg)
1726 : OperandRenderer(OR_CopySubReg), NewInsnID(NewInsnID),
Daniel Sanderscc36dbf2017-06-27 10:11:39 +00001727 SymbolicName(SymbolicName), SubReg(SubReg) {}
1728
1729 static bool classof(const OperandRenderer *R) {
1730 return R->getKind() == OR_CopySubReg;
1731 }
1732
1733 const StringRef getSymbolicName() const { return SymbolicName; }
1734
Daniel Sanders7aac7cc2017-07-20 09:25:44 +00001735 void emitRenderOpcodes(MatchTable &Table, RuleMatcher &Rule) const override {
Daniel Sandersbfa9e2c2017-10-14 00:31:58 +00001736 const OperandMatcher &Operand = Rule.getOperandMatcher(SymbolicName);
Daniel Sandersd93a35a2017-07-05 09:39:33 +00001737 unsigned OldInsnVarID = Rule.getInsnVarID(Operand.getInstructionMatcher());
Daniel Sanders7aac7cc2017-07-20 09:25:44 +00001738 Table << MatchTable::Opcode("GIR_CopySubReg")
1739 << MatchTable::Comment("NewInsnID") << MatchTable::IntValue(NewInsnID)
1740 << MatchTable::Comment("OldInsnID")
1741 << MatchTable::IntValue(OldInsnVarID) << MatchTable::Comment("OpIdx")
1742 << MatchTable::IntValue(Operand.getOperandIndex())
1743 << MatchTable::Comment("SubRegIdx")
1744 << MatchTable::IntValue(SubReg->EnumValue)
1745 << MatchTable::Comment(SymbolicName) << MatchTable::LineBreak;
Daniel Sanderscc36dbf2017-06-27 10:11:39 +00001746 }
1747};
1748
Daniel Sanders066ebbf2017-02-24 15:43:30 +00001749/// Adds a specific physical register to the instruction being built.
1750/// This is typically useful for WZR/XZR on AArch64.
1751class AddRegisterRenderer : public OperandRenderer {
1752protected:
Daniel Sandersd93a35a2017-07-05 09:39:33 +00001753 unsigned InsnID;
Daniel Sanders066ebbf2017-02-24 15:43:30 +00001754 const Record *RegisterDef;
1755
1756public:
Daniel Sandersd93a35a2017-07-05 09:39:33 +00001757 AddRegisterRenderer(unsigned InsnID, const Record *RegisterDef)
1758 : OperandRenderer(OR_Register), InsnID(InsnID), RegisterDef(RegisterDef) {
1759 }
Daniel Sanders066ebbf2017-02-24 15:43:30 +00001760
1761 static bool classof(const OperandRenderer *R) {
1762 return R->getKind() == OR_Register;
1763 }
1764
Daniel Sanders7aac7cc2017-07-20 09:25:44 +00001765 void emitRenderOpcodes(MatchTable &Table, RuleMatcher &Rule) const override {
1766 Table << MatchTable::Opcode("GIR_AddRegister")
1767 << MatchTable::Comment("InsnID") << MatchTable::IntValue(InsnID)
1768 << MatchTable::NamedValue(
1769 (RegisterDef->getValue("Namespace")
1770 ? RegisterDef->getValueAsString("Namespace")
1771 : ""),
1772 RegisterDef->getName())
1773 << MatchTable::LineBreak;
Daniel Sanders066ebbf2017-02-24 15:43:30 +00001774 }
1775};
1776
Daniel Sanders9cbe7c72017-11-01 19:57:57 +00001777/// Adds a specific temporary virtual register to the instruction being built.
1778/// This is used to chain instructions together when emitting multiple
1779/// instructions.
1780class TempRegRenderer : public OperandRenderer {
1781protected:
1782 unsigned InsnID;
1783 unsigned TempRegID;
1784 bool IsDef;
1785
1786public:
1787 TempRegRenderer(unsigned InsnID, unsigned TempRegID, bool IsDef = false)
1788 : OperandRenderer(OR_Register), InsnID(InsnID), TempRegID(TempRegID),
1789 IsDef(IsDef) {}
1790
1791 static bool classof(const OperandRenderer *R) {
1792 return R->getKind() == OR_TempRegister;
1793 }
1794
1795 void emitRenderOpcodes(MatchTable &Table, RuleMatcher &Rule) const override {
1796 Table << MatchTable::Opcode("GIR_AddTempRegister")
1797 << MatchTable::Comment("InsnID") << MatchTable::IntValue(InsnID)
1798 << MatchTable::Comment("TempRegID") << MatchTable::IntValue(TempRegID)
1799 << MatchTable::Comment("TempRegFlags");
1800 if (IsDef)
1801 Table << MatchTable::NamedValue("RegState::Define");
1802 else
1803 Table << MatchTable::IntValue(0);
1804 Table << MatchTable::LineBreak;
1805 }
1806};
1807
Daniel Sanders0ed28822017-04-12 08:23:08 +00001808/// Adds a specific immediate to the instruction being built.
1809class ImmRenderer : public OperandRenderer {
1810protected:
Daniel Sandersd93a35a2017-07-05 09:39:33 +00001811 unsigned InsnID;
Daniel Sanders0ed28822017-04-12 08:23:08 +00001812 int64_t Imm;
1813
1814public:
Daniel Sandersd93a35a2017-07-05 09:39:33 +00001815 ImmRenderer(unsigned InsnID, int64_t Imm)
1816 : OperandRenderer(OR_Imm), InsnID(InsnID), Imm(Imm) {}
Daniel Sanders0ed28822017-04-12 08:23:08 +00001817
1818 static bool classof(const OperandRenderer *R) {
1819 return R->getKind() == OR_Imm;
1820 }
1821
Daniel Sanders7aac7cc2017-07-20 09:25:44 +00001822 void emitRenderOpcodes(MatchTable &Table, RuleMatcher &Rule) const override {
1823 Table << MatchTable::Opcode("GIR_AddImm") << MatchTable::Comment("InsnID")
1824 << MatchTable::IntValue(InsnID) << MatchTable::Comment("Imm")
1825 << MatchTable::IntValue(Imm) << MatchTable::LineBreak;
Daniel Sanders0ed28822017-04-12 08:23:08 +00001826 }
1827};
1828
Daniel Sanders2deea182017-04-22 15:11:04 +00001829/// Adds operands by calling a renderer function supplied by the ComplexPattern
1830/// matcher function.
Daniel Sanders8a4bae92017-03-14 21:32:08 +00001831class RenderComplexPatternOperand : public OperandRenderer {
1832private:
Daniel Sandersd93a35a2017-07-05 09:39:33 +00001833 unsigned InsnID;
Daniel Sanders8a4bae92017-03-14 21:32:08 +00001834 const Record &TheDef;
Daniel Sanders2deea182017-04-22 15:11:04 +00001835 /// The name of the operand.
1836 const StringRef SymbolicName;
1837 /// The renderer number. This must be unique within a rule since it's used to
1838 /// identify a temporary variable to hold the renderer function.
1839 unsigned RendererID;
Daniel Sandersdf39cba2017-10-15 18:22:54 +00001840 /// When provided, this is the suboperand of the ComplexPattern operand to
1841 /// render. Otherwise all the suboperands will be rendered.
1842 Optional<unsigned> SubOperand;
Daniel Sanders8a4bae92017-03-14 21:32:08 +00001843
1844 unsigned getNumOperands() const {
1845 return TheDef.getValueAsDag("Operands")->getNumArgs();
1846 }
1847
1848public:
Daniel Sandersd93a35a2017-07-05 09:39:33 +00001849 RenderComplexPatternOperand(unsigned InsnID, const Record &TheDef,
Daniel Sandersdf39cba2017-10-15 18:22:54 +00001850 StringRef SymbolicName, unsigned RendererID,
1851 Optional<unsigned> SubOperand = None)
Daniel Sandersd93a35a2017-07-05 09:39:33 +00001852 : OperandRenderer(OR_ComplexPattern), InsnID(InsnID), TheDef(TheDef),
Daniel Sandersdf39cba2017-10-15 18:22:54 +00001853 SymbolicName(SymbolicName), RendererID(RendererID),
1854 SubOperand(SubOperand) {}
Daniel Sanders8a4bae92017-03-14 21:32:08 +00001855
1856 static bool classof(const OperandRenderer *R) {
1857 return R->getKind() == OR_ComplexPattern;
1858 }
1859
Daniel Sanders7aac7cc2017-07-20 09:25:44 +00001860 void emitRenderOpcodes(MatchTable &Table, RuleMatcher &Rule) const override {
Daniel Sandersdf39cba2017-10-15 18:22:54 +00001861 Table << MatchTable::Opcode(SubOperand.hasValue() ? "GIR_ComplexSubOperandRenderer"
1862 : "GIR_ComplexRenderer")
Daniel Sanders7aac7cc2017-07-20 09:25:44 +00001863 << MatchTable::Comment("InsnID") << MatchTable::IntValue(InsnID)
1864 << MatchTable::Comment("RendererID")
Daniel Sandersdf39cba2017-10-15 18:22:54 +00001865 << MatchTable::IntValue(RendererID);
1866 if (SubOperand.hasValue())
1867 Table << MatchTable::Comment("SubOperand")
1868 << MatchTable::IntValue(SubOperand.getValue());
1869 Table << MatchTable::Comment(SymbolicName) << MatchTable::LineBreak;
Daniel Sanders8a4bae92017-03-14 21:32:08 +00001870 }
1871};
1872
Ahmed Bougacha56ca3a92017-02-04 00:47:10 +00001873/// An action taken when all Matcher predicates succeeded for a parent rule.
1874///
1875/// Typical actions include:
1876/// * Changing the opcode of an instruction.
1877/// * Adding an operand to an instruction.
Daniel Sanders43c882c2017-02-01 10:53:10 +00001878class MatchAction {
1879public:
1880 virtual ~MatchAction() {}
Daniel Sanders066ebbf2017-02-24 15:43:30 +00001881
Daniel Sanders7aac7cc2017-07-20 09:25:44 +00001882 /// Emit the MatchTable opcodes to implement the action.
Daniel Sandersa7b75262017-10-31 18:50:24 +00001883 virtual void emitActionOpcodes(MatchTable &Table,
1884 RuleMatcher &Rule) const = 0;
Daniel Sanders43c882c2017-02-01 10:53:10 +00001885};
1886
Ahmed Bougacha9aa4c102017-02-04 00:47:08 +00001887/// Generates a comment describing the matched rule being acted upon.
1888class DebugCommentAction : public MatchAction {
1889private:
Daniel Sanders6ea17ed2017-10-31 18:07:03 +00001890 std::string S;
Ahmed Bougacha9aa4c102017-02-04 00:47:08 +00001891
1892public:
Daniel Sanders6ea17ed2017-10-31 18:07:03 +00001893 DebugCommentAction(StringRef S) : S(S) {}
Ahmed Bougacha9aa4c102017-02-04 00:47:08 +00001894
Daniel Sandersa7b75262017-10-31 18:50:24 +00001895 void emitActionOpcodes(MatchTable &Table, RuleMatcher &Rule) const override {
Daniel Sanders6ea17ed2017-10-31 18:07:03 +00001896 Table << MatchTable::Comment(S) << MatchTable::LineBreak;
Ahmed Bougacha9aa4c102017-02-04 00:47:08 +00001897 }
1898};
1899
Daniel Sanders066ebbf2017-02-24 15:43:30 +00001900/// Generates code to build an instruction or mutate an existing instruction
1901/// into the desired instruction when this is possible.
1902class BuildMIAction : public MatchAction {
Daniel Sanders43c882c2017-02-01 10:53:10 +00001903private:
Daniel Sandersd93a35a2017-07-05 09:39:33 +00001904 unsigned InsnID;
Ahmed Bougacha36f70352016-12-21 23:26:20 +00001905 const CodeGenInstruction *I;
Daniel Sanders64f745c2017-10-24 17:08:43 +00001906 const InstructionMatcher *Matched;
Daniel Sanders066ebbf2017-02-24 15:43:30 +00001907 std::vector<std::unique_ptr<OperandRenderer>> OperandRenderers;
1908
1909 /// True if the instruction can be built solely by mutating the opcode.
Daniel Sandersa7b75262017-10-31 18:50:24 +00001910 bool canMutate(RuleMatcher &Rule, const InstructionMatcher *Insn) const {
1911 if (!Insn)
Daniel Sandersab1d1192017-10-24 18:11:54 +00001912 return false;
1913
Daniel Sandersa7b75262017-10-31 18:50:24 +00001914 if (OperandRenderers.size() != Insn->getNumOperands())
Daniel Sanderse9fdba32017-04-29 17:30:09 +00001915 return false;
1916
Daniel Sanders066ebbf2017-02-24 15:43:30 +00001917 for (const auto &Renderer : enumerate(OperandRenderers)) {
Zachary Turner309a0882017-03-13 16:24:10 +00001918 if (const auto *Copy = dyn_cast<CopyRenderer>(&*Renderer.value())) {
Daniel Sandersbfa9e2c2017-10-14 00:31:58 +00001919 const OperandMatcher &OM = Rule.getOperandMatcher(Copy->getSymbolicName());
Daniel Sandersa7b75262017-10-31 18:50:24 +00001920 if (Insn != &OM.getInstructionMatcher() ||
Daniel Sanders3016d3c2017-04-22 14:31:28 +00001921 OM.getOperandIndex() != Renderer.index())
Daniel Sanders066ebbf2017-02-24 15:43:30 +00001922 return false;
1923 } else
1924 return false;
1925 }
1926
1927 return true;
1928 }
Ahmed Bougacha36f70352016-12-21 23:26:20 +00001929
Daniel Sanders43c882c2017-02-01 10:53:10 +00001930public:
Daniel Sandersa7b75262017-10-31 18:50:24 +00001931 BuildMIAction(unsigned InsnID, const CodeGenInstruction *I)
1932 : InsnID(InsnID), I(I), Matched(nullptr) {}
1933
Daniel Sandersdf258e32017-10-31 19:09:29 +00001934 const CodeGenInstruction *getCGI() const { return I; }
1935
Daniel Sandersa7b75262017-10-31 18:50:24 +00001936 void chooseInsnToMutate(RuleMatcher &Rule) {
1937 for (const auto *MutateCandidate : Rule.mutatable_insns()) {
1938 if (canMutate(Rule, MutateCandidate)) {
1939 // Take the first one we're offered that we're able to mutate.
1940 Rule.reserveInsnMatcherForMutation(MutateCandidate);
1941 Matched = MutateCandidate;
1942 return;
1943 }
1944 }
1945 }
Daniel Sanders43c882c2017-02-01 10:53:10 +00001946
Daniel Sanders066ebbf2017-02-24 15:43:30 +00001947 template <class Kind, class... Args>
1948 Kind &addRenderer(Args&&... args) {
1949 OperandRenderers.emplace_back(
Daniel Sanders198447a2017-11-01 00:29:47 +00001950 llvm::make_unique<Kind>(InsnID, std::forward<Args>(args)...));
Daniel Sanders066ebbf2017-02-24 15:43:30 +00001951 return *static_cast<Kind *>(OperandRenderers.back().get());
1952 }
1953
Daniel Sandersa7b75262017-10-31 18:50:24 +00001954 void emitActionOpcodes(MatchTable &Table, RuleMatcher &Rule) const override {
1955 if (Matched) {
1956 assert(canMutate(Rule, Matched) &&
1957 "Arranged to mutate an insn that isn't mutatable");
1958
1959 unsigned RecycleInsnID = Rule.getInsnVarID(*Matched);
Daniel Sanders7aac7cc2017-07-20 09:25:44 +00001960 Table << MatchTable::Opcode("GIR_MutateOpcode")
1961 << MatchTable::Comment("InsnID") << MatchTable::IntValue(InsnID)
1962 << MatchTable::Comment("RecycleInsnID")
1963 << MatchTable::IntValue(RecycleInsnID)
1964 << MatchTable::Comment("Opcode")
1965 << MatchTable::NamedValue(I->Namespace, I->TheDef->getName())
1966 << MatchTable::LineBreak;
Tim Northover4340d642017-03-20 21:58:23 +00001967
1968 if (!I->ImplicitDefs.empty() || !I->ImplicitUses.empty()) {
Tim Northover4340d642017-03-20 21:58:23 +00001969 for (auto Def : I->ImplicitDefs) {
Diana Picus8abcbbb2017-05-02 09:40:49 +00001970 auto Namespace = Def->getValue("Namespace")
1971 ? Def->getValueAsString("Namespace")
1972 : "";
Daniel Sanders7aac7cc2017-07-20 09:25:44 +00001973 Table << MatchTable::Opcode("GIR_AddImplicitDef")
1974 << MatchTable::Comment("InsnID") << MatchTable::IntValue(InsnID)
1975 << MatchTable::NamedValue(Namespace, Def->getName())
1976 << MatchTable::LineBreak;
Tim Northover4340d642017-03-20 21:58:23 +00001977 }
1978 for (auto Use : I->ImplicitUses) {
Diana Picus8abcbbb2017-05-02 09:40:49 +00001979 auto Namespace = Use->getValue("Namespace")
1980 ? Use->getValueAsString("Namespace")
1981 : "";
Daniel Sanders7aac7cc2017-07-20 09:25:44 +00001982 Table << MatchTable::Opcode("GIR_AddImplicitUse")
1983 << MatchTable::Comment("InsnID") << MatchTable::IntValue(InsnID)
1984 << MatchTable::NamedValue(Namespace, Use->getName())
1985 << MatchTable::LineBreak;
Tim Northover4340d642017-03-20 21:58:23 +00001986 }
1987 }
Daniel Sanders066ebbf2017-02-24 15:43:30 +00001988 return;
1989 }
1990
1991 // TODO: Simple permutation looks like it could be almost as common as
1992 // mutation due to commutative operations.
1993
Daniel Sanders7aac7cc2017-07-20 09:25:44 +00001994 Table << MatchTable::Opcode("GIR_BuildMI") << MatchTable::Comment("InsnID")
1995 << MatchTable::IntValue(InsnID) << MatchTable::Comment("Opcode")
1996 << MatchTable::NamedValue(I->Namespace, I->TheDef->getName())
1997 << MatchTable::LineBreak;
Daniel Sanders066ebbf2017-02-24 15:43:30 +00001998 for (const auto &Renderer : OperandRenderers)
Daniel Sanders7aac7cc2017-07-20 09:25:44 +00001999 Renderer->emitRenderOpcodes(Table, Rule);
Daniel Sandersd93a35a2017-07-05 09:39:33 +00002000
Florian Hahn3bc3ec62017-08-03 14:48:22 +00002001 if (I->mayLoad || I->mayStore) {
2002 Table << MatchTable::Opcode("GIR_MergeMemOperands")
2003 << MatchTable::Comment("InsnID") << MatchTable::IntValue(InsnID)
2004 << MatchTable::Comment("MergeInsnID's");
2005 // Emit the ID's for all the instructions that are matched by this rule.
2006 // TODO: Limit this to matched instructions that mayLoad/mayStore or have
2007 // some other means of having a memoperand. Also limit this to
2008 // emitted instructions that expect to have a memoperand too. For
2009 // example, (G_SEXT (G_LOAD x)) that results in separate load and
2010 // sign-extend instructions shouldn't put the memoperand on the
2011 // sign-extend since it has no effect there.
2012 std::vector<unsigned> MergeInsnIDs;
2013 for (const auto &IDMatcherPair : Rule.defined_insn_vars())
2014 MergeInsnIDs.push_back(IDMatcherPair.second);
2015 std::sort(MergeInsnIDs.begin(), MergeInsnIDs.end());
2016 for (const auto &MergeInsnID : MergeInsnIDs)
2017 Table << MatchTable::IntValue(MergeInsnID);
Daniel Sanders05540042017-08-08 10:44:31 +00002018 Table << MatchTable::NamedValue("GIU_MergeMemOperands_EndOfList")
2019 << MatchTable::LineBreak;
Florian Hahn3bc3ec62017-08-03 14:48:22 +00002020 }
2021
Daniel Sanders9cbe7c72017-11-01 19:57:57 +00002022 // FIXME: This is a hack but it's sufficient for ISel. We'll need to do
2023 // better for combines. Particularly when there are multiple match
2024 // roots.
2025 if (InsnID == 0)
2026 Table << MatchTable::Opcode("GIR_EraseFromParent")
2027 << MatchTable::Comment("InsnID") << MatchTable::IntValue(InsnID)
2028 << MatchTable::LineBreak;
Daniel Sandersa6e2ceb2017-06-20 12:36:34 +00002029 }
2030};
2031
2032/// Generates code to constrain the operands of an output instruction to the
2033/// register classes specified by the definition of that instruction.
2034class ConstrainOperandsToDefinitionAction : public MatchAction {
Daniel Sandersd93a35a2017-07-05 09:39:33 +00002035 unsigned InsnID;
Daniel Sandersa6e2ceb2017-06-20 12:36:34 +00002036
2037public:
Daniel Sandersd93a35a2017-07-05 09:39:33 +00002038 ConstrainOperandsToDefinitionAction(unsigned InsnID) : InsnID(InsnID) {}
Daniel Sandersa6e2ceb2017-06-20 12:36:34 +00002039
Daniel Sandersa7b75262017-10-31 18:50:24 +00002040 void emitActionOpcodes(MatchTable &Table, RuleMatcher &Rule) const override {
Daniel Sanders7aac7cc2017-07-20 09:25:44 +00002041 Table << MatchTable::Opcode("GIR_ConstrainSelectedInstOperands")
2042 << MatchTable::Comment("InsnID") << MatchTable::IntValue(InsnID)
2043 << MatchTable::LineBreak;
Daniel Sandersa6e2ceb2017-06-20 12:36:34 +00002044 }
2045};
2046
2047/// Generates code to constrain the specified operand of an output instruction
2048/// to the specified register class.
2049class ConstrainOperandToRegClassAction : public MatchAction {
Daniel Sandersd93a35a2017-07-05 09:39:33 +00002050 unsigned InsnID;
Daniel Sandersa6e2ceb2017-06-20 12:36:34 +00002051 unsigned OpIdx;
2052 const CodeGenRegisterClass &RC;
2053
2054public:
Daniel Sandersd93a35a2017-07-05 09:39:33 +00002055 ConstrainOperandToRegClassAction(unsigned InsnID, unsigned OpIdx,
Daniel Sandersa6e2ceb2017-06-20 12:36:34 +00002056 const CodeGenRegisterClass &RC)
Daniel Sandersd93a35a2017-07-05 09:39:33 +00002057 : InsnID(InsnID), OpIdx(OpIdx), RC(RC) {}
Daniel Sandersa6e2ceb2017-06-20 12:36:34 +00002058
Daniel Sandersa7b75262017-10-31 18:50:24 +00002059 void emitActionOpcodes(MatchTable &Table, RuleMatcher &Rule) const override {
Daniel Sanders7aac7cc2017-07-20 09:25:44 +00002060 Table << MatchTable::Opcode("GIR_ConstrainOperandRC")
2061 << MatchTable::Comment("InsnID") << MatchTable::IntValue(InsnID)
2062 << MatchTable::Comment("Op") << MatchTable::IntValue(OpIdx)
2063 << MatchTable::Comment("RC " + RC.getName())
2064 << MatchTable::IntValue(RC.EnumValue) << MatchTable::LineBreak;
Ahmed Bougacha36f70352016-12-21 23:26:20 +00002065 }
2066};
2067
Daniel Sanders9cbe7c72017-11-01 19:57:57 +00002068/// Generates code to create a temporary register which can be used to chain
2069/// instructions together.
2070class MakeTempRegisterAction : public MatchAction {
2071private:
2072 LLTCodeGen Ty;
2073 unsigned TempRegID;
2074
2075public:
2076 MakeTempRegisterAction(const LLTCodeGen &Ty, unsigned TempRegID)
2077 : Ty(Ty), TempRegID(TempRegID) {}
2078
2079 void emitActionOpcodes(MatchTable &Table, RuleMatcher &Rule) const override {
2080 Table << MatchTable::Opcode("GIR_MakeTempReg")
2081 << MatchTable::Comment("TempRegID") << MatchTable::IntValue(TempRegID)
2082 << MatchTable::Comment("TypeID")
2083 << MatchTable::NamedValue(Ty.getCxxEnumValue())
2084 << MatchTable::LineBreak;
2085 }
2086};
2087
Daniel Sanders05540042017-08-08 10:44:31 +00002088InstructionMatcher &RuleMatcher::addInstructionMatcher(StringRef SymbolicName) {
Daniel Sandersbfa9e2c2017-10-14 00:31:58 +00002089 Matchers.emplace_back(new InstructionMatcher(*this, SymbolicName));
Daniel Sandersa7b75262017-10-31 18:50:24 +00002090 MutatableInsns.insert(Matchers.back().get());
Daniel Sandersbdfebb82017-03-15 20:18:38 +00002091 return *Matchers.back();
2092}
Ahmed Bougacha56ca3a92017-02-04 00:47:10 +00002093
Daniel Sanderse7b0d662017-04-21 15:59:56 +00002094void RuleMatcher::addRequiredFeature(Record *Feature) {
2095 RequiredFeatures.push_back(Feature);
2096}
2097
Daniel Sanders6ab0daa2017-07-04 14:35:06 +00002098const std::vector<Record *> &RuleMatcher::getRequiredFeatures() const {
2099 return RequiredFeatures;
2100}
2101
Daniel Sanders7438b262017-10-31 23:03:18 +00002102// Emplaces an action of the specified Kind at the end of the action list.
2103//
2104// Returns a reference to the newly created action.
2105//
2106// Like std::vector::emplace_back(), may invalidate all iterators if the new
2107// size exceeds the capacity. Otherwise, only invalidates the past-the-end
2108// iterator.
Daniel Sandersbdfebb82017-03-15 20:18:38 +00002109template <class Kind, class... Args>
2110Kind &RuleMatcher::addAction(Args &&... args) {
2111 Actions.emplace_back(llvm::make_unique<Kind>(std::forward<Args>(args)...));
2112 return *static_cast<Kind *>(Actions.back().get());
2113}
Daniel Sanders9cbe7c72017-11-01 19:57:57 +00002114
Daniel Sanders7438b262017-10-31 23:03:18 +00002115// Emplaces an action of the specified Kind before the given insertion point.
2116//
2117// Returns an iterator pointing at the newly created instruction.
2118//
2119// Like std::vector::insert(), may invalidate all iterators if the new size
2120// exceeds the capacity. Otherwise, only invalidates the iterators from the
2121// insertion point onwards.
2122template <class Kind, class... Args>
2123action_iterator RuleMatcher::insertAction(action_iterator InsertPt,
2124 Args &&... args) {
Daniel Sanders9cbe7c72017-11-01 19:57:57 +00002125 return Actions.emplace(InsertPt,
2126 llvm::make_unique<Kind>(std::forward<Args>(args)...));
Daniel Sanders7438b262017-10-31 23:03:18 +00002127}
Daniel Sandersdc662ff2017-01-26 11:10:14 +00002128
Daniel Sanders6ab0daa2017-07-04 14:35:06 +00002129unsigned
2130RuleMatcher::implicitlyDefineInsnVar(const InstructionMatcher &Matcher) {
2131 unsigned NewInsnVarID = NextInsnVarID++;
2132 InsnVariableIDs[&Matcher] = NewInsnVarID;
2133 return NewInsnVarID;
Daniel Sandersb96f40d2017-03-20 15:20:42 +00002134}
2135
Daniel Sanders7aac7cc2017-07-20 09:25:44 +00002136unsigned RuleMatcher::defineInsnVar(MatchTable &Table,
Daniel Sanders6ab0daa2017-07-04 14:35:06 +00002137 const InstructionMatcher &Matcher,
2138 unsigned InsnID, unsigned OpIdx) {
2139 unsigned NewInsnVarID = implicitlyDefineInsnVar(Matcher);
Daniel Sanders7aac7cc2017-07-20 09:25:44 +00002140 Table << MatchTable::Opcode("GIM_RecordInsn")
2141 << MatchTable::Comment("DefineMI") << MatchTable::IntValue(NewInsnVarID)
2142 << MatchTable::Comment("MI") << MatchTable::IntValue(InsnID)
2143 << MatchTable::Comment("OpIdx") << MatchTable::IntValue(OpIdx)
2144 << MatchTable::Comment("MIs[" + llvm::to_string(NewInsnVarID) + "]")
2145 << MatchTable::LineBreak;
Daniel Sanders6ab0daa2017-07-04 14:35:06 +00002146 return NewInsnVarID;
2147}
2148
2149unsigned RuleMatcher::getInsnVarID(const InstructionMatcher &InsnMatcher) const {
2150 const auto &I = InsnVariableIDs.find(&InsnMatcher);
2151 if (I != InsnVariableIDs.end())
Daniel Sandersb96f40d2017-03-20 15:20:42 +00002152 return I->second;
2153 llvm_unreachable("Matched Insn was not captured in a local variable");
2154}
2155
Daniel Sandersbfa9e2c2017-10-14 00:31:58 +00002156void RuleMatcher::defineOperand(StringRef SymbolicName, OperandMatcher &OM) {
2157 if (DefinedOperands.find(SymbolicName) == DefinedOperands.end()) {
2158 DefinedOperands[SymbolicName] = &OM;
2159 return;
2160 }
2161
2162 // If the operand is already defined, then we must ensure both references in
2163 // the matcher have the exact same node.
2164 OM.addPredicate<SameOperandMatcher>(OM.getSymbolicName());
2165}
2166
Daniel Sanders05540042017-08-08 10:44:31 +00002167const InstructionMatcher &
2168RuleMatcher::getInstructionMatcher(StringRef SymbolicName) const {
2169 for (const auto &I : InsnVariableIDs)
2170 if (I.first->getSymbolicName() == SymbolicName)
2171 return *I.first;
2172 llvm_unreachable(
2173 ("Failed to lookup instruction " + SymbolicName).str().c_str());
2174}
2175
Daniel Sandersbfa9e2c2017-10-14 00:31:58 +00002176const OperandMatcher &
2177RuleMatcher::getOperandMatcher(StringRef Name) const {
2178 const auto &I = DefinedOperands.find(Name);
2179
2180 if (I == DefinedOperands.end())
2181 PrintFatalError(SrcLoc, "Operand " + Name + " was not declared in matcher");
2182
2183 return *I->second;
2184}
2185
Daniel Sanders9d662d22017-07-06 10:06:12 +00002186/// Emit MatchTable opcodes to check the shape of the match and capture
Daniel Sandersb96f40d2017-03-20 15:20:42 +00002187/// instructions into local variables.
Daniel Sanders7aac7cc2017-07-20 09:25:44 +00002188void RuleMatcher::emitCaptureOpcodes(MatchTable &Table) {
Daniel Sandersb96f40d2017-03-20 15:20:42 +00002189 assert(Matchers.size() == 1 && "Cannot handle multi-root matchers yet");
Daniel Sanders6ab0daa2017-07-04 14:35:06 +00002190 unsigned InsnVarID = implicitlyDefineInsnVar(*Matchers.front());
Daniel Sanders7aac7cc2017-07-20 09:25:44 +00002191 Matchers.front()->emitCaptureOpcodes(Table, *this, InsnVarID);
Daniel Sandersb96f40d2017-03-20 15:20:42 +00002192}
2193
Daniel Sanders8e82af22017-07-27 11:03:45 +00002194void RuleMatcher::emit(MatchTable &Table) {
Daniel Sandersbdfebb82017-03-15 20:18:38 +00002195 if (Matchers.empty())
2196 llvm_unreachable("Unexpected empty matcher!");
Daniel Sandersdc662ff2017-01-26 11:10:14 +00002197
Daniel Sandersbdfebb82017-03-15 20:18:38 +00002198 // The representation supports rules that require multiple roots such as:
2199 // %ptr(p0) = ...
2200 // %elt0(s32) = G_LOAD %ptr
2201 // %1(p0) = G_ADD %ptr, 4
2202 // %elt1(s32) = G_LOAD p0 %1
2203 // which could be usefully folded into:
2204 // %ptr(p0) = ...
2205 // %elt0(s32), %elt1(s32) = TGT_LOAD_PAIR %ptr
2206 // on some targets but we don't need to make use of that yet.
2207 assert(Matchers.size() == 1 && "Cannot handle multi-root matchers yet");
Daniel Sanderse7b0d662017-04-21 15:59:56 +00002208
Daniel Sanders8e82af22017-07-27 11:03:45 +00002209 unsigned LabelID = Table.allocateLabelID();
Daniel Sanders7aac7cc2017-07-20 09:25:44 +00002210 Table << MatchTable::Opcode("GIM_Try", +1)
Daniel Sanders8e82af22017-07-27 11:03:45 +00002211 << MatchTable::Comment("On fail goto") << MatchTable::JumpTarget(LabelID)
Daniel Sanders7aac7cc2017-07-20 09:25:44 +00002212 << MatchTable::LineBreak;
2213
Daniel Sanderse7b0d662017-04-21 15:59:56 +00002214 if (!RequiredFeatures.empty()) {
Daniel Sanders7aac7cc2017-07-20 09:25:44 +00002215 Table << MatchTable::Opcode("GIM_CheckFeatures")
2216 << MatchTable::NamedValue(getNameForFeatureBitset(RequiredFeatures))
2217 << MatchTable::LineBreak;
Daniel Sanderse7b0d662017-04-21 15:59:56 +00002218 }
Daniel Sandersb96f40d2017-03-20 15:20:42 +00002219
Daniel Sanders7aac7cc2017-07-20 09:25:44 +00002220 emitCaptureOpcodes(Table);
Daniel Sanders6ab0daa2017-07-04 14:35:06 +00002221
Daniel Sanders7aac7cc2017-07-20 09:25:44 +00002222 Matchers.front()->emitPredicateOpcodes(Table, *this,
Daniel Sanders6ab0daa2017-07-04 14:35:06 +00002223 getInsnVarID(*Matchers.front()));
2224
Daniel Sandersbee57392017-04-04 13:25:23 +00002225 // We must also check if it's safe to fold the matched instructions.
Daniel Sanders6ab0daa2017-07-04 14:35:06 +00002226 if (InsnVariableIDs.size() >= 2) {
Galina Kistanova1754fee2017-05-25 01:51:53 +00002227 // Invert the map to create stable ordering (by var names)
Daniel Sanders6ab0daa2017-07-04 14:35:06 +00002228 SmallVector<unsigned, 2> InsnIDs;
2229 for (const auto &Pair : InsnVariableIDs) {
Daniel Sandersbee57392017-04-04 13:25:23 +00002230 // Skip the root node since it isn't moving anywhere. Everything else is
2231 // sinking to meet it.
2232 if (Pair.first == Matchers.front().get())
2233 continue;
2234
Daniel Sanders6ab0daa2017-07-04 14:35:06 +00002235 InsnIDs.push_back(Pair.second);
Galina Kistanova1754fee2017-05-25 01:51:53 +00002236 }
Daniel Sanders6ab0daa2017-07-04 14:35:06 +00002237 std::sort(InsnIDs.begin(), InsnIDs.end());
Galina Kistanova1754fee2017-05-25 01:51:53 +00002238
Daniel Sanders6ab0daa2017-07-04 14:35:06 +00002239 for (const auto &InsnID : InsnIDs) {
Daniel Sandersbee57392017-04-04 13:25:23 +00002240 // Reject the difficult cases until we have a more accurate check.
Daniel Sanders7aac7cc2017-07-20 09:25:44 +00002241 Table << MatchTable::Opcode("GIM_CheckIsSafeToFold")
2242 << MatchTable::Comment("InsnID") << MatchTable::IntValue(InsnID)
2243 << MatchTable::LineBreak;
Daniel Sandersbee57392017-04-04 13:25:23 +00002244
2245 // FIXME: Emit checks to determine it's _actually_ safe to fold and/or
2246 // account for unsafe cases.
2247 //
2248 // Example:
2249 // MI1--> %0 = ...
2250 // %1 = ... %0
2251 // MI0--> %2 = ... %0
2252 // It's not safe to erase MI1. We currently handle this by not
2253 // erasing %0 (even when it's dead).
2254 //
2255 // Example:
2256 // MI1--> %0 = load volatile @a
2257 // %1 = load volatile @a
2258 // MI0--> %2 = ... %0
2259 // It's not safe to sink %0's def past %1. We currently handle
2260 // this by rejecting all loads.
2261 //
2262 // Example:
2263 // MI1--> %0 = load @a
2264 // %1 = store @a
2265 // MI0--> %2 = ... %0
2266 // It's not safe to sink %0's def past %1. We currently handle
2267 // this by rejecting all loads.
2268 //
2269 // Example:
2270 // G_CONDBR %cond, @BB1
2271 // BB0:
2272 // MI1--> %0 = load @a
2273 // G_BR @BB1
2274 // BB1:
2275 // MI0--> %2 = ... %0
2276 // It's not always safe to sink %0 across control flow. In this
2277 // case it may introduce a memory fault. We currentl handle this
2278 // by rejecting all loads.
2279 }
2280 }
2281
Daniel Sandersd93a35a2017-07-05 09:39:33 +00002282 for (const auto &MA : Actions)
Daniel Sandersa7b75262017-10-31 18:50:24 +00002283 MA->emitActionOpcodes(Table, *this);
Daniel Sandersf76f3152017-11-16 00:46:35 +00002284
2285 if (GenerateCoverage)
2286 Table << MatchTable::Opcode("GIR_Coverage") << MatchTable::IntValue(RuleID)
2287 << MatchTable::LineBreak;
2288
Daniel Sanders7aac7cc2017-07-20 09:25:44 +00002289 Table << MatchTable::Opcode("GIR_Done", -1) << MatchTable::LineBreak
Daniel Sanders8e82af22017-07-27 11:03:45 +00002290 << MatchTable::Label(LabelID);
Daniel Sandersbdfebb82017-03-15 20:18:38 +00002291}
Daniel Sanders43c882c2017-02-01 10:53:10 +00002292
Daniel Sandersbdfebb82017-03-15 20:18:38 +00002293bool RuleMatcher::isHigherPriorityThan(const RuleMatcher &B) const {
2294 // Rules involving more match roots have higher priority.
2295 if (Matchers.size() > B.Matchers.size())
2296 return true;
2297 if (Matchers.size() < B.Matchers.size())
Daniel Sanders759ff412017-02-24 13:58:11 +00002298 return false;
Daniel Sanders8a4bae92017-03-14 21:32:08 +00002299
Daniel Sandersbdfebb82017-03-15 20:18:38 +00002300 for (const auto &Matcher : zip(Matchers, B.Matchers)) {
2301 if (std::get<0>(Matcher)->isHigherPriorityThan(*std::get<1>(Matcher)))
2302 return true;
2303 if (std::get<1>(Matcher)->isHigherPriorityThan(*std::get<0>(Matcher)))
2304 return false;
Daniel Sanders8a4bae92017-03-14 21:32:08 +00002305 }
Daniel Sandersbdfebb82017-03-15 20:18:38 +00002306
2307 return false;
Simon Pilgrima7d1da82017-03-15 22:50:47 +00002308}
Ahmed Bougacha36f70352016-12-21 23:26:20 +00002309
Daniel Sanders2deea182017-04-22 15:11:04 +00002310unsigned RuleMatcher::countRendererFns() const {
Daniel Sandersbdfebb82017-03-15 20:18:38 +00002311 return std::accumulate(
2312 Matchers.begin(), Matchers.end(), 0,
2313 [](unsigned A, const std::unique_ptr<InstructionMatcher> &Matcher) {
Daniel Sanders2deea182017-04-22 15:11:04 +00002314 return A + Matcher->countRendererFns();
Daniel Sandersbdfebb82017-03-15 20:18:38 +00002315 });
2316}
2317
Daniel Sanders05540042017-08-08 10:44:31 +00002318bool OperandPredicateMatcher::isHigherPriorityThan(
2319 const OperandPredicateMatcher &B) const {
2320 // Generally speaking, an instruction is more important than an Int or a
2321 // LiteralInt because it can cover more nodes but theres an exception to
2322 // this. G_CONSTANT's are less important than either of those two because they
2323 // are more permissive.
Daniel Sandersedd07842017-08-17 09:26:14 +00002324
2325 const InstructionOperandMatcher *AOM =
2326 dyn_cast<InstructionOperandMatcher>(this);
2327 const InstructionOperandMatcher *BOM =
2328 dyn_cast<InstructionOperandMatcher>(&B);
2329 bool AIsConstantInsn = AOM && AOM->getInsnMatcher().isConstantInstruction();
2330 bool BIsConstantInsn = BOM && BOM->getInsnMatcher().isConstantInstruction();
2331
2332 if (AOM && BOM) {
2333 // The relative priorities between a G_CONSTANT and any other instruction
2334 // don't actually matter but this code is needed to ensure a strict weak
2335 // ordering. This is particularly important on Windows where the rules will
2336 // be incorrectly sorted without it.
2337 if (AIsConstantInsn != BIsConstantInsn)
2338 return AIsConstantInsn < BIsConstantInsn;
2339 return false;
Daniel Sanders05540042017-08-08 10:44:31 +00002340 }
Daniel Sandersedd07842017-08-17 09:26:14 +00002341
2342 if (AOM && AIsConstantInsn && (B.Kind == OPM_Int || B.Kind == OPM_LiteralInt))
2343 return false;
2344 if (BOM && BIsConstantInsn && (Kind == OPM_Int || Kind == OPM_LiteralInt))
2345 return true;
Daniel Sanders05540042017-08-08 10:44:31 +00002346
2347 return Kind < B.Kind;
Daniel Sanders75b84fc2017-08-08 13:21:26 +00002348}
Daniel Sanders05540042017-08-08 10:44:31 +00002349
Daniel Sandersbfa9e2c2017-10-14 00:31:58 +00002350void SameOperandMatcher::emitPredicateOpcodes(MatchTable &Table,
2351 RuleMatcher &Rule,
2352 unsigned InsnVarID,
2353 unsigned OpIdx) const {
Daniel Sanders1e4569f2017-10-20 20:55:29 +00002354 const OperandMatcher &OtherOM = Rule.getOperandMatcher(MatchingName);
Daniel Sandersbfa9e2c2017-10-14 00:31:58 +00002355 unsigned OtherInsnVarID = Rule.getInsnVarID(OtherOM.getInstructionMatcher());
2356
2357 Table << MatchTable::Opcode("GIM_CheckIsSameOperand")
2358 << MatchTable::Comment("MI") << MatchTable::IntValue(InsnVarID)
2359 << MatchTable::Comment("OpIdx") << MatchTable::IntValue(OpIdx)
2360 << MatchTable::Comment("OtherMI")
2361 << MatchTable::IntValue(OtherInsnVarID)
2362 << MatchTable::Comment("OtherOpIdx")
2363 << MatchTable::IntValue(OtherOM.getOperandIndex())
2364 << MatchTable::LineBreak;
2365}
2366
Ahmed Bougacha36f70352016-12-21 23:26:20 +00002367//===- GlobalISelEmitter class --------------------------------------------===//
2368
Ahmed Bougacha982c5eb2017-02-10 04:00:17 +00002369class GlobalISelEmitter {
2370public:
2371 explicit GlobalISelEmitter(RecordKeeper &RK);
2372 void run(raw_ostream &OS);
2373
2374private:
2375 const RecordKeeper &RK;
2376 const CodeGenDAGPatterns CGP;
2377 const CodeGenTarget &Target;
Daniel Sanderscc36dbf2017-06-27 10:11:39 +00002378 CodeGenRegBank CGRegs;
Ahmed Bougacha982c5eb2017-02-10 04:00:17 +00002379
Daniel Sanders39690bd2017-10-15 02:41:12 +00002380 /// Keep track of the equivalence between SDNodes and Instruction by mapping
Daniel Sanders3c1c4c02017-12-05 05:52:07 +00002381 /// SDNodes to the GINodeEquiv mapping. We need to map to the GINodeEquiv to
2382 /// check for attributes on the relation such as CheckMMOIsNonAtomic.
Ahmed Bougacha982c5eb2017-02-10 04:00:17 +00002383 /// This is defined using 'GINodeEquiv' in the target description.
Daniel Sanders39690bd2017-10-15 02:41:12 +00002384 DenseMap<Record *, Record *> NodeEquivs;
Ahmed Bougacha982c5eb2017-02-10 04:00:17 +00002385
Daniel Sanders8a4bae92017-03-14 21:32:08 +00002386 /// Keep track of the equivalence between ComplexPattern's and
2387 /// GIComplexOperandMatcher. Map entries are specified by subclassing
2388 /// GIComplexPatternEquiv.
2389 DenseMap<const Record *, const Record *> ComplexPatternEquivs;
2390
Daniel Sanderse7b0d662017-04-21 15:59:56 +00002391 // Map of predicates to their subtarget features.
Daniel Sanderse9fdba32017-04-29 17:30:09 +00002392 SubtargetFeatureInfoMap SubtargetFeatures;
Daniel Sanderse7b0d662017-04-21 15:59:56 +00002393
Daniel Sandersf76f3152017-11-16 00:46:35 +00002394 // Rule coverage information.
2395 Optional<CodeGenCoverage> RuleCoverage;
2396
Ahmed Bougacha982c5eb2017-02-10 04:00:17 +00002397 void gatherNodeEquivs();
Daniel Sanders39690bd2017-10-15 02:41:12 +00002398 Record *findNodeEquiv(Record *N) const;
Ahmed Bougacha982c5eb2017-02-10 04:00:17 +00002399
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +00002400 Error importRulePredicates(RuleMatcher &M, ArrayRef<Predicate> Predicates);
Daniel Sandersdf39cba2017-10-15 18:22:54 +00002401 Expected<InstructionMatcher &> createAndImportSelDAGMatcher(
2402 RuleMatcher &Rule, InstructionMatcher &InsnMatcher,
2403 const TreePatternNode *Src, unsigned &TempOpIdx) const;
2404 Error importComplexPatternOperandMatcher(OperandMatcher &OM, Record *R,
2405 unsigned &TempOpIdx) const;
2406 Error importChildMatcher(RuleMatcher &Rule, InstructionMatcher &InsnMatcher,
Daniel Sandersa71f4542017-10-16 00:56:30 +00002407 const TreePatternNode *SrcChild,
2408 bool OperandIsAPointer, unsigned OpIdx,
Daniel Sandersc270c502017-03-30 09:36:33 +00002409 unsigned &TempOpIdx) const;
Daniel Sandersdf258e32017-10-31 19:09:29 +00002410
Daniel Sanderscc36dbf2017-06-27 10:11:39 +00002411 Expected<BuildMIAction &>
Daniel Sandersa7b75262017-10-31 18:50:24 +00002412 createAndImportInstructionRenderer(RuleMatcher &M,
2413 const TreePatternNode *Dst);
Daniel Sanders9cbe7c72017-11-01 19:57:57 +00002414 Expected<action_iterator> createAndImportSubInstructionRenderer(
2415 action_iterator InsertPt, RuleMatcher &M, const TreePatternNode *Dst,
2416 unsigned TempReg);
Daniel Sanders7438b262017-10-31 23:03:18 +00002417 Expected<action_iterator>
2418 createInstructionRenderer(action_iterator InsertPt, RuleMatcher &M,
2419 const TreePatternNode *Dst);
Daniel Sandersdf258e32017-10-31 19:09:29 +00002420 void importExplicitDefRenderers(BuildMIAction &DstMIBuilder);
Daniel Sanders7438b262017-10-31 23:03:18 +00002421 Expected<action_iterator>
2422 importExplicitUseRenderers(action_iterator InsertPt, RuleMatcher &M,
2423 BuildMIAction &DstMIBuilder,
Daniel Sandersdf258e32017-10-31 19:09:29 +00002424 const llvm::TreePatternNode *Dst);
Daniel Sanders7438b262017-10-31 23:03:18 +00002425 Expected<action_iterator>
2426 importExplicitUseRenderer(action_iterator InsertPt, RuleMatcher &Rule,
2427 BuildMIAction &DstMIBuilder,
Daniel Sanders9cbe7c72017-11-01 19:57:57 +00002428 TreePatternNode *DstChild);
Diana Picus382602f2017-05-17 08:57:28 +00002429 Error importDefaultOperandRenderers(BuildMIAction &DstMIBuilder,
2430 DagInit *DefaultOps) const;
Daniel Sandersc270c502017-03-30 09:36:33 +00002431 Error
Daniel Sandersffc7d582017-03-29 15:37:18 +00002432 importImplicitDefRenderers(BuildMIAction &DstMIBuilder,
2433 const std::vector<Record *> &ImplicitDefs) const;
2434
Daniel Sanders11300ce2017-10-13 21:28:03 +00002435 void emitImmPredicates(raw_ostream &OS, StringRef TypeIdentifier,
2436 StringRef Type,
Daniel Sanders649c5852017-10-13 20:42:18 +00002437 std::function<bool(const Record *R)> Filter);
2438
Ahmed Bougacha982c5eb2017-02-10 04:00:17 +00002439 /// Analyze pattern \p P, returning a matcher for it if possible.
2440 /// Otherwise, return an Error explaining why we don't support it.
2441 Expected<RuleMatcher> runOnPattern(const PatternToMatch &P);
Daniel Sanderse7b0d662017-04-21 15:59:56 +00002442
2443 void declareSubtargetFeature(Record *Predicate);
Daniel Sanders7e523672017-11-11 03:23:44 +00002444
2445 TreePatternNode *fixupPatternNode(TreePatternNode *N);
2446 void fixupPatternTrees(TreePattern *P);
Ahmed Bougacha982c5eb2017-02-10 04:00:17 +00002447};
2448
Ahmed Bougacha36f70352016-12-21 23:26:20 +00002449void GlobalISelEmitter::gatherNodeEquivs() {
2450 assert(NodeEquivs.empty());
2451 for (Record *Equiv : RK.getAllDerivedDefinitions("GINodeEquiv"))
Daniel Sanders39690bd2017-10-15 02:41:12 +00002452 NodeEquivs[Equiv->getValueAsDef("Node")] = Equiv;
Daniel Sanders8a4bae92017-03-14 21:32:08 +00002453
2454 assert(ComplexPatternEquivs.empty());
2455 for (Record *Equiv : RK.getAllDerivedDefinitions("GIComplexPatternEquiv")) {
2456 Record *SelDAGEquiv = Equiv->getValueAsDef("SelDAGEquivalent");
2457 if (!SelDAGEquiv)
2458 continue;
2459 ComplexPatternEquivs[SelDAGEquiv] = Equiv;
2460 }
Ahmed Bougacha36f70352016-12-21 23:26:20 +00002461}
2462
Daniel Sanders39690bd2017-10-15 02:41:12 +00002463Record *GlobalISelEmitter::findNodeEquiv(Record *N) const {
Ahmed Bougacha36f70352016-12-21 23:26:20 +00002464 return NodeEquivs.lookup(N);
2465}
2466
2467GlobalISelEmitter::GlobalISelEmitter(RecordKeeper &RK)
Daniel Sanders7e523672017-11-11 03:23:44 +00002468 : RK(RK), CGP(RK, [&](TreePattern *P) { fixupPatternTrees(P); }),
2469 Target(CGP.getTargetInfo()), CGRegs(RK, Target.getHwModes()) {}
Ahmed Bougacha36f70352016-12-21 23:26:20 +00002470
2471//===- Emitter ------------------------------------------------------------===//
2472
Daniel Sandersc270c502017-03-30 09:36:33 +00002473Error
Daniel Sandersffc7d582017-03-29 15:37:18 +00002474GlobalISelEmitter::importRulePredicates(RuleMatcher &M,
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +00002475 ArrayRef<Predicate> Predicates) {
2476 for (const Predicate &P : Predicates) {
2477 if (!P.Def)
2478 continue;
2479 declareSubtargetFeature(P.Def);
2480 M.addRequiredFeature(P.Def);
Daniel Sanderse7b0d662017-04-21 15:59:56 +00002481 }
2482
Daniel Sandersc270c502017-03-30 09:36:33 +00002483 return Error::success();
Daniel Sandersffc7d582017-03-29 15:37:18 +00002484}
Daniel Sanders8a4bae92017-03-14 21:32:08 +00002485
Daniel Sanders3c1c4c02017-12-05 05:52:07 +00002486Expected<InstructionMatcher &> GlobalISelEmitter::createAndImportSelDAGMatcher(
2487 RuleMatcher &Rule, InstructionMatcher &InsnMatcher,
2488 const TreePatternNode *Src, unsigned &TempOpIdx) const {
2489 Record *SrcGIEquivOrNull = nullptr;
2490 const CodeGenInstruction *SrcGIOrNull = nullptr;
2491
2492 // Start with the defined operands (i.e., the results of the root operator).
2493 if (Src->getExtTypes().size() > 1)
2494 return failedImport("Src pattern has multiple results");
2495
2496 if (Src->isLeaf()) {
2497 Init *SrcInit = Src->getLeafValue();
2498 if (isa<IntInit>(SrcInit)) {
2499 InsnMatcher.addPredicate<InstructionOpcodeMatcher>(
2500 &Target.getInstruction(RK.getDef("G_CONSTANT")));
2501 } else
2502 return failedImport(
2503 "Unable to deduce gMIR opcode to handle Src (which is a leaf)");
2504 } else {
2505 SrcGIEquivOrNull = findNodeEquiv(Src->getOperator());
2506 if (!SrcGIEquivOrNull)
2507 return failedImport("Pattern operator lacks an equivalent Instruction" +
2508 explainOperator(Src->getOperator()));
2509 SrcGIOrNull = &Target.getInstruction(SrcGIEquivOrNull->getValueAsDef("I"));
2510
2511 // The operators look good: match the opcode
2512 InsnMatcher.addPredicate<InstructionOpcodeMatcher>(SrcGIOrNull);
2513 }
2514
2515 unsigned OpIdx = 0;
2516 for (const TypeSetByHwMode &VTy : Src->getExtTypes()) {
2517 // Results don't have a name unless they are the root node. The caller will
2518 // set the name if appropriate.
2519 OperandMatcher &OM = InsnMatcher.addOperand(OpIdx++, "", TempOpIdx);
2520 if (auto Error = OM.addTypeCheckPredicate(VTy, false /* OperandIsAPointer */))
2521 return failedImport(toString(std::move(Error)) +
2522 " for result of Src pattern operator");
2523 }
2524
Daniel Sanders2c269f62017-08-24 09:11:20 +00002525 for (const auto &Predicate : Src->getPredicateFns()) {
2526 if (Predicate.isAlwaysTrue())
2527 continue;
2528
2529 if (Predicate.isImmediatePattern()) {
2530 InsnMatcher.addPredicate<InstructionImmPredicateMatcher>(Predicate);
2531 continue;
2532 }
2533
Daniel Sandersa71f4542017-10-16 00:56:30 +00002534 // No check required. G_LOAD by itself is a non-extending load.
2535 if (Predicate.isNonExtLoad())
2536 continue;
2537
Daniel Sandersd66e0902017-10-23 18:19:24 +00002538 // No check required. G_STORE by itself is a non-extending store.
2539 if (Predicate.isNonTruncStore())
2540 continue;
2541
Daniel Sanders76664652017-11-28 22:07:05 +00002542 if (Predicate.isLoad() || Predicate.isStore() || Predicate.isAtomic()) {
2543 if (Predicate.getMemoryVT() != nullptr) {
2544 Optional<LLTCodeGen> MemTyOrNone =
2545 MVTToLLT(getValueType(Predicate.getMemoryVT()));
Daniel Sandersd66e0902017-10-23 18:19:24 +00002546
Daniel Sanders76664652017-11-28 22:07:05 +00002547 if (!MemTyOrNone)
2548 return failedImport("MemVT could not be converted to LLT");
Daniel Sandersd66e0902017-10-23 18:19:24 +00002549
Daniel Sanders76664652017-11-28 22:07:05 +00002550 InsnMatcher.getOperand(0).addPredicate<LLTOperandMatcher>(
2551 MemTyOrNone.getValue());
2552 continue;
2553 }
2554 }
2555
2556 if (Predicate.isLoad() || Predicate.isStore()) {
2557 // No check required. A G_LOAD/G_STORE is an unindexed load.
2558 if (Predicate.isUnindexed())
2559 continue;
2560 }
2561
2562 if (Predicate.isAtomic()) {
2563 if (Predicate.isAtomicOrderingMonotonic()) {
2564 InsnMatcher.addPredicate<AtomicOrderingMMOPredicateMatcher>(
2565 "Monotonic");
2566 continue;
2567 }
2568 if (Predicate.isAtomicOrderingAcquire()) {
2569 InsnMatcher.addPredicate<AtomicOrderingMMOPredicateMatcher>("Acquire");
2570 continue;
2571 }
2572 if (Predicate.isAtomicOrderingRelease()) {
2573 InsnMatcher.addPredicate<AtomicOrderingMMOPredicateMatcher>("Release");
2574 continue;
2575 }
2576 if (Predicate.isAtomicOrderingAcquireRelease()) {
2577 InsnMatcher.addPredicate<AtomicOrderingMMOPredicateMatcher>(
2578 "AcquireRelease");
2579 continue;
2580 }
2581 if (Predicate.isAtomicOrderingSequentiallyConsistent()) {
2582 InsnMatcher.addPredicate<AtomicOrderingMMOPredicateMatcher>(
2583 "SequentiallyConsistent");
2584 continue;
2585 }
Daniel Sanders0c43b3a2017-11-30 21:05:59 +00002586
2587 if (Predicate.isAtomicOrderingAcquireOrStronger()) {
2588 InsnMatcher.addPredicate<AtomicOrderingMMOPredicateMatcher>(
2589 "Acquire", AtomicOrderingMMOPredicateMatcher::AO_OrStronger);
2590 continue;
2591 }
2592 if (Predicate.isAtomicOrderingWeakerThanAcquire()) {
2593 InsnMatcher.addPredicate<AtomicOrderingMMOPredicateMatcher>(
2594 "Acquire", AtomicOrderingMMOPredicateMatcher::AO_WeakerThan);
2595 continue;
2596 }
2597
2598 if (Predicate.isAtomicOrderingReleaseOrStronger()) {
2599 InsnMatcher.addPredicate<AtomicOrderingMMOPredicateMatcher>(
2600 "Release", AtomicOrderingMMOPredicateMatcher::AO_OrStronger);
2601 continue;
2602 }
2603 if (Predicate.isAtomicOrderingWeakerThanRelease()) {
2604 InsnMatcher.addPredicate<AtomicOrderingMMOPredicateMatcher>(
2605 "Release", AtomicOrderingMMOPredicateMatcher::AO_WeakerThan);
2606 continue;
2607 }
Daniel Sandersd66e0902017-10-23 18:19:24 +00002608 }
2609
Daniel Sanders2c269f62017-08-24 09:11:20 +00002610 return failedImport("Src pattern child has predicate (" +
2611 explainPredicates(Src) + ")");
2612 }
Daniel Sanders3c1c4c02017-12-05 05:52:07 +00002613 if (SrcGIEquivOrNull && SrcGIEquivOrNull->getValueAsBit("CheckMMOIsNonAtomic"))
2614 InsnMatcher.addPredicate<AtomicOrderingMMOPredicateMatcher>("NotAtomic");
Daniel Sanders2c269f62017-08-24 09:11:20 +00002615
Daniel Sanders452c8ae2017-05-23 19:33:16 +00002616 if (Src->isLeaf()) {
2617 Init *SrcInit = Src->getLeafValue();
2618 if (IntInit *SrcIntInit = dyn_cast<IntInit>(SrcInit)) {
Daniel Sandersbfa9e2c2017-10-14 00:31:58 +00002619 OperandMatcher &OM =
2620 InsnMatcher.addOperand(OpIdx++, Src->getName(), TempOpIdx);
Daniel Sanders452c8ae2017-05-23 19:33:16 +00002621 OM.addPredicate<LiteralIntOperandMatcher>(SrcIntInit->getValue());
2622 } else
Daniel Sanders32291982017-06-28 13:50:04 +00002623 return failedImport(
2624 "Unable to deduce gMIR opcode to handle Src (which is a leaf)");
Daniel Sanders452c8ae2017-05-23 19:33:16 +00002625 } else {
Daniel Sanders85ffd362017-07-06 08:12:20 +00002626 assert(SrcGIOrNull &&
2627 "Expected to have already found an equivalent Instruction");
Daniel Sanders11300ce2017-10-13 21:28:03 +00002628 if (SrcGIOrNull->TheDef->getName() == "G_CONSTANT" ||
2629 SrcGIOrNull->TheDef->getName() == "G_FCONSTANT") {
2630 // imm/fpimm still have operands but we don't need to do anything with it
Daniel Sanders05540042017-08-08 10:44:31 +00002631 // here since we don't support ImmLeaf predicates yet. However, we still
2632 // need to note the hidden operand to get GIM_CheckNumOperands correct.
2633 InsnMatcher.addOperand(OpIdx++, "", TempOpIdx);
2634 return InsnMatcher;
2635 }
2636
Daniel Sanders452c8ae2017-05-23 19:33:16 +00002637 // Match the used operands (i.e. the children of the operator).
2638 for (unsigned i = 0, e = Src->getNumChildren(); i != e; ++i) {
Daniel Sanders85ffd362017-07-06 08:12:20 +00002639 TreePatternNode *SrcChild = Src->getChild(i);
2640
Daniel Sandersa71f4542017-10-16 00:56:30 +00002641 // SelectionDAG allows pointers to be represented with iN since it doesn't
2642 // distinguish between pointers and integers but they are different types in GlobalISel.
2643 // Coerce integers to pointers to address space 0 if the context indicates a pointer.
Daniel Sandersc54aa9c2017-11-18 00:16:44 +00002644 bool OperandIsAPointer = SrcGIOrNull->isOperandAPointer(i);
Daniel Sandersa71f4542017-10-16 00:56:30 +00002645
Daniel Sanders28887fe2017-09-19 12:56:36 +00002646 // For G_INTRINSIC/G_INTRINSIC_W_SIDE_EFFECTS, the operand immediately
2647 // following the defs is an intrinsic ID.
2648 if ((SrcGIOrNull->TheDef->getName() == "G_INTRINSIC" ||
2649 SrcGIOrNull->TheDef->getName() == "G_INTRINSIC_W_SIDE_EFFECTS") &&
2650 i == 0) {
Daniel Sandersfe12c0f2017-07-11 08:57:29 +00002651 if (const CodeGenIntrinsic *II = Src->getIntrinsicInfo(CGP)) {
Daniel Sanders85ffd362017-07-06 08:12:20 +00002652 OperandMatcher &OM =
2653 InsnMatcher.addOperand(OpIdx++, SrcChild->getName(), TempOpIdx);
Daniel Sandersfe12c0f2017-07-11 08:57:29 +00002654 OM.addPredicate<IntrinsicIDOperandMatcher>(II);
Daniel Sanders85ffd362017-07-06 08:12:20 +00002655 continue;
2656 }
2657
2658 return failedImport("Expected IntInit containing instrinsic ID)");
2659 }
2660
Daniel Sandersa71f4542017-10-16 00:56:30 +00002661 if (auto Error =
2662 importChildMatcher(Rule, InsnMatcher, SrcChild, OperandIsAPointer,
2663 OpIdx++, TempOpIdx))
Daniel Sanders452c8ae2017-05-23 19:33:16 +00002664 return std::move(Error);
2665 }
Daniel Sandersffc7d582017-03-29 15:37:18 +00002666 }
2667
2668 return InsnMatcher;
2669}
2670
Daniel Sandersdf39cba2017-10-15 18:22:54 +00002671Error GlobalISelEmitter::importComplexPatternOperandMatcher(
2672 OperandMatcher &OM, Record *R, unsigned &TempOpIdx) const {
2673 const auto &ComplexPattern = ComplexPatternEquivs.find(R);
2674 if (ComplexPattern == ComplexPatternEquivs.end())
2675 return failedImport("SelectionDAG ComplexPattern (" + R->getName() +
2676 ") not mapped to GlobalISel");
2677
2678 OM.addPredicate<ComplexPatternOperandMatcher>(OM, *ComplexPattern->second);
2679 TempOpIdx++;
2680 return Error::success();
2681}
2682
2683Error GlobalISelEmitter::importChildMatcher(RuleMatcher &Rule,
2684 InstructionMatcher &InsnMatcher,
Daniel Sanders452c8ae2017-05-23 19:33:16 +00002685 const TreePatternNode *SrcChild,
Daniel Sandersa71f4542017-10-16 00:56:30 +00002686 bool OperandIsAPointer,
Daniel Sandersc270c502017-03-30 09:36:33 +00002687 unsigned OpIdx,
2688 unsigned &TempOpIdx) const {
Daniel Sanders4f3eb242017-04-05 13:14:03 +00002689 OperandMatcher &OM =
2690 InsnMatcher.addOperand(OpIdx, SrcChild->getName(), TempOpIdx);
Daniel Sandersbfa9e2c2017-10-14 00:31:58 +00002691 if (OM.isSameAsAnotherOperand())
2692 return Error::success();
Daniel Sandersffc7d582017-03-29 15:37:18 +00002693
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +00002694 ArrayRef<TypeSetByHwMode> ChildTypes = SrcChild->getExtTypes();
Daniel Sandersffc7d582017-03-29 15:37:18 +00002695 if (ChildTypes.size() != 1)
2696 return failedImport("Src pattern child has multiple results");
2697
2698 // Check MBB's before the type check since they are not a known type.
2699 if (!SrcChild->isLeaf()) {
2700 if (SrcChild->getOperator()->isSubClassOf("SDNode")) {
2701 auto &ChildSDNI = CGP.getSDNodeInfo(SrcChild->getOperator());
2702 if (ChildSDNI.getSDClassName() == "BasicBlockSDNode") {
2703 OM.addPredicate<MBBOperandMatcher>();
Daniel Sandersc270c502017-03-30 09:36:33 +00002704 return Error::success();
Daniel Sandersffc7d582017-03-29 15:37:18 +00002705 }
2706 }
Daniel Sandersffc7d582017-03-29 15:37:18 +00002707 }
2708
Daniel Sandersa71f4542017-10-16 00:56:30 +00002709 if (auto Error =
2710 OM.addTypeCheckPredicate(ChildTypes.front(), OperandIsAPointer))
2711 return failedImport(toString(std::move(Error)) + " for Src operand (" +
2712 to_string(*SrcChild) + ")");
Daniel Sandersffc7d582017-03-29 15:37:18 +00002713
Daniel Sandersbee57392017-04-04 13:25:23 +00002714 // Check for nested instructions.
2715 if (!SrcChild->isLeaf()) {
Daniel Sandersdf39cba2017-10-15 18:22:54 +00002716 if (SrcChild->getOperator()->isSubClassOf("ComplexPattern")) {
2717 // When a ComplexPattern is used as an operator, it should do the same
2718 // thing as when used as a leaf. However, the children of the operator
2719 // name the sub-operands that make up the complex operand and we must
2720 // prepare to reference them in the renderer too.
2721 unsigned RendererID = TempOpIdx;
2722 if (auto Error = importComplexPatternOperandMatcher(
2723 OM, SrcChild->getOperator(), TempOpIdx))
2724 return Error;
2725
2726 for (unsigned i = 0, e = SrcChild->getNumChildren(); i != e; ++i) {
2727 auto *SubOperand = SrcChild->getChild(i);
2728 if (!SubOperand->getName().empty())
2729 Rule.defineComplexSubOperand(SubOperand->getName(),
2730 SrcChild->getOperator(), RendererID, i);
2731 }
2732
2733 return Error::success();
2734 }
2735
Daniel Sandersbfa9e2c2017-10-14 00:31:58 +00002736 auto MaybeInsnOperand = OM.addPredicate<InstructionOperandMatcher>(
2737 InsnMatcher.getRuleMatcher(), SrcChild->getName());
2738 if (!MaybeInsnOperand.hasValue()) {
2739 // This isn't strictly true. If the user were to provide exactly the same
2740 // matchers as the original operand then we could allow it. However, it's
2741 // simpler to not permit the redundant specification.
2742 return failedImport("Nested instruction cannot be the same as another operand");
2743 }
2744
Daniel Sandersbee57392017-04-04 13:25:23 +00002745 // Map the node to a gMIR instruction.
Daniel Sandersbfa9e2c2017-10-14 00:31:58 +00002746 InstructionOperandMatcher &InsnOperand = **MaybeInsnOperand;
Daniel Sanders57938df2017-07-11 10:40:18 +00002747 auto InsnMatcherOrError = createAndImportSelDAGMatcher(
Daniel Sandersdf39cba2017-10-15 18:22:54 +00002748 Rule, InsnOperand.getInsnMatcher(), SrcChild, TempOpIdx);
Daniel Sandersbee57392017-04-04 13:25:23 +00002749 if (auto Error = InsnMatcherOrError.takeError())
2750 return Error;
2751
2752 return Error::success();
2753 }
2754
Diana Picusd1b61812017-11-03 10:30:19 +00002755 if (SrcChild->hasAnyPredicate())
2756 return failedImport("Src pattern child has unsupported predicate");
2757
Daniel Sandersffc7d582017-03-29 15:37:18 +00002758 // Check for constant immediates.
2759 if (auto *ChildInt = dyn_cast<IntInit>(SrcChild->getLeafValue())) {
Daniel Sanders452c8ae2017-05-23 19:33:16 +00002760 OM.addPredicate<ConstantIntOperandMatcher>(ChildInt->getValue());
Daniel Sandersc270c502017-03-30 09:36:33 +00002761 return Error::success();
Daniel Sandersffc7d582017-03-29 15:37:18 +00002762 }
2763
2764 // Check for def's like register classes or ComplexPattern's.
2765 if (auto *ChildDefInit = dyn_cast<DefInit>(SrcChild->getLeafValue())) {
2766 auto *ChildRec = ChildDefInit->getDef();
2767
2768 // Check for register classes.
Daniel Sandersa6e2ceb2017-06-20 12:36:34 +00002769 if (ChildRec->isSubClassOf("RegisterClass") ||
2770 ChildRec->isSubClassOf("RegisterOperand")) {
Daniel Sandersffc7d582017-03-29 15:37:18 +00002771 OM.addPredicate<RegisterBankOperandMatcher>(
Daniel Sandersa6e2ceb2017-06-20 12:36:34 +00002772 Target.getRegisterClass(getInitValueAsRegClass(ChildDefInit)));
Daniel Sanders658541f2017-04-22 15:53:21 +00002773 return Error::success();
2774 }
2775
Daniel Sanders4d4e7652017-10-09 18:14:53 +00002776 // Check for ValueType.
2777 if (ChildRec->isSubClassOf("ValueType")) {
2778 // We already added a type check as standard practice so this doesn't need
2779 // to do anything.
2780 return Error::success();
2781 }
2782
Daniel Sandersffc7d582017-03-29 15:37:18 +00002783 // Check for ComplexPattern's.
Daniel Sandersdf39cba2017-10-15 18:22:54 +00002784 if (ChildRec->isSubClassOf("ComplexPattern"))
2785 return importComplexPatternOperandMatcher(OM, ChildRec, TempOpIdx);
Daniel Sandersffc7d582017-03-29 15:37:18 +00002786
Daniel Sandersd0656a32017-04-13 09:45:37 +00002787 if (ChildRec->isSubClassOf("ImmLeaf")) {
2788 return failedImport(
2789 "Src pattern child def is an unsupported tablegen class (ImmLeaf)");
2790 }
2791
Daniel Sandersffc7d582017-03-29 15:37:18 +00002792 return failedImport(
2793 "Src pattern child def is an unsupported tablegen class");
2794 }
2795
2796 return failedImport("Src pattern child is an unsupported kind");
2797}
2798
Daniel Sanders7438b262017-10-31 23:03:18 +00002799Expected<action_iterator> GlobalISelEmitter::importExplicitUseRenderer(
2800 action_iterator InsertPt, RuleMatcher &Rule, BuildMIAction &DstMIBuilder,
Daniel Sanders9cbe7c72017-11-01 19:57:57 +00002801 TreePatternNode *DstChild) {
Daniel Sanders2c269f62017-08-24 09:11:20 +00002802 if (DstChild->getTransformFn() != nullptr) {
2803 return failedImport("Dst pattern child has transform fn " +
2804 DstChild->getTransformFn()->getName());
2805 }
2806
Daniel Sandersdf39cba2017-10-15 18:22:54 +00002807 const auto &SubOperand = Rule.getComplexSubOperand(DstChild->getName());
2808 if (SubOperand.hasValue()) {
2809 DstMIBuilder.addRenderer<RenderComplexPatternOperand>(
Daniel Sanders198447a2017-11-01 00:29:47 +00002810 *std::get<0>(*SubOperand), DstChild->getName(),
Daniel Sandersdf39cba2017-10-15 18:22:54 +00002811 std::get<1>(*SubOperand), std::get<2>(*SubOperand));
Daniel Sanders7438b262017-10-31 23:03:18 +00002812 return InsertPt;
Daniel Sandersdf39cba2017-10-15 18:22:54 +00002813 }
2814
Daniel Sandersffc7d582017-03-29 15:37:18 +00002815 if (!DstChild->isLeaf()) {
Daniel Sanders05540042017-08-08 10:44:31 +00002816 // We accept 'bb' here. It's an operator because BasicBlockSDNode isn't
2817 // inline, but in MI it's just another operand.
Daniel Sandersffc7d582017-03-29 15:37:18 +00002818 if (DstChild->getOperator()->isSubClassOf("SDNode")) {
2819 auto &ChildSDNI = CGP.getSDNodeInfo(DstChild->getOperator());
2820 if (ChildSDNI.getSDClassName() == "BasicBlockSDNode") {
Daniel Sanders198447a2017-11-01 00:29:47 +00002821 DstMIBuilder.addRenderer<CopyRenderer>(DstChild->getName());
Daniel Sanders7438b262017-10-31 23:03:18 +00002822 return InsertPt;
Daniel Sandersffc7d582017-03-29 15:37:18 +00002823 }
2824 }
Daniel Sanders05540042017-08-08 10:44:31 +00002825
2826 // Similarly, imm is an operator in TreePatternNode's view but must be
2827 // rendered as operands.
2828 // FIXME: The target should be able to choose sign-extended when appropriate
2829 // (e.g. on Mips).
2830 if (DstChild->getOperator()->getName() == "imm") {
Daniel Sanders198447a2017-11-01 00:29:47 +00002831 DstMIBuilder.addRenderer<CopyConstantAsImmRenderer>(DstChild->getName());
Daniel Sanders7438b262017-10-31 23:03:18 +00002832 return InsertPt;
Daniel Sanders11300ce2017-10-13 21:28:03 +00002833 } else if (DstChild->getOperator()->getName() == "fpimm") {
2834 DstMIBuilder.addRenderer<CopyFConstantAsFPImmRenderer>(
Daniel Sanders198447a2017-11-01 00:29:47 +00002835 DstChild->getName());
Daniel Sanders7438b262017-10-31 23:03:18 +00002836 return InsertPt;
Daniel Sanders05540042017-08-08 10:44:31 +00002837 }
2838
Daniel Sanders9cbe7c72017-11-01 19:57:57 +00002839 if (DstChild->getOperator()->isSubClassOf("Instruction")) {
2840 ArrayRef<TypeSetByHwMode> ChildTypes = DstChild->getExtTypes();
2841 if (ChildTypes.size() != 1)
2842 return failedImport("Dst pattern child has multiple results");
2843
2844 Optional<LLTCodeGen> OpTyOrNone = None;
2845 if (ChildTypes.front().isMachineValueType())
2846 OpTyOrNone =
2847 MVTToLLT(ChildTypes.front().getMachineValueType().SimpleTy);
2848 if (!OpTyOrNone)
2849 return failedImport("Dst operand has an unsupported type");
2850
2851 unsigned TempRegID = Rule.allocateTempRegID();
2852 InsertPt = Rule.insertAction<MakeTempRegisterAction>(
2853 InsertPt, OpTyOrNone.getValue(), TempRegID);
2854 DstMIBuilder.addRenderer<TempRegRenderer>(TempRegID);
2855
2856 auto InsertPtOrError = createAndImportSubInstructionRenderer(
2857 ++InsertPt, Rule, DstChild, TempRegID);
2858 if (auto Error = InsertPtOrError.takeError())
2859 return std::move(Error);
2860 return InsertPtOrError.get();
2861 }
2862
Daniel Sanders2c269f62017-08-24 09:11:20 +00002863 return failedImport("Dst pattern child isn't a leaf node or an MBB" + llvm::to_string(*DstChild));
Daniel Sandersffc7d582017-03-29 15:37:18 +00002864 }
2865
Daniel Sandersf499b2b2017-11-30 18:48:35 +00002866 // It could be a specific immediate in which case we should just check for
2867 // that immediate.
2868 if (const IntInit *ChildIntInit =
2869 dyn_cast<IntInit>(DstChild->getLeafValue())) {
2870 DstMIBuilder.addRenderer<ImmRenderer>(ChildIntInit->getValue());
2871 return InsertPt;
2872 }
2873
Daniel Sandersffc7d582017-03-29 15:37:18 +00002874 // Otherwise, we're looking for a bog-standard RegisterClass operand.
Daniel Sandersffc7d582017-03-29 15:37:18 +00002875 if (auto *ChildDefInit = dyn_cast<DefInit>(DstChild->getLeafValue())) {
2876 auto *ChildRec = ChildDefInit->getDef();
2877
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +00002878 ArrayRef<TypeSetByHwMode> ChildTypes = DstChild->getExtTypes();
Daniel Sandersffc7d582017-03-29 15:37:18 +00002879 if (ChildTypes.size() != 1)
2880 return failedImport("Dst pattern child has multiple results");
2881
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +00002882 Optional<LLTCodeGen> OpTyOrNone = None;
2883 if (ChildTypes.front().isMachineValueType())
2884 OpTyOrNone = MVTToLLT(ChildTypes.front().getMachineValueType().SimpleTy);
Daniel Sandersffc7d582017-03-29 15:37:18 +00002885 if (!OpTyOrNone)
2886 return failedImport("Dst operand has an unsupported type");
2887
2888 if (ChildRec->isSubClassOf("Register")) {
Daniel Sanders198447a2017-11-01 00:29:47 +00002889 DstMIBuilder.addRenderer<AddRegisterRenderer>(ChildRec);
Daniel Sanders7438b262017-10-31 23:03:18 +00002890 return InsertPt;
Daniel Sandersffc7d582017-03-29 15:37:18 +00002891 }
2892
Daniel Sanders658541f2017-04-22 15:53:21 +00002893 if (ChildRec->isSubClassOf("RegisterClass") ||
Daniel Sanders4d4e7652017-10-09 18:14:53 +00002894 ChildRec->isSubClassOf("RegisterOperand") ||
2895 ChildRec->isSubClassOf("ValueType")) {
Daniel Sandersd66e0902017-10-23 18:19:24 +00002896 if (ChildRec->isSubClassOf("RegisterOperand") &&
2897 !ChildRec->isValueUnset("GIZeroRegister")) {
2898 DstMIBuilder.addRenderer<CopyOrAddZeroRegRenderer>(
Daniel Sanders198447a2017-11-01 00:29:47 +00002899 DstChild->getName(), ChildRec->getValueAsDef("GIZeroRegister"));
Daniel Sanders7438b262017-10-31 23:03:18 +00002900 return InsertPt;
Daniel Sandersd66e0902017-10-23 18:19:24 +00002901 }
2902
Daniel Sanders198447a2017-11-01 00:29:47 +00002903 DstMIBuilder.addRenderer<CopyRenderer>(DstChild->getName());
Daniel Sanders7438b262017-10-31 23:03:18 +00002904 return InsertPt;
Daniel Sandersffc7d582017-03-29 15:37:18 +00002905 }
2906
2907 if (ChildRec->isSubClassOf("ComplexPattern")) {
2908 const auto &ComplexPattern = ComplexPatternEquivs.find(ChildRec);
2909 if (ComplexPattern == ComplexPatternEquivs.end())
2910 return failedImport(
2911 "SelectionDAG ComplexPattern not mapped to GlobalISel");
2912
Daniel Sandersbfa9e2c2017-10-14 00:31:58 +00002913 const OperandMatcher &OM = Rule.getOperandMatcher(DstChild->getName());
Daniel Sandersffc7d582017-03-29 15:37:18 +00002914 DstMIBuilder.addRenderer<RenderComplexPatternOperand>(
Daniel Sanders198447a2017-11-01 00:29:47 +00002915 *ComplexPattern->second, DstChild->getName(),
Daniel Sanders2deea182017-04-22 15:11:04 +00002916 OM.getAllocatedTemporariesBaseID());
Daniel Sanders7438b262017-10-31 23:03:18 +00002917 return InsertPt;
Daniel Sandersffc7d582017-03-29 15:37:18 +00002918 }
2919
Daniel Sandersd0656a32017-04-13 09:45:37 +00002920 if (ChildRec->isSubClassOf("SDNodeXForm"))
2921 return failedImport("Dst pattern child def is an unsupported tablegen "
2922 "class (SDNodeXForm)");
2923
Daniel Sandersffc7d582017-03-29 15:37:18 +00002924 return failedImport(
2925 "Dst pattern child def is an unsupported tablegen class");
2926 }
2927
2928 return failedImport("Dst pattern child is an unsupported kind");
2929}
2930
Daniel Sandersc270c502017-03-30 09:36:33 +00002931Expected<BuildMIAction &> GlobalISelEmitter::createAndImportInstructionRenderer(
Daniel Sandersa7b75262017-10-31 18:50:24 +00002932 RuleMatcher &M, const TreePatternNode *Dst) {
Daniel Sanders7438b262017-10-31 23:03:18 +00002933 auto InsertPtOrError = createInstructionRenderer(M.actions_end(), M, Dst);
2934 if (auto Error = InsertPtOrError.takeError())
Daniel Sandersdf258e32017-10-31 19:09:29 +00002935 return std::move(Error);
2936
Daniel Sanders7438b262017-10-31 23:03:18 +00002937 action_iterator InsertPt = InsertPtOrError.get();
2938 BuildMIAction &DstMIBuilder = *static_cast<BuildMIAction *>(InsertPt->get());
Daniel Sandersdf258e32017-10-31 19:09:29 +00002939
2940 importExplicitDefRenderers(DstMIBuilder);
2941
Daniel Sanders7438b262017-10-31 23:03:18 +00002942 if (auto Error = importExplicitUseRenderers(InsertPt, M, DstMIBuilder, Dst)
2943 .takeError())
Daniel Sandersdf258e32017-10-31 19:09:29 +00002944 return std::move(Error);
2945
2946 return DstMIBuilder;
2947}
2948
Daniel Sanders9cbe7c72017-11-01 19:57:57 +00002949Expected<action_iterator>
2950GlobalISelEmitter::createAndImportSubInstructionRenderer(
2951 action_iterator InsertPt, RuleMatcher &M, const TreePatternNode *Dst,
2952 unsigned TempRegID) {
2953 auto InsertPtOrError = createInstructionRenderer(InsertPt, M, Dst);
2954
2955 // TODO: Assert there's exactly one result.
2956
2957 if (auto Error = InsertPtOrError.takeError())
2958 return std::move(Error);
2959 InsertPt = InsertPtOrError.get();
2960
2961 BuildMIAction &DstMIBuilder =
2962 *static_cast<BuildMIAction *>(InsertPtOrError.get()->get());
2963
2964 // Assign the result to TempReg.
2965 DstMIBuilder.addRenderer<TempRegRenderer>(TempRegID, true);
2966
2967 InsertPtOrError = importExplicitUseRenderers(InsertPt, M, DstMIBuilder, Dst);
2968 if (auto Error = InsertPtOrError.takeError())
2969 return std::move(Error);
2970
2971 return InsertPtOrError.get();
2972}
2973
Daniel Sanders7438b262017-10-31 23:03:18 +00002974Expected<action_iterator> GlobalISelEmitter::createInstructionRenderer(
2975 action_iterator InsertPt, RuleMatcher &M, const TreePatternNode *Dst) {
Daniel Sandersffc7d582017-03-29 15:37:18 +00002976 Record *DstOp = Dst->getOperator();
Daniel Sandersd0656a32017-04-13 09:45:37 +00002977 if (!DstOp->isSubClassOf("Instruction")) {
2978 if (DstOp->isSubClassOf("ValueType"))
2979 return failedImport(
2980 "Pattern operator isn't an instruction (it's a ValueType)");
Daniel Sandersffc7d582017-03-29 15:37:18 +00002981 return failedImport("Pattern operator isn't an instruction");
Daniel Sandersd0656a32017-04-13 09:45:37 +00002982 }
Daniel Sandersa6e2ceb2017-06-20 12:36:34 +00002983 CodeGenInstruction *DstI = &Target.getInstruction(DstOp);
Daniel Sandersffc7d582017-03-29 15:37:18 +00002984
Daniel Sandersa6e2ceb2017-06-20 12:36:34 +00002985 // COPY_TO_REGCLASS is just a copy with a ConstrainOperandToRegClassAction
Daniel Sanderscc36dbf2017-06-27 10:11:39 +00002986 // attached. Similarly for EXTRACT_SUBREG except that's a subregister copy.
Daniel Sandersdf258e32017-10-31 19:09:29 +00002987 if (DstI->TheDef->getName() == "COPY_TO_REGCLASS")
Daniel Sandersa6e2ceb2017-06-20 12:36:34 +00002988 DstI = &Target.getInstruction(RK.getDef("COPY"));
Daniel Sandersdf258e32017-10-31 19:09:29 +00002989 else if (DstI->TheDef->getName() == "EXTRACT_SUBREG")
Daniel Sanderscc36dbf2017-06-27 10:11:39 +00002990 DstI = &Target.getInstruction(RK.getDef("COPY"));
Daniel Sanders9cbe7c72017-11-01 19:57:57 +00002991 else if (DstI->TheDef->getName() == "REG_SEQUENCE")
2992 return failedImport("Unable to emit REG_SEQUENCE");
Daniel Sandersa6e2ceb2017-06-20 12:36:34 +00002993
Daniel Sanders198447a2017-11-01 00:29:47 +00002994 return M.insertAction<BuildMIAction>(InsertPt, M.allocateOutputInsnID(),
2995 DstI);
Daniel Sandersdf258e32017-10-31 19:09:29 +00002996}
2997
2998void GlobalISelEmitter::importExplicitDefRenderers(
2999 BuildMIAction &DstMIBuilder) {
3000 const CodeGenInstruction *DstI = DstMIBuilder.getCGI();
Daniel Sandersa6e2ceb2017-06-20 12:36:34 +00003001 for (unsigned I = 0; I < DstI->Operands.NumDefs; ++I) {
3002 const CGIOperandList::OperandInfo &DstIOperand = DstI->Operands[I];
Daniel Sanders198447a2017-11-01 00:29:47 +00003003 DstMIBuilder.addRenderer<CopyRenderer>(DstIOperand.Name);
Daniel Sandersffc7d582017-03-29 15:37:18 +00003004 }
Daniel Sandersdf258e32017-10-31 19:09:29 +00003005}
3006
Daniel Sanders7438b262017-10-31 23:03:18 +00003007Expected<action_iterator> GlobalISelEmitter::importExplicitUseRenderers(
3008 action_iterator InsertPt, RuleMatcher &M, BuildMIAction &DstMIBuilder,
Daniel Sandersdf258e32017-10-31 19:09:29 +00003009 const llvm::TreePatternNode *Dst) {
3010 const CodeGenInstruction *DstI = DstMIBuilder.getCGI();
3011 CodeGenInstruction *OrigDstI = &Target.getInstruction(Dst->getOperator());
Daniel Sandersffc7d582017-03-29 15:37:18 +00003012
Daniel Sanderscc36dbf2017-06-27 10:11:39 +00003013 // EXTRACT_SUBREG needs to use a subregister COPY.
Daniel Sandersdf258e32017-10-31 19:09:29 +00003014 if (OrigDstI->TheDef->getName() == "EXTRACT_SUBREG") {
Daniel Sanderscc36dbf2017-06-27 10:11:39 +00003015 if (!Dst->getChild(0)->isLeaf())
3016 return failedImport("EXTRACT_SUBREG child #1 is not a leaf");
3017
Daniel Sanders32291982017-06-28 13:50:04 +00003018 if (DefInit *SubRegInit =
3019 dyn_cast<DefInit>(Dst->getChild(1)->getLeafValue())) {
Daniel Sanders9cbe7c72017-11-01 19:57:57 +00003020 Record *RCDef = getInitValueAsRegClass(Dst->getChild(0)->getLeafValue());
3021 if (!RCDef)
3022 return failedImport("EXTRACT_SUBREG child #0 could not "
3023 "be coerced to a register class");
3024
3025 CodeGenRegisterClass *RC = CGRegs.getRegClass(RCDef);
Daniel Sanderscc36dbf2017-06-27 10:11:39 +00003026 CodeGenSubRegIndex *SubIdx = CGRegs.getSubRegIdx(SubRegInit->getDef());
3027
3028 const auto &SrcRCDstRCPair =
3029 RC->getMatchingSubClassWithSubRegs(CGRegs, SubIdx);
3030 if (SrcRCDstRCPair.hasValue()) {
3031 assert(SrcRCDstRCPair->second && "Couldn't find a matching subclass");
3032 if (SrcRCDstRCPair->first != RC)
3033 return failedImport("EXTRACT_SUBREG requires an additional COPY");
3034 }
3035
Daniel Sanders198447a2017-11-01 00:29:47 +00003036 DstMIBuilder.addRenderer<CopySubRegRenderer>(Dst->getChild(0)->getName(),
3037 SubIdx);
Daniel Sanders7438b262017-10-31 23:03:18 +00003038 return InsertPt;
Daniel Sanderscc36dbf2017-06-27 10:11:39 +00003039 }
3040
3041 return failedImport("EXTRACT_SUBREG child #1 is not a subreg index");
3042 }
3043
Daniel Sandersffc7d582017-03-29 15:37:18 +00003044 // Render the explicit uses.
Daniel Sandersdf258e32017-10-31 19:09:29 +00003045 unsigned DstINumUses = OrigDstI->Operands.size() - OrigDstI->Operands.NumDefs;
3046 unsigned ExpectedDstINumUses = Dst->getNumChildren();
3047 if (OrigDstI->TheDef->getName() == "COPY_TO_REGCLASS") {
3048 DstINumUses--; // Ignore the class constraint.
3049 ExpectedDstINumUses--;
3050 }
3051
Daniel Sanders0ed28822017-04-12 08:23:08 +00003052 unsigned Child = 0;
Diana Picus382602f2017-05-17 08:57:28 +00003053 unsigned NumDefaultOps = 0;
Daniel Sanders0ed28822017-04-12 08:23:08 +00003054 for (unsigned I = 0; I != DstINumUses; ++I) {
Daniel Sandersa6e2ceb2017-06-20 12:36:34 +00003055 const CGIOperandList::OperandInfo &DstIOperand =
3056 DstI->Operands[DstI->Operands.NumDefs + I];
Daniel Sanders0ed28822017-04-12 08:23:08 +00003057
Diana Picus382602f2017-05-17 08:57:28 +00003058 // If the operand has default values, introduce them now.
3059 // FIXME: Until we have a decent test case that dictates we should do
3060 // otherwise, we're going to assume that operands with default values cannot
3061 // be specified in the patterns. Therefore, adding them will not cause us to
3062 // end up with too many rendered operands.
3063 if (DstIOperand.Rec->isSubClassOf("OperandWithDefaultOps")) {
Daniel Sanders0ed28822017-04-12 08:23:08 +00003064 DagInit *DefaultOps = DstIOperand.Rec->getValueAsDag("DefaultOps");
Diana Picus382602f2017-05-17 08:57:28 +00003065 if (auto Error = importDefaultOperandRenderers(DstMIBuilder, DefaultOps))
3066 return std::move(Error);
3067 ++NumDefaultOps;
Daniel Sanders0ed28822017-04-12 08:23:08 +00003068 continue;
3069 }
3070
Daniel Sanders7438b262017-10-31 23:03:18 +00003071 auto InsertPtOrError = importExplicitUseRenderer(InsertPt, M, DstMIBuilder,
3072 Dst->getChild(Child));
3073 if (auto Error = InsertPtOrError.takeError())
Daniel Sandersffc7d582017-03-29 15:37:18 +00003074 return std::move(Error);
Daniel Sanders7438b262017-10-31 23:03:18 +00003075 InsertPt = InsertPtOrError.get();
Daniel Sanders0ed28822017-04-12 08:23:08 +00003076 ++Child;
Daniel Sandersffc7d582017-03-29 15:37:18 +00003077 }
3078
Daniel Sandersa6e2ceb2017-06-20 12:36:34 +00003079 if (NumDefaultOps + ExpectedDstINumUses != DstINumUses)
Diana Picuseb2057c2017-05-17 09:25:08 +00003080 return failedImport("Expected " + llvm::to_string(DstINumUses) +
Diana Picus382602f2017-05-17 08:57:28 +00003081 " used operands but found " +
Daniel Sandersa6e2ceb2017-06-20 12:36:34 +00003082 llvm::to_string(ExpectedDstINumUses) +
Diana Picuseb2057c2017-05-17 09:25:08 +00003083 " explicit ones and " + llvm::to_string(NumDefaultOps) +
Diana Picus382602f2017-05-17 08:57:28 +00003084 " default ones");
3085
Daniel Sanders7438b262017-10-31 23:03:18 +00003086 return InsertPt;
Daniel Sandersffc7d582017-03-29 15:37:18 +00003087}
3088
Diana Picus382602f2017-05-17 08:57:28 +00003089Error GlobalISelEmitter::importDefaultOperandRenderers(
3090 BuildMIAction &DstMIBuilder, DagInit *DefaultOps) const {
Craig Topper481ff702017-05-29 21:49:34 +00003091 for (const auto *DefaultOp : DefaultOps->getArgs()) {
Diana Picus382602f2017-05-17 08:57:28 +00003092 // Look through ValueType operators.
3093 if (const DagInit *DefaultDagOp = dyn_cast<DagInit>(DefaultOp)) {
3094 if (const DefInit *DefaultDagOperator =
3095 dyn_cast<DefInit>(DefaultDagOp->getOperator())) {
3096 if (DefaultDagOperator->getDef()->isSubClassOf("ValueType"))
3097 DefaultOp = DefaultDagOp->getArg(0);
3098 }
3099 }
3100
3101 if (const DefInit *DefaultDefOp = dyn_cast<DefInit>(DefaultOp)) {
Daniel Sanders198447a2017-11-01 00:29:47 +00003102 DstMIBuilder.addRenderer<AddRegisterRenderer>(DefaultDefOp->getDef());
Diana Picus382602f2017-05-17 08:57:28 +00003103 continue;
3104 }
3105
3106 if (const IntInit *DefaultIntOp = dyn_cast<IntInit>(DefaultOp)) {
Daniel Sanders198447a2017-11-01 00:29:47 +00003107 DstMIBuilder.addRenderer<ImmRenderer>(DefaultIntOp->getValue());
Diana Picus382602f2017-05-17 08:57:28 +00003108 continue;
3109 }
3110
3111 return failedImport("Could not add default op");
3112 }
3113
3114 return Error::success();
3115}
3116
Daniel Sandersc270c502017-03-30 09:36:33 +00003117Error GlobalISelEmitter::importImplicitDefRenderers(
Daniel Sandersffc7d582017-03-29 15:37:18 +00003118 BuildMIAction &DstMIBuilder,
3119 const std::vector<Record *> &ImplicitDefs) const {
3120 if (!ImplicitDefs.empty())
3121 return failedImport("Pattern defines a physical register");
Daniel Sandersc270c502017-03-30 09:36:33 +00003122 return Error::success();
Daniel Sandersffc7d582017-03-29 15:37:18 +00003123}
3124
3125Expected<RuleMatcher> GlobalISelEmitter::runOnPattern(const PatternToMatch &P) {
Ahmed Bougacha36f70352016-12-21 23:26:20 +00003126 // Keep track of the matchers and actions to emit.
Daniel Sandersbfa9e2c2017-10-14 00:31:58 +00003127 RuleMatcher M(P.getSrcRecord()->getLoc());
Daniel Sanders6ea17ed2017-10-31 18:07:03 +00003128 M.addAction<DebugCommentAction>(llvm::to_string(*P.getSrcPattern()) +
3129 " => " +
3130 llvm::to_string(*P.getDstPattern()));
Ahmed Bougacha36f70352016-12-21 23:26:20 +00003131
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +00003132 if (auto Error = importRulePredicates(M, P.getPredicates()))
Daniel Sandersffc7d582017-03-29 15:37:18 +00003133 return std::move(Error);
Ahmed Bougacha36f70352016-12-21 23:26:20 +00003134
3135 // Next, analyze the pattern operators.
3136 TreePatternNode *Src = P.getSrcPattern();
3137 TreePatternNode *Dst = P.getDstPattern();
3138
3139 // If the root of either pattern isn't a simple operator, ignore it.
Daniel Sandersd0656a32017-04-13 09:45:37 +00003140 if (auto Err = isTrivialOperatorNode(Dst))
3141 return failedImport("Dst pattern root isn't a trivial operator (" +
3142 toString(std::move(Err)) + ")");
3143 if (auto Err = isTrivialOperatorNode(Src))
3144 return failedImport("Src pattern root isn't a trivial operator (" +
3145 toString(std::move(Err)) + ")");
Ahmed Bougacha36f70352016-12-21 23:26:20 +00003146
Daniel Sandersedd07842017-08-17 09:26:14 +00003147 InstructionMatcher &InsnMatcherTemp = M.addInstructionMatcher(Src->getName());
3148 unsigned TempOpIdx = 0;
3149 auto InsnMatcherOrError =
Daniel Sandersdf39cba2017-10-15 18:22:54 +00003150 createAndImportSelDAGMatcher(M, InsnMatcherTemp, Src, TempOpIdx);
Daniel Sandersedd07842017-08-17 09:26:14 +00003151 if (auto Error = InsnMatcherOrError.takeError())
3152 return std::move(Error);
3153 InstructionMatcher &InsnMatcher = InsnMatcherOrError.get();
3154
3155 if (Dst->isLeaf()) {
3156 Record *RCDef = getInitValueAsRegClass(Dst->getLeafValue());
3157
3158 const CodeGenRegisterClass &RC = Target.getRegisterClass(RCDef);
3159 if (RCDef) {
3160 // We need to replace the def and all its uses with the specified
3161 // operand. However, we must also insert COPY's wherever needed.
3162 // For now, emit a copy and let the register allocator clean up.
3163 auto &DstI = Target.getInstruction(RK.getDef("COPY"));
3164 const auto &DstIOperand = DstI.Operands[0];
3165
3166 OperandMatcher &OM0 = InsnMatcher.getOperand(0);
3167 OM0.setSymbolicName(DstIOperand.Name);
Daniel Sandersbfa9e2c2017-10-14 00:31:58 +00003168 M.defineOperand(OM0.getSymbolicName(), OM0);
Daniel Sandersedd07842017-08-17 09:26:14 +00003169 OM0.addPredicate<RegisterBankOperandMatcher>(RC);
3170
Daniel Sanders198447a2017-11-01 00:29:47 +00003171 auto &DstMIBuilder =
3172 M.addAction<BuildMIAction>(M.allocateOutputInsnID(), &DstI);
3173 DstMIBuilder.addRenderer<CopyRenderer>(DstIOperand.Name);
3174 DstMIBuilder.addRenderer<CopyRenderer>(Dst->getName());
Daniel Sandersedd07842017-08-17 09:26:14 +00003175 M.addAction<ConstrainOperandToRegClassAction>(0, 0, RC);
3176
3177 // We're done with this pattern! It's eligible for GISel emission; return
3178 // it.
3179 ++NumPatternImported;
3180 return std::move(M);
3181 }
3182
Daniel Sanders452c8ae2017-05-23 19:33:16 +00003183 return failedImport("Dst pattern root isn't a known leaf");
Daniel Sandersedd07842017-08-17 09:26:14 +00003184 }
Daniel Sanders452c8ae2017-05-23 19:33:16 +00003185
Daniel Sandersbee57392017-04-04 13:25:23 +00003186 // Start with the defined operands (i.e., the results of the root operator).
Ahmed Bougacha36f70352016-12-21 23:26:20 +00003187 Record *DstOp = Dst->getOperator();
3188 if (!DstOp->isSubClassOf("Instruction"))
Ahmed Bougacha982c5eb2017-02-10 04:00:17 +00003189 return failedImport("Pattern operator isn't an instruction");
Ahmed Bougacha36f70352016-12-21 23:26:20 +00003190
3191 auto &DstI = Target.getInstruction(DstOp);
Ahmed Bougacha36f70352016-12-21 23:26:20 +00003192 if (DstI.Operands.NumDefs != Src->getExtTypes().size())
Daniel Sandersd0656a32017-04-13 09:45:37 +00003193 return failedImport("Src pattern results and dst MI defs are different (" +
3194 to_string(Src->getExtTypes().size()) + " def(s) vs " +
3195 to_string(DstI.Operands.NumDefs) + " def(s))");
Ahmed Bougacha36f70352016-12-21 23:26:20 +00003196
Daniel Sandersffc7d582017-03-29 15:37:18 +00003197 // The root of the match also has constraints on the register bank so that it
3198 // matches the result instruction.
3199 unsigned OpIdx = 0;
Krzysztof Parzyszek779d98e2017-09-14 16:56:21 +00003200 for (const TypeSetByHwMode &VTy : Src->getExtTypes()) {
3201 (void)VTy;
Daniel Sandersffc7d582017-03-29 15:37:18 +00003202
Daniel Sanders066ebbf2017-02-24 15:43:30 +00003203 const auto &DstIOperand = DstI.Operands[OpIdx];
3204 Record *DstIOpRec = DstIOperand.Rec;
Daniel Sandersa6e2ceb2017-06-20 12:36:34 +00003205 if (DstI.TheDef->getName() == "COPY_TO_REGCLASS") {
3206 DstIOpRec = getInitValueAsRegClass(Dst->getChild(1)->getLeafValue());
3207
3208 if (DstIOpRec == nullptr)
3209 return failedImport(
3210 "COPY_TO_REGCLASS operand #1 isn't a register class");
Daniel Sanderscc36dbf2017-06-27 10:11:39 +00003211 } else if (DstI.TheDef->getName() == "EXTRACT_SUBREG") {
3212 if (!Dst->getChild(0)->isLeaf())
3213 return failedImport("EXTRACT_SUBREG operand #0 isn't a leaf");
3214
Daniel Sanders32291982017-06-28 13:50:04 +00003215 // We can assume that a subregister is in the same bank as it's super
3216 // register.
Daniel Sanderscc36dbf2017-06-27 10:11:39 +00003217 DstIOpRec = getInitValueAsRegClass(Dst->getChild(0)->getLeafValue());
3218
3219 if (DstIOpRec == nullptr)
3220 return failedImport(
3221 "EXTRACT_SUBREG operand #0 isn't a register class");
Daniel Sandersa6e2ceb2017-06-20 12:36:34 +00003222 } else if (DstIOpRec->isSubClassOf("RegisterOperand"))
Daniel Sanders658541f2017-04-22 15:53:21 +00003223 DstIOpRec = DstIOpRec->getValueAsDef("RegClass");
Daniel Sandersa6e2ceb2017-06-20 12:36:34 +00003224 else if (!DstIOpRec->isSubClassOf("RegisterClass"))
Daniel Sanders32291982017-06-28 13:50:04 +00003225 return failedImport("Dst MI def isn't a register class" +
3226 to_string(*Dst));
Ahmed Bougacha36f70352016-12-21 23:26:20 +00003227
Daniel Sandersffc7d582017-03-29 15:37:18 +00003228 OperandMatcher &OM = InsnMatcher.getOperand(OpIdx);
3229 OM.setSymbolicName(DstIOperand.Name);
Daniel Sandersbfa9e2c2017-10-14 00:31:58 +00003230 M.defineOperand(OM.getSymbolicName(), OM);
Daniel Sandersdc662ff2017-01-26 11:10:14 +00003231 OM.addPredicate<RegisterBankOperandMatcher>(
3232 Target.getRegisterClass(DstIOpRec));
Ahmed Bougacha36f70352016-12-21 23:26:20 +00003233 ++OpIdx;
3234 }
3235
Daniel Sandersa7b75262017-10-31 18:50:24 +00003236 auto DstMIBuilderOrError = createAndImportInstructionRenderer(M, Dst);
Daniel Sandersffc7d582017-03-29 15:37:18 +00003237 if (auto Error = DstMIBuilderOrError.takeError())
3238 return std::move(Error);
3239 BuildMIAction &DstMIBuilder = DstMIBuilderOrError.get();
Ahmed Bougacha36f70352016-12-21 23:26:20 +00003240
Daniel Sandersffc7d582017-03-29 15:37:18 +00003241 // Render the implicit defs.
3242 // These are only added to the root of the result.
Daniel Sandersc270c502017-03-30 09:36:33 +00003243 if (auto Error = importImplicitDefRenderers(DstMIBuilder, P.getDstRegs()))
Daniel Sandersffc7d582017-03-29 15:37:18 +00003244 return std::move(Error);
Ahmed Bougacha36f70352016-12-21 23:26:20 +00003245
Daniel Sandersa7b75262017-10-31 18:50:24 +00003246 DstMIBuilder.chooseInsnToMutate(M);
3247
Daniel Sandersa6e2ceb2017-06-20 12:36:34 +00003248 // Constrain the registers to classes. This is normally derived from the
3249 // emitted instruction but a few instructions require special handling.
3250 if (DstI.TheDef->getName() == "COPY_TO_REGCLASS") {
3251 // COPY_TO_REGCLASS does not provide operand constraints itself but the
3252 // result is constrained to the class given by the second child.
3253 Record *DstIOpRec =
3254 getInitValueAsRegClass(Dst->getChild(1)->getLeafValue());
3255
3256 if (DstIOpRec == nullptr)
3257 return failedImport("COPY_TO_REGCLASS operand #1 isn't a register class");
3258
3259 M.addAction<ConstrainOperandToRegClassAction>(
Daniel Sandersd93a35a2017-07-05 09:39:33 +00003260 0, 0, Target.getRegisterClass(DstIOpRec));
Daniel Sanderscc36dbf2017-06-27 10:11:39 +00003261
3262 // We're done with this pattern! It's eligible for GISel emission; return
3263 // it.
3264 ++NumPatternImported;
3265 return std::move(M);
3266 }
3267
3268 if (DstI.TheDef->getName() == "EXTRACT_SUBREG") {
3269 // EXTRACT_SUBREG selects into a subregister COPY but unlike most
3270 // instructions, the result register class is controlled by the
3271 // subregisters of the operand. As a result, we must constrain the result
3272 // class rather than check that it's already the right one.
3273 if (!Dst->getChild(0)->isLeaf())
3274 return failedImport("EXTRACT_SUBREG child #1 is not a leaf");
3275
Daniel Sanders320390b2017-06-28 15:16:03 +00003276 DefInit *SubRegInit = dyn_cast<DefInit>(Dst->getChild(1)->getLeafValue());
3277 if (!SubRegInit)
3278 return failedImport("EXTRACT_SUBREG child #1 is not a subreg index");
Daniel Sanderscc36dbf2017-06-27 10:11:39 +00003279
Daniel Sanders320390b2017-06-28 15:16:03 +00003280 // Constrain the result to the same register bank as the operand.
3281 Record *DstIOpRec =
3282 getInitValueAsRegClass(Dst->getChild(0)->getLeafValue());
Daniel Sanderscc36dbf2017-06-27 10:11:39 +00003283
Daniel Sanders320390b2017-06-28 15:16:03 +00003284 if (DstIOpRec == nullptr)
3285 return failedImport("EXTRACT_SUBREG operand #1 isn't a register class");
Daniel Sanderscc36dbf2017-06-27 10:11:39 +00003286
Daniel Sanders320390b2017-06-28 15:16:03 +00003287 CodeGenSubRegIndex *SubIdx = CGRegs.getSubRegIdx(SubRegInit->getDef());
Daniel Sandersd93a35a2017-07-05 09:39:33 +00003288 CodeGenRegisterClass *SrcRC = CGRegs.getRegClass(DstIOpRec);
Daniel Sanderscc36dbf2017-06-27 10:11:39 +00003289
Daniel Sanders320390b2017-06-28 15:16:03 +00003290 // It would be nice to leave this constraint implicit but we're required
3291 // to pick a register class so constrain the result to a register class
3292 // that can hold the correct MVT.
3293 //
3294 // FIXME: This may introduce an extra copy if the chosen class doesn't
3295 // actually contain the subregisters.
3296 assert(Src->getExtTypes().size() == 1 &&
3297 "Expected Src of EXTRACT_SUBREG to have one result type");
Daniel Sanderscc36dbf2017-06-27 10:11:39 +00003298
Daniel Sanders320390b2017-06-28 15:16:03 +00003299 const auto &SrcRCDstRCPair =
3300 SrcRC->getMatchingSubClassWithSubRegs(CGRegs, SubIdx);
3301 assert(SrcRCDstRCPair->second && "Couldn't find a matching subclass");
Daniel Sandersd93a35a2017-07-05 09:39:33 +00003302 M.addAction<ConstrainOperandToRegClassAction>(0, 0, *SrcRCDstRCPair->second);
3303 M.addAction<ConstrainOperandToRegClassAction>(0, 1, *SrcRCDstRCPair->first);
3304
3305 // We're done with this pattern! It's eligible for GISel emission; return
3306 // it.
3307 ++NumPatternImported;
3308 return std::move(M);
3309 }
3310
3311 M.addAction<ConstrainOperandsToDefinitionAction>(0);
Daniel Sandersa6e2ceb2017-06-20 12:36:34 +00003312
Ahmed Bougacha982c5eb2017-02-10 04:00:17 +00003313 // We're done with this pattern! It's eligible for GISel emission; return it.
Daniel Sandersb41ce2b2017-02-20 14:31:27 +00003314 ++NumPatternImported;
Ahmed Bougacha982c5eb2017-02-10 04:00:17 +00003315 return std::move(M);
Ahmed Bougacha36f70352016-12-21 23:26:20 +00003316}
3317
Daniel Sanders649c5852017-10-13 20:42:18 +00003318// Emit imm predicate table and an enum to reference them with.
3319// The 'Predicate_' part of the name is redundant but eliminating it is more
3320// trouble than it's worth.
3321void GlobalISelEmitter::emitImmPredicates(
Daniel Sanders11300ce2017-10-13 21:28:03 +00003322 raw_ostream &OS, StringRef TypeIdentifier, StringRef Type,
3323 std::function<bool(const Record *R)> Filter) {
Daniel Sanders649c5852017-10-13 20:42:18 +00003324 std::vector<const Record *> MatchedRecords;
3325 const auto &Defs = RK.getAllDerivedDefinitions("PatFrag");
3326 std::copy_if(Defs.begin(), Defs.end(), std::back_inserter(MatchedRecords),
3327 [&](Record *Record) {
3328 return !Record->getValueAsString("ImmediateCode").empty() &&
3329 Filter(Record);
3330 });
3331
Daniel Sanders11300ce2017-10-13 21:28:03 +00003332 if (!MatchedRecords.empty()) {
3333 OS << "// PatFrag predicates.\n"
3334 << "enum {\n";
Daniel Sanders2fed4ff2017-10-13 21:51:20 +00003335 std::string EnumeratorSeparator =
Daniel Sanders11300ce2017-10-13 21:28:03 +00003336 (" = GIPFP_" + TypeIdentifier + "_Invalid + 1,\n").str();
3337 for (const auto *Record : MatchedRecords) {
3338 OS << " GIPFP_" << TypeIdentifier << "_Predicate_" << Record->getName()
3339 << EnumeratorSeparator;
3340 EnumeratorSeparator = ",\n";
3341 }
3342 OS << "};\n";
Daniel Sanders649c5852017-10-13 20:42:18 +00003343 }
Daniel Sanders11300ce2017-10-13 21:28:03 +00003344
Daniel Sanders649c5852017-10-13 20:42:18 +00003345 for (const auto *Record : MatchedRecords)
Daniel Sanders11300ce2017-10-13 21:28:03 +00003346 OS << "static bool Predicate_" << Record->getName() << "(" << Type
3347 << " Imm) {" << Record->getValueAsString("ImmediateCode") << "}\n";
3348
3349 OS << "static InstructionSelector::" << TypeIdentifier
3350 << "ImmediatePredicateFn " << TypeIdentifier << "ImmPredicateFns[] = {\n"
Daniel Sanders649c5852017-10-13 20:42:18 +00003351 << " nullptr,\n";
3352 for (const auto *Record : MatchedRecords)
3353 OS << " Predicate_" << Record->getName() << ",\n";
3354 OS << "};\n";
3355}
3356
Ahmed Bougacha36f70352016-12-21 23:26:20 +00003357void GlobalISelEmitter::run(raw_ostream &OS) {
Daniel Sandersf76f3152017-11-16 00:46:35 +00003358 if (!UseCoverageFile.empty()) {
3359 RuleCoverage = CodeGenCoverage();
3360 auto RuleCoverageBufOrErr = MemoryBuffer::getFile(UseCoverageFile);
3361 if (!RuleCoverageBufOrErr) {
3362 PrintWarning(SMLoc(), "Missing rule coverage data");
3363 RuleCoverage = None;
3364 } else {
3365 if (!RuleCoverage->parse(*RuleCoverageBufOrErr.get(), Target.getName())) {
3366 PrintWarning(SMLoc(), "Ignoring invalid or missing rule coverage data");
3367 RuleCoverage = None;
3368 }
3369 }
3370 }
3371
Ahmed Bougacha36f70352016-12-21 23:26:20 +00003372 // Track the GINodeEquiv definitions.
3373 gatherNodeEquivs();
3374
3375 emitSourceFileHeader(("Global Instruction Selector for the " +
3376 Target.getName() + " target").str(), OS);
Daniel Sandersb41ce2b2017-02-20 14:31:27 +00003377 std::vector<RuleMatcher> Rules;
Ahmed Bougacha36f70352016-12-21 23:26:20 +00003378 // Look through the SelectionDAG patterns we found, possibly emitting some.
3379 for (const PatternToMatch &Pat : CGP.ptms()) {
3380 ++NumPatternTotal;
Daniel Sanders7e523672017-11-11 03:23:44 +00003381
Ahmed Bougacha982c5eb2017-02-10 04:00:17 +00003382 auto MatcherOrErr = runOnPattern(Pat);
3383
3384 // The pattern analysis can fail, indicating an unsupported pattern.
3385 // Report that if we've been asked to do so.
3386 if (auto Err = MatcherOrErr.takeError()) {
Ahmed Bougacha36f70352016-12-21 23:26:20 +00003387 if (WarnOnSkippedPatterns) {
3388 PrintWarning(Pat.getSrcRecord()->getLoc(),
Ahmed Bougacha982c5eb2017-02-10 04:00:17 +00003389 "Skipped pattern: " + toString(std::move(Err)));
3390 } else {
3391 consumeError(std::move(Err));
Ahmed Bougacha36f70352016-12-21 23:26:20 +00003392 }
Daniel Sandersb41ce2b2017-02-20 14:31:27 +00003393 ++NumPatternImportsSkipped;
Ahmed Bougacha982c5eb2017-02-10 04:00:17 +00003394 continue;
Ahmed Bougacha36f70352016-12-21 23:26:20 +00003395 }
Ahmed Bougacha982c5eb2017-02-10 04:00:17 +00003396
Daniel Sandersf76f3152017-11-16 00:46:35 +00003397 if (RuleCoverage) {
3398 if (RuleCoverage->isCovered(MatcherOrErr->getRuleID()))
3399 ++NumPatternsTested;
3400 else
3401 PrintWarning(Pat.getSrcRecord()->getLoc(),
3402 "Pattern is not covered by a test");
3403 }
Daniel Sandersb41ce2b2017-02-20 14:31:27 +00003404 Rules.push_back(std::move(MatcherOrErr.get()));
3405 }
3406
Daniel Sanderseb2f5f32017-08-15 15:10:31 +00003407 std::stable_sort(Rules.begin(), Rules.end(),
3408 [&](const RuleMatcher &A, const RuleMatcher &B) {
3409 if (A.isHigherPriorityThan(B)) {
3410 assert(!B.isHigherPriorityThan(A) && "Cannot be more important "
3411 "and less important at "
3412 "the same time");
3413 return true;
3414 }
3415 return false;
3416 });
Daniel Sanders759ff412017-02-24 13:58:11 +00003417
Daniel Sanders6ab0daa2017-07-04 14:35:06 +00003418 std::vector<Record *> ComplexPredicates =
3419 RK.getAllDerivedDefinitions("GIComplexOperandMatcher");
3420 std::sort(ComplexPredicates.begin(), ComplexPredicates.end(),
3421 [](const Record *A, const Record *B) {
3422 if (A->getName() < B->getName())
3423 return true;
3424 return false;
3425 });
Daniel Sanders8a4bae92017-03-14 21:32:08 +00003426 unsigned MaxTemporaries = 0;
3427 for (const auto &Rule : Rules)
Daniel Sanders2deea182017-04-22 15:11:04 +00003428 MaxTemporaries = std::max(MaxTemporaries, Rule.countRendererFns());
Daniel Sanders8a4bae92017-03-14 21:32:08 +00003429
Daniel Sanderse7b0d662017-04-21 15:59:56 +00003430 OS << "#ifdef GET_GLOBALISEL_PREDICATE_BITSET\n"
3431 << "const unsigned MAX_SUBTARGET_PREDICATES = " << SubtargetFeatures.size()
3432 << ";\n"
3433 << "using PredicateBitset = "
3434 "llvm::PredicateBitsetImpl<MAX_SUBTARGET_PREDICATES>;\n"
3435 << "#endif // ifdef GET_GLOBALISEL_PREDICATE_BITSET\n\n";
3436
Daniel Sanders6ab0daa2017-07-04 14:35:06 +00003437 OS << "#ifdef GET_GLOBALISEL_TEMPORARIES_DECL\n"
3438 << " mutable MatcherState State;\n"
3439 << " typedef "
Daniel Sanders1e4569f2017-10-20 20:55:29 +00003440 "ComplexRendererFns("
Daniel Sanders6ab0daa2017-07-04 14:35:06 +00003441 << Target.getName()
3442 << "InstructionSelector::*ComplexMatcherMemFn)(MachineOperand &) const;\n"
Daniel Sandersea8711b2017-10-16 03:36:29 +00003443 << " const MatcherInfoTy<PredicateBitset, ComplexMatcherMemFn> "
Daniel Sanders6ab0daa2017-07-04 14:35:06 +00003444 "MatcherInfo;\n"
Daniel Sandersea8711b2017-10-16 03:36:29 +00003445 << " static " << Target.getName()
3446 << "InstructionSelector::ComplexMatcherMemFn ComplexPredicateFns[];\n"
Daniel Sanders6ab0daa2017-07-04 14:35:06 +00003447 << "#endif // ifdef GET_GLOBALISEL_TEMPORARIES_DECL\n\n";
Daniel Sanders8a4bae92017-03-14 21:32:08 +00003448
Daniel Sanders6ab0daa2017-07-04 14:35:06 +00003449 OS << "#ifdef GET_GLOBALISEL_TEMPORARIES_INIT\n"
3450 << ", State(" << MaxTemporaries << "),\n"
Daniel Sanders11300ce2017-10-13 21:28:03 +00003451 << "MatcherInfo({TypeObjects, FeatureBitsets, I64ImmPredicateFns, "
Daniel Sandersea8711b2017-10-16 03:36:29 +00003452 "APIntImmPredicateFns, APFloatImmPredicateFns, ComplexPredicateFns})\n"
Daniel Sanders6ab0daa2017-07-04 14:35:06 +00003453 << "#endif // ifdef GET_GLOBALISEL_TEMPORARIES_INIT\n\n";
Daniel Sanders8a4bae92017-03-14 21:32:08 +00003454
Daniel Sanderse7b0d662017-04-21 15:59:56 +00003455 OS << "#ifdef GET_GLOBALISEL_IMPL\n";
3456 SubtargetFeatureInfo::emitSubtargetFeatureBitEnumeration(SubtargetFeatures,
3457 OS);
Daniel Sanderse9fdba32017-04-29 17:30:09 +00003458
3459 // Separate subtarget features by how often they must be recomputed.
3460 SubtargetFeatureInfoMap ModuleFeatures;
3461 std::copy_if(SubtargetFeatures.begin(), SubtargetFeatures.end(),
3462 std::inserter(ModuleFeatures, ModuleFeatures.end()),
3463 [](const SubtargetFeatureInfoMap::value_type &X) {
3464 return !X.second.mustRecomputePerFunction();
3465 });
3466 SubtargetFeatureInfoMap FunctionFeatures;
3467 std::copy_if(SubtargetFeatures.begin(), SubtargetFeatures.end(),
3468 std::inserter(FunctionFeatures, FunctionFeatures.end()),
3469 [](const SubtargetFeatureInfoMap::value_type &X) {
3470 return X.second.mustRecomputePerFunction();
3471 });
3472
Daniel Sanderse7b0d662017-04-21 15:59:56 +00003473 SubtargetFeatureInfo::emitComputeAvailableFeatures(
Daniel Sanderse9fdba32017-04-29 17:30:09 +00003474 Target.getName(), "InstructionSelector", "computeAvailableModuleFeatures",
3475 ModuleFeatures, OS);
3476 SubtargetFeatureInfo::emitComputeAvailableFeatures(
3477 Target.getName(), "InstructionSelector",
3478 "computeAvailableFunctionFeatures", FunctionFeatures, OS,
3479 "const MachineFunction *MF");
Daniel Sanderse7b0d662017-04-21 15:59:56 +00003480
Daniel Sanders6ab0daa2017-07-04 14:35:06 +00003481 // Emit a table containing the LLT objects needed by the matcher and an enum
3482 // for the matcher to reference them with.
Daniel Sanders032e7f22017-08-17 13:18:35 +00003483 std::vector<LLTCodeGen> TypeObjects;
3484 for (const auto &Ty : LLTOperandMatcher::KnownTypes)
3485 TypeObjects.push_back(Ty);
Daniel Sanders6ab0daa2017-07-04 14:35:06 +00003486 std::sort(TypeObjects.begin(), TypeObjects.end());
Daniel Sanders49980702017-08-23 10:09:25 +00003487 OS << "// LLT Objects.\n"
3488 << "enum {\n";
Daniel Sanders6ab0daa2017-07-04 14:35:06 +00003489 for (const auto &TypeObject : TypeObjects) {
3490 OS << " ";
3491 TypeObject.emitCxxEnumValue(OS);
3492 OS << ",\n";
3493 }
3494 OS << "};\n"
3495 << "const static LLT TypeObjects[] = {\n";
3496 for (const auto &TypeObject : TypeObjects) {
3497 OS << " ";
3498 TypeObject.emitCxxConstructorCall(OS);
3499 OS << ",\n";
3500 }
3501 OS << "};\n\n";
3502
3503 // Emit a table containing the PredicateBitsets objects needed by the matcher
3504 // and an enum for the matcher to reference them with.
3505 std::vector<std::vector<Record *>> FeatureBitsets;
3506 for (auto &Rule : Rules)
3507 FeatureBitsets.push_back(Rule.getRequiredFeatures());
3508 std::sort(
3509 FeatureBitsets.begin(), FeatureBitsets.end(),
3510 [&](const std::vector<Record *> &A, const std::vector<Record *> &B) {
3511 if (A.size() < B.size())
3512 return true;
3513 if (A.size() > B.size())
3514 return false;
3515 for (const auto &Pair : zip(A, B)) {
3516 if (std::get<0>(Pair)->getName() < std::get<1>(Pair)->getName())
3517 return true;
3518 if (std::get<0>(Pair)->getName() > std::get<1>(Pair)->getName())
3519 return false;
3520 }
3521 return false;
3522 });
3523 FeatureBitsets.erase(
3524 std::unique(FeatureBitsets.begin(), FeatureBitsets.end()),
3525 FeatureBitsets.end());
Daniel Sanders49980702017-08-23 10:09:25 +00003526 OS << "// Feature bitsets.\n"
3527 << "enum {\n"
Daniel Sanders6ab0daa2017-07-04 14:35:06 +00003528 << " GIFBS_Invalid,\n";
3529 for (const auto &FeatureBitset : FeatureBitsets) {
3530 if (FeatureBitset.empty())
3531 continue;
3532 OS << " " << getNameForFeatureBitset(FeatureBitset) << ",\n";
3533 }
3534 OS << "};\n"
3535 << "const static PredicateBitset FeatureBitsets[] {\n"
3536 << " {}, // GIFBS_Invalid\n";
3537 for (const auto &FeatureBitset : FeatureBitsets) {
3538 if (FeatureBitset.empty())
3539 continue;
3540 OS << " {";
3541 for (const auto &Feature : FeatureBitset) {
3542 const auto &I = SubtargetFeatures.find(Feature);
3543 assert(I != SubtargetFeatures.end() && "Didn't import predicate?");
3544 OS << I->second.getEnumBitName() << ", ";
3545 }
3546 OS << "},\n";
3547 }
3548 OS << "};\n\n";
3549
3550 // Emit complex predicate table and an enum to reference them with.
Daniel Sanders49980702017-08-23 10:09:25 +00003551 OS << "// ComplexPattern predicates.\n"
3552 << "enum {\n"
Daniel Sanders6ab0daa2017-07-04 14:35:06 +00003553 << " GICP_Invalid,\n";
3554 for (const auto &Record : ComplexPredicates)
3555 OS << " GICP_" << Record->getName() << ",\n";
3556 OS << "};\n"
3557 << "// See constructor for table contents\n\n";
3558
Daniel Sanders11300ce2017-10-13 21:28:03 +00003559 emitImmPredicates(OS, "I64", "int64_t", [](const Record *R) {
Daniel Sanders649c5852017-10-13 20:42:18 +00003560 bool Unset;
3561 return !R->getValueAsBitOrUnset("IsAPFloat", Unset) &&
3562 !R->getValueAsBit("IsAPInt");
3563 });
Daniel Sanders11300ce2017-10-13 21:28:03 +00003564 emitImmPredicates(OS, "APFloat", "const APFloat &", [](const Record *R) {
3565 bool Unset;
3566 return R->getValueAsBitOrUnset("IsAPFloat", Unset);
3567 });
3568 emitImmPredicates(OS, "APInt", "const APInt &", [](const Record *R) {
3569 return R->getValueAsBit("IsAPInt");
3570 });
Daniel Sandersea8711b2017-10-16 03:36:29 +00003571 OS << "\n";
3572
3573 OS << Target.getName() << "InstructionSelector::ComplexMatcherMemFn\n"
3574 << Target.getName() << "InstructionSelector::ComplexPredicateFns[] = {\n"
3575 << " nullptr, // GICP_Invalid\n";
3576 for (const auto &Record : ComplexPredicates)
3577 OS << " &" << Target.getName()
3578 << "InstructionSelector::" << Record->getValueAsString("MatcherFn")
3579 << ", // " << Record->getName() << "\n";
3580 OS << "};\n\n";
Daniel Sanders2c269f62017-08-24 09:11:20 +00003581
Daniel Sanderse7b0d662017-04-21 15:59:56 +00003582 OS << "bool " << Target.getName()
Daniel Sandersf76f3152017-11-16 00:46:35 +00003583 << "InstructionSelector::selectImpl(MachineInstr &I, CodeGenCoverage "
3584 "&CoverageInfo) const {\n"
Daniel Sanders8a4bae92017-03-14 21:32:08 +00003585 << " MachineFunction &MF = *I.getParent()->getParent();\n"
Daniel Sanders6ab0daa2017-07-04 14:35:06 +00003586 << " MachineRegisterInfo &MRI = MF.getRegInfo();\n"
Daniel Sanders32291982017-06-28 13:50:04 +00003587 << " // FIXME: This should be computed on a per-function basis rather "
3588 "than per-insn.\n"
3589 << " AvailableFunctionFeatures = computeAvailableFunctionFeatures(&STI, "
3590 "&MF);\n"
Daniel Sandersa6cfce62017-07-05 14:50:18 +00003591 << " const PredicateBitset AvailableFeatures = getAvailableFeatures();\n"
3592 << " NewMIVector OutMIs;\n"
3593 << " State.MIs.clear();\n"
3594 << " State.MIs.push_back(&I);\n\n";
Daniel Sanders8a4bae92017-03-14 21:32:08 +00003595
Daniel Sanders8e82af22017-07-27 11:03:45 +00003596 MatchTable Table(0);
Daniel Sandersb96f40d2017-03-20 15:20:42 +00003597 for (auto &Rule : Rules) {
Daniel Sanders8e82af22017-07-27 11:03:45 +00003598 Rule.emit(Table);
Ahmed Bougacha982c5eb2017-02-10 04:00:17 +00003599 ++NumPatternEmitted;
Ahmed Bougacha36f70352016-12-21 23:26:20 +00003600 }
Daniel Sanders8e82af22017-07-27 11:03:45 +00003601 Table << MatchTable::Opcode("GIM_Reject") << MatchTable::LineBreak;
3602 Table.emitDeclaration(OS);
3603 OS << " if (executeMatchTable(*this, OutMIs, State, MatcherInfo, ";
3604 Table.emitUse(OS);
Daniel Sandersf76f3152017-11-16 00:46:35 +00003605 OS << ", TII, MRI, TRI, RBI, AvailableFeatures, CoverageInfo)) {\n"
Daniel Sanders8e82af22017-07-27 11:03:45 +00003606 << " return true;\n"
3607 << " }\n\n";
Ahmed Bougacha36f70352016-12-21 23:26:20 +00003608
Daniel Sanders8a4bae92017-03-14 21:32:08 +00003609 OS << " return false;\n"
3610 << "}\n"
3611 << "#endif // ifdef GET_GLOBALISEL_IMPL\n";
Daniel Sanderse9fdba32017-04-29 17:30:09 +00003612
3613 OS << "#ifdef GET_GLOBALISEL_PREDICATES_DECL\n"
3614 << "PredicateBitset AvailableModuleFeatures;\n"
3615 << "mutable PredicateBitset AvailableFunctionFeatures;\n"
3616 << "PredicateBitset getAvailableFeatures() const {\n"
3617 << " return AvailableModuleFeatures | AvailableFunctionFeatures;\n"
3618 << "}\n"
3619 << "PredicateBitset\n"
3620 << "computeAvailableModuleFeatures(const " << Target.getName()
3621 << "Subtarget *Subtarget) const;\n"
3622 << "PredicateBitset\n"
3623 << "computeAvailableFunctionFeatures(const " << Target.getName()
3624 << "Subtarget *Subtarget,\n"
3625 << " const MachineFunction *MF) const;\n"
3626 << "#endif // ifdef GET_GLOBALISEL_PREDICATES_DECL\n";
3627
3628 OS << "#ifdef GET_GLOBALISEL_PREDICATES_INIT\n"
3629 << "AvailableModuleFeatures(computeAvailableModuleFeatures(&STI)),\n"
3630 << "AvailableFunctionFeatures()\n"
3631 << "#endif // ifdef GET_GLOBALISEL_PREDICATES_INIT\n";
Ahmed Bougacha36f70352016-12-21 23:26:20 +00003632}
3633
Daniel Sanderse7b0d662017-04-21 15:59:56 +00003634void GlobalISelEmitter::declareSubtargetFeature(Record *Predicate) {
3635 if (SubtargetFeatures.count(Predicate) == 0)
3636 SubtargetFeatures.emplace(
3637 Predicate, SubtargetFeatureInfo(Predicate, SubtargetFeatures.size()));
3638}
3639
Daniel Sanders7e523672017-11-11 03:23:44 +00003640TreePatternNode *GlobalISelEmitter::fixupPatternNode(TreePatternNode *N) {
3641 if (!N->isLeaf()) {
3642 for (unsigned I = 0, E = N->getNumChildren(); I < E; ++I) {
3643 TreePatternNode *OrigChild = N->getChild(I);
3644 TreePatternNode *NewChild = fixupPatternNode(OrigChild);
3645 if (OrigChild != NewChild)
3646 N->setChild(I, NewChild);
3647 }
3648
3649 if (N->getOperator()->getName() == "ld") {
3650 // If it's a signext-load we need to adapt the pattern slightly. We need
3651 // to split the node into (sext (ld ...)), remove the <<signext>> predicate,
3652 // and then apply the <<signextTY>> predicate by updating the result type
3653 // of the load.
3654 //
3655 // For example:
3656 // (ld:[i32] [iPTR])<<unindexed>><<signext>><<signexti16>>
3657 // must be transformed into:
3658 // (sext:[i32] (ld:[i16] [iPTR])<<unindexed>>)
3659 //
Daniel Sandersb78ac6e2017-11-13 18:30:23 +00003660 // Likewise for zeroext-load and anyext-load.
Daniel Sanders7e523672017-11-11 03:23:44 +00003661
3662 std::vector<TreePredicateFn> Predicates;
3663 bool IsSignExtLoad = false;
3664 bool IsZeroExtLoad = false;
Daniel Sandersb78ac6e2017-11-13 18:30:23 +00003665 bool IsAnyExtLoad = false;
Daniel Sanders7e523672017-11-11 03:23:44 +00003666 Record *MemVT = nullptr;
3667 for (const auto &P : N->getPredicateFns()) {
3668 if (P.isLoad() && P.isSignExtLoad()) {
3669 IsSignExtLoad = true;
3670 continue;
3671 }
3672 if (P.isLoad() && P.isZeroExtLoad()) {
3673 IsZeroExtLoad = true;
3674 continue;
3675 }
Daniel Sandersb78ac6e2017-11-13 18:30:23 +00003676 if (P.isLoad() && P.isAnyExtLoad()) {
3677 IsAnyExtLoad = true;
3678 continue;
3679 }
Daniel Sanders7e523672017-11-11 03:23:44 +00003680 if (P.isLoad() && P.getMemoryVT()) {
3681 MemVT = P.getMemoryVT();
3682 continue;
3683 }
3684 Predicates.push_back(P);
3685 }
3686
Daniel Sandersb78ac6e2017-11-13 18:30:23 +00003687 if ((IsSignExtLoad || IsZeroExtLoad || IsAnyExtLoad) && MemVT) {
3688 assert((IsSignExtLoad + IsZeroExtLoad + IsAnyExtLoad) == 1 &&
3689 "IsSignExtLoad, IsZeroExtLoad, IsAnyExtLoad are mutually exclusive");
Daniel Sanders7e523672017-11-11 03:23:44 +00003690 TreePatternNode *Ext = new TreePatternNode(
Daniel Sandersb78ac6e2017-11-13 18:30:23 +00003691 RK.getDef(IsSignExtLoad ? "sext"
3692 : IsZeroExtLoad ? "zext" : "anyext"),
3693 {N}, 1);
Daniel Sanders7e523672017-11-11 03:23:44 +00003694 Ext->setType(0, N->getType(0));
3695 N->clearPredicateFns();
3696 N->setPredicateFns(Predicates);
3697 N->setType(0, getValueType(MemVT));
3698 return Ext;
3699 }
3700 }
3701 }
3702
3703 return N;
3704}
3705
3706void GlobalISelEmitter::fixupPatternTrees(TreePattern *P) {
3707 for (unsigned I = 0, E = P->getNumTrees(); I < E; ++I) {
3708 TreePatternNode *OrigTree = P->getTree(I);
3709 TreePatternNode *NewTree = fixupPatternNode(OrigTree);
3710 if (OrigTree != NewTree)
3711 P->setTree(I, NewTree);
3712 }
3713}
3714
Ahmed Bougacha982c5eb2017-02-10 04:00:17 +00003715} // end anonymous namespace
3716
Ahmed Bougacha36f70352016-12-21 23:26:20 +00003717//===----------------------------------------------------------------------===//
3718
3719namespace llvm {
3720void EmitGlobalISel(RecordKeeper &RK, raw_ostream &OS) {
3721 GlobalISelEmitter(RK).run(OS);
3722}
3723} // End llvm namespace