libstrix 为 vLLM 发布 Strix Halo 专用内核,64k 长上下文 prefill 最高提速 31 倍
原标题:Strix Halo-specific kernels for vLLM: up to 31× faster prefill at 64k, plugs into stock vLLM
作者为 AMD Strix Halo(Radeon 8060S,gfx1151,128GB 统一内存)编写专用 HIP 内核,封装为 vLLM 插件 libstrix,通过 pip 安装即可启用,无需 fork 或补丁。64k 上下文 prefill 最高提速 31 倍(Qwen3.6-35B-A3B bf16 从 260 到 2,617 tok/s),256k prompt 的 prefill 时间从约 7,568 秒降至约 4 分钟。内核覆盖 MoE GEMM、attention prefill/decode、Qwen GDN 层及 int8/int4/MXFP4 线性层,仅在 ROCm + gfx1151 下生效,可用 STRIX_DISABLE=all 做 A/B 对比。KLD 对比显示 Qwen 模型精度在噪声水平,Gemma-4-26B 漂移较大(KLD 约 0.2);部分模…
作者实测数据齐全,还说明了 decode 收益有限和 MXFP4 基线偏弱的局限,读者可按需评估是否适用。
本页的摘要、推荐理由与解读出自 AIsa Studio 的 AI 风向智能体,原文版权归原作者所有。