!fixup use first successor for middle block.

fhahn · fhahn · commit 4a51d2900da0 · 2024-12-12T09:35:07.000Z
diff --git a/llvm/lib/Transforms/Vectorize/VPlan.h b/llvm/lib/Transforms/Vectorize/VPlan.h
@@ -3891,10 +3891,10 @@ class VPlan {
   /// whether to execute the scalar tail loop or the exit block from the loop
   /// latch.
   const VPBasicBlock *getMiddleBlock() const {
-    return cast<VPBasicBlock>(getScalarPreheader()->getSinglePredecessor());
+    return cast<VPBasicBlock>(getScalarPreheader()->getPredecessors()[0]);
   }
   VPBasicBlock *getMiddleBlock() {
-    return cast<VPBasicBlock>(getScalarPreheader()->getSinglePredecessor());
+    return cast<VPBasicBlock>(getScalarPreheader()->getPredecessors()[0]);
   }
 
   /// Return the VPBasicBlock for the preheader of the scalar loop.

Original file line number	Diff line number	Diff line change
`@@ -3891,10 +3891,10 @@ class VPlan {`
`3891`	`3891`	`/// whether to execute the scalar tail loop or the exit block from the loop`
`3892`	`3892`	`/// latch.`
`3893`	`3893`	`const VPBasicBlock *getMiddleBlock() const {`
`3894`		`- return cast<VPBasicBlock>(getScalarPreheader()->getSinglePredecessor());`
	`3894`	`+ return cast<VPBasicBlock>(getScalarPreheader()->getPredecessors()[0]);`
`3895`	`3895`	`}`
`3896`	`3896`	`VPBasicBlock *getMiddleBlock() {`
`3897`		`- return cast<VPBasicBlock>(getScalarPreheader()->getSinglePredecessor());`
	`3897`	`+ return cast<VPBasicBlock>(getScalarPreheader()->getPredecessors()[0]);`
`3898`	`3898`	`}`
`3899`	`3899`
`3900`	`3900`	`/// Return the VPBasicBlock for the preheader of the scalar loop.`