大模型推理引擎横评:vLLM、SGLang、TensorRT-LLM
推理引擎:LLM服务的核心基础设施 模型只是"大脑",推理引擎是"神经系统"——它决定了模型服务能多快、能并发多少请求、成本多低。选对推理引擎,单GPU的吞吐量差异可达5-10倍。 三大主流引擎 vLLM:社区标准 架构特点: PagedAttention:操作系统级KV Cache管理 Continuous Batching:迭代级动态批处理 纯Python实现(核心算子用CUDA/Triton) 支持大多数主流开源模型 核心技术:PagedAttention 传统KV Cache: 请求A: [已分配 5/5 blocks] [等待分配...] 请求B: [已分配 3/3 blocks] → 内存碎片化,浪费30-60% PagedAttention: Block Pool: [A1][B1][A2][C1][B2][A3]... 每个请求通过页表映射到物理block → 零碎片,内存利用率接近100% 性能特征: 吞吐量:比HuggingFace Transformers高24倍 首token延迟:中等 显存利用率:90%+ 支持多模型共享GPU 优势: 社区支持最广,几乎所有模型首发支持 部署简单(一行命令启动) 持续迭代,功能快速扩展 劣势: 纯Python实现,某些热路径不如C++优化 不支持一些极致优化(如FP8推理) 调度策略相对简单 SGLang:结构化生成之王 架构特点: 前缀缓存(Prefix Caching):跨请求共享计算 程序化生成:支持结构化输出约束 原生支持JSON/正则约束 C++核心,Python接口 核心技术:RadixAttention 请求1: "system: 你是助手\nuser: 天气" → 缓存KV Cache 请求2: "system: 你是助手\nuser: 股票" → 复用"system: 你是助手\n"的前缀KV Cache → 只需计算新的部分 多请求共享相同system prompt时,前缀缓存可节省50%+计算。 ...







