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 인터페이스를 제공합니다.

관련

  • 프로젝트
  • 프로젝트
  • 프로젝트
  • 프로젝트
  • 프로젝트