Skip to content

feat: add Stem sparse attention module#286

Merged
yghstill merged 4 commits intoTencent:mainfrom
loloto:stem
Apr 21, 2026
Merged

feat: add Stem sparse attention module#286
yghstill merged 4 commits intoTencent:mainfrom
loloto:stem

Conversation

@loloto
Copy link
Copy Markdown
Contributor

@loloto loloto commented Apr 17, 2026

添加 Stem 稀疏注意力模块(支持 torch/hpc bf16/fp8 后端),含推理脚本和文档。

Comment thread angelslim/compressor/sparsity/stem/backends/dispatcher.py
Comment thread angelslim/compressor/sparsity/stem/ops/stem_kernel.py
Comment thread docs/source/index.md
Comment thread tools/stem.py Outdated
Comment thread angelslim/compressor/sparsity/stem/stem.py
@yghstill yghstill merged commit 3a9c1a3 into Tencent:main Apr 21, 2026
5 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants