- 06 Mar, 2026 1 commit
-
-
zhanghj2 authored
-
- 05 Mar, 2026 2 commits
- 04 Mar, 2026 1 commit
-
-
zhanghj2 authored
-
- 27 Feb, 2026 1 commit
-
-
zhanghj2 authored
-
- 11 Feb, 2026 1 commit
-
-
zhanghj2 authored
-
- 06 Feb, 2026 1 commit
-
-
zhanghj2 authored
-
- 25 Jan, 2026 1 commit
-
-
zhanghj2 authored
-
- 20 Jan, 2026 1 commit
-
-
Shengyu Liu authored
-
- 16 Jan, 2026 1 commit
-
-
Shengyu Liu authored
* Multiple updates and refactorings * Remove dead code
-
- 24 Sep, 2025 1 commit
-
-
Shengyu Liu authored
-
- 27 Aug, 2025 1 commit
-
-
Zeyu WANG authored
* fix calc space bug * use python code to allocate the buffer for backward kernel
-
- 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>
-
- 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
-
- 24 Feb, 2025 2 commits
-
-
chunyang.wen authored
-
Jiashi Li authored
i
-
