1 //===- GlobalISelEmitter.cpp - Generate an instruction selector -----------===// 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 /// \file 10 /// This tablegen backend emits code for use by the GlobalISel instruction 11 /// selector. See include/llvm/CodeGen/TargetGlobalISel.td. 12 /// 13 /// This file analyzes the patterns recognized by the SelectionDAGISel tablegen 14 /// backend, filters out the ones that are unsupported, maps 15 /// SelectionDAG-specific constructs to their GlobalISel counterpart 16 /// (when applicable: MVT to LLT; SDNode to generic Instruction). 17 /// 18 /// Not all patterns are supported: pass the tablegen invocation 19 /// "-warn-on-skipped-patterns" to emit a warning when a pattern is skipped, 20 /// as well as why. 21 /// 22 /// The generated file defines a single method: 23 /// bool <Target>InstructionSelector::selectImpl(MachineInstr &I) const; 24 /// intended to be used in InstructionSelector::select as the first-step 25 /// selector for the patterns that don't require complex C++. 26 /// 27 /// FIXME: We'll probably want to eventually define a base 28 /// "TargetGenInstructionSelector" class. 29 /// 30 //===----------------------------------------------------------------------===// 31 32 #include "CodeGenDAGPatterns.h" 33 #include "SubtargetFeatureInfo.h" 34 #include "llvm/ADT/Optional.h" 35 #include "llvm/ADT/SmallSet.h" 36 #include "llvm/ADT/Statistic.h" 37 #include "llvm/Support/CodeGenCoverage.h" 38 #include "llvm/Support/CommandLine.h" 39 #include "llvm/Support/Error.h" 40 #include "llvm/Support/LowLevelTypeImpl.h" 41 #include "llvm/Support/MachineValueType.h" 42 #include "llvm/Support/ScopedPrinter.h" 43 #include "llvm/TableGen/Error.h" 44 #include "llvm/TableGen/Record.h" 45 #include "llvm/TableGen/TableGenBackend.h" 46 #include <numeric> 47 #include <string> 48 using namespace llvm; 49 50 #define DEBUG_TYPE "gisel-emitter" 51 52 STATISTIC(NumPatternTotal, "Total number of patterns"); 53 STATISTIC(NumPatternImported, "Number of patterns imported from SelectionDAG"); 54 STATISTIC(NumPatternImportsSkipped, "Number of SelectionDAG imports skipped"); 55 STATISTIC(NumPatternsTested, "Number of patterns executed according to coverage information"); 56 STATISTIC(NumPatternEmitted, "Number of patterns emitted"); 57 58 cl::OptionCategory GlobalISelEmitterCat("Options for -gen-global-isel"); 59 60 static cl::opt<bool> WarnOnSkippedPatterns( 61 "warn-on-skipped-patterns", 62 cl::desc("Explain why a pattern was skipped for inclusion " 63 "in the GlobalISel selector"), 64 cl::init(false), cl::cat(GlobalISelEmitterCat)); 65 66 static cl::opt<bool> GenerateCoverage( 67 "instrument-gisel-coverage", 68 cl::desc("Generate coverage instrumentation for GlobalISel"), 69 cl::init(false), cl::cat(GlobalISelEmitterCat)); 70 71 static cl::opt<std::string> UseCoverageFile( 72 "gisel-coverage-file", cl::init(""), 73 cl::desc("Specify file to retrieve coverage information from"), 74 cl::cat(GlobalISelEmitterCat)); 75 76 static cl::opt<bool> OptimizeMatchTable( 77 "optimize-match-table", 78 cl::desc("Generate an optimized version of the match table"), 79 cl::init(true), cl::cat(GlobalISelEmitterCat)); 80 81 namespace { 82 //===- Helper functions ---------------------------------------------------===// 83 84 /// Get the name of the enum value used to number the predicate function. 85 std::string getEnumNameForPredicate(const TreePredicateFn &Predicate) { 86 if (Predicate.hasGISelPredicateCode()) 87 return "GIPFP_MI_" + Predicate.getFnName(); 88 return "GIPFP_" + Predicate.getImmTypeIdentifier().str() + "_" + 89 Predicate.getFnName(); 90 } 91 92 /// Get the opcode used to check this predicate. 93 std::string getMatchOpcodeForPredicate(const TreePredicateFn &Predicate) { 94 return "GIM_Check" + Predicate.getImmTypeIdentifier().str() + "ImmPredicate"; 95 } 96 97 /// This class stands in for LLT wherever we want to tablegen-erate an 98 /// equivalent at compiler run-time. 99 class LLTCodeGen { 100 private: 101 LLT Ty; 102 103 public: 104 LLTCodeGen() = default; 105 LLTCodeGen(const LLT &Ty) : Ty(Ty) {} 106 107 std::string getCxxEnumValue() const { 108 std::string Str; 109 raw_string_ostream OS(Str); 110 111 emitCxxEnumValue(OS); 112 return OS.str(); 113 } 114 115 void emitCxxEnumValue(raw_ostream &OS) const { 116 if (Ty.isScalar()) { 117 OS << "GILLT_s" << Ty.getSizeInBits(); 118 return; 119 } 120 if (Ty.isVector()) { 121 OS << "GILLT_v" << Ty.getNumElements() << "s" << Ty.getScalarSizeInBits(); 122 return; 123 } 124 if (Ty.isPointer()) { 125 OS << "GILLT_p" << Ty.getAddressSpace(); 126 if (Ty.getSizeInBits() > 0) 127 OS << "s" << Ty.getSizeInBits(); 128 return; 129 } 130 llvm_unreachable("Unhandled LLT"); 131 } 132 133 void emitCxxConstructorCall(raw_ostream &OS) const { 134 if (Ty.isScalar()) { 135 OS << "LLT::scalar(" << Ty.getSizeInBits() << ")"; 136 return; 137 } 138 if (Ty.isVector()) { 139 OS << "LLT::vector(" << Ty.getNumElements() << ", " 140 << Ty.getScalarSizeInBits() << ")"; 141 return; 142 } 143 if (Ty.isPointer() && Ty.getSizeInBits() > 0) { 144 OS << "LLT::pointer(" << Ty.getAddressSpace() << ", " 145 << Ty.getSizeInBits() << ")"; 146 return; 147 } 148 llvm_unreachable("Unhandled LLT"); 149 } 150 151 const LLT &get() const { return Ty; } 152 153 /// This ordering is used for std::unique() and llvm::sort(). There's no 154 /// particular logic behind the order but either A < B or B < A must be 155 /// true if A != B. 156 bool operator<(const LLTCodeGen &Other) const { 157 if (Ty.isValid() != Other.Ty.isValid()) 158 return Ty.isValid() < Other.Ty.isValid(); 159 if (!Ty.isValid()) 160 return false; 161 162 if (Ty.isVector() != Other.Ty.isVector()) 163 return Ty.isVector() < Other.Ty.isVector(); 164 if (Ty.isScalar() != Other.Ty.isScalar()) 165 return Ty.isScalar() < Other.Ty.isScalar(); 166 if (Ty.isPointer() != Other.Ty.isPointer()) 167 return Ty.isPointer() < Other.Ty.isPointer(); 168 169 if (Ty.isPointer() && Ty.getAddressSpace() != Other.Ty.getAddressSpace()) 170 return Ty.getAddressSpace() < Other.Ty.getAddressSpace(); 171 172 if (Ty.isVector() && Ty.getNumElements() != Other.Ty.getNumElements()) 173 return Ty.getNumElements() < Other.Ty.getNumElements(); 174 175 return Ty.getSizeInBits() < Other.Ty.getSizeInBits(); 176 } 177 178 bool operator==(const LLTCodeGen &B) const { return Ty == B.Ty; } 179 }; 180 181 // Track all types that are used so we can emit the corresponding enum. 182 std::set<LLTCodeGen> KnownTypes; 183 184 class InstructionMatcher; 185 /// Convert an MVT to an equivalent LLT if possible, or the invalid LLT() for 186 /// MVTs that don't map cleanly to an LLT (e.g., iPTR, *any, ...). 187 static Optional<LLTCodeGen> MVTToLLT(MVT::SimpleValueType SVT) { 188 MVT VT(SVT); 189 190 if (VT.isVector() && VT.getVectorNumElements() != 1) 191 return LLTCodeGen( 192 LLT::vector(VT.getVectorNumElements(), VT.getScalarSizeInBits())); 193 194 if (VT.isInteger() || VT.isFloatingPoint()) 195 return LLTCodeGen(LLT::scalar(VT.getSizeInBits())); 196 return None; 197 } 198 199 static std::string explainPredicates(const TreePatternNode *N) { 200 std::string Explanation = ""; 201 StringRef Separator = ""; 202 for (const TreePredicateCall &Call : N->getPredicateCalls()) { 203 const TreePredicateFn &P = Call.Fn; 204 Explanation += 205 (Separator + P.getOrigPatFragRecord()->getRecord()->getName()).str(); 206 Separator = ", "; 207 208 if (P.isAlwaysTrue()) 209 Explanation += " always-true"; 210 if (P.isImmediatePattern()) 211 Explanation += " immediate"; 212 213 if (P.isUnindexed()) 214 Explanation += " unindexed"; 215 216 if (P.isNonExtLoad()) 217 Explanation += " non-extload"; 218 if (P.isAnyExtLoad()) 219 Explanation += " extload"; 220 if (P.isSignExtLoad()) 221 Explanation += " sextload"; 222 if (P.isZeroExtLoad()) 223 Explanation += " zextload"; 224 225 if (P.isNonTruncStore()) 226 Explanation += " non-truncstore"; 227 if (P.isTruncStore()) 228 Explanation += " truncstore"; 229 230 if (Record *VT = P.getMemoryVT()) 231 Explanation += (" MemVT=" + VT->getName()).str(); 232 if (Record *VT = P.getScalarMemoryVT()) 233 Explanation += (" ScalarVT(MemVT)=" + VT->getName()).str(); 234 235 if (P.isAtomicOrderingMonotonic()) 236 Explanation += " monotonic"; 237 if (P.isAtomicOrderingAcquire()) 238 Explanation += " acquire"; 239 if (P.isAtomicOrderingRelease()) 240 Explanation += " release"; 241 if (P.isAtomicOrderingAcquireRelease()) 242 Explanation += " acq_rel"; 243 if (P.isAtomicOrderingSequentiallyConsistent()) 244 Explanation += " seq_cst"; 245 if (P.isAtomicOrderingAcquireOrStronger()) 246 Explanation += " >=acquire"; 247 if (P.isAtomicOrderingWeakerThanAcquire()) 248 Explanation += " <acquire"; 249 if (P.isAtomicOrderingReleaseOrStronger()) 250 Explanation += " >=release"; 251 if (P.isAtomicOrderingWeakerThanRelease()) 252 Explanation += " <release"; 253 } 254 return Explanation; 255 } 256 257 std::string explainOperator(Record *Operator) { 258 if (Operator->isSubClassOf("SDNode")) 259 return (" (" + Operator->getValueAsString("Opcode") + ")").str(); 260 261 if (Operator->isSubClassOf("Intrinsic")) 262 return (" (Operator is an Intrinsic, " + Operator->getName() + ")").str(); 263 264 if (Operator->isSubClassOf("ComplexPattern")) 265 return (" (Operator is an unmapped ComplexPattern, " + Operator->getName() + 266 ")") 267 .str(); 268 269 if (Operator->isSubClassOf("SDNodeXForm")) 270 return (" (Operator is an unmapped SDNodeXForm, " + Operator->getName() + 271 ")") 272 .str(); 273 274 return (" (Operator " + Operator->getName() + " not understood)").str(); 275 } 276 277 /// Helper function to let the emitter report skip reason error messages. 278 static Error failedImport(const Twine &Reason) { 279 return make_error<StringError>(Reason, inconvertibleErrorCode()); 280 } 281 282 static Error isTrivialOperatorNode(const TreePatternNode *N) { 283 std::string Explanation = ""; 284 std::string Separator = ""; 285 286 bool HasUnsupportedPredicate = false; 287 for (const TreePredicateCall &Call : N->getPredicateCalls()) { 288 const TreePredicateFn &Predicate = Call.Fn; 289 290 if (Predicate.isAlwaysTrue()) 291 continue; 292 293 if (Predicate.isImmediatePattern()) 294 continue; 295 296 if (Predicate.isNonExtLoad() || Predicate.isAnyExtLoad() || 297 Predicate.isSignExtLoad() || Predicate.isZeroExtLoad()) 298 continue; 299 300 if (Predicate.isNonTruncStore()) 301 continue; 302 303 if (Predicate.isLoad() && Predicate.getMemoryVT()) 304 continue; 305 306 if (Predicate.isLoad() || Predicate.isStore()) { 307 if (Predicate.isUnindexed()) 308 continue; 309 } 310 311 if (Predicate.isAtomic() && Predicate.getMemoryVT()) 312 continue; 313 314 if (Predicate.isAtomic() && 315 (Predicate.isAtomicOrderingMonotonic() || 316 Predicate.isAtomicOrderingAcquire() || 317 Predicate.isAtomicOrderingRelease() || 318 Predicate.isAtomicOrderingAcquireRelease() || 319 Predicate.isAtomicOrderingSequentiallyConsistent() || 320 Predicate.isAtomicOrderingAcquireOrStronger() || 321 Predicate.isAtomicOrderingWeakerThanAcquire() || 322 Predicate.isAtomicOrderingReleaseOrStronger() || 323 Predicate.isAtomicOrderingWeakerThanRelease())) 324 continue; 325 326 if (Predicate.hasGISelPredicateCode()) 327 continue; 328 329 HasUnsupportedPredicate = true; 330 Explanation = Separator + "Has a predicate (" + explainPredicates(N) + ")"; 331 Separator = ", "; 332 Explanation += (Separator + "first-failing:" + 333 Predicate.getOrigPatFragRecord()->getRecord()->getName()) 334 .str(); 335 break; 336 } 337 338 if (!HasUnsupportedPredicate) 339 return Error::success(); 340 341 return failedImport(Explanation); 342 } 343 344 static Record *getInitValueAsRegClass(Init *V) { 345 if (DefInit *VDefInit = dyn_cast<DefInit>(V)) { 346 if (VDefInit->getDef()->isSubClassOf("RegisterOperand")) 347 return VDefInit->getDef()->getValueAsDef("RegClass"); 348 if (VDefInit->getDef()->isSubClassOf("RegisterClass")) 349 return VDefInit->getDef(); 350 } 351 return nullptr; 352 } 353 354 std::string 355 getNameForFeatureBitset(const std::vector<Record *> &FeatureBitset) { 356 std::string Name = "GIFBS"; 357 for (const auto &Feature : FeatureBitset) 358 Name += ("_" + Feature->getName()).str(); 359 return Name; 360 } 361 362 //===- MatchTable Helpers -------------------------------------------------===// 363 364 class MatchTable; 365 366 /// A record to be stored in a MatchTable. 367 /// 368 /// This class represents any and all output that may be required to emit the 369 /// MatchTable. Instances are most often configured to represent an opcode or 370 /// value that will be emitted to the table with some formatting but it can also 371 /// represent commas, comments, and other formatting instructions. 372 struct MatchTableRecord { 373 enum RecordFlagsBits { 374 MTRF_None = 0x0, 375 /// Causes EmitStr to be formatted as comment when emitted. 376 MTRF_Comment = 0x1, 377 /// Causes the record value to be followed by a comma when emitted. 378 MTRF_CommaFollows = 0x2, 379 /// Causes the record value to be followed by a line break when emitted. 380 MTRF_LineBreakFollows = 0x4, 381 /// Indicates that the record defines a label and causes an additional 382 /// comment to be emitted containing the index of the label. 383 MTRF_Label = 0x8, 384 /// Causes the record to be emitted as the index of the label specified by 385 /// LabelID along with a comment indicating where that label is. 386 MTRF_JumpTarget = 0x10, 387 /// Causes the formatter to add a level of indentation before emitting the 388 /// record. 389 MTRF_Indent = 0x20, 390 /// Causes the formatter to remove a level of indentation after emitting the 391 /// record. 392 MTRF_Outdent = 0x40, 393 }; 394 395 /// When MTRF_Label or MTRF_JumpTarget is used, indicates a label id to 396 /// reference or define. 397 unsigned LabelID; 398 /// The string to emit. Depending on the MTRF_* flags it may be a comment, a 399 /// value, a label name. 400 std::string EmitStr; 401 402 private: 403 /// The number of MatchTable elements described by this record. Comments are 0 404 /// while values are typically 1. Values >1 may occur when we need to emit 405 /// values that exceed the size of a MatchTable element. 406 unsigned NumElements; 407 408 public: 409 /// A bitfield of RecordFlagsBits flags. 410 unsigned Flags; 411 412 /// The actual run-time value, if known 413 int64_t RawValue; 414 415 MatchTableRecord(Optional<unsigned> LabelID_, StringRef EmitStr, 416 unsigned NumElements, unsigned Flags, 417 int64_t RawValue = std::numeric_limits<int64_t>::min()) 418 : LabelID(LabelID_.hasValue() ? LabelID_.getValue() : ~0u), 419 EmitStr(EmitStr), NumElements(NumElements), Flags(Flags), 420 RawValue(RawValue) { 421 422 assert((!LabelID_.hasValue() || LabelID != ~0u) && 423 "This value is reserved for non-labels"); 424 } 425 MatchTableRecord(const MatchTableRecord &Other) = default; 426 MatchTableRecord(MatchTableRecord &&Other) = default; 427 428 /// Useful if a Match Table Record gets optimized out 429 void turnIntoComment() { 430 Flags |= MTRF_Comment; 431 Flags &= ~MTRF_CommaFollows; 432 NumElements = 0; 433 } 434 435 /// For Jump Table generation purposes 436 bool operator<(const MatchTableRecord &Other) const { 437 return RawValue < Other.RawValue; 438 } 439 int64_t getRawValue() const { return RawValue; } 440 441 void emit(raw_ostream &OS, bool LineBreakNextAfterThis, 442 const MatchTable &Table) const; 443 unsigned size() const { return NumElements; } 444 }; 445 446 class Matcher; 447 448 /// Holds the contents of a generated MatchTable to enable formatting and the 449 /// necessary index tracking needed to support GIM_Try. 450 class MatchTable { 451 /// An unique identifier for the table. The generated table will be named 452 /// MatchTable${ID}. 453 unsigned ID; 454 /// The records that make up the table. Also includes comments describing the 455 /// values being emitted and line breaks to format it. 456 std::vector<MatchTableRecord> Contents; 457 /// The currently defined labels. 458 DenseMap<unsigned, unsigned> LabelMap; 459 /// Tracks the sum of MatchTableRecord::NumElements as the table is built. 460 unsigned CurrentSize = 0; 461 /// A unique identifier for a MatchTable label. 462 unsigned CurrentLabelID = 0; 463 /// Determines if the table should be instrumented for rule coverage tracking. 464 bool IsWithCoverage; 465 466 public: 467 static MatchTableRecord LineBreak; 468 static MatchTableRecord Comment(StringRef Comment) { 469 return MatchTableRecord(None, Comment, 0, MatchTableRecord::MTRF_Comment); 470 } 471 static MatchTableRecord Opcode(StringRef Opcode, int IndentAdjust = 0) { 472 unsigned ExtraFlags = 0; 473 if (IndentAdjust > 0) 474 ExtraFlags |= MatchTableRecord::MTRF_Indent; 475 if (IndentAdjust < 0) 476 ExtraFlags |= MatchTableRecord::MTRF_Outdent; 477 478 return MatchTableRecord(None, Opcode, 1, 479 MatchTableRecord::MTRF_CommaFollows | ExtraFlags); 480 } 481 static MatchTableRecord NamedValue(StringRef NamedValue) { 482 return MatchTableRecord(None, NamedValue, 1, 483 MatchTableRecord::MTRF_CommaFollows); 484 } 485 static MatchTableRecord NamedValue(StringRef NamedValue, int64_t RawValue) { 486 return MatchTableRecord(None, NamedValue, 1, 487 MatchTableRecord::MTRF_CommaFollows, RawValue); 488 } 489 static MatchTableRecord NamedValue(StringRef Namespace, 490 StringRef NamedValue) { 491 return MatchTableRecord(None, (Namespace + "::" + NamedValue).str(), 1, 492 MatchTableRecord::MTRF_CommaFollows); 493 } 494 static MatchTableRecord NamedValue(StringRef Namespace, StringRef NamedValue, 495 int64_t RawValue) { 496 return MatchTableRecord(None, (Namespace + "::" + NamedValue).str(), 1, 497 MatchTableRecord::MTRF_CommaFollows, RawValue); 498 } 499 static MatchTableRecord IntValue(int64_t IntValue) { 500 return MatchTableRecord(None, llvm::to_string(IntValue), 1, 501 MatchTableRecord::MTRF_CommaFollows); 502 } 503 static MatchTableRecord Label(unsigned LabelID) { 504 return MatchTableRecord(LabelID, "Label " + llvm::to_string(LabelID), 0, 505 MatchTableRecord::MTRF_Label | 506 MatchTableRecord::MTRF_Comment | 507 MatchTableRecord::MTRF_LineBreakFollows); 508 } 509 static MatchTableRecord JumpTarget(unsigned LabelID) { 510 return MatchTableRecord(LabelID, "Label " + llvm::to_string(LabelID), 1, 511 MatchTableRecord::MTRF_JumpTarget | 512 MatchTableRecord::MTRF_Comment | 513 MatchTableRecord::MTRF_CommaFollows); 514 } 515 516 static MatchTable buildTable(ArrayRef<Matcher *> Rules, bool WithCoverage); 517 518 MatchTable(bool WithCoverage, unsigned ID = 0) 519 : ID(ID), IsWithCoverage(WithCoverage) {} 520 521 bool isWithCoverage() const { return IsWithCoverage; } 522 523 void push_back(const MatchTableRecord &Value) { 524 if (Value.Flags & MatchTableRecord::MTRF_Label) 525 defineLabel(Value.LabelID); 526 Contents.push_back(Value); 527 CurrentSize += Value.size(); 528 } 529 530 unsigned allocateLabelID() { return CurrentLabelID++; } 531 532 void defineLabel(unsigned LabelID) { 533 LabelMap.insert(std::make_pair(LabelID, CurrentSize)); 534 } 535 536 unsigned getLabelIndex(unsigned LabelID) const { 537 const auto I = LabelMap.find(LabelID); 538 assert(I != LabelMap.end() && "Use of undeclared label"); 539 return I->second; 540 } 541 542 void emitUse(raw_ostream &OS) const { OS << "MatchTable" << ID; } 543 544 void emitDeclaration(raw_ostream &OS) const { 545 unsigned Indentation = 4; 546 OS << " constexpr static int64_t MatchTable" << ID << "[] = {"; 547 LineBreak.emit(OS, true, *this); 548 OS << std::string(Indentation, ' '); 549 550 for (auto I = Contents.begin(), E = Contents.end(); I != E; 551 ++I) { 552 bool LineBreakIsNext = false; 553 const auto &NextI = std::next(I); 554 555 if (NextI != E) { 556 if (NextI->EmitStr == "" && 557 NextI->Flags == MatchTableRecord::MTRF_LineBreakFollows) 558 LineBreakIsNext = true; 559 } 560 561 if (I->Flags & MatchTableRecord::MTRF_Indent) 562 Indentation += 2; 563 564 I->emit(OS, LineBreakIsNext, *this); 565 if (I->Flags & MatchTableRecord::MTRF_LineBreakFollows) 566 OS << std::string(Indentation, ' '); 567 568 if (I->Flags & MatchTableRecord::MTRF_Outdent) 569 Indentation -= 2; 570 } 571 OS << "};\n"; 572 } 573 }; 574 575 MatchTableRecord MatchTable::LineBreak = { 576 None, "" /* Emit String */, 0 /* Elements */, 577 MatchTableRecord::MTRF_LineBreakFollows}; 578 579 void MatchTableRecord::emit(raw_ostream &OS, bool LineBreakIsNextAfterThis, 580 const MatchTable &Table) const { 581 bool UseLineComment = 582 LineBreakIsNextAfterThis | (Flags & MTRF_LineBreakFollows); 583 if (Flags & (MTRF_JumpTarget | MTRF_CommaFollows)) 584 UseLineComment = false; 585 586 if (Flags & MTRF_Comment) 587 OS << (UseLineComment ? "// " : "/*"); 588 589 OS << EmitStr; 590 if (Flags & MTRF_Label) 591 OS << ": @" << Table.getLabelIndex(LabelID); 592 593 if (Flags & MTRF_Comment && !UseLineComment) 594 OS << "*/"; 595 596 if (Flags & MTRF_JumpTarget) { 597 if (Flags & MTRF_Comment) 598 OS << " "; 599 OS << Table.getLabelIndex(LabelID); 600 } 601 602 if (Flags & MTRF_CommaFollows) { 603 OS << ","; 604 if (!LineBreakIsNextAfterThis && !(Flags & MTRF_LineBreakFollows)) 605 OS << " "; 606 } 607 608 if (Flags & MTRF_LineBreakFollows) 609 OS << "\n"; 610 } 611 612 MatchTable &operator<<(MatchTable &Table, const MatchTableRecord &Value) { 613 Table.push_back(Value); 614 return Table; 615 } 616 617 //===- Matchers -----------------------------------------------------------===// 618 619 class OperandMatcher; 620 class MatchAction; 621 class PredicateMatcher; 622 class RuleMatcher; 623 624 class Matcher { 625 public: 626 virtual ~Matcher() = default; 627 virtual void optimize() {} 628 virtual void emit(MatchTable &Table) = 0; 629 630 virtual bool hasFirstCondition() const = 0; 631 virtual const PredicateMatcher &getFirstCondition() const = 0; 632 virtual std::unique_ptr<PredicateMatcher> popFirstCondition() = 0; 633 }; 634 635 MatchTable MatchTable::buildTable(ArrayRef<Matcher *> Rules, 636 bool WithCoverage) { 637 MatchTable Table(WithCoverage); 638 for (Matcher *Rule : Rules) 639 Rule->emit(Table); 640 641 return Table << MatchTable::Opcode("GIM_Reject") << MatchTable::LineBreak; 642 } 643 644 class GroupMatcher final : public Matcher { 645 /// Conditions that form a common prefix of all the matchers contained. 646 SmallVector<std::unique_ptr<PredicateMatcher>, 1> Conditions; 647 648 /// All the nested matchers, sharing a common prefix. 649 std::vector<Matcher *> Matchers; 650 651 /// An owning collection for any auxiliary matchers created while optimizing 652 /// nested matchers contained. 653 std::vector<std::unique_ptr<Matcher>> MatcherStorage; 654 655 public: 656 /// Add a matcher to the collection of nested matchers if it meets the 657 /// requirements, and return true. If it doesn't, do nothing and return false. 658 /// 659 /// Expected to preserve its argument, so it could be moved out later on. 660 bool addMatcher(Matcher &Candidate); 661 662 /// Mark the matcher as fully-built and ensure any invariants expected by both 663 /// optimize() and emit(...) methods. Generally, both sequences of calls 664 /// are expected to lead to a sensible result: 665 /// 666 /// addMatcher(...)*; finalize(); optimize(); emit(...); and 667 /// addMatcher(...)*; finalize(); emit(...); 668 /// 669 /// or generally 670 /// 671 /// addMatcher(...)*; finalize(); { optimize()*; emit(...); }* 672 /// 673 /// Multiple calls to optimize() are expected to be handled gracefully, though 674 /// optimize() is not expected to be idempotent. Multiple calls to finalize() 675 /// aren't generally supported. emit(...) is expected to be non-mutating and 676 /// producing the exact same results upon repeated calls. 677 /// 678 /// addMatcher() calls after the finalize() call are not supported. 679 /// 680 /// finalize() and optimize() are both allowed to mutate the contained 681 /// matchers, so moving them out after finalize() is not supported. 682 void finalize(); 683 void optimize() override; 684 void emit(MatchTable &Table) override; 685 686 /// Could be used to move out the matchers added previously, unless finalize() 687 /// has been already called. If any of the matchers are moved out, the group 688 /// becomes safe to destroy, but not safe to re-use for anything else. 689 iterator_range<std::vector<Matcher *>::iterator> matchers() { 690 return make_range(Matchers.begin(), Matchers.end()); 691 } 692 size_t size() const { return Matchers.size(); } 693 bool empty() const { return Matchers.empty(); } 694 695 std::unique_ptr<PredicateMatcher> popFirstCondition() override { 696 assert(!Conditions.empty() && 697 "Trying to pop a condition from a condition-less group"); 698 std::unique_ptr<PredicateMatcher> P = std::move(Conditions.front()); 699 Conditions.erase(Conditions.begin()); 700 return P; 701 } 702 const PredicateMatcher &getFirstCondition() const override { 703 assert(!Conditions.empty() && 704 "Trying to get a condition from a condition-less group"); 705 return *Conditions.front(); 706 } 707 bool hasFirstCondition() const override { return !Conditions.empty(); } 708 709 private: 710 /// See if a candidate matcher could be added to this group solely by 711 /// analyzing its first condition. 712 bool candidateConditionMatches(const PredicateMatcher &Predicate) const; 713 }; 714 715 class SwitchMatcher : public Matcher { 716 /// All the nested matchers, representing distinct switch-cases. The first 717 /// conditions (as Matcher::getFirstCondition() reports) of all the nested 718 /// matchers must share the same type and path to a value they check, in other 719 /// words, be isIdenticalDownToValue, but have different values they check 720 /// against. 721 std::vector<Matcher *> Matchers; 722 723 /// The representative condition, with a type and a path (InsnVarID and OpIdx 724 /// in most cases) shared by all the matchers contained. 725 std::unique_ptr<PredicateMatcher> Condition = nullptr; 726 727 /// Temporary set used to check that the case values don't repeat within the 728 /// same switch. 729 std::set<MatchTableRecord> Values; 730 731 /// An owning collection for any auxiliary matchers created while optimizing 732 /// nested matchers contained. 733 std::vector<std::unique_ptr<Matcher>> MatcherStorage; 734 735 public: 736 bool addMatcher(Matcher &Candidate); 737 738 void finalize(); 739 void emit(MatchTable &Table) override; 740 741 iterator_range<std::vector<Matcher *>::iterator> matchers() { 742 return make_range(Matchers.begin(), Matchers.end()); 743 } 744 size_t size() const { return Matchers.size(); } 745 bool empty() const { return Matchers.empty(); } 746 747 std::unique_ptr<PredicateMatcher> popFirstCondition() override { 748 // SwitchMatcher doesn't have a common first condition for its cases, as all 749 // the cases only share a kind of a value (a type and a path to it) they 750 // match, but deliberately differ in the actual value they match. 751 llvm_unreachable("Trying to pop a condition from a condition-less group"); 752 } 753 const PredicateMatcher &getFirstCondition() const override { 754 llvm_unreachable("Trying to pop a condition from a condition-less group"); 755 } 756 bool hasFirstCondition() const override { return false; } 757 758 private: 759 /// See if the predicate type has a Switch-implementation for it. 760 static bool isSupportedPredicateType(const PredicateMatcher &Predicate); 761 762 bool candidateConditionMatches(const PredicateMatcher &Predicate) const; 763 764 /// emit()-helper 765 static void emitPredicateSpecificOpcodes(const PredicateMatcher &P, 766 MatchTable &Table); 767 }; 768 769 /// Generates code to check that a match rule matches. 770 class RuleMatcher : public Matcher { 771 public: 772 using ActionList = std::list<std::unique_ptr<MatchAction>>; 773 using action_iterator = ActionList::iterator; 774 775 protected: 776 /// A list of matchers that all need to succeed for the current rule to match. 777 /// FIXME: This currently supports a single match position but could be 778 /// extended to support multiple positions to support div/rem fusion or 779 /// load-multiple instructions. 780 using MatchersTy = std::vector<std::unique_ptr<InstructionMatcher>> ; 781 MatchersTy Matchers; 782 783 /// A list of actions that need to be taken when all predicates in this rule 784 /// have succeeded. 785 ActionList Actions; 786 787 using DefinedInsnVariablesMap = std::map<InstructionMatcher *, unsigned>; 788 789 /// A map of instruction matchers to the local variables 790 DefinedInsnVariablesMap InsnVariableIDs; 791 792 using MutatableInsnSet = SmallPtrSet<InstructionMatcher *, 4>; 793 794 // The set of instruction matchers that have not yet been claimed for mutation 795 // by a BuildMI. 796 MutatableInsnSet MutatableInsns; 797 798 /// A map of named operands defined by the matchers that may be referenced by 799 /// the renderers. 800 StringMap<OperandMatcher *> DefinedOperands; 801 802 /// ID for the next instruction variable defined with implicitlyDefineInsnVar() 803 unsigned NextInsnVarID; 804 805 /// ID for the next output instruction allocated with allocateOutputInsnID() 806 unsigned NextOutputInsnID; 807 808 /// ID for the next temporary register ID allocated with allocateTempRegID() 809 unsigned NextTempRegID; 810 811 std::vector<Record *> RequiredFeatures; 812 std::vector<std::unique_ptr<PredicateMatcher>> EpilogueMatchers; 813 814 ArrayRef<SMLoc> SrcLoc; 815 816 typedef std::tuple<Record *, unsigned, unsigned> 817 DefinedComplexPatternSubOperand; 818 typedef StringMap<DefinedComplexPatternSubOperand> 819 DefinedComplexPatternSubOperandMap; 820 /// A map of Symbolic Names to ComplexPattern sub-operands. 821 DefinedComplexPatternSubOperandMap ComplexSubOperands; 822 823 uint64_t RuleID; 824 static uint64_t NextRuleID; 825 826 public: 827 RuleMatcher(ArrayRef<SMLoc> SrcLoc) 828 : Matchers(), Actions(), InsnVariableIDs(), MutatableInsns(), 829 DefinedOperands(), NextInsnVarID(0), NextOutputInsnID(0), 830 NextTempRegID(0), SrcLoc(SrcLoc), ComplexSubOperands(), 831 RuleID(NextRuleID++) {} 832 RuleMatcher(RuleMatcher &&Other) = default; 833 RuleMatcher &operator=(RuleMatcher &&Other) = default; 834 835 uint64_t getRuleID() const { return RuleID; } 836 837 InstructionMatcher &addInstructionMatcher(StringRef SymbolicName); 838 void addRequiredFeature(Record *Feature); 839 const std::vector<Record *> &getRequiredFeatures() const; 840 841 template <class Kind, class... Args> Kind &addAction(Args &&... args); 842 template <class Kind, class... Args> 843 action_iterator insertAction(action_iterator InsertPt, Args &&... args); 844 845 /// Define an instruction without emitting any code to do so. 846 unsigned implicitlyDefineInsnVar(InstructionMatcher &Matcher); 847 848 unsigned getInsnVarID(InstructionMatcher &InsnMatcher) const; 849 DefinedInsnVariablesMap::const_iterator defined_insn_vars_begin() const { 850 return InsnVariableIDs.begin(); 851 } 852 DefinedInsnVariablesMap::const_iterator defined_insn_vars_end() const { 853 return InsnVariableIDs.end(); 854 } 855 iterator_range<typename DefinedInsnVariablesMap::const_iterator> 856 defined_insn_vars() const { 857 return make_range(defined_insn_vars_begin(), defined_insn_vars_end()); 858 } 859 860 MutatableInsnSet::const_iterator mutatable_insns_begin() const { 861 return MutatableInsns.begin(); 862 } 863 MutatableInsnSet::const_iterator mutatable_insns_end() const { 864 return MutatableInsns.end(); 865 } 866 iterator_range<typename MutatableInsnSet::const_iterator> 867 mutatable_insns() const { 868 return make_range(mutatable_insns_begin(), mutatable_insns_end()); 869 } 870 void reserveInsnMatcherForMutation(InstructionMatcher *InsnMatcher) { 871 bool R = MutatableInsns.erase(InsnMatcher); 872 assert(R && "Reserving a mutatable insn that isn't available"); 873 (void)R; 874 } 875 876 action_iterator actions_begin() { return Actions.begin(); } 877 action_iterator actions_end() { return Actions.end(); } 878 iterator_range<action_iterator> actions() { 879 return make_range(actions_begin(), actions_end()); 880 } 881 882 void defineOperand(StringRef SymbolicName, OperandMatcher &OM); 883 884 Error defineComplexSubOperand(StringRef SymbolicName, Record *ComplexPattern, 885 unsigned RendererID, unsigned SubOperandID) { 886 if (ComplexSubOperands.count(SymbolicName)) 887 return failedImport( 888 "Complex suboperand referenced more than once (Operand: " + 889 SymbolicName + ")"); 890 891 ComplexSubOperands[SymbolicName] = 892 std::make_tuple(ComplexPattern, RendererID, SubOperandID); 893 894 return Error::success(); 895 } 896 897 Optional<DefinedComplexPatternSubOperand> 898 getComplexSubOperand(StringRef SymbolicName) const { 899 const auto &I = ComplexSubOperands.find(SymbolicName); 900 if (I == ComplexSubOperands.end()) 901 return None; 902 return I->second; 903 } 904 905 InstructionMatcher &getInstructionMatcher(StringRef SymbolicName) const; 906 const OperandMatcher &getOperandMatcher(StringRef Name) const; 907 908 void optimize() override; 909 void emit(MatchTable &Table) override; 910 911 /// Compare the priority of this object and B. 912 /// 913 /// Returns true if this object is more important than B. 914 bool isHigherPriorityThan(const RuleMatcher &B) const; 915 916 /// Report the maximum number of temporary operands needed by the rule 917 /// matcher. 918 unsigned countRendererFns() const; 919 920 std::unique_ptr<PredicateMatcher> popFirstCondition() override; 921 const PredicateMatcher &getFirstCondition() const override; 922 LLTCodeGen getFirstConditionAsRootType(); 923 bool hasFirstCondition() const override; 924 unsigned getNumOperands() const; 925 StringRef getOpcode() const; 926 927 // FIXME: Remove this as soon as possible 928 InstructionMatcher &insnmatchers_front() const { return *Matchers.front(); } 929 930 unsigned allocateOutputInsnID() { return NextOutputInsnID++; } 931 unsigned allocateTempRegID() { return NextTempRegID++; } 932 933 iterator_range<MatchersTy::iterator> insnmatchers() { 934 return make_range(Matchers.begin(), Matchers.end()); 935 } 936 bool insnmatchers_empty() const { return Matchers.empty(); } 937 void insnmatchers_pop_front() { Matchers.erase(Matchers.begin()); } 938 }; 939 940 uint64_t RuleMatcher::NextRuleID = 0; 941 942 using action_iterator = RuleMatcher::action_iterator; 943 944 template <class PredicateTy> class PredicateListMatcher { 945 private: 946 /// Template instantiations should specialize this to return a string to use 947 /// for the comment emitted when there are no predicates. 948 std::string getNoPredicateComment() const; 949 950 protected: 951 using PredicatesTy = std::deque<std::unique_ptr<PredicateTy>>; 952 PredicatesTy Predicates; 953 954 /// Track if the list of predicates was manipulated by one of the optimization 955 /// methods. 956 bool Optimized = false; 957 958 public: 959 /// Construct a new predicate and add it to the matcher. 960 template <class Kind, class... Args> 961 Optional<Kind *> addPredicate(Args &&... args); 962 963 typename PredicatesTy::iterator predicates_begin() { 964 return Predicates.begin(); 965 } 966 typename PredicatesTy::iterator predicates_end() { 967 return Predicates.end(); 968 } 969 iterator_range<typename PredicatesTy::iterator> predicates() { 970 return make_range(predicates_begin(), predicates_end()); 971 } 972 typename PredicatesTy::size_type predicates_size() const { 973 return Predicates.size(); 974 } 975 bool predicates_empty() const { return Predicates.empty(); } 976 977 std::unique_ptr<PredicateTy> predicates_pop_front() { 978 std::unique_ptr<PredicateTy> Front = std::move(Predicates.front()); 979 Predicates.pop_front(); 980 Optimized = true; 981 return Front; 982 } 983 984 void prependPredicate(std::unique_ptr<PredicateTy> &&Predicate) { 985 Predicates.push_front(std::move(Predicate)); 986 } 987 988 void eraseNullPredicates() { 989 const auto NewEnd = 990 std::stable_partition(Predicates.begin(), Predicates.end(), 991 std::logical_not<std::unique_ptr<PredicateTy>>()); 992 if (NewEnd != Predicates.begin()) { 993 Predicates.erase(Predicates.begin(), NewEnd); 994 Optimized = true; 995 } 996 } 997 998 /// Emit MatchTable opcodes that tests whether all the predicates are met. 999 template <class... Args> 1000 void emitPredicateListOpcodes(MatchTable &Table, Args &&... args) { 1001 if (Predicates.empty() && !Optimized) { 1002 Table << MatchTable::Comment(getNoPredicateComment()) 1003 << MatchTable::LineBreak; 1004 return; 1005 } 1006 1007 for (const auto &Predicate : predicates()) 1008 Predicate->emitPredicateOpcodes(Table, std::forward<Args>(args)...); 1009 } 1010 }; 1011 1012 class PredicateMatcher { 1013 public: 1014 /// This enum is used for RTTI and also defines the priority that is given to 1015 /// the predicate when generating the matcher code. Kinds with higher priority 1016 /// must be tested first. 1017 /// 1018 /// The relative priority of OPM_LLT, OPM_RegBank, and OPM_MBB do not matter 1019 /// but OPM_Int must have priority over OPM_RegBank since constant integers 1020 /// are represented by a virtual register defined by a G_CONSTANT instruction. 1021 /// 1022 /// Note: The relative priority between IPM_ and OPM_ does not matter, they 1023 /// are currently not compared between each other. 1024 enum PredicateKind { 1025 IPM_Opcode, 1026 IPM_NumOperands, 1027 IPM_ImmPredicate, 1028 IPM_AtomicOrderingMMO, 1029 IPM_MemoryLLTSize, 1030 IPM_MemoryVsLLTSize, 1031 IPM_GenericPredicate, 1032 OPM_SameOperand, 1033 OPM_ComplexPattern, 1034 OPM_IntrinsicID, 1035 OPM_Instruction, 1036 OPM_Int, 1037 OPM_LiteralInt, 1038 OPM_LLT, 1039 OPM_PointerToAny, 1040 OPM_RegBank, 1041 OPM_MBB, 1042 }; 1043 1044 protected: 1045 PredicateKind Kind; 1046 unsigned InsnVarID; 1047 unsigned OpIdx; 1048 1049 public: 1050 PredicateMatcher(PredicateKind Kind, unsigned InsnVarID, unsigned OpIdx = ~0) 1051 : Kind(Kind), InsnVarID(InsnVarID), OpIdx(OpIdx) {} 1052 1053 unsigned getInsnVarID() const { return InsnVarID; } 1054 unsigned getOpIdx() const { return OpIdx; } 1055 1056 virtual ~PredicateMatcher() = default; 1057 /// Emit MatchTable opcodes that check the predicate for the given operand. 1058 virtual void emitPredicateOpcodes(MatchTable &Table, 1059 RuleMatcher &Rule) const = 0; 1060 1061 PredicateKind getKind() const { return Kind; } 1062 1063 virtual bool isIdentical(const PredicateMatcher &B) const { 1064 return B.getKind() == getKind() && InsnVarID == B.InsnVarID && 1065 OpIdx == B.OpIdx; 1066 } 1067 1068 virtual bool isIdenticalDownToValue(const PredicateMatcher &B) const { 1069 return hasValue() && PredicateMatcher::isIdentical(B); 1070 } 1071 1072 virtual MatchTableRecord getValue() const { 1073 assert(hasValue() && "Can not get a value of a value-less predicate!"); 1074 llvm_unreachable("Not implemented yet"); 1075 } 1076 virtual bool hasValue() const { return false; } 1077 1078 /// Report the maximum number of temporary operands needed by the predicate 1079 /// matcher. 1080 virtual unsigned countRendererFns() const { return 0; } 1081 }; 1082 1083 /// Generates code to check a predicate of an operand. 1084 /// 1085 /// Typical predicates include: 1086 /// * Operand is a particular register. 1087 /// * Operand is assigned a particular register bank. 1088 /// * Operand is an MBB. 1089 class OperandPredicateMatcher : public PredicateMatcher { 1090 public: 1091 OperandPredicateMatcher(PredicateKind Kind, unsigned InsnVarID, 1092 unsigned OpIdx) 1093 : PredicateMatcher(Kind, InsnVarID, OpIdx) {} 1094 virtual ~OperandPredicateMatcher() {} 1095 1096 /// Compare the priority of this object and B. 1097 /// 1098 /// Returns true if this object is more important than B. 1099 virtual bool isHigherPriorityThan(const OperandPredicateMatcher &B) const; 1100 }; 1101 1102 template <> 1103 std::string 1104 PredicateListMatcher<OperandPredicateMatcher>::getNoPredicateComment() const { 1105 return "No operand predicates"; 1106 } 1107 1108 /// Generates code to check that a register operand is defined by the same exact 1109 /// one as another. 1110 class SameOperandMatcher : public OperandPredicateMatcher { 1111 std::string MatchingName; 1112 1113 public: 1114 SameOperandMatcher(unsigned InsnVarID, unsigned OpIdx, StringRef MatchingName) 1115 : OperandPredicateMatcher(OPM_SameOperand, InsnVarID, OpIdx), 1116 MatchingName(MatchingName) {} 1117 1118 static bool classof(const PredicateMatcher *P) { 1119 return P->getKind() == OPM_SameOperand; 1120 } 1121 1122 void emitPredicateOpcodes(MatchTable &Table, 1123 RuleMatcher &Rule) const override; 1124 1125 bool isIdentical(const PredicateMatcher &B) const override { 1126 return OperandPredicateMatcher::isIdentical(B) && 1127 MatchingName == cast<SameOperandMatcher>(&B)->MatchingName; 1128 } 1129 }; 1130 1131 /// Generates code to check that an operand is a particular LLT. 1132 class LLTOperandMatcher : public OperandPredicateMatcher { 1133 protected: 1134 LLTCodeGen Ty; 1135 1136 public: 1137 static std::map<LLTCodeGen, unsigned> TypeIDValues; 1138 1139 static void initTypeIDValuesMap() { 1140 TypeIDValues.clear(); 1141 1142 unsigned ID = 0; 1143 for (const LLTCodeGen LLTy : KnownTypes) 1144 TypeIDValues[LLTy] = ID++; 1145 } 1146 1147 LLTOperandMatcher(unsigned InsnVarID, unsigned OpIdx, const LLTCodeGen &Ty) 1148 : OperandPredicateMatcher(OPM_LLT, InsnVarID, OpIdx), Ty(Ty) { 1149 KnownTypes.insert(Ty); 1150 } 1151 1152 static bool classof(const PredicateMatcher *P) { 1153 return P->getKind() == OPM_LLT; 1154 } 1155 bool isIdentical(const PredicateMatcher &B) const override { 1156 return OperandPredicateMatcher::isIdentical(B) && 1157 Ty == cast<LLTOperandMatcher>(&B)->Ty; 1158 } 1159 MatchTableRecord getValue() const override { 1160 const auto VI = TypeIDValues.find(Ty); 1161 if (VI == TypeIDValues.end()) 1162 return MatchTable::NamedValue(getTy().getCxxEnumValue()); 1163 return MatchTable::NamedValue(getTy().getCxxEnumValue(), VI->second); 1164 } 1165 bool hasValue() const override { 1166 if (TypeIDValues.size() != KnownTypes.size()) 1167 initTypeIDValuesMap(); 1168 return TypeIDValues.count(Ty); 1169 } 1170 1171 LLTCodeGen getTy() const { return Ty; } 1172 1173 void emitPredicateOpcodes(MatchTable &Table, 1174 RuleMatcher &Rule) const override { 1175 Table << MatchTable::Opcode("GIM_CheckType") << MatchTable::Comment("MI") 1176 << MatchTable::IntValue(InsnVarID) << MatchTable::Comment("Op") 1177 << MatchTable::IntValue(OpIdx) << MatchTable::Comment("Type") 1178 << getValue() << MatchTable::LineBreak; 1179 } 1180 }; 1181 1182 std::map<LLTCodeGen, unsigned> LLTOperandMatcher::TypeIDValues; 1183 1184 /// Generates code to check that an operand is a pointer to any address space. 1185 /// 1186 /// In SelectionDAG, the types did not describe pointers or address spaces. As a 1187 /// result, iN is used to describe a pointer of N bits to any address space and 1188 /// PatFrag predicates are typically used to constrain the address space. There's 1189 /// no reliable means to derive the missing type information from the pattern so 1190 /// imported rules must test the components of a pointer separately. 1191 /// 1192 /// If SizeInBits is zero, then the pointer size will be obtained from the 1193 /// subtarget. 1194 class PointerToAnyOperandMatcher : public OperandPredicateMatcher { 1195 protected: 1196 unsigned SizeInBits; 1197 1198 public: 1199 PointerToAnyOperandMatcher(unsigned InsnVarID, unsigned OpIdx, 1200 unsigned SizeInBits) 1201 : OperandPredicateMatcher(OPM_PointerToAny, InsnVarID, OpIdx), 1202 SizeInBits(SizeInBits) {} 1203 1204 static bool classof(const OperandPredicateMatcher *P) { 1205 return P->getKind() == OPM_PointerToAny; 1206 } 1207 1208 void emitPredicateOpcodes(MatchTable &Table, 1209 RuleMatcher &Rule) const override { 1210 Table << MatchTable::Opcode("GIM_CheckPointerToAny") 1211 << MatchTable::Comment("MI") << MatchTable::IntValue(InsnVarID) 1212 << MatchTable::Comment("Op") << MatchTable::IntValue(OpIdx) 1213 << MatchTable::Comment("SizeInBits") 1214 << MatchTable::IntValue(SizeInBits) << MatchTable::LineBreak; 1215 } 1216 }; 1217 1218 /// Generates code to check that an operand is a particular target constant. 1219 class ComplexPatternOperandMatcher : public OperandPredicateMatcher { 1220 protected: 1221 const OperandMatcher &Operand; 1222 const Record &TheDef; 1223 1224 unsigned getAllocatedTemporariesBaseID() const; 1225 1226 public: 1227 bool isIdentical(const PredicateMatcher &B) const override { return false; } 1228 1229 ComplexPatternOperandMatcher(unsigned InsnVarID, unsigned OpIdx, 1230 const OperandMatcher &Operand, 1231 const Record &TheDef) 1232 : OperandPredicateMatcher(OPM_ComplexPattern, InsnVarID, OpIdx), 1233 Operand(Operand), TheDef(TheDef) {} 1234 1235 static bool classof(const PredicateMatcher *P) { 1236 return P->getKind() == OPM_ComplexPattern; 1237 } 1238 1239 void emitPredicateOpcodes(MatchTable &Table, 1240 RuleMatcher &Rule) const override { 1241 unsigned ID = getAllocatedTemporariesBaseID(); 1242 Table << MatchTable::Opcode("GIM_CheckComplexPattern") 1243 << MatchTable::Comment("MI") << MatchTable::IntValue(InsnVarID) 1244 << MatchTable::Comment("Op") << MatchTable::IntValue(OpIdx) 1245 << MatchTable::Comment("Renderer") << MatchTable::IntValue(ID) 1246 << MatchTable::NamedValue(("GICP_" + TheDef.getName()).str()) 1247 << MatchTable::LineBreak; 1248 } 1249 1250 unsigned countRendererFns() const override { 1251 return 1; 1252 } 1253 }; 1254 1255 /// Generates code to check that an operand is in a particular register bank. 1256 class RegisterBankOperandMatcher : public OperandPredicateMatcher { 1257 protected: 1258 const CodeGenRegisterClass &RC; 1259 1260 public: 1261 RegisterBankOperandMatcher(unsigned InsnVarID, unsigned OpIdx, 1262 const CodeGenRegisterClass &RC) 1263 : OperandPredicateMatcher(OPM_RegBank, InsnVarID, OpIdx), RC(RC) {} 1264 1265 bool isIdentical(const PredicateMatcher &B) const override { 1266 return OperandPredicateMatcher::isIdentical(B) && 1267 RC.getDef() == cast<RegisterBankOperandMatcher>(&B)->RC.getDef(); 1268 } 1269 1270 static bool classof(const PredicateMatcher *P) { 1271 return P->getKind() == OPM_RegBank; 1272 } 1273 1274 void emitPredicateOpcodes(MatchTable &Table, 1275 RuleMatcher &Rule) const override { 1276 Table << MatchTable::Opcode("GIM_CheckRegBankForClass") 1277 << MatchTable::Comment("MI") << MatchTable::IntValue(InsnVarID) 1278 << MatchTable::Comment("Op") << MatchTable::IntValue(OpIdx) 1279 << MatchTable::Comment("RC") 1280 << MatchTable::NamedValue(RC.getQualifiedName() + "RegClassID") 1281 << MatchTable::LineBreak; 1282 } 1283 }; 1284 1285 /// Generates code to check that an operand is a basic block. 1286 class MBBOperandMatcher : public OperandPredicateMatcher { 1287 public: 1288 MBBOperandMatcher(unsigned InsnVarID, unsigned OpIdx) 1289 : OperandPredicateMatcher(OPM_MBB, InsnVarID, OpIdx) {} 1290 1291 static bool classof(const PredicateMatcher *P) { 1292 return P->getKind() == OPM_MBB; 1293 } 1294 1295 void emitPredicateOpcodes(MatchTable &Table, 1296 RuleMatcher &Rule) const override { 1297 Table << MatchTable::Opcode("GIM_CheckIsMBB") << MatchTable::Comment("MI") 1298 << MatchTable::IntValue(InsnVarID) << MatchTable::Comment("Op") 1299 << MatchTable::IntValue(OpIdx) << MatchTable::LineBreak; 1300 } 1301 }; 1302 1303 /// Generates code to check that an operand is a G_CONSTANT with a particular 1304 /// int. 1305 class ConstantIntOperandMatcher : public OperandPredicateMatcher { 1306 protected: 1307 int64_t Value; 1308 1309 public: 1310 ConstantIntOperandMatcher(unsigned InsnVarID, unsigned OpIdx, int64_t Value) 1311 : OperandPredicateMatcher(OPM_Int, InsnVarID, OpIdx), Value(Value) {} 1312 1313 bool isIdentical(const PredicateMatcher &B) const override { 1314 return OperandPredicateMatcher::isIdentical(B) && 1315 Value == cast<ConstantIntOperandMatcher>(&B)->Value; 1316 } 1317 1318 static bool classof(const PredicateMatcher *P) { 1319 return P->getKind() == OPM_Int; 1320 } 1321 1322 void emitPredicateOpcodes(MatchTable &Table, 1323 RuleMatcher &Rule) const override { 1324 Table << MatchTable::Opcode("GIM_CheckConstantInt") 1325 << MatchTable::Comment("MI") << MatchTable::IntValue(InsnVarID) 1326 << MatchTable::Comment("Op") << MatchTable::IntValue(OpIdx) 1327 << MatchTable::IntValue(Value) << MatchTable::LineBreak; 1328 } 1329 }; 1330 1331 /// Generates code to check that an operand is a raw int (where MO.isImm() or 1332 /// MO.isCImm() is true). 1333 class LiteralIntOperandMatcher : public OperandPredicateMatcher { 1334 protected: 1335 int64_t Value; 1336 1337 public: 1338 LiteralIntOperandMatcher(unsigned InsnVarID, unsigned OpIdx, int64_t Value) 1339 : OperandPredicateMatcher(OPM_LiteralInt, InsnVarID, OpIdx), 1340 Value(Value) {} 1341 1342 bool isIdentical(const PredicateMatcher &B) const override { 1343 return OperandPredicateMatcher::isIdentical(B) && 1344 Value == cast<LiteralIntOperandMatcher>(&B)->Value; 1345 } 1346 1347 static bool classof(const PredicateMatcher *P) { 1348 return P->getKind() == OPM_LiteralInt; 1349 } 1350 1351 void emitPredicateOpcodes(MatchTable &Table, 1352 RuleMatcher &Rule) const override { 1353 Table << MatchTable::Opcode("GIM_CheckLiteralInt") 1354 << MatchTable::Comment("MI") << MatchTable::IntValue(InsnVarID) 1355 << MatchTable::Comment("Op") << MatchTable::IntValue(OpIdx) 1356 << MatchTable::IntValue(Value) << MatchTable::LineBreak; 1357 } 1358 }; 1359 1360 /// Generates code to check that an operand is an intrinsic ID. 1361 class IntrinsicIDOperandMatcher : public OperandPredicateMatcher { 1362 protected: 1363 const CodeGenIntrinsic *II; 1364 1365 public: 1366 IntrinsicIDOperandMatcher(unsigned InsnVarID, unsigned OpIdx, 1367 const CodeGenIntrinsic *II) 1368 : OperandPredicateMatcher(OPM_IntrinsicID, InsnVarID, OpIdx), II(II) {} 1369 1370 bool isIdentical(const PredicateMatcher &B) const override { 1371 return OperandPredicateMatcher::isIdentical(B) && 1372 II == cast<IntrinsicIDOperandMatcher>(&B)->II; 1373 } 1374 1375 static bool classof(const PredicateMatcher *P) { 1376 return P->getKind() == OPM_IntrinsicID; 1377 } 1378 1379 void emitPredicateOpcodes(MatchTable &Table, 1380 RuleMatcher &Rule) const override { 1381 Table << MatchTable::Opcode("GIM_CheckIntrinsicID") 1382 << MatchTable::Comment("MI") << MatchTable::IntValue(InsnVarID) 1383 << MatchTable::Comment("Op") << MatchTable::IntValue(OpIdx) 1384 << MatchTable::NamedValue("Intrinsic::" + II->EnumName) 1385 << MatchTable::LineBreak; 1386 } 1387 }; 1388 1389 /// Generates code to check that a set of predicates match for a particular 1390 /// operand. 1391 class OperandMatcher : public PredicateListMatcher<OperandPredicateMatcher> { 1392 protected: 1393 InstructionMatcher &Insn; 1394 unsigned OpIdx; 1395 std::string SymbolicName; 1396 1397 /// The index of the first temporary variable allocated to this operand. The 1398 /// number of allocated temporaries can be found with 1399 /// countRendererFns(). 1400 unsigned AllocatedTemporariesBaseID; 1401 1402 public: 1403 OperandMatcher(InstructionMatcher &Insn, unsigned OpIdx, 1404 const std::string &SymbolicName, 1405 unsigned AllocatedTemporariesBaseID) 1406 : Insn(Insn), OpIdx(OpIdx), SymbolicName(SymbolicName), 1407 AllocatedTemporariesBaseID(AllocatedTemporariesBaseID) {} 1408 1409 bool hasSymbolicName() const { return !SymbolicName.empty(); } 1410 const StringRef getSymbolicName() const { return SymbolicName; } 1411 void setSymbolicName(StringRef Name) { 1412 assert(SymbolicName.empty() && "Operand already has a symbolic name"); 1413 SymbolicName = Name; 1414 } 1415 1416 /// Construct a new operand predicate and add it to the matcher. 1417 template <class Kind, class... Args> 1418 Optional<Kind *> addPredicate(Args &&... args) { 1419 if (isSameAsAnotherOperand()) 1420 return None; 1421 Predicates.emplace_back(llvm::make_unique<Kind>( 1422 getInsnVarID(), getOpIdx(), std::forward<Args>(args)...)); 1423 return static_cast<Kind *>(Predicates.back().get()); 1424 } 1425 1426 unsigned getOpIdx() const { return OpIdx; } 1427 unsigned getInsnVarID() const; 1428 1429 std::string getOperandExpr(unsigned InsnVarID) const { 1430 return "State.MIs[" + llvm::to_string(InsnVarID) + "]->getOperand(" + 1431 llvm::to_string(OpIdx) + ")"; 1432 } 1433 1434 InstructionMatcher &getInstructionMatcher() const { return Insn; } 1435 1436 Error addTypeCheckPredicate(const TypeSetByHwMode &VTy, 1437 bool OperandIsAPointer); 1438 1439 /// Emit MatchTable opcodes that test whether the instruction named in 1440 /// InsnVarID matches all the predicates and all the operands. 1441 void emitPredicateOpcodes(MatchTable &Table, RuleMatcher &Rule) { 1442 if (!Optimized) { 1443 std::string Comment; 1444 raw_string_ostream CommentOS(Comment); 1445 CommentOS << "MIs[" << getInsnVarID() << "] "; 1446 if (SymbolicName.empty()) 1447 CommentOS << "Operand " << OpIdx; 1448 else 1449 CommentOS << SymbolicName; 1450 Table << MatchTable::Comment(CommentOS.str()) << MatchTable::LineBreak; 1451 } 1452 1453 emitPredicateListOpcodes(Table, Rule); 1454 } 1455 1456 /// Compare the priority of this object and B. 1457 /// 1458 /// Returns true if this object is more important than B. 1459 bool isHigherPriorityThan(OperandMatcher &B) { 1460 // Operand matchers involving more predicates have higher priority. 1461 if (predicates_size() > B.predicates_size()) 1462 return true; 1463 if (predicates_size() < B.predicates_size()) 1464 return false; 1465 1466 // This assumes that predicates are added in a consistent order. 1467 for (auto &&Predicate : zip(predicates(), B.predicates())) { 1468 if (std::get<0>(Predicate)->isHigherPriorityThan(*std::get<1>(Predicate))) 1469 return true; 1470 if (std::get<1>(Predicate)->isHigherPriorityThan(*std::get<0>(Predicate))) 1471 return false; 1472 } 1473 1474 return false; 1475 }; 1476 1477 /// Report the maximum number of temporary operands needed by the operand 1478 /// matcher. 1479 unsigned countRendererFns() { 1480 return std::accumulate( 1481 predicates().begin(), predicates().end(), 0, 1482 [](unsigned A, 1483 const std::unique_ptr<OperandPredicateMatcher> &Predicate) { 1484 return A + Predicate->countRendererFns(); 1485 }); 1486 } 1487 1488 unsigned getAllocatedTemporariesBaseID() const { 1489 return AllocatedTemporariesBaseID; 1490 } 1491 1492 bool isSameAsAnotherOperand() { 1493 for (const auto &Predicate : predicates()) 1494 if (isa<SameOperandMatcher>(Predicate)) 1495 return true; 1496 return false; 1497 } 1498 }; 1499 1500 Error OperandMatcher::addTypeCheckPredicate(const TypeSetByHwMode &VTy, 1501 bool OperandIsAPointer) { 1502 if (!VTy.isMachineValueType()) 1503 return failedImport("unsupported typeset"); 1504 1505 if (VTy.getMachineValueType() == MVT::iPTR && OperandIsAPointer) { 1506 addPredicate<PointerToAnyOperandMatcher>(0); 1507 return Error::success(); 1508 } 1509 1510 auto OpTyOrNone = MVTToLLT(VTy.getMachineValueType().SimpleTy); 1511 if (!OpTyOrNone) 1512 return failedImport("unsupported type"); 1513 1514 if (OperandIsAPointer) 1515 addPredicate<PointerToAnyOperandMatcher>(OpTyOrNone->get().getSizeInBits()); 1516 else if (VTy.isPointer()) 1517 addPredicate<LLTOperandMatcher>(LLT::pointer(VTy.getPtrAddrSpace(), 1518 OpTyOrNone->get().getSizeInBits())); 1519 else 1520 addPredicate<LLTOperandMatcher>(*OpTyOrNone); 1521 return Error::success(); 1522 } 1523 1524 unsigned ComplexPatternOperandMatcher::getAllocatedTemporariesBaseID() const { 1525 return Operand.getAllocatedTemporariesBaseID(); 1526 } 1527 1528 /// Generates code to check a predicate on an instruction. 1529 /// 1530 /// Typical predicates include: 1531 /// * The opcode of the instruction is a particular value. 1532 /// * The nsw/nuw flag is/isn't set. 1533 class InstructionPredicateMatcher : public PredicateMatcher { 1534 public: 1535 InstructionPredicateMatcher(PredicateKind Kind, unsigned InsnVarID) 1536 : PredicateMatcher(Kind, InsnVarID) {} 1537 virtual ~InstructionPredicateMatcher() {} 1538 1539 /// Compare the priority of this object and B. 1540 /// 1541 /// Returns true if this object is more important than B. 1542 virtual bool 1543 isHigherPriorityThan(const InstructionPredicateMatcher &B) const { 1544 return Kind < B.Kind; 1545 }; 1546 }; 1547 1548 template <> 1549 std::string 1550 PredicateListMatcher<PredicateMatcher>::getNoPredicateComment() const { 1551 return "No instruction predicates"; 1552 } 1553 1554 /// Generates code to check the opcode of an instruction. 1555 class InstructionOpcodeMatcher : public InstructionPredicateMatcher { 1556 protected: 1557 const CodeGenInstruction *I; 1558 1559 static DenseMap<const CodeGenInstruction *, unsigned> OpcodeValues; 1560 1561 public: 1562 static void initOpcodeValuesMap(const CodeGenTarget &Target) { 1563 OpcodeValues.clear(); 1564 1565 unsigned OpcodeValue = 0; 1566 for (const CodeGenInstruction *I : Target.getInstructionsByEnumValue()) 1567 OpcodeValues[I] = OpcodeValue++; 1568 } 1569 1570 InstructionOpcodeMatcher(unsigned InsnVarID, const CodeGenInstruction *I) 1571 : InstructionPredicateMatcher(IPM_Opcode, InsnVarID), I(I) {} 1572 1573 static bool classof(const PredicateMatcher *P) { 1574 return P->getKind() == IPM_Opcode; 1575 } 1576 1577 bool isIdentical(const PredicateMatcher &B) const override { 1578 return InstructionPredicateMatcher::isIdentical(B) && 1579 I == cast<InstructionOpcodeMatcher>(&B)->I; 1580 } 1581 MatchTableRecord getValue() const override { 1582 const auto VI = OpcodeValues.find(I); 1583 if (VI != OpcodeValues.end()) 1584 return MatchTable::NamedValue(I->Namespace, I->TheDef->getName(), 1585 VI->second); 1586 return MatchTable::NamedValue(I->Namespace, I->TheDef->getName()); 1587 } 1588 bool hasValue() const override { return OpcodeValues.count(I); } 1589 1590 void emitPredicateOpcodes(MatchTable &Table, 1591 RuleMatcher &Rule) const override { 1592 Table << MatchTable::Opcode("GIM_CheckOpcode") << MatchTable::Comment("MI") 1593 << MatchTable::IntValue(InsnVarID) << getValue() 1594 << MatchTable::LineBreak; 1595 } 1596 1597 /// Compare the priority of this object and B. 1598 /// 1599 /// Returns true if this object is more important than B. 1600 bool 1601 isHigherPriorityThan(const InstructionPredicateMatcher &B) const override { 1602 if (InstructionPredicateMatcher::isHigherPriorityThan(B)) 1603 return true; 1604 if (B.InstructionPredicateMatcher::isHigherPriorityThan(*this)) 1605 return false; 1606 1607 // Prioritize opcodes for cosmetic reasons in the generated source. Although 1608 // this is cosmetic at the moment, we may want to drive a similar ordering 1609 // using instruction frequency information to improve compile time. 1610 if (const InstructionOpcodeMatcher *BO = 1611 dyn_cast<InstructionOpcodeMatcher>(&B)) 1612 return I->TheDef->getName() < BO->I->TheDef->getName(); 1613 1614 return false; 1615 }; 1616 1617 bool isConstantInstruction() const { 1618 return I->TheDef->getName() == "G_CONSTANT"; 1619 } 1620 1621 StringRef getOpcode() const { return I->TheDef->getName(); } 1622 unsigned getNumOperands() const { return I->Operands.size(); } 1623 1624 StringRef getOperandType(unsigned OpIdx) const { 1625 return I->Operands[OpIdx].OperandType; 1626 } 1627 }; 1628 1629 DenseMap<const CodeGenInstruction *, unsigned> 1630 InstructionOpcodeMatcher::OpcodeValues; 1631 1632 class InstructionNumOperandsMatcher final : public InstructionPredicateMatcher { 1633 unsigned NumOperands = 0; 1634 1635 public: 1636 InstructionNumOperandsMatcher(unsigned InsnVarID, unsigned NumOperands) 1637 : InstructionPredicateMatcher(IPM_NumOperands, InsnVarID), 1638 NumOperands(NumOperands) {} 1639 1640 static bool classof(const PredicateMatcher *P) { 1641 return P->getKind() == IPM_NumOperands; 1642 } 1643 1644 bool isIdentical(const PredicateMatcher &B) const override { 1645 return InstructionPredicateMatcher::isIdentical(B) && 1646 NumOperands == cast<InstructionNumOperandsMatcher>(&B)->NumOperands; 1647 } 1648 1649 void emitPredicateOpcodes(MatchTable &Table, 1650 RuleMatcher &Rule) const override { 1651 Table << MatchTable::Opcode("GIM_CheckNumOperands") 1652 << MatchTable::Comment("MI") << MatchTable::IntValue(InsnVarID) 1653 << MatchTable::Comment("Expected") 1654 << MatchTable::IntValue(NumOperands) << MatchTable::LineBreak; 1655 } 1656 }; 1657 1658 /// Generates code to check that this instruction is a constant whose value 1659 /// meets an immediate predicate. 1660 /// 1661 /// Immediates are slightly odd since they are typically used like an operand 1662 /// but are represented as an operator internally. We typically write simm8:$src 1663 /// in a tablegen pattern, but this is just syntactic sugar for 1664 /// (imm:i32)<<P:Predicate_simm8>>:$imm which more directly describes the nodes 1665 /// that will be matched and the predicate (which is attached to the imm 1666 /// operator) that will be tested. In SelectionDAG this describes a 1667 /// ConstantSDNode whose internal value will be tested using the simm8 predicate. 1668 /// 1669 /// The corresponding GlobalISel representation is %1 = G_CONSTANT iN Value. In 1670 /// this representation, the immediate could be tested with an 1671 /// InstructionMatcher, InstructionOpcodeMatcher, OperandMatcher, and a 1672 /// OperandPredicateMatcher-subclass to check the Value meets the predicate but 1673 /// there are two implementation issues with producing that matcher 1674 /// configuration from the SelectionDAG pattern: 1675 /// * ImmLeaf is a PatFrag whose root is an InstructionMatcher. This means that 1676 /// were we to sink the immediate predicate to the operand we would have to 1677 /// have two partial implementations of PatFrag support, one for immediates 1678 /// and one for non-immediates. 1679 /// * At the point we handle the predicate, the OperandMatcher hasn't been 1680 /// created yet. If we were to sink the predicate to the OperandMatcher we 1681 /// would also have to complicate (or duplicate) the code that descends and 1682 /// creates matchers for the subtree. 1683 /// Overall, it's simpler to handle it in the place it was found. 1684 class InstructionImmPredicateMatcher : public InstructionPredicateMatcher { 1685 protected: 1686 TreePredicateFn Predicate; 1687 1688 public: 1689 InstructionImmPredicateMatcher(unsigned InsnVarID, 1690 const TreePredicateFn &Predicate) 1691 : InstructionPredicateMatcher(IPM_ImmPredicate, InsnVarID), 1692 Predicate(Predicate) {} 1693 1694 bool isIdentical(const PredicateMatcher &B) const override { 1695 return InstructionPredicateMatcher::isIdentical(B) && 1696 Predicate.getOrigPatFragRecord() == 1697 cast<InstructionImmPredicateMatcher>(&B) 1698 ->Predicate.getOrigPatFragRecord(); 1699 } 1700 1701 static bool classof(const PredicateMatcher *P) { 1702 return P->getKind() == IPM_ImmPredicate; 1703 } 1704 1705 void emitPredicateOpcodes(MatchTable &Table, 1706 RuleMatcher &Rule) const override { 1707 Table << MatchTable::Opcode(getMatchOpcodeForPredicate(Predicate)) 1708 << MatchTable::Comment("MI") << MatchTable::IntValue(InsnVarID) 1709 << MatchTable::Comment("Predicate") 1710 << MatchTable::NamedValue(getEnumNameForPredicate(Predicate)) 1711 << MatchTable::LineBreak; 1712 } 1713 }; 1714 1715 /// Generates code to check that a memory instruction has a atomic ordering 1716 /// MachineMemoryOperand. 1717 class AtomicOrderingMMOPredicateMatcher : public InstructionPredicateMatcher { 1718 public: 1719 enum AOComparator { 1720 AO_Exactly, 1721 AO_OrStronger, 1722 AO_WeakerThan, 1723 }; 1724 1725 protected: 1726 StringRef Order; 1727 AOComparator Comparator; 1728 1729 public: 1730 AtomicOrderingMMOPredicateMatcher(unsigned InsnVarID, StringRef Order, 1731 AOComparator Comparator = AO_Exactly) 1732 : InstructionPredicateMatcher(IPM_AtomicOrderingMMO, InsnVarID), 1733 Order(Order), Comparator(Comparator) {} 1734 1735 static bool classof(const PredicateMatcher *P) { 1736 return P->getKind() == IPM_AtomicOrderingMMO; 1737 } 1738 1739 bool isIdentical(const PredicateMatcher &B) const override { 1740 if (!InstructionPredicateMatcher::isIdentical(B)) 1741 return false; 1742 const auto &R = *cast<AtomicOrderingMMOPredicateMatcher>(&B); 1743 return Order == R.Order && Comparator == R.Comparator; 1744 } 1745 1746 void emitPredicateOpcodes(MatchTable &Table, 1747 RuleMatcher &Rule) const override { 1748 StringRef Opcode = "GIM_CheckAtomicOrdering"; 1749 1750 if (Comparator == AO_OrStronger) 1751 Opcode = "GIM_CheckAtomicOrderingOrStrongerThan"; 1752 if (Comparator == AO_WeakerThan) 1753 Opcode = "GIM_CheckAtomicOrderingWeakerThan"; 1754 1755 Table << MatchTable::Opcode(Opcode) << MatchTable::Comment("MI") 1756 << MatchTable::IntValue(InsnVarID) << MatchTable::Comment("Order") 1757 << MatchTable::NamedValue(("(int64_t)AtomicOrdering::" + Order).str()) 1758 << MatchTable::LineBreak; 1759 } 1760 }; 1761 1762 /// Generates code to check that the size of an MMO is exactly N bytes. 1763 class MemorySizePredicateMatcher : public InstructionPredicateMatcher { 1764 protected: 1765 unsigned MMOIdx; 1766 uint64_t Size; 1767 1768 public: 1769 MemorySizePredicateMatcher(unsigned InsnVarID, unsigned MMOIdx, unsigned Size) 1770 : InstructionPredicateMatcher(IPM_MemoryLLTSize, InsnVarID), 1771 MMOIdx(MMOIdx), Size(Size) {} 1772 1773 static bool classof(const PredicateMatcher *P) { 1774 return P->getKind() == IPM_MemoryLLTSize; 1775 } 1776 bool isIdentical(const PredicateMatcher &B) const override { 1777 return InstructionPredicateMatcher::isIdentical(B) && 1778 MMOIdx == cast<MemorySizePredicateMatcher>(&B)->MMOIdx && 1779 Size == cast<MemorySizePredicateMatcher>(&B)->Size; 1780 } 1781 1782 void emitPredicateOpcodes(MatchTable &Table, 1783 RuleMatcher &Rule) const override { 1784 Table << MatchTable::Opcode("GIM_CheckMemorySizeEqualTo") 1785 << MatchTable::Comment("MI") << MatchTable::IntValue(InsnVarID) 1786 << MatchTable::Comment("MMO") << MatchTable::IntValue(MMOIdx) 1787 << MatchTable::Comment("Size") << MatchTable::IntValue(Size) 1788 << MatchTable::LineBreak; 1789 } 1790 }; 1791 1792 /// Generates code to check that the size of an MMO is less-than, equal-to, or 1793 /// greater than a given LLT. 1794 class MemoryVsLLTSizePredicateMatcher : public InstructionPredicateMatcher { 1795 public: 1796 enum RelationKind { 1797 GreaterThan, 1798 EqualTo, 1799 LessThan, 1800 }; 1801 1802 protected: 1803 unsigned MMOIdx; 1804 RelationKind Relation; 1805 unsigned OpIdx; 1806 1807 public: 1808 MemoryVsLLTSizePredicateMatcher(unsigned InsnVarID, unsigned MMOIdx, 1809 enum RelationKind Relation, 1810 unsigned OpIdx) 1811 : InstructionPredicateMatcher(IPM_MemoryVsLLTSize, InsnVarID), 1812 MMOIdx(MMOIdx), Relation(Relation), OpIdx(OpIdx) {} 1813 1814 static bool classof(const PredicateMatcher *P) { 1815 return P->getKind() == IPM_MemoryVsLLTSize; 1816 } 1817 bool isIdentical(const PredicateMatcher &B) const override { 1818 return InstructionPredicateMatcher::isIdentical(B) && 1819 MMOIdx == cast<MemoryVsLLTSizePredicateMatcher>(&B)->MMOIdx && 1820 Relation == cast<MemoryVsLLTSizePredicateMatcher>(&B)->Relation && 1821 OpIdx == cast<MemoryVsLLTSizePredicateMatcher>(&B)->OpIdx; 1822 } 1823 1824 void emitPredicateOpcodes(MatchTable &Table, 1825 RuleMatcher &Rule) const override { 1826 Table << MatchTable::Opcode(Relation == EqualTo 1827 ? "GIM_CheckMemorySizeEqualToLLT" 1828 : Relation == GreaterThan 1829 ? "GIM_CheckMemorySizeGreaterThanLLT" 1830 : "GIM_CheckMemorySizeLessThanLLT") 1831 << MatchTable::Comment("MI") << MatchTable::IntValue(InsnVarID) 1832 << MatchTable::Comment("MMO") << MatchTable::IntValue(MMOIdx) 1833 << MatchTable::Comment("OpIdx") << MatchTable::IntValue(OpIdx) 1834 << MatchTable::LineBreak; 1835 } 1836 }; 1837 1838 /// Generates code to check an arbitrary C++ instruction predicate. 1839 class GenericInstructionPredicateMatcher : public InstructionPredicateMatcher { 1840 protected: 1841 TreePredicateFn Predicate; 1842 1843 public: 1844 GenericInstructionPredicateMatcher(unsigned InsnVarID, 1845 TreePredicateFn Predicate) 1846 : InstructionPredicateMatcher(IPM_GenericPredicate, InsnVarID), 1847 Predicate(Predicate) {} 1848 1849 static bool classof(const InstructionPredicateMatcher *P) { 1850 return P->getKind() == IPM_GenericPredicate; 1851 } 1852 bool isIdentical(const PredicateMatcher &B) const override { 1853 return InstructionPredicateMatcher::isIdentical(B) && 1854 Predicate == 1855 static_cast<const GenericInstructionPredicateMatcher &>(B) 1856 .Predicate; 1857 } 1858 void emitPredicateOpcodes(MatchTable &Table, 1859 RuleMatcher &Rule) const override { 1860 Table << MatchTable::Opcode("GIM_CheckCxxInsnPredicate") 1861 << MatchTable::Comment("MI") << MatchTable::IntValue(InsnVarID) 1862 << MatchTable::Comment("FnId") 1863 << MatchTable::NamedValue(getEnumNameForPredicate(Predicate)) 1864 << MatchTable::LineBreak; 1865 } 1866 }; 1867 1868 /// Generates code to check that a set of predicates and operands match for a 1869 /// particular instruction. 1870 /// 1871 /// Typical predicates include: 1872 /// * Has a specific opcode. 1873 /// * Has an nsw/nuw flag or doesn't. 1874 class InstructionMatcher final : public PredicateListMatcher<PredicateMatcher> { 1875 protected: 1876 typedef std::vector<std::unique_ptr<OperandMatcher>> OperandVec; 1877 1878 RuleMatcher &Rule; 1879 1880 /// The operands to match. All rendered operands must be present even if the 1881 /// condition is always true. 1882 OperandVec Operands; 1883 bool NumOperandsCheck = true; 1884 1885 std::string SymbolicName; 1886 unsigned InsnVarID; 1887 1888 public: 1889 InstructionMatcher(RuleMatcher &Rule, StringRef SymbolicName) 1890 : Rule(Rule), SymbolicName(SymbolicName) { 1891 // We create a new instruction matcher. 1892 // Get a new ID for that instruction. 1893 InsnVarID = Rule.implicitlyDefineInsnVar(*this); 1894 } 1895 1896 /// Construct a new instruction predicate and add it to the matcher. 1897 template <class Kind, class... Args> 1898 Optional<Kind *> addPredicate(Args &&... args) { 1899 Predicates.emplace_back( 1900 llvm::make_unique<Kind>(getInsnVarID(), std::forward<Args>(args)...)); 1901 return static_cast<Kind *>(Predicates.back().get()); 1902 } 1903 1904 RuleMatcher &getRuleMatcher() const { return Rule; } 1905 1906 unsigned getInsnVarID() const { return InsnVarID; } 1907 1908 /// Add an operand to the matcher. 1909 OperandMatcher &addOperand(unsigned OpIdx, const std::string &SymbolicName, 1910 unsigned AllocatedTemporariesBaseID) { 1911 Operands.emplace_back(new OperandMatcher(*this, OpIdx, SymbolicName, 1912 AllocatedTemporariesBaseID)); 1913 if (!SymbolicName.empty()) 1914 Rule.defineOperand(SymbolicName, *Operands.back()); 1915 1916 return *Operands.back(); 1917 } 1918 1919 OperandMatcher &getOperand(unsigned OpIdx) { 1920 auto I = std::find_if(Operands.begin(), Operands.end(), 1921 [&OpIdx](const std::unique_ptr<OperandMatcher> &X) { 1922 return X->getOpIdx() == OpIdx; 1923 }); 1924 if (I != Operands.end()) 1925 return **I; 1926 llvm_unreachable("Failed to lookup operand"); 1927 } 1928 1929 StringRef getSymbolicName() const { return SymbolicName; } 1930 unsigned getNumOperands() const { return Operands.size(); } 1931 OperandVec::iterator operands_begin() { return Operands.begin(); } 1932 OperandVec::iterator operands_end() { return Operands.end(); } 1933 iterator_range<OperandVec::iterator> operands() { 1934 return make_range(operands_begin(), operands_end()); 1935 } 1936 OperandVec::const_iterator operands_begin() const { return Operands.begin(); } 1937 OperandVec::const_iterator operands_end() const { return Operands.end(); } 1938 iterator_range<OperandVec::const_iterator> operands() const { 1939 return make_range(operands_begin(), operands_end()); 1940 } 1941 bool operands_empty() const { return Operands.empty(); } 1942 1943 void pop_front() { Operands.erase(Operands.begin()); } 1944 1945 void optimize(); 1946 1947 /// Emit MatchTable opcodes that test whether the instruction named in 1948 /// InsnVarName matches all the predicates and all the operands. 1949 void emitPredicateOpcodes(MatchTable &Table, RuleMatcher &Rule) { 1950 if (NumOperandsCheck) 1951 InstructionNumOperandsMatcher(InsnVarID, getNumOperands()) 1952 .emitPredicateOpcodes(Table, Rule); 1953 1954 emitPredicateListOpcodes(Table, Rule); 1955 1956 for (const auto &Operand : Operands) 1957 Operand->emitPredicateOpcodes(Table, Rule); 1958 } 1959 1960 /// Compare the priority of this object and B. 1961 /// 1962 /// Returns true if this object is more important than B. 1963 bool isHigherPriorityThan(InstructionMatcher &B) { 1964 // Instruction matchers involving more operands have higher priority. 1965 if (Operands.size() > B.Operands.size()) 1966 return true; 1967 if (Operands.size() < B.Operands.size()) 1968 return false; 1969 1970 for (auto &&P : zip(predicates(), B.predicates())) { 1971 auto L = static_cast<InstructionPredicateMatcher *>(std::get<0>(P).get()); 1972 auto R = static_cast<InstructionPredicateMatcher *>(std::get<1>(P).get()); 1973 if (L->isHigherPriorityThan(*R)) 1974 return true; 1975 if (R->isHigherPriorityThan(*L)) 1976 return false; 1977 } 1978 1979 for (const auto &Operand : zip(Operands, B.Operands)) { 1980 if (std::get<0>(Operand)->isHigherPriorityThan(*std::get<1>(Operand))) 1981 return true; 1982 if (std::get<1>(Operand)->isHigherPriorityThan(*std::get<0>(Operand))) 1983 return false; 1984 } 1985 1986 return false; 1987 }; 1988 1989 /// Report the maximum number of temporary operands needed by the instruction 1990 /// matcher. 1991 unsigned countRendererFns() { 1992 return std::accumulate( 1993 predicates().begin(), predicates().end(), 0, 1994 [](unsigned A, 1995 const std::unique_ptr<PredicateMatcher> &Predicate) { 1996 return A + Predicate->countRendererFns(); 1997 }) + 1998 std::accumulate( 1999 Operands.begin(), Operands.end(), 0, 2000 [](unsigned A, const std::unique_ptr<OperandMatcher> &Operand) { 2001 return A + Operand->countRendererFns(); 2002 }); 2003 } 2004 2005 InstructionOpcodeMatcher &getOpcodeMatcher() { 2006 for (auto &P : predicates()) 2007 if (auto *OpMatcher = dyn_cast<InstructionOpcodeMatcher>(P.get())) 2008 return *OpMatcher; 2009 llvm_unreachable("Didn't find an opcode matcher"); 2010 } 2011 2012 bool isConstantInstruction() { 2013 return getOpcodeMatcher().isConstantInstruction(); 2014 } 2015 2016 StringRef getOpcode() { return getOpcodeMatcher().getOpcode(); } 2017 }; 2018 2019 StringRef RuleMatcher::getOpcode() const { 2020 return Matchers.front()->getOpcode(); 2021 } 2022 2023 unsigned RuleMatcher::getNumOperands() const { 2024 return Matchers.front()->getNumOperands(); 2025 } 2026 2027 LLTCodeGen RuleMatcher::getFirstConditionAsRootType() { 2028 InstructionMatcher &InsnMatcher = *Matchers.front(); 2029 if (!InsnMatcher.predicates_empty()) 2030 if (const auto *TM = 2031 dyn_cast<LLTOperandMatcher>(&**InsnMatcher.predicates_begin())) 2032 if (TM->getInsnVarID() == 0 && TM->getOpIdx() == 0) 2033 return TM->getTy(); 2034 return {}; 2035 } 2036 2037 /// Generates code to check that the operand is a register defined by an 2038 /// instruction that matches the given instruction matcher. 2039 /// 2040 /// For example, the pattern: 2041 /// (set $dst, (G_MUL (G_ADD $src1, $src2), $src3)) 2042 /// would use an InstructionOperandMatcher for operand 1 of the G_MUL to match 2043 /// the: 2044 /// (G_ADD $src1, $src2) 2045 /// subpattern. 2046 class InstructionOperandMatcher : public OperandPredicateMatcher { 2047 protected: 2048 std::unique_ptr<InstructionMatcher> InsnMatcher; 2049 2050 public: 2051 InstructionOperandMatcher(unsigned InsnVarID, unsigned OpIdx, 2052 RuleMatcher &Rule, StringRef SymbolicName) 2053 : OperandPredicateMatcher(OPM_Instruction, InsnVarID, OpIdx), 2054 InsnMatcher(new InstructionMatcher(Rule, SymbolicName)) {} 2055 2056 static bool classof(const PredicateMatcher *P) { 2057 return P->getKind() == OPM_Instruction; 2058 } 2059 2060 InstructionMatcher &getInsnMatcher() const { return *InsnMatcher; } 2061 2062 void emitCaptureOpcodes(MatchTable &Table, RuleMatcher &Rule) const { 2063 const unsigned NewInsnVarID = InsnMatcher->getInsnVarID(); 2064 Table << MatchTable::Opcode("GIM_RecordInsn") 2065 << MatchTable::Comment("DefineMI") 2066 << MatchTable::IntValue(NewInsnVarID) << MatchTable::Comment("MI") 2067 << MatchTable::IntValue(getInsnVarID()) 2068 << MatchTable::Comment("OpIdx") << MatchTable::IntValue(getOpIdx()) 2069 << MatchTable::Comment("MIs[" + llvm::to_string(NewInsnVarID) + "]") 2070 << MatchTable::LineBreak; 2071 } 2072 2073 void emitPredicateOpcodes(MatchTable &Table, 2074 RuleMatcher &Rule) const override { 2075 emitCaptureOpcodes(Table, Rule); 2076 InsnMatcher->emitPredicateOpcodes(Table, Rule); 2077 } 2078 2079 bool isHigherPriorityThan(const OperandPredicateMatcher &B) const override { 2080 if (OperandPredicateMatcher::isHigherPriorityThan(B)) 2081 return true; 2082 if (B.OperandPredicateMatcher::isHigherPriorityThan(*this)) 2083 return false; 2084 2085 if (const InstructionOperandMatcher *BP = 2086 dyn_cast<InstructionOperandMatcher>(&B)) 2087 if (InsnMatcher->isHigherPriorityThan(*BP->InsnMatcher)) 2088 return true; 2089 return false; 2090 } 2091 }; 2092 2093 void InstructionMatcher::optimize() { 2094 SmallVector<std::unique_ptr<PredicateMatcher>, 8> Stash; 2095 const auto &OpcMatcher = getOpcodeMatcher(); 2096 2097 Stash.push_back(predicates_pop_front()); 2098 if (Stash.back().get() == &OpcMatcher) { 2099 if (NumOperandsCheck && OpcMatcher.getNumOperands() < getNumOperands()) 2100 Stash.emplace_back( 2101 new InstructionNumOperandsMatcher(InsnVarID, getNumOperands())); 2102 NumOperandsCheck = false; 2103 2104 for (auto &OM : Operands) 2105 for (auto &OP : OM->predicates()) 2106 if (isa<IntrinsicIDOperandMatcher>(OP)) { 2107 Stash.push_back(std::move(OP)); 2108 OM->eraseNullPredicates(); 2109 break; 2110 } 2111 } 2112 2113 if (InsnVarID > 0) { 2114 assert(!Operands.empty() && "Nested instruction is expected to def a vreg"); 2115 for (auto &OP : Operands[0]->predicates()) 2116 OP.reset(); 2117 Operands[0]->eraseNullPredicates(); 2118 } 2119 for (auto &OM : Operands) { 2120 for (auto &OP : OM->predicates()) 2121 if (isa<LLTOperandMatcher>(OP)) 2122 Stash.push_back(std::move(OP)); 2123 OM->eraseNullPredicates(); 2124 } 2125 while (!Stash.empty()) 2126 prependPredicate(Stash.pop_back_val()); 2127 } 2128 2129 //===- Actions ------------------------------------------------------------===// 2130 class OperandRenderer { 2131 public: 2132 enum RendererKind { 2133 OR_Copy, 2134 OR_CopyOrAddZeroReg, 2135 OR_CopySubReg, 2136 OR_CopyConstantAsImm, 2137 OR_CopyFConstantAsFPImm, 2138 OR_Imm, 2139 OR_Register, 2140 OR_TempRegister, 2141 OR_ComplexPattern, 2142 OR_Custom 2143 }; 2144 2145 protected: 2146 RendererKind Kind; 2147 2148 public: 2149 OperandRenderer(RendererKind Kind) : Kind(Kind) {} 2150 virtual ~OperandRenderer() {} 2151 2152 RendererKind getKind() const { return Kind; } 2153 2154 virtual void emitRenderOpcodes(MatchTable &Table, 2155 RuleMatcher &Rule) const = 0; 2156 }; 2157 2158 /// A CopyRenderer emits code to copy a single operand from an existing 2159 /// instruction to the one being built. 2160 class CopyRenderer : public OperandRenderer { 2161 protected: 2162 unsigned NewInsnID; 2163 /// The name of the operand. 2164 const StringRef SymbolicName; 2165 2166 public: 2167 CopyRenderer(unsigned NewInsnID, StringRef SymbolicName) 2168 : OperandRenderer(OR_Copy), NewInsnID(NewInsnID), 2169 SymbolicName(SymbolicName) { 2170 assert(!SymbolicName.empty() && "Cannot copy from an unspecified source"); 2171 } 2172 2173 static bool classof(const OperandRenderer *R) { 2174 return R->getKind() == OR_Copy; 2175 } 2176 2177 const StringRef getSymbolicName() const { return SymbolicName; } 2178 2179 void emitRenderOpcodes(MatchTable &Table, RuleMatcher &Rule) const override { 2180 const OperandMatcher &Operand = Rule.getOperandMatcher(SymbolicName); 2181 unsigned OldInsnVarID = Rule.getInsnVarID(Operand.getInstructionMatcher()); 2182 Table << MatchTable::Opcode("GIR_Copy") << MatchTable::Comment("NewInsnID") 2183 << MatchTable::IntValue(NewInsnID) << MatchTable::Comment("OldInsnID") 2184 << MatchTable::IntValue(OldInsnVarID) << MatchTable::Comment("OpIdx") 2185 << MatchTable::IntValue(Operand.getOpIdx()) 2186 << MatchTable::Comment(SymbolicName) << MatchTable::LineBreak; 2187 } 2188 }; 2189 2190 /// A CopyOrAddZeroRegRenderer emits code to copy a single operand from an 2191 /// existing instruction to the one being built. If the operand turns out to be 2192 /// a 'G_CONSTANT 0' then it replaces the operand with a zero register. 2193 class CopyOrAddZeroRegRenderer : public OperandRenderer { 2194 protected: 2195 unsigned NewInsnID; 2196 /// The name of the operand. 2197 const StringRef SymbolicName; 2198 const Record *ZeroRegisterDef; 2199 2200 public: 2201 CopyOrAddZeroRegRenderer(unsigned NewInsnID, 2202 StringRef SymbolicName, Record *ZeroRegisterDef) 2203 : OperandRenderer(OR_CopyOrAddZeroReg), NewInsnID(NewInsnID), 2204 SymbolicName(SymbolicName), ZeroRegisterDef(ZeroRegisterDef) { 2205 assert(!SymbolicName.empty() && "Cannot copy from an unspecified source"); 2206 } 2207 2208 static bool classof(const OperandRenderer *R) { 2209 return R->getKind() == OR_CopyOrAddZeroReg; 2210 } 2211 2212 const StringRef getSymbolicName() const { return SymbolicName; } 2213 2214 void emitRenderOpcodes(MatchTable &Table, RuleMatcher &Rule) const override { 2215 const OperandMatcher &Operand = Rule.getOperandMatcher(SymbolicName); 2216 unsigned OldInsnVarID = Rule.getInsnVarID(Operand.getInstructionMatcher()); 2217 Table << MatchTable::Opcode("GIR_CopyOrAddZeroReg") 2218 << MatchTable::Comment("NewInsnID") << MatchTable::IntValue(NewInsnID) 2219 << MatchTable::Comment("OldInsnID") 2220 << MatchTable::IntValue(OldInsnVarID) << MatchTable::Comment("OpIdx") 2221 << MatchTable::IntValue(Operand.getOpIdx()) 2222 << MatchTable::NamedValue( 2223 (ZeroRegisterDef->getValue("Namespace") 2224 ? ZeroRegisterDef->getValueAsString("Namespace") 2225 : ""), 2226 ZeroRegisterDef->getName()) 2227 << MatchTable::Comment(SymbolicName) << MatchTable::LineBreak; 2228 } 2229 }; 2230 2231 /// A CopyConstantAsImmRenderer emits code to render a G_CONSTANT instruction to 2232 /// an extended immediate operand. 2233 class CopyConstantAsImmRenderer : public OperandRenderer { 2234 protected: 2235 unsigned NewInsnID; 2236 /// The name of the operand. 2237 const std::string SymbolicName; 2238 bool Signed; 2239 2240 public: 2241 CopyConstantAsImmRenderer(unsigned NewInsnID, StringRef SymbolicName) 2242 : OperandRenderer(OR_CopyConstantAsImm), NewInsnID(NewInsnID), 2243 SymbolicName(SymbolicName), Signed(true) {} 2244 2245 static bool classof(const OperandRenderer *R) { 2246 return R->getKind() == OR_CopyConstantAsImm; 2247 } 2248 2249 const StringRef getSymbolicName() const { return SymbolicName; } 2250 2251 void emitRenderOpcodes(MatchTable &Table, RuleMatcher &Rule) const override { 2252 InstructionMatcher &InsnMatcher = Rule.getInstructionMatcher(SymbolicName); 2253 unsigned OldInsnVarID = Rule.getInsnVarID(InsnMatcher); 2254 Table << MatchTable::Opcode(Signed ? "GIR_CopyConstantAsSImm" 2255 : "GIR_CopyConstantAsUImm") 2256 << MatchTable::Comment("NewInsnID") << MatchTable::IntValue(NewInsnID) 2257 << MatchTable::Comment("OldInsnID") 2258 << MatchTable::IntValue(OldInsnVarID) 2259 << MatchTable::Comment(SymbolicName) << MatchTable::LineBreak; 2260 } 2261 }; 2262 2263 /// A CopyFConstantAsFPImmRenderer emits code to render a G_FCONSTANT 2264 /// instruction to an extended immediate operand. 2265 class CopyFConstantAsFPImmRenderer : public OperandRenderer { 2266 protected: 2267 unsigned NewInsnID; 2268 /// The name of the operand. 2269 const std::string SymbolicName; 2270 2271 public: 2272 CopyFConstantAsFPImmRenderer(unsigned NewInsnID, StringRef SymbolicName) 2273 : OperandRenderer(OR_CopyFConstantAsFPImm), NewInsnID(NewInsnID), 2274 SymbolicName(SymbolicName) {} 2275 2276 static bool classof(const OperandRenderer *R) { 2277 return R->getKind() == OR_CopyFConstantAsFPImm; 2278 } 2279 2280 const StringRef getSymbolicName() const { return SymbolicName; } 2281 2282 void emitRenderOpcodes(MatchTable &Table, RuleMatcher &Rule) const override { 2283 InstructionMatcher &InsnMatcher = Rule.getInstructionMatcher(SymbolicName); 2284 unsigned OldInsnVarID = Rule.getInsnVarID(InsnMatcher); 2285 Table << MatchTable::Opcode("GIR_CopyFConstantAsFPImm") 2286 << MatchTable::Comment("NewInsnID") << MatchTable::IntValue(NewInsnID) 2287 << MatchTable::Comment("OldInsnID") 2288 << MatchTable::IntValue(OldInsnVarID) 2289 << MatchTable::Comment(SymbolicName) << MatchTable::LineBreak; 2290 } 2291 }; 2292 2293 /// A CopySubRegRenderer emits code to copy a single register operand from an 2294 /// existing instruction to the one being built and indicate that only a 2295 /// subregister should be copied. 2296 class CopySubRegRenderer : public OperandRenderer { 2297 protected: 2298 unsigned NewInsnID; 2299 /// The name of the operand. 2300 const StringRef SymbolicName; 2301 /// The subregister to extract. 2302 const CodeGenSubRegIndex *SubReg; 2303 2304 public: 2305 CopySubRegRenderer(unsigned NewInsnID, StringRef SymbolicName, 2306 const CodeGenSubRegIndex *SubReg) 2307 : OperandRenderer(OR_CopySubReg), NewInsnID(NewInsnID), 2308 SymbolicName(SymbolicName), SubReg(SubReg) {} 2309 2310 static bool classof(const OperandRenderer *R) { 2311 return R->getKind() == OR_CopySubReg; 2312 } 2313 2314 const StringRef getSymbolicName() const { return SymbolicName; } 2315 2316 void emitRenderOpcodes(MatchTable &Table, RuleMatcher &Rule) const override { 2317 const OperandMatcher &Operand = Rule.getOperandMatcher(SymbolicName); 2318 unsigned OldInsnVarID = Rule.getInsnVarID(Operand.getInstructionMatcher()); 2319 Table << MatchTable::Opcode("GIR_CopySubReg") 2320 << MatchTable::Comment("NewInsnID") << MatchTable::IntValue(NewInsnID) 2321 << MatchTable::Comment("OldInsnID") 2322 << MatchTable::IntValue(OldInsnVarID) << MatchTable::Comment("OpIdx") 2323 << MatchTable::IntValue(Operand.getOpIdx()) 2324 << MatchTable::Comment("SubRegIdx") 2325 << MatchTable::IntValue(SubReg->EnumValue) 2326 << MatchTable::Comment(SymbolicName) << MatchTable::LineBreak; 2327 } 2328 }; 2329 2330 /// Adds a specific physical register to the instruction being built. 2331 /// This is typically useful for WZR/XZR on AArch64. 2332 class AddRegisterRenderer : public OperandRenderer { 2333 protected: 2334 unsigned InsnID; 2335 const Record *RegisterDef; 2336 2337 public: 2338 AddRegisterRenderer(unsigned InsnID, const Record *RegisterDef) 2339 : OperandRenderer(OR_Register), InsnID(InsnID), RegisterDef(RegisterDef) { 2340 } 2341 2342 static bool classof(const OperandRenderer *R) { 2343 return R->getKind() == OR_Register; 2344 } 2345 2346 void emitRenderOpcodes(MatchTable &Table, RuleMatcher &Rule) const override { 2347 Table << MatchTable::Opcode("GIR_AddRegister") 2348 << MatchTable::Comment("InsnID") << MatchTable::IntValue(InsnID) 2349 << MatchTable::NamedValue( 2350 (RegisterDef->getValue("Namespace") 2351 ? RegisterDef->getValueAsString("Namespace") 2352 : ""), 2353 RegisterDef->getName()) 2354 << MatchTable::LineBreak; 2355 } 2356 }; 2357 2358 /// Adds a specific temporary virtual register to the instruction being built. 2359 /// This is used to chain instructions together when emitting multiple 2360 /// instructions. 2361 class TempRegRenderer : public OperandRenderer { 2362 protected: 2363 unsigned InsnID; 2364 unsigned TempRegID; 2365 bool IsDef; 2366 2367 public: 2368 TempRegRenderer(unsigned InsnID, unsigned TempRegID, bool IsDef = false) 2369 : OperandRenderer(OR_Register), InsnID(InsnID), TempRegID(TempRegID), 2370 IsDef(IsDef) {} 2371 2372 static bool classof(const OperandRenderer *R) { 2373 return R->getKind() == OR_TempRegister; 2374 } 2375 2376 void emitRenderOpcodes(MatchTable &Table, RuleMatcher &Rule) const override { 2377 Table << MatchTable::Opcode("GIR_AddTempRegister") 2378 << MatchTable::Comment("InsnID") << MatchTable::IntValue(InsnID) 2379 << MatchTable::Comment("TempRegID") << MatchTable::IntValue(TempRegID) 2380 << MatchTable::Comment("TempRegFlags"); 2381 if (IsDef) 2382 Table << MatchTable::NamedValue("RegState::Define"); 2383 else 2384 Table << MatchTable::IntValue(0); 2385 Table << MatchTable::LineBreak; 2386 } 2387 }; 2388 2389 /// Adds a specific immediate to the instruction being built. 2390 class ImmRenderer : public OperandRenderer { 2391 protected: 2392 unsigned InsnID; 2393 int64_t Imm; 2394 2395 public: 2396 ImmRenderer(unsigned InsnID, int64_t Imm) 2397 : OperandRenderer(OR_Imm), InsnID(InsnID), Imm(Imm) {} 2398 2399 static bool classof(const OperandRenderer *R) { 2400 return R->getKind() == OR_Imm; 2401 } 2402 2403 void emitRenderOpcodes(MatchTable &Table, RuleMatcher &Rule) const override { 2404 Table << MatchTable::Opcode("GIR_AddImm") << MatchTable::Comment("InsnID") 2405 << MatchTable::IntValue(InsnID) << MatchTable::Comment("Imm") 2406 << MatchTable::IntValue(Imm) << MatchTable::LineBreak; 2407 } 2408 }; 2409 2410 /// Adds operands by calling a renderer function supplied by the ComplexPattern 2411 /// matcher function. 2412 class RenderComplexPatternOperand : public OperandRenderer { 2413 private: 2414 unsigned InsnID; 2415 const Record &TheDef; 2416 /// The name of the operand. 2417 const StringRef SymbolicName; 2418 /// The renderer number. This must be unique within a rule since it's used to 2419 /// identify a temporary variable to hold the renderer function. 2420 unsigned RendererID; 2421 /// When provided, this is the suboperand of the ComplexPattern operand to 2422 /// render. Otherwise all the suboperands will be rendered. 2423 Optional<unsigned> SubOperand; 2424 2425 unsigned getNumOperands() const { 2426 return TheDef.getValueAsDag("Operands")->getNumArgs(); 2427 } 2428 2429 public: 2430 RenderComplexPatternOperand(unsigned InsnID, const Record &TheDef, 2431 StringRef SymbolicName, unsigned RendererID, 2432 Optional<unsigned> SubOperand = None) 2433 : OperandRenderer(OR_ComplexPattern), InsnID(InsnID), TheDef(TheDef), 2434 SymbolicName(SymbolicName), RendererID(RendererID), 2435 SubOperand(SubOperand) {} 2436 2437 static bool classof(const OperandRenderer *R) { 2438 return R->getKind() == OR_ComplexPattern; 2439 } 2440 2441 void emitRenderOpcodes(MatchTable &Table, RuleMatcher &Rule) const override { 2442 Table << MatchTable::Opcode(SubOperand.hasValue() ? "GIR_ComplexSubOperandRenderer" 2443 : "GIR_ComplexRenderer") 2444 << MatchTable::Comment("InsnID") << MatchTable::IntValue(InsnID) 2445 << MatchTable::Comment("RendererID") 2446 << MatchTable::IntValue(RendererID); 2447 if (SubOperand.hasValue()) 2448 Table << MatchTable::Comment("SubOperand") 2449 << MatchTable::IntValue(SubOperand.getValue()); 2450 Table << MatchTable::Comment(SymbolicName) << MatchTable::LineBreak; 2451 } 2452 }; 2453 2454 class CustomRenderer : public OperandRenderer { 2455 protected: 2456 unsigned InsnID; 2457 const Record &Renderer; 2458 /// The name of the operand. 2459 const std::string SymbolicName; 2460 2461 public: 2462 CustomRenderer(unsigned InsnID, const Record &Renderer, 2463 StringRef SymbolicName) 2464 : OperandRenderer(OR_Custom), InsnID(InsnID), Renderer(Renderer), 2465 SymbolicName(SymbolicName) {} 2466 2467 static bool classof(const OperandRenderer *R) { 2468 return R->getKind() == OR_Custom; 2469 } 2470 2471 void emitRenderOpcodes(MatchTable &Table, RuleMatcher &Rule) const override { 2472 InstructionMatcher &InsnMatcher = Rule.getInstructionMatcher(SymbolicName); 2473 unsigned OldInsnVarID = Rule.getInsnVarID(InsnMatcher); 2474 Table << MatchTable::Opcode("GIR_CustomRenderer") 2475 << MatchTable::Comment("InsnID") << MatchTable::IntValue(InsnID) 2476 << MatchTable::Comment("OldInsnID") 2477 << MatchTable::IntValue(OldInsnVarID) 2478 << MatchTable::Comment("Renderer") 2479 << MatchTable::NamedValue( 2480 "GICR_" + Renderer.getValueAsString("RendererFn").str()) 2481 << MatchTable::Comment(SymbolicName) << MatchTable::LineBreak; 2482 } 2483 }; 2484 2485 /// An action taken when all Matcher predicates succeeded for a parent rule. 2486 /// 2487 /// Typical actions include: 2488 /// * Changing the opcode of an instruction. 2489 /// * Adding an operand to an instruction. 2490 class MatchAction { 2491 public: 2492 virtual ~MatchAction() {} 2493 2494 /// Emit the MatchTable opcodes to implement the action. 2495 virtual void emitActionOpcodes(MatchTable &Table, 2496 RuleMatcher &Rule) const = 0; 2497 }; 2498 2499 /// Generates a comment describing the matched rule being acted upon. 2500 class DebugCommentAction : public MatchAction { 2501 private: 2502 std::string S; 2503 2504 public: 2505 DebugCommentAction(StringRef S) : S(S) {} 2506 2507 void emitActionOpcodes(MatchTable &Table, RuleMatcher &Rule) const override { 2508 Table << MatchTable::Comment(S) << MatchTable::LineBreak; 2509 } 2510 }; 2511 2512 /// Generates code to build an instruction or mutate an existing instruction 2513 /// into the desired instruction when this is possible. 2514 class BuildMIAction : public MatchAction { 2515 private: 2516 unsigned InsnID; 2517 const CodeGenInstruction *I; 2518 InstructionMatcher *Matched; 2519 std::vector<std::unique_ptr<OperandRenderer>> OperandRenderers; 2520 2521 /// True if the instruction can be built solely by mutating the opcode. 2522 bool canMutate(RuleMatcher &Rule, const InstructionMatcher *Insn) const { 2523 if (!Insn) 2524 return false; 2525 2526 if (OperandRenderers.size() != Insn->getNumOperands()) 2527 return false; 2528 2529 for (const auto &Renderer : enumerate(OperandRenderers)) { 2530 if (const auto *Copy = dyn_cast<CopyRenderer>(&*Renderer.value())) { 2531 const OperandMatcher &OM = Rule.getOperandMatcher(Copy->getSymbolicName()); 2532 if (Insn != &OM.getInstructionMatcher() || 2533 OM.getOpIdx() != Renderer.index()) 2534 return false; 2535 } else 2536 return false; 2537 } 2538 2539 return true; 2540 } 2541 2542 public: 2543 BuildMIAction(unsigned InsnID, const CodeGenInstruction *I) 2544 : InsnID(InsnID), I(I), Matched(nullptr) {} 2545 2546 unsigned getInsnID() const { return InsnID; } 2547 const CodeGenInstruction *getCGI() const { return I; } 2548 2549 void chooseInsnToMutate(RuleMatcher &Rule) { 2550 for (auto *MutateCandidate : Rule.mutatable_insns()) { 2551 if (canMutate(Rule, MutateCandidate)) { 2552 // Take the first one we're offered that we're able to mutate. 2553 Rule.reserveInsnMatcherForMutation(MutateCandidate); 2554 Matched = MutateCandidate; 2555 return; 2556 } 2557 } 2558 } 2559 2560 template <class Kind, class... Args> 2561 Kind &addRenderer(Args&&... args) { 2562 OperandRenderers.emplace_back( 2563 llvm::make_unique<Kind>(InsnID, std::forward<Args>(args)...)); 2564 return *static_cast<Kind *>(OperandRenderers.back().get()); 2565 } 2566 2567 void emitActionOpcodes(MatchTable &Table, RuleMatcher &Rule) const override { 2568 if (Matched) { 2569 assert(canMutate(Rule, Matched) && 2570 "Arranged to mutate an insn that isn't mutatable"); 2571 2572 unsigned RecycleInsnID = Rule.getInsnVarID(*Matched); 2573 Table << MatchTable::Opcode("GIR_MutateOpcode") 2574 << MatchTable::Comment("InsnID") << MatchTable::IntValue(InsnID) 2575 << MatchTable::Comment("RecycleInsnID") 2576 << MatchTable::IntValue(RecycleInsnID) 2577 << MatchTable::Comment("Opcode") 2578 << MatchTable::NamedValue(I->Namespace, I->TheDef->getName()) 2579 << MatchTable::LineBreak; 2580 2581 if (!I->ImplicitDefs.empty() || !I->ImplicitUses.empty()) { 2582 for (auto Def : I->ImplicitDefs) { 2583 auto Namespace = Def->getValue("Namespace") 2584 ? Def->getValueAsString("Namespace") 2585 : ""; 2586 Table << MatchTable::Opcode("GIR_AddImplicitDef") 2587 << MatchTable::Comment("InsnID") << MatchTable::IntValue(InsnID) 2588 << MatchTable::NamedValue(Namespace, Def->getName()) 2589 << MatchTable::LineBreak; 2590 } 2591 for (auto Use : I->ImplicitUses) { 2592 auto Namespace = Use->getValue("Namespace") 2593 ? Use->getValueAsString("Namespace") 2594 : ""; 2595 Table << MatchTable::Opcode("GIR_AddImplicitUse") 2596 << MatchTable::Comment("InsnID") << MatchTable::IntValue(InsnID) 2597 << MatchTable::NamedValue(Namespace, Use->getName()) 2598 << MatchTable::LineBreak; 2599 } 2600 } 2601 return; 2602 } 2603 2604 // TODO: Simple permutation looks like it could be almost as common as 2605 // mutation due to commutative operations. 2606 2607 Table << MatchTable::Opcode("GIR_BuildMI") << MatchTable::Comment("InsnID") 2608 << MatchTable::IntValue(InsnID) << MatchTable::Comment("Opcode") 2609 << MatchTable::NamedValue(I->Namespace, I->TheDef->getName()) 2610 << MatchTable::LineBreak; 2611 for (const auto &Renderer : OperandRenderers) 2612 Renderer->emitRenderOpcodes(Table, Rule); 2613 2614 if (I->mayLoad || I->mayStore) { 2615 Table << MatchTable::Opcode("GIR_MergeMemOperands") 2616 << MatchTable::Comment("InsnID") << MatchTable::IntValue(InsnID) 2617 << MatchTable::Comment("MergeInsnID's"); 2618 // Emit the ID's for all the instructions that are matched by this rule. 2619 // TODO: Limit this to matched instructions that mayLoad/mayStore or have 2620 // some other means of having a memoperand. Also limit this to 2621 // emitted instructions that expect to have a memoperand too. For 2622 // example, (G_SEXT (G_LOAD x)) that results in separate load and 2623 // sign-extend instructions shouldn't put the memoperand on the 2624 // sign-extend since it has no effect there. 2625 std::vector<unsigned> MergeInsnIDs; 2626 for (const auto &IDMatcherPair : Rule.defined_insn_vars()) 2627 MergeInsnIDs.push_back(IDMatcherPair.second); 2628 llvm::sort(MergeInsnIDs); 2629 for (const auto &MergeInsnID : MergeInsnIDs) 2630 Table << MatchTable::IntValue(MergeInsnID); 2631 Table << MatchTable::NamedValue("GIU_MergeMemOperands_EndOfList") 2632 << MatchTable::LineBreak; 2633 } 2634 2635 // FIXME: This is a hack but it's sufficient for ISel. We'll need to do 2636 // better for combines. Particularly when there are multiple match 2637 // roots. 2638 if (InsnID == 0) 2639 Table << MatchTable::Opcode("GIR_EraseFromParent") 2640 << MatchTable::Comment("InsnID") << MatchTable::IntValue(InsnID) 2641 << MatchTable::LineBreak; 2642 } 2643 }; 2644 2645 /// Generates code to constrain the operands of an output instruction to the 2646 /// register classes specified by the definition of that instruction. 2647 class ConstrainOperandsToDefinitionAction : public MatchAction { 2648 unsigned InsnID; 2649 2650 public: 2651 ConstrainOperandsToDefinitionAction(unsigned InsnID) : InsnID(InsnID) {} 2652 2653 void emitActionOpcodes(MatchTable &Table, RuleMatcher &Rule) const override { 2654 Table << MatchTable::Opcode("GIR_ConstrainSelectedInstOperands") 2655 << MatchTable::Comment("InsnID") << MatchTable::IntValue(InsnID) 2656 << MatchTable::LineBreak; 2657 } 2658 }; 2659 2660 /// Generates code to constrain the specified operand of an output instruction 2661 /// to the specified register class. 2662 class ConstrainOperandToRegClassAction : public MatchAction { 2663 unsigned InsnID; 2664 unsigned OpIdx; 2665 const CodeGenRegisterClass &RC; 2666 2667 public: 2668 ConstrainOperandToRegClassAction(unsigned InsnID, unsigned OpIdx, 2669 const CodeGenRegisterClass &RC) 2670 : InsnID(InsnID), OpIdx(OpIdx), RC(RC) {} 2671 2672 void emitActionOpcodes(MatchTable &Table, RuleMatcher &Rule) const override { 2673 Table << MatchTable::Opcode("GIR_ConstrainOperandRC") 2674 << MatchTable::Comment("InsnID") << MatchTable::IntValue(InsnID) 2675 << MatchTable::Comment("Op") << MatchTable::IntValue(OpIdx) 2676 << MatchTable::Comment("RC " + RC.getName()) 2677 << MatchTable::IntValue(RC.EnumValue) << MatchTable::LineBreak; 2678 } 2679 }; 2680 2681 /// Generates code to create a temporary register which can be used to chain 2682 /// instructions together. 2683 class MakeTempRegisterAction : public MatchAction { 2684 private: 2685 LLTCodeGen Ty; 2686 unsigned TempRegID; 2687 2688 public: 2689 MakeTempRegisterAction(const LLTCodeGen &Ty, unsigned TempRegID) 2690 : Ty(Ty), TempRegID(TempRegID) {} 2691 2692 void emitActionOpcodes(MatchTable &Table, RuleMatcher &Rule) const override { 2693 Table << MatchTable::Opcode("GIR_MakeTempReg") 2694 << MatchTable::Comment("TempRegID") << MatchTable::IntValue(TempRegID) 2695 << MatchTable::Comment("TypeID") 2696 << MatchTable::NamedValue(Ty.getCxxEnumValue()) 2697 << MatchTable::LineBreak; 2698 } 2699 }; 2700 2701 InstructionMatcher &RuleMatcher::addInstructionMatcher(StringRef SymbolicName) { 2702 Matchers.emplace_back(new InstructionMatcher(*this, SymbolicName)); 2703 MutatableInsns.insert(Matchers.back().get()); 2704 return *Matchers.back(); 2705 } 2706 2707 void RuleMatcher::addRequiredFeature(Record *Feature) { 2708 RequiredFeatures.push_back(Feature); 2709 } 2710 2711 const std::vector<Record *> &RuleMatcher::getRequiredFeatures() const { 2712 return RequiredFeatures; 2713 } 2714 2715 // Emplaces an action of the specified Kind at the end of the action list. 2716 // 2717 // Returns a reference to the newly created action. 2718 // 2719 // Like std::vector::emplace_back(), may invalidate all iterators if the new 2720 // size exceeds the capacity. Otherwise, only invalidates the past-the-end 2721 // iterator. 2722 template <class Kind, class... Args> 2723 Kind &RuleMatcher::addAction(Args &&... args) { 2724 Actions.emplace_back(llvm::make_unique<Kind>(std::forward<Args>(args)...)); 2725 return *static_cast<Kind *>(Actions.back().get()); 2726 } 2727 2728 // Emplaces an action of the specified Kind before the given insertion point. 2729 // 2730 // Returns an iterator pointing at the newly created instruction. 2731 // 2732 // Like std::vector::insert(), may invalidate all iterators if the new size 2733 // exceeds the capacity. Otherwise, only invalidates the iterators from the 2734 // insertion point onwards. 2735 template <class Kind, class... Args> 2736 action_iterator RuleMatcher::insertAction(action_iterator InsertPt, 2737 Args &&... args) { 2738 return Actions.emplace(InsertPt, 2739 llvm::make_unique<Kind>(std::forward<Args>(args)...)); 2740 } 2741 2742 unsigned RuleMatcher::implicitlyDefineInsnVar(InstructionMatcher &Matcher) { 2743 unsigned NewInsnVarID = NextInsnVarID++; 2744 InsnVariableIDs[&Matcher] = NewInsnVarID; 2745 return NewInsnVarID; 2746 } 2747 2748 unsigned RuleMatcher::getInsnVarID(InstructionMatcher &InsnMatcher) const { 2749 const auto &I = InsnVariableIDs.find(&InsnMatcher); 2750 if (I != InsnVariableIDs.end()) 2751 return I->second; 2752 llvm_unreachable("Matched Insn was not captured in a local variable"); 2753 } 2754 2755 void RuleMatcher::defineOperand(StringRef SymbolicName, OperandMatcher &OM) { 2756 if (DefinedOperands.find(SymbolicName) == DefinedOperands.end()) { 2757 DefinedOperands[SymbolicName] = &OM; 2758 return; 2759 } 2760 2761 // If the operand is already defined, then we must ensure both references in 2762 // the matcher have the exact same node. 2763 OM.addPredicate<SameOperandMatcher>(OM.getSymbolicName()); 2764 } 2765 2766 InstructionMatcher & 2767 RuleMatcher::getInstructionMatcher(StringRef SymbolicName) const { 2768 for (const auto &I : InsnVariableIDs) 2769 if (I.first->getSymbolicName() == SymbolicName) 2770 return *I.first; 2771 llvm_unreachable( 2772 ("Failed to lookup instruction " + SymbolicName).str().c_str()); 2773 } 2774 2775 const OperandMatcher & 2776 RuleMatcher::getOperandMatcher(StringRef Name) const { 2777 const auto &I = DefinedOperands.find(Name); 2778 2779 if (I == DefinedOperands.end()) 2780 PrintFatalError(SrcLoc, "Operand " + Name + " was not declared in matcher"); 2781 2782 return *I->second; 2783 } 2784 2785 void RuleMatcher::emit(MatchTable &Table) { 2786 if (Matchers.empty()) 2787 llvm_unreachable("Unexpected empty matcher!"); 2788 2789 // The representation supports rules that require multiple roots such as: 2790 // %ptr(p0) = ... 2791 // %elt0(s32) = G_LOAD %ptr 2792 // %1(p0) = G_ADD %ptr, 4 2793 // %elt1(s32) = G_LOAD p0 %1 2794 // which could be usefully folded into: 2795 // %ptr(p0) = ... 2796 // %elt0(s32), %elt1(s32) = TGT_LOAD_PAIR %ptr 2797 // on some targets but we don't need to make use of that yet. 2798 assert(Matchers.size() == 1 && "Cannot handle multi-root matchers yet"); 2799 2800 unsigned LabelID = Table.allocateLabelID(); 2801 Table << MatchTable::Opcode("GIM_Try", +1) 2802 << MatchTable::Comment("On fail goto") 2803 << MatchTable::JumpTarget(LabelID) 2804 << MatchTable::Comment(("Rule ID " + Twine(RuleID) + " //").str()) 2805 << MatchTable::LineBreak; 2806 2807 if (!RequiredFeatures.empty()) { 2808 Table << MatchTable::Opcode("GIM_CheckFeatures") 2809 << MatchTable::NamedValue(getNameForFeatureBitset(RequiredFeatures)) 2810 << MatchTable::LineBreak; 2811 } 2812 2813 Matchers.front()->emitPredicateOpcodes(Table, *this); 2814 2815 // We must also check if it's safe to fold the matched instructions. 2816 if (InsnVariableIDs.size() >= 2) { 2817 // Invert the map to create stable ordering (by var names) 2818 SmallVector<unsigned, 2> InsnIDs; 2819 for (const auto &Pair : InsnVariableIDs) { 2820 // Skip the root node since it isn't moving anywhere. Everything else is 2821 // sinking to meet it. 2822 if (Pair.first == Matchers.front().get()) 2823 continue; 2824 2825 InsnIDs.push_back(Pair.second); 2826 } 2827 llvm::sort(InsnIDs); 2828 2829 for (const auto &InsnID : InsnIDs) { 2830 // Reject the difficult cases until we have a more accurate check. 2831 Table << MatchTable::Opcode("GIM_CheckIsSafeToFold") 2832 << MatchTable::Comment("InsnID") << MatchTable::IntValue(InsnID) 2833 << MatchTable::LineBreak; 2834 2835 // FIXME: Emit checks to determine it's _actually_ safe to fold and/or 2836 // account for unsafe cases. 2837 // 2838 // Example: 2839 // MI1--> %0 = ... 2840 // %1 = ... %0 2841 // MI0--> %2 = ... %0 2842 // It's not safe to erase MI1. We currently handle this by not 2843 // erasing %0 (even when it's dead). 2844 // 2845 // Example: 2846 // MI1--> %0 = load volatile @a 2847 // %1 = load volatile @a 2848 // MI0--> %2 = ... %0 2849 // It's not safe to sink %0's def past %1. We currently handle 2850 // this by rejecting all loads. 2851 // 2852 // Example: 2853 // MI1--> %0 = load @a 2854 // %1 = store @a 2855 // MI0--> %2 = ... %0 2856 // It's not safe to sink %0's def past %1. We currently handle 2857 // this by rejecting all loads. 2858 // 2859 // Example: 2860 // G_CONDBR %cond, @BB1 2861 // BB0: 2862 // MI1--> %0 = load @a 2863 // G_BR @BB1 2864 // BB1: 2865 // MI0--> %2 = ... %0 2866 // It's not always safe to sink %0 across control flow. In this 2867 // case it may introduce a memory fault. We currentl handle this 2868 // by rejecting all loads. 2869 } 2870 } 2871 2872 for (const auto &PM : EpilogueMatchers) 2873 PM->emitPredicateOpcodes(Table, *this); 2874 2875 for (const auto &MA : Actions) 2876 MA->emitActionOpcodes(Table, *this); 2877 2878 if (Table.isWithCoverage()) 2879 Table << MatchTable::Opcode("GIR_Coverage") << MatchTable::IntValue(RuleID) 2880 << MatchTable::LineBreak; 2881 else 2882 Table << MatchTable::Comment(("GIR_Coverage, " + Twine(RuleID) + ",").str()) 2883 << MatchTable::LineBreak; 2884 2885 Table << MatchTable::Opcode("GIR_Done", -1) << MatchTable::LineBreak 2886 << MatchTable::Label(LabelID); 2887 ++NumPatternEmitted; 2888 } 2889 2890 bool RuleMatcher::isHigherPriorityThan(const RuleMatcher &B) const { 2891 // Rules involving more match roots have higher priority. 2892 if (Matchers.size() > B.Matchers.size()) 2893 return true; 2894 if (Matchers.size() < B.Matchers.size()) 2895 return false; 2896 2897 for (const auto &Matcher : zip(Matchers, B.Matchers)) { 2898 if (std::get<0>(Matcher)->isHigherPriorityThan(*std::get<1>(Matcher))) 2899 return true; 2900 if (std::get<1>(Matcher)->isHigherPriorityThan(*std::get<0>(Matcher))) 2901 return false; 2902 } 2903 2904 return false; 2905 } 2906 2907 unsigned RuleMatcher::countRendererFns() const { 2908 return std::accumulate( 2909 Matchers.begin(), Matchers.end(), 0, 2910 [](unsigned A, const std::unique_ptr<InstructionMatcher> &Matcher) { 2911 return A + Matcher->countRendererFns(); 2912 }); 2913 } 2914 2915 bool OperandPredicateMatcher::isHigherPriorityThan( 2916 const OperandPredicateMatcher &B) const { 2917 // Generally speaking, an instruction is more important than an Int or a 2918 // LiteralInt because it can cover more nodes but theres an exception to 2919 // this. G_CONSTANT's are less important than either of those two because they 2920 // are more permissive. 2921 2922 const InstructionOperandMatcher *AOM = 2923 dyn_cast<InstructionOperandMatcher>(this); 2924 const InstructionOperandMatcher *BOM = 2925 dyn_cast<InstructionOperandMatcher>(&B); 2926 bool AIsConstantInsn = AOM && AOM->getInsnMatcher().isConstantInstruction(); 2927 bool BIsConstantInsn = BOM && BOM->getInsnMatcher().isConstantInstruction(); 2928 2929 if (AOM && BOM) { 2930 // The relative priorities between a G_CONSTANT and any other instruction 2931 // don't actually matter but this code is needed to ensure a strict weak 2932 // ordering. This is particularly important on Windows where the rules will 2933 // be incorrectly sorted without it. 2934 if (AIsConstantInsn != BIsConstantInsn) 2935 return AIsConstantInsn < BIsConstantInsn; 2936 return false; 2937 } 2938 2939 if (AOM && AIsConstantInsn && (B.Kind == OPM_Int || B.Kind == OPM_LiteralInt)) 2940 return false; 2941 if (BOM && BIsConstantInsn && (Kind == OPM_Int || Kind == OPM_LiteralInt)) 2942 return true; 2943 2944 return Kind < B.Kind; 2945 } 2946 2947 void SameOperandMatcher::emitPredicateOpcodes(MatchTable &Table, 2948 RuleMatcher &Rule) const { 2949 const OperandMatcher &OtherOM = Rule.getOperandMatcher(MatchingName); 2950 unsigned OtherInsnVarID = Rule.getInsnVarID(OtherOM.getInstructionMatcher()); 2951 assert(OtherInsnVarID == OtherOM.getInstructionMatcher().getInsnVarID()); 2952 2953 Table << MatchTable::Opcode("GIM_CheckIsSameOperand") 2954 << MatchTable::Comment("MI") << MatchTable::IntValue(InsnVarID) 2955 << MatchTable::Comment("OpIdx") << MatchTable::IntValue(OpIdx) 2956 << MatchTable::Comment("OtherMI") 2957 << MatchTable::IntValue(OtherInsnVarID) 2958 << MatchTable::Comment("OtherOpIdx") 2959 << MatchTable::IntValue(OtherOM.getOpIdx()) 2960 << MatchTable::LineBreak; 2961 } 2962 2963 //===- GlobalISelEmitter class --------------------------------------------===// 2964 2965 class GlobalISelEmitter { 2966 public: 2967 explicit GlobalISelEmitter(RecordKeeper &RK); 2968 void run(raw_ostream &OS); 2969 2970 private: 2971 const RecordKeeper &RK; 2972 const CodeGenDAGPatterns CGP; 2973 const CodeGenTarget &Target; 2974 CodeGenRegBank CGRegs; 2975 2976 /// Keep track of the equivalence between SDNodes and Instruction by mapping 2977 /// SDNodes to the GINodeEquiv mapping. We need to map to the GINodeEquiv to 2978 /// check for attributes on the relation such as CheckMMOIsNonAtomic. 2979 /// This is defined using 'GINodeEquiv' in the target description. 2980 DenseMap<Record *, Record *> NodeEquivs; 2981 2982 /// Keep track of the equivalence between ComplexPattern's and 2983 /// GIComplexOperandMatcher. Map entries are specified by subclassing 2984 /// GIComplexPatternEquiv. 2985 DenseMap<const Record *, const Record *> ComplexPatternEquivs; 2986 2987 /// Keep track of the equivalence between SDNodeXForm's and 2988 /// GICustomOperandRenderer. Map entries are specified by subclassing 2989 /// GISDNodeXFormEquiv. 2990 DenseMap<const Record *, const Record *> SDNodeXFormEquivs; 2991 2992 /// Keep track of Scores of PatternsToMatch similar to how the DAG does. 2993 /// This adds compatibility for RuleMatchers to use this for ordering rules. 2994 DenseMap<uint64_t, int> RuleMatcherScores; 2995 2996 // Map of predicates to their subtarget features. 2997 SubtargetFeatureInfoMap SubtargetFeatures; 2998 2999 // Rule coverage information. 3000 Optional<CodeGenCoverage> RuleCoverage; 3001 3002 void gatherOpcodeValues(); 3003 void gatherTypeIDValues(); 3004 void gatherNodeEquivs(); 3005 3006 Record *findNodeEquiv(Record *N) const; 3007 const CodeGenInstruction *getEquivNode(Record &Equiv, 3008 const TreePatternNode *N) const; 3009 3010 Error importRulePredicates(RuleMatcher &M, ArrayRef<Predicate> Predicates); 3011 Expected<InstructionMatcher &> 3012 createAndImportSelDAGMatcher(RuleMatcher &Rule, 3013 InstructionMatcher &InsnMatcher, 3014 const TreePatternNode *Src, unsigned &TempOpIdx); 3015 Error importComplexPatternOperandMatcher(OperandMatcher &OM, Record *R, 3016 unsigned &TempOpIdx) const; 3017 Error importChildMatcher(RuleMatcher &Rule, InstructionMatcher &InsnMatcher, 3018 const TreePatternNode *SrcChild, 3019 bool OperandIsAPointer, unsigned OpIdx, 3020 unsigned &TempOpIdx); 3021 3022 Expected<BuildMIAction &> 3023 createAndImportInstructionRenderer(RuleMatcher &M, 3024 const TreePatternNode *Dst); 3025 Expected<action_iterator> createAndImportSubInstructionRenderer( 3026 action_iterator InsertPt, RuleMatcher &M, const TreePatternNode *Dst, 3027 unsigned TempReg); 3028 Expected<action_iterator> 3029 createInstructionRenderer(action_iterator InsertPt, RuleMatcher &M, 3030 const TreePatternNode *Dst); 3031 void importExplicitDefRenderers(BuildMIAction &DstMIBuilder); 3032 Expected<action_iterator> 3033 importExplicitUseRenderers(action_iterator InsertPt, RuleMatcher &M, 3034 BuildMIAction &DstMIBuilder, 3035 const llvm::TreePatternNode *Dst); 3036 Expected<action_iterator> 3037 importExplicitUseRenderer(action_iterator InsertPt, RuleMatcher &Rule, 3038 BuildMIAction &DstMIBuilder, 3039 TreePatternNode *DstChild); 3040 Error importDefaultOperandRenderers(BuildMIAction &DstMIBuilder, 3041 DagInit *DefaultOps) const; 3042 Error 3043 importImplicitDefRenderers(BuildMIAction &DstMIBuilder, 3044 const std::vector<Record *> &ImplicitDefs) const; 3045 3046 void emitCxxPredicateFns(raw_ostream &OS, StringRef CodeFieldName, 3047 StringRef TypeIdentifier, StringRef ArgType, 3048 StringRef ArgName, StringRef AdditionalDeclarations, 3049 std::function<bool(const Record *R)> Filter); 3050 void emitImmPredicateFns(raw_ostream &OS, StringRef TypeIdentifier, 3051 StringRef ArgType, 3052 std::function<bool(const Record *R)> Filter); 3053 void emitMIPredicateFns(raw_ostream &OS); 3054 3055 /// Analyze pattern \p P, returning a matcher for it if possible. 3056 /// Otherwise, return an Error explaining why we don't support it. 3057 Expected<RuleMatcher> runOnPattern(const PatternToMatch &P); 3058 3059 void declareSubtargetFeature(Record *Predicate); 3060 3061 MatchTable buildMatchTable(MutableArrayRef<RuleMatcher> Rules, bool Optimize, 3062 bool WithCoverage); 3063 3064 public: 3065 /// Takes a sequence of \p Rules and group them based on the predicates 3066 /// they share. \p MatcherStorage is used as a memory container 3067 /// for the group that are created as part of this process. 3068 /// 3069 /// What this optimization does looks like if GroupT = GroupMatcher: 3070 /// Output without optimization: 3071 /// \verbatim 3072 /// # R1 3073 /// # predicate A 3074 /// # predicate B 3075 /// ... 3076 /// # R2 3077 /// # predicate A // <-- effectively this is going to be checked twice. 3078 /// // Once in R1 and once in R2. 3079 /// # predicate C 3080 /// \endverbatim 3081 /// Output with optimization: 3082 /// \verbatim 3083 /// # Group1_2 3084 /// # predicate A // <-- Check is now shared. 3085 /// # R1 3086 /// # predicate B 3087 /// # R2 3088 /// # predicate C 3089 /// \endverbatim 3090 template <class GroupT> 3091 static std::vector<Matcher *> optimizeRules( 3092 ArrayRef<Matcher *> Rules, 3093 std::vector<std::unique_ptr<Matcher>> &MatcherStorage); 3094 }; 3095 3096 void GlobalISelEmitter::gatherOpcodeValues() { 3097 InstructionOpcodeMatcher::initOpcodeValuesMap(Target); 3098 } 3099 3100 void GlobalISelEmitter::gatherTypeIDValues() { 3101 LLTOperandMatcher::initTypeIDValuesMap(); 3102 } 3103 3104 void GlobalISelEmitter::gatherNodeEquivs() { 3105 assert(NodeEquivs.empty()); 3106 for (Record *Equiv : RK.getAllDerivedDefinitions("GINodeEquiv")) 3107 NodeEquivs[Equiv->getValueAsDef("Node")] = Equiv; 3108 3109 assert(ComplexPatternEquivs.empty()); 3110 for (Record *Equiv : RK.getAllDerivedDefinitions("GIComplexPatternEquiv")) { 3111 Record *SelDAGEquiv = Equiv->getValueAsDef("SelDAGEquivalent"); 3112 if (!SelDAGEquiv) 3113 continue; 3114 ComplexPatternEquivs[SelDAGEquiv] = Equiv; 3115 } 3116 3117 assert(SDNodeXFormEquivs.empty()); 3118 for (Record *Equiv : RK.getAllDerivedDefinitions("GISDNodeXFormEquiv")) { 3119 Record *SelDAGEquiv = Equiv->getValueAsDef("SelDAGEquivalent"); 3120 if (!SelDAGEquiv) 3121 continue; 3122 SDNodeXFormEquivs[SelDAGEquiv] = Equiv; 3123 } 3124 } 3125 3126 Record *GlobalISelEmitter::findNodeEquiv(Record *N) const { 3127 return NodeEquivs.lookup(N); 3128 } 3129 3130 const CodeGenInstruction * 3131 GlobalISelEmitter::getEquivNode(Record &Equiv, const TreePatternNode *N) const { 3132 for (const TreePredicateCall &Call : N->getPredicateCalls()) { 3133 const TreePredicateFn &Predicate = Call.Fn; 3134 if (!Equiv.isValueUnset("IfSignExtend") && Predicate.isLoad() && 3135 Predicate.isSignExtLoad()) 3136 return &Target.getInstruction(Equiv.getValueAsDef("IfSignExtend")); 3137 if (!Equiv.isValueUnset("IfZeroExtend") && Predicate.isLoad() && 3138 Predicate.isZeroExtLoad()) 3139 return &Target.getInstruction(Equiv.getValueAsDef("IfZeroExtend")); 3140 } 3141 return &Target.getInstruction(Equiv.getValueAsDef("I")); 3142 } 3143 3144 GlobalISelEmitter::GlobalISelEmitter(RecordKeeper &RK) 3145 : RK(RK), CGP(RK), Target(CGP.getTargetInfo()), 3146 CGRegs(RK, Target.getHwModes()) {} 3147 3148 //===- Emitter ------------------------------------------------------------===// 3149 3150 Error 3151 GlobalISelEmitter::importRulePredicates(RuleMatcher &M, 3152 ArrayRef<Predicate> Predicates) { 3153 for (const Predicate &P : Predicates) { 3154 if (!P.Def) 3155 continue; 3156 declareSubtargetFeature(P.Def); 3157 M.addRequiredFeature(P.Def); 3158 } 3159 3160 return Error::success(); 3161 } 3162 3163 Expected<InstructionMatcher &> GlobalISelEmitter::createAndImportSelDAGMatcher( 3164 RuleMatcher &Rule, InstructionMatcher &InsnMatcher, 3165 const TreePatternNode *Src, unsigned &TempOpIdx) { 3166 Record *SrcGIEquivOrNull = nullptr; 3167 const CodeGenInstruction *SrcGIOrNull = nullptr; 3168 3169 // Start with the defined operands (i.e., the results of the root operator). 3170 if (Src->getExtTypes().size() > 1) 3171 return failedImport("Src pattern has multiple results"); 3172 3173 if (Src->isLeaf()) { 3174 Init *SrcInit = Src->getLeafValue(); 3175 if (isa<IntInit>(SrcInit)) { 3176 InsnMatcher.addPredicate<InstructionOpcodeMatcher>( 3177 &Target.getInstruction(RK.getDef("G_CONSTANT"))); 3178 } else 3179 return failedImport( 3180 "Unable to deduce gMIR opcode to handle Src (which is a leaf)"); 3181 } else { 3182 SrcGIEquivOrNull = findNodeEquiv(Src->getOperator()); 3183 if (!SrcGIEquivOrNull) 3184 return failedImport("Pattern operator lacks an equivalent Instruction" + 3185 explainOperator(Src->getOperator())); 3186 SrcGIOrNull = getEquivNode(*SrcGIEquivOrNull, Src); 3187 3188 // The operators look good: match the opcode 3189 InsnMatcher.addPredicate<InstructionOpcodeMatcher>(SrcGIOrNull); 3190 } 3191 3192 unsigned OpIdx = 0; 3193 for (const TypeSetByHwMode &VTy : Src->getExtTypes()) { 3194 // Results don't have a name unless they are the root node. The caller will 3195 // set the name if appropriate. 3196 OperandMatcher &OM = InsnMatcher.addOperand(OpIdx++, "", TempOpIdx); 3197 if (auto Error = OM.addTypeCheckPredicate(VTy, false /* OperandIsAPointer */)) 3198 return failedImport(toString(std::move(Error)) + 3199 " for result of Src pattern operator"); 3200 } 3201 3202 for (const TreePredicateCall &Call : Src->getPredicateCalls()) { 3203 const TreePredicateFn &Predicate = Call.Fn; 3204 if (Predicate.isAlwaysTrue()) 3205 continue; 3206 3207 if (Predicate.isImmediatePattern()) { 3208 InsnMatcher.addPredicate<InstructionImmPredicateMatcher>(Predicate); 3209 continue; 3210 } 3211 3212 // G_LOAD is used for both non-extending and any-extending loads. 3213 if (Predicate.isLoad() && Predicate.isNonExtLoad()) { 3214 InsnMatcher.addPredicate<MemoryVsLLTSizePredicateMatcher>( 3215 0, MemoryVsLLTSizePredicateMatcher::EqualTo, 0); 3216 continue; 3217 } 3218 if (Predicate.isLoad() && Predicate.isAnyExtLoad()) { 3219 InsnMatcher.addPredicate<MemoryVsLLTSizePredicateMatcher>( 3220 0, MemoryVsLLTSizePredicateMatcher::LessThan, 0); 3221 continue; 3222 } 3223 3224 // No check required. We already did it by swapping the opcode. 3225 if (!SrcGIEquivOrNull->isValueUnset("IfSignExtend") && 3226 Predicate.isSignExtLoad()) 3227 continue; 3228 3229 // No check required. We already did it by swapping the opcode. 3230 if (!SrcGIEquivOrNull->isValueUnset("IfZeroExtend") && 3231 Predicate.isZeroExtLoad()) 3232 continue; 3233 3234 // No check required. G_STORE by itself is a non-extending store. 3235 if (Predicate.isNonTruncStore()) 3236 continue; 3237 3238 if (Predicate.isLoad() || Predicate.isStore() || Predicate.isAtomic()) { 3239 if (Predicate.getMemoryVT() != nullptr) { 3240 Optional<LLTCodeGen> MemTyOrNone = 3241 MVTToLLT(getValueType(Predicate.getMemoryVT())); 3242 3243 if (!MemTyOrNone) 3244 return failedImport("MemVT could not be converted to LLT"); 3245 3246 // MMO's work in bytes so we must take care of unusual types like i1 3247 // don't round down. 3248 unsigned MemSizeInBits = 3249 llvm::alignTo(MemTyOrNone->get().getSizeInBits(), 8); 3250 3251 InsnMatcher.addPredicate<MemorySizePredicateMatcher>( 3252 0, MemSizeInBits / 8); 3253 continue; 3254 } 3255 } 3256 3257 if (Predicate.isLoad() || Predicate.isStore()) { 3258 // No check required. A G_LOAD/G_STORE is an unindexed load. 3259 if (Predicate.isUnindexed()) 3260 continue; 3261 } 3262 3263 if (Predicate.isAtomic()) { 3264 if (Predicate.isAtomicOrderingMonotonic()) { 3265 InsnMatcher.addPredicate<AtomicOrderingMMOPredicateMatcher>( 3266 "Monotonic"); 3267 continue; 3268 } 3269 if (Predicate.isAtomicOrderingAcquire()) { 3270 InsnMatcher.addPredicate<AtomicOrderingMMOPredicateMatcher>("Acquire"); 3271 continue; 3272 } 3273 if (Predicate.isAtomicOrderingRelease()) { 3274 InsnMatcher.addPredicate<AtomicOrderingMMOPredicateMatcher>("Release"); 3275 continue; 3276 } 3277 if (Predicate.isAtomicOrderingAcquireRelease()) { 3278 InsnMatcher.addPredicate<AtomicOrderingMMOPredicateMatcher>( 3279 "AcquireRelease"); 3280 continue; 3281 } 3282 if (Predicate.isAtomicOrderingSequentiallyConsistent()) { 3283 InsnMatcher.addPredicate<AtomicOrderingMMOPredicateMatcher>( 3284 "SequentiallyConsistent"); 3285 continue; 3286 } 3287 3288 if (Predicate.isAtomicOrderingAcquireOrStronger()) { 3289 InsnMatcher.addPredicate<AtomicOrderingMMOPredicateMatcher>( 3290 "Acquire", AtomicOrderingMMOPredicateMatcher::AO_OrStronger); 3291 continue; 3292 } 3293 if (Predicate.isAtomicOrderingWeakerThanAcquire()) { 3294 InsnMatcher.addPredicate<AtomicOrderingMMOPredicateMatcher>( 3295 "Acquire", AtomicOrderingMMOPredicateMatcher::AO_WeakerThan); 3296 continue; 3297 } 3298 3299 if (Predicate.isAtomicOrderingReleaseOrStronger()) { 3300 InsnMatcher.addPredicate<AtomicOrderingMMOPredicateMatcher>( 3301 "Release", AtomicOrderingMMOPredicateMatcher::AO_OrStronger); 3302 continue; 3303 } 3304 if (Predicate.isAtomicOrderingWeakerThanRelease()) { 3305 InsnMatcher.addPredicate<AtomicOrderingMMOPredicateMatcher>( 3306 "Release", AtomicOrderingMMOPredicateMatcher::AO_WeakerThan); 3307 continue; 3308 } 3309 } 3310 3311 if (Predicate.hasGISelPredicateCode()) { 3312 InsnMatcher.addPredicate<GenericInstructionPredicateMatcher>(Predicate); 3313 continue; 3314 } 3315 3316 return failedImport("Src pattern child has predicate (" + 3317 explainPredicates(Src) + ")"); 3318 } 3319 if (SrcGIEquivOrNull && SrcGIEquivOrNull->getValueAsBit("CheckMMOIsNonAtomic")) 3320 InsnMatcher.addPredicate<AtomicOrderingMMOPredicateMatcher>("NotAtomic"); 3321 3322 if (Src->isLeaf()) { 3323 Init *SrcInit = Src->getLeafValue(); 3324 if (IntInit *SrcIntInit = dyn_cast<IntInit>(SrcInit)) { 3325 OperandMatcher &OM = 3326 InsnMatcher.addOperand(OpIdx++, Src->getName(), TempOpIdx); 3327 OM.addPredicate<LiteralIntOperandMatcher>(SrcIntInit->getValue()); 3328 } else 3329 return failedImport( 3330 "Unable to deduce gMIR opcode to handle Src (which is a leaf)"); 3331 } else { 3332 assert(SrcGIOrNull && 3333 "Expected to have already found an equivalent Instruction"); 3334 if (SrcGIOrNull->TheDef->getName() == "G_CONSTANT" || 3335 SrcGIOrNull->TheDef->getName() == "G_FCONSTANT") { 3336 // imm/fpimm still have operands but we don't need to do anything with it 3337 // here since we don't support ImmLeaf predicates yet. However, we still 3338 // need to note the hidden operand to get GIM_CheckNumOperands correct. 3339 InsnMatcher.addOperand(OpIdx++, "", TempOpIdx); 3340 return InsnMatcher; 3341 } 3342 3343 // Match the used operands (i.e. the children of the operator). 3344 for (unsigned i = 0, e = Src->getNumChildren(); i != e; ++i) { 3345 TreePatternNode *SrcChild = Src->getChild(i); 3346 3347 // SelectionDAG allows pointers to be represented with iN since it doesn't 3348 // distinguish between pointers and integers but they are different types in GlobalISel. 3349 // Coerce integers to pointers to address space 0 if the context indicates a pointer. 3350 bool OperandIsAPointer = SrcGIOrNull->isOperandAPointer(i); 3351 3352 // For G_INTRINSIC/G_INTRINSIC_W_SIDE_EFFECTS, the operand immediately 3353 // following the defs is an intrinsic ID. 3354 if ((SrcGIOrNull->TheDef->getName() == "G_INTRINSIC" || 3355 SrcGIOrNull->TheDef->getName() == "G_INTRINSIC_W_SIDE_EFFECTS") && 3356 i == 0) { 3357 if (const CodeGenIntrinsic *II = Src->getIntrinsicInfo(CGP)) { 3358 OperandMatcher &OM = 3359 InsnMatcher.addOperand(OpIdx++, SrcChild->getName(), TempOpIdx); 3360 OM.addPredicate<IntrinsicIDOperandMatcher>(II); 3361 continue; 3362 } 3363 3364 return failedImport("Expected IntInit containing instrinsic ID)"); 3365 } 3366 3367 if (auto Error = 3368 importChildMatcher(Rule, InsnMatcher, SrcChild, OperandIsAPointer, 3369 OpIdx++, TempOpIdx)) 3370 return std::move(Error); 3371 } 3372 } 3373 3374 return InsnMatcher; 3375 } 3376 3377 Error GlobalISelEmitter::importComplexPatternOperandMatcher( 3378 OperandMatcher &OM, Record *R, unsigned &TempOpIdx) const { 3379 const auto &ComplexPattern = ComplexPatternEquivs.find(R); 3380 if (ComplexPattern == ComplexPatternEquivs.end()) 3381 return failedImport("SelectionDAG ComplexPattern (" + R->getName() + 3382 ") not mapped to GlobalISel"); 3383 3384 OM.addPredicate<ComplexPatternOperandMatcher>(OM, *ComplexPattern->second); 3385 TempOpIdx++; 3386 return Error::success(); 3387 } 3388 3389 Error GlobalISelEmitter::importChildMatcher(RuleMatcher &Rule, 3390 InstructionMatcher &InsnMatcher, 3391 const TreePatternNode *SrcChild, 3392 bool OperandIsAPointer, 3393 unsigned OpIdx, 3394 unsigned &TempOpIdx) { 3395 OperandMatcher &OM = 3396 InsnMatcher.addOperand(OpIdx, SrcChild->getName(), TempOpIdx); 3397 if (OM.isSameAsAnotherOperand()) 3398 return Error::success(); 3399 3400 ArrayRef<TypeSetByHwMode> ChildTypes = SrcChild->getExtTypes(); 3401 if (ChildTypes.size() != 1) 3402 return failedImport("Src pattern child has multiple results"); 3403 3404 // Check MBB's before the type check since they are not a known type. 3405 if (!SrcChild->isLeaf()) { 3406 if (SrcChild->getOperator()->isSubClassOf("SDNode")) { 3407 auto &ChildSDNI = CGP.getSDNodeInfo(SrcChild->getOperator()); 3408 if (ChildSDNI.getSDClassName() == "BasicBlockSDNode") { 3409 OM.addPredicate<MBBOperandMatcher>(); 3410 return Error::success(); 3411 } 3412 } 3413 } 3414 3415 if (auto Error = 3416 OM.addTypeCheckPredicate(ChildTypes.front(), OperandIsAPointer)) 3417 return failedImport(toString(std::move(Error)) + " for Src operand (" + 3418 to_string(*SrcChild) + ")"); 3419 3420 // Check for nested instructions. 3421 if (!SrcChild->isLeaf()) { 3422 if (SrcChild->getOperator()->isSubClassOf("ComplexPattern")) { 3423 // When a ComplexPattern is used as an operator, it should do the same 3424 // thing as when used as a leaf. However, the children of the operator 3425 // name the sub-operands that make up the complex operand and we must 3426 // prepare to reference them in the renderer too. 3427 unsigned RendererID = TempOpIdx; 3428 if (auto Error = importComplexPatternOperandMatcher( 3429 OM, SrcChild->getOperator(), TempOpIdx)) 3430 return Error; 3431 3432 for (unsigned i = 0, e = SrcChild->getNumChildren(); i != e; ++i) { 3433 auto *SubOperand = SrcChild->getChild(i); 3434 if (!SubOperand->getName().empty()) { 3435 if (auto Error = Rule.defineComplexSubOperand(SubOperand->getName(), 3436 SrcChild->getOperator(), 3437 RendererID, i)) 3438 return Error; 3439 } 3440 } 3441 3442 return Error::success(); 3443 } 3444 3445 auto MaybeInsnOperand = OM.addPredicate<InstructionOperandMatcher>( 3446 InsnMatcher.getRuleMatcher(), SrcChild->getName()); 3447 if (!MaybeInsnOperand.hasValue()) { 3448 // This isn't strictly true. If the user were to provide exactly the same 3449 // matchers as the original operand then we could allow it. However, it's 3450 // simpler to not permit the redundant specification. 3451 return failedImport("Nested instruction cannot be the same as another operand"); 3452 } 3453 3454 // Map the node to a gMIR instruction. 3455 InstructionOperandMatcher &InsnOperand = **MaybeInsnOperand; 3456 auto InsnMatcherOrError = createAndImportSelDAGMatcher( 3457 Rule, InsnOperand.getInsnMatcher(), SrcChild, TempOpIdx); 3458 if (auto Error = InsnMatcherOrError.takeError()) 3459 return Error; 3460 3461 return Error::success(); 3462 } 3463 3464 if (SrcChild->hasAnyPredicate()) 3465 return failedImport("Src pattern child has unsupported predicate"); 3466 3467 // Check for constant immediates. 3468 if (auto *ChildInt = dyn_cast<IntInit>(SrcChild->getLeafValue())) { 3469 OM.addPredicate<ConstantIntOperandMatcher>(ChildInt->getValue()); 3470 return Error::success(); 3471 } 3472 3473 // Check for def's like register classes or ComplexPattern's. 3474 if (auto *ChildDefInit = dyn_cast<DefInit>(SrcChild->getLeafValue())) { 3475 auto *ChildRec = ChildDefInit->getDef(); 3476 3477 // Check for register classes. 3478 if (ChildRec->isSubClassOf("RegisterClass") || 3479 ChildRec->isSubClassOf("RegisterOperand")) { 3480 OM.addPredicate<RegisterBankOperandMatcher>( 3481 Target.getRegisterClass(getInitValueAsRegClass(ChildDefInit))); 3482 return Error::success(); 3483 } 3484 3485 // Check for ValueType. 3486 if (ChildRec->isSubClassOf("ValueType")) { 3487 // We already added a type check as standard practice so this doesn't need 3488 // to do anything. 3489 return Error::success(); 3490 } 3491 3492 // Check for ComplexPattern's. 3493 if (ChildRec->isSubClassOf("ComplexPattern")) 3494 return importComplexPatternOperandMatcher(OM, ChildRec, TempOpIdx); 3495 3496 if (ChildRec->isSubClassOf("ImmLeaf")) { 3497 return failedImport( 3498 "Src pattern child def is an unsupported tablegen class (ImmLeaf)"); 3499 } 3500 3501 return failedImport( 3502 "Src pattern child def is an unsupported tablegen class"); 3503 } 3504 3505 return failedImport("Src pattern child is an unsupported kind"); 3506 } 3507 3508 Expected<action_iterator> GlobalISelEmitter::importExplicitUseRenderer( 3509 action_iterator InsertPt, RuleMatcher &Rule, BuildMIAction &DstMIBuilder, 3510 TreePatternNode *DstChild) { 3511 3512 const auto &SubOperand = Rule.getComplexSubOperand(DstChild->getName()); 3513 if (SubOperand.hasValue()) { 3514 DstMIBuilder.addRenderer<RenderComplexPatternOperand>( 3515 *std::get<0>(*SubOperand), DstChild->getName(), 3516 std::get<1>(*SubOperand), std::get<2>(*SubOperand)); 3517 return InsertPt; 3518 } 3519 3520 if (!DstChild->isLeaf()) { 3521 3522 if (DstChild->getOperator()->isSubClassOf("SDNodeXForm")) { 3523 auto Child = DstChild->getChild(0); 3524 auto I = SDNodeXFormEquivs.find(DstChild->getOperator()); 3525 if (I != SDNodeXFormEquivs.end()) { 3526 DstMIBuilder.addRenderer<CustomRenderer>(*I->second, Child->getName()); 3527 return InsertPt; 3528 } 3529 return failedImport("SDNodeXForm " + Child->getName() + 3530 " has no custom renderer"); 3531 } 3532 3533 // We accept 'bb' here. It's an operator because BasicBlockSDNode isn't 3534 // inline, but in MI it's just another operand. 3535 if (DstChild->getOperator()->isSubClassOf("SDNode")) { 3536 auto &ChildSDNI = CGP.getSDNodeInfo(DstChild->getOperator()); 3537 if (ChildSDNI.getSDClassName() == "BasicBlockSDNode") { 3538 DstMIBuilder.addRenderer<CopyRenderer>(DstChild->getName()); 3539 return InsertPt; 3540 } 3541 } 3542 3543 // Similarly, imm is an operator in TreePatternNode's view but must be 3544 // rendered as operands. 3545 // FIXME: The target should be able to choose sign-extended when appropriate 3546 // (e.g. on Mips). 3547 if (DstChild->getOperator()->getName() == "imm") { 3548 DstMIBuilder.addRenderer<CopyConstantAsImmRenderer>(DstChild->getName()); 3549 return InsertPt; 3550 } else if (DstChild->getOperator()->getName() == "fpimm") { 3551 DstMIBuilder.addRenderer<CopyFConstantAsFPImmRenderer>( 3552 DstChild->getName()); 3553 return InsertPt; 3554 } 3555 3556 if (DstChild->getOperator()->isSubClassOf("Instruction")) { 3557 ArrayRef<TypeSetByHwMode> ChildTypes = DstChild->getExtTypes(); 3558 if (ChildTypes.size() != 1) 3559 return failedImport("Dst pattern child has multiple results"); 3560 3561 Optional<LLTCodeGen> OpTyOrNone = None; 3562 if (ChildTypes.front().isMachineValueType()) 3563 OpTyOrNone = 3564 MVTToLLT(ChildTypes.front().getMachineValueType().SimpleTy); 3565 if (!OpTyOrNone) 3566 return failedImport("Dst operand has an unsupported type"); 3567 3568 unsigned TempRegID = Rule.allocateTempRegID(); 3569 InsertPt = Rule.insertAction<MakeTempRegisterAction>( 3570 InsertPt, OpTyOrNone.getValue(), TempRegID); 3571 DstMIBuilder.addRenderer<TempRegRenderer>(TempRegID); 3572 3573 auto InsertPtOrError = createAndImportSubInstructionRenderer( 3574 ++InsertPt, Rule, DstChild, TempRegID); 3575 if (auto Error = InsertPtOrError.takeError()) 3576 return std::move(Error); 3577 return InsertPtOrError.get(); 3578 } 3579 3580 return failedImport("Dst pattern child isn't a leaf node or an MBB" + llvm::to_string(*DstChild)); 3581 } 3582 3583 // It could be a specific immediate in which case we should just check for 3584 // that immediate. 3585 if (const IntInit *ChildIntInit = 3586 dyn_cast<IntInit>(DstChild->getLeafValue())) { 3587 DstMIBuilder.addRenderer<ImmRenderer>(ChildIntInit->getValue()); 3588 return InsertPt; 3589 } 3590 3591 // Otherwise, we're looking for a bog-standard RegisterClass operand. 3592 if (auto *ChildDefInit = dyn_cast<DefInit>(DstChild->getLeafValue())) { 3593 auto *ChildRec = ChildDefInit->getDef(); 3594 3595 ArrayRef<TypeSetByHwMode> ChildTypes = DstChild->getExtTypes(); 3596 if (ChildTypes.size() != 1) 3597 return failedImport("Dst pattern child has multiple results"); 3598 3599 Optional<LLTCodeGen> OpTyOrNone = None; 3600 if (ChildTypes.front().isMachineValueType()) 3601 OpTyOrNone = MVTToLLT(ChildTypes.front().getMachineValueType().SimpleTy); 3602 if (!OpTyOrNone) 3603 return failedImport("Dst operand has an unsupported type"); 3604 3605 if (ChildRec->isSubClassOf("Register")) { 3606 DstMIBuilder.addRenderer<AddRegisterRenderer>(ChildRec); 3607 return InsertPt; 3608 } 3609 3610 if (ChildRec->isSubClassOf("RegisterClass") || 3611 ChildRec->isSubClassOf("RegisterOperand") || 3612 ChildRec->isSubClassOf("ValueType")) { 3613 if (ChildRec->isSubClassOf("RegisterOperand") && 3614 !ChildRec->isValueUnset("GIZeroRegister")) { 3615 DstMIBuilder.addRenderer<CopyOrAddZeroRegRenderer>( 3616 DstChild->getName(), ChildRec->getValueAsDef("GIZeroRegister")); 3617 return InsertPt; 3618 } 3619 3620 DstMIBuilder.addRenderer<CopyRenderer>(DstChild->getName()); 3621 return InsertPt; 3622 } 3623 3624 if (ChildRec->isSubClassOf("ComplexPattern")) { 3625 const auto &ComplexPattern = ComplexPatternEquivs.find(ChildRec); 3626 if (ComplexPattern == ComplexPatternEquivs.end()) 3627 return failedImport( 3628 "SelectionDAG ComplexPattern not mapped to GlobalISel"); 3629 3630 const OperandMatcher &OM = Rule.getOperandMatcher(DstChild->getName()); 3631 DstMIBuilder.addRenderer<RenderComplexPatternOperand>( 3632 *ComplexPattern->second, DstChild->getName(), 3633 OM.getAllocatedTemporariesBaseID()); 3634 return InsertPt; 3635 } 3636 3637 return failedImport( 3638 "Dst pattern child def is an unsupported tablegen class"); 3639 } 3640 3641 return failedImport("Dst pattern child is an unsupported kind"); 3642 } 3643 3644 Expected<BuildMIAction &> GlobalISelEmitter::createAndImportInstructionRenderer( 3645 RuleMatcher &M, const TreePatternNode *Dst) { 3646 auto InsertPtOrError = createInstructionRenderer(M.actions_end(), M, Dst); 3647 if (auto Error = InsertPtOrError.takeError()) 3648 return std::move(Error); 3649 3650 action_iterator InsertPt = InsertPtOrError.get(); 3651 BuildMIAction &DstMIBuilder = *static_cast<BuildMIAction *>(InsertPt->get()); 3652 3653 importExplicitDefRenderers(DstMIBuilder); 3654 3655 if (auto Error = importExplicitUseRenderers(InsertPt, M, DstMIBuilder, Dst) 3656 .takeError()) 3657 return std::move(Error); 3658 3659 return DstMIBuilder; 3660 } 3661 3662 Expected<action_iterator> 3663 GlobalISelEmitter::createAndImportSubInstructionRenderer( 3664 const action_iterator InsertPt, RuleMatcher &M, const TreePatternNode *Dst, 3665 unsigned TempRegID) { 3666 auto InsertPtOrError = createInstructionRenderer(InsertPt, M, Dst); 3667 3668 // TODO: Assert there's exactly one result. 3669 3670 if (auto Error = InsertPtOrError.takeError()) 3671 return std::move(Error); 3672 3673 BuildMIAction &DstMIBuilder = 3674 *static_cast<BuildMIAction *>(InsertPtOrError.get()->get()); 3675 3676 // Assign the result to TempReg. 3677 DstMIBuilder.addRenderer<TempRegRenderer>(TempRegID, true); 3678 3679 InsertPtOrError = 3680 importExplicitUseRenderers(InsertPtOrError.get(), M, DstMIBuilder, Dst); 3681 if (auto Error = InsertPtOrError.takeError()) 3682 return std::move(Error); 3683 3684 M.insertAction<ConstrainOperandsToDefinitionAction>(InsertPt, 3685 DstMIBuilder.getInsnID()); 3686 return InsertPtOrError.get(); 3687 } 3688 3689 Expected<action_iterator> GlobalISelEmitter::createInstructionRenderer( 3690 action_iterator InsertPt, RuleMatcher &M, const TreePatternNode *Dst) { 3691 Record *DstOp = Dst->getOperator(); 3692 if (!DstOp->isSubClassOf("Instruction")) { 3693 if (DstOp->isSubClassOf("ValueType")) 3694 return failedImport( 3695 "Pattern operator isn't an instruction (it's a ValueType)"); 3696 return failedImport("Pattern operator isn't an instruction"); 3697 } 3698 CodeGenInstruction *DstI = &Target.getInstruction(DstOp); 3699 3700 // COPY_TO_REGCLASS is just a copy with a ConstrainOperandToRegClassAction 3701 // attached. Similarly for EXTRACT_SUBREG except that's a subregister copy. 3702 if (DstI->TheDef->getName() == "COPY_TO_REGCLASS") 3703 DstI = &Target.getInstruction(RK.getDef("COPY")); 3704 else if (DstI->TheDef->getName() == "EXTRACT_SUBREG") 3705 DstI = &Target.getInstruction(RK.getDef("COPY")); 3706 else if (DstI->TheDef->getName() == "REG_SEQUENCE") 3707 return failedImport("Unable to emit REG_SEQUENCE"); 3708 3709 return M.insertAction<BuildMIAction>(InsertPt, M.allocateOutputInsnID(), 3710 DstI); 3711 } 3712 3713 void GlobalISelEmitter::importExplicitDefRenderers( 3714 BuildMIAction &DstMIBuilder) { 3715 const CodeGenInstruction *DstI = DstMIBuilder.getCGI(); 3716 for (unsigned I = 0; I < DstI->Operands.NumDefs; ++I) { 3717 const CGIOperandList::OperandInfo &DstIOperand = DstI->Operands[I]; 3718 DstMIBuilder.addRenderer<CopyRenderer>(DstIOperand.Name); 3719 } 3720 } 3721 3722 Expected<action_iterator> GlobalISelEmitter::importExplicitUseRenderers( 3723 action_iterator InsertPt, RuleMatcher &M, BuildMIAction &DstMIBuilder, 3724 const llvm::TreePatternNode *Dst) { 3725 const CodeGenInstruction *DstI = DstMIBuilder.getCGI(); 3726 CodeGenInstruction *OrigDstI = &Target.getInstruction(Dst->getOperator()); 3727 3728 // EXTRACT_SUBREG needs to use a subregister COPY. 3729 if (OrigDstI->TheDef->getName() == "EXTRACT_SUBREG") { 3730 if (!Dst->getChild(0)->isLeaf()) 3731 return failedImport("EXTRACT_SUBREG child #1 is not a leaf"); 3732 3733 if (DefInit *SubRegInit = 3734 dyn_cast<DefInit>(Dst->getChild(1)->getLeafValue())) { 3735 Record *RCDef = getInitValueAsRegClass(Dst->getChild(0)->getLeafValue()); 3736 if (!RCDef) 3737 return failedImport("EXTRACT_SUBREG child #0 could not " 3738 "be coerced to a register class"); 3739 3740 CodeGenRegisterClass *RC = CGRegs.getRegClass(RCDef); 3741 CodeGenSubRegIndex *SubIdx = CGRegs.getSubRegIdx(SubRegInit->getDef()); 3742 3743 const auto &SrcRCDstRCPair = 3744 RC->getMatchingSubClassWithSubRegs(CGRegs, SubIdx); 3745 if (SrcRCDstRCPair.hasValue()) { 3746 assert(SrcRCDstRCPair->second && "Couldn't find a matching subclass"); 3747 if (SrcRCDstRCPair->first != RC) 3748 return failedImport("EXTRACT_SUBREG requires an additional COPY"); 3749 } 3750 3751 DstMIBuilder.addRenderer<CopySubRegRenderer>(Dst->getChild(0)->getName(), 3752 SubIdx); 3753 return InsertPt; 3754 } 3755 3756 return failedImport("EXTRACT_SUBREG child #1 is not a subreg index"); 3757 } 3758 3759 // Render the explicit uses. 3760 unsigned DstINumUses = OrigDstI->Operands.size() - OrigDstI->Operands.NumDefs; 3761 unsigned ExpectedDstINumUses = Dst->getNumChildren(); 3762 if (OrigDstI->TheDef->getName() == "COPY_TO_REGCLASS") { 3763 DstINumUses--; // Ignore the class constraint. 3764 ExpectedDstINumUses--; 3765 } 3766 3767 unsigned Child = 0; 3768 unsigned NumDefaultOps = 0; 3769 for (unsigned I = 0; I != DstINumUses; ++I) { 3770 const CGIOperandList::OperandInfo &DstIOperand = 3771 DstI->Operands[DstI->Operands.NumDefs + I]; 3772 3773 // If the operand has default values, introduce them now. 3774 // FIXME: Until we have a decent test case that dictates we should do 3775 // otherwise, we're going to assume that operands with default values cannot 3776 // be specified in the patterns. Therefore, adding them will not cause us to 3777 // end up with too many rendered operands. 3778 if (DstIOperand.Rec->isSubClassOf("OperandWithDefaultOps")) { 3779 DagInit *DefaultOps = DstIOperand.Rec->getValueAsDag("DefaultOps"); 3780 if (auto Error = importDefaultOperandRenderers(DstMIBuilder, DefaultOps)) 3781 return std::move(Error); 3782 ++NumDefaultOps; 3783 continue; 3784 } 3785 3786 auto InsertPtOrError = importExplicitUseRenderer(InsertPt, M, DstMIBuilder, 3787 Dst->getChild(Child)); 3788 if (auto Error = InsertPtOrError.takeError()) 3789 return std::move(Error); 3790 InsertPt = InsertPtOrError.get(); 3791 ++Child; 3792 } 3793 3794 if (NumDefaultOps + ExpectedDstINumUses != DstINumUses) 3795 return failedImport("Expected " + llvm::to_string(DstINumUses) + 3796 " used operands but found " + 3797 llvm::to_string(ExpectedDstINumUses) + 3798 " explicit ones and " + llvm::to_string(NumDefaultOps) + 3799 " default ones"); 3800 3801 return InsertPt; 3802 } 3803 3804 Error GlobalISelEmitter::importDefaultOperandRenderers( 3805 BuildMIAction &DstMIBuilder, DagInit *DefaultOps) const { 3806 for (const auto *DefaultOp : DefaultOps->getArgs()) { 3807 // Look through ValueType operators. 3808 if (const DagInit *DefaultDagOp = dyn_cast<DagInit>(DefaultOp)) { 3809 if (const DefInit *DefaultDagOperator = 3810 dyn_cast<DefInit>(DefaultDagOp->getOperator())) { 3811 if (DefaultDagOperator->getDef()->isSubClassOf("ValueType")) 3812 DefaultOp = DefaultDagOp->getArg(0); 3813 } 3814 } 3815 3816 if (const DefInit *DefaultDefOp = dyn_cast<DefInit>(DefaultOp)) { 3817 DstMIBuilder.addRenderer<AddRegisterRenderer>(DefaultDefOp->getDef()); 3818 continue; 3819 } 3820 3821 if (const IntInit *DefaultIntOp = dyn_cast<IntInit>(DefaultOp)) { 3822 DstMIBuilder.addRenderer<ImmRenderer>(DefaultIntOp->getValue()); 3823 continue; 3824 } 3825 3826 return failedImport("Could not add default op"); 3827 } 3828 3829 return Error::success(); 3830 } 3831 3832 Error GlobalISelEmitter::importImplicitDefRenderers( 3833 BuildMIAction &DstMIBuilder, 3834 const std::vector<Record *> &ImplicitDefs) const { 3835 if (!ImplicitDefs.empty()) 3836 return failedImport("Pattern defines a physical register"); 3837 return Error::success(); 3838 } 3839 3840 Expected<RuleMatcher> GlobalISelEmitter::runOnPattern(const PatternToMatch &P) { 3841 // Keep track of the matchers and actions to emit. 3842 int Score = P.getPatternComplexity(CGP); 3843 RuleMatcher M(P.getSrcRecord()->getLoc()); 3844 RuleMatcherScores[M.getRuleID()] = Score; 3845 M.addAction<DebugCommentAction>(llvm::to_string(*P.getSrcPattern()) + 3846 " => " + 3847 llvm::to_string(*P.getDstPattern())); 3848 3849 if (auto Error = importRulePredicates(M, P.getPredicates())) 3850 return std::move(Error); 3851 3852 // Next, analyze the pattern operators. 3853 TreePatternNode *Src = P.getSrcPattern(); 3854 TreePatternNode *Dst = P.getDstPattern(); 3855 3856 // If the root of either pattern isn't a simple operator, ignore it. 3857 if (auto Err = isTrivialOperatorNode(Dst)) 3858 return failedImport("Dst pattern root isn't a trivial operator (" + 3859 toString(std::move(Err)) + ")"); 3860 if (auto Err = isTrivialOperatorNode(Src)) 3861 return failedImport("Src pattern root isn't a trivial operator (" + 3862 toString(std::move(Err)) + ")"); 3863 3864 // The different predicates and matchers created during 3865 // addInstructionMatcher use the RuleMatcher M to set up their 3866 // instruction ID (InsnVarID) that are going to be used when 3867 // M is going to be emitted. 3868 // However, the code doing the emission still relies on the IDs 3869 // returned during that process by the RuleMatcher when issuing 3870 // the recordInsn opcodes. 3871 // Because of that: 3872 // 1. The order in which we created the predicates 3873 // and such must be the same as the order in which we emit them, 3874 // and 3875 // 2. We need to reset the generation of the IDs in M somewhere between 3876 // addInstructionMatcher and emit 3877 // 3878 // FIXME: Long term, we don't want to have to rely on this implicit 3879 // naming being the same. One possible solution would be to have 3880 // explicit operator for operation capture and reference those. 3881 // The plus side is that it would expose opportunities to share 3882 // the capture accross rules. The downside is that it would 3883 // introduce a dependency between predicates (captures must happen 3884 // before their first use.) 3885 InstructionMatcher &InsnMatcherTemp = M.addInstructionMatcher(Src->getName()); 3886 unsigned TempOpIdx = 0; 3887 auto InsnMatcherOrError = 3888 createAndImportSelDAGMatcher(M, InsnMatcherTemp, Src, TempOpIdx); 3889 if (auto Error = InsnMatcherOrError.takeError()) 3890 return std::move(Error); 3891 InstructionMatcher &InsnMatcher = InsnMatcherOrError.get(); 3892 3893 if (Dst->isLeaf()) { 3894 Record *RCDef = getInitValueAsRegClass(Dst->getLeafValue()); 3895 3896 const CodeGenRegisterClass &RC = Target.getRegisterClass(RCDef); 3897 if (RCDef) { 3898 // We need to replace the def and all its uses with the specified 3899 // operand. However, we must also insert COPY's wherever needed. 3900 // For now, emit a copy and let the register allocator clean up. 3901 auto &DstI = Target.getInstruction(RK.getDef("COPY")); 3902 const auto &DstIOperand = DstI.Operands[0]; 3903 3904 OperandMatcher &OM0 = InsnMatcher.getOperand(0); 3905 OM0.setSymbolicName(DstIOperand.Name); 3906 M.defineOperand(OM0.getSymbolicName(), OM0); 3907 OM0.addPredicate<RegisterBankOperandMatcher>(RC); 3908 3909 auto &DstMIBuilder = 3910 M.addAction<BuildMIAction>(M.allocateOutputInsnID(), &DstI); 3911 DstMIBuilder.addRenderer<CopyRenderer>(DstIOperand.Name); 3912 DstMIBuilder.addRenderer<CopyRenderer>(Dst->getName()); 3913 M.addAction<ConstrainOperandToRegClassAction>(0, 0, RC); 3914 3915 // We're done with this pattern! It's eligible for GISel emission; return 3916 // it. 3917 ++NumPatternImported; 3918 return std::move(M); 3919 } 3920 3921 return failedImport("Dst pattern root isn't a known leaf"); 3922 } 3923 3924 // Start with the defined operands (i.e., the results of the root operator). 3925 Record *DstOp = Dst->getOperator(); 3926 if (!DstOp->isSubClassOf("Instruction")) 3927 return failedImport("Pattern operator isn't an instruction"); 3928 3929 auto &DstI = Target.getInstruction(DstOp); 3930 if (DstI.Operands.NumDefs != Src->getExtTypes().size()) 3931 return failedImport("Src pattern results and dst MI defs are different (" + 3932 to_string(Src->getExtTypes().size()) + " def(s) vs " + 3933 to_string(DstI.Operands.NumDefs) + " def(s))"); 3934 3935 // The root of the match also has constraints on the register bank so that it 3936 // matches the result instruction. 3937 unsigned OpIdx = 0; 3938 for (const TypeSetByHwMode &VTy : Src->getExtTypes()) { 3939 (void)VTy; 3940 3941 const auto &DstIOperand = DstI.Operands[OpIdx]; 3942 Record *DstIOpRec = DstIOperand.Rec; 3943 if (DstI.TheDef->getName() == "COPY_TO_REGCLASS") { 3944 DstIOpRec = getInitValueAsRegClass(Dst->getChild(1)->getLeafValue()); 3945 3946 if (DstIOpRec == nullptr) 3947 return failedImport( 3948 "COPY_TO_REGCLASS operand #1 isn't a register class"); 3949 } else if (DstI.TheDef->getName() == "EXTRACT_SUBREG") { 3950 if (!Dst->getChild(0)->isLeaf()) 3951 return failedImport("EXTRACT_SUBREG operand #0 isn't a leaf"); 3952 3953 // We can assume that a subregister is in the same bank as it's super 3954 // register. 3955 DstIOpRec = getInitValueAsRegClass(Dst->getChild(0)->getLeafValue()); 3956 3957 if (DstIOpRec == nullptr) 3958 return failedImport( 3959 "EXTRACT_SUBREG operand #0 isn't a register class"); 3960 } else if (DstIOpRec->isSubClassOf("RegisterOperand")) 3961 DstIOpRec = DstIOpRec->getValueAsDef("RegClass"); 3962 else if (!DstIOpRec->isSubClassOf("RegisterClass")) 3963 return failedImport("Dst MI def isn't a register class" + 3964 to_string(*Dst)); 3965 3966 OperandMatcher &OM = InsnMatcher.getOperand(OpIdx); 3967 OM.setSymbolicName(DstIOperand.Name); 3968 M.defineOperand(OM.getSymbolicName(), OM); 3969 OM.addPredicate<RegisterBankOperandMatcher>( 3970 Target.getRegisterClass(DstIOpRec)); 3971 ++OpIdx; 3972 } 3973 3974 auto DstMIBuilderOrError = createAndImportInstructionRenderer(M, Dst); 3975 if (auto Error = DstMIBuilderOrError.takeError()) 3976 return std::move(Error); 3977 BuildMIAction &DstMIBuilder = DstMIBuilderOrError.get(); 3978 3979 // Render the implicit defs. 3980 // These are only added to the root of the result. 3981 if (auto Error = importImplicitDefRenderers(DstMIBuilder, P.getDstRegs())) 3982 return std::move(Error); 3983 3984 DstMIBuilder.chooseInsnToMutate(M); 3985 3986 // Constrain the registers to classes. This is normally derived from the 3987 // emitted instruction but a few instructions require special handling. 3988 if (DstI.TheDef->getName() == "COPY_TO_REGCLASS") { 3989 // COPY_TO_REGCLASS does not provide operand constraints itself but the 3990 // result is constrained to the class given by the second child. 3991 Record *DstIOpRec = 3992 getInitValueAsRegClass(Dst->getChild(1)->getLeafValue()); 3993 3994 if (DstIOpRec == nullptr) 3995 return failedImport("COPY_TO_REGCLASS operand #1 isn't a register class"); 3996 3997 M.addAction<ConstrainOperandToRegClassAction>( 3998 0, 0, Target.getRegisterClass(DstIOpRec)); 3999 4000 // We're done with this pattern! It's eligible for GISel emission; return 4001 // it. 4002 ++NumPatternImported; 4003 return std::move(M); 4004 } 4005 4006 if (DstI.TheDef->getName() == "EXTRACT_SUBREG") { 4007 // EXTRACT_SUBREG selects into a subregister COPY but unlike most 4008 // instructions, the result register class is controlled by the 4009 // subregisters of the operand. As a result, we must constrain the result 4010 // class rather than check that it's already the right one. 4011 if (!Dst->getChild(0)->isLeaf()) 4012 return failedImport("EXTRACT_SUBREG child #1 is not a leaf"); 4013 4014 DefInit *SubRegInit = dyn_cast<DefInit>(Dst->getChild(1)->getLeafValue()); 4015 if (!SubRegInit) 4016 return failedImport("EXTRACT_SUBREG child #1 is not a subreg index"); 4017 4018 // Constrain the result to the same register bank as the operand. 4019 Record *DstIOpRec = 4020 getInitValueAsRegClass(Dst->getChild(0)->getLeafValue()); 4021 4022 if (DstIOpRec == nullptr) 4023 return failedImport("EXTRACT_SUBREG operand #1 isn't a register class"); 4024 4025 CodeGenSubRegIndex *SubIdx = CGRegs.getSubRegIdx(SubRegInit->getDef()); 4026 CodeGenRegisterClass *SrcRC = CGRegs.getRegClass(DstIOpRec); 4027 4028 // It would be nice to leave this constraint implicit but we're required 4029 // to pick a register class so constrain the result to a register class 4030 // that can hold the correct MVT. 4031 // 4032 // FIXME: This may introduce an extra copy if the chosen class doesn't 4033 // actually contain the subregisters. 4034 assert(Src->getExtTypes().size() == 1 && 4035 "Expected Src of EXTRACT_SUBREG to have one result type"); 4036 4037 const auto &SrcRCDstRCPair = 4038 SrcRC->getMatchingSubClassWithSubRegs(CGRegs, SubIdx); 4039 assert(SrcRCDstRCPair->second && "Couldn't find a matching subclass"); 4040 M.addAction<ConstrainOperandToRegClassAction>(0, 0, *SrcRCDstRCPair->second); 4041 M.addAction<ConstrainOperandToRegClassAction>(0, 1, *SrcRCDstRCPair->first); 4042 4043 // We're done with this pattern! It's eligible for GISel emission; return 4044 // it. 4045 ++NumPatternImported; 4046 return std::move(M); 4047 } 4048 4049 M.addAction<ConstrainOperandsToDefinitionAction>(0); 4050 4051 // We're done with this pattern! It's eligible for GISel emission; return it. 4052 ++NumPatternImported; 4053 return std::move(M); 4054 } 4055 4056 // Emit imm predicate table and an enum to reference them with. 4057 // The 'Predicate_' part of the name is redundant but eliminating it is more 4058 // trouble than it's worth. 4059 void GlobalISelEmitter::emitCxxPredicateFns( 4060 raw_ostream &OS, StringRef CodeFieldName, StringRef TypeIdentifier, 4061 StringRef ArgType, StringRef ArgName, StringRef AdditionalDeclarations, 4062 std::function<bool(const Record *R)> Filter) { 4063 std::vector<const Record *> MatchedRecords; 4064 const auto &Defs = RK.getAllDerivedDefinitions("PatFrag"); 4065 std::copy_if(Defs.begin(), Defs.end(), std::back_inserter(MatchedRecords), 4066 [&](Record *Record) { 4067 return !Record->getValueAsString(CodeFieldName).empty() && 4068 Filter(Record); 4069 }); 4070 4071 if (!MatchedRecords.empty()) { 4072 OS << "// PatFrag predicates.\n" 4073 << "enum {\n"; 4074 std::string EnumeratorSeparator = 4075 (" = GIPFP_" + TypeIdentifier + "_Invalid + 1,\n").str(); 4076 for (const auto *Record : MatchedRecords) { 4077 OS << " GIPFP_" << TypeIdentifier << "_Predicate_" << Record->getName() 4078 << EnumeratorSeparator; 4079 EnumeratorSeparator = ",\n"; 4080 } 4081 OS << "};\n"; 4082 } 4083 4084 OS << "bool " << Target.getName() << "InstructionSelector::test" << ArgName 4085 << "Predicate_" << TypeIdentifier << "(unsigned PredicateID, " << ArgType << " " 4086 << ArgName << ") const {\n" 4087 << AdditionalDeclarations; 4088 if (!AdditionalDeclarations.empty()) 4089 OS << "\n"; 4090 if (!MatchedRecords.empty()) 4091 OS << " switch (PredicateID) {\n"; 4092 for (const auto *Record : MatchedRecords) { 4093 OS << " case GIPFP_" << TypeIdentifier << "_Predicate_" 4094 << Record->getName() << ": {\n" 4095 << " " << Record->getValueAsString(CodeFieldName) << "\n" 4096 << " llvm_unreachable(\"" << CodeFieldName 4097 << " should have returned\");\n" 4098 << " return false;\n" 4099 << " }\n"; 4100 } 4101 if (!MatchedRecords.empty()) 4102 OS << " }\n"; 4103 OS << " llvm_unreachable(\"Unknown predicate\");\n" 4104 << " return false;\n" 4105 << "}\n"; 4106 } 4107 4108 void GlobalISelEmitter::emitImmPredicateFns( 4109 raw_ostream &OS, StringRef TypeIdentifier, StringRef ArgType, 4110 std::function<bool(const Record *R)> Filter) { 4111 return emitCxxPredicateFns(OS, "ImmediateCode", TypeIdentifier, ArgType, 4112 "Imm", "", Filter); 4113 } 4114 4115 void GlobalISelEmitter::emitMIPredicateFns(raw_ostream &OS) { 4116 return emitCxxPredicateFns( 4117 OS, "GISelPredicateCode", "MI", "const MachineInstr &", "MI", 4118 " const MachineFunction &MF = *MI.getParent()->getParent();\n" 4119 " const MachineRegisterInfo &MRI = MF.getRegInfo();\n" 4120 " (void)MRI;", 4121 [](const Record *R) { return true; }); 4122 } 4123 4124 template <class GroupT> 4125 std::vector<Matcher *> GlobalISelEmitter::optimizeRules( 4126 ArrayRef<Matcher *> Rules, 4127 std::vector<std::unique_ptr<Matcher>> &MatcherStorage) { 4128 4129 std::vector<Matcher *> OptRules; 4130 std::unique_ptr<GroupT> CurrentGroup = make_unique<GroupT>(); 4131 assert(CurrentGroup->empty() && "Newly created group isn't empty!"); 4132 unsigned NumGroups = 0; 4133 4134 auto ProcessCurrentGroup = [&]() { 4135 if (CurrentGroup->empty()) 4136 // An empty group is good to be reused: 4137 return; 4138 4139 // If the group isn't large enough to provide any benefit, move all the 4140 // added rules out of it and make sure to re-create the group to properly 4141 // re-initialize it: 4142 if (CurrentGroup->size() < 2) 4143 for (Matcher *M : CurrentGroup->matchers()) 4144 OptRules.push_back(M); 4145 else { 4146 CurrentGroup->finalize(); 4147 OptRules.push_back(CurrentGroup.get()); 4148 MatcherStorage.emplace_back(std::move(CurrentGroup)); 4149 ++NumGroups; 4150 } 4151 CurrentGroup = make_unique<GroupT>(); 4152 }; 4153 for (Matcher *Rule : Rules) { 4154 // Greedily add as many matchers as possible to the current group: 4155 if (CurrentGroup->addMatcher(*Rule)) 4156 continue; 4157 4158 ProcessCurrentGroup(); 4159 assert(CurrentGroup->empty() && "A group wasn't properly re-initialized"); 4160 4161 // Try to add the pending matcher to a newly created empty group: 4162 if (!CurrentGroup->addMatcher(*Rule)) 4163 // If we couldn't add the matcher to an empty group, that group type 4164 // doesn't support that kind of matchers at all, so just skip it: 4165 OptRules.push_back(Rule); 4166 } 4167 ProcessCurrentGroup(); 4168 4169 LLVM_DEBUG(dbgs() << "NumGroups: " << NumGroups << "\n"); 4170 assert(CurrentGroup->empty() && "The last group wasn't properly processed"); 4171 return OptRules; 4172 } 4173 4174 MatchTable 4175 GlobalISelEmitter::buildMatchTable(MutableArrayRef<RuleMatcher> Rules, 4176 bool Optimize, bool WithCoverage) { 4177 std::vector<Matcher *> InputRules; 4178 for (Matcher &Rule : Rules) 4179 InputRules.push_back(&Rule); 4180 4181 if (!Optimize) 4182 return MatchTable::buildTable(InputRules, WithCoverage); 4183 4184 unsigned CurrentOrdering = 0; 4185 StringMap<unsigned> OpcodeOrder; 4186 for (RuleMatcher &Rule : Rules) { 4187 const StringRef Opcode = Rule.getOpcode(); 4188 assert(!Opcode.empty() && "Didn't expect an undefined opcode"); 4189 if (OpcodeOrder.count(Opcode) == 0) 4190 OpcodeOrder[Opcode] = CurrentOrdering++; 4191 } 4192 4193 std::stable_sort(InputRules.begin(), InputRules.end(), 4194 [&OpcodeOrder](const Matcher *A, const Matcher *B) { 4195 auto *L = static_cast<const RuleMatcher *>(A); 4196 auto *R = static_cast<const RuleMatcher *>(B); 4197 return std::make_tuple(OpcodeOrder[L->getOpcode()], 4198 L->getNumOperands()) < 4199 std::make_tuple(OpcodeOrder[R->getOpcode()], 4200 R->getNumOperands()); 4201 }); 4202 4203 for (Matcher *Rule : InputRules) 4204 Rule->optimize(); 4205 4206 std::vector<std::unique_ptr<Matcher>> MatcherStorage; 4207 std::vector<Matcher *> OptRules = 4208 optimizeRules<GroupMatcher>(InputRules, MatcherStorage); 4209 4210 for (Matcher *Rule : OptRules) 4211 Rule->optimize(); 4212 4213 OptRules = optimizeRules<SwitchMatcher>(OptRules, MatcherStorage); 4214 4215 return MatchTable::buildTable(OptRules, WithCoverage); 4216 } 4217 4218 void GroupMatcher::optimize() { 4219 // Make sure we only sort by a specific predicate within a range of rules that 4220 // all have that predicate checked against a specific value (not a wildcard): 4221 auto F = Matchers.begin(); 4222 auto T = F; 4223 auto E = Matchers.end(); 4224 while (T != E) { 4225 while (T != E) { 4226 auto *R = static_cast<RuleMatcher *>(*T); 4227 if (!R->getFirstConditionAsRootType().get().isValid()) 4228 break; 4229 ++T; 4230 } 4231 std::stable_sort(F, T, [](Matcher *A, Matcher *B) { 4232 auto *L = static_cast<RuleMatcher *>(A); 4233 auto *R = static_cast<RuleMatcher *>(B); 4234 return L->getFirstConditionAsRootType() < 4235 R->getFirstConditionAsRootType(); 4236 }); 4237 if (T != E) 4238 F = ++T; 4239 } 4240 GlobalISelEmitter::optimizeRules<GroupMatcher>(Matchers, MatcherStorage) 4241 .swap(Matchers); 4242 GlobalISelEmitter::optimizeRules<SwitchMatcher>(Matchers, MatcherStorage) 4243 .swap(Matchers); 4244 } 4245 4246 void GlobalISelEmitter::run(raw_ostream &OS) { 4247 if (!UseCoverageFile.empty()) { 4248 RuleCoverage = CodeGenCoverage(); 4249 auto RuleCoverageBufOrErr = MemoryBuffer::getFile(UseCoverageFile); 4250 if (!RuleCoverageBufOrErr) { 4251 PrintWarning(SMLoc(), "Missing rule coverage data"); 4252 RuleCoverage = None; 4253 } else { 4254 if (!RuleCoverage->parse(*RuleCoverageBufOrErr.get(), Target.getName())) { 4255 PrintWarning(SMLoc(), "Ignoring invalid or missing rule coverage data"); 4256 RuleCoverage = None; 4257 } 4258 } 4259 } 4260 4261 // Track the run-time opcode values 4262 gatherOpcodeValues(); 4263 // Track the run-time LLT ID values 4264 gatherTypeIDValues(); 4265 4266 // Track the GINodeEquiv definitions. 4267 gatherNodeEquivs(); 4268 4269 emitSourceFileHeader(("Global Instruction Selector for the " + 4270 Target.getName() + " target").str(), OS); 4271 std::vector<RuleMatcher> Rules; 4272 // Look through the SelectionDAG patterns we found, possibly emitting some. 4273 for (const PatternToMatch &Pat : CGP.ptms()) { 4274 ++NumPatternTotal; 4275 4276 auto MatcherOrErr = runOnPattern(Pat); 4277 4278 // The pattern analysis can fail, indicating an unsupported pattern. 4279 // Report that if we've been asked to do so. 4280 if (auto Err = MatcherOrErr.takeError()) { 4281 if (WarnOnSkippedPatterns) { 4282 PrintWarning(Pat.getSrcRecord()->getLoc(), 4283 "Skipped pattern: " + toString(std::move(Err))); 4284 } else { 4285 consumeError(std::move(Err)); 4286 } 4287 ++NumPatternImportsSkipped; 4288 continue; 4289 } 4290 4291 if (RuleCoverage) { 4292 if (RuleCoverage->isCovered(MatcherOrErr->getRuleID())) 4293 ++NumPatternsTested; 4294 else 4295 PrintWarning(Pat.getSrcRecord()->getLoc(), 4296 "Pattern is not covered by a test"); 4297 } 4298 Rules.push_back(std::move(MatcherOrErr.get())); 4299 } 4300 4301 // Comparison function to order records by name. 4302 auto orderByName = [](const Record *A, const Record *B) { 4303 return A->getName() < B->getName(); 4304 }; 4305 4306 std::vector<Record *> ComplexPredicates = 4307 RK.getAllDerivedDefinitions("GIComplexOperandMatcher"); 4308 llvm::sort(ComplexPredicates, orderByName); 4309 4310 std::vector<Record *> CustomRendererFns = 4311 RK.getAllDerivedDefinitions("GICustomOperandRenderer"); 4312 llvm::sort(CustomRendererFns, orderByName); 4313 4314 unsigned MaxTemporaries = 0; 4315 for (const auto &Rule : Rules) 4316 MaxTemporaries = std::max(MaxTemporaries, Rule.countRendererFns()); 4317 4318 OS << "#ifdef GET_GLOBALISEL_PREDICATE_BITSET\n" 4319 << "const unsigned MAX_SUBTARGET_PREDICATES = " << SubtargetFeatures.size() 4320 << ";\n" 4321 << "using PredicateBitset = " 4322 "llvm::PredicateBitsetImpl<MAX_SUBTARGET_PREDICATES>;\n" 4323 << "#endif // ifdef GET_GLOBALISEL_PREDICATE_BITSET\n\n"; 4324 4325 OS << "#ifdef GET_GLOBALISEL_TEMPORARIES_DECL\n" 4326 << " mutable MatcherState State;\n" 4327 << " typedef " 4328 "ComplexRendererFns(" 4329 << Target.getName() 4330 << "InstructionSelector::*ComplexMatcherMemFn)(MachineOperand &) const;\n" 4331 4332 << " typedef void(" << Target.getName() 4333 << "InstructionSelector::*CustomRendererFn)(MachineInstrBuilder &, const " 4334 "MachineInstr&) " 4335 "const;\n" 4336 << " const ISelInfoTy<PredicateBitset, ComplexMatcherMemFn, " 4337 "CustomRendererFn> " 4338 "ISelInfo;\n"; 4339 OS << " static " << Target.getName() 4340 << "InstructionSelector::ComplexMatcherMemFn ComplexPredicateFns[];\n" 4341 << " static " << Target.getName() 4342 << "InstructionSelector::CustomRendererFn CustomRenderers[];\n" 4343 << " bool testImmPredicate_I64(unsigned PredicateID, int64_t Imm) const " 4344 "override;\n" 4345 << " bool testImmPredicate_APInt(unsigned PredicateID, const APInt &Imm) " 4346 "const override;\n" 4347 << " bool testImmPredicate_APFloat(unsigned PredicateID, const APFloat " 4348 "&Imm) const override;\n" 4349 << " const int64_t *getMatchTable() const override;\n" 4350 << " bool testMIPredicate_MI(unsigned PredicateID, const MachineInstr &MI) " 4351 "const override;\n" 4352 << "#endif // ifdef GET_GLOBALISEL_TEMPORARIES_DECL\n\n"; 4353 4354 OS << "#ifdef GET_GLOBALISEL_TEMPORARIES_INIT\n" 4355 << ", State(" << MaxTemporaries << "),\n" 4356 << "ISelInfo(TypeObjects, NumTypeObjects, FeatureBitsets" 4357 << ", ComplexPredicateFns, CustomRenderers)\n" 4358 << "#endif // ifdef GET_GLOBALISEL_TEMPORARIES_INIT\n\n"; 4359 4360 OS << "#ifdef GET_GLOBALISEL_IMPL\n"; 4361 SubtargetFeatureInfo::emitSubtargetFeatureBitEnumeration(SubtargetFeatures, 4362 OS); 4363 4364 // Separate subtarget features by how often they must be recomputed. 4365 SubtargetFeatureInfoMap ModuleFeatures; 4366 std::copy_if(SubtargetFeatures.begin(), SubtargetFeatures.end(), 4367 std::inserter(ModuleFeatures, ModuleFeatures.end()), 4368 [](const SubtargetFeatureInfoMap::value_type &X) { 4369 return !X.second.mustRecomputePerFunction(); 4370 }); 4371 SubtargetFeatureInfoMap FunctionFeatures; 4372 std::copy_if(SubtargetFeatures.begin(), SubtargetFeatures.end(), 4373 std::inserter(FunctionFeatures, FunctionFeatures.end()), 4374 [](const SubtargetFeatureInfoMap::value_type &X) { 4375 return X.second.mustRecomputePerFunction(); 4376 }); 4377 4378 SubtargetFeatureInfo::emitComputeAvailableFeatures( 4379 Target.getName(), "InstructionSelector", "computeAvailableModuleFeatures", 4380 ModuleFeatures, OS); 4381 SubtargetFeatureInfo::emitComputeAvailableFeatures( 4382 Target.getName(), "InstructionSelector", 4383 "computeAvailableFunctionFeatures", FunctionFeatures, OS, 4384 "const MachineFunction *MF"); 4385 4386 // Emit a table containing the LLT objects needed by the matcher and an enum 4387 // for the matcher to reference them with. 4388 std::vector<LLTCodeGen> TypeObjects; 4389 for (const auto &Ty : KnownTypes) 4390 TypeObjects.push_back(Ty); 4391 llvm::sort(TypeObjects); 4392 OS << "// LLT Objects.\n" 4393 << "enum {\n"; 4394 for (const auto &TypeObject : TypeObjects) { 4395 OS << " "; 4396 TypeObject.emitCxxEnumValue(OS); 4397 OS << ",\n"; 4398 } 4399 OS << "};\n"; 4400 OS << "const static size_t NumTypeObjects = " << TypeObjects.size() << ";\n" 4401 << "const static LLT TypeObjects[] = {\n"; 4402 for (const auto &TypeObject : TypeObjects) { 4403 OS << " "; 4404 TypeObject.emitCxxConstructorCall(OS); 4405 OS << ",\n"; 4406 } 4407 OS << "};\n\n"; 4408 4409 // Emit a table containing the PredicateBitsets objects needed by the matcher 4410 // and an enum for the matcher to reference them with. 4411 std::vector<std::vector<Record *>> FeatureBitsets; 4412 for (auto &Rule : Rules) 4413 FeatureBitsets.push_back(Rule.getRequiredFeatures()); 4414 llvm::sort(FeatureBitsets, [&](const std::vector<Record *> &A, 4415 const std::vector<Record *> &B) { 4416 if (A.size() < B.size()) 4417 return true; 4418 if (A.size() > B.size()) 4419 return false; 4420 for (const auto &Pair : zip(A, B)) { 4421 if (std::get<0>(Pair)->getName() < std::get<1>(Pair)->getName()) 4422 return true; 4423 if (std::get<0>(Pair)->getName() > std::get<1>(Pair)->getName()) 4424 return false; 4425 } 4426 return false; 4427 }); 4428 FeatureBitsets.erase( 4429 std::unique(FeatureBitsets.begin(), FeatureBitsets.end()), 4430 FeatureBitsets.end()); 4431 OS << "// Feature bitsets.\n" 4432 << "enum {\n" 4433 << " GIFBS_Invalid,\n"; 4434 for (const auto &FeatureBitset : FeatureBitsets) { 4435 if (FeatureBitset.empty()) 4436 continue; 4437 OS << " " << getNameForFeatureBitset(FeatureBitset) << ",\n"; 4438 } 4439 OS << "};\n" 4440 << "const static PredicateBitset FeatureBitsets[] {\n" 4441 << " {}, // GIFBS_Invalid\n"; 4442 for (const auto &FeatureBitset : FeatureBitsets) { 4443 if (FeatureBitset.empty()) 4444 continue; 4445 OS << " {"; 4446 for (const auto &Feature : FeatureBitset) { 4447 const auto &I = SubtargetFeatures.find(Feature); 4448 assert(I != SubtargetFeatures.end() && "Didn't import predicate?"); 4449 OS << I->second.getEnumBitName() << ", "; 4450 } 4451 OS << "},\n"; 4452 } 4453 OS << "};\n\n"; 4454 4455 // Emit complex predicate table and an enum to reference them with. 4456 OS << "// ComplexPattern predicates.\n" 4457 << "enum {\n" 4458 << " GICP_Invalid,\n"; 4459 for (const auto &Record : ComplexPredicates) 4460 OS << " GICP_" << Record->getName() << ",\n"; 4461 OS << "};\n" 4462 << "// See constructor for table contents\n\n"; 4463 4464 emitImmPredicateFns(OS, "I64", "int64_t", [](const Record *R) { 4465 bool Unset; 4466 return !R->getValueAsBitOrUnset("IsAPFloat", Unset) && 4467 !R->getValueAsBit("IsAPInt"); 4468 }); 4469 emitImmPredicateFns(OS, "APFloat", "const APFloat &", [](const Record *R) { 4470 bool Unset; 4471 return R->getValueAsBitOrUnset("IsAPFloat", Unset); 4472 }); 4473 emitImmPredicateFns(OS, "APInt", "const APInt &", [](const Record *R) { 4474 return R->getValueAsBit("IsAPInt"); 4475 }); 4476 emitMIPredicateFns(OS); 4477 OS << "\n"; 4478 4479 OS << Target.getName() << "InstructionSelector::ComplexMatcherMemFn\n" 4480 << Target.getName() << "InstructionSelector::ComplexPredicateFns[] = {\n" 4481 << " nullptr, // GICP_Invalid\n"; 4482 for (const auto &Record : ComplexPredicates) 4483 OS << " &" << Target.getName() 4484 << "InstructionSelector::" << Record->getValueAsString("MatcherFn") 4485 << ", // " << Record->getName() << "\n"; 4486 OS << "};\n\n"; 4487 4488 OS << "// Custom renderers.\n" 4489 << "enum {\n" 4490 << " GICR_Invalid,\n"; 4491 for (const auto &Record : CustomRendererFns) 4492 OS << " GICR_" << Record->getValueAsString("RendererFn") << ", \n"; 4493 OS << "};\n"; 4494 4495 OS << Target.getName() << "InstructionSelector::CustomRendererFn\n" 4496 << Target.getName() << "InstructionSelector::CustomRenderers[] = {\n" 4497 << " nullptr, // GICP_Invalid\n"; 4498 for (const auto &Record : CustomRendererFns) 4499 OS << " &" << Target.getName() 4500 << "InstructionSelector::" << Record->getValueAsString("RendererFn") 4501 << ", // " << Record->getName() << "\n"; 4502 OS << "};\n\n"; 4503 4504 llvm::stable_sort(Rules, [&](const RuleMatcher &A, const RuleMatcher &B) { 4505 int ScoreA = RuleMatcherScores[A.getRuleID()]; 4506 int ScoreB = RuleMatcherScores[B.getRuleID()]; 4507 if (ScoreA > ScoreB) 4508 return true; 4509 if (ScoreB > ScoreA) 4510 return false; 4511 if (A.isHigherPriorityThan(B)) { 4512 assert(!B.isHigherPriorityThan(A) && "Cannot be more important " 4513 "and less important at " 4514 "the same time"); 4515 return true; 4516 } 4517 return false; 4518 }); 4519 4520 OS << "bool " << Target.getName() 4521 << "InstructionSelector::selectImpl(MachineInstr &I, CodeGenCoverage " 4522 "&CoverageInfo) const {\n" 4523 << " MachineFunction &MF = *I.getParent()->getParent();\n" 4524 << " MachineRegisterInfo &MRI = MF.getRegInfo();\n" 4525 << " // FIXME: This should be computed on a per-function basis rather " 4526 "than per-insn.\n" 4527 << " AvailableFunctionFeatures = computeAvailableFunctionFeatures(&STI, " 4528 "&MF);\n" 4529 << " const PredicateBitset AvailableFeatures = getAvailableFeatures();\n" 4530 << " NewMIVector OutMIs;\n" 4531 << " State.MIs.clear();\n" 4532 << " State.MIs.push_back(&I);\n\n" 4533 << " if (executeMatchTable(*this, OutMIs, State, ISelInfo" 4534 << ", getMatchTable(), TII, MRI, TRI, RBI, AvailableFeatures" 4535 << ", CoverageInfo)) {\n" 4536 << " return true;\n" 4537 << " }\n\n" 4538 << " return false;\n" 4539 << "}\n\n"; 4540 4541 const MatchTable Table = 4542 buildMatchTable(Rules, OptimizeMatchTable, GenerateCoverage); 4543 OS << "const int64_t *" << Target.getName() 4544 << "InstructionSelector::getMatchTable() const {\n"; 4545 Table.emitDeclaration(OS); 4546 OS << " return "; 4547 Table.emitUse(OS); 4548 OS << ";\n}\n"; 4549 OS << "#endif // ifdef GET_GLOBALISEL_IMPL\n"; 4550 4551 OS << "#ifdef GET_GLOBALISEL_PREDICATES_DECL\n" 4552 << "PredicateBitset AvailableModuleFeatures;\n" 4553 << "mutable PredicateBitset AvailableFunctionFeatures;\n" 4554 << "PredicateBitset getAvailableFeatures() const {\n" 4555 << " return AvailableModuleFeatures | AvailableFunctionFeatures;\n" 4556 << "}\n" 4557 << "PredicateBitset\n" 4558 << "computeAvailableModuleFeatures(const " << Target.getName() 4559 << "Subtarget *Subtarget) const;\n" 4560 << "PredicateBitset\n" 4561 << "computeAvailableFunctionFeatures(const " << Target.getName() 4562 << "Subtarget *Subtarget,\n" 4563 << " const MachineFunction *MF) const;\n" 4564 << "#endif // ifdef GET_GLOBALISEL_PREDICATES_DECL\n"; 4565 4566 OS << "#ifdef GET_GLOBALISEL_PREDICATES_INIT\n" 4567 << "AvailableModuleFeatures(computeAvailableModuleFeatures(&STI)),\n" 4568 << "AvailableFunctionFeatures()\n" 4569 << "#endif // ifdef GET_GLOBALISEL_PREDICATES_INIT\n"; 4570 } 4571 4572 void GlobalISelEmitter::declareSubtargetFeature(Record *Predicate) { 4573 if (SubtargetFeatures.count(Predicate) == 0) 4574 SubtargetFeatures.emplace( 4575 Predicate, SubtargetFeatureInfo(Predicate, SubtargetFeatures.size())); 4576 } 4577 4578 void RuleMatcher::optimize() { 4579 for (auto &Item : InsnVariableIDs) { 4580 InstructionMatcher &InsnMatcher = *Item.first; 4581 for (auto &OM : InsnMatcher.operands()) { 4582 // Complex Patterns are usually expensive and they relatively rarely fail 4583 // on their own: more often we end up throwing away all the work done by a 4584 // matching part of a complex pattern because some other part of the 4585 // enclosing pattern didn't match. All of this makes it beneficial to 4586 // delay complex patterns until the very end of the rule matching, 4587 // especially for targets having lots of complex patterns. 4588 for (auto &OP : OM->predicates()) 4589 if (isa<ComplexPatternOperandMatcher>(OP)) 4590 EpilogueMatchers.emplace_back(std::move(OP)); 4591 OM->eraseNullPredicates(); 4592 } 4593 InsnMatcher.optimize(); 4594 } 4595 llvm::sort(EpilogueMatchers, [](const std::unique_ptr<PredicateMatcher> &L, 4596 const std::unique_ptr<PredicateMatcher> &R) { 4597 return std::make_tuple(L->getKind(), L->getInsnVarID(), L->getOpIdx()) < 4598 std::make_tuple(R->getKind(), R->getInsnVarID(), R->getOpIdx()); 4599 }); 4600 } 4601 4602 bool RuleMatcher::hasFirstCondition() const { 4603 if (insnmatchers_empty()) 4604 return false; 4605 InstructionMatcher &Matcher = insnmatchers_front(); 4606 if (!Matcher.predicates_empty()) 4607 return true; 4608 for (auto &OM : Matcher.operands()) 4609 for (auto &OP : OM->predicates()) 4610 if (!isa<InstructionOperandMatcher>(OP)) 4611 return true; 4612 return false; 4613 } 4614 4615 const PredicateMatcher &RuleMatcher::getFirstCondition() const { 4616 assert(!insnmatchers_empty() && 4617 "Trying to get a condition from an empty RuleMatcher"); 4618 4619 InstructionMatcher &Matcher = insnmatchers_front(); 4620 if (!Matcher.predicates_empty()) 4621 return **Matcher.predicates_begin(); 4622 // If there is no more predicate on the instruction itself, look at its 4623 // operands. 4624 for (auto &OM : Matcher.operands()) 4625 for (auto &OP : OM->predicates()) 4626 if (!isa<InstructionOperandMatcher>(OP)) 4627 return *OP; 4628 4629 llvm_unreachable("Trying to get a condition from an InstructionMatcher with " 4630 "no conditions"); 4631 } 4632 4633 std::unique_ptr<PredicateMatcher> RuleMatcher::popFirstCondition() { 4634 assert(!insnmatchers_empty() && 4635 "Trying to pop a condition from an empty RuleMatcher"); 4636 4637 InstructionMatcher &Matcher = insnmatchers_front(); 4638 if (!Matcher.predicates_empty()) 4639 return Matcher.predicates_pop_front(); 4640 // If there is no more predicate on the instruction itself, look at its 4641 // operands. 4642 for (auto &OM : Matcher.operands()) 4643 for (auto &OP : OM->predicates()) 4644 if (!isa<InstructionOperandMatcher>(OP)) { 4645 std::unique_ptr<PredicateMatcher> Result = std::move(OP); 4646 OM->eraseNullPredicates(); 4647 return Result; 4648 } 4649 4650 llvm_unreachable("Trying to pop a condition from an InstructionMatcher with " 4651 "no conditions"); 4652 } 4653 4654 bool GroupMatcher::candidateConditionMatches( 4655 const PredicateMatcher &Predicate) const { 4656 4657 if (empty()) { 4658 // Sharing predicates for nested instructions is not supported yet as we 4659 // currently don't hoist the GIM_RecordInsn's properly, therefore we can 4660 // only work on the original root instruction (InsnVarID == 0): 4661 if (Predicate.getInsnVarID() != 0) 4662 return false; 4663 // ... otherwise an empty group can handle any predicate with no specific 4664 // requirements: 4665 return true; 4666 } 4667 4668 const Matcher &Representative = **Matchers.begin(); 4669 const auto &RepresentativeCondition = Representative.getFirstCondition(); 4670 // ... if not empty, the group can only accomodate matchers with the exact 4671 // same first condition: 4672 return Predicate.isIdentical(RepresentativeCondition); 4673 } 4674 4675 bool GroupMatcher::addMatcher(Matcher &Candidate) { 4676 if (!Candidate.hasFirstCondition()) 4677 return false; 4678 4679 const PredicateMatcher &Predicate = Candidate.getFirstCondition(); 4680 if (!candidateConditionMatches(Predicate)) 4681 return false; 4682 4683 Matchers.push_back(&Candidate); 4684 return true; 4685 } 4686 4687 void GroupMatcher::finalize() { 4688 assert(Conditions.empty() && "Already finalized?"); 4689 if (empty()) 4690 return; 4691 4692 Matcher &FirstRule = **Matchers.begin(); 4693 for (;;) { 4694 // All the checks are expected to succeed during the first iteration: 4695 for (const auto &Rule : Matchers) 4696 if (!Rule->hasFirstCondition()) 4697 return; 4698 const auto &FirstCondition = FirstRule.getFirstCondition(); 4699 for (unsigned I = 1, E = Matchers.size(); I < E; ++I) 4700 if (!Matchers[I]->getFirstCondition().isIdentical(FirstCondition)) 4701 return; 4702 4703 Conditions.push_back(FirstRule.popFirstCondition()); 4704 for (unsigned I = 1, E = Matchers.size(); I < E; ++I) 4705 Matchers[I]->popFirstCondition(); 4706 } 4707 } 4708 4709 void GroupMatcher::emit(MatchTable &Table) { 4710 unsigned LabelID = ~0U; 4711 if (!Conditions.empty()) { 4712 LabelID = Table.allocateLabelID(); 4713 Table << MatchTable::Opcode("GIM_Try", +1) 4714 << MatchTable::Comment("On fail goto") 4715 << MatchTable::JumpTarget(LabelID) << MatchTable::LineBreak; 4716 } 4717 for (auto &Condition : Conditions) 4718 Condition->emitPredicateOpcodes( 4719 Table, *static_cast<RuleMatcher *>(*Matchers.begin())); 4720 4721 for (const auto &M : Matchers) 4722 M->emit(Table); 4723 4724 // Exit the group 4725 if (!Conditions.empty()) 4726 Table << MatchTable::Opcode("GIM_Reject", -1) << MatchTable::LineBreak 4727 << MatchTable::Label(LabelID); 4728 } 4729 4730 bool SwitchMatcher::isSupportedPredicateType(const PredicateMatcher &P) { 4731 return isa<InstructionOpcodeMatcher>(P) || isa<LLTOperandMatcher>(P); 4732 } 4733 4734 bool SwitchMatcher::candidateConditionMatches( 4735 const PredicateMatcher &Predicate) const { 4736 4737 if (empty()) { 4738 // Sharing predicates for nested instructions is not supported yet as we 4739 // currently don't hoist the GIM_RecordInsn's properly, therefore we can 4740 // only work on the original root instruction (InsnVarID == 0): 4741 if (Predicate.getInsnVarID() != 0) 4742 return false; 4743 // ... while an attempt to add even a root matcher to an empty SwitchMatcher 4744 // could fail as not all the types of conditions are supported: 4745 if (!isSupportedPredicateType(Predicate)) 4746 return false; 4747 // ... or the condition might not have a proper implementation of 4748 // getValue() / isIdenticalDownToValue() yet: 4749 if (!Predicate.hasValue()) 4750 return false; 4751 // ... otherwise an empty Switch can accomodate the condition with no 4752 // further requirements: 4753 return true; 4754 } 4755 4756 const Matcher &CaseRepresentative = **Matchers.begin(); 4757 const auto &RepresentativeCondition = CaseRepresentative.getFirstCondition(); 4758 // Switch-cases must share the same kind of condition and path to the value it 4759 // checks: 4760 if (!Predicate.isIdenticalDownToValue(RepresentativeCondition)) 4761 return false; 4762 4763 const auto Value = Predicate.getValue(); 4764 // ... but be unique with respect to the actual value they check: 4765 return Values.count(Value) == 0; 4766 } 4767 4768 bool SwitchMatcher::addMatcher(Matcher &Candidate) { 4769 if (!Candidate.hasFirstCondition()) 4770 return false; 4771 4772 const PredicateMatcher &Predicate = Candidate.getFirstCondition(); 4773 if (!candidateConditionMatches(Predicate)) 4774 return false; 4775 const auto Value = Predicate.getValue(); 4776 Values.insert(Value); 4777 4778 Matchers.push_back(&Candidate); 4779 return true; 4780 } 4781 4782 void SwitchMatcher::finalize() { 4783 assert(Condition == nullptr && "Already finalized"); 4784 assert(Values.size() == Matchers.size() && "Broken SwitchMatcher"); 4785 if (empty()) 4786 return; 4787 4788 std::stable_sort(Matchers.begin(), Matchers.end(), 4789 [](const Matcher *L, const Matcher *R) { 4790 return L->getFirstCondition().getValue() < 4791 R->getFirstCondition().getValue(); 4792 }); 4793 Condition = Matchers[0]->popFirstCondition(); 4794 for (unsigned I = 1, E = Values.size(); I < E; ++I) 4795 Matchers[I]->popFirstCondition(); 4796 } 4797 4798 void SwitchMatcher::emitPredicateSpecificOpcodes(const PredicateMatcher &P, 4799 MatchTable &Table) { 4800 assert(isSupportedPredicateType(P) && "Predicate type is not supported"); 4801 4802 if (const auto *Condition = dyn_cast<InstructionOpcodeMatcher>(&P)) { 4803 Table << MatchTable::Opcode("GIM_SwitchOpcode") << MatchTable::Comment("MI") 4804 << MatchTable::IntValue(Condition->getInsnVarID()); 4805 return; 4806 } 4807 if (const auto *Condition = dyn_cast<LLTOperandMatcher>(&P)) { 4808 Table << MatchTable::Opcode("GIM_SwitchType") << MatchTable::Comment("MI") 4809 << MatchTable::IntValue(Condition->getInsnVarID()) 4810 << MatchTable::Comment("Op") 4811 << MatchTable::IntValue(Condition->getOpIdx()); 4812 return; 4813 } 4814 4815 llvm_unreachable("emitPredicateSpecificOpcodes is broken: can not handle a " 4816 "predicate type that is claimed to be supported"); 4817 } 4818 4819 void SwitchMatcher::emit(MatchTable &Table) { 4820 assert(Values.size() == Matchers.size() && "Broken SwitchMatcher"); 4821 if (empty()) 4822 return; 4823 assert(Condition != nullptr && 4824 "Broken SwitchMatcher, hasn't been finalized?"); 4825 4826 std::vector<unsigned> LabelIDs(Values.size()); 4827 std::generate(LabelIDs.begin(), LabelIDs.end(), 4828 [&Table]() { return Table.allocateLabelID(); }); 4829 const unsigned Default = Table.allocateLabelID(); 4830 4831 const int64_t LowerBound = Values.begin()->getRawValue(); 4832 const int64_t UpperBound = Values.rbegin()->getRawValue() + 1; 4833 4834 emitPredicateSpecificOpcodes(*Condition, Table); 4835 4836 Table << MatchTable::Comment("[") << MatchTable::IntValue(LowerBound) 4837 << MatchTable::IntValue(UpperBound) << MatchTable::Comment(")") 4838 << MatchTable::Comment("default:") << MatchTable::JumpTarget(Default); 4839 4840 int64_t J = LowerBound; 4841 auto VI = Values.begin(); 4842 for (unsigned I = 0, E = Values.size(); I < E; ++I) { 4843 auto V = *VI++; 4844 while (J++ < V.getRawValue()) 4845 Table << MatchTable::IntValue(0); 4846 V.turnIntoComment(); 4847 Table << MatchTable::LineBreak << V << MatchTable::JumpTarget(LabelIDs[I]); 4848 } 4849 Table << MatchTable::LineBreak; 4850 4851 for (unsigned I = 0, E = Values.size(); I < E; ++I) { 4852 Table << MatchTable::Label(LabelIDs[I]); 4853 Matchers[I]->emit(Table); 4854 Table << MatchTable::Opcode("GIM_Reject") << MatchTable::LineBreak; 4855 } 4856 Table << MatchTable::Label(Default); 4857 } 4858 4859 unsigned OperandMatcher::getInsnVarID() const { return Insn.getInsnVarID(); } 4860 4861 } // end anonymous namespace 4862 4863 //===----------------------------------------------------------------------===// 4864 4865 namespace llvm { 4866 void EmitGlobalISel(RecordKeeper &RK, raw_ostream &OS) { 4867 GlobalISelEmitter(RK).run(OS); 4868 } 4869 } // End llvm namespace 4870