[llvm-commits] [llvm] r43061 - /llvm/trunk/lib/CodeGen/RegAllocLinearScan.cpp

[Evan Cheng]

Author: evancheng
Date: Tue Oct 16 21:12:22 2007
New Revision: 43061

URL: http://llvm.org/viewvc/llvm-project?rev=43061&view=rev
Log:
Clean up code that calculate MBB live-in's.

Modified:
    llvm/trunk/lib/CodeGen/RegAllocLinearScan.cpp

Modified: llvm/trunk/lib/CodeGen/RegAllocLinearScan.cpp
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/lib/CodeGen/RegAllocLinearScan.cpp?rev=43061&r1=43060&r2=43061&view=diff

==============================================================================
--- llvm/trunk/lib/CodeGen/RegAllocLinearScan.cpp (original)
+++ llvm/trunk/lib/CodeGen/RegAllocLinearScan.cpp Tue Oct 16 21:12:22 2007
@@ -288,23 +288,25 @@
         DOUT << "\tinterval " << *i->first << " expired\n");
   inactive_.clear();
 
-  // A brute force way of adding live-ins to every BB.
-  MachineFunction::iterator MBB = mf_->begin();
-  ++MBB; // Skip entry MBB.
-  for (MachineFunction::iterator E = mf_->end(); MBB != E; ++MBB) {
-    unsigned StartIdx = li_->getMBBStartIdx(MBB->getNumber());
-    for (IntervalPtrs::iterator i = fixed_.begin(), e = fixed_.end();
-         i != e; ++i)
-      if (i->first->liveAt(StartIdx))
-        MBB->addLiveIn(i->first->reg);
-
-    for (unsigned i = 0, e = handled_.size(); i != e; ++i) { 
-      LiveInterval *HI = handled_[i];
-      unsigned Reg = HI->reg;
-      if (vrm_->isAssignedReg(Reg) && HI->liveAt(StartIdx)) {
-        assert(MRegisterInfo::isVirtualRegister(Reg));
-        Reg = vrm_->getPhys(Reg);
-        MBB->addLiveIn(Reg);
+  // Add live-ins to every BB except for entry.
+  MachineFunction::iterator EntryMBB = mf_->begin();
+  for (LiveIntervals::iterator i = li_->begin(), e = li_->end(); i != e; ++i) {
+    const LiveInterval &cur = i->second;
+    unsigned Reg = 0;
+    if (MRegisterInfo::isPhysicalRegister(cur.reg))
+      Reg = i->second.reg;
+    else if (vrm_->isAssignedReg(cur.reg))
+      Reg = vrm_->getPhys(cur.reg);
+    if (!Reg)
+      continue;
+    for (LiveInterval::Ranges::const_iterator I = cur.begin(), E = cur.end();
+         I != E; ++I) {
+      const LiveRange &LR = *I;
+      SmallVector<MachineBasicBlock*, 4> LiveInMBBs;
+      if (li_->findLiveInMBBs(LR, LiveInMBBs)) {
+        for (unsigned i = 0, e = LiveInMBBs.size(); i != e; ++i)
+          if (LiveInMBBs[i] != EntryMBB)
+            LiveInMBBs[i]->addLiveIn(Reg);
       }
     }
   }