基于SLURM的HPC架构实现大模型高效推理,支持多模型并发部署。
Scalable Engine and the Performance of Different LLM Models in a SLURM based HPC architecture
- 利用SLURM动态调度容器化服务,实现多节点资源协同管理。
- 1B/3B模型可支持128并发、延迟低于50ms,70B模型仅2并发即超2秒延迟。
- 适用于聊天机器人、批量推理等场景,适合大规模模型部署者。
本文提出一种基于Simple Linux Utility for Resource Management(SLURM)的高性能计算(HPC)架构,用于在异构环境下部署大规模语言模型(LLMs)并构建可扩展的推理引擎。通过动态资源调度与容器化微服务无缝集成,该架构在多节点集群中高效管理CPU、GPU和内存分配。实验使用Llama 3.2(1B和3B参数)及Llama 3.1(8B和70B参数)模型,评估了吞吐量、延迟和并发性能。结果表明,小模型可支持最多128个并发请求且延迟低于50毫秒;而大模型在仅2个并发用户时即出现饱和,延迟超过2秒。系统提供REST API接口支持单次与批量推理,以及多步骤“法庭式”精炼等高级工作流。实验验证了容器与调度开销极低,且在批处理与交互式场景下均具可靠可扩展性。此外,通过检索增强生成的聊天机器人部署案例,展示了架构的灵活性与鲁棒性。研究为大规模HPC基础设施上的高效、低延迟、高容错的LLM推理提供了可行路径。
原文摘要 · Abstract (English)
This work elaborates on a High performance computing (HPC) architecture based on Simple Linux Utility for Resource Management (SLURM) [1] for deploying heterogeneous Large Language Models (LLMs) into a scalable inference engine. Dynamic resource scheduling and seamless integration of containerized microservices have been leveraged herein to manage CPU, GPU, and memory allocations efficiently in multi-node clusters. Extensive experiments, using Llama 3.2 (1B and 3B parameters) [2] and Llama 3.1 (8B and 70B) [3], probe throughput, latency, and concurrency and show that small models can handle up to 128 concurrent requests at sub-50 ms latency, while for larger models, saturation happens with as few as two concurrent users, with a latency of more than 2 seconds. This architecture includes Representational State Transfer Application Programming Interfaces (REST APIs) [4] endpoints for single and bulk inferences, as well as advanced workflows such as multi-step "tribunal" refinement. Experimental results confirm minimal overhead from container and scheduling activities and show that the approach scales reliably both for batch and interactive settings. We further illustrate real-world scenarios, including the deployment of chatbots with retrievalaugmented generation, which helps to demonstrate the flexibility and robustness of the architecture. The obtained results pave ways for significantly more efficient, responsive, and fault-tolerant LLM inference on large-scale HPC infrastructures.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。