tilelang

trending/tilelang

Fork 0

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

Files

T

History

2cccf1f5dd [Feature] Upgrade cutlass version and support fp8 T.gemm (#202 )

* upgrade cutlass to upstream v3.8.0

* Implement fp8 gemm and add example script

* Fix dtype retrieval with map_torch_type for fp8 inputs

* Disable vectorization of fp8 values

* Make MMA declaration compatible with cutlass 3.4.0+

* Add test for fp8 T.gemm

* fix indent

* fix indent

* Add copyright and license header

* Add copyright and license header

* lint fix

* Refactor matmul_nt and assert_matmul_correctness functions for improved readability by consolidating parameter definitions and adjusting formatting.

* clang format lint

---------

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

2025-03-13 19:51:03 +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 )