1//===- DAGISelMatcherEmitter.cpp - Matcher Emitter ------------------------===//
2//
3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4// See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6//
7//===----------------------------------------------------------------------===//
8//
9// This file contains code to generate C++ code for a matcher.
10//
11//===----------------------------------------------------------------------===//
12
13#include "Basic/SDNodeProperties.h"
14#include "Common/CodeGenDAGPatterns.h"
15#include "Common/CodeGenInstruction.h"
16#include "Common/CodeGenRegisters.h"
17#include "Common/CodeGenTarget.h"
18#include "Common/DAGISelMatcher.h"
19#include "llvm/ADT/DenseMap.h"
20#include "llvm/ADT/MapVector.h"
21#include "llvm/ADT/StringMap.h"
22#include "llvm/ADT/TinyPtrVector.h"
23#include "llvm/Support/CommandLine.h"
24#include "llvm/Support/Format.h"
25#include "llvm/Support/SourceMgr.h"
26#include "llvm/TableGen/Error.h"
27#include "llvm/TableGen/Record.h"
28
29using namespace llvm;
30
31enum {
32 IndexWidth = 6,
33 FullIndexWidth = IndexWidth + 4,
34 HistOpcWidth = 40,
35};
36
37cl::OptionCategory DAGISelCat("Options for -gen-dag-isel");
38
39// To reduce generated source code size.
40static cl::opt<bool> OmitComments("omit-comments",
41 cl::desc("Do not generate comments"),
42 cl::init(Val: false), cl::cat(DAGISelCat));
43
44static cl::opt<bool> InstrumentCoverage(
45 "instrument-coverage",
46 cl::desc("Generates tables to help identify patterns matched"),
47 cl::init(Val: false), cl::cat(DAGISelCat));
48
49namespace {
50class MatcherTableEmitter {
51 const CodeGenDAGPatterns &CGP;
52
53 SmallVector<unsigned, Matcher::HighestKind + 1> OpcodeCounts;
54
55 std::vector<TreePattern *> NodePredicates;
56 std::vector<TreePattern *> NodePredicatesWithOperands;
57
58 // We de-duplicate the predicates by code string, and use this map to track
59 // all the patterns with "identical" predicates.
60 MapVector<std::string, TinyPtrVector<TreePattern *>, StringMap<unsigned>>
61 NodePredicatesByCodeToRun;
62
63 std::vector<std::string> PatternPredicates;
64
65 std::vector<const ComplexPattern *> ComplexPatterns;
66
67 DenseMap<Record *, unsigned> NodeXFormMap;
68 std::vector<Record *> NodeXForms;
69
70 std::vector<std::string> VecIncludeStrings;
71 MapVector<std::string, unsigned, StringMap<unsigned>> VecPatterns;
72
73 unsigned getPatternIdxFromTable(std::string &&P, std::string &&include_loc) {
74 const auto It = VecPatterns.find(Key: P);
75 if (It == VecPatterns.end()) {
76 VecPatterns.insert(KV: std::pair(std::move(P), VecPatterns.size()));
77 VecIncludeStrings.push_back(x: std::move(include_loc));
78 return VecIncludeStrings.size() - 1;
79 }
80 return It->second;
81 }
82
83public:
84 MatcherTableEmitter(const Matcher *TheMatcher, const CodeGenDAGPatterns &cgp)
85 : CGP(cgp), OpcodeCounts(Matcher::HighestKind + 1, 0) {
86 // Record the usage of ComplexPattern.
87 MapVector<const ComplexPattern *, unsigned> ComplexPatternUsage;
88 // Record the usage of PatternPredicate.
89 MapVector<StringRef, unsigned> PatternPredicateUsage;
90 // Record the usage of Predicate.
91 MapVector<TreePattern *, unsigned> PredicateUsage;
92
93 // Iterate the whole MatcherTable once and do some statistics.
94 std::function<void(const Matcher *)> Statistic = [&](const Matcher *N) {
95 while (N) {
96 if (auto *SM = dyn_cast<ScopeMatcher>(Val: N))
97 for (unsigned I = 0; I < SM->getNumChildren(); I++)
98 Statistic(SM->getChild(i: I));
99 else if (auto *SOM = dyn_cast<SwitchOpcodeMatcher>(Val: N))
100 for (unsigned I = 0; I < SOM->getNumCases(); I++)
101 Statistic(SOM->getCaseMatcher(i: I));
102 else if (auto *STM = dyn_cast<SwitchTypeMatcher>(Val: N))
103 for (unsigned I = 0; I < STM->getNumCases(); I++)
104 Statistic(STM->getCaseMatcher(i: I));
105 else if (auto *CPM = dyn_cast<CheckComplexPatMatcher>(Val: N))
106 ++ComplexPatternUsage[&CPM->getPattern()];
107 else if (auto *CPPM = dyn_cast<CheckPatternPredicateMatcher>(Val: N))
108 ++PatternPredicateUsage[CPPM->getPredicate()];
109 else if (auto *PM = dyn_cast<CheckPredicateMatcher>(Val: N))
110 ++PredicateUsage[PM->getPredicate().getOrigPatFragRecord()];
111 N = N->getNext();
112 }
113 };
114 Statistic(TheMatcher);
115
116 // Sort ComplexPatterns by usage.
117 std::vector<std::pair<const ComplexPattern *, unsigned>> ComplexPatternList(
118 ComplexPatternUsage.begin(), ComplexPatternUsage.end());
119 stable_sort(Range&: ComplexPatternList, C: [](const auto &A, const auto &B) {
120 return A.second > B.second;
121 });
122 for (const auto &ComplexPattern : ComplexPatternList)
123 ComplexPatterns.push_back(x: ComplexPattern.first);
124
125 // Sort PatternPredicates by usage.
126 std::vector<std::pair<std::string, unsigned>> PatternPredicateList(
127 PatternPredicateUsage.begin(), PatternPredicateUsage.end());
128 stable_sort(Range&: PatternPredicateList, C: [](const auto &A, const auto &B) {
129 return A.second > B.second;
130 });
131 for (const auto &PatternPredicate : PatternPredicateList)
132 PatternPredicates.push_back(x: PatternPredicate.first);
133
134 // Sort Predicates by usage.
135 // Merge predicates with same code.
136 for (const auto &Usage : PredicateUsage) {
137 TreePattern *TP = Usage.first;
138 TreePredicateFn Pred(TP);
139 NodePredicatesByCodeToRun[Pred.getCodeToRunOnSDNode()].push_back(NewVal: TP);
140 }
141
142 std::vector<std::pair<TreePattern *, unsigned>> PredicateList;
143 // Sum the usage.
144 for (auto &Predicate : NodePredicatesByCodeToRun) {
145 TinyPtrVector<TreePattern *> &TPs = Predicate.second;
146 stable_sort(Range&: TPs, C: [](const auto *A, const auto *B) {
147 return A->getRecord()->getName() < B->getRecord()->getName();
148 });
149 unsigned Uses = 0;
150 for (TreePattern *TP : TPs)
151 Uses += PredicateUsage[TP];
152
153 // We only add the first predicate here since they are with the same code.
154 PredicateList.push_back(x: {TPs[0], Uses});
155 }
156
157 stable_sort(Range&: PredicateList, C: [](const auto &A, const auto &B) {
158 return A.second > B.second;
159 });
160 for (const auto &Predicate : PredicateList) {
161 TreePattern *TP = Predicate.first;
162 if (TreePredicateFn(TP).usesOperands())
163 NodePredicatesWithOperands.push_back(x: TP);
164 else
165 NodePredicates.push_back(x: TP);
166 }
167 }
168
169 unsigned EmitMatcherList(const Matcher *N, const unsigned Indent,
170 unsigned StartIdx, raw_ostream &OS);
171
172 unsigned SizeMatcherList(Matcher *N, raw_ostream &OS);
173
174 void EmitPredicateFunctions(raw_ostream &OS);
175
176 void EmitHistogram(const Matcher *N, raw_ostream &OS);
177
178 void EmitPatternMatchTable(raw_ostream &OS);
179
180private:
181 void EmitNodePredicatesFunction(const std::vector<TreePattern *> &Preds,
182 StringRef Decl, raw_ostream &OS);
183
184 unsigned SizeMatcher(Matcher *N, raw_ostream &OS);
185
186 unsigned EmitMatcher(const Matcher *N, const unsigned Indent,
187 unsigned CurrentIdx, raw_ostream &OS);
188
189 unsigned getNodePredicate(TreePredicateFn Pred) {
190 // We use the first predicate.
191 TreePattern *PredPat =
192 NodePredicatesByCodeToRun[Pred.getCodeToRunOnSDNode()][0];
193 return Pred.usesOperands()
194 ? llvm::find(Range&: NodePredicatesWithOperands, Val: PredPat) -
195 NodePredicatesWithOperands.begin()
196 : llvm::find(Range&: NodePredicates, Val: PredPat) - NodePredicates.begin();
197 }
198
199 unsigned getPatternPredicate(StringRef PredName) {
200 return llvm::find(Range&: PatternPredicates, Val: PredName) - PatternPredicates.begin();
201 }
202 unsigned getComplexPat(const ComplexPattern &P) {
203 return llvm::find(Range&: ComplexPatterns, Val: &P) - ComplexPatterns.begin();
204 }
205
206 unsigned getNodeXFormID(Record *Rec) {
207 unsigned &Entry = NodeXFormMap[Rec];
208 if (Entry == 0) {
209 NodeXForms.push_back(x: Rec);
210 Entry = NodeXForms.size();
211 }
212 return Entry - 1;
213 }
214};
215} // end anonymous namespace.
216
217static std::string GetPatFromTreePatternNode(const TreePatternNode &N) {
218 std::string str;
219 raw_string_ostream Stream(str);
220 Stream << N;
221 return str;
222}
223
224static unsigned GetVBRSize(unsigned Val) {
225 if (Val <= 127)
226 return 1;
227
228 unsigned NumBytes = 0;
229 while (Val >= 128) {
230 Val >>= 7;
231 ++NumBytes;
232 }
233 return NumBytes + 1;
234}
235
236/// EmitVBRValue - Emit the specified value as a VBR, returning the number of
237/// bytes emitted.
238static unsigned EmitVBRValue(uint64_t Val, raw_ostream &OS) {
239 if (Val <= 127) {
240 OS << Val << ", ";
241 return 1;
242 }
243
244 uint64_t InVal = Val;
245 unsigned NumBytes = 0;
246 while (Val >= 128) {
247 OS << (Val & 127) << "|128,";
248 Val >>= 7;
249 ++NumBytes;
250 }
251 OS << Val;
252 if (!OmitComments)
253 OS << "/*" << InVal << "*/";
254 OS << ", ";
255 return NumBytes + 1;
256}
257
258/// Emit the specified signed value as a VBR. To improve compression we encode
259/// positive numbers shifted left by 1 and negative numbers negated and shifted
260/// left by 1 with bit 0 set.
261static unsigned EmitSignedVBRValue(uint64_t Val, raw_ostream &OS) {
262 if ((int64_t)Val >= 0)
263 Val = Val << 1;
264 else
265 Val = (-Val << 1) | 1;
266
267 return EmitVBRValue(Val, OS);
268}
269
270// This is expensive and slow.
271static std::string getIncludePath(const Record *R) {
272 std::string str;
273 raw_string_ostream Stream(str);
274 auto Locs = R->getLoc();
275 SMLoc L;
276 if (Locs.size() > 1) {
277 // Get where the pattern prototype was instantiated
278 L = Locs[1];
279 } else if (Locs.size() == 1) {
280 L = Locs[0];
281 }
282 unsigned CurBuf = SrcMgr.FindBufferContainingLoc(Loc: L);
283 assert(CurBuf && "Invalid or unspecified location!");
284
285 Stream << SrcMgr.getBufferInfo(i: CurBuf).Buffer->getBufferIdentifier() << ":"
286 << SrcMgr.FindLineNumber(Loc: L, BufferID: CurBuf);
287 return str;
288}
289
290/// This function traverses the matcher tree and sizes all the nodes
291/// that are children of the three kinds of nodes that have them.
292unsigned MatcherTableEmitter::SizeMatcherList(Matcher *N, raw_ostream &OS) {
293 unsigned Size = 0;
294 while (N) {
295 Size += SizeMatcher(N, OS);
296 N = N->getNext();
297 }
298 return Size;
299}
300
301/// This function sizes the children of the three kinds of nodes that
302/// have them. It does so by using special cases for those three
303/// nodes, but sharing the code in EmitMatcher() for the other kinds.
304unsigned MatcherTableEmitter::SizeMatcher(Matcher *N, raw_ostream &OS) {
305 unsigned Idx = 0;
306
307 ++OpcodeCounts[N->getKind()];
308 switch (N->getKind()) {
309 // The Scope matcher has its kind, a series of child size + child,
310 // and a trailing zero.
311 case Matcher::Scope: {
312 ScopeMatcher *SM = cast<ScopeMatcher>(Val: N);
313 assert(SM->getNext() == nullptr && "Scope matcher should not have next");
314 unsigned Size = 1; // Count the kind.
315 for (unsigned i = 0, e = SM->getNumChildren(); i != e; ++i) {
316 const unsigned ChildSize = SizeMatcherList(N: SM->getChild(i), OS);
317 assert(ChildSize != 0 && "Matcher cannot have child of size 0");
318 SM->getChild(i)->setSize(ChildSize);
319 Size += GetVBRSize(Val: ChildSize) + ChildSize; // Count VBR and child size.
320 }
321 ++Size; // Count the zero sentinel.
322 return Size;
323 }
324
325 // SwitchOpcode and SwitchType have their kind, a series of child size +
326 // opcode/type + child, and a trailing zero.
327 case Matcher::SwitchOpcode:
328 case Matcher::SwitchType: {
329 unsigned Size = 1; // Count the kind.
330 unsigned NumCases;
331 if (const SwitchOpcodeMatcher *SOM = dyn_cast<SwitchOpcodeMatcher>(Val: N))
332 NumCases = SOM->getNumCases();
333 else
334 NumCases = cast<SwitchTypeMatcher>(Val: N)->getNumCases();
335 for (unsigned i = 0, e = NumCases; i != e; ++i) {
336 Matcher *Child;
337 if (SwitchOpcodeMatcher *SOM = dyn_cast<SwitchOpcodeMatcher>(Val: N)) {
338 Child = SOM->getCaseMatcher(i);
339 Size += 2; // Count the child's opcode.
340 } else {
341 Child = cast<SwitchTypeMatcher>(Val: N)->getCaseMatcher(i);
342 ++Size; // Count the child's type.
343 }
344 const unsigned ChildSize = SizeMatcherList(N: Child, OS);
345 assert(ChildSize != 0 && "Matcher cannot have child of size 0");
346 Child->setSize(ChildSize);
347 Size += GetVBRSize(Val: ChildSize) + ChildSize; // Count VBR and child size.
348 }
349 ++Size; // Count the zero sentinel.
350 return Size;
351 }
352
353 default:
354 // Employ the matcher emitter to size other matchers.
355 return EmitMatcher(N, Indent: 0, CurrentIdx: Idx, OS);
356 }
357 llvm_unreachable("Unreachable");
358}
359
360static void BeginEmitFunction(raw_ostream &OS, StringRef RetType,
361 StringRef Decl, bool AddOverride) {
362 OS << "#ifdef GET_DAGISEL_DECL\n";
363 OS << RetType << ' ' << Decl;
364 if (AddOverride)
365 OS << " override";
366 OS << ";\n"
367 "#endif\n"
368 "#if defined(GET_DAGISEL_BODY) || DAGISEL_INLINE\n";
369 OS << RetType << " DAGISEL_CLASS_COLONCOLON " << Decl << "\n";
370 if (AddOverride) {
371 OS << "#if DAGISEL_INLINE\n"
372 " override\n"
373 "#endif\n";
374 }
375}
376
377static void EndEmitFunction(raw_ostream &OS) {
378 OS << "#endif // GET_DAGISEL_BODY\n\n";
379}
380
381void MatcherTableEmitter::EmitPatternMatchTable(raw_ostream &OS) {
382
383 assert(isUInt<16>(VecPatterns.size()) &&
384 "Using only 16 bits to encode offset into Pattern Table");
385 assert(VecPatterns.size() == VecIncludeStrings.size() &&
386 "The sizes of Pattern and include vectors should be the same");
387
388 BeginEmitFunction(OS, RetType: "StringRef", Decl: "getPatternForIndex(unsigned Index)",
389 AddOverride: true /*AddOverride*/);
390 OS << "{\n";
391 OS << "static const char *PATTERN_MATCH_TABLE[] = {\n";
392
393 for (const auto &It : VecPatterns) {
394 OS << "\"" << It.first << "\",\n";
395 }
396
397 OS << "\n};";
398 OS << "\nreturn StringRef(PATTERN_MATCH_TABLE[Index]);";
399 OS << "\n}\n";
400 EndEmitFunction(OS);
401
402 BeginEmitFunction(OS, RetType: "StringRef", Decl: "getIncludePathForIndex(unsigned Index)",
403 AddOverride: true /*AddOverride*/);
404 OS << "{\n";
405 OS << "static const char *INCLUDE_PATH_TABLE[] = {\n";
406
407 for (const auto &It : VecIncludeStrings) {
408 OS << "\"" << It << "\",\n";
409 }
410
411 OS << "\n};";
412 OS << "\nreturn StringRef(INCLUDE_PATH_TABLE[Index]);";
413 OS << "\n}\n";
414 EndEmitFunction(OS);
415}
416
417/// EmitMatcher - Emit bytes for the specified matcher and return
418/// the number of bytes emitted.
419unsigned MatcherTableEmitter::EmitMatcher(const Matcher *N,
420 const unsigned Indent,
421 unsigned CurrentIdx,
422 raw_ostream &OS) {
423 OS.indent(NumSpaces: Indent);
424
425 switch (N->getKind()) {
426 case Matcher::Scope: {
427 const ScopeMatcher *SM = cast<ScopeMatcher>(Val: N);
428 unsigned StartIdx = CurrentIdx;
429
430 // Emit all of the children.
431 for (unsigned i = 0, e = SM->getNumChildren(); i != e; ++i) {
432 if (i == 0) {
433 OS << "OPC_Scope, ";
434 ++CurrentIdx;
435 } else {
436 if (!OmitComments) {
437 OS << "/*" << format_decimal(N: CurrentIdx, Width: IndexWidth) << "*/";
438 OS.indent(NumSpaces: Indent) << "/*Scope*/ ";
439 } else
440 OS.indent(NumSpaces: Indent);
441 }
442
443 unsigned ChildSize = SM->getChild(i)->getSize();
444 unsigned VBRSize = EmitVBRValue(Val: ChildSize, OS);
445 if (!OmitComments) {
446 OS << "/*->" << CurrentIdx + VBRSize + ChildSize << "*/";
447 if (i == 0)
448 OS << " // " << SM->getNumChildren() << " children in Scope";
449 }
450 OS << '\n';
451
452 ChildSize = EmitMatcherList(N: SM->getChild(i), Indent: Indent + 1,
453 StartIdx: CurrentIdx + VBRSize, OS);
454 assert(ChildSize == SM->getChild(i)->getSize() &&
455 "Emitted child size does not match calculated size");
456 CurrentIdx += VBRSize + ChildSize;
457 }
458
459 // Emit a zero as a sentinel indicating end of 'Scope'.
460 if (!OmitComments)
461 OS << "/*" << format_decimal(N: CurrentIdx, Width: IndexWidth) << "*/";
462 OS.indent(NumSpaces: Indent) << "0, ";
463 if (!OmitComments)
464 OS << "/*End of Scope*/";
465 OS << '\n';
466 return CurrentIdx - StartIdx + 1;
467 }
468
469 case Matcher::RecordNode:
470 OS << "OPC_RecordNode,";
471 if (!OmitComments)
472 OS << " // #" << cast<RecordMatcher>(Val: N)->getResultNo() << " = "
473 << cast<RecordMatcher>(Val: N)->getWhatFor();
474 OS << '\n';
475 return 1;
476
477 case Matcher::RecordChild:
478 OS << "OPC_RecordChild" << cast<RecordChildMatcher>(Val: N)->getChildNo() << ',';
479 if (!OmitComments)
480 OS << " // #" << cast<RecordChildMatcher>(Val: N)->getResultNo() << " = "
481 << cast<RecordChildMatcher>(Val: N)->getWhatFor();
482 OS << '\n';
483 return 1;
484
485 case Matcher::RecordMemRef:
486 OS << "OPC_RecordMemRef,\n";
487 return 1;
488
489 case Matcher::CaptureGlueInput:
490 OS << "OPC_CaptureGlueInput,\n";
491 return 1;
492
493 case Matcher::MoveChild: {
494 const auto *MCM = cast<MoveChildMatcher>(Val: N);
495
496 OS << "OPC_MoveChild";
497 // Handle the specialized forms.
498 if (MCM->getChildNo() >= 8)
499 OS << ", ";
500 OS << MCM->getChildNo() << ",\n";
501 return (MCM->getChildNo() >= 8) ? 2 : 1;
502 }
503
504 case Matcher::MoveSibling: {
505 const auto *MSM = cast<MoveSiblingMatcher>(Val: N);
506
507 OS << "OPC_MoveSibling";
508 // Handle the specialized forms.
509 if (MSM->getSiblingNo() >= 8)
510 OS << ", ";
511 OS << MSM->getSiblingNo() << ",\n";
512 return (MSM->getSiblingNo() >= 8) ? 2 : 1;
513 }
514
515 case Matcher::MoveParent:
516 OS << "OPC_MoveParent,\n";
517 return 1;
518
519 case Matcher::CheckSame:
520 OS << "OPC_CheckSame, " << cast<CheckSameMatcher>(Val: N)->getMatchNumber()
521 << ",\n";
522 return 2;
523
524 case Matcher::CheckChildSame:
525 OS << "OPC_CheckChild" << cast<CheckChildSameMatcher>(Val: N)->getChildNo()
526 << "Same, " << cast<CheckChildSameMatcher>(Val: N)->getMatchNumber() << ",\n";
527 return 2;
528
529 case Matcher::CheckPatternPredicate: {
530 StringRef Pred = cast<CheckPatternPredicateMatcher>(Val: N)->getPredicate();
531 unsigned PredNo = getPatternPredicate(PredName: Pred);
532 if (PredNo > 255)
533 OS << "OPC_CheckPatternPredicateTwoByte, TARGET_VAL(" << PredNo << "),";
534 else if (PredNo < 8)
535 OS << "OPC_CheckPatternPredicate" << PredNo << ',';
536 else
537 OS << "OPC_CheckPatternPredicate, " << PredNo << ',';
538 if (!OmitComments)
539 OS << " // " << Pred;
540 OS << '\n';
541 return 2 + (PredNo > 255) - (PredNo < 8);
542 }
543 case Matcher::CheckPredicate: {
544 TreePredicateFn Pred = cast<CheckPredicateMatcher>(Val: N)->getPredicate();
545 unsigned OperandBytes = 0;
546 unsigned PredNo = getNodePredicate(Pred);
547
548 if (Pred.usesOperands()) {
549 unsigned NumOps = cast<CheckPredicateMatcher>(Val: N)->getNumOperands();
550 OS << "OPC_CheckPredicateWithOperands, " << NumOps << "/*#Ops*/, ";
551 for (unsigned i = 0; i < NumOps; ++i)
552 OS << cast<CheckPredicateMatcher>(Val: N)->getOperandNo(i) << ", ";
553 OperandBytes = 1 + NumOps;
554 } else {
555 if (PredNo < 8) {
556 OperandBytes = -1;
557 OS << "OPC_CheckPredicate" << PredNo << ", ";
558 } else
559 OS << "OPC_CheckPredicate, ";
560 }
561
562 if (PredNo >= 8 || Pred.usesOperands())
563 OS << PredNo << ',';
564 if (!OmitComments)
565 OS << " // " << Pred.getFnName();
566 OS << '\n';
567 return 2 + OperandBytes;
568 }
569
570 case Matcher::CheckOpcode:
571 OS << "OPC_CheckOpcode, TARGET_VAL("
572 << cast<CheckOpcodeMatcher>(Val: N)->getOpcode().getEnumName() << "),\n";
573 return 3;
574
575 case Matcher::SwitchOpcode:
576 case Matcher::SwitchType: {
577 unsigned StartIdx = CurrentIdx;
578
579 unsigned NumCases;
580 if (const SwitchOpcodeMatcher *SOM = dyn_cast<SwitchOpcodeMatcher>(Val: N)) {
581 OS << "OPC_SwitchOpcode ";
582 NumCases = SOM->getNumCases();
583 } else {
584 OS << "OPC_SwitchType ";
585 NumCases = cast<SwitchTypeMatcher>(Val: N)->getNumCases();
586 }
587
588 if (!OmitComments)
589 OS << "/*" << NumCases << " cases */";
590 OS << ", ";
591 ++CurrentIdx;
592
593 // For each case we emit the size, then the opcode, then the matcher.
594 for (unsigned i = 0, e = NumCases; i != e; ++i) {
595 const Matcher *Child;
596 unsigned IdxSize;
597 if (const SwitchOpcodeMatcher *SOM = dyn_cast<SwitchOpcodeMatcher>(Val: N)) {
598 Child = SOM->getCaseMatcher(i);
599 IdxSize = 2; // size of opcode in table is 2 bytes.
600 } else {
601 Child = cast<SwitchTypeMatcher>(Val: N)->getCaseMatcher(i);
602 IdxSize = 1; // size of type in table is 1 byte.
603 }
604
605 if (i != 0) {
606 if (!OmitComments)
607 OS << "/*" << format_decimal(N: CurrentIdx, Width: IndexWidth) << "*/";
608 OS.indent(NumSpaces: Indent);
609 if (!OmitComments)
610 OS << (isa<SwitchOpcodeMatcher>(Val: N) ? "/*SwitchOpcode*/ "
611 : "/*SwitchType*/ ");
612 }
613
614 unsigned ChildSize = Child->getSize();
615 CurrentIdx += EmitVBRValue(Val: ChildSize, OS) + IdxSize;
616 if (const SwitchOpcodeMatcher *SOM = dyn_cast<SwitchOpcodeMatcher>(Val: N))
617 OS << "TARGET_VAL(" << SOM->getCaseOpcode(i).getEnumName() << "),";
618 else
619 OS << getEnumName(T: cast<SwitchTypeMatcher>(Val: N)->getCaseType(i)) << ',';
620 if (!OmitComments)
621 OS << "// ->" << CurrentIdx + ChildSize;
622 OS << '\n';
623
624 ChildSize = EmitMatcherList(N: Child, Indent: Indent + 1, StartIdx: CurrentIdx, OS);
625 assert(ChildSize == Child->getSize() &&
626 "Emitted child size does not match calculated size");
627 CurrentIdx += ChildSize;
628 }
629
630 // Emit the final zero to terminate the switch.
631 if (!OmitComments)
632 OS << "/*" << format_decimal(N: CurrentIdx, Width: IndexWidth) << "*/";
633 OS.indent(NumSpaces: Indent) << "0,";
634 if (!OmitComments)
635 OS << (isa<SwitchOpcodeMatcher>(Val: N) ? " // EndSwitchOpcode"
636 : " // EndSwitchType");
637
638 OS << '\n';
639 return CurrentIdx - StartIdx + 1;
640 }
641
642 case Matcher::CheckType:
643 if (cast<CheckTypeMatcher>(Val: N)->getResNo() == 0) {
644 MVT::SimpleValueType VT = cast<CheckTypeMatcher>(Val: N)->getType();
645 switch (VT) {
646 case MVT::i32:
647 case MVT::i64:
648 OS << "OPC_CheckTypeI" << MVT(VT).getSizeInBits() << ",\n";
649 return 1;
650 default:
651 OS << "OPC_CheckType, " << getEnumName(T: VT) << ",\n";
652 return 2;
653 }
654 }
655 OS << "OPC_CheckTypeRes, " << cast<CheckTypeMatcher>(Val: N)->getResNo() << ", "
656 << getEnumName(T: cast<CheckTypeMatcher>(Val: N)->getType()) << ",\n";
657 return 3;
658
659 case Matcher::CheckChildType: {
660 MVT::SimpleValueType VT = cast<CheckChildTypeMatcher>(Val: N)->getType();
661 switch (VT) {
662 case MVT::i32:
663 case MVT::i64:
664 OS << "OPC_CheckChild" << cast<CheckChildTypeMatcher>(Val: N)->getChildNo()
665 << "TypeI" << MVT(VT).getSizeInBits() << ",\n";
666 return 1;
667 default:
668 OS << "OPC_CheckChild" << cast<CheckChildTypeMatcher>(Val: N)->getChildNo()
669 << "Type, " << getEnumName(T: VT) << ",\n";
670 return 2;
671 }
672 }
673
674 case Matcher::CheckInteger: {
675 OS << "OPC_CheckInteger, ";
676 unsigned Bytes =
677 1 + EmitSignedVBRValue(Val: cast<CheckIntegerMatcher>(Val: N)->getValue(), OS);
678 OS << '\n';
679 return Bytes;
680 }
681 case Matcher::CheckChildInteger: {
682 OS << "OPC_CheckChild" << cast<CheckChildIntegerMatcher>(Val: N)->getChildNo()
683 << "Integer, ";
684 unsigned Bytes = 1 + EmitSignedVBRValue(
685 Val: cast<CheckChildIntegerMatcher>(Val: N)->getValue(), OS);
686 OS << '\n';
687 return Bytes;
688 }
689 case Matcher::CheckCondCode:
690 OS << "OPC_CheckCondCode, ISD::"
691 << cast<CheckCondCodeMatcher>(Val: N)->getCondCodeName() << ",\n";
692 return 2;
693
694 case Matcher::CheckChild2CondCode:
695 OS << "OPC_CheckChild2CondCode, ISD::"
696 << cast<CheckChild2CondCodeMatcher>(Val: N)->getCondCodeName() << ",\n";
697 return 2;
698
699 case Matcher::CheckValueType:
700 OS << "OPC_CheckValueType, MVT::"
701 << cast<CheckValueTypeMatcher>(Val: N)->getTypeName() << ",\n";
702 return 2;
703
704 case Matcher::CheckComplexPat: {
705 const CheckComplexPatMatcher *CCPM = cast<CheckComplexPatMatcher>(Val: N);
706 const ComplexPattern &Pattern = CCPM->getPattern();
707 unsigned PatternNo = getComplexPat(P: Pattern);
708 if (PatternNo < 8)
709 OS << "OPC_CheckComplexPat" << PatternNo << ", /*#*/"
710 << CCPM->getMatchNumber() << ',';
711 else
712 OS << "OPC_CheckComplexPat, /*CP*/" << PatternNo << ", /*#*/"
713 << CCPM->getMatchNumber() << ',';
714
715 if (!OmitComments) {
716 OS << " // " << Pattern.getSelectFunc();
717 OS << ":$" << CCPM->getName();
718 for (unsigned i = 0, e = Pattern.getNumOperands(); i != e; ++i)
719 OS << " #" << CCPM->getFirstResult() + i;
720
721 if (Pattern.hasProperty(Prop: SDNPHasChain))
722 OS << " + chain result";
723 }
724 OS << '\n';
725 return PatternNo < 8 ? 2 : 3;
726 }
727
728 case Matcher::CheckAndImm: {
729 OS << "OPC_CheckAndImm, ";
730 unsigned Bytes =
731 1 + EmitVBRValue(Val: cast<CheckAndImmMatcher>(Val: N)->getValue(), OS);
732 OS << '\n';
733 return Bytes;
734 }
735
736 case Matcher::CheckOrImm: {
737 OS << "OPC_CheckOrImm, ";
738 unsigned Bytes =
739 1 + EmitVBRValue(Val: cast<CheckOrImmMatcher>(Val: N)->getValue(), OS);
740 OS << '\n';
741 return Bytes;
742 }
743
744 case Matcher::CheckFoldableChainNode:
745 OS << "OPC_CheckFoldableChainNode,\n";
746 return 1;
747
748 case Matcher::CheckImmAllOnesV:
749 OS << "OPC_CheckImmAllOnesV,\n";
750 return 1;
751
752 case Matcher::CheckImmAllZerosV:
753 OS << "OPC_CheckImmAllZerosV,\n";
754 return 1;
755
756 case Matcher::EmitInteger: {
757 int64_t Val = cast<EmitIntegerMatcher>(Val: N)->getValue();
758 MVT::SimpleValueType VT = cast<EmitIntegerMatcher>(Val: N)->getVT();
759 unsigned OpBytes;
760 switch (VT) {
761 case MVT::i8:
762 case MVT::i16:
763 case MVT::i32:
764 case MVT::i64:
765 OpBytes = 1;
766 OS << "OPC_EmitInteger" << MVT(VT).getSizeInBits() << ", ";
767 break;
768 default:
769 OpBytes = 2;
770 OS << "OPC_EmitInteger, " << getEnumName(T: VT) << ", ";
771 break;
772 }
773 unsigned Bytes = OpBytes + EmitSignedVBRValue(Val, OS);
774 OS << '\n';
775 return Bytes;
776 }
777 case Matcher::EmitStringInteger: {
778 const std::string &Val = cast<EmitStringIntegerMatcher>(Val: N)->getValue();
779 MVT::SimpleValueType VT = cast<EmitStringIntegerMatcher>(Val: N)->getVT();
780 // These should always fit into 7 bits.
781 unsigned OpBytes;
782 switch (VT) {
783 case MVT::i32:
784 OpBytes = 1;
785 OS << "OPC_EmitStringInteger" << MVT(VT).getSizeInBits() << ", ";
786 break;
787 default:
788 OpBytes = 2;
789 OS << "OPC_EmitStringInteger, " << getEnumName(T: VT) << ", ";
790 break;
791 }
792 OS << Val << ",\n";
793 return OpBytes + 1;
794 }
795
796 case Matcher::EmitRegister: {
797 const EmitRegisterMatcher *Matcher = cast<EmitRegisterMatcher>(Val: N);
798 const CodeGenRegister *Reg = Matcher->getReg();
799 MVT::SimpleValueType VT = Matcher->getVT();
800 // If the enum value of the register is larger than one byte can handle,
801 // use EmitRegister2.
802 if (Reg && Reg->EnumValue > 255) {
803 OS << "OPC_EmitRegister2, " << getEnumName(T: VT) << ", ";
804 OS << "TARGET_VAL(" << getQualifiedName(R: Reg->TheDef) << "),\n";
805 return 4;
806 }
807 unsigned OpBytes;
808 switch (VT) {
809 case MVT::i32:
810 case MVT::i64:
811 OpBytes = 1;
812 OS << "OPC_EmitRegisterI" << MVT(VT).getSizeInBits() << ", ";
813 break;
814 default:
815 OpBytes = 2;
816 OS << "OPC_EmitRegister, " << getEnumName(T: VT) << ", ";
817 break;
818 }
819 if (Reg) {
820 OS << getQualifiedName(R: Reg->TheDef) << ",\n";
821 } else {
822 OS << "0 ";
823 if (!OmitComments)
824 OS << "/*zero_reg*/";
825 OS << ",\n";
826 }
827 return OpBytes + 1;
828 }
829
830 case Matcher::EmitConvertToTarget: {
831 unsigned Slot = cast<EmitConvertToTargetMatcher>(Val: N)->getSlot();
832 if (Slot < 8) {
833 OS << "OPC_EmitConvertToTarget" << Slot << ",\n";
834 return 1;
835 }
836 OS << "OPC_EmitConvertToTarget, " << Slot << ",\n";
837 return 2;
838 }
839
840 case Matcher::EmitMergeInputChains: {
841 const EmitMergeInputChainsMatcher *MN =
842 cast<EmitMergeInputChainsMatcher>(Val: N);
843
844 // Handle the specialized forms OPC_EmitMergeInputChains1_0, 1_1, and 1_2.
845 if (MN->getNumNodes() == 1 && MN->getNode(i: 0) < 3) {
846 OS << "OPC_EmitMergeInputChains1_" << MN->getNode(i: 0) << ",\n";
847 return 1;
848 }
849
850 OS << "OPC_EmitMergeInputChains, " << MN->getNumNodes() << ", ";
851 for (unsigned i = 0, e = MN->getNumNodes(); i != e; ++i)
852 OS << MN->getNode(i) << ", ";
853 OS << '\n';
854 return 2 + MN->getNumNodes();
855 }
856 case Matcher::EmitCopyToReg: {
857 const auto *C2RMatcher = cast<EmitCopyToRegMatcher>(Val: N);
858 int Bytes = 3;
859 const CodeGenRegister *Reg = C2RMatcher->getDestPhysReg();
860 unsigned Slot = C2RMatcher->getSrcSlot();
861 if (Reg->EnumValue > 255) {
862 assert(isUInt<16>(Reg->EnumValue) && "not handled");
863 OS << "OPC_EmitCopyToRegTwoByte, " << Slot << ", "
864 << "TARGET_VAL(" << getQualifiedName(R: Reg->TheDef) << "),\n";
865 ++Bytes;
866 } else {
867 if (Slot < 8) {
868 OS << "OPC_EmitCopyToReg" << Slot << ", "
869 << getQualifiedName(R: Reg->TheDef) << ",\n";
870 --Bytes;
871 } else
872 OS << "OPC_EmitCopyToReg, " << Slot << ", "
873 << getQualifiedName(R: Reg->TheDef) << ",\n";
874 }
875
876 return Bytes;
877 }
878 case Matcher::EmitNodeXForm: {
879 const EmitNodeXFormMatcher *XF = cast<EmitNodeXFormMatcher>(Val: N);
880 OS << "OPC_EmitNodeXForm, " << getNodeXFormID(Rec: XF->getNodeXForm()) << ", "
881 << XF->getSlot() << ',';
882 if (!OmitComments)
883 OS << " // " << XF->getNodeXForm()->getName();
884 OS << '\n';
885 return 3;
886 }
887
888 case Matcher::EmitNode:
889 case Matcher::MorphNodeTo: {
890 auto NumCoveredBytes = 0;
891 if (InstrumentCoverage) {
892 if (const MorphNodeToMatcher *SNT = dyn_cast<MorphNodeToMatcher>(Val: N)) {
893 NumCoveredBytes = 3;
894 OS << "OPC_Coverage, ";
895 std::string src =
896 GetPatFromTreePatternNode(N: SNT->getPattern().getSrcPattern());
897 std::string dst =
898 GetPatFromTreePatternNode(N: SNT->getPattern().getDstPattern());
899 Record *PatRecord = SNT->getPattern().getSrcRecord();
900 std::string include_src = getIncludePath(R: PatRecord);
901 unsigned Offset =
902 getPatternIdxFromTable(P: src + " -> " + dst, include_loc: std::move(include_src));
903 OS << "TARGET_VAL(" << Offset << "),\n";
904 OS.indent(NumSpaces: FullIndexWidth + Indent);
905 }
906 }
907 const EmitNodeMatcherCommon *EN = cast<EmitNodeMatcherCommon>(Val: N);
908 bool IsEmitNode = isa<EmitNodeMatcher>(Val: EN);
909 OS << (IsEmitNode ? "OPC_EmitNode" : "OPC_MorphNodeTo");
910 bool CompressVTs = EN->getNumVTs() < 3;
911 bool CompressNodeInfo = false;
912 if (CompressVTs) {
913 OS << EN->getNumVTs();
914 if (!EN->hasChain() && !EN->hasInGlue() && !EN->hasOutGlue() &&
915 !EN->hasMemRefs() && EN->getNumFixedArityOperands() == -1) {
916 CompressNodeInfo = true;
917 OS << "None";
918 } else if (EN->hasChain() && !EN->hasInGlue() && !EN->hasOutGlue() &&
919 !EN->hasMemRefs() && EN->getNumFixedArityOperands() == -1) {
920 CompressNodeInfo = true;
921 OS << "Chain";
922 } else if (!IsEmitNode && !EN->hasChain() && EN->hasInGlue() &&
923 !EN->hasOutGlue() && !EN->hasMemRefs() &&
924 EN->getNumFixedArityOperands() == -1) {
925 CompressNodeInfo = true;
926 OS << "GlueInput";
927 } else if (!IsEmitNode && !EN->hasChain() && !EN->hasInGlue() &&
928 EN->hasOutGlue() && !EN->hasMemRefs() &&
929 EN->getNumFixedArityOperands() == -1) {
930 CompressNodeInfo = true;
931 OS << "GlueOutput";
932 }
933 }
934
935 const CodeGenInstruction &CGI = EN->getInstruction();
936 OS << ", TARGET_VAL(" << CGI.Namespace << "::" << CGI.TheDef->getName()
937 << ")";
938
939 if (!CompressNodeInfo) {
940 OS << ", 0";
941 if (EN->hasChain())
942 OS << "|OPFL_Chain";
943 if (EN->hasInGlue())
944 OS << "|OPFL_GlueInput";
945 if (EN->hasOutGlue())
946 OS << "|OPFL_GlueOutput";
947 if (EN->hasMemRefs())
948 OS << "|OPFL_MemRefs";
949 if (EN->getNumFixedArityOperands() != -1)
950 OS << "|OPFL_Variadic" << EN->getNumFixedArityOperands();
951 }
952 OS << ",\n";
953
954 OS.indent(NumSpaces: FullIndexWidth + Indent + 4);
955 if (!CompressVTs) {
956 OS << EN->getNumVTs();
957 if (!OmitComments)
958 OS << "/*#VTs*/";
959 OS << ", ";
960 }
961 for (unsigned i = 0, e = EN->getNumVTs(); i != e; ++i)
962 OS << getEnumName(T: EN->getVT(i)) << ", ";
963
964 OS << EN->getNumOperands();
965 if (!OmitComments)
966 OS << "/*#Ops*/";
967 OS << ", ";
968 unsigned NumOperandBytes = 0;
969 for (unsigned i = 0, e = EN->getNumOperands(); i != e; ++i)
970 NumOperandBytes += EmitVBRValue(Val: EN->getOperand(i), OS);
971
972 if (!OmitComments) {
973 // Print the result #'s for EmitNode.
974 if (const EmitNodeMatcher *E = dyn_cast<EmitNodeMatcher>(Val: EN)) {
975 if (unsigned NumResults = EN->getNumVTs()) {
976 OS << " // Results =";
977 unsigned First = E->getFirstResultSlot();
978 for (unsigned i = 0; i != NumResults; ++i)
979 OS << " #" << First + i;
980 }
981 }
982 OS << '\n';
983
984 if (const MorphNodeToMatcher *SNT = dyn_cast<MorphNodeToMatcher>(Val: N)) {
985 OS.indent(NumSpaces: FullIndexWidth + Indent)
986 << "// Src: " << SNT->getPattern().getSrcPattern()
987 << " - Complexity = " << SNT->getPattern().getPatternComplexity(CGP)
988 << '\n';
989 OS.indent(NumSpaces: FullIndexWidth + Indent)
990 << "// Dst: " << SNT->getPattern().getDstPattern() << '\n';
991 }
992 } else
993 OS << '\n';
994
995 return 4 + !CompressVTs + !CompressNodeInfo + EN->getNumVTs() +
996 NumOperandBytes + NumCoveredBytes;
997 }
998 case Matcher::CompleteMatch: {
999 const CompleteMatchMatcher *CM = cast<CompleteMatchMatcher>(Val: N);
1000 auto NumCoveredBytes = 0;
1001 if (InstrumentCoverage) {
1002 NumCoveredBytes = 3;
1003 OS << "OPC_Coverage, ";
1004 std::string src =
1005 GetPatFromTreePatternNode(N: CM->getPattern().getSrcPattern());
1006 std::string dst =
1007 GetPatFromTreePatternNode(N: CM->getPattern().getDstPattern());
1008 Record *PatRecord = CM->getPattern().getSrcRecord();
1009 std::string include_src = getIncludePath(R: PatRecord);
1010 unsigned Offset =
1011 getPatternIdxFromTable(P: src + " -> " + dst, include_loc: std::move(include_src));
1012 OS << "TARGET_VAL(" << Offset << "),\n";
1013 OS.indent(NumSpaces: FullIndexWidth + Indent);
1014 }
1015 OS << "OPC_CompleteMatch, " << CM->getNumResults() << ", ";
1016 unsigned NumResultBytes = 0;
1017 for (unsigned i = 0, e = CM->getNumResults(); i != e; ++i)
1018 NumResultBytes += EmitVBRValue(Val: CM->getResult(R: i), OS);
1019 OS << '\n';
1020 if (!OmitComments) {
1021 OS.indent(NumSpaces: FullIndexWidth + Indent)
1022 << " // Src: " << CM->getPattern().getSrcPattern()
1023 << " - Complexity = " << CM->getPattern().getPatternComplexity(CGP)
1024 << '\n';
1025 OS.indent(NumSpaces: FullIndexWidth + Indent)
1026 << " // Dst: " << CM->getPattern().getDstPattern();
1027 }
1028 OS << '\n';
1029 return 2 + NumResultBytes + NumCoveredBytes;
1030 }
1031 }
1032 llvm_unreachable("Unreachable");
1033}
1034
1035/// This function traverses the matcher tree and emits all the nodes.
1036/// The nodes have already been sized.
1037unsigned MatcherTableEmitter::EmitMatcherList(const Matcher *N,
1038 const unsigned Indent,
1039 unsigned CurrentIdx,
1040 raw_ostream &OS) {
1041 unsigned Size = 0;
1042 while (N) {
1043 if (!OmitComments)
1044 OS << "/*" << format_decimal(N: CurrentIdx, Width: IndexWidth) << "*/";
1045 unsigned MatcherSize = EmitMatcher(N, Indent, CurrentIdx, OS);
1046 Size += MatcherSize;
1047 CurrentIdx += MatcherSize;
1048
1049 // If there are other nodes in this list, iterate to them, otherwise we're
1050 // done.
1051 N = N->getNext();
1052 }
1053 return Size;
1054}
1055
1056void MatcherTableEmitter::EmitNodePredicatesFunction(
1057 const std::vector<TreePattern *> &Preds, StringRef Decl, raw_ostream &OS) {
1058 if (Preds.empty())
1059 return;
1060
1061 BeginEmitFunction(OS, RetType: "bool", Decl, AddOverride: true /*AddOverride*/);
1062 OS << "{\n";
1063 OS << " switch (PredNo) {\n";
1064 OS << " default: llvm_unreachable(\"Invalid predicate in table?\");\n";
1065 for (unsigned i = 0, e = Preds.size(); i != e; ++i) {
1066 // Emit the predicate code corresponding to this pattern.
1067 TreePredicateFn PredFn(Preds[i]);
1068 assert(!PredFn.isAlwaysTrue() && "No code in this predicate");
1069 std::string PredFnCodeStr = PredFn.getCodeToRunOnSDNode();
1070
1071 OS << " case " << i << ": {\n";
1072 for (auto *SimilarPred : NodePredicatesByCodeToRun[PredFnCodeStr])
1073 OS << " // " << TreePredicateFn(SimilarPred).getFnName() << '\n';
1074 OS << PredFnCodeStr << "\n }\n";
1075 }
1076 OS << " }\n";
1077 OS << "}\n";
1078 EndEmitFunction(OS);
1079}
1080
1081void MatcherTableEmitter::EmitPredicateFunctions(raw_ostream &OS) {
1082 // Emit pattern predicates.
1083 if (!PatternPredicates.empty()) {
1084 BeginEmitFunction(OS, RetType: "bool",
1085 Decl: "CheckPatternPredicate(unsigned PredNo) const",
1086 AddOverride: true /*AddOverride*/);
1087 OS << "{\n";
1088 OS << " switch (PredNo) {\n";
1089 OS << " default: llvm_unreachable(\"Invalid predicate in table?\");\n";
1090 for (unsigned i = 0, e = PatternPredicates.size(); i != e; ++i)
1091 OS << " case " << i << ": return " << PatternPredicates[i] << ";\n";
1092 OS << " }\n";
1093 OS << "}\n";
1094 EndEmitFunction(OS);
1095 }
1096
1097 // Emit Node predicates.
1098 EmitNodePredicatesFunction(
1099 Preds: NodePredicates, Decl: "CheckNodePredicate(SDNode *Node, unsigned PredNo) const",
1100 OS);
1101 EmitNodePredicatesFunction(
1102 Preds: NodePredicatesWithOperands,
1103 Decl: "CheckNodePredicateWithOperands(SDNode *Node, unsigned PredNo, "
1104 "const SmallVectorImpl<SDValue> &Operands) const",
1105 OS);
1106
1107 // Emit CompletePattern matchers.
1108 // FIXME: This should be const.
1109 if (!ComplexPatterns.empty()) {
1110 BeginEmitFunction(
1111 OS, RetType: "bool",
1112 Decl: "CheckComplexPattern(SDNode *Root, SDNode *Parent,\n"
1113 " SDValue N, unsigned PatternNo,\n"
1114 " SmallVectorImpl<std::pair<SDValue, SDNode *>> &Result)",
1115 AddOverride: true /*AddOverride*/);
1116 OS << "{\n";
1117 OS << " unsigned NextRes = Result.size();\n";
1118 OS << " switch (PatternNo) {\n";
1119 OS << " default: llvm_unreachable(\"Invalid pattern # in table?\");\n";
1120 for (unsigned i = 0, e = ComplexPatterns.size(); i != e; ++i) {
1121 const ComplexPattern &P = *ComplexPatterns[i];
1122 unsigned NumOps = P.getNumOperands();
1123
1124 if (P.hasProperty(Prop: SDNPHasChain))
1125 ++NumOps; // Get the chained node too.
1126
1127 OS << " case " << i << ":\n";
1128 if (InstrumentCoverage)
1129 OS << " {\n";
1130 OS << " Result.resize(NextRes+" << NumOps << ");\n";
1131 if (InstrumentCoverage)
1132 OS << " bool Succeeded = " << P.getSelectFunc();
1133 else
1134 OS << " return " << P.getSelectFunc();
1135
1136 OS << "(";
1137 // If the complex pattern wants the root of the match, pass it in as the
1138 // first argument.
1139 if (P.hasProperty(Prop: SDNPWantRoot))
1140 OS << "Root, ";
1141
1142 // If the complex pattern wants the parent of the operand being matched,
1143 // pass it in as the next argument.
1144 if (P.hasProperty(Prop: SDNPWantParent))
1145 OS << "Parent, ";
1146
1147 OS << "N";
1148 for (unsigned i = 0; i != NumOps; ++i)
1149 OS << ", Result[NextRes+" << i << "].first";
1150 OS << ");\n";
1151 if (InstrumentCoverage) {
1152 OS << " if (Succeeded)\n";
1153 OS << " dbgs() << \"\\nCOMPLEX_PATTERN: " << P.getSelectFunc()
1154 << "\\n\" ;\n";
1155 OS << " return Succeeded;\n";
1156 OS << " }\n";
1157 }
1158 }
1159 OS << " }\n";
1160 OS << "}\n";
1161 EndEmitFunction(OS);
1162 }
1163
1164 // Emit SDNodeXForm handlers.
1165 // FIXME: This should be const.
1166 if (!NodeXForms.empty()) {
1167 BeginEmitFunction(OS, RetType: "SDValue",
1168 Decl: "RunSDNodeXForm(SDValue V, unsigned XFormNo)",
1169 AddOverride: true /*AddOverride*/);
1170 OS << "{\n";
1171 OS << " switch (XFormNo) {\n";
1172 OS << " default: llvm_unreachable(\"Invalid xform # in table?\");\n";
1173
1174 // FIXME: The node xform could take SDValue's instead of SDNode*'s.
1175 for (unsigned i = 0, e = NodeXForms.size(); i != e; ++i) {
1176 const CodeGenDAGPatterns::NodeXForm &Entry =
1177 CGP.getSDNodeTransform(R: NodeXForms[i]);
1178
1179 Record *SDNode = Entry.first;
1180 const std::string &Code = Entry.second;
1181
1182 OS << " case " << i << ": { ";
1183 if (!OmitComments)
1184 OS << "// " << NodeXForms[i]->getName();
1185 OS << '\n';
1186
1187 std::string ClassName =
1188 std::string(CGP.getSDNodeInfo(R: SDNode).getSDClassName());
1189 if (ClassName == "SDNode")
1190 OS << " SDNode *N = V.getNode();\n";
1191 else
1192 OS << " " << ClassName << " *N = cast<" << ClassName
1193 << ">(V.getNode());\n";
1194 OS << Code << "\n }\n";
1195 }
1196 OS << " }\n";
1197 OS << "}\n";
1198 EndEmitFunction(OS);
1199 }
1200}
1201
1202static StringRef getOpcodeString(Matcher::KindTy Kind) {
1203 switch (Kind) {
1204 case Matcher::Scope:
1205 return "OPC_Scope";
1206 case Matcher::RecordNode:
1207 return "OPC_RecordNode";
1208 case Matcher::RecordChild:
1209 return "OPC_RecordChild";
1210 case Matcher::RecordMemRef:
1211 return "OPC_RecordMemRef";
1212 case Matcher::CaptureGlueInput:
1213 return "OPC_CaptureGlueInput";
1214 case Matcher::MoveChild:
1215 return "OPC_MoveChild";
1216 case Matcher::MoveSibling:
1217 return "OPC_MoveSibling";
1218 case Matcher::MoveParent:
1219 return "OPC_MoveParent";
1220 case Matcher::CheckSame:
1221 return "OPC_CheckSame";
1222 case Matcher::CheckChildSame:
1223 return "OPC_CheckChildSame";
1224 case Matcher::CheckPatternPredicate:
1225 return "OPC_CheckPatternPredicate";
1226 case Matcher::CheckPredicate:
1227 return "OPC_CheckPredicate";
1228 case Matcher::CheckOpcode:
1229 return "OPC_CheckOpcode";
1230 case Matcher::SwitchOpcode:
1231 return "OPC_SwitchOpcode";
1232 case Matcher::CheckType:
1233 return "OPC_CheckType";
1234 case Matcher::SwitchType:
1235 return "OPC_SwitchType";
1236 case Matcher::CheckChildType:
1237 return "OPC_CheckChildType";
1238 case Matcher::CheckInteger:
1239 return "OPC_CheckInteger";
1240 case Matcher::CheckChildInteger:
1241 return "OPC_CheckChildInteger";
1242 case Matcher::CheckCondCode:
1243 return "OPC_CheckCondCode";
1244 case Matcher::CheckChild2CondCode:
1245 return "OPC_CheckChild2CondCode";
1246 case Matcher::CheckValueType:
1247 return "OPC_CheckValueType";
1248 case Matcher::CheckComplexPat:
1249 return "OPC_CheckComplexPat";
1250 case Matcher::CheckAndImm:
1251 return "OPC_CheckAndImm";
1252 case Matcher::CheckOrImm:
1253 return "OPC_CheckOrImm";
1254 case Matcher::CheckFoldableChainNode:
1255 return "OPC_CheckFoldableChainNode";
1256 case Matcher::CheckImmAllOnesV:
1257 return "OPC_CheckImmAllOnesV";
1258 case Matcher::CheckImmAllZerosV:
1259 return "OPC_CheckImmAllZerosV";
1260 case Matcher::EmitInteger:
1261 return "OPC_EmitInteger";
1262 case Matcher::EmitStringInteger:
1263 return "OPC_EmitStringInteger";
1264 case Matcher::EmitRegister:
1265 return "OPC_EmitRegister";
1266 case Matcher::EmitConvertToTarget:
1267 return "OPC_EmitConvertToTarget";
1268 case Matcher::EmitMergeInputChains:
1269 return "OPC_EmitMergeInputChains";
1270 case Matcher::EmitCopyToReg:
1271 return "OPC_EmitCopyToReg";
1272 case Matcher::EmitNode:
1273 return "OPC_EmitNode";
1274 case Matcher::MorphNodeTo:
1275 return "OPC_MorphNodeTo";
1276 case Matcher::EmitNodeXForm:
1277 return "OPC_EmitNodeXForm";
1278 case Matcher::CompleteMatch:
1279 return "OPC_CompleteMatch";
1280 }
1281
1282 llvm_unreachable("Unhandled opcode?");
1283}
1284
1285void MatcherTableEmitter::EmitHistogram(const Matcher *M, raw_ostream &OS) {
1286 if (OmitComments)
1287 return;
1288
1289 OS << " // Opcode Histogram:\n";
1290 for (unsigned i = 0, e = OpcodeCounts.size(); i != e; ++i) {
1291 OS << " // #"
1292 << left_justify(Str: getOpcodeString(Kind: (Matcher::KindTy)i), Width: HistOpcWidth)
1293 << " = " << OpcodeCounts[i] << '\n';
1294 }
1295 OS << '\n';
1296}
1297
1298void llvm::EmitMatcherTable(Matcher *TheMatcher, const CodeGenDAGPatterns &CGP,
1299 raw_ostream &OS) {
1300 OS << "#if defined(GET_DAGISEL_DECL) && defined(GET_DAGISEL_BODY)\n";
1301 OS << "#error GET_DAGISEL_DECL and GET_DAGISEL_BODY cannot be both defined, ";
1302 OS << "undef both for inline definitions\n";
1303 OS << "#endif\n\n";
1304
1305 // Emit a check for omitted class name.
1306 OS << "#ifdef GET_DAGISEL_BODY\n";
1307 OS << "#define LOCAL_DAGISEL_STRINGIZE(X) LOCAL_DAGISEL_STRINGIZE_(X)\n";
1308 OS << "#define LOCAL_DAGISEL_STRINGIZE_(X) #X\n";
1309 OS << "static_assert(sizeof(LOCAL_DAGISEL_STRINGIZE(GET_DAGISEL_BODY)) > 1,"
1310 "\n";
1311 OS << " \"GET_DAGISEL_BODY is empty: it should be defined with the class "
1312 "name\");\n";
1313 OS << "#undef LOCAL_DAGISEL_STRINGIZE_\n";
1314 OS << "#undef LOCAL_DAGISEL_STRINGIZE\n";
1315 OS << "#endif\n\n";
1316
1317 OS << "#if !defined(GET_DAGISEL_DECL) && !defined(GET_DAGISEL_BODY)\n";
1318 OS << "#define DAGISEL_INLINE 1\n";
1319 OS << "#else\n";
1320 OS << "#define DAGISEL_INLINE 0\n";
1321 OS << "#endif\n\n";
1322
1323 OS << "#if !DAGISEL_INLINE\n";
1324 OS << "#define DAGISEL_CLASS_COLONCOLON GET_DAGISEL_BODY ::\n";
1325 OS << "#else\n";
1326 OS << "#define DAGISEL_CLASS_COLONCOLON\n";
1327 OS << "#endif\n\n";
1328
1329 BeginEmitFunction(OS, RetType: "void", Decl: "SelectCode(SDNode *N)", AddOverride: false /*AddOverride*/);
1330 MatcherTableEmitter MatcherEmitter(TheMatcher, CGP);
1331
1332 // First we size all the children of the three kinds of matchers that have
1333 // them. This is done by sharing the code in EmitMatcher(). but we don't
1334 // want to emit anything, so we turn off comments and use a null stream.
1335 bool SaveOmitComments = OmitComments;
1336 OmitComments = true;
1337 raw_null_ostream NullOS;
1338 unsigned TotalSize = MatcherEmitter.SizeMatcherList(N: TheMatcher, OS&: NullOS);
1339 OmitComments = SaveOmitComments;
1340
1341 // Now that the matchers are sized, we can emit the code for them to the
1342 // final stream.
1343 OS << "{\n";
1344 OS << " // Some target values are emitted as 2 bytes, TARGET_VAL handles\n";
1345 OS << " // this.\n";
1346 OS << " #define TARGET_VAL(X) X & 255, unsigned(X) >> 8\n";
1347 OS << " static const unsigned char MatcherTable[] = {\n";
1348 TotalSize = MatcherEmitter.EmitMatcherList(N: TheMatcher, Indent: 1, CurrentIdx: 0, OS);
1349 OS << " 0\n }; // Total Array size is " << (TotalSize + 1)
1350 << " bytes\n\n";
1351
1352 MatcherEmitter.EmitHistogram(M: TheMatcher, OS);
1353
1354 OS << " #undef TARGET_VAL\n";
1355 OS << " SelectCodeCommon(N, MatcherTable,sizeof(MatcherTable));\n";
1356 OS << "}\n";
1357 EndEmitFunction(OS);
1358
1359 // Next up, emit the function for node and pattern predicates:
1360 MatcherEmitter.EmitPredicateFunctions(OS);
1361
1362 if (InstrumentCoverage)
1363 MatcherEmitter.EmitPatternMatchTable(OS);
1364
1365 // Clean up the preprocessor macros.
1366 OS << "\n";
1367 OS << "#ifdef DAGISEL_INLINE\n";
1368 OS << "#undef DAGISEL_INLINE\n";
1369 OS << "#endif\n";
1370 OS << "#ifdef DAGISEL_CLASS_COLONCOLON\n";
1371 OS << "#undef DAGISEL_CLASS_COLONCOLON\n";
1372 OS << "#endif\n";
1373 OS << "#ifdef GET_DAGISEL_DECL\n";
1374 OS << "#undef GET_DAGISEL_DECL\n";
1375 OS << "#endif\n";
1376 OS << "#ifdef GET_DAGISEL_BODY\n";
1377 OS << "#undef GET_DAGISEL_BODY\n";
1378 OS << "#endif\n";
1379}
1380

source code of llvm/utils/TableGen/DAGISelMatcherEmitter.cpp