跳到正文
arXiv cs.AI· Joon Ha Kim, Geon-Woo Kim, Anoop Rachakonda, Daehyeok Kim·· 12 小时前AI 评分31

Dooly:面向 LLM 推理模拟的配置无关、冗余感知性能分析工具

Dooly: Configuration-Agnostic, Redundancy-Aware Profiling for LLM Inference Simulation

AI 导读

Dooly 通过一次推理遍历与污点传播标记输入维度来源,只对延迟数据库中缺失的算子做性能分析,实现配置无关、冗余感知的 profiling。在两种 GPU 平台、三种 attention 后端上,其 TTFT 模拟精度在 5% MAPE 内、TPOT 在 8% 内,12 个模型的 profiling GPU 时相比现有方法减少 56.4%。Dooly 已开源,可作为现有模拟器的即插即用后端。

正文 · AI 翻译

查看 PDF HTML(实验性)

摘要:选择最优的 LLM 推理配置需要在硬件、服务引擎、注意力后端和模型架构之间进行评估,因为没有任何单一选择能在所有工作负载下都表现最佳。基于性能分析的模拟器是标准工具,但它们将操作集硬编码为特定配置,并从头重新分析每个操作,使得探索成本高得令人望而却步。这一成本源于缺少对结构的理解:每个操作的每个输入维度要么由模型配置固定,要么由传入请求决定。许多模型配置值(例如头大小、层数)在不同模型间重复出现,因此同一操作会在多种配置下运行;对依赖请求的维度进行一次扫描即可服务所有这些配置。我们提出 Dooly,它利用这一结构实现与配置无关、感知冗余的性能分析。Dooly 执行一次推理过程,通过污点传播为每个输入维度标注其来源,并仅选择性地分析其延迟数据库中不存在的操作;注意力等有状态操作通过复用服务引擎自身的初始化代码进行隔离,从而无需手动插桩。基于该数据库,它构建延迟回归模型,作为现有模拟器的即插即用后端。在两个 GPU 平台、三种注意力后端和多种模型架构上,Dooly 对 TTFT 的模拟精度达到 5% MAPE 以内,对 TPOT 达到 8% 以内,同时与现有性能分析方法相比,将 12 个模型的性能分析 GPU 小时数减少了 56.4%。我们已在 此 https URL 开源 Dooly。
主题: 分布式、并行与集群计算(cs.DC);人工智能(cs.AI)
引用为: arXiv:2605.07985 [cs.DC]
  (或 arXiv:2605.07985v3 [cs.DC] 用于此版本)
  https://doi.org/10.48550/arXiv.2605.07985

由 DataCite 颁发的 arXiv DOI

提交历史

来自:Joon Ha Kim [查看邮箱]
[v1] 2026年5月8日星期五 16:44:47 UTC(766 KB)
[v2] 2026年5月21日星期四 13:49:15 UTC(1,185 KB)
[v3] 2026年10月20日星期二 18:41:32 UTC(774 KB)

来源:arXiv cs.AI · arxiv.org