跳到正文
elvis· @omarsar0 · X·· 3 小时前AI 评分66
AI 导读

Google 团队用 AI 智能体集群查找并修复 TPU 编译器中的静默错误,目前已解决 142 个以上,另有 103 项发现仍在审查中。团队把合成图与真实生产图上的 fuzzing 与现有 autotuner 旁的检查结合,用二分流水线把整个工作负载的失败定位到融合内核中的单个算子,智能体负责生成压力测试、复现失败并编写修复。

正文

Banger paper from Google on using agent swarms to fix Google's TPU compiler.

The team used swarms of AI agents to find and fix silent compiler bugs, and the framework has resolved over 142 of them so far.

They find silent miscompilations are the most dangerous. The compiler raises no error, but numbers drift, or buffers get corrupted, which can ruin a large training run or degrade live serving.

The team combines fuzzing on synthetic and real production graphs with checks that run alongside the existing autotuner. A bisection pipeline narrows a failure in a full workload down to one operation inside a fused kernel.

The agents generate stress tests, reproduce failures, and write fixes.

The team steered them away from easy crash bugs and toward silent data corruption, and 103 more findings are under review.

They also report that reviewer effort grew faster than the complexity of the fixes, and agent patches needed adversarial pre-filtering before humans looked at them.

https://arxiv.org/abs/2611.00624

Chat with Paper: https://academy.dair.ai/papers/xla-from-the-trenches-correctness-on-tpu-workloads-at-scale-2611.00624

来源:elvis · x.com