[tvm] branch main updated: [TIR] Fix remaining dtype mismatch issue caused by SubspaceDivide (#13558)

[wu...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

wuwei pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git


The following commit(s) were added to refs/heads/main by this push:
     new a1d46645d2 [TIR] Fix remaining dtype mismatch issue caused by SubspaceDivide (#13558)
a1d46645d2 is described below

commit a1d46645d27e376afaa2c141493854e0dced7621
Author: masahi <ma...@gmail.com>
AuthorDate: Wed Dec 7 04:07:55 2022 +0900

    [TIR] Fix remaining dtype mismatch issue caused by SubspaceDivide (#13558)
---
 src/arith/iter_affine_map.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/arith/iter_affine_map.cc b/src/arith/iter_affine_map.cc
index fa4f0fd514..adba61632f 100644
--- a/src/arith/iter_affine_map.cc
+++ b/src/arith/iter_affine_map.cc
@@ -1910,7 +1910,7 @@ class SubspaceDivider {
       return DivisionResult::Failure();
     }
     bool need_predicate = !analyzer_->CanProveEqual(extent, mark_extent);
-    const IterMark& outer_mark = MarkFromArgsAndBase(outer_args, 0);
+    const IterMark& outer_mark = MarkFromArgsAndBase(outer_args, make_const(dtype, 0));
     const IterMark& inner_mark = MarkFromArgsAndBase(inner_args, expr->base);
     IterSumExpr outer_source = Downcast<IterSumExpr>(outer_mark->source);
     IterSumExpr inner_source = Downcast<IterSumExpr>(inner_mark->source);