[Refactor] Unify @jit and @lazy_jit into a single @jit decorator#1617

Closed

LeiWang1999 wants to merge 13 commits intomainfrom

refactor/unify-jit-decorator

Commits on Jan 6, 2026

Refactor: Replace @tilelang.lazy_jit with @tilelang.jit in examples and tests
LeiWang1999
committed
Enhancement: Improve layout annotation handling in T.copy and parallel loops
LeiWang1999
committed
Merge branch 'main' of https://github.com/tile-ai/tilelang into refactor/unify-jit-decorator
LeiWang1999
committed
Add out_idx parameter to @jit and validate it's only used in lazy mode
LeiWang1999
committed
lintfix
LeiWang1999
committed
Merge branch 'main' of https://github.com/tile-ai/tilelang into refactor/unify-jit-decorator
LeiWang1999
committed
Refactor: centralize mode inference in JITImpl.__call__ before parse_args
LeiWang1999
committed
test fix
LeiWang1999
committed
Enhancement: Improve JIT mode inference and AST analysis for lazy style detection
LeiWang1999
committed
Refactor: Enhance JIT mode handling and streamline decorator usage
LeiWang1999
committed
Refactor: Update JIT mode handling and remove unused decorator
LeiWang1999
committed
lint fix
LeiWang1999
committed
Refactor: Simplify JIT function definitions and enhance lazy style detection
LeiWang1999
committed