baidu/vLLM-Kunlun

vLLM Kunlun (vllm-kunlun) is a community-maintained hardware plugin designed to seamlessly run vLLM on the Kunlun XPU.

解決する課題

Kunlun XPUハードウェア上でvLLM推論エンジンを実行できるようにします。ハードウェアプラグイン可能なインターフェースを提供することで、ユーザーはvLLMのコアソースコードを変更することなく、人気のあるオープンソースLLMやマルチモーダルモデルをKunlun XPU上にデプロイできます。

仕組み

このプロジェクトは、vLLM用のコミュニティ保守型ハードウェアプラグインとして機能します。プラットフォーム固有のロジック、カスタムオペレーター(FlashMLAや融合MoEカーネルなど)、量子化サポートを実装し、vLLMの高レベルエンジンとKunlun XPUアーキテクチャの間のギャップを埋めます。また、ハードウェアアクセラレーションによるグラフ最適化のための「Piecewise Kunlun Graph」も含まれています。

対象ユーザー

Kunlun3 P800ハードウェアを使用し、さまざまな生成モデルやマルチモーダルモデルに対してvLLMの高性能推論機能を活用したい開発者やAIエンジニア向けです。

主な特徴

  • 幅広いモデルサポート: Qwen、Llama、DeepSeek、GLM、Gemma4など20以上の主流モデルに対応。
  • ハードウェアアクセラレーション: Piecewise Kunlun GraphとFlashMLAなどの最適化されたアテンションメカニズムを搭載。
  • 量子化: W8A8、AWQ、GPTQ、圧縮テンソルW4A16をサポート。
  • 高度なデコーディング: MTP(マルチトークン予測)とDFlash/EAGLEスタイルのプロポーザーによる投機的デコーディングを実装。
  • OpenAI互換: モデルを提供するための標準的なOpenAI APIインターフェースを提供。

関連

  • プロジェクト
  • プロジェクト
  • プロジェクト
  • プロジェクト
  • プロジェクト