tilelang

mirror of https://github.com/tile-ai/tilelang.git synced 2026-10-21 06:40:23 +08:00

Files

T

Lei Wang 7bde63d526 [Feat] Introduce new caching mechanism for compiled kernels (#176 )

* Add kernel caching mechanism to TileLang

- Implement a new `cached` function in `tilelang/cache/__init__.py` to cache and reuse compiled kernels
- Expose the `cached` function in the main `tilelang/__init__.py`
- Add a test case for cached matrix multiplication in `testing/python/cache/test_tilelang_cache_matmul.py`
- Provide a `clear_cache()` function to reset the kernel cache when needed

* Refactor kernel caching test and implementation

- Simplify the `cached` function in `tilelang/cache/__init__.py`
- Update test script `test_tilelang_cache_matmul.py` to use `tilelang.testing.main()`
- Remove unnecessary whitespace and improve code formatting

* Update import for `cached` function in MHA examples

- Modify import statement in `example_mha_bwd.py` and `test_tilelang_kernel_mha_bwd.py`
- Change import from `tilelang.profiler import cached` to `tilelang import cached`
- Align with recent refactoring of kernel caching mechanism

* Refactor `cached` function signature in kernel caching

- Update function signature to use keyword-only arguments for `target` and `target_host`
- Improve parameter order and readability of the `cached` decorator
- Maintain existing functionality while enhancing function definition

2025-03-09 20:29:37 +08:00

test_tilelang_kernel_bf16_gemm_mma.py

[Enhancement] Enable runtime tensor data type validation (#146 )

2025-03-05 21:56:27 +08:00

test_tilelang_kernel_convolution.py

[CI][Test] Add test cases for tilelang kernel convolution (#51 )

2025-01-25 18:27:12 +08:00

test_tilelang_kernel_dequantize_gemm.py

[Refactor] Replace T.thread_binding with T.get_thread_binding in examples and test cases (#163 )

2025-03-07 16:10:26 +08:00

test_tilelang_kernel_element_wise_add.py

[CI][Test] Add test cases for element_add (#47 )