- 28 Jan, 2026 1 commit
-
-
zhanghj2 authored
-
- 27 Jan, 2026 4 commits
- 26 Jan, 2026 5 commits
- 25 Jan, 2026 6 commits
- 19 Jan, 2026 1 commit
-
-
Jiashi Li authored
Co-authored-by:baowending.bwd <baowending.bwd@alibaba-inc.com>
-
- 16 Jan, 2026 1 commit
-
-
Shengyu Liu authored
* Multiple updates and refactorings * Remove dead code
-
- 30 Sep, 2025 1 commit
-
-
Jiashi Li authored
-
- 29 Sep, 2025 2 commits
-
-
Shengyu Liu authored
-
Simon Mo authored
Signed-off-by:simon-mo <simon.mo@hey.com>
-
- 24 Sep, 2025 1 commit
-
-
Shengyu Liu authored
-
- 22 Sep, 2025 1 commit
-
-
zhang authored
-
- 27 Aug, 2025 1 commit
-
-
Zeyu WANG authored
* fix calc space bug * use python code to allocate the buffer for backward kernel
-
- 25 Aug, 2025 2 commits
-
-
Li Xiang authored
* get rid of cudaMalloc and cudaFree * minor fix --------- Co-authored-by:Jiashi Li <js.li@high-flyer.cn>
-
zhang authored
-
- 14 Aug, 2025 1 commit
-
-
Jiashi Li authored
-
- 01 Aug, 2025 1 commit
-
-
Zeyu WANG authored
* Add more GPU architctures support * Merge fmha and mla runner * add varlen & non varlen support, and add incontiguous tensor support * update readme * add varlen api --------- Co-authored-by:dianzhangc <dianzhangc@nvidia.com>
-
- 29 Apr, 2025 1 commit
-
-
ljss authored
-
- 28 Apr, 2025 1 commit
-
-
ljss authored
-
- 22 Apr, 2025 1 commit
-
-
Shengyu Liu authored
* 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
-
- 01 Mar, 2025 1 commit
-
-
ljss authored
-
- 24 Feb, 2025 4 commits
-
-
Sijia Chen authored
-
Sijia Chen authored
-
Sijia Chen authored
-
Jiashi Li authored
i
-