From 9a19cf64929ca1e2f2801cefaf8cab26423cbb39 Mon Sep 17 00:00:00 2001
From: wrongtest <wrongtest@gmail.com>
Date: Fri, 18 Jul 2025 08:23:46 +0000
Subject: [PATCH] fix trivial index map [] -> [0]

---
 src/tir/ir/index_map.cc                             | 2 +-
 src/tir/schedule/primitive/layout_transformation.cc | 4 ++++
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/tir/ir/index_map.cc b/src/tir/ir/index_map.cc
index c722af555a39..3b91c5e84b53 100644
--- a/src/tir/ir/index_map.cc
+++ b/src/tir/ir/index_map.cc
@@ -218,7 +218,7 @@ Array<Range> IndexMapNode::MapRanges(const Array<Range>& ranges, arith::Analyzer
     }
   }
   auto output_dtype = [&]() {
-    int max_bits = 0;
+    int max_bits = ranges.empty() ? 32 : 0;
     for (const auto& range : ranges) {
       max_bits = std::max(max_bits, range->extent.dtype().bits());
     }
diff --git a/src/tir/schedule/primitive/layout_transformation.cc b/src/tir/schedule/primitive/layout_transformation.cc
index 5cce3ea7758f..8931c0e71c11 100644
--- a/src/tir/schedule/primitive/layout_transformation.cc
+++ b/src/tir/schedule/primitive/layout_transformation.cc
@@ -1464,6 +1464,10 @@ void TransformBlockLayout(ScheduleState self, const StmtSRef& block_sref,
   }
 
   // Step 6: Do the actual replacement
+  if (scope_sref->StmtAs<BlockNode>()) {
+    ICHECK(new_loop_vars.empty()) << "Invalid block to loop replacement due to layout transform "
+                                  << index_map;
+  }
   self->Replace(scope_sref, body, {{block, new_block}});
 }