commit ff5d6bd441cd9826eba2fb4b8a67d52fed38aade
author Uday Bondhugula <bondhugula@google.com> Thu Sep 27 18:03:27 2018 -0700
committer jpienaar <jpienaar@google.com> Fri Mar 29 13:21:13 2019 -0700
tree ae4024cf31d6a9df0709506ff276bb60cbaf4efb
parent b1b3496a368cb680c67c961e9e7ff981cc431bb3

Change loop step to be a positive integral constant

Changing this per discussion on mlir-team. Spec updated.

PiperOrigin-RevId: 214868483

lib/Analysis/LoopAnalysis.cpp

diff --git a/lib/Analysis/LoopAnalysis.cpp b/lib/Analysis/LoopAnalysis.cpp
index fa283ae..f4e607e 100644
--- a/lib/Analysis/LoopAnalysis.cpp
+++ b/lib/Analysis/LoopAnalysis.cpp
@@ -66,13 +66,12 @@
 
     auto *cExpr = dyn_cast<AffineConstantExpr>(loopSpanExpr);
     if (!cExpr)
-      return AffineBinaryOpExpr::getCeilDiv(loopSpanExpr, std::abs(step),
-                                            context);
+      return AffineBinaryOpExpr::getCeilDiv(loopSpanExpr, step, context);
     loopSpan = cExpr->getValue();
   }
 
   // 0 iteration loops.
-  if ((loopSpan < 0 && step >= 1) || (loopSpan > 0 && step <= -1))
+  if (loopSpan < 0)
     return 0;
 
   return AffineConstantExpr::get(