← 最新论文
🤖 machine learning

Instant GPU Efficiency Visibility at Fleet Scale

本文介绍了整体浮点运算利用率(OFU),这是一种基于片上性能计数器、无需仪器化介入的硬件级 GPU 效率指标,它能够在多样化的工作负载和 GPU 代际中高精度地预测应用级的 MFU,从而支持大规模集群的有效监控并检测效率回退。

原作者: Connor Pedersen, Dong H. Ahn, Michel Migdal, Collin Neale, Nik Konyuchenko

发布于 2026-05-21
📖 1 分钟阅读☕ 轻松阅读

原作者: Connor Pedersen, Dong H. Ahn, Michel Migdal, Collin Neale, Nik Konyuchenko

原始论文采用 CC BY 4.0 许可(http://creativecommons.org/licenses/by/4.0/)。 这是对下方论文的AI生成解释。它不是由作者撰写或认可的。如需技术准确性,请参阅原始论文。 阅读完整免责声明

想象一下,你拥有一支庞大的送货卡车车队(GPU),它们本应负责在全国范围内运输重型货物(AI 计算)。你想知道:这些卡车真的在移动,还是仅仅在交通中怠速空转?

长期以来,检查这一点曾是一场噩梦。你不得不要求每一位司机(软件)手动记录他们运送了多少货物。但司机们很忙,可能会忘记,或者可能谎报运载量。此外,如果你明年购买了新型卡车,你就必须重写所有关于如何计算货物的规则。

本文介绍了一种更聪明的检查方法:总体浮点运算利用率(OFU)

以下是作者所做工作及其重要性的简明分解。

问题所在:“司机日志”已失效

目前,大型科技公司试图通过让 AI 软件自行统计其工作量来衡量效率。

  • 缺陷: 软件经常算错数学题。它可能以为自己运送了 100 个箱子,而实际上只运送了 50 个。
  • 后果: 在作者发现的一个真实案例中,一个训练任务看起来运行效率高达54%(很棒!),但当他们检查实际硬件时,发现其运行效率仅为25%(糟糕透顶!)。软件之所以撒谎,是因为它不理解自己正在运送的那种新型、复杂的货物。

解决方案:“速度表与发动机警示灯”

作者没有询问司机做了什么,而是构建了一个直接查看卡车仪表盘的系统。他们利用每个 NVIDIA GPU 都已报告的两个简单信号,创建了一个名为OFU的指标:

  1. 张量管道活动(Tensor Pipe Activity): 这就像一盏灯,每当发动机实际在进行数值运算时就会亮起。
  2. SM 时钟频率(SM Clock Frequency): 这是速度表,告诉你发动机转动的速度。

通过将“灯亮”的时间乘以“发动机转速”,他们获得了一个完美、实时的 GPU 工作强度估算。无论运送的是什么货物(AI 模型),也无论司机说什么语言,硬件都知道自己是否在运转。

“隐藏成本”(为何并非 100% 完美)

作者意识到,即使查看仪表盘也存在一些怪异之处,因此他们进行了数千次测试以绘制出这些情况:

  • “瓦片”问题: 想象你在打包一个箱子。如果箱子比物品稍大,你就必须用气泡膜(填充物)填满空隙。GPU 也会这样做。它会计算一些实际上对 AI 无用的“气泡膜”数学运算。作者精确计算出了这种额外运算的量,以便将其从总量中扣除。
  • “速度表故障”: 有时,由于发动机转速变化迅速,速度表会闪烁。作者发现,如果你检查频率足够高(每隔几秒),这些闪烁就会相互抵消,得出的数值非常准确。
  • “小部件”问题: GPU 有一个主发动机(张量核心)和一个用于处理小任务的微型侧发动机(CUDA 核心)。作者发现,主发动机承担了 99.8% 的重活,因此忽略那个微小的侧发动机并不会真正改变结果。

结果:捕捉漏洞

当他们用这种新的“仪表盘”方法测试 608 个真实世界的 AI 训练任务时,效果惊人:

  • 与事实相符: 它与实际完成的工作量高度相关。
  • 识破谎言: 它发现了两个软件错误计算工作量的重大案例。其中一个是一个复杂的“混合专家”(Mixture of Experts)模型,软件忘记计算一个步骤,使得该任务看起来比实际效率高出一倍。
  • 节省资金: 在一个机器人训练 AI 的案例中,仪表盘显示卡车正在怠速。团队调查发现,一个“调试模式”被意外开启,迫使卡车不断停下来检查文件。他们关闭了该模式,效率提升了2.5 倍

结论

作者不仅发明了一个新的数学公式,他们构建了一种通用、即时且诚实的方法,用于查看你的 AI 计算机是否真正在工作

  • 无需安装: 你无需更改 AI 代码。
  • 无处不在: 它适用于新旧 GPU,以及任何类型的 AI 模型。
  • 即时可见: 它能立即告诉你某个任务是否在浪费资金,让团队在损失数百万之前进行修复。

简而言之,OFU 就像在你的整个车队中的每辆卡车上安装了一个防篡改的燃油表,这样你就不必再猜测司机们是否真的在运送货物了。

您所在领域的论文太多了?

获取与您研究关键词匹配的最新论文每日摘要——附技术摘要,使用您的语言。

试用 Digest →