tilelang

mirror of https://github.com/tile-ai/tilelang.git synced 2026-10-20 06:30:07 +08:00

Files

T

Lei Wang 6cede73dce [Enhancement] Update warp specialization checking (#580 )

* Fix L2 cache size calculation to handle symbolic expressions and ensure float conversion of hit ratios in annotation

* [Enhancement] Update warp specialization check in phase.py

* lint fix

* [Enhancement] Add ContainsSeqStmt method to improve statement handling in merge_shared_memory_allocations.cc

* [Refactor] Simplify memory copy operations in GEMM kernel tests

- Updated memory copy operations in `test_tilelang_kernel_gemm.py` to use shared memory allocations for both A and B matrices, improving clarity and performance.
- Adjusted the main execution block to include a new `run_gemm_rs` function call for testing, enhancing the test structure.

* revert memory reuse pass.

* revert the memory resue and thread sync pass/

* Update test_tilelang_kernel_gemm.py

* Update test_tilelang_kernel_mha_bwd.py

2025-06-19 02:08:54 +08:00

test_tilelang_kernel_bf16_gemm_mma.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_kernel_convolution.py

[AutoTune] Refactor AutoTuneArtifact to utilize kernel as context instead of profiler (#344 )

2025-04-07 13:36:35 +08:00

test_tilelang_kernel_deepseek_nsa.py

[Enhancement][Pipeline] More precise copy code block detection in pipeline (#384 )

2025-04-12 23:13:21 +08:00

test_tilelang_kernel_dequantize_gemm.py

[Refactor] Update barrier functions and add new example for GEMM with warp specialization (#456 )