From 0fb0678106c490b38343487ab121461f635875a0 Mon Sep 17 00:00:00 2001
From: Andrew Trick <atrick@apple.com>
Date: Tue, 10 Apr 2012 03:36:49 +0000
Subject: [PATCH] Fix for register pressure tables.

Recent refactoring introduced a bug. Fix: added buildRegUnitSets.

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@154382 91177308-0d34-0410-b5e6-96231b3b80d8
---
 utils/TableGen/CodeGenRegisters.cpp | 28 +++++++++++++++-------------
 1 file changed, 15 insertions(+), 13 deletions(-)

diff --git a/utils/TableGen/CodeGenRegisters.cpp b/utils/TableGen/CodeGenRegisters.cpp
index bd1eefe36f4..f071749ec6d 100644
--- a/utils/TableGen/CodeGenRegisters.cpp
+++ b/utils/TableGen/CodeGenRegisters.cpp
@@ -1118,6 +1118,17 @@ void CodeGenRegBank::computeRegUnitWeights() {
   }
 }
 
+// Populate a unique sorted list of units from a register set.
+static void buildRegUnitSet(const CodeGenRegister::Set &Regs,
+                            std::vector<unsigned> &RegUnits) {
+  std::vector<unsigned> TmpUnits;
+  for (RegUnitIterator UnitI(Regs); UnitI.isValid(); ++UnitI)
+    TmpUnits.push_back(*UnitI);
+  std::sort(TmpUnits.begin(), TmpUnits.end());
+  std::unique_copy(TmpUnits.begin(), TmpUnits.end(),
+                   std::back_inserter(RegUnits));
+}
+
 // Find a set in UniqueSets with the same elements as Set.
 // Return an iterator into UniqueSets.
 static std::vector<RegUnitSet>::const_iterator
@@ -1185,18 +1196,12 @@ void CodeGenRegBank::computeRegUnitSets() {
   unsigned NumRegClasses = RegClasses.size();
   for (unsigned RCIdx = 0, RCEnd = NumRegClasses; RCIdx != RCEnd; ++RCIdx) {
 
-    // Compute a sorted list of units in this class.
-    std::vector<unsigned> RegUnits;
-    const CodeGenRegister::Set &Regs = RegClasses[RCIdx]->getMembers();
-    for (RegUnitIterator UnitI(Regs); UnitI.isValid(); ++UnitI)
-      RegUnits.push_back(*UnitI);
-    std::sort(RegUnits.begin(), RegUnits.end());
-
     // Speculatively grow the RegUnitSets to hold the new set.
     RegUnitSets.resize(RegUnitSets.size() + 1);
     RegUnitSets.back().Name = RegClasses[RCIdx]->getName();
-    std::unique_copy(RegUnits.begin(), RegUnits.end(),
-                     std::back_inserter(RegUnitSets.back().Units));
+
+    // Compute a sorted list of units in this class.
+    buildRegUnitSet(RegClasses[RCIdx]->getMembers(), RegUnitSets.back().Units);
 
     // Find an existing RegUnitSet.
     std::vector<RegUnitSet>::const_iterator SetI =
@@ -1256,10 +1261,7 @@ void CodeGenRegBank::computeRegUnitSets() {
   for (unsigned RCIdx = 0, RCEnd = NumRegClasses; RCIdx != RCEnd; ++RCIdx) {
     // Recompute the sorted list of units in this class.
     std::vector<unsigned> RegUnits;
-    const CodeGenRegister::Set &Regs = RegClasses[RCIdx]->getMembers();
-    for (RegUnitIterator UnitI(Regs); UnitI.isValid(); ++UnitI)
-      RegUnits.push_back(*UnitI);
-    std::sort(RegUnits.begin(), RegUnits.end());
+    buildRegUnitSet(RegClasses[RCIdx]->getMembers(), RegUnits);
 
     // Don't increase pressure for unallocatable regclasses.
     if (RegUnits.empty())
-- 
2.34.1