KurisandLei Wang 0921328dc5 [Language] Tilelang LazyJIT Experimental Version (#1337) ...

* initial step

* modify builder

* scratch version of new frontend

* write some tests

* add many tests

* add typing stub for tir.ir

* remove idents

* minor update

* minor update

* First version of jitv2 (renamed to LazyJIT)

* fix pre-commit error

* minor fix

* fix lint error

* fix lint error

* Fix conditional check for PrimFunc instance

---------

Co-authored-by: Lei Wang <34334180+LeiWang1999@users.noreply.github.com>

2025-12-06 19:18:49 +08:00

..

test_tilelang_capture.py

[Fix] Fix memory leak bug (#1281)

2025-11-19 14:17:45 +08:00

test_tilelang_intimm.py

[Fix] Remove unused let_bindings_ in CodeGenC to fix #1300 (#1305)

2025-11-21 17:04:52 +08:00

test_tilelang_language_alias.py

Refactor to support upstream tvm (#595)

2025-07-30 14:00:15 +08:00

test_tilelang_language_all_of.py

[Bugfix][CI] Bug fixing and migrate CI from ada to hopper (#652)

2025-07-23 16:54:03 +08:00

test_tilelang_language_alloc.py

[FFI] Use tvm ffi as the default execution backend (#1259)

2025-11-18 23:50:57 +08:00

test_tilelang_language_annot.py

[Feat] Add support for using T.Tensor(n * 2 + 1) in function annotation (#1285)

2025-11-20 13:56:09 +08:00

test_tilelang_language_annotate_safe_value.py

[Refactor] Refactor Pass LegalizeSafeMemoryAccess to support recursive load/store rewrite (#1050)

2025-10-17 20:56:01 +08:00

test_tilelang_language_any_of.py

[Bugfix][CI] Bug fixing and migrate CI from ada to hopper (#652)

2025-07-23 16:54:03 +08:00

test_tilelang_language_assume.py

Add unit tests for T.assume (#1341)

2025-11-26 12:57:48 +08:00

test_tilelang_language_atomic_add.py

[BugFix] Use BufferRegion in tl.cumsum to infer buffer shape (#1321)

2025-11-25 01:36:17 +08:00

test_tilelang_language_ceildiv.py

[Transform] Migrate LowerIntrin from tvm into tilelang (#999)

2025-10-14 10:39:38 +08:00

test_tilelang_language_chain_equal.py

[Language] Initial version of tilelang frontend v2 (#1120)

2025-11-03 18:15:56 +08:00

test_tilelang_language_clamp.py

[Refactor] Deprecated T.Buffer as arguments and rename related calls into T.Tensor (#281)

2025-03-26 13:42:52 +08:00

test_tilelang_language_clear.py

[Refactor] refactor tilelang.jit to support a faster and more flexible kernel cache (#501)

2025-05-19 02:21:11 +08:00

test_tilelang_language_composable_index.py

[Refactor] Add parallel loop transform pass for condition extraction (#618)

2025-07-09 15:07:05 +08:00

test_tilelang_language_copy.py

[Language] Recommend using T.dynamic instead of T.symbolic (#1076)

2025-10-20 19:48:31 +08:00

test_tilelang_language_cumsum.py

[TileOp] Implememt CumSum1D (#978)

2025-10-11 15:05:25 +08:00

test_tilelang_language_frontend_v2.py

[Language] Tilelang LazyJIT Experimental Version (#1337)

2025-12-06 19:18:49 +08:00

test_tilelang_language_get_warp_info.py

[Language] Expose T.warpgroup_fence_operand for nvcc code motion (#986)

2025-11-03 00:50:31 +08:00

test_tilelang_language_if_range.py

[Language] Support sequence comparisons (#872)

2025-09-25 12:05:52 +08:00

test_tilelang_language_infinity.py

[Feature] Add tl.infinity operator for infinity handling of bfloat16 (#1175)

2025-11-06 01:22:29 +08:00

test_tilelang_language_int64.py

Fix various issues under int64_t static and dynamic shape. (#1218)

2025-11-18 15:28:23 +08:00

test_tilelang_language_lazy_jit.py

[Language] Tilelang LazyJIT Experimental Version (#1337)

2025-12-06 19:18:49 +08:00

test_tilelang_language_let.py

[Language] Initial version of tilelang frontend v2 (#1120)

2025-11-03 18:15:56 +08:00

test_tilelang_language_mask_op.py

[Refactor] refactor tilelang.jit to support a faster and more flexible kernel cache (#501)

2025-05-19 02:21:11 +08:00

test_tilelang_language_negative_index.py

[Feat] Add A Pass to Handle Negative Index (#1192)

2025-11-06 18:08:36 +08:00

test_tilelang_language_parallel.py

[Parallel] Support T.Parallel with dynamic extents (#990)

2025-10-20 15:50:38 +08:00

test_tilelang_language_pipeline.py

[Pipeline] Optimize inject software pipeline and pipeline planing pass (#706)

2025-08-10 18:36:10 +08:00

test_tilelang_language_ptr.py

[Refactor] Deprecated T.Buffer as arguments and rename related calls into T.Tensor (#281)

2025-03-26 13:42:52 +08:00

test_tilelang_language_reduce.py

[Enhancement] Add thread count validation for ReduceOp fragment layout inference (#1225)

2025-11-11 16:32:52 +08:00

test_tilelang_language_reshape.py

[Language] Add shape check in T.view/reshape (#1277)

2025-11-18 23:17:49 +08:00

test_tilelang_language_ternary.py

[DSL] Support python tenary if then else expression (#822)

2025-09-17 13:49:57 +08:00

test_tilelang_language_tma_1d.py

[Testing] Move TMA 1D and test for its functionality (#1167)

2025-11-01 15:17:01 +08:00

test_tilelang_language_unroll.py

[Enhancement] Implement dynamic unroll factor in CUDA code generation (#1360)

2025-12-02 00:15:52 +08:00

test_tilelang_language_var_init.py

[BugFix] alloc_var init failed to handle complex expression (#1144)

2025-10-28 22:41:05 +08:00

test_tilelang_language_vectorize.py

[Enhancement] Remove constraint requiring last dimension stride to be 1 (#1040)

2025-10-17 13:43:08 +08:00

test_tilelang_language_vectorized_cast.py

[Enhancement] Enhance Cast operations Vectorization (#1156)

2025-10-29 18:22:43 +08:00

test_tilelang_language_view.py

[Language] Add shape check in T.view/reshape (#1277)

2025-11-18 23:17:49 +08:00

test_tilelang_language_warp_reduce.py

[Feat] Support warp reduce (#1316)

2025-11-24 19:38:14 +08:00