Adjust how we iterate over vplan blocks

john-brawn-arm · john-brawn-arm · commit 1f72ed321548 · 2025-04-04T15:22:43.000+01:00
diff --git a/llvm/lib/Transforms/Vectorize/LoopVectorize.cpp b/llvm/lib/Transforms/Vectorize/LoopVectorize.cpp
@@ -4547,8 +4547,8 @@ static bool willGenerateVectors(VPlan &Plan, ElementCount VF,
 }
 
 static bool hasReplicatorRegion(VPlan &Plan) {
-  return any_of(VPBlockUtils::blocksOnly<VPRegionBlock>(
-                    vp_depth_first_deep(Plan.getEntry())),
+  return any_of(VPBlockUtils::blocksOnly<VPRegionBlock>(vp_depth_first_shallow(
+                    Plan.getVectorLoopRegion()->getEntry())),
                 [](auto *VPRB) { return VPRB->isReplicator(); });
 }
 

Original file line number	Diff line number	Diff line change
`@@ -4547,8 +4547,8 @@ static bool willGenerateVectors(VPlan &Plan, ElementCount VF,`
`4547`	`4547`	`}`
`4548`	`4548`
`4549`	`4549`	`static bool hasReplicatorRegion(VPlan &Plan) {`
`4550`		`- return any_of(VPBlockUtils::blocksOnly<VPRegionBlock>(`
`4551`		`- vp_depth_first_deep(Plan.getEntry())),`
	`4550`	`+ return any_of(VPBlockUtils::blocksOnly<VPRegionBlock>(vp_depth_first_shallow(`
	`4551`	`+ Plan.getVectorLoopRegion()->getEntry())),`
`4552`	`4552`	`[](auto *VPRB) { return VPRB->isReplicator(); });`
`4553`	`4553`	`}`
`4554`	`4554`