arXiv:2504.15303cs.DCcs.AI2025-04被引 3

优化异构集群的LLM推理配置与调度,提升吞吐量最多达122.5%。

High-Throughput LLM inference on Heterogeneous Clusters

  • 通过资源与吞吐建模,穷举搜索最优部署配置。
  • 在两个异构集群上分别提升吞吐量122.5%和33.6%。
  • 适合需要高效部署LLM推理服务的工业级团队。

当前许多公司拥有多种AI加速器,形成异构计算集群。高效利用这些集群进行大规模语言模型(LLM)推理服务,可显著降低成本并加快任务处理速度。然而,异构集群上的LLM推理面临两大挑战:其一,不同部署配置带来的性能差异巨大,配置空间庞大且评估复杂,难以找到最优方案;其二,集群内各实例处理能力不同,导致请求处理速度不一,准确评估其能力并设计能充分发挥每台设备潜力的调度算法极具挑战。本文提出一种面向异构集群的高吞吐推理服务系统:首先通过建模资源量与预期吞吐量,采用穷举搜索法优化部署配置;其次提出新型请求调度机制,充分考虑各实例的异构处理能力。大量实验表明,所提调度器在两个异构集群上分别实现122.5%和33.6%的吞吐量提升。

原文摘要 · Abstract (English)

Nowadays, many companies possess various types of AI accelerators, forming heterogeneous clusters. Efficiently leveraging these clusters for high-throughput large language model (LLM) inference services can significantly reduce costs and expedite task processing. However, LLM inference on heterogeneous clusters presents two main challenges. Firstly, different deployment configurations can result in vastly different performance. The number of possible configurations is large, and evaluating the effectiveness of a specific setup is complex. Thus, finding an optimal configuration is not an easy task. Secondly, LLM inference instances within a heterogeneous cluster possess varying processing capacities, leading to different processing speeds for handling inference requests. Evaluating these capacities and designing a request scheduling algorithm that fully maximizes the potential of each instance is challenging. In this paper, we propose a high-throughput inference service system on heterogeneous clusters. First, the deployment configuration is optimized by modeling the resource amount and expected throughput and using the exhaustive search method. Second, a novel mechanism is proposed to schedule requests among instances, which fully considers the different processing capabilities of various instances. Extensive experiments show that the proposed scheduler improves throughput by 122.5% and 33.6% on two heterogeneous clusters, respectively.

LLM推理异构集群调度优化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。