1dff0c46cSDimitry Andric //===-- InterferenceCache.cpp - Caching per-block interference ---------*--===//
23b0f4066SDimitry Andric //
33b0f4066SDimitry Andric //                     The LLVM Compiler Infrastructure
43b0f4066SDimitry Andric //
53b0f4066SDimitry Andric // This file is distributed under the University of Illinois Open Source
63b0f4066SDimitry Andric // License. See LICENSE.TXT for details.
73b0f4066SDimitry Andric //
83b0f4066SDimitry Andric //===----------------------------------------------------------------------===//
93b0f4066SDimitry Andric //
103b0f4066SDimitry Andric // InterferenceCache remembers per-block interference in LiveIntervalUnions.
113b0f4066SDimitry Andric //
123b0f4066SDimitry Andric //===----------------------------------------------------------------------===//
133b0f4066SDimitry Andric 
143b0f4066SDimitry Andric #define DEBUG_TYPE "regalloc"
153b0f4066SDimitry Andric #include "InterferenceCache.h"
16dff0c46cSDimitry Andric #include "llvm/CodeGen/LiveIntervalAnalysis.h"
17139f7f9bSDimitry Andric #include "llvm/Support/ErrorHandling.h"
18139f7f9bSDimitry Andric #include "llvm/Target/TargetRegisterInfo.h"
193b0f4066SDimitry Andric 
203b0f4066SDimitry Andric using namespace llvm;
213b0f4066SDimitry Andric 
226122f3e6SDimitry Andric // Static member used for null interference cursors.
236122f3e6SDimitry Andric InterferenceCache::BlockInterference InterferenceCache::Cursor::NoInterference;
246122f3e6SDimitry Andric 
253b0f4066SDimitry Andric void InterferenceCache::init(MachineFunction *mf,
263b0f4066SDimitry Andric                              LiveIntervalUnion *liuarray,
273b0f4066SDimitry Andric                              SlotIndexes *indexes,
28dff0c46cSDimitry Andric                              LiveIntervals *lis,
293b0f4066SDimitry Andric                              const TargetRegisterInfo *tri) {
303b0f4066SDimitry Andric   MF = mf;
313b0f4066SDimitry Andric   LIUArray = liuarray;
323b0f4066SDimitry Andric   TRI = tri;
333b0f4066SDimitry Andric   PhysRegEntries.assign(TRI->getNumRegs(), 0);
343b0f4066SDimitry Andric   for (unsigned i = 0; i != CacheEntries; ++i)
35dff0c46cSDimitry Andric     Entries[i].clear(mf, indexes, lis);
363b0f4066SDimitry Andric }
373b0f4066SDimitry Andric 
383b0f4066SDimitry Andric InterferenceCache::Entry *InterferenceCache::get(unsigned PhysReg) {
393b0f4066SDimitry Andric   unsigned E = PhysRegEntries[PhysReg];
403b0f4066SDimitry Andric   if (E < CacheEntries && Entries[E].getPhysReg() == PhysReg) {
413b0f4066SDimitry Andric     if (!Entries[E].valid(LIUArray, TRI))
427ae0e2c9SDimitry Andric       Entries[E].revalidate(LIUArray, TRI);
433b0f4066SDimitry Andric     return &Entries[E];
443b0f4066SDimitry Andric   }
453b0f4066SDimitry Andric   // No valid entry exists, pick the next round-robin entry.
463b0f4066SDimitry Andric   E = RoundRobin;
473b0f4066SDimitry Andric   if (++RoundRobin == CacheEntries)
483b0f4066SDimitry Andric     RoundRobin = 0;
4917a519f9SDimitry Andric   for (unsigned i = 0; i != CacheEntries; ++i) {
5017a519f9SDimitry Andric     // Skip entries that are in use.
5117a519f9SDimitry Andric     if (Entries[E].hasRefs()) {
5217a519f9SDimitry Andric       if (++E == CacheEntries)
5317a519f9SDimitry Andric         E = 0;
5417a519f9SDimitry Andric       continue;
5517a519f9SDimitry Andric     }
563b0f4066SDimitry Andric     Entries[E].reset(PhysReg, LIUArray, TRI, MF);
573b0f4066SDimitry Andric     PhysRegEntries[PhysReg] = E;
583b0f4066SDimitry Andric     return &Entries[E];
593b0f4066SDimitry Andric   }
6017a519f9SDimitry Andric   llvm_unreachable("Ran out of interference cache entries.");
6117a519f9SDimitry Andric }
623b0f4066SDimitry Andric 
633b0f4066SDimitry Andric /// revalidate - LIU contents have changed, update tags.
647ae0e2c9SDimitry Andric void InterferenceCache::Entry::revalidate(LiveIntervalUnion *LIUArray,
657ae0e2c9SDimitry Andric                                           const TargetRegisterInfo *TRI) {
663b0f4066SDimitry Andric   // Invalidate all block entries.
673b0f4066SDimitry Andric   ++Tag;
683b0f4066SDimitry Andric   // Invalidate all iterators.
693b0f4066SDimitry Andric   PrevPos = SlotIndex();
707ae0e2c9SDimitry Andric   unsigned i = 0;
717ae0e2c9SDimitry Andric   for (MCRegUnitIterator Units(PhysReg, TRI); Units.isValid(); ++Units, ++i)
727ae0e2c9SDimitry Andric     RegUnits[i].VirtTag = LIUArray[*Units].getTag();
733b0f4066SDimitry Andric }
743b0f4066SDimitry Andric 
753b0f4066SDimitry Andric void InterferenceCache::Entry::reset(unsigned physReg,
763b0f4066SDimitry Andric                                      LiveIntervalUnion *LIUArray,
773b0f4066SDimitry Andric                                      const TargetRegisterInfo *TRI,
783b0f4066SDimitry Andric                                      const MachineFunction *MF) {
7917a519f9SDimitry Andric   assert(!hasRefs() && "Cannot reset cache entry with references");
803b0f4066SDimitry Andric   // LIU's changed, invalidate cache.
813b0f4066SDimitry Andric   ++Tag;
823b0f4066SDimitry Andric   PhysReg = physReg;
833b0f4066SDimitry Andric   Blocks.resize(MF->getNumBlockIDs());
843b0f4066SDimitry Andric 
853b0f4066SDimitry Andric   // Reset iterators.
863b0f4066SDimitry Andric   PrevPos = SlotIndex();
877ae0e2c9SDimitry Andric   RegUnits.clear();
887ae0e2c9SDimitry Andric   for (MCRegUnitIterator Units(PhysReg, TRI); Units.isValid(); ++Units) {
897ae0e2c9SDimitry Andric     RegUnits.push_back(LIUArray[*Units]);
907ae0e2c9SDimitry Andric     RegUnits.back().Fixed = &LIS->getRegUnit(*Units);
917ae0e2c9SDimitry Andric   }
923b0f4066SDimitry Andric }
933b0f4066SDimitry Andric 
943b0f4066SDimitry Andric bool InterferenceCache::Entry::valid(LiveIntervalUnion *LIUArray,
953b0f4066SDimitry Andric                                      const TargetRegisterInfo *TRI) {
967ae0e2c9SDimitry Andric   unsigned i = 0, e = RegUnits.size();
977ae0e2c9SDimitry Andric   for (MCRegUnitIterator Units(PhysReg, TRI); Units.isValid(); ++Units, ++i) {
987ae0e2c9SDimitry Andric     if (i == e)
993b0f4066SDimitry Andric       return false;
1007ae0e2c9SDimitry Andric     if (LIUArray[*Units].changedSince(RegUnits[i].VirtTag))
1013b0f4066SDimitry Andric       return false;
1023b0f4066SDimitry Andric   }
1033b0f4066SDimitry Andric   return i == e;
1043b0f4066SDimitry Andric }
1053b0f4066SDimitry Andric 
1063b0f4066SDimitry Andric void InterferenceCache::Entry::update(unsigned MBBNum) {
1073b0f4066SDimitry Andric   SlotIndex Start, Stop;
1083b0f4066SDimitry Andric   tie(Start, Stop) = Indexes->getMBBRange(MBBNum);
1093b0f4066SDimitry Andric 
1103b0f4066SDimitry Andric   // Use advanceTo only when possible.
1113b0f4066SDimitry Andric   if (PrevPos != Start) {
1127ae0e2c9SDimitry Andric     if (!PrevPos.isValid() || Start < PrevPos) {
1137ae0e2c9SDimitry Andric       for (unsigned i = 0, e = RegUnits.size(); i != e; ++i) {
1147ae0e2c9SDimitry Andric         RegUnitInfo &RUI = RegUnits[i];
1157ae0e2c9SDimitry Andric         RUI.VirtI.find(Start);
1167ae0e2c9SDimitry Andric         RUI.FixedI = RUI.Fixed->find(Start);
1177ae0e2c9SDimitry Andric       }
1187ae0e2c9SDimitry Andric     } else {
1197ae0e2c9SDimitry Andric       for (unsigned i = 0, e = RegUnits.size(); i != e; ++i) {
1207ae0e2c9SDimitry Andric         RegUnitInfo &RUI = RegUnits[i];
1217ae0e2c9SDimitry Andric         RUI.VirtI.advanceTo(Start);
1227ae0e2c9SDimitry Andric         if (RUI.FixedI != RUI.Fixed->end())
1237ae0e2c9SDimitry Andric           RUI.FixedI = RUI.Fixed->advanceTo(RUI.FixedI, Start);
1247ae0e2c9SDimitry Andric       }
1257ae0e2c9SDimitry Andric     }
1263b0f4066SDimitry Andric     PrevPos = Start;
1273b0f4066SDimitry Andric   }
1283b0f4066SDimitry Andric 
1293b0f4066SDimitry Andric   MachineFunction::const_iterator MFI = MF->getBlockNumbered(MBBNum);
1303b0f4066SDimitry Andric   BlockInterference *BI = &Blocks[MBBNum];
131dff0c46cSDimitry Andric   ArrayRef<SlotIndex> RegMaskSlots;
132dff0c46cSDimitry Andric   ArrayRef<const uint32_t*> RegMaskBits;
1333b0f4066SDimitry Andric   for (;;) {
1343b0f4066SDimitry Andric     BI->Tag = Tag;
1353b0f4066SDimitry Andric     BI->First = BI->Last = SlotIndex();
1363b0f4066SDimitry Andric 
1377ae0e2c9SDimitry Andric     // Check for first interference from virtregs.
1387ae0e2c9SDimitry Andric     for (unsigned i = 0, e = RegUnits.size(); i != e; ++i) {
1397ae0e2c9SDimitry Andric       LiveIntervalUnion::SegmentIter &I = RegUnits[i].VirtI;
1403b0f4066SDimitry Andric       if (!I.valid())
1413b0f4066SDimitry Andric         continue;
1423b0f4066SDimitry Andric       SlotIndex StartI = I.start();
1433b0f4066SDimitry Andric       if (StartI >= Stop)
1443b0f4066SDimitry Andric         continue;
1453b0f4066SDimitry Andric       if (!BI->First.isValid() || StartI < BI->First)
1463b0f4066SDimitry Andric         BI->First = StartI;
1473b0f4066SDimitry Andric     }
1483b0f4066SDimitry Andric 
1497ae0e2c9SDimitry Andric     // Same thing for fixed interference.
1507ae0e2c9SDimitry Andric     for (unsigned i = 0, e = RegUnits.size(); i != e; ++i) {
1517ae0e2c9SDimitry Andric       LiveInterval::const_iterator I = RegUnits[i].FixedI;
1527ae0e2c9SDimitry Andric       LiveInterval::const_iterator E = RegUnits[i].Fixed->end();
1537ae0e2c9SDimitry Andric       if (I == E)
1547ae0e2c9SDimitry Andric         continue;
1557ae0e2c9SDimitry Andric       SlotIndex StartI = I->start;
1567ae0e2c9SDimitry Andric       if (StartI >= Stop)
1577ae0e2c9SDimitry Andric         continue;
1587ae0e2c9SDimitry Andric       if (!BI->First.isValid() || StartI < BI->First)
1597ae0e2c9SDimitry Andric         BI->First = StartI;
1607ae0e2c9SDimitry Andric     }
1617ae0e2c9SDimitry Andric 
162dff0c46cSDimitry Andric     // Also check for register mask interference.
163dff0c46cSDimitry Andric     RegMaskSlots = LIS->getRegMaskSlotsInBlock(MBBNum);
164dff0c46cSDimitry Andric     RegMaskBits = LIS->getRegMaskBitsInBlock(MBBNum);
165dff0c46cSDimitry Andric     SlotIndex Limit = BI->First.isValid() ? BI->First : Stop;
166dff0c46cSDimitry Andric     for (unsigned i = 0, e = RegMaskSlots.size();
167dff0c46cSDimitry Andric          i != e && RegMaskSlots[i] < Limit; ++i)
168dff0c46cSDimitry Andric       if (MachineOperand::clobbersPhysReg(RegMaskBits[i], PhysReg)) {
169dff0c46cSDimitry Andric         // Register mask i clobbers PhysReg before the LIU interference.
170dff0c46cSDimitry Andric         BI->First = RegMaskSlots[i];
171dff0c46cSDimitry Andric         break;
172dff0c46cSDimitry Andric       }
173dff0c46cSDimitry Andric 
1743b0f4066SDimitry Andric     PrevPos = Stop;
1753b0f4066SDimitry Andric     if (BI->First.isValid())
1763b0f4066SDimitry Andric       break;
1773b0f4066SDimitry Andric 
1783b0f4066SDimitry Andric     // No interference in this block? Go ahead and precompute the next block.
1793b0f4066SDimitry Andric     if (++MFI == MF->end())
1803b0f4066SDimitry Andric       return;
1813b0f4066SDimitry Andric     MBBNum = MFI->getNumber();
1823b0f4066SDimitry Andric     BI = &Blocks[MBBNum];
1833b0f4066SDimitry Andric     if (BI->Tag == Tag)
1843b0f4066SDimitry Andric       return;
1853b0f4066SDimitry Andric     tie(Start, Stop) = Indexes->getMBBRange(MBBNum);
1863b0f4066SDimitry Andric   }
1873b0f4066SDimitry Andric 
1883b0f4066SDimitry Andric   // Check for last interference in block.
1897ae0e2c9SDimitry Andric   for (unsigned i = 0, e = RegUnits.size(); i != e; ++i) {
1907ae0e2c9SDimitry Andric     LiveIntervalUnion::SegmentIter &I = RegUnits[i].VirtI;
1913b0f4066SDimitry Andric     if (!I.valid() || I.start() >= Stop)
1923b0f4066SDimitry Andric       continue;
1933b0f4066SDimitry Andric     I.advanceTo(Stop);
1943b0f4066SDimitry Andric     bool Backup = !I.valid() || I.start() >= Stop;
1953b0f4066SDimitry Andric     if (Backup)
1963b0f4066SDimitry Andric       --I;
1973b0f4066SDimitry Andric     SlotIndex StopI = I.stop();
1983b0f4066SDimitry Andric     if (!BI->Last.isValid() || StopI > BI->Last)
1993b0f4066SDimitry Andric       BI->Last = StopI;
2003b0f4066SDimitry Andric     if (Backup)
2013b0f4066SDimitry Andric       ++I;
2023b0f4066SDimitry Andric   }
203dff0c46cSDimitry Andric 
2047ae0e2c9SDimitry Andric   // Fixed interference.
2057ae0e2c9SDimitry Andric   for (unsigned i = 0, e = RegUnits.size(); i != e; ++i) {
2067ae0e2c9SDimitry Andric     LiveInterval::iterator &I = RegUnits[i].FixedI;
207f785676fSDimitry Andric     LiveRange *LR = RegUnits[i].Fixed;
208f785676fSDimitry Andric     if (I == LR->end() || I->start >= Stop)
2097ae0e2c9SDimitry Andric       continue;
210f785676fSDimitry Andric     I = LR->advanceTo(I, Stop);
211f785676fSDimitry Andric     bool Backup = I == LR->end() || I->start >= Stop;
2127ae0e2c9SDimitry Andric     if (Backup)
2137ae0e2c9SDimitry Andric       --I;
2147ae0e2c9SDimitry Andric     SlotIndex StopI = I->end;
2157ae0e2c9SDimitry Andric     if (!BI->Last.isValid() || StopI > BI->Last)
2167ae0e2c9SDimitry Andric       BI->Last = StopI;
2177ae0e2c9SDimitry Andric     if (Backup)
2187ae0e2c9SDimitry Andric       ++I;
2197ae0e2c9SDimitry Andric   }
2207ae0e2c9SDimitry Andric 
221dff0c46cSDimitry Andric   // Also check for register mask interference.
222dff0c46cSDimitry Andric   SlotIndex Limit = BI->Last.isValid() ? BI->Last : Start;
223dff0c46cSDimitry Andric   for (unsigned i = RegMaskSlots.size();
224dff0c46cSDimitry Andric        i && RegMaskSlots[i-1].getDeadSlot() > Limit; --i)
225dff0c46cSDimitry Andric     if (MachineOperand::clobbersPhysReg(RegMaskBits[i-1], PhysReg)) {
226dff0c46cSDimitry Andric       // Register mask i-1 clobbers PhysReg after the LIU interference.
227dff0c46cSDimitry Andric       // Model the regmask clobber as a dead def.
228dff0c46cSDimitry Andric       BI->Last = RegMaskSlots[i-1].getDeadSlot();
229dff0c46cSDimitry Andric       break;
230dff0c46cSDimitry Andric     }
2313b0f4066SDimitry Andric }
232