diff options
Diffstat (limited to 'mlir')
| -rw-r--r-- | mlir/lib/IR/MLIRContext.cpp | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/mlir/lib/IR/MLIRContext.cpp b/mlir/lib/IR/MLIRContext.cpp index 9394d6ea9ce..1e847456c88 100644 --- a/mlir/lib/IR/MLIRContext.cpp +++ b/mlir/lib/IR/MLIRContext.cpp @@ -947,7 +947,7 @@ static AffineExpr simplifyFloorDiv(AffineExpr lhs, AffineExpr rhs) { auto lhsConst = lhs.dyn_cast<AffineConstantExpr>(); auto rhsConst = rhs.dyn_cast<AffineConstantExpr>(); - if (lhsConst && rhsConst) + if (lhsConst && rhsConst && rhsConst.getValue() >= 1) return getAffineConstantExpr( floorDiv(lhsConst.getValue(), rhsConst.getValue()), lhs.getContext()); @@ -974,7 +974,7 @@ static AffineExpr simplifyCeilDiv(AffineExpr lhs, AffineExpr rhs) { auto lhsConst = lhs.dyn_cast<AffineConstantExpr>(); auto rhsConst = rhs.dyn_cast<AffineConstantExpr>(); - if (lhsConst && rhsConst) + if (lhsConst && rhsConst && rhsConst.getValue() >= 1) return getAffineConstantExpr( ceilDiv(lhsConst.getValue(), rhsConst.getValue()), lhs.getContext()); @@ -1001,7 +1001,7 @@ static AffineExpr simplifyMod(AffineExpr lhs, AffineExpr rhs) { auto lhsConst = lhs.dyn_cast<AffineConstantExpr>(); auto rhsConst = rhs.dyn_cast<AffineConstantExpr>(); - if (lhsConst && rhsConst) + if (lhsConst && rhsConst && rhsConst.getValue() >= 1) return getAffineConstantExpr(mod(lhsConst.getValue(), rhsConst.getValue()), lhs.getContext()); |
