[llvm] [LV] Optimize VPWidenIntOrFpInductionRecipe for known TC (PR #118828)
Florian Hahn via llvm-commits
llvm-commits at lists.llvm.org
Fri Dec 20 02:02:29 PST 2024
================
@@ -661,11 +663,65 @@ static void recursivelyDeleteDeadRecipes(VPValue *V) {
}
}
-void VPlanTransforms::optimizeForVFAndUF(VPlan &Plan, ElementCount BestVF,
- unsigned BestUF,
- PredicatedScalarEvolution &PSE) {
- assert(Plan.hasVF(BestVF) && "BestVF is not available in Plan");
- assert(Plan.hasUF(BestUF) && "BestUF is not available in Plan");
+/// Optimize the width of vector induction variables in \p Plan based on a known
+/// constant Trip Count, \p BestVF and \p BestUF.
+static bool optimizeVectorInductionWidthForTCAndVFUF(VPlan &Plan,
+ ElementCount BestVF,
+ unsigned BestUF) {
+ auto *TC = dyn_cast_if_present<ConstantInt>(
+ Plan.getTripCount()->getUnderlyingValue());
+ if (!TC || !BestVF.isFixed())
+ return false;
+
+ auto TCVal = TC->getValue().tryZExtValue();
+ if (!TCVal)
+ return false;
----------------
fhahn wrote:
Can we instead do the math on APInt to not require this restriction?
Could you also add a test with TC > 64 bits?
https://github.com/llvm/llvm-project/pull/118828
More information about the llvm-commits
mailing list