37#ifndef LLVM_SUPPORT_GENERICDOMTREECONSTRUCTION_H
38#define LLVM_SUPPORT_GENERICDOMTREECONSTRUCTION_H
49#define DEBUG_TYPE "dom-tree-builder"
52namespace DomTreeBuilder {
55 using NodePtr =
typename DomTreeT::NodePtr;
56 using NodeT =
typename DomTreeT::NodeType;
58 using RootsT =
decltype(DomTreeT::Roots);
59 static constexpr bool IsPostDom = DomTreeT::IsPostDominator;
81 using UpdateT =
typename DomTreeT::UpdateType;
110 template <
bool Inversed>
124 using DirectedNodeT =
125 std::conditional_t<Inversed, Inverse<NodePtr>,
NodePtr>;
130 if constexpr (std::is_same_v<std::decay_t<
decltype(*R.begin())>,
NodePtr>)
142 Max =
GraphTraits<
decltype(BB->getParent())>::getMaxNumber(
145 NodeInfos.resize(Max ? Max + 1 : Idx + 1);
163 assert(IDom || DT.getNode(
nullptr));
168 return DT.createNode(BB, IDomNode);
183 BP.
N->printAsOperand(O,
false);
201 template <
bool IsReverse = false,
typename DescendCondition>
203 unsigned AttachToNum,
209 while (!WorkList.
empty()) {
212 BBInfo.ReverseChildren.push_back(ParentNum);
215 if (BBInfo.DFSNum != 0)
217 BBInfo.Parent = ParentNum;
218 BBInfo.DFSNum = BBInfo.Semi = BBInfo.Label = ++LastNum;
227 if (Condition(BB, Succ))
233 if (SuccOrder && Successors.size() > 1)
236 return SuccOrder->find(A)->second < SuccOrder->find(B)->second;
239 for (
const NodePtr Succ : Successors) {
240 if (!Condition(BB, Succ))
263 unsigned eval(
unsigned V,
unsigned LastLinked,
267 if (VInfo->
Parent < LastLinked)
273 Stack.push_back(VInfo);
274 VInfo = NumToInfo[VInfo->
Parent];
275 }
while (VInfo->
Parent >= LastLinked);
282 VInfo = Stack.pop_back_val();
285 if (PLabelInfo->
Semi < VLabelInfo->
Semi)
288 PLabelInfo = VLabelInfo;
290 }
while (!Stack.empty());
296 const unsigned NextDFSNum(
NumToNode.size());
304 for (
unsigned i = 1; i < NextDFSNum; ++i) {
306 IDoms[i] = VInfo.Parent;
307 NumToInfo[i] = &VInfo;
312 for (
unsigned i = NextDFSNum - 1; i >= 2; --i) {
313 auto &WInfo = *NumToInfo[i];
316 WInfo.Semi = WInfo.Parent;
317 for (
unsigned N : WInfo.ReverseChildren) {
318 unsigned SemiU = NumToInfo[
eval(
N, i + 1, EvalStack, NumToInfo)]->Semi;
319 if (SemiU < WInfo.Semi)
327 for (
unsigned i = 2; i < NextDFSNum; ++i) {
328 auto &WInfo = *NumToInfo[i];
330 unsigned WIDom = IDoms[i];
331 while (WIDom > WInfo.Semi)
332 WIDom = IDoms[WIDom];
345 assert(
NumToNode.size() == 1 &&
"SNCAInfo must be freshly constructed");
348 BBInfo.DFSNum = BBInfo.Semi = BBInfo.Label = 1;
357 assert(
N &&
"N must be a valid node");
362 assert(DT.Parent &&
"Parent not set");
370 assert(DT.Parent &&
"Parent pointer is not set");
414 bool HasNonTrivialRoots =
false;
417 if (
Total + 1 != Num) {
418 HasNonTrivialRoots =
true;
425 std::optional<NodeOrderMap> SuccOrder;
426 auto InitSuccOrderOnce = [&]() {
431 SuccOrder->try_emplace(Succ, 0);
434 unsigned NodeNum = 0;
435 for (
const auto Node :
nodes(DT.Parent)) {
437 auto Order = SuccOrder->find(
Node);
438 if (Order != SuccOrder->end()) {
439 assert(Order->second == 0);
440 Order->second = NodeNum;
472 const unsigned NewNum =
476 <<
"(non-trivial root): "
478 Roots.push_back(FurthestAway);
479 LLVM_DEBUG(
dbgs() <<
"\t\t\tPrev DFSNum: " << Num <<
", new DFSNum: "
480 << NewNum <<
"\n\t\t\tRemoving DFS info\n");
481 for (
unsigned i = NewNum; i > Num; --i) {
488 const unsigned PrevNum = Num;
491 for (
unsigned i = PrevNum + 1; i <= Num; ++i)
503 assert((
Total + 1 == Num) &&
"Everything should have been visited");
506 if (HasNonTrivialRoots)
532 for (
unsigned i = 0; i < Roots.size(); ++i) {
533 auto &Root = Roots[i];
538 <<
" remains a root\n");
544 for (
unsigned x = 2; x <= Num; ++x) {
565 template <
typename DescendCondition>
568 assert(DT.Roots.size() == 1 &&
"Dominators should have a singe root");
569 runDFS(DT.Roots[0], 0, DC, 0);
575 for (
const NodePtr Root : DT.Roots)
576 Num =
runDFS(Root, Num, DC, 1);
580 auto *Parent = DT.Parent;
604 dbgs() <<
"DomTree recalculated, skipping future batch updates\n");
607 if (DT.Roots.empty())
615 DT.RootNode = DT.createNode(Root);
634 DT.createNode(W, IDomNode);
652 return LHS->getLevel() <
RHS->getLevel();
658 std::priority_queue<TreeNodePtr, SmallVector<TreeNodePtr, 8>, Compare>
662#if LLVM_ENABLE_ABI_BREAKING_CHECKS
670 "From has to be a valid CFG node or a virtual root");
671 assert(To &&
"Cannot be a nullptr");
683 FromTN = DT.createNode(From, VirtualRoot);
684 DT.Roots.push_back(From);
687 DT.DFSInfoValid =
false;
704 if (!DT.isVirtualRoot(To->
getIDom()))
711 <<
" is no longer a root\n\t\tRebuilding the tree!!!\n");
719 if (
A.size() !=
B.size())
723 if (Set.count(
N) == 0)
736 return HasForwardSuccessors(N, BUI);
749 <<
"The entire tree needs to be rebuilt\n");
770 assert(NCDBlock || DT.isPostDominator());
775 const unsigned NCDLevel = NCD->
getLevel();
793 II.Visited.insert(To);
795 while (!
II.Bucket.empty()) {
798 II.Affected.push_back(TN);
800 const unsigned CurrentLevel = TN->
getLevel();
802 <<
"as affected, CurrentLevel " << CurrentLevel
819 "Unreachable successor found at reachable insertion");
820 const unsigned SuccLevel = SuccTN->
getLevel();
823 <<
", level = " << SuccLevel <<
"\n");
832 if (SuccLevel <= NCDLevel + 1 || !
II.Visited.insert(SuccTN).second)
835 if (SuccLevel > CurrentLevel) {
840 UnaffectedOnCurrentLevel.
push_back(SuccTN);
841#if LLVM_ENABLE_ABI_BREAKING_CHECKS
842 II.VisitedUnaffected.push_back(SuccTN);
848 <<
" to a Bucket\n");
849 II.Bucket.push(SuccTN);
853 if (UnaffectedOnCurrentLevel.
empty())
875#if LLVM_ENABLE_ABI_BREAKING_CHECKS && !defined(NDEBUG)
878 "TN should have been updated by an affected ancestor");
902 for (
const auto &Edge : DiscoveredEdgesToReachable) {
915 &DiscoveredConnectingEdges) {
916 assert(!DT.getNode(Root) &&
"Root must not be reachable");
919 auto UnreachableDescender = [&DT, &DiscoveredConnectingEdges](
NodePtr From,
925 DiscoveredConnectingEdges.push_back({From, ToTN});
930 SNCA.
runDFS(Root, 0, UnreachableDescender, 0);
939 assert(From && To &&
"Cannot disconnect nullptrs");
943#if LLVM_ENABLE_ABI_BREAKING_CHECKS
947 auto IsSuccessor = [BUI](
const NodePtr SuccCandidate,
const NodePtr Of) {
952 assert(!IsSuccessor(To, From) &&
"Deleted edge still exists in the CFG!");
964 <<
") already unreachable -- there is no edge to delete\n");
968 const NodePtr NCDBlock = DT.findNearestCommonDominator(From, To);
973 DT.DFSInfoValid =
false;
1003 assert(ToIDom || DT.isPostDominator());
1009 if (!PrevIDomSubTree) {
1016 const unsigned Level = ToIDomTN->
getLevel();
1018 return DT.getNode(To)->getLevel() > Level;
1025 SNCA.
runDFS(ToIDom, 0, DescendBelow, 0);
1040 if (!DT.getNode(Pred))
1043 const NodePtr Support = DT.findNearestCommonDominator(TNB, Pred);
1045 if (Support != TNB) {
1047 <<
" is reachable from support "
1069 LLVM_DEBUG(
dbgs() <<
"\tDeletion made a region reverse-unreachable\n");
1072 DT.Roots.push_back(ToTN->
getBlock());
1078 const unsigned Level = ToTN->
getLevel();
1082 auto DescendAndCollect = [Level, &AffectedQueue, &DT](
NodePtr,
NodePtr To) {
1094 unsigned LastDFSNum =
1101 for (
const NodePtr N : AffectedQueue) {
1105 assert(NCDBlock || DT.isPostDominator());
1125 for (
unsigned i = LastDFSNum; i > 0; --i) {
1133 if (MinNode == ToTN)
1136 LLVM_DEBUG(
dbgs() <<
"DeleteUnreachable: running DFS with MinNode = "
1138 const unsigned MinLevel = MinNode->
getLevel();
1147 return ToTN->
getLevel() > MinLevel;
1148 DT.createNode(To, DT.getNode(R));
1170 if (NumUpdates == 0)
1175 if (NumUpdates == 1) {
1178 if (Update.getKind() == UpdateKind::Insert)
1179 InsertEdge(DT,
nullptr, Update.getFrom(), Update.getTo());
1181 DeleteEdge(DT,
nullptr, Update.getFrom(), Update.getTo());
1184 if (Update.getKind() == UpdateKind::Insert)
1185 InsertEdge(DT, &BUI, Update.getFrom(), Update.getTo());
1187 DeleteEdge(DT, &BUI, Update.getFrom(), Update.getTo());
1201 if (DT.DomTreeNodes.size() <= 100) {
1204 }
else if (BUI.
NumLegalized > DT.DomTreeNodes.size() / 40)
1225 if (CurrentUpdate.getKind() == UpdateKind::Insert)
1226 InsertEdge(DT, &BUI, CurrentUpdate.getFrom(), CurrentUpdate.getTo());
1228 DeleteEdge(DT, &BUI, CurrentUpdate.getFrom(), CurrentUpdate.getTo());
1241 if (!DT.Parent && !DT.Roots.empty()) {
1242 errs() <<
"Tree has no parent but has roots!\n";
1248 if (DT.Roots.empty()) {
1249 errs() <<
"Tree doesn't have a root!\n";
1255 errs() <<
"Tree's root is not its parent's entry node!\n";
1263 errs() <<
"Tree has different roots than freshly computed ones!\n";
1264 errs() <<
"\tPDT roots: ";
1267 errs() <<
"\n\tComputed roots: ";
1268 for (
const NodePtr N : ComputedRoots)
1284 for (
auto *TN : DT.DomTreeNodes) {
1287 const NodePtr BB = TN->getBlock();
1290 if (DT.isVirtualRoot(TN))
1295 <<
" not found by DFS walk!\n";
1303 if (
N && !DT.getNode(
N)) {
1305 <<
" not found in the DomTree!\n";
1319 for (
auto *TN : DT.DomTreeNodes) {
1322 const NodePtr BB = TN->getBlock();
1327 if (!IDom && TN->
getLevel() != 0) {
1329 <<
" has a nonzero level " << TN->getLevel() <<
"!\n";
1337 << TN->getLevel() <<
" while its IDom "
1353 if (!DT.DFSInfoValid || !DT.Parent)
1359 auto PrintNodeAndDFSNums = [](
const TreeNodePtr TN) {
1361 << TN->getDFSNumOut() <<
'}';
1367 errs() <<
"DFSIn number for the tree root is not:\n\t";
1368 PrintNodeAndDFSNums(Root);
1376 for (
auto *
Node : DT.DomTreeNodes) {
1381 if (
Node->isLeaf()) {
1382 if (
Node->getDFSNumIn() + 1 !=
Node->getDFSNumOut()) {
1383 errs() <<
"Tree leaf should have DFSOut = DFSIn + 1:\n\t";
1384 PrintNodeAndDFSNums(
Node);
1400 auto PrintChildrenError =
1405 errs() <<
"Incorrect DFS numbers for:\n\tParent ";
1406 PrintNodeAndDFSNums(
Node);
1408 errs() <<
"\n\tChild ";
1409 PrintNodeAndDFSNums(FirstCh);
1412 errs() <<
"\n\tSecond child ";
1413 PrintNodeAndDFSNums(SecondCh);
1416 errs() <<
"\nAll children: ";
1418 PrintNodeAndDFSNums(Ch);
1426 if (Children.front()->getDFSNumIn() !=
Node->getDFSNumIn() + 1) {
1427 PrintChildrenError(Children.front(),
nullptr);
1431 if (Children.back()->getDFSNumOut() + 1 !=
Node->getDFSNumOut()) {
1432 PrintChildrenError(Children.back(),
nullptr);
1436 for (
size_t i = 0, e = Children.size() - 1; i != e; ++i) {
1437 if (Children[i]->getDFSNumOut() + 1 != Children[i + 1]->getDFSNumIn()) {
1438 PrintChildrenError(Children[i], Children[i + 1]);
1491 for (
auto *TN : DT.DomTreeNodes) {
1494 const NodePtr BB = TN->getBlock();
1495 if (!BB || TN->isLeaf())
1502 return From != BB && To != BB;
1509 <<
" is removed!\n";
1526 for (
auto *TN : DT.DomTreeNodes) {
1529 const NodePtr BB = TN->getBlock();
1530 if (!BB || TN->isLeaf())
1537 return From != BBN && To != BBN;
1547 <<
" is removed!\n";
1568 FreshTree.recalculate(*DT.Parent);
1569 const bool Different = DT.compare(FreshTree);
1572 errs() << (DT.isPostDominator() ?
"Post" :
"")
1573 <<
"DominatorTree is different than a freshly computed one!\n"
1576 errs() <<
"\n\tFreshly computed tree:\n";
1577 FreshTree.print(
errs());
1589template <
typename DomTreeT>
1600template <
class DomTreeT>
1602 typename DomTreeT::NodePtr To) {
1603 if (DT.isPostDominator())
1608template <
class DomTreeT>
1610 typename DomTreeT::NodePtr To) {
1611 if (DT.isPostDominator())
1616template <
class DomTreeT>
1619 DomTreeT::IsPostDominator> &PreViewCFG,
1621 DomTreeT::IsPostDominator> *PostViewCFG) {
1625template <
class DomTreeT>
1626bool Verify(
const DomTreeT &DT,
typename DomTreeT::VerificationLevel VL) {
1640 if (VL == DomTreeT::VerificationLevel::Basic ||
1641 VL == DomTreeT::VerificationLevel::Full)
1644 if (VL == DomTreeT::VerificationLevel::Full)
assert(UImm &&(UImm !=~static_cast< T >(0)) &&"Invalid immediate!")
Unify divergent function exit nodes
static GCRegistry::Add< ErlangGC > A("erlang", "erlang-compatible garbage collector")
static GCRegistry::Add< OcamlGC > B("ocaml", "ocaml 3.10-compatible GC")
This file defines the DenseSet and SmallDenseSet classes.
This file builds on the ADT/GraphTraits.h file to build generic depth first graph iterator.
This file defines a set of templates that efficiently compute a dominator tree over a generic graph.
Loop::LoopBounds::Direction Direction
uint64_t IntrinsicInst * II
ppc ctr loops PowerPC CTR Loops Verify
This file defines the SmallPtrSet class.
Represent a constant reference to an array (0 or more elements consecutively in memory),...
Base class for the actual dominator tree node.
void setIDom(DomTreeNodeBase *NewIDom)
DomTreeNodeBase * getIDom() const
unsigned getDFSNumIn() const
getDFSNumIn/getDFSNumOut - These return the DFS visitation order for nodes in the dominator tree.
unsigned getLevel() const
cfg::Update< NodePtr > popUpdateForIncrementalUpdates()
unsigned getNumLegalizedUpdates() const
Implements a dense probed hash-table based set with some number of buckets stored inline.
SmallPtrSet - This class implements a set which is optimized for holding SmallSize or less elements.
This class consists of common code factored out of the SmallVector class to reduce code duplication b...
void resize_for_overwrite(size_type N)
Like resize, but T is POD, the new values won't be initialized.
void push_back(const T &Elt)
This is a 'vector' (really, a variable-sized array), optimized for the case when the array is small.
This class implements an extremely fast bulk output stream that can only output to a stream.
@ C
The default llvm calling convention, compatible with C.
void CalculateWithUpdates(DomTreeT &DT, ArrayRef< typename DomTreeT::UpdateType > Updates)
void DeleteEdge(DomTreeT &DT, typename DomTreeT::NodePtr From, typename DomTreeT::NodePtr To)
void Calculate(DomTreeT &DT)
void ApplyUpdates(DomTreeT &DT, GraphDiff< typename DomTreeT::NodePtr, DomTreeT::IsPostDominator > &PreViewCFG, GraphDiff< typename DomTreeT::NodePtr, DomTreeT::IsPostDominator > *PostViewCFG)
void InsertEdge(DomTreeT &DT, typename DomTreeT::NodePtr From, typename DomTreeT::NodePtr To)
auto reverse_if(Range &&R)
This is an optimization pass for GlobalISel generic memory operations.
auto drop_begin(T &&RangeOrContainer, size_t N=1)
Return a range covering RangeOrContainer with the first N elements excluded.
constexpr from_range_t from_range
constexpr bool GraphHasNodeNumbers
Indicate whether a GraphTraits<NodeT>::getNumber() is supported.
void sort(IteratorTy Start, IteratorTy End)
LLVM_ABI raw_ostream & dbgs()
dbgs() - This returns a reference to a raw_ostream for debugging messages.
bool none_of(R &&Range, UnaryPredicate P)
Provide wrappers to std::none_of which take ranges instead of having to pass begin/end explicitly.
iterator_range< filter_iterator< detail::IterOfRange< RangeT >, PredicateT > > make_filter_range(RangeT &&Range, PredicateT Pred)
Convenience function that takes a range of elements and a predicate, and return a new filter_iterator...
LLVM_ABI raw_fd_ostream & errs()
This returns a reference to a raw_ostream for standard error.
iterator_range< typename GraphTraits< GraphType >::ChildIteratorType > children(const typename GraphTraits< GraphType >::NodeRef &G)
bool is_contained(R &&Range, const E &Element)
Returns true if Element is found in Range.
void swap(llvm::BitVector &LHS, llvm::BitVector &RHS)
Implement std::swap in terms of BitVector swap.
BatchUpdateInfo(GraphDiffT &PreViewCFG, GraphDiffT *PostViewCFG=nullptr)
const size_t NumLegalized
friend raw_ostream & operator<<(raw_ostream &O, const BlockNamePrinter &BP)
BlockNamePrinter(NodePtr Block)
BlockNamePrinter(TreeNodePtr TN)
SmallVector< unsigned, 4 > ReverseChildren
bool operator()(TreeNodePtr LHS, TreeNodePtr RHS) const
std::priority_queue< TreeNodePtr, SmallVector< TreeNodePtr, 8 >, Compare > Bucket
SmallVector< TreeNodePtr, 8 > Affected
SmallDenseSet< TreeNodePtr, 8 > Visited
SemiNCAInfo(BatchUpdatePtr BUI)
static void UpdateInsertion(DomTreeT &DT, const BatchUpdatePtr BUI, const TreeNodePtr NCD, InsertionInfo &II)
static void DeleteEdge(DomTreeT &DT, const BatchUpdatePtr BUI, const NodePtr From, const NodePtr To)
NodePtr getIDom(NodePtr BB)
InfoRec & getNodeInfo(NodePtr BB)
void doFullDFSWalk(const DomTreeT &DT, DescendCondition DC)
DenseMap< NodePtr, unsigned > NodeOrderMap
static RootsT FindRoots(const DomTreeT &DT, BatchUpdatePtr BUI)
static SmallVector< NodePtr, 8 > getChildren(NodePtr N, BatchUpdatePtr BUI)
static void ComputeUnreachableDominators(DomTreeT &DT, const BatchUpdatePtr BUI, const NodePtr Root, const TreeNodePtr Incoming, SmallVectorImpl< std::pair< NodePtr, TreeNodePtr > > &DiscoveredConnectingEdges)
BatchUpdateInfo * BatchUpdatePtr
static bool VerifyLevels(const DomTreeT &DT)
decltype(DomTreeT::Roots) RootsT
bool verifyReachability(const DomTreeT &DT)
unsigned eval(unsigned V, unsigned LastLinked, SmallVectorImpl< InfoRec * > &Stack, ArrayRef< InfoRec * > NumToInfo)
static bool IsSameAsFreshTree(const DomTreeT &DT)
GraphDiff< NodePtr, IsPostDom > GraphDiffT
static constexpr bool IsPostDom
typename DomTreeT::NodePtr NodePtr
static void ApplyUpdates(DomTreeT &DT, GraphDiffT &PreViewCFG, GraphDiffT *PostViewCFG)
typename DomTreeT::UpdateKind UpdateKind
static bool UpdateRootsBeforeInsertion(DomTreeT &DT, const BatchUpdatePtr BUI, const TreeNodePtr From, const TreeNodePtr To)
bool verifySiblingProperty(const DomTreeT &DT)
typename DomTreeT::NodeType NodeT
void reattachExistingSubtree(DomTreeT &DT, const TreeNodePtr AttachTo)
static NodePtr GetEntryNode(const DomTreeT &DT)
static bool AlwaysDescend(NodePtr, NodePtr)
static void UpdateRootsAfterUpdate(DomTreeT &DT, const BatchUpdatePtr BUI)
SmallVector< NodePtr, 64 > NumToNode
unsigned runDFS(NodePtr V, unsigned LastNum, DescendCondition Condition, unsigned AttachToNum, const NodeOrderMap *SuccOrder=nullptr)
static void DeleteReachable(DomTreeT &DT, const BatchUpdatePtr BUI, const TreeNodePtr FromTN, const TreeNodePtr ToTN)
static void RemoveRedundantRoots(const DomTreeT &DT, BatchUpdatePtr BUI, RootsT &Roots)
static bool HasProperSupport(DomTreeT &DT, const BatchUpdatePtr BUI, const TreeNodePtr TN)
static bool isPermutation(const SmallVectorImpl< NodePtr > &A, const SmallVectorImpl< NodePtr > &B)
static void CalculateFromScratch(DomTreeT &DT, BatchUpdatePtr BUI)
TreeNodePtr getNodeForBlock(NodePtr BB, DomTreeT &DT)
typename DomTreeT::UpdateType UpdateT
static void InsertReachable(DomTreeT &DT, const BatchUpdatePtr BUI, const TreeNodePtr From, const TreeNodePtr To)
static bool HasForwardSuccessors(const NodePtr N, BatchUpdatePtr BUI)
static void InsertUnreachable(DomTreeT &DT, const BatchUpdatePtr BUI, const TreeNodePtr From, const NodePtr To)
static void ApplyNextUpdate(DomTreeT &DT, BatchUpdateInfo &BUI)
DomTreeNodeBase< NodeT > * TreeNodePtr
static bool VerifyDFSNumbers(const DomTreeT &DT)
static void DeleteUnreachable(DomTreeT &DT, const BatchUpdatePtr BUI, const TreeNodePtr ToTN)
void attachNewSubtree(DomTreeT &DT, const TreeNodePtr AttachTo)
static auto getChildren(NodePtr N)
static void InsertEdge(DomTreeT &DT, const BatchUpdatePtr BUI, const NodePtr From, const NodePtr To)
BatchUpdateInfo * BatchUpdates
bool verifyParentProperty(const DomTreeT &DT)
bool verifyRoots(const DomTreeT &DT)
std::conditional_t< GraphHasNodeNumbers< NodePtr >, SmallVector< InfoRec, 64 >, DenseMap< NodePtr, InfoRec > > NodeInfos