tvm-commits mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From tqc...@apache.org
Subject [incubator-tvm] branch v0.6 updated: Fix Canonical Simplifier for v0.6 (#5509)
Date Mon, 04 May 2020 20:04:00 GMT
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 <kevinthesunwy@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():


Mime
View raw message