yangjianfengo1 3754a9906d [Feature] block sparse attention (#3668) ...

* 支持稀疏attn

* fix bug

* code style

* fix moba attn get kv shape

* 修复a100编译

* codestyle

* code style

* code style

* code style

* fix conflict

* 增加单侧

* code style

* 增加eblite 加载时间

* fix bug

* for ci

* for ci

* for ci

* for ci

* 支持mlp block size 128

* 增加小算子单测

* fix 单测 mlp

* 将环境变量加入到config里面

* fix rollout config

* 修复显存

* add test server

* add test server

* fix mlp  最后一层使用full attn

2025-08-29 19:46:30 +08:00

..

moba_decoder_attn_kernel.h

[Feature] block sparse attention (#3668)

2025-08-29 19:46:30 +08:00

moba_decoder_attn.cu

[Feature] block sparse attention (#3668)

2025-08-29 19:46:30 +08:00

moba_decoder_write_cache.cu

[Feature] block sparse attention (#3668)

2025-08-29 19:46:30 +08:00

moba_qk_sort_decoder.cu

[Feature] block sparse attention (#3668)

2025-08-29 19:46:30 +08:00