[llvm] 3b35113 - [VPlan] Add VPBlockBase::successors() returning an iterator_range (NFC).

[Florian Hahn via llvm-commits]

Author: Florian Hahn
Date: 2021-12-16T14:28:50Z
New Revision: 3b35113ff0963c536a19b4950b5b99aca2e02372

URL: https://github.com/llvm/llvm-project/commit/3b35113ff0963c536a19b4950b5b99aca2e02372
DIFF: https://github.com/llvm/llvm-project/commit/3b35113ff0963c536a19b4950b5b99aca2e02372.diff

LOG: [VPlan] Add VPBlockBase::successors() returning an iterator_range (NFC).

This will also be helpful for D115793.

Added: 
    

Modified: 
    llvm/lib/Transforms/Vectorize/VPlan.cpp
    llvm/lib/Transforms/Vectorize/VPlan.h

Removed: 
    


################################################################################
diff  --git a/llvm/lib/Transforms/Vectorize/VPlan.cpp b/llvm/lib/Transforms/Vectorize/VPlan.cpp
index 44b5e1df08399..aa501e94484e0 100644
--- a/llvm/lib/Transforms/Vectorize/VPlan.cpp
+++ b/llvm/lib/Transforms/Vectorize/VPlan.cpp
@@ -374,8 +374,7 @@ VPBasicBlock *VPBasicBlock::splitAt(iterator SplitAt) {
   assert((SplitAt == end() || SplitAt->getParent() == this) &&
          "can only split at a position in the same block");
 
-  SmallVector<VPBlockBase *, 2> Succs(getSuccessors().begin(),
-                                      getSuccessors().end());
+  SmallVector<VPBlockBase *, 2> Succs(successors());
   // First, disconnect the current block from its successors.
   for (VPBlockBase *Succ : Succs)
     VPBlockUtils::disconnectBlocks(this, Succ);

diff  --git a/llvm/lib/Transforms/Vectorize/VPlan.h b/llvm/lib/Transforms/Vectorize/VPlan.h
index 16dfbfbe863d1..8024bcab1a019 100644
--- a/llvm/lib/Transforms/Vectorize/VPlan.h
+++ b/llvm/lib/Transforms/Vectorize/VPlan.h
@@ -501,6 +501,8 @@ class VPBlockBase {
   const VPBlocksTy &getSuccessors() const { return Successors; }
   VPBlocksTy &getSuccessors() { return Successors; }
 
+  iterator_range<VPBlockBase **> successors() { return Successors; }
+
   const VPBlocksTy &getPredecessors() const { return Predecessors; }
   VPBlocksTy &getPredecessors() { return Predecessors; }