fix comments

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

Author: yaoyaoding

python/tilus/backends/codegen.py

@@ -288,7 +288,9 @@ def visit_ThreadGroupStmt(self, stmt: ThreadGroupStmt) -> None:
             cond = (self.current_thread // stmt.num_threads) == (stmt.thread_begin // stmt.num_threads)
             tid_value = self.current_thread % stmt.num_threads
         else:
-            cond = logical_and(0 <= self.current_thread, self.current_thread < stmt.thread_begin + stmt.num_threads)
+            cond = logical_and(
+                stmt.thread_begin <= self.current_thread, self.current_thread < stmt.thread_begin + stmt.num_threads
+            )
             tid_value = self.current_thread - stmt.thread_begin
         with self.builder.if_then(cond=cond):
             tid = self.builder.declare_var("tid", tp=int32, init=tid_value)

python/tilus/ir/tools/printer.py

@@ -269,9 +269,9 @@ def visit_ForStmt(self, stmt: ForStmt) -> Doc:
     def visit_ThreadGroupStmt(self, stmt: ThreadGroupStmt) -> Doc:
         head_doc = (
             NewLine()
-            + Text("with thread_group(group_index=")
+            + Text("with thread_group(thread_begin=")
             + self.visit(stmt.thread_begin)
-            + ", group_size="
+            + ", num_threads="
             + self.visit(stmt.num_threads)
             + "):"
         )