[LoopVectorizer] Propagate underlying instruction to the cloned instances of VPPartialReductionRecipes

NickGuy-Arm · NickGuy-Arm · commit cf5656206712 · 2025-01-20T16:01:19.000Z
diff --git a/llvm/lib/Transforms/Vectorize/VPlan.h b/llvm/lib/Transforms/Vectorize/VPlan.h
@@ -2459,7 +2459,8 @@ class VPPartialReductionRecipe : public VPSingleDefRecipe {
   ~VPPartialReductionRecipe() override = default;
 
   VPPartialReductionRecipe *clone() override {
-    return new VPPartialReductionRecipe(Opcode, getOperand(0), getOperand(1));
+    return new VPPartialReductionRecipe(Opcode, getOperand(0), getOperand(1),
+       getUnderlyingInstr());
   }
 
   VP_CLASSOF_IMPL(VPDef::VPPartialReductionSC)

Original file line number	Diff line number	Diff line change
`@@ -2459,7 +2459,8 @@ class VPPartialReductionRecipe : public VPSingleDefRecipe {`
`2459`	`2459`	`~VPPartialReductionRecipe() override = default;`
`2460`	`2460`
`2461`	`2461`	`VPPartialReductionRecipe *clone() override {`
`2462`		`- return new VPPartialReductionRecipe(Opcode, getOperand(0), getOperand(1));`
	`2462`	`+ return new VPPartialReductionRecipe(Opcode, getOperand(0), getOperand(1),`
	`2463`	`+ getUnderlyingInstr());`
`2463`	`2464`	`}`
`2464`	`2465`
`2465`	`2466`	`VP_CLASSOF_IMPL(VPDef::VPPartialReductionSC)`