[llvm] dfc5f37 - [VPlan] Move onlyFirstLaneUsed to VPWidenInductionRecipe (NFC).

[Florian Hahn via llvm-commits]

Author: Florian Hahn
Date: 2025-03-03T22:43:47Z
New Revision: dfc5f37e3a62a93ce16935cea1d9822bec649a2b

URL: https://github.com/llvm/llvm-project/commit/dfc5f37e3a62a93ce16935cea1d9822bec649a2b
DIFF: https://github.com/llvm/llvm-project/commit/dfc5f37e3a62a93ce16935cea1d9822bec649a2b.diff

LOG: [VPlan] Move onlyFirstLaneUsed to VPWidenInductionRecipe (NFC).

Move onlyFirstLaneUsed from VPWidenIntOrFpInductionRecipe and
VPWidenPointerInduction to VPWidenInductionRecipe. Also mark step value
as having only its first lane used.

Added: 
    

Modified: 
    llvm/lib/Transforms/Vectorize/VPlan.h

Removed: 
    


################################################################################
diff  --git a/llvm/lib/Transforms/Vectorize/VPlan.h b/llvm/lib/Transforms/Vectorize/VPlan.h
index 57ac1a743bbf3..24440bc64d2a7 100644
--- a/llvm/lib/Transforms/Vectorize/VPlan.h
+++ b/llvm/lib/Transforms/Vectorize/VPlan.h
@@ -1755,6 +1755,16 @@ class VPWidenInductionRecipe : public VPHeaderPHIRecipe {
     llvm_unreachable(
         "VPWidenIntOrFpInductionRecipe generates its own backedge value");
   }
+
+  /// Returns true if the recipe only uses the first lane of operand \p Op.
+  bool onlyFirstLaneUsed(const VPValue *Op) const override {
+    assert(is_contained(operands(), Op) &&
+           "Op must be an operand of the recipe");
+    // The recipe creates its own wide start value, so it only requests the
+    // first lane of the operand.
+    // TODO: Remove once creating the start value is modeled separately.
+    return Op == getStartValue() || Op == getStepValue();
+  }
 };
 
 /// A recipe for handling phi nodes of integer and floating-point inductions,
@@ -1831,16 +1841,6 @@ class VPWidenIntOrFpInductionRecipe : public VPWidenInductionRecipe {
   VPValue *getLastUnrolledPartOperand() {
     return getNumOperands() == 5 ? getOperand(4) : this;
   }
-
-  /// Returns true if the recipe only uses the first lane of operand \p Op.
-  bool onlyFirstLaneUsed(const VPValue *Op) const override {
-    assert(is_contained(operands(), Op) &&
-           "Op must be an operand of the recipe");
-    // The recipe creates its own wide start value, so it only requests the
-    // first lane of the operand.
-    // TODO: Remove once creating the start value is modeled separately.
-    return Op == getStartValue();
-  }
 };
 
 class VPWidenPointerInductionRecipe : public VPWidenInductionRecipe,
@@ -1873,13 +1873,6 @@ class VPWidenPointerInductionRecipe : public VPWidenInductionRecipe,
   /// Returns true if only scalar values will be generated.
   bool onlyScalarsGenerated(bool IsScalable);
 
-  /// Returns true if the recipe only uses the first lane of operand \p Op.
-  bool onlyFirstLaneUsed(const VPValue *Op) const override {
-    assert(is_contained(operands(), Op) &&
-           "Op must be an operand of the recipe");
-    return Op == getOperand(0);
-  }
-
   /// Returns the VPValue representing the value of this induction at
   /// the first unrolled part, if it exists. Returns itself if unrolling did not
   /// take place.