DeepSeek/FlashMLA
1
0
Fork 0
mirror of https://github.com/deepseek-ai/FlashMLA synced 2025-05-14 16:45:59 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity
35 Commits 1 Branch 0 Tags 293 KiB
main
Commit Graph

6 Commits

Author SHA1 Message Date
Shengyu Liu
c2067be3ea
Performance Update (2025.04.22) (#71) 
* Fix benchmark script

* Performance optimization for compute-bound cases

* Add new testcase (s_k = 16384)

* Update README.md

* Update comment

* Update README.md

* Add the deep-dive blog

* Add background color for MLA Kernel Sched.drawio.svg

* Use relative path for the schedule image

* Move flash_mla.h to kernels/params.h
 2025-04-22 17:50:57 +08:00
ljss
b31bfe72a8 add missing copyright 2025-03-01 18:24:24 +08:00
Sijia Chen
a3b74b8574 add flag to disable FP16 compile 2025-02-24 10:01:59 -08:00
Sijia Chen
65fb7732fc support fp16 2025-02-24 01:58:53 -08:00
Sijia Chen
15a82b81b8 replace c10 optional with std optional 2025-02-24 00:25:40 -08:00
Jiashi Li
414a2f3eed Initial commit 
i
 2025-02-24 09:20:23 +08:00