Pallas 内核怎么调优?Google 给出的答案是“少信模型,多看寄存器”
Google 为 XProf 添加了一个内核性能分析套件。XProf 是其面向 TPU 工作负载的开源性能分析器。现在,开发者可以查看自定义 Pallas 内核中的周期级细节。此前,这些内核在跟踪记录中显示为单个不透明块。在 TPU v7(Ironwood)上,XProf 会在运行时对硬件性能计数器进行采样。在一个分块矩阵乘法示例中,Google 发现了内存停顿问题。通过添加三重缓冲,他们将内核执
事件进展(1 篇报道)
-
2026-10-10 22:22