Update python/tilus/extensions/hidet/transforms/declare_to_let.py

Co-authored-by: Copilot <[email protected]>
Signed-off-by: Yaoyao Ding <[email protected]>

Author: yaoyaoding

python/tilus/extensions/hidet/transforms/declare_to_let.py

@@ -97,7 +97,11 @@ def visit_SeqStmt(self, seq_stmt: SeqStmt) -> Stmt:
                 let_stmt = LetStmt(bind_vars=[stmt.var], bind_values=[stmt.value], body=self.concat(seq[i + 1 :]))
                 seq = seq[:i] + [let_stmt]
                 self.declare_to_remove.add(stmt.var)
-            elif isinstance(stmt, DeclareStmt) and stmt.var in self.declare_to_remove:
+            elif (
+                isinstance(stmt, DeclareStmt)
+                and stmt.var in self.declare_to_remove
+                and stmt.init is None
+            ):
                 # declare var (var is never assigned)
                 # ...
                 # assign var = value (var is only assigned here and never modified later)