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接口用于模型服务。

相关

  • 项目
  • 项目
  • 项目
  • 项目
  • 项目