commit 48e4c4b5e9011b477381de9c13f390300c6474a8
author Uday Bondhugula <bondhugula@google.com> Wed Oct 03 10:07:54 2018 -0700
committer jpienaar <jpienaar@google.com> Fri Mar 29 13:21:53 2019 -0700
tree 8b25c15ee816a56a45725d68bf526a4259d5d798
parent 62df2737cbda858ca8f502ec7f001bdbe1f5aceb

Fix MLIR's floordiv, ceildiv, and mod for constant inputs (for negative lhs's)

- introduce mlir::{floorDiv, ceilDiv, mod} for constant inputs in
  mlir/Support/MathExtras.h
- consistently use these everywhere in IR, Analysis, and Transforms.

PiperOrigin-RevId: 215580677

lib/Analysis/LoopAnalysis.cpp

diff --git a/lib/Analysis/LoopAnalysis.cpp b/lib/Analysis/LoopAnalysis.cpp
index f4e607e..1549182 100644
--- a/lib/Analysis/LoopAnalysis.cpp
+++ b/lib/Analysis/LoopAnalysis.cpp
@@ -25,6 +25,7 @@
 #include "mlir/IR/AffineExpr.h"
 #include "mlir/IR/AffineMap.h"
 #include "mlir/IR/Statements.h"
+#include "mlir/Support/MathExtras.h"
 
 using mlir::AffineExpr;
 
@@ -74,10 +75,8 @@
   if (loopSpan < 0)
     return 0;
 
-  return AffineConstantExpr::get(
-      static_cast<uint64_t>(loopSpan % step == 0 ? loopSpan / step
-                                                 : loopSpan / step + 1),
-      context);
+  return AffineConstantExpr::get(static_cast<uint64_t>(ceilDiv(loopSpan, step)),
+                                 context);
 }
 
 /// Returns the trip count of the loop if it's a constant, None otherwise. This