Index: lib/Target/X86/X86InstrInfo.h =================================================================== --- lib/Target/X86/X86InstrInfo.h +++ lib/Target/X86/X86InstrInfo.h @@ -558,6 +558,9 @@ MachineBasicBlock::iterator &It, MachineFunction &MF, const outliner::Candidate &C) const override; +#define GET_TII_HELPER_DECLS +#include "X86GenInstrInfo.inc" + protected: /// Commutes the operands in the given instruction by changing the operands /// order and/or changing the instruction's opcode and/or the immediate value Index: lib/Target/X86/X86InstrInfo.cpp =================================================================== --- lib/Target/X86/X86InstrInfo.cpp +++ lib/Target/X86/X86InstrInfo.cpp @@ -7844,3 +7844,6 @@ return It; } + +#define GET_TII_HELPERS +#include "X86GenInstrInfo.inc" Index: utils/TableGen/InstrInfoEmitter.cpp =================================================================== --- utils/TableGen/InstrInfoEmitter.cpp +++ utils/TableGen/InstrInfoEmitter.cpp @@ -66,7 +66,8 @@ /// This method is used to custom expand TIIPredicate definitions. /// See file llvm/Target/TargetInstPredicates.td for a description of what is /// a TIIPredicate and how to use it. - void emitTIIHelperMethods(raw_ostream &OS, StringRef TargetName); + void emitTIIHelperMethods(raw_ostream &OS, StringRef TargetName, + bool ExpandDefinition = true); /// Expand TIIPredicate definitions to functions that accept a const MCInst /// reference. @@ -400,7 +401,8 @@ } void InstrInfoEmitter::emitTIIHelperMethods(raw_ostream &OS, - StringRef TargetName) { + StringRef TargetName, + bool ExpandDefinition) { RecVec TIIPredicates = Records.getAllDerivedDefinitions("TIIPredicate"); if (TIIPredicates.empty()) return; @@ -410,8 +412,17 @@ PE.setIndentLevel(2); for (const Record *Rec : TIIPredicates) { - OS << "\n static bool " << Rec->getValueAsString("FunctionName"); - OS << "(const MachineInstr &MI) {\n"; + OS << "\n " << (ExpandDefinition ? "" : "static ") << "bool "; + if (ExpandDefinition) + OS << TargetName << "InstrInfo::"; + OS << Rec->getValueAsString("FunctionName"); + OS << "(const MachineInstr &MI)"; + if (!ExpandDefinition) { + OS << ";\n"; + continue; + } + + OS << " {\n"; OS.indent(PE.getIndentLevel() * 2); PE.expandStatement(OS, Rec->getValueAsDef("Body")); @@ -517,12 +528,21 @@ << "(int CFSetupOpcode = -1, int CFDestroyOpcode = -1, int CatchRetOpcode = -1, int ReturnOpcode = -1);\n" << " ~" << ClassName << "() override = default;\n"; - emitTIIHelperMethods(OS, TargetName); OS << "\n};\n} // end llvm namespace\n"; OS << "#endif // GET_INSTRINFO_HEADER\n\n"; + OS << "#ifdef GET_TII_HELPER_DECLS\n"; + OS << "#undef GET_TII_HELPER_DECLS\n"; + emitTIIHelperMethods(OS, TargetName, /* ExpandDefintion = */false); + OS << "#endif // GET_TII_HELPER_DECLS\n\n"; + + OS << "#ifdef GET_TII_HELPERS\n"; + OS << "#undef GET_TII_HELPERS\n"; + emitTIIHelperMethods(OS, TargetName, /* ExpandDefintion = */true); + OS << "#endif // GET_TTI_HELPERS\n\n"; + OS << "#ifdef GET_INSTRINFO_CTOR_DTOR\n"; OS << "#undef GET_INSTRINFO_CTOR_DTOR\n"; Index: utils/TableGen/PredicateExpander.h =================================================================== --- utils/TableGen/PredicateExpander.h +++ utils/TableGen/PredicateExpander.h @@ -56,9 +56,16 @@ using RecVec = std::vector; void expandTrue(raw_ostream &OS); void expandFalse(raw_ostream &OS); - void expandCheckImmOperand(raw_ostream &OS, int OpIndex, int ImmVal); - void expandCheckImmOperand(raw_ostream &OS, int OpIndex, StringRef ImmVal); - void expandCheckRegOperand(raw_ostream &OS, int OpIndex, const Record *Reg); + void expandCheckImmOperand(raw_ostream &OS, int OpIndex, int ImmVal, + StringRef FunctionExpander); + void expandCheckImmOperand(raw_ostream &OS, int OpIndex, StringRef ImmVal, + StringRef FunctionExpander); + void expandCheckImmOperandSimple(raw_ostream &OS, int OpIndex, + StringRef FunctionExpander); + void expandCheckRegOperand(raw_ostream &OS, int OpIndex, const Record *Reg, + StringRef FunctionExpander); + void expandCheckRegOperandSimple(raw_ostream &OS, int OpIndex, + StringRef FunctionExpander); void expandCheckSameRegOperand(raw_ostream &OS, int First, int Second); void expandCheckNumOperands(raw_ostream &OS, int NumOps); void expandCheckOpcode(raw_ostream &OS, const Record *Inst); Index: utils/TableGen/PredicateExpander.cpp =================================================================== --- utils/TableGen/PredicateExpander.cpp +++ utils/TableGen/PredicateExpander.cpp @@ -20,23 +20,43 @@ void PredicateExpander::expandFalse(raw_ostream &OS) { OS << "false"; } void PredicateExpander::expandCheckImmOperand(raw_ostream &OS, int OpIndex, - int ImmVal) { + int ImmVal, + StringRef FunctionMapper) { + if (!FunctionMapper.empty()) + OS << FunctionMapper << "("; OS << "MI" << (isByRef() ? "." : "->") << "getOperand(" << OpIndex - << ").getImm() " << (shouldNegate() ? "!= " : "== ") << ImmVal; + << ").getImm()"; + OS << (FunctionMapper.empty() ? " " : ") "); + OS << (shouldNegate() ? "!= " : "== ") << ImmVal; } void PredicateExpander::expandCheckImmOperand(raw_ostream &OS, int OpIndex, - StringRef ImmVal) { + StringRef ImmVal, + StringRef FunctionMapper) { + if (!FunctionMapper.empty()) + OS << FunctionMapper << "("; OS << "MI" << (isByRef() ? "." : "->") << "getOperand(" << OpIndex - << ").getImm() " << (shouldNegate() ? "!= " : "== ") << ImmVal; + << ").getImm()"; + + OS << (FunctionMapper.empty() ? "" : ")"); + if (ImmVal.empty()) + return; + OS << (shouldNegate() ? " != " : " == ") << ImmVal; } void PredicateExpander::expandCheckRegOperand(raw_ostream &OS, int OpIndex, - const Record *Reg) { + const Record *Reg, + StringRef FunctionMapper) { assert(Reg->isSubClassOf("Register") && "Expected a register Record!"); + if (!FunctionMapper.empty()) + OS << FunctionMapper << "("; OS << "MI" << (isByRef() ? "." : "->") << "getOperand(" << OpIndex - << ").getReg() " << (shouldNegate() ? "!= " : "== "); + << ").getReg()"; + OS << (FunctionMapper.empty() ? "" : ")"); + if (!Reg) + return; + OS << (shouldNegate() ? " != " : " == "); const StringRef Str = Reg->getValueAsString("Namespace"); if (!Str.empty()) OS << Str << "::"; @@ -137,7 +157,7 @@ void PredicateExpander::expandTIIFunctionCall(raw_ostream &OS, StringRef MethodName) { OS << (shouldNegate() ? "!" : ""); - OS << TargetName << (shouldExpandForMC() ? "_MC::" : "GenInstrInfo::"); + OS << TargetName << (shouldExpandForMC() ? "_MC::" : "InstrInfo::"); OS << MethodName << (isByRef() ? "(MI)" : "(*MI)"); } @@ -266,18 +286,30 @@ if (Rec->isSubClassOf("CheckRegOperand")) return expandCheckRegOperand(OS, Rec->getValueAsInt("OpIndex"), - Rec->getValueAsDef("Reg")); + Rec->getValueAsDef("Reg"), + Rec->getValueAsString("FunctionMapper")); + + if (Rec->isSubClassOf("CheckRegOperandSimple")) + return expandCheckRegOperand(OS, Rec->getValueAsInt("OpIndex"), + nullptr, + Rec->getValueAsString("FunctionMapper")); if (Rec->isSubClassOf("CheckInvalidRegOperand")) return expandCheckInvalidRegOperand(OS, Rec->getValueAsInt("OpIndex")); if (Rec->isSubClassOf("CheckImmOperand")) return expandCheckImmOperand(OS, Rec->getValueAsInt("OpIndex"), - Rec->getValueAsInt("ImmVal")); + Rec->getValueAsInt("ImmVal"), + Rec->getValueAsString("FunctionMapper")); if (Rec->isSubClassOf("CheckImmOperand_s")) return expandCheckImmOperand(OS, Rec->getValueAsInt("OpIndex"), - Rec->getValueAsString("ImmVal")); + Rec->getValueAsString("ImmVal"), + Rec->getValueAsString("FunctionMapper")); + + if (Rec->isSubClassOf("CheckImmOperandSimple")) + return expandCheckImmOperand(OS, Rec->getValueAsInt("OpIndex"), "", + Rec->getValueAsString("FunctionMapper")); if (Rec->isSubClassOf("CheckSameRegOperand")) return expandCheckSameRegOperand(OS, Rec->getValueAsInt("FirstIndex"),