Recommit [NFC] Better encapsulation of llvm::Optional Storage
[llvm-complete.git] / include / llvm / CodeGen / LiveInterval.h
blob622c1248b36a7857a851b7af60fbccc59e7935cc
1 //===- llvm/CodeGen/LiveInterval.h - Interval representation ----*- C++ -*-===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 //
9 // This file implements the LiveRange and LiveInterval classes. Given some
10 // numbering of each the machine instructions an interval [i, j) is said to be a
11 // live range for register v if there is no instruction with number j' >= j
12 // such that v is live at j' and there is no instruction with number i' < i such
13 // that v is live at i'. In this implementation ranges can have holes,
14 // i.e. a range might look like [1,20), [50,65), [1000,1001). Each
15 // individual segment is represented as an instance of LiveRange::Segment,
16 // and the whole range is represented as an instance of LiveRange.
18 //===----------------------------------------------------------------------===//
20 #ifndef LLVM_CODEGEN_LIVEINTERVAL_H
21 #define LLVM_CODEGEN_LIVEINTERVAL_H
23 #include "llvm/ADT/ArrayRef.h"
24 #include "llvm/ADT/IntEqClasses.h"
25 #include "llvm/ADT/STLExtras.h"
26 #include "llvm/ADT/SmallVector.h"
27 #include "llvm/ADT/iterator_range.h"
28 #include "llvm/CodeGen/SlotIndexes.h"
29 #include "llvm/MC/LaneBitmask.h"
30 #include "llvm/Support/Allocator.h"
31 #include "llvm/Support/MathExtras.h"
32 #include <algorithm>
33 #include <cassert>
34 #include <cstddef>
35 #include <functional>
36 #include <memory>
37 #include <set>
38 #include <tuple>
39 #include <utility>
41 namespace llvm {
43 class CoalescerPair;
44 class LiveIntervals;
45 class MachineRegisterInfo;
46 class raw_ostream;
48 /// VNInfo - Value Number Information.
49 /// This class holds information about a machine level values, including
50 /// definition and use points.
51 ///
52 class VNInfo {
53 public:
54 using Allocator = BumpPtrAllocator;
56 /// The ID number of this value.
57 unsigned id;
59 /// The index of the defining instruction.
60 SlotIndex def;
62 /// VNInfo constructor.
63 VNInfo(unsigned i, SlotIndex d) : id(i), def(d) {}
65 /// VNInfo constructor, copies values from orig, except for the value number.
66 VNInfo(unsigned i, const VNInfo &orig) : id(i), def(orig.def) {}
68 /// Copy from the parameter into this VNInfo.
69 void copyFrom(VNInfo &src) {
70 def = src.def;
73 /// Returns true if this value is defined by a PHI instruction (or was,
74 /// PHI instructions may have been eliminated).
75 /// PHI-defs begin at a block boundary, all other defs begin at register or
76 /// EC slots.
77 bool isPHIDef() const { return def.isBlock(); }
79 /// Returns true if this value is unused.
80 bool isUnused() const { return !def.isValid(); }
82 /// Mark this value as unused.
83 void markUnused() { def = SlotIndex(); }
86 /// Result of a LiveRange query. This class hides the implementation details
87 /// of live ranges, and it should be used as the primary interface for
88 /// examining live ranges around instructions.
89 class LiveQueryResult {
90 VNInfo *const EarlyVal;
91 VNInfo *const LateVal;
92 const SlotIndex EndPoint;
93 const bool Kill;
95 public:
96 LiveQueryResult(VNInfo *EarlyVal, VNInfo *LateVal, SlotIndex EndPoint,
97 bool Kill)
98 : EarlyVal(EarlyVal), LateVal(LateVal), EndPoint(EndPoint), Kill(Kill)
101 /// Return the value that is live-in to the instruction. This is the value
102 /// that will be read by the instruction's use operands. Return NULL if no
103 /// value is live-in.
104 VNInfo *valueIn() const {
105 return EarlyVal;
108 /// Return true if the live-in value is killed by this instruction. This
109 /// means that either the live range ends at the instruction, or it changes
110 /// value.
111 bool isKill() const {
112 return Kill;
115 /// Return true if this instruction has a dead def.
116 bool isDeadDef() const {
117 return EndPoint.isDead();
120 /// Return the value leaving the instruction, if any. This can be a
121 /// live-through value, or a live def. A dead def returns NULL.
122 VNInfo *valueOut() const {
123 return isDeadDef() ? nullptr : LateVal;
126 /// Returns the value alive at the end of the instruction, if any. This can
127 /// be a live-through value, a live def or a dead def.
128 VNInfo *valueOutOrDead() const {
129 return LateVal;
132 /// Return the value defined by this instruction, if any. This includes
133 /// dead defs, it is the value created by the instruction's def operands.
134 VNInfo *valueDefined() const {
135 return EarlyVal == LateVal ? nullptr : LateVal;
138 /// Return the end point of the last live range segment to interact with
139 /// the instruction, if any.
141 /// The end point is an invalid SlotIndex only if the live range doesn't
142 /// intersect the instruction at all.
144 /// The end point may be at or past the end of the instruction's basic
145 /// block. That means the value was live out of the block.
146 SlotIndex endPoint() const {
147 return EndPoint;
151 /// This class represents the liveness of a register, stack slot, etc.
152 /// It manages an ordered list of Segment objects.
153 /// The Segments are organized in a static single assignment form: At places
154 /// where a new value is defined or different values reach a CFG join a new
155 /// segment with a new value number is used.
156 class LiveRange {
157 public:
158 /// This represents a simple continuous liveness interval for a value.
159 /// The start point is inclusive, the end point exclusive. These intervals
160 /// are rendered as [start,end).
161 struct Segment {
162 SlotIndex start; // Start point of the interval (inclusive)
163 SlotIndex end; // End point of the interval (exclusive)
164 VNInfo *valno = nullptr; // identifier for the value contained in this
165 // segment.
167 Segment() = default;
169 Segment(SlotIndex S, SlotIndex E, VNInfo *V)
170 : start(S), end(E), valno(V) {
171 assert(S < E && "Cannot create empty or backwards segment");
174 /// Return true if the index is covered by this segment.
175 bool contains(SlotIndex I) const {
176 return start <= I && I < end;
179 /// Return true if the given interval, [S, E), is covered by this segment.
180 bool containsInterval(SlotIndex S, SlotIndex E) const {
181 assert((S < E) && "Backwards interval?");
182 return (start <= S && S < end) && (start < E && E <= end);
185 bool operator<(const Segment &Other) const {
186 return std::tie(start, end) < std::tie(Other.start, Other.end);
188 bool operator==(const Segment &Other) const {
189 return start == Other.start && end == Other.end;
192 void dump() const;
195 using Segments = SmallVector<Segment, 2>;
196 using VNInfoList = SmallVector<VNInfo *, 2>;
198 Segments segments; // the liveness segments
199 VNInfoList valnos; // value#'s
201 // The segment set is used temporarily to accelerate initial computation
202 // of live ranges of physical registers in computeRegUnitRange.
203 // After that the set is flushed to the segment vector and deleted.
204 using SegmentSet = std::set<Segment>;
205 std::unique_ptr<SegmentSet> segmentSet;
207 using iterator = Segments::iterator;
208 using const_iterator = Segments::const_iterator;
210 iterator begin() { return segments.begin(); }
211 iterator end() { return segments.end(); }
213 const_iterator begin() const { return segments.begin(); }
214 const_iterator end() const { return segments.end(); }
216 using vni_iterator = VNInfoList::iterator;
217 using const_vni_iterator = VNInfoList::const_iterator;
219 vni_iterator vni_begin() { return valnos.begin(); }
220 vni_iterator vni_end() { return valnos.end(); }
222 const_vni_iterator vni_begin() const { return valnos.begin(); }
223 const_vni_iterator vni_end() const { return valnos.end(); }
225 /// Constructs a new LiveRange object.
226 LiveRange(bool UseSegmentSet = false)
227 : segmentSet(UseSegmentSet ? llvm::make_unique<SegmentSet>()
228 : nullptr) {}
230 /// Constructs a new LiveRange object by copying segments and valnos from
231 /// another LiveRange.
232 LiveRange(const LiveRange &Other, BumpPtrAllocator &Allocator) {
233 assert(Other.segmentSet == nullptr &&
234 "Copying of LiveRanges with active SegmentSets is not supported");
235 assign(Other, Allocator);
238 /// Copies values numbers and live segments from \p Other into this range.
239 void assign(const LiveRange &Other, BumpPtrAllocator &Allocator) {
240 if (this == &Other)
241 return;
243 assert(Other.segmentSet == nullptr &&
244 "Copying of LiveRanges with active SegmentSets is not supported");
245 // Duplicate valnos.
246 for (const VNInfo *VNI : Other.valnos)
247 createValueCopy(VNI, Allocator);
248 // Now we can copy segments and remap their valnos.
249 for (const Segment &S : Other.segments)
250 segments.push_back(Segment(S.start, S.end, valnos[S.valno->id]));
253 /// advanceTo - Advance the specified iterator to point to the Segment
254 /// containing the specified position, or end() if the position is past the
255 /// end of the range. If no Segment contains this position, but the
256 /// position is in a hole, this method returns an iterator pointing to the
257 /// Segment immediately after the hole.
258 iterator advanceTo(iterator I, SlotIndex Pos) {
259 assert(I != end());
260 if (Pos >= endIndex())
261 return end();
262 while (I->end <= Pos) ++I;
263 return I;
266 const_iterator advanceTo(const_iterator I, SlotIndex Pos) const {
267 assert(I != end());
268 if (Pos >= endIndex())
269 return end();
270 while (I->end <= Pos) ++I;
271 return I;
274 /// find - Return an iterator pointing to the first segment that ends after
275 /// Pos, or end(). This is the same as advanceTo(begin(), Pos), but faster
276 /// when searching large ranges.
278 /// If Pos is contained in a Segment, that segment is returned.
279 /// If Pos is in a hole, the following Segment is returned.
280 /// If Pos is beyond endIndex, end() is returned.
281 iterator find(SlotIndex Pos);
283 const_iterator find(SlotIndex Pos) const {
284 return const_cast<LiveRange*>(this)->find(Pos);
287 void clear() {
288 valnos.clear();
289 segments.clear();
292 size_t size() const {
293 return segments.size();
296 bool hasAtLeastOneValue() const { return !valnos.empty(); }
298 bool containsOneValue() const { return valnos.size() == 1; }
300 unsigned getNumValNums() const { return (unsigned)valnos.size(); }
302 /// getValNumInfo - Returns pointer to the specified val#.
304 inline VNInfo *getValNumInfo(unsigned ValNo) {
305 return valnos[ValNo];
307 inline const VNInfo *getValNumInfo(unsigned ValNo) const {
308 return valnos[ValNo];
311 /// containsValue - Returns true if VNI belongs to this range.
312 bool containsValue(const VNInfo *VNI) const {
313 return VNI && VNI->id < getNumValNums() && VNI == getValNumInfo(VNI->id);
316 /// getNextValue - Create a new value number and return it. MIIdx specifies
317 /// the instruction that defines the value number.
318 VNInfo *getNextValue(SlotIndex def, VNInfo::Allocator &VNInfoAllocator) {
319 VNInfo *VNI =
320 new (VNInfoAllocator) VNInfo((unsigned)valnos.size(), def);
321 valnos.push_back(VNI);
322 return VNI;
325 /// createDeadDef - Make sure the range has a value defined at Def.
326 /// If one already exists, return it. Otherwise allocate a new value and
327 /// add liveness for a dead def.
328 VNInfo *createDeadDef(SlotIndex Def, VNInfo::Allocator &VNIAlloc);
330 /// Create a def of value @p VNI. Return @p VNI. If there already exists
331 /// a definition at VNI->def, the value defined there must be @p VNI.
332 VNInfo *createDeadDef(VNInfo *VNI);
334 /// Create a copy of the given value. The new value will be identical except
335 /// for the Value number.
336 VNInfo *createValueCopy(const VNInfo *orig,
337 VNInfo::Allocator &VNInfoAllocator) {
338 VNInfo *VNI =
339 new (VNInfoAllocator) VNInfo((unsigned)valnos.size(), *orig);
340 valnos.push_back(VNI);
341 return VNI;
344 /// RenumberValues - Renumber all values in order of appearance and remove
345 /// unused values.
346 void RenumberValues();
348 /// MergeValueNumberInto - This method is called when two value numbers
349 /// are found to be equivalent. This eliminates V1, replacing all
350 /// segments with the V1 value number with the V2 value number. This can
351 /// cause merging of V1/V2 values numbers and compaction of the value space.
352 VNInfo* MergeValueNumberInto(VNInfo *V1, VNInfo *V2);
354 /// Merge all of the live segments of a specific val# in RHS into this live
355 /// range as the specified value number. The segments in RHS are allowed
356 /// to overlap with segments in the current range, it will replace the
357 /// value numbers of the overlaped live segments with the specified value
358 /// number.
359 void MergeSegmentsInAsValue(const LiveRange &RHS, VNInfo *LHSValNo);
361 /// MergeValueInAsValue - Merge all of the segments of a specific val#
362 /// in RHS into this live range as the specified value number.
363 /// The segments in RHS are allowed to overlap with segments in the
364 /// current range, but only if the overlapping segments have the
365 /// specified value number.
366 void MergeValueInAsValue(const LiveRange &RHS,
367 const VNInfo *RHSValNo, VNInfo *LHSValNo);
369 bool empty() const { return segments.empty(); }
371 /// beginIndex - Return the lowest numbered slot covered.
372 SlotIndex beginIndex() const {
373 assert(!empty() && "Call to beginIndex() on empty range.");
374 return segments.front().start;
377 /// endNumber - return the maximum point of the range of the whole,
378 /// exclusive.
379 SlotIndex endIndex() const {
380 assert(!empty() && "Call to endIndex() on empty range.");
381 return segments.back().end;
384 bool expiredAt(SlotIndex index) const {
385 return index >= endIndex();
388 bool liveAt(SlotIndex index) const {
389 const_iterator r = find(index);
390 return r != end() && r->start <= index;
393 /// Return the segment that contains the specified index, or null if there
394 /// is none.
395 const Segment *getSegmentContaining(SlotIndex Idx) const {
396 const_iterator I = FindSegmentContaining(Idx);
397 return I == end() ? nullptr : &*I;
400 /// Return the live segment that contains the specified index, or null if
401 /// there is none.
402 Segment *getSegmentContaining(SlotIndex Idx) {
403 iterator I = FindSegmentContaining(Idx);
404 return I == end() ? nullptr : &*I;
407 /// getVNInfoAt - Return the VNInfo that is live at Idx, or NULL.
408 VNInfo *getVNInfoAt(SlotIndex Idx) const {
409 const_iterator I = FindSegmentContaining(Idx);
410 return I == end() ? nullptr : I->valno;
413 /// getVNInfoBefore - Return the VNInfo that is live up to but not
414 /// necessarilly including Idx, or NULL. Use this to find the reaching def
415 /// used by an instruction at this SlotIndex position.
416 VNInfo *getVNInfoBefore(SlotIndex Idx) const {
417 const_iterator I = FindSegmentContaining(Idx.getPrevSlot());
418 return I == end() ? nullptr : I->valno;
421 /// Return an iterator to the segment that contains the specified index, or
422 /// end() if there is none.
423 iterator FindSegmentContaining(SlotIndex Idx) {
424 iterator I = find(Idx);
425 return I != end() && I->start <= Idx ? I : end();
428 const_iterator FindSegmentContaining(SlotIndex Idx) const {
429 const_iterator I = find(Idx);
430 return I != end() && I->start <= Idx ? I : end();
433 /// overlaps - Return true if the intersection of the two live ranges is
434 /// not empty.
435 bool overlaps(const LiveRange &other) const {
436 if (other.empty())
437 return false;
438 return overlapsFrom(other, other.begin());
441 /// overlaps - Return true if the two ranges have overlapping segments
442 /// that are not coalescable according to CP.
444 /// Overlapping segments where one range is defined by a coalescable
445 /// copy are allowed.
446 bool overlaps(const LiveRange &Other, const CoalescerPair &CP,
447 const SlotIndexes&) const;
449 /// overlaps - Return true if the live range overlaps an interval specified
450 /// by [Start, End).
451 bool overlaps(SlotIndex Start, SlotIndex End) const;
453 /// overlapsFrom - Return true if the intersection of the two live ranges
454 /// is not empty. The specified iterator is a hint that we can begin
455 /// scanning the Other range starting at I.
456 bool overlapsFrom(const LiveRange &Other, const_iterator StartPos) const;
458 /// Returns true if all segments of the @p Other live range are completely
459 /// covered by this live range.
460 /// Adjacent live ranges do not affect the covering:the liverange
461 /// [1,5](5,10] covers (3,7].
462 bool covers(const LiveRange &Other) const;
464 /// Add the specified Segment to this range, merging segments as
465 /// appropriate. This returns an iterator to the inserted segment (which
466 /// may have grown since it was inserted).
467 iterator addSegment(Segment S);
469 /// Attempt to extend a value defined after @p StartIdx to include @p Use.
470 /// Both @p StartIdx and @p Use should be in the same basic block. In case
471 /// of subranges, an extension could be prevented by an explicit "undef"
472 /// caused by a <def,read-undef> on a non-overlapping lane. The list of
473 /// location of such "undefs" should be provided in @p Undefs.
474 /// The return value is a pair: the first element is VNInfo of the value
475 /// that was extended (possibly nullptr), the second is a boolean value
476 /// indicating whether an "undef" was encountered.
477 /// If this range is live before @p Use in the basic block that starts at
478 /// @p StartIdx, and there is no intervening "undef", extend it to be live
479 /// up to @p Use, and return the pair {value, false}. If there is no
480 /// segment before @p Use and there is no "undef" between @p StartIdx and
481 /// @p Use, return {nullptr, false}. If there is an "undef" before @p Use,
482 /// return {nullptr, true}.
483 std::pair<VNInfo*,bool> extendInBlock(ArrayRef<SlotIndex> Undefs,
484 SlotIndex StartIdx, SlotIndex Kill);
486 /// Simplified version of the above "extendInBlock", which assumes that
487 /// no register lanes are undefined by <def,read-undef> operands.
488 /// If this range is live before @p Use in the basic block that starts
489 /// at @p StartIdx, extend it to be live up to @p Use, and return the
490 /// value. If there is no segment before @p Use, return nullptr.
491 VNInfo *extendInBlock(SlotIndex StartIdx, SlotIndex Kill);
493 /// join - Join two live ranges (this, and other) together. This applies
494 /// mappings to the value numbers in the LHS/RHS ranges as specified. If
495 /// the ranges are not joinable, this aborts.
496 void join(LiveRange &Other,
497 const int *ValNoAssignments,
498 const int *RHSValNoAssignments,
499 SmallVectorImpl<VNInfo *> &NewVNInfo);
501 /// True iff this segment is a single segment that lies between the
502 /// specified boundaries, exclusively. Vregs live across a backedge are not
503 /// considered local. The boundaries are expected to lie within an extended
504 /// basic block, so vregs that are not live out should contain no holes.
505 bool isLocal(SlotIndex Start, SlotIndex End) const {
506 return beginIndex() > Start.getBaseIndex() &&
507 endIndex() < End.getBoundaryIndex();
510 /// Remove the specified segment from this range. Note that the segment
511 /// must be a single Segment in its entirety.
512 void removeSegment(SlotIndex Start, SlotIndex End,
513 bool RemoveDeadValNo = false);
515 void removeSegment(Segment S, bool RemoveDeadValNo = false) {
516 removeSegment(S.start, S.end, RemoveDeadValNo);
519 /// Remove segment pointed to by iterator @p I from this range. This does
520 /// not remove dead value numbers.
521 iterator removeSegment(iterator I) {
522 return segments.erase(I);
525 /// Query Liveness at Idx.
526 /// The sub-instruction slot of Idx doesn't matter, only the instruction
527 /// it refers to is considered.
528 LiveQueryResult Query(SlotIndex Idx) const {
529 // Find the segment that enters the instruction.
530 const_iterator I = find(Idx.getBaseIndex());
531 const_iterator E = end();
532 if (I == E)
533 return LiveQueryResult(nullptr, nullptr, SlotIndex(), false);
535 // Is this an instruction live-in segment?
536 // If Idx is the start index of a basic block, include live-in segments
537 // that start at Idx.getBaseIndex().
538 VNInfo *EarlyVal = nullptr;
539 VNInfo *LateVal = nullptr;
540 SlotIndex EndPoint;
541 bool Kill = false;
542 if (I->start <= Idx.getBaseIndex()) {
543 EarlyVal = I->valno;
544 EndPoint = I->end;
545 // Move to the potentially live-out segment.
546 if (SlotIndex::isSameInstr(Idx, I->end)) {
547 Kill = true;
548 if (++I == E)
549 return LiveQueryResult(EarlyVal, LateVal, EndPoint, Kill);
551 // Special case: A PHIDef value can have its def in the middle of a
552 // segment if the value happens to be live out of the layout
553 // predecessor.
554 // Such a value is not live-in.
555 if (EarlyVal->def == Idx.getBaseIndex())
556 EarlyVal = nullptr;
558 // I now points to the segment that may be live-through, or defined by
559 // this instr. Ignore segments starting after the current instr.
560 if (!SlotIndex::isEarlierInstr(Idx, I->start)) {
561 LateVal = I->valno;
562 EndPoint = I->end;
564 return LiveQueryResult(EarlyVal, LateVal, EndPoint, Kill);
567 /// removeValNo - Remove all the segments defined by the specified value#.
568 /// Also remove the value# from value# list.
569 void removeValNo(VNInfo *ValNo);
571 /// Returns true if the live range is zero length, i.e. no live segments
572 /// span instructions. It doesn't pay to spill such a range.
573 bool isZeroLength(SlotIndexes *Indexes) const {
574 for (const Segment &S : segments)
575 if (Indexes->getNextNonNullIndex(S.start).getBaseIndex() <
576 S.end.getBaseIndex())
577 return false;
578 return true;
581 // Returns true if any segment in the live range contains any of the
582 // provided slot indexes. Slots which occur in holes between
583 // segments will not cause the function to return true.
584 bool isLiveAtIndexes(ArrayRef<SlotIndex> Slots) const;
586 bool operator<(const LiveRange& other) const {
587 const SlotIndex &thisIndex = beginIndex();
588 const SlotIndex &otherIndex = other.beginIndex();
589 return thisIndex < otherIndex;
592 /// Returns true if there is an explicit "undef" between @p Begin
593 /// @p End.
594 bool isUndefIn(ArrayRef<SlotIndex> Undefs, SlotIndex Begin,
595 SlotIndex End) const {
596 return std::any_of(Undefs.begin(), Undefs.end(),
597 [Begin,End] (SlotIndex Idx) -> bool {
598 return Begin <= Idx && Idx < End;
602 /// Flush segment set into the regular segment vector.
603 /// The method is to be called after the live range
604 /// has been created, if use of the segment set was
605 /// activated in the constructor of the live range.
606 void flushSegmentSet();
608 void print(raw_ostream &OS) const;
609 void dump() const;
611 /// Walk the range and assert if any invariants fail to hold.
613 /// Note that this is a no-op when asserts are disabled.
614 #ifdef NDEBUG
615 void verify() const {}
616 #else
617 void verify() const;
618 #endif
620 protected:
621 /// Append a segment to the list of segments.
622 void append(const LiveRange::Segment S);
624 private:
625 friend class LiveRangeUpdater;
626 void addSegmentToSet(Segment S);
627 void markValNoForDeletion(VNInfo *V);
630 inline raw_ostream &operator<<(raw_ostream &OS, const LiveRange &LR) {
631 LR.print(OS);
632 return OS;
635 /// LiveInterval - This class represents the liveness of a register,
636 /// or stack slot.
637 class LiveInterval : public LiveRange {
638 public:
639 using super = LiveRange;
641 /// A live range for subregisters. The LaneMask specifies which parts of the
642 /// super register are covered by the interval.
643 /// (@sa TargetRegisterInfo::getSubRegIndexLaneMask()).
644 class SubRange : public LiveRange {
645 public:
646 SubRange *Next = nullptr;
647 LaneBitmask LaneMask;
649 /// Constructs a new SubRange object.
650 SubRange(LaneBitmask LaneMask) : LaneMask(LaneMask) {}
652 /// Constructs a new SubRange object by copying liveness from @p Other.
653 SubRange(LaneBitmask LaneMask, const LiveRange &Other,
654 BumpPtrAllocator &Allocator)
655 : LiveRange(Other, Allocator), LaneMask(LaneMask) {}
657 void print(raw_ostream &OS) const;
658 void dump() const;
661 private:
662 SubRange *SubRanges = nullptr; ///< Single linked list of subregister live
663 /// ranges.
665 public:
666 const unsigned reg; // the register or stack slot of this interval.
667 float weight; // weight of this interval
669 LiveInterval(unsigned Reg, float Weight) : reg(Reg), weight(Weight) {}
671 ~LiveInterval() {
672 clearSubRanges();
675 template<typename T>
676 class SingleLinkedListIterator {
677 T *P;
679 public:
680 SingleLinkedListIterator<T>(T *P) : P(P) {}
682 SingleLinkedListIterator<T> &operator++() {
683 P = P->Next;
684 return *this;
686 SingleLinkedListIterator<T> operator++(int) {
687 SingleLinkedListIterator res = *this;
688 ++*this;
689 return res;
691 bool operator!=(const SingleLinkedListIterator<T> &Other) {
692 return P != Other.operator->();
694 bool operator==(const SingleLinkedListIterator<T> &Other) {
695 return P == Other.operator->();
697 T &operator*() const {
698 return *P;
700 T *operator->() const {
701 return P;
705 using subrange_iterator = SingleLinkedListIterator<SubRange>;
706 using const_subrange_iterator = SingleLinkedListIterator<const SubRange>;
708 subrange_iterator subrange_begin() {
709 return subrange_iterator(SubRanges);
711 subrange_iterator subrange_end() {
712 return subrange_iterator(nullptr);
715 const_subrange_iterator subrange_begin() const {
716 return const_subrange_iterator(SubRanges);
718 const_subrange_iterator subrange_end() const {
719 return const_subrange_iterator(nullptr);
722 iterator_range<subrange_iterator> subranges() {
723 return make_range(subrange_begin(), subrange_end());
726 iterator_range<const_subrange_iterator> subranges() const {
727 return make_range(subrange_begin(), subrange_end());
730 /// Creates a new empty subregister live range. The range is added at the
731 /// beginning of the subrange list; subrange iterators stay valid.
732 SubRange *createSubRange(BumpPtrAllocator &Allocator,
733 LaneBitmask LaneMask) {
734 SubRange *Range = new (Allocator) SubRange(LaneMask);
735 appendSubRange(Range);
736 return Range;
739 /// Like createSubRange() but the new range is filled with a copy of the
740 /// liveness information in @p CopyFrom.
741 SubRange *createSubRangeFrom(BumpPtrAllocator &Allocator,
742 LaneBitmask LaneMask,
743 const LiveRange &CopyFrom) {
744 SubRange *Range = new (Allocator) SubRange(LaneMask, CopyFrom, Allocator);
745 appendSubRange(Range);
746 return Range;
749 /// Returns true if subregister liveness information is available.
750 bool hasSubRanges() const {
751 return SubRanges != nullptr;
754 /// Removes all subregister liveness information.
755 void clearSubRanges();
757 /// Removes all subranges without any segments (subranges without segments
758 /// are not considered valid and should only exist temporarily).
759 void removeEmptySubRanges();
761 /// getSize - Returns the sum of sizes of all the LiveRange's.
763 unsigned getSize() const;
765 /// isSpillable - Can this interval be spilled?
766 bool isSpillable() const {
767 return weight != huge_valf;
770 /// markNotSpillable - Mark interval as not spillable
771 void markNotSpillable() {
772 weight = huge_valf;
775 /// For a given lane mask @p LaneMask, compute indexes at which the
776 /// lane is marked undefined by subregister <def,read-undef> definitions.
777 void computeSubRangeUndefs(SmallVectorImpl<SlotIndex> &Undefs,
778 LaneBitmask LaneMask,
779 const MachineRegisterInfo &MRI,
780 const SlotIndexes &Indexes) const;
782 /// Refines the subranges to support \p LaneMask. This may only be called
783 /// for LI.hasSubrange()==true. Subregister ranges are split or created
784 /// until \p LaneMask can be matched exactly. \p Mod is executed on the
785 /// matching subranges.
787 /// Example:
788 /// Given an interval with subranges with lanemasks L0F00, L00F0 and
789 /// L000F, refining for mask L0018. Will split the L00F0 lane into
790 /// L00E0 and L0010 and the L000F lane into L0007 and L0008. The Mod
791 /// function will be applied to the L0010 and L0008 subranges.
792 void refineSubRanges(BumpPtrAllocator &Allocator, LaneBitmask LaneMask,
793 std::function<void(LiveInterval::SubRange&)> Apply);
795 bool operator<(const LiveInterval& other) const {
796 const SlotIndex &thisIndex = beginIndex();
797 const SlotIndex &otherIndex = other.beginIndex();
798 return std::tie(thisIndex, reg) < std::tie(otherIndex, other.reg);
801 void print(raw_ostream &OS) const;
802 void dump() const;
804 /// Walks the interval and assert if any invariants fail to hold.
806 /// Note that this is a no-op when asserts are disabled.
807 #ifdef NDEBUG
808 void verify(const MachineRegisterInfo *MRI = nullptr) const {}
809 #else
810 void verify(const MachineRegisterInfo *MRI = nullptr) const;
811 #endif
813 private:
814 /// Appends @p Range to SubRanges list.
815 void appendSubRange(SubRange *Range) {
816 Range->Next = SubRanges;
817 SubRanges = Range;
820 /// Free memory held by SubRange.
821 void freeSubRange(SubRange *S);
824 inline raw_ostream &operator<<(raw_ostream &OS,
825 const LiveInterval::SubRange &SR) {
826 SR.print(OS);
827 return OS;
830 inline raw_ostream &operator<<(raw_ostream &OS, const LiveInterval &LI) {
831 LI.print(OS);
832 return OS;
835 raw_ostream &operator<<(raw_ostream &OS, const LiveRange::Segment &S);
837 inline bool operator<(SlotIndex V, const LiveRange::Segment &S) {
838 return V < S.start;
841 inline bool operator<(const LiveRange::Segment &S, SlotIndex V) {
842 return S.start < V;
845 /// Helper class for performant LiveRange bulk updates.
847 /// Calling LiveRange::addSegment() repeatedly can be expensive on large
848 /// live ranges because segments after the insertion point may need to be
849 /// shifted. The LiveRangeUpdater class can defer the shifting when adding
850 /// many segments in order.
852 /// The LiveRange will be in an invalid state until flush() is called.
853 class LiveRangeUpdater {
854 LiveRange *LR;
855 SlotIndex LastStart;
856 LiveRange::iterator WriteI;
857 LiveRange::iterator ReadI;
858 SmallVector<LiveRange::Segment, 16> Spills;
859 void mergeSpills();
861 public:
862 /// Create a LiveRangeUpdater for adding segments to LR.
863 /// LR will temporarily be in an invalid state until flush() is called.
864 LiveRangeUpdater(LiveRange *lr = nullptr) : LR(lr) {}
866 ~LiveRangeUpdater() { flush(); }
868 /// Add a segment to LR and coalesce when possible, just like
869 /// LR.addSegment(). Segments should be added in increasing start order for
870 /// best performance.
871 void add(LiveRange::Segment);
873 void add(SlotIndex Start, SlotIndex End, VNInfo *VNI) {
874 add(LiveRange::Segment(Start, End, VNI));
877 /// Return true if the LR is currently in an invalid state, and flush()
878 /// needs to be called.
879 bool isDirty() const { return LastStart.isValid(); }
881 /// Flush the updater state to LR so it is valid and contains all added
882 /// segments.
883 void flush();
885 /// Select a different destination live range.
886 void setDest(LiveRange *lr) {
887 if (LR != lr && isDirty())
888 flush();
889 LR = lr;
892 /// Get the current destination live range.
893 LiveRange *getDest() const { return LR; }
895 void dump() const;
896 void print(raw_ostream&) const;
899 inline raw_ostream &operator<<(raw_ostream &OS, const LiveRangeUpdater &X) {
900 X.print(OS);
901 return OS;
904 /// ConnectedVNInfoEqClasses - Helper class that can divide VNInfos in a
905 /// LiveInterval into equivalence clases of connected components. A
906 /// LiveInterval that has multiple connected components can be broken into
907 /// multiple LiveIntervals.
909 /// Given a LiveInterval that may have multiple connected components, run:
911 /// unsigned numComps = ConEQ.Classify(LI);
912 /// if (numComps > 1) {
913 /// // allocate numComps-1 new LiveIntervals into LIS[1..]
914 /// ConEQ.Distribute(LIS);
915 /// }
917 class ConnectedVNInfoEqClasses {
918 LiveIntervals &LIS;
919 IntEqClasses EqClass;
921 public:
922 explicit ConnectedVNInfoEqClasses(LiveIntervals &lis) : LIS(lis) {}
924 /// Classify the values in \p LR into connected components.
925 /// Returns the number of connected components.
926 unsigned Classify(const LiveRange &LR);
928 /// getEqClass - Classify creates equivalence classes numbered 0..N. Return
929 /// the equivalence class assigned the VNI.
930 unsigned getEqClass(const VNInfo *VNI) const { return EqClass[VNI->id]; }
932 /// Distribute values in \p LI into a separate LiveIntervals
933 /// for each connected component. LIV must have an empty LiveInterval for
934 /// each additional connected component. The first connected component is
935 /// left in \p LI.
936 void Distribute(LiveInterval &LI, LiveInterval *LIV[],
937 MachineRegisterInfo &MRI);
940 } // end namespace llvm
942 #endif // LLVM_CODEGEN_LIVEINTERVAL_H