Also check other op info in isExtended()

SamTebbs33 · SamTebbs33 · commit d604652fbef1 · 2025-05-19T16:48:07.000+01:00
diff --git a/llvm/lib/Transforms/Vectorize/VPlan.h b/llvm/lib/Transforms/Vectorize/VPlan.h
@@ -2589,7 +2589,8 @@ class VPMulAccumulateReductionRecipe : public VPReductionRecipe {
 
   /// Return if this MulAcc recipe contains extend instructions.
   bool isExtended() const {
-    return getVecOp0Info().ExtOp != Instruction::CastOps::CastOpsEnd;
+    return getVecOp0Info().ExtOp != Instruction::CastOps::CastOpsEnd ||
+           getVecOp1Info().ExtOp != Instruction::CastOps::CastOpsEnd;
   }
 
   /// Return if the operands of mul instruction come from same extend.

Original file line number	Diff line number	Diff line change
`@@ -2589,7 +2589,8 @@ class VPMulAccumulateReductionRecipe : public VPReductionRecipe {`
`2589`	`2589`
`2590`	`2590`	`/// Return if this MulAcc recipe contains extend instructions.`
`2591`	`2591`	`bool isExtended() const {`
`2592`		`- return getVecOp0Info().ExtOp != Instruction::CastOps::CastOpsEnd;`
	`2592`	`+ return getVecOp0Info().ExtOp != Instruction::CastOps::CastOpsEnd \|\|`
	`2593`	`+ getVecOp1Info().ExtOp != Instruction::CastOps::CastOpsEnd;`
`2593`	`2594`	`}`
`2594`	`2595`
`2595`	`2596`	`/// Return if the operands of mul instruction come from same extend.`