tilelang

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

Files

T

Lei Wang dca2fb489a [Enhancement] Introduce flag to visualize shared memory merge plan (#496 )

* Remove debug print statement from block_sparse_attn_triton.py and implement a timeout handler in autotuner for function execution. This enhances the robustness of the autotuner by allowing it to handle timeouts gracefully.

* Enhance the autotuner module by adding a timeout handler for function execution, improving robustness in handling long-running tasks. This change includes the introduction of a custom TimeoutException and updates to the run_with_timeout function for better signal management.

* Add merge shared memory allocations pass and related configurations

- Introduced a new pass for merging shared memory allocations in GPU kernels, allowing for more efficient memory usage.
- Registered configuration options for debugging and controlling the merging behavior.
- Updated relevant files to integrate the new pass into the TileLang engine and transform modules.
- Adjusted import paths and added documentation for the new functionality.

* Reduce num_stages parameter in GEMM functions from 3 to 1 for improved performance in test_tilelang_kernel_gemm.py

2025-05-16 21:00:18 +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 )