---
id: pr-tilelang-1632
repo: tile-ai/tilelang
pr: 1632
title: '[Refactor] Unify @jit and @lazy_jit into a single @jit decorator'
author: LeiWang1999
date: '2026-01-09'
url: https://github.com/tile-ai/tilelang/pull/1632
source_category: upstream-code
architectures:
- sm100
tags:
- tile-dsl
- triton kernel
techniques: []
hardware_features: []
kernel_types: []
languages:
- cuda-cpp
- triton
captured_at: '2026-05-16'
status: merged
merge_sha: 6e439530b5db4ae927ae348519060f5e7d3a7ebf
inclusion_reason: Search-matched CUDA kernel optimization PR; materialize diff/source before synthesis.
changed_paths:
- examples/blocksparse_attention/example_triton_sparse_gqa_decode_varlen_indice.py
- examples/gemm/example_gemm_intrinsics.py
- examples/gemm_fp8/example_tilelang_gemm_fp8_intrinsic.py
- examples/lazy_jit/lazyjit.en.ipynb
- examples/lazy_jit/lazyjit.zh.ipynb
- src/transform/layout_inference.cc
- src/transform/lower_tile_op.cc
- testing/python/arith/test_arith_hard.py
- testing/python/issue/test_tilelang_issue_1549.py
- testing/python/issue/test_tilelang_issue_1601.py
- testing/python/kernel/test_tilelang_kernel_int4_gemm_mma.py
- testing/python/language/test_tilelang_language_frontend_v2.py
- testing/python/language/test_tilelang_language_lazy_jit.py
- testing/python/language/test_tilelang_language_ptr.py
- testing/python/language/test_tilelang_language_subtype.py
- testing/python/layout/test_tilelang_annotate_loop_layout.py
- tilelang/__init__.py
- tilelang/jit/__init__.py
- tilelang/jit/adapter/tvm_ffi.py
- tilelang/jit/exceptions.py
- tilelang/language/__init__.py
- tilelang/language/allocate.py
- tilelang/language/eager/__init__.py
- tilelang/language/eager/ast.py
- tilelang/language/eager/builder.py
- tilelang/language/eager/dtypes.py
- tilelang/language/eager/utils.py
- tilelang/language/kernel.py
- tilelang/language/loop.py
- tilelang/language/print_op.py
- tilelang/language/proxy.py
- tilelang/language/v2/__init__.py
artifact_dir: evidence/pull-bundles/tilelang/gh-1632
head_ref_oid: d86f8478e1194d85ab6c02fdbeb62940c0609471
---
# [Refactor] Unify @jit and @lazy_jit into a single @jit decorator

- Repository: `tile-ai/tilelang`
- PR: [1632](https://github.com/tile-ai/tilelang/pull/1632)
- Decision: `Search-matched CUDA kernel optimization PR; materialize diff/source before synthesis.`
- Reason: Search-matched CUDA kernel optimization PR; materialize diff/source before synthesis.

## Changed Paths

- `examples/blocksparse_attention/example_triton_sparse_gqa_decode_varlen_indice.py`
- `examples/gemm/example_gemm_intrinsics.py`
- `examples/gemm_fp8/example_tilelang_gemm_fp8_intrinsic.py`
- `examples/lazy_jit/lazyjit.en.ipynb`
- `examples/lazy_jit/lazyjit.zh.ipynb`
- `src/transform/layout_inference.cc`
- `src/transform/lower_tile_op.cc`
- `testing/python/arith/test_arith_hard.py`
- `testing/python/issue/test_tilelang_issue_1549.py`
- `testing/python/issue/test_tilelang_issue_1601.py`
- `testing/python/kernel/test_tilelang_kernel_int4_gemm_mma.py`
- `testing/python/language/test_tilelang_language_frontend_v2.py`
- `testing/python/language/test_tilelang_language_lazy_jit.py`
- `testing/python/language/test_tilelang_language_ptr.py`
- `testing/python/language/test_tilelang_language_subtype.py`
- `testing/python/layout/test_tilelang_annotate_loop_layout.py`
- `tilelang/__init__.py`
- `tilelang/jit/__init__.py`
- `tilelang/jit/adapter/tvm_ffi.py`
- `tilelang/jit/exceptions.py`
- `tilelang/language/__init__.py`
- `tilelang/language/allocate.py`
- `tilelang/language/eager/__init__.py`
- `tilelang/language/eager/ast.py`
- `tilelang/language/eager/builder.py`
- `tilelang/language/eager/dtypes.py`
- `tilelang/language/eager/utils.py`
- `tilelang/language/kernel.py`
- `tilelang/language/loop.py`
- `tilelang/language/print_op.py`
- `tilelang/language/proxy.py`
- `tilelang/language/v2/__init__.py`

## Evidence Bundle

`evidence/pull-bundles/tilelang/gh-1632`
