diff --git a/utils/TableGen/CodeGenRegisters.cpp b/utils/TableGen/CodeGenRegisters.cpp index 95394bdee02..145ec26c9eb 100644 --- a/utils/TableGen/CodeGenRegisters.cpp +++ b/utils/TableGen/CodeGenRegisters.cpp @@ -191,6 +191,9 @@ CodeGenRegister::computeSubRegs(CodeGenRegBank &RegBank) { if (!SubRegs.insert(std::make_pair(Idx, SR)).second) throw TGError(TheDef->getLoc(), "SubRegIndex " + Idx->getName() + " appears twice in Register " + getName()); + // Map explicit sub-registers first, so the names take precedence. + // The inherited sub-registers are mapped below. + SubReg2Idx.insert(std::make_pair(SR, Idx)); } // Keep track of inherited subregs and how they can be reached. @@ -309,6 +312,19 @@ CodeGenRegister::computeSubRegs(CodeGenRegBank &RegBank) { SubRegs[RegBank.getCompositeSubRegIndex(Idx, SI->first)] = SI->second; } + // Compute the inverse SubReg -> Idx map. + for (SubRegMap::const_iterator SI = SubRegs.begin(), SE = SubRegs.end(); + SI != SE; ++SI) { + // Ignore idempotent sub-register indices. + if (SI->second == this) + continue; + // Is is possible to have multiple names for the same sub-register. + // For example, XMM0 appears as sub_xmm, sub_sd, and sub_ss in YMM0. + // Eventually, this degeneration should go away, but for now we simply give + // precedence to the explicit sub-register index over the inherited ones. + SubReg2Idx.insert(std::make_pair(SI->second, SI->first)); + } + // Initialize RegUnitList. A register with no subregisters creates its own // unit. Otherwise, it inherits all its subregister's units. Because // getSubRegs is called recursively, this processes the register hierarchy in diff --git a/utils/TableGen/CodeGenRegisters.h b/utils/TableGen/CodeGenRegisters.h index b0f8f4683ff..dce08b4ff53 100644 --- a/utils/TableGen/CodeGenRegisters.h +++ b/utils/TableGen/CodeGenRegisters.h @@ -113,6 +113,12 @@ namespace llvm { void addSubRegsPreOrder(SetVector &OSet, CodeGenRegBank&) const; + // Return the sub-register index naming Reg as a sub-register of this + // register. Returns NULL if Reg is not a sub-register. + CodeGenSubRegIndex *getSubRegIndex(const CodeGenRegister *Reg) const { + return SubReg2Idx.lookup(Reg); + } + // List of super-registers in topological order, small to large. typedef std::vector SuperRegList; @@ -157,6 +163,7 @@ namespace llvm { bool SubRegsComplete; SubRegMap SubRegs; SuperRegList SuperRegs; + DenseMap SubReg2Idx; RegUnitList RegUnits; };