细粒度调度让消费级设备高效运行大模型
Automated Tensor Scheduling for Hybrid CPU-GPU LLM Inference on Consumer Devices
- 按张量级别动态调度,突破传统层级调度局限
- 预填充吞吐提升1.94倍,解码速度最高快3.29倍
- 适合本地部署大模型的开发者与终端用户
在笔记本等消费级设备上运行大语言模型面临挑战,因模型权重常超出GPU显存容量,需借助CPU内存进行计算卸载以扩展有效模型规模。现有卸载系统多依赖粗粒度的层级或专家级调度,忽视同一层内张量间的显著异构性,且难以适应设备硬件负载的动态变化。本文提出ATSInfer,一种面向消费设备的混合CPU-GPU推理系统,实现张量级卸载。该系统结合静态张量放置与负载感知的动态传输,并引入异步CPU-GPU协调机制,高效调度跨异构后端的存储、数据移动与计算。我们在代表性消费平台使用稠密模型和MoE模型对ATSInfer进行了实现与评估。相比现有系统,其预填充吞吐最高提升1.94倍,解码吞吐最高提升3.29倍,同时提高GPU利用率并更充分使用PCIe带宽。结果表明,ATSInfer可显著提升个人设备本地部署大模型的用户体验。
原文摘要 · Abstract (English)
Running large language models on consumer devices such as laptops and desktops is challenging because model weights often exceed GPU memory capacity, making offloading inference necessary to extend effective model capacity with CPU memory. Existing offloading systems, however, typically rely on coarse layer-level or expert-level scheduling, which overlooks substantial heterogeneity among tensors within the same layer and adapts poorly to changing hardware load conditions on such devices. This paper presents ATSInfer, a hybrid CPU-GPU inference system for consumer devices that performs offloading at tensor granularity. ATSInfer combines static tensor placement with load-aware dynamic transfer, and introduces asynchronous CPU-GPU coordination to efficiently schedule hardware storage, data movement, and computation across heterogeneous backends. We implement ATSInfer and evaluate it on representative consumer platforms using both dense and MoE models. Compared with existing systems, ATSInfer improves prefill throughput by up to 1.94$\times$ and decode throughput by up to 3.29$\times$, while also increasing GPU utilization and making more effective use of PCIe bandwidth. These results show that ATSInfer can substantially improve the user experience of local LLM deployment on personal consumer devices.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。