baidu/vLLM-Kunlun
vLLM Kunlun (vllm-kunlun) is a community-maintained hardware plugin designed to seamlessly run vLLM on the Kunlun XPU.
解决的问题
它使vLLM推理引擎能够在Kunlun XPU硬件上运行。通过提供硬件可插拔接口,用户无需修改vLLM核心源代码,即可在Kunlun XPU上部署流行的开源LLM和多模态模型。
工作原理
该项目作为vLLM的社区维护硬件插件。它实现了必要的平台特定逻辑、自定义算子(如FlashMLA和融合MoE内核)以及量化支持,以弥合vLLM高级引擎与Kunlun XPU架构之间的差距。它还包含用于硬件加速图优化的“Piecewise Kunlun Graph”。
适用对象
适用于使用Kunlun3 P800硬件,并希望利用vLLM高性能推理能力来处理各种生成式和多模态模型的开发者和AI工程师。
亮点
- 广泛的模型支持:兼容20多种主流模型,包括Qwen、Llama、DeepSeek、GLM和Gemma4。
- 硬件加速:具有Piecewise Kunlun Graph和FlashMLA等优化注意力机制。
- 量化:支持W8A8、AWQ、GPTQ和压缩张量W4A16。
- 高级解码:通过MTP(多令牌预测)和DFlash/EAGLE风格提议器实现推测解码。
- OpenAI兼容:提供标准OpenAI API接口用于模型服务。
相关
- 项目
- 项目
- 项目
- 项目
- 项目