SuperInfer通过智能调度与内存优化,让超芯片高效响应大模型请求。
SuperInfer: SLO-Aware Rotary Scheduling and Memory Management for LLM Inference on Superchips
- 设计旋转调度器,主动轮换请求保持低延迟
- 在GH200上使首次词生成延迟达标率提升74.7%
- 适合追求高响应速度的AI服务部署者
大型语言模型(LLM)服务面临严格延迟服务等级目标(SLO)与有限GPU内存之间的根本矛盾。当高请求速率耗尽KV缓存预算时,现有系统常出现严重队头阻塞。尽管先前工作探索了基于PCIe的卸载方案,但在高请求率下难以维持响应性,往往无法满足严格的首词生成时间(TTFT)和词间间隔(TBT)SLO。我们提出SuperInfer,一种专为新兴超芯片(如NVIDIA GH200)设计的高性能LLM推理系统,利用通过NVLink-C2C紧密耦合的GPU-CPU架构。SuperInfer引入RotaSched,首个主动式、SLO感知的旋转调度器,可在超芯片上维持响应性;并设计DuplexKV,一种优化的旋转引擎,实现NVLink-C2C上的全双工传输。在GH200上使用多种模型和数据集的评估显示,SuperInfer将TTFT SLO达成率最高提升74.7%,同时保持与顶尖系统相当的TBT和吞吐量,证明了SLO感知调度与内存协同设计能充分释放超芯片在响应式LLM服务中的潜力。代码已开源于https://github.com/Supercomputing-System-AI-Lab/SuperInfer。
原文摘要 · Abstract (English)
Large Language Model (LLM) serving faces a fundamental tension between stringent latency Service Level Objectives (SLOs) and limited GPU memory capacity. When high request rates exhaust the KV cache budget, existing LLM inference systems often suffer severe head-of-line (HOL) blocking. While prior work explored PCIe-based offloading, these approaches cannot sustain responsiveness under high request rates, often failing to meet tight Time-To-First-Token (TTFT) and Time-Between-Tokens (TBT) SLOs. We present SuperInfer, a high-performance LLM inference system designed for emerging Superchips (e.g., NVIDIA GH200) with tightly coupled GPU-CPU architecture via NVLink-C2C. SuperInfer introduces RotaSched, the first proactive, SLO-aware rotary scheduler that rotates requests to maintain responsiveness on Superchips, and DuplexKV, an optimized rotation engine that enables full-duplex transfer over NVLink-C2C. Evaluations on GH200 using various models and datasets show that SuperInfer improves TTFT SLO attainment rates by up to 74.7% while maintaining comparable TBT and throughput compared to state-of-the-art systems, demonstrating that SLO-aware scheduling and memory co-design unlocks the full potential of Superchips for responsive LLM serving. Code is available in https://github.com/Supercomputing-System-AI-Lab/SuperInfer.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。