[schedule] Improve ceil_divide in tile/split

apache · Aug 27, 2019 · 885204a · 885204a
1 parent 347e3d9
commit 885204a
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/schedule/message_passing.cc b/src/schedule/message_passing.cc
@@ -56,6 +56,9 @@ void PassDownDomain(const Stage& stage,
                     arith::Analyzer* actx,
                     bool allow_missing) {
   auto ceil_div = [actx](Expr a, Expr b) {
+    if (actx->CanProve(a % b == 0)) {
+      return actx->Simplify(a / b);
+    }
     return actx->Simplify((a + (b - 1)) / b);
   };