[llvm] [VPlan] Generalize collectUsersInExitBlocks for multiple exit bbs. (PR #115066)
Florian Hahn via llvm-commits
llvm-commits at lists.llvm.org
Sun Nov 17 11:34:12 PST 2024
================
@@ -3837,6 +3837,10 @@ class VPlan {
return cast<VPBasicBlock>(getVectorLoopRegion()->getSingleSuccessor());
}
+ /// Return the exit blocks of the VPlan, that is leaf nodes except the scalar
+ /// header.
+ auto getExitBlocks();
----------------
fhahn wrote:
It is a bit unfortunate but I couldn't find a way to do that, as it needs accesses to the definition of `VPBlockShallowTraversalWrapper` to spell out the full type (other than moving everything in VPlanCFG back to this file).
There may be another way to spell out the type I am not aware of (it is a few nests of various iterator range adopters). For now I extended the comment to provide more information
https://github.com/llvm/llvm-project/pull/115066
More information about the llvm-commits
mailing list