Commits · 902e303211d773d8cabe368ff9dab18e75305acb · OpenDAS / FlashMLA

27 May, 2026 1 commit
- Fix precision issue · 902e3032
  zhanghj2 authored May 27, 2026
  
  902e3032
26 May, 2026 1 commit
- opt h_q 128 sparse prefill · b4f69d84
  zhanghj2 authored May 26, 2026
  
  b4f69d84
25 May, 2026 3 commits
- tail guard to opt sparse prefill · e83a4119
  zhanghj2 authored May 25, 2026
  
  e83a4119
- Clean up redundant code · 9a805181
  zhanghj2 authored May 25, 2026
  
  9a805181
- test have_topk_length performance · bd0083f5
  zhanghj2 authored May 25, 2026
  
  bd0083f5
22 May, 2026 2 commits
- 优化sparse prefill · 0728420c
  zhanghj2 authored May 22, 2026
  
  0728420c
- Optimize sparse prefill d512 h64 topk sink path · 1cb8a563
  zhanghj2 authored May 22, 2026
  
  1cb8a563
21 May, 2026 1 commit
- update error info · 79408d6d
  zhanghj2 authored May 21, 2026
  
  79408d6d
29 Apr, 2026 2 commits
- Merge branch 'feature/kme-bf16' into 'master' · 92a05388
  zhanghj2 authored Apr 29, 2026
```
支持kme dense bf16

See merge request dcutoolkit/deeplearing/flashmla!11
```
  92a05388
- 支持kme dense bf16 · 8b581a54
  zhanghj2 authored Apr 29, 2026
  
  8b581a54
20 Apr, 2026 2 commits
- Merge branch 'feature/kimi-nhead64-dense' into 'master' · c85c787e
  zhanghj2 authored Apr 20, 2026
```
Feature/kimi nhead64 dense

See merge request dcutoolkit/deeplearing/flashmla!10
```
  c85c787e
- Feature/kimi nhead64 dense · aec17474
  zhanghj2 authored Apr 20, 2026
  
  aec17474
14 Apr, 2026 3 commits
- Merge branch 'feature/decode-fp8-opt' into 'master' · a45f646b
  zhanghj2 authored Apr 14, 2026
```
优化sparse decode fp8

See merge request dcutoolkit/deeplearing/flashmla!8
```
  a45f646b
- 优化sparse decode fp8 · 29b0d13b
  zhanghj2 authored Apr 14, 2026
  
  29b0d13b
- update version to 1.2.0 · 4648ec2f
  zhanghj2 authored Apr 14, 2026
  
  4648ec2f
10 Apr, 2026 1 commit
- FLASH_MLA_BF16_TYPE=1 · 261aa80c
  zhanghj2 authored Apr 10, 2026
  
  261aa80c
08 Apr, 2026 1 commit
- Merge branch 'fix/data-hazard' into 'master' · fd342087
  zhanghj2 authored Apr 08, 2026
```
fix buffer load lds data hazard

See merge request dcutoolkit/deeplearing/flashmla!7
```
  fd342087
07 Apr, 2026 3 commits
- fix buffer load lds data hazard · 1dea361d
  zhanghj2 authored Apr 07, 2026
  
  1dea361d
- Merge branch 'feature/bmz-das-prefill-glm5-nheads64' into 'master' · 8a69b46c
  zhanghj2 authored Apr 07, 2026
```
优化 nmz和bmz dsa prefill，nhead=64

See merge request dcutoolkit/deeplearing/flashmla!6
```
  8a69b46c
- 优化 nmz和bmz dsa prefill，nhead=64 · 14b2cfc5
  zhanghj2 authored Apr 07, 2026
  
  14b2cfc5
31 Mar, 2026 2 commits
- Merge branch 'flashmla-setup-with-torchver' into 'master' · a9ef79c6
  zhangyh15 authored Mar 31, 2026
```
[FEATURE]打包时指定环境变量PYTORCH_VERSION依赖可以带上torch版本

See merge request dcutoolkit/deeplearing/flashmla!5
```
  a9ef79c6
- 打包时依赖可以带上torch版本 · b9856c3a
  zhangyh15 authored Mar 31, 2026
  
  b9856c3a
23 Mar, 2026 2 commits
- 优化prefill sparse写出 · f01246b6
  zhanghj2 authored Mar 23, 2026
  
  f01246b6
- 加入打印使用的环境变量 · c3a5b02a
  zhanghj2 authored Mar 23, 2026
  
  c3a5b02a
20 Mar, 2026 2 commits
- fix fp8 e5m2融合问题 · fd2b2d8f
  zhanghj2 authored Mar 20, 2026
  
  fd2b2d8f
- 加入print_param · 2ff5a773
  zhanghj2 authored Mar 20, 2026
  
  2ff5a773
17 Mar, 2026 1 commit
- 使用__builtin_hcu_ds_read_m32x32_i8_alt2指令 · 79096f6b
  zhanghj2 authored Mar 17, 2026
  
  79096f6b
14 Mar, 2026 1 commit
- 使用64位计算stride,避免越界 · 4b3bcb50
  zhanghj2 authored Mar 14, 2026
  
  4b3bcb50
13 Mar, 2026 1 commit
- fix h_q < 16 bug · 84f83034
  zhanghj2 authored Mar 13, 2026
  
  84f83034
12 Mar, 2026 1 commit
- 优化nmz fp8 tp1 · 40f4bf39
  zhanghj2 authored Mar 12, 2026
  
  40f4bf39
09 Mar, 2026 1 commit
- fix mha_fwd_kvcache_mla_nope_pe接口架构判断 · 5e577dee
  zhanghj2 authored Mar 09, 2026
  
  5e577dee
07 Mar, 2026 1 commit
- 优化nmz buffer load提升fp8性能 · 79c06a56
  zhanghj2 authored Mar 07, 2026
  
  79c06a56
06 Mar, 2026 2 commits
- 添加编译选项 · 276a1fb7
  zhanghj2 authored Mar 06, 2026
  
  276a1fb7
- 整理接口 · ed798e2e
  zhanghj2 authored Mar 06, 2026
  
  ed798e2e
05 Mar, 2026 3 commits
- 融合类算子合并到flash_mla_with_kvcache_fp8_with_cat · 1b95bb9e
  zhanghj2 authored Mar 05, 2026
  
  1b95bb9e
- fix k_scale 未定义 · 113ee450
  zhanghj2 authored Mar 05, 2026
  
  113ee450
- e5m2接口合并到flash_mla_with_kvcache_fp8 · 702e8c22
  zhanghj2 authored Mar 05, 2026
  
  702e8c22
04 Mar, 2026 1 commit
- get_mla_decoding_metadata_dense_fp8和社区保持一致 · 7949f854
  zhanghj2 authored Mar 04, 2026
  
  7949f854
03 Mar, 2026 1 commit
- 优化mtp场景 · b894e2da
  zhanghj2 authored Mar 03, 2026
  
  b894e2da
02 Mar, 2026 1 commit
- 优化nmz fp8 tp8 · 7efb944d
  zhanghj2 authored Mar 02, 2026
  
  7efb944d