[llvm] [LV] Vectorize selecting last IV of min/max element. (PR #141431)
via llvm-commits
llvm-commits at lists.llvm.org
Sun Nov 23 12:58:47 PST 2025
================
@@ -965,3 +965,101 @@ bool VPlanTransforms::handleMaxMinNumReductions(VPlan &Plan) {
MiddleTerm->setOperand(0, NewCond);
return true;
}
+
+bool VPlanTransforms::legalizeMultiUseReductions(VPlan &Plan) {
+ for (auto &PhiR : make_early_inc_range(
+ Plan.getVectorLoopRegion()->getEntryBasicBlock()->phis())) {
+ auto *MinMaxPhiR = dyn_cast<VPReductionPHIRecipe>(&PhiR);
+ if (!MinMaxPhiR)
+ continue;
+
+ RecurKind RdxKind = MinMaxPhiR->getRecurrenceKind();
+ // TODO: check for multi-uses in VPlan directly.
+ if (!RecurrenceDescriptor::isIntMinMaxRecurrenceKind(RdxKind) ||
+ !MinMaxPhiR->isPhiMultiUse())
+ continue;
+
+ // One user of MinMaxPhiR is MinMaxOp, the other users must be a compare
+ // that's part of a FindLastIV chain.
+ auto *MinMaxOp =
+ dyn_cast<VPRecipeWithIRFlags>(MinMaxPhiR->getBackedgeValue());
+ if (!MinMaxOp || MinMaxOp->getNumUsers() != 2)
----------------
ayalz wrote:
Better clarify what patterns are supported as accurately as possible at the outset.
Somewhat confusing - `isMinMaxReductionWithLoopUsersOutsideReductionChain()` checks that Phi has more than one user (in the loop) and Inc has one - here MinMaxOp (aka Inc?) has two? Should MinMaxOp be checked to have 2 operands but (asserted to have) 1 user?
https://github.com/llvm/llvm-project/pull/141431
More information about the llvm-commits
mailing list