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 60c5e48 Fix Canonical Simplifier for v0.6 (#5509) 60c5e48 is described below commit 60c5e485f873803b9bcf5811fd212968506391fd Author: Yao Wang <kevinthesu...@gmail.com> AuthorDate: Mon May 4 13:03:47 2020 -0700 Fix Canonical Simplifier for v0.6 (#5509) --- src/arithmetic/canonical_simplify.cc | 1 + tests/python/unittest/test_arith_canonical_simplify.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/src/arithmetic/canonical_simplify.cc b/src/arithmetic/canonical_simplify.cc index 1b576a6..2a05834 100644 --- a/src/arithmetic/canonical_simplify.cc +++ b/src/arithmetic/canonical_simplify.cc @@ -860,6 +860,7 @@ SplitModConst(SplitExpr lhs, int64_t cval, DivMode div_mode) { lhs->upper_factor != SplitExprNode::kPosInf) { auto updated = ToSplitExpr(Mutate(ModImpl( lhs->index, make_const(lhs.type(), new_upper_factor), div_mode))); + updated.CopyOnWrite()->scale = lhs->scale; // re-apply the lower_factor if (lhs->lower_factor != 1) { return SplitDivConst(updated, lhs->lower_factor, div_mode); diff --git a/tests/python/unittest/test_arith_canonical_simplify.py b/tests/python/unittest/test_arith_canonical_simplify.py index 28a4388..08038ff 100644 --- a/tests/python/unittest/test_arith_canonical_simplify.py +++ b/tests/python/unittest/test_arith_canonical_simplify.py @@ -123,7 +123,7 @@ def test_floormod_simplify(): x, y = tvm.var("x"), tvm.var("y") ck.verify(flm(flm((x*4) + y - 466036, 24528) - 24512, 16), flm((x*4) + y + 12, 16)) - + ck.verify(flm(flm((x*4), 16), 8), flm(x, 2) * 4) def test_canonical_mixed():