From 3b6eb1e5ea1b090cefc5591b2b8b610e267d535d Mon Sep 17 00:00:00 2001
From: Andrew Trick <atrick@apple.com>
Date: Wed, 7 Mar 2012 00:18:05 +0000
Subject: misched: Use the StartBlock/FinishBlock hooks

llvm-svn: 152172
---
 llvm/lib/CodeGen/MachineScheduler.cpp | 2 ++
 1 file changed, 2 insertions(+)

(limited to 'llvm/lib/CodeGen/MachineScheduler.cpp')

diff --git a/llvm/lib/CodeGen/MachineScheduler.cpp b/llvm/lib/CodeGen/MachineScheduler.cpp
index 8a485e0fdef..55b931ff80f 100644
--- a/llvm/lib/CodeGen/MachineScheduler.cpp
+++ b/llvm/lib/CodeGen/MachineScheduler.cpp
@@ -264,6 +264,7 @@ bool MachineScheduler::runOnMachineFunction(MachineFunction &mf) {
     unsigned RemainingCount = MBB->size();
     for(MachineBasicBlock::iterator RegionEnd = MBB->end();
         RegionEnd != MBB->begin();) {
+      Scheduler->StartBlock(MBB);
       // The next region starts above the previous region. Look backward in the
       // instruction stream until we find the nearest boundary.
       MachineBasicBlock::iterator I = RegionEnd;
@@ -294,6 +295,7 @@ bool MachineScheduler::runOnMachineFunction(MachineFunction &mf) {
       RegionEnd = Scheduler->Begin;
     }
     assert(RemainingCount == 0 && "Instruction count mismatch!");
+    Scheduler->FinishBlock();
   }
   return true;
 }
-- 
cgit v1.2.3