Tao20060602 开源 llm-gpu-kernels:RTX 3080 单GPU上的 MoE grouped GEMM 与 GQA attention 算子项目
作者开源了独立于 NanoKV 的 LLM GPU 算子项目 llm-gpu-kernels,包含 BF16 MoE grouped GEMM 局部融合与四段KV GQA prefill attention,硬件为 RTX 3080 Laptop 16 GiB(SM86)。attention 短query冻结选择在合成验证矩阵中相对完整 FA2 链路 Graph 耗时几何平均减少 22.09%,最坏慢 2.17%;M24 对 W4A16 packed gate/up 的布局机制对照中六个固定 case 相对同量化 vLLM 低 21.17%。项目按 M0–M24 分阶段报告,记录了失败实验、适用形状边界与回退策略,作者明确说明为单 GPU 原型,不构成 expert-parallel 或 serving 经验。
推荐理由:作者用逐阶段报告和固定输入复测记录了 MoE 与 attention 算子在 RTX 3080 上的实测收益和明确边界,可迁移到类似单GPU原型工作。