[llvm] f1d4db4 - [GraphDiff] Use class method getChildren instead of GraphTraits.
Alina Sbirlea via llvm-commits
llvm-commits at lists.llvm.org
Mon Jul 27 16:13:16 PDT 2020
Author: Alina Sbirlea
Date: 2020-07-27T16:12:34-07:00
New Revision: f1d4db4f0cdcbfeaee0840bf8a4fb5dc1b9b56fd
URL: https://github.com/llvm/llvm-project/commit/f1d4db4f0cdcbfeaee0840bf8a4fb5dc1b9b56fd
DIFF: https://github.com/llvm/llvm-project/commit/f1d4db4f0cdcbfeaee0840bf8a4fb5dc1b9b56fd.diff
LOG: [GraphDiff] Use class method getChildren instead of GraphTraits.
Summary:
Use getChildren() method in GraphDiff instead of GraphTraits.
This simplifies the code and allows for refactorigns inside GraphDiff.
All usecase need not have a light-weight/copyable range.
Clean GraphTraits implementation.
Reviewers: dblaikie
Subscribers: hiraditya, llvm-commits, george.burgess.iv
Tags: #llvm
Differential Revision: https://reviews.llvm.org/D84562
Added:
Modified:
llvm/include/llvm/Analysis/IteratedDominanceFrontier.h
llvm/include/llvm/Analysis/MemorySSAUpdater.h
llvm/include/llvm/Support/CFGDiff.h
llvm/lib/Analysis/MemorySSAUpdater.cpp
Removed:
################################################################################
diff --git a/llvm/include/llvm/Analysis/IteratedDominanceFrontier.h b/llvm/include/llvm/Analysis/IteratedDominanceFrontier.h
index fb6605285156..8166b52aa226 100644
--- a/llvm/include/llvm/Analysis/IteratedDominanceFrontier.h
+++ b/llvm/include/llvm/Analysis/IteratedDominanceFrontier.h
@@ -73,13 +73,7 @@ ChildrenGetterTy<BasicBlock, IsPostDom>::get(const NodeRef &N) {
return {Children.begin(), Children.end()};
}
- using SnapShotBBPairTy =
- std::pair<const GraphDiff<BasicBlock *, IsPostDom> *, OrderedNodeTy>;
-
- ChildrenTy Ret;
- for (const auto &SnapShotBBPair : children<SnapShotBBPairTy>({GD, N}))
- Ret.emplace_back(SnapShotBBPair.second);
- return Ret;
+ return GD->template getChildren<IsPostDom>(N);
}
} // end of namespace IDFCalculatorDetail
diff --git a/llvm/include/llvm/Analysis/MemorySSAUpdater.h b/llvm/include/llvm/Analysis/MemorySSAUpdater.h
index 20588ef083c5..d41b93209979 100644
--- a/llvm/include/llvm/Analysis/MemorySSAUpdater.h
+++ b/llvm/include/llvm/Analysis/MemorySSAUpdater.h
@@ -52,8 +52,6 @@ class LoopBlocksRPO;
using ValueToValueMapTy = ValueMap<const Value *, WeakTrackingVH>;
using PhiToDefMap = SmallDenseMap<MemoryPhi *, MemoryAccess *>;
using CFGUpdate = cfg::Update<BasicBlock *>;
-using GraphDiffInvBBPair =
- std::pair<const GraphDiff<BasicBlock *> *, Inverse<BasicBlock *>>;
class MemorySSAUpdater {
private:
diff --git a/llvm/include/llvm/Support/CFGDiff.h b/llvm/include/llvm/Support/CFGDiff.h
index 269984872bfa..9cbf311f6801 100644
--- a/llvm/include/llvm/Support/CFGDiff.h
+++ b/llvm/include/llvm/Support/CFGDiff.h
@@ -30,44 +30,6 @@
// a non-inversed graph, the children are naturally the successors when
// InverseEdge is false and the predecessors when InverseEdge is true.
-// We define two base clases that call into GraphDiff, one for successors
-// (CFGSuccessors), where InverseEdge is false, and one for predecessors
-// (CFGPredecessors), where InverseEdge is true.
-// FIXME: Further refactoring may merge the two base classes into a single one
-// templated / parametrized on using succ_iterator/pred_iterator and false/true
-// for the InverseEdge.
-
-// CFGViewChildren and CFGViewPredecessors, both can be parametrized to
-// consider the graph inverted or not (i.e. InverseGraph). Successors
-// implicitly has InverseEdge = false and Predecessors implicitly has
-// InverseEdge = true (see calls to GraphDiff methods in there). The GraphTraits
-// instantiations that follow define the value of InverseGraph.
-
-// GraphTraits instantiations:
-// - GraphDiff<BasicBlock *> is equivalent to InverseGraph = false
-// - GraphDiff<Inverse<BasicBlock *>> is equivalent to InverseGraph = true
-// - second pair item is BasicBlock *, then InverseEdge = false (so it inherits
-// from CFGViewChildren).
-// - second pair item is Inverse<BasicBlock *>, then InverseEdge = true (so it
-// inherits from CFGViewPredecessors).
-
-// The 4 GraphTraits are as follows:
-// 1. std::pair<const GraphDiff<BasicBlock *> *, BasicBlock *>> :
-// CFGViewChildren<false>
-// Regular CFG, children means successors, InverseGraph = false,
-// InverseEdge = false.
-// 2. std::pair<const GraphDiff<Inverse<BasicBlock *>> *, BasicBlock *>> :
-// CFGViewChildren<true>
-// Reverse the graph, get successors but reverse-apply updates,
-// InverseGraph = true, InverseEdge = false.
-// 3. std::pair<const GraphDiff<BasicBlock *> *, Inverse<BasicBlock *>>> :
-// CFGViewPredecessors<false>
-// Regular CFG, reverse edges, so children mean predecessors,
-// InverseGraph = false, InverseEdge = true.
-// 4. std::pair<const GraphDiff<Inverse<BasicBlock *>> *, Inverse<BasicBlock *>>
-// : CFGViewPredecessors<true>
-// Reverse the graph and the edges, InverseGraph = true, InverseEdge = true.
-
namespace llvm {
namespace detail {
@@ -87,9 +49,9 @@ template <bool B, typename Range> auto reverse_if(Range &&R) {
}
} // namespace detail
-// GraphDiff defines a CFG snapshot: given a set of Update<NodePtr>, provide
-// utilities to skip edges marked as deleted and return a set of edges marked as
-// newly inserted. The current
diff treats the CFG as a graph rather than a
+// GraphDiff defines a CFG snapshot: given a set of Update<NodePtr>, provides
+// a getChildren method to get a Node's children based on the additional updates
+// in the snapshot. The current
diff treats the CFG as a graph rather than a
// multigraph. Added edges are pruned to be unique, and deleted edges will
// remove all existing edges between two blocks.
template <typename NodePtr, bool InverseGraph = false> class GraphDiff {
@@ -191,7 +153,6 @@ template <typename NodePtr, bool InverseGraph = false> class GraphDiff {
}
using VectRet = SmallVector<NodePtr, 8>;
-
template <bool InverseEdge> VectRet getChildren(NodePtr N) const {
using DirectedNodeT =
std::conditional_t<InverseEdge, Inverse<NodePtr>, NodePtr>;
@@ -228,59 +189,6 @@ template <typename NodePtr, bool InverseGraph = false> class GraphDiff {
LLVM_DUMP_METHOD void dump() const { print(dbgs()); }
#endif
};
-
-template <typename GraphT, bool InverseGraph = false, bool InverseEdge = false,
- typename GT = GraphTraits<GraphT>>
-struct CFGViewChildren {
- using DataRef = const GraphDiff<typename GT::NodeRef, InverseGraph> *;
- using NodeRef = std::pair<DataRef, typename GT::NodeRef>;
-
- template<typename Range>
- static auto makeChildRange(Range &&R, DataRef DR) {
- using Iter = WrappedPairNodeDataIterator<decltype(std::forward<Range>(R).begin()), NodeRef, DataRef>;
- return make_range(Iter(R.begin(), DR), Iter(R.end(), DR));
- }
-
- static auto children(NodeRef N) {
-
- // filter iterator init:
- auto R = make_range(GT::child_begin(N.second), GT::child_end(N.second));
- auto RR = detail::reverse_if<!InverseEdge>(R);
- // This lambda is copied into the iterators and persists to callers, ensure
- // captures are by value or otherwise have sufficient lifetime.
- auto First = make_filter_range(makeChildRange(RR, N.first), [N](NodeRef C) {
- return !C.first->ignoreChild(N.second, C.second, InverseEdge);
- });
-
- // new inserts iterator init:
- auto InsertVec = N.first->getAddedChildren(N.second, InverseEdge);
- auto Second = makeChildRange(InsertVec, N.first);
-
- auto CR = concat<NodeRef>(First, Second);
-
- // concat_range contains references to other ranges, returning it would
- // leave those references dangling - the iterators contain
- // other iterators by value so they're safe to return.
- return make_range(CR.begin(), CR.end());
- }
-
- static auto child_begin(NodeRef N) {
- return children(N).begin();
- }
-
- static auto child_end(NodeRef N) {
- return children(N).end();
- }
-
- using ChildIteratorType = decltype(child_end(std::declval<NodeRef>()));
-};
-
-template <typename T, bool B>
-struct GraphTraits<std::pair<const GraphDiff<T, B> *, T>>
- : CFGViewChildren<T, B> {};
-template <typename T, bool B>
-struct GraphTraits<std::pair<const GraphDiff<T, B> *, Inverse<T>>>
- : CFGViewChildren<Inverse<T>, B, true> {};
} // end namespace llvm
#endif // LLVM_SUPPORT_CFGDIFF_H
diff --git a/llvm/lib/Analysis/MemorySSAUpdater.cpp b/llvm/lib/Analysis/MemorySSAUpdater.cpp
index 85af091772e7..21cbdcd67147 100644
--- a/llvm/lib/Analysis/MemorySSAUpdater.cpp
+++ b/llvm/lib/Analysis/MemorySSAUpdater.cpp
@@ -832,8 +832,8 @@ void MemorySSAUpdater::applyInsertUpdates(ArrayRef<CFGUpdate> Updates,
// Check number of predecessors, we only care if there's more than one.
unsigned Count = 0;
BasicBlock *Pred = nullptr;
- for (auto &Pair : children<GraphDiffInvBBPair>({GD, BB})) {
- Pred = Pair.second;
+ for (auto *Pi : GD->template getChildren</*InverseEdge=*/true>(BB)) {
+ Pred = Pi;
Count++;
if (Count == 2)
break;
@@ -926,8 +926,7 @@ void MemorySSAUpdater::applyInsertUpdates(ArrayRef<CFGUpdate> Updates,
auto *BB = BBPredPair.first;
const auto &AddedBlockSet = BBPredPair.second.Added;
auto &PrevBlockSet = BBPredPair.second.Prev;
- for (auto &Pair : children<GraphDiffInvBBPair>({GD, BB})) {
- BasicBlock *Pi = Pair.second;
+ for (auto *Pi : GD->template getChildren</*InverseEdge=*/true>(BB)) {
if (!AddedBlockSet.count(Pi))
PrevBlockSet.insert(Pi);
EdgeCountMap[{Pi, BB}]++;
@@ -1078,10 +1077,8 @@ void MemorySSAUpdater::applyInsertUpdates(ArrayRef<CFGUpdate> Updates,
for (unsigned I = 0, E = IDFPhi->getNumIncomingValues(); I < E; ++I)
IDFPhi->setIncomingValue(I, GetLastDef(IDFPhi->getIncomingBlock(I)));
} else {
- for (auto &Pair : children<GraphDiffInvBBPair>({GD, BBIDF})) {
- BasicBlock *Pi = Pair.second;
+ for (auto *Pi : GD->template getChildren</*InverseEdge=*/true>(BBIDF))
IDFPhi->addIncoming(GetLastDef(Pi), Pi);
- }
}
}
}
More information about the llvm-commits
mailing list