[llvm] [LV] Use IVUpdateMayOverflow to set HasNUW. (PR #111758)

Ramkumar Ramachandra via llvm-commits llvm-commits at lists.llvm.org
Tue Oct 22 13:21:53 PDT 2024


================
@@ -9014,9 +9014,11 @@ LoopVectorizationPlanner::tryToBuildVPlanWithVPRecipes(VFRange &Range) {
 
   DebugLoc DL = getDebugLocFromInstOrOperands(Legal->getPrimaryInduction());
   TailFoldingStyle Style = CM.getTailFoldingStyle(IVUpdateMayOverflow);
-  // When not folding the tail, we know that the induction increment will not
-  // overflow.
-  bool HasNUW = Style == TailFoldingStyle::None;
+  // Use NUW for the induction increment if we proved that it won't overflow in
+  // the vector loop or when not folding the tail. Then we know that the
+  // induction increment will not overflow as the vector trip count is >=
+  // increment and a multiple of the increment.
+  bool HasNUW = !IVUpdateMayOverflow || Style == TailFoldingStyle::None;
----------------
artagnon wrote:

I think (although I'm not sure) you can drop the `Style == TailFoldingStyle::None`, as I think `!IVUpdateMayOverflow` subsumes it.

https://github.com/llvm/llvm-project/pull/111758


More information about the llvm-commits mailing list