[VPlan] Manually jumpthread a bit of reduction code for readability […

…nfc]
Sterling-Augustine · Aug 30, 2024 · c53008d · c53008d
1 parent a3f8790
commit c53008d
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/llvm/lib/Transforms/Vectorize/VPlanRecipes.cpp b/llvm/lib/Transforms/Vectorize/VPlanRecipes.cpp
@@ -1802,18 +1802,18 @@ void VPReductionRecipe::execute(VPTransformState &State) {
  (Instruction::BinaryOps)RdxDesc.getOpcode(Kind), PrevInChain,
  NewVecOp);
  PrevInChain = NewRed;
+ NextInChain = NewRed;
  } else {
  PrevInChain = State.get(getChainOp(), Part, /*IsScalar*/ true);
  NewRed = createTargetReduction(State.Builder, RdxDesc, NewVecOp);
+ if (RecurrenceDescriptor::isMinMaxRecurrenceKind(Kind))
+ NextInChain = createMinMaxOp(State.Builder, RdxDesc.getRecurrenceKind(),
+ NewRed, PrevInChain);
+ else
+ NextInChain = State.Builder.CreateBinOp(
+ (Instruction::BinaryOps)RdxDesc.getOpcode(Kind), NewRed,
+ PrevInChain);
  }
- if (RecurrenceDescriptor::isMinMaxRecurrenceKind(Kind)) {
- NextInChain = createMinMaxOp(State.Builder, RdxDesc.getRecurrenceKind(),
- NewRed, PrevInChain);
- } else if (IsOrdered)
- NextInChain = NewRed;
- else
- NextInChain = State.Builder.CreateBinOp(
- (Instruction::BinaryOps)RdxDesc.getOpcode(Kind), NewRed, PrevInChain);
  State.set(this, NextInChain, Part, /*IsScalar*/ true);
  }
 }