diff --git a/include/llvm/CodeGen/CalcSpillWeights.h b/include/llvm/CodeGen/CalcSpillWeights.h index 99703a1c10a..a47d8477393 100644 --- a/include/llvm/CodeGen/CalcSpillWeights.h +++ b/include/llvm/CodeGen/CalcSpillWeights.h @@ -25,11 +25,11 @@ namespace llvm { class VirtRegAuxInfo { MachineFunction &mf_; LiveIntervals &lis_; - MachineLoopInfo &loops_; + const MachineLoopInfo &loops_; DenseMap hint_; public: VirtRegAuxInfo(MachineFunction &mf, LiveIntervals &lis, - MachineLoopInfo &loops) : + const MachineLoopInfo &loops) : mf_(mf), lis_(lis), loops_(loops) {} /// CalculateRegClass - recompute the register class for li from its uses. diff --git a/lib/CodeGen/SplitKit.cpp b/lib/CodeGen/SplitKit.cpp index e6a7b70c7d6..b7af73f627a 100644 --- a/lib/CodeGen/SplitKit.cpp +++ b/lib/CodeGen/SplitKit.cpp @@ -15,6 +15,7 @@ #define DEBUG_TYPE "splitter" #include "SplitKit.h" #include "VirtRegMap.h" +#include "llvm/CodeGen/CalcSpillWeights.h" #include "llvm/CodeGen/LiveIntervalAnalysis.h" #include "llvm/CodeGen/MachineFunctionPass.h" #include "llvm/CodeGen/MachineInstrBuilder.h" @@ -566,8 +567,12 @@ void SplitEditor::rewrite() { intervals_.push_back(dupli_); } - // FIXME: *Calculate spill weights, allocation hints, and register classes for - // firstInterval.. + // Calculate spill weight and allocation hints for new intervals. + VirtRegAuxInfo vrai(vrm_.getMachineFunction(), lis_, sa_.loops_); + for (unsigned i = firstInterval, e = intervals_.size(); i != e; ++i) { + LiveInterval &li = *intervals_[i]; + vrai.CalculateWeightAndHint(li); + } } diff --git a/lib/CodeGen/SplitKit.h b/lib/CodeGen/SplitKit.h index d5136877de6..d125a453437 100644 --- a/lib/CodeGen/SplitKit.h +++ b/lib/CodeGen/SplitKit.h @@ -31,11 +31,13 @@ class VNInfo; /// SplitAnalysis - Analyze a LiveInterval, looking for live range splitting /// opportunities. class SplitAnalysis { +public: const MachineFunction &mf_; const LiveIntervals &lis_; const MachineLoopInfo &loops_; const TargetInstrInfo &tii_; +private: // Current live interval. const LiveInterval *curli_;