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

Merged

LeiWang1999 merged 31 commits intotile-ai:mainfrom

LeiWang1999:jit_merge_0107

Jan 9, 2026

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

Commits on Jan 7, 2026

Commits on Jan 8, 2026

Commits on Jan 9, 2026