This is an automated email from the ASF dual-hosted git repository. tqchen pushed a commit to branch v0.6 in repository https://gitbox.apache.org/repos/asf/incubator-tvm.git
The following commit(s) were added to refs/heads/v0.6 by this push: new 802f055 [BACKPORT-0.6][Bugfix][Arith] keep div_mode during floordiv simplify (#5927) 802f055 is described below commit 802f0555553312d71e66489f92fcec9ea00b31df Author: Yizhi Liu <liuyi...@apache.org> AuthorDate: Thu Jun 25 11:33:18 2020 -0700 [BACKPORT-0.6][Bugfix][Arith] keep div_mode during floordiv simplify (#5927) --- src/arithmetic/canonical_simplify.cc | 1 + tests/python/unittest/test_arith_canonical_simplify.py | 1 + 2 files changed, 2 insertions(+) diff --git a/src/arithmetic/canonical_simplify.cc b/src/arithmetic/canonical_simplify.cc index 2a05834..52104f3 100644 --- a/src/arithmetic/canonical_simplify.cc +++ b/src/arithmetic/canonical_simplify.cc @@ -715,6 +715,7 @@ SplitDivConst(SplitExpr lhs, int64_t cval, DivMode div_mode) { CHECK(lhs->DivModeCompatibleTo(div_mode)); CHECK_EQ(lhs->scale, 1); lhs.CopyOnWrite()->lower_factor *= cval; + lhs.CopyOnWrite()->div_mode = div_mode; return lhs; } diff --git a/tests/python/unittest/test_arith_canonical_simplify.py b/tests/python/unittest/test_arith_canonical_simplify.py index 08038ff..aa41d71 100644 --- a/tests/python/unittest/test_arith_canonical_simplify.py +++ b/tests/python/unittest/test_arith_canonical_simplify.py @@ -82,6 +82,7 @@ def test_split_index_simplify(): # floordiv fld = tvm.floordiv flm = tvm.floormod + ck.verify(fld(x*5, 2), fld(x*5, 2)) ck.verify(fld(x, 3) * 3 + flm(x, 3), x) ck.verify(fld(x, 6) * 6 + flm(fld(x, 3), 2) * 3 + flm(x, 3), x) ck.verify(fld(fld(flm(x, 16), 2) * 2, 4), fld(flm(x, 16), 4))