[llvm] [LV][EVL] Support in-loop reduction using tail folding with EVL. (PR #90184)
via llvm-commits
llvm-commits at lists.llvm.org
Sun Jul 14 19:41:56 PDT 2024
================
@@ -1429,11 +1429,20 @@ bool VPlanTransforms::tryAddExplicitVectorLength(VPlan &Plan) {
// The transform updates all users of inductions to work based on EVL, instead
// of the VF directly. At the moment, widened inductions cannot be updated, so
// bail out if the plan contains any.
- if (any_of(Header->phis(), [](VPRecipeBase &Phi) {
- return (isa<VPWidenIntOrFpInductionRecipe>(&Phi) ||
- isa<VPWidenPointerInductionRecipe>(&Phi));
- }))
+ bool ContainsWidenInductions = any_of(Header->phis(), [](VPRecipeBase &Phi) {
+ return isa<VPWidenIntOrFpInductionRecipe, VPWidenPointerInductionRecipe>(
+ &Phi);
+ });
+ // FIXME: Remove this once we can transform (select header_mask, true_value,
+ // false_value) into vp.merge.
+ bool ContainsOutloopReductions =
+ any_of(Header->phis(), [&](VPRecipeBase &Phi) {
+ auto *R = dyn_cast<VPReductionPHIRecipe>(&Phi);
+ return R && !R->isInLoop();
+ });
+ if (ContainsWidenInductions || ContainsOutloopReductions)
return false;
----------------
sanggyu-shin wrote:
Could you explain why loop has out loop reductions can't use EVL ?
I think EVL can be used even if loop contains out loop reductions.
https://github.com/llvm/llvm-project/pull/90184
More information about the llvm-commits
mailing list