trending/tilelang
1
0
Fork 0
mirror of https://github.com/tile-ai/tilelang.git synced 2026-10-21 06:40:23 +08:00
Code Issues Packages Projects Releases Wiki Activity
107 Commits 9 Branches 24 Tags
159af5dfb322df80356130e30130db20a0ebdf21
Commit Graph
2 Commits
Author SHA1 Message Date
Yu Cheng 465f01072c [CI][Test] Add test cases for tilelang transform MultiVersionBuffer and WarpSpecialized (#72) 
* [CI][Test] Add test cases for tilelang transform MultiVersionBuffer and WarpSpecialized

* Relax the mismatch ratio restrictions in the flash_linear_attention and mha tests
 2025-02-11 16:04:28 +08:00
Yu Cheng bedab1a0f0 [CI][Test] Add test cases for tilelang kernel FlashAttention (#54) 
* [Dev] Add FlashDecoding example

* [CI][Test] Add test cases for tilelang kernel convolution

* [CI][Test] Add test cases for tilelang kernel FlashAttention

* Reduce the number of stages to ensure the shared memory allocation is valid

* Temporarily remove the dim128 case

* lint

* update einops in requirements-dev.txt

* update einops in requirements-test.txt

* remove einops in requirements-dev.txt
 2025-01-26 00:46:59 +08:00