Skip to content

Commit 31803c7

Browse files
committed
Update DetectIterMap to recognize floormod 2 patterns
1 parent 5323627 commit 31803c7

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

src/arith/iter_affine_map.cc

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -898,6 +898,11 @@ class IterMapRewriter : public ExprMutator {
898898
PrimExpr SplitFloorModConst(IterSplitExpr lhs, PrimExpr base, PrimExpr rhs);
899899

900900
static void AddToLhs(IterSumExprNode* lhs, IterSplitExpr rhs, int sign) {
901+
if (sign < 0 && is_const_int(rhs->extent, 2)) {
902+
lhs->base -= rhs->scale;
903+
sign = 1;
904+
}
905+
901906
tir::ExprDeepEqual equal;
902907
for (size_t i = 0; i < lhs->args.size(); ++i) {
903908
IterSplitExpr lvalue = lhs->args[i];

0 commit comments

Comments
 (0)