Skip to content

Commit 8a8a874

Browse files
committed
Add extra regression tests on floormod(x, 2) change with itermap simplify.
1 parent 8a24623 commit 8a8a874

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

tests/python/unittest/test_arith_iter_affine_map.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1190,6 +1190,12 @@ def test_iter_map_simplify_unit_loop_order():
11901190
simplify_trivial_iterators=False,
11911191
)
11921192

1193+
assert_iter_map_simplfy(
1194+
{y + 64 - x % 2 * 64: y + 64 - x % 2 * 64},
1195+
var_dom([(x, 6), (y, 64)]),
1196+
simplify_trivial_iterators=False,
1197+
)
1198+
11931199

11941200
if __name__ == "__main__":
11951201
tvm.testing.main()

0 commit comments

Comments
 (0)