Add to VPReductionRecipe::classof

SamTebbs33 · SamTebbs33 · commit 8290d2f1b3cf · 2025-06-02T17:13:05.000+01:00
diff --git a/llvm/lib/Transforms/Vectorize/VPlan.h b/llvm/lib/Transforms/Vectorize/VPlan.h
@@ -2470,7 +2470,8 @@ class VPReductionRecipe : public VPRecipeWithIRFlags {
     return R->getVPDefID() == VPRecipeBase::VPReductionSC ||
            R->getVPDefID() == VPRecipeBase::VPReductionEVLSC ||
            R->getVPDefID() == VPRecipeBase::VPExtendedReductionSC ||
-           R->getVPDefID() == VPRecipeBase::VPMulAccumulateReductionSC;
+           R->getVPDefID() == VPRecipeBase::VPMulAccumulateReductionSC ||
+           R->getVPDefID() == VPRecipeBase::VPPartialReductionSC;
   }
 
   static inline bool classof(const VPUser *U) {

Original file line number	Diff line number	Diff line change
`@@ -2470,7 +2470,8 @@ class VPReductionRecipe : public VPRecipeWithIRFlags {`
`2470`	`2470`	`return R->getVPDefID() == VPRecipeBase::VPReductionSC \|\|`
`2471`	`2471`	`R->getVPDefID() == VPRecipeBase::VPReductionEVLSC \|\|`
`2472`	`2472`	`R->getVPDefID() == VPRecipeBase::VPExtendedReductionSC \|\|`
`2473`		`- R->getVPDefID() == VPRecipeBase::VPMulAccumulateReductionSC;`
	`2473`	`+ R->getVPDefID() == VPRecipeBase::VPMulAccumulateReductionSC \|\|`
	`2474`	`+ R->getVPDefID() == VPRecipeBase::VPPartialReductionSC;`
`2474`	`2475`	`}`
`2475`	`2476`
`2476`	`2477`	`static inline bool classof(const VPUser *U) {`