[mlir][Analysis] Fix assertion in FlatLinearConstraints

Differential Revision: https://reviews.llvm.org/D146869

diff --git a/mlir/lib/Analysis/FlatLinearValueConstraints.cpp b/mlir/lib/Analysis/FlatLinearValueConstraints.cpp
index 24c8d87..b2d8bd9 100644 (file)
--- a/mlir/lib/Analysis/FlatLinearValueConstraints.cpp
+++ b/mlir/lib/Analysis/FlatLinearValueConstraints.cpp
@@ -457,7 +457,7 @@ void FlatLinearConstraints::getSliceBounds(unsigned offset, unsigned num,
                                            SmallVectorImpl<AffineMap> *lbMaps,
                                            SmallVectorImpl<AffineMap> *ubMaps,
                                            bool closedUB) {
-  assert(num < getNumDimVars() && "invalid range");
+  assert(offset + num <= getNumDimVars() && "invalid range");
 
   // Basic simplification.
   normalizeConstraintsByGCD();