From 680363b5fdf7ce6e188d4c8c7508fc819061ba6d Mon Sep 17 00:00:00 2001
From: Andrew Trick <atrick@apple.com>
Date: Wed, 11 Apr 2012 04:31:33 +0000
Subject: [PATCH] Tablegen'd regpressure: emit the weighted pressure limit.

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@154477 91177308-0d34-0410-b5e6-96231b3b80d8
---
 utils/TableGen/RegisterInfoEmitter.cpp | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/utils/TableGen/RegisterInfoEmitter.cpp b/utils/TableGen/RegisterInfoEmitter.cpp
index ac89e258431..8085955baf1 100644
--- a/utils/TableGen/RegisterInfoEmitter.cpp
+++ b/utils/TableGen/RegisterInfoEmitter.cpp
@@ -152,7 +152,13 @@ EmitRegUnitPressure(raw_ostream &OS, const CodeGenRegBank &RegBank,
      << "getRegPressureSetLimit(unsigned Idx) const {\n"
      << "  static const unsigned PressureLimitTable[] = {\n";
   for (unsigned i = 0; i < NumSets; ++i ) {
-    OS << "    " << RegBank.getRegPressureSet(i).Units.size()
+    const RegUnitSet &RegUnits = RegBank.getRegPressureSet(i);
+    unsigned Weight = 0;
+    for (RegUnitSet::iterator
+           I = RegUnits.Units.begin(), E = RegUnits.Units.end(); I != E; ++I) {
+      Weight += RegBank.getRegUnitWeight(*I);
+    }
+    OS << "    " << Weight
        << ",  \t// " << i << ": " << RegBank.getRegPressureSet(i).Name << "\n";
   }
   OS << "    0 };\n"