diff --git a/src/arith/iter_affine_map.cc b/src/arith/iter_affine_map.cc index 7529019abda8..14bc35530d4a 100644 --- a/src/arith/iter_affine_map.cc +++ b/src/arith/iter_affine_map.cc @@ -1912,7 +1912,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(outer_mark->source); IterSumExpr inner_source = Downcast(inner_mark->source);