TPU 的 fp32 精度陷阱
JFPuget · x · 2026-07-20
这条帖结合配图指出了一个 TPU 上很容易踩坑的细节：即使你请求的是 fp32 matmul，TPU 默认也可能按 bf16 执行。
作者提到，训练命令里的 --lm-head-precision highest 看起来像是在保护质量，但实际上并不起作用，因为 TPU 的默认行为会把这类计算落到 bf16。配图里的 JAX 例子也说明了这一点：输入是 fp32，但默认精度路径仍然会被使用。
作者强调，这不是隐藏 bug，而是文档里写明的默认行为；甚至连 Google 研究员都曾因此反复踩坑。核心提醒是：在 TPU 上不要被参数名字误导，必须真正理解平台的精度执行语义。
「Infra」频道最新
- 500 万美元种子轮不够 DeepSeek 训一遍:没 GPU 的 AI 创业只剩三条路 — FinanceYF5 · 2026-10-26
- HBM 需求转向:低端推理砍单,Rubin 将全系 12-high HBM — BenBajarin · 2026-10-26
- CircuitHub 发布工厂级机器人 The Grid，美国产 PCB 提速两倍对标深圳 — garrytan · 2026-10-26
- Ubicloud 发文修复 vLLM 工具调用准确率问题 — petrusenko_max · 2026-10-26
- Meta 批准无投票权 C 类股，2026 年资本开支指引达 1350–1450 亿美元 — thione · 2026-10-26
- Unsloth 发布 MiniMax-M3.1-Flash 量化：2-bit 保留 91.5% 精度，150GB 内存可跑 — danielhanchen · 2026-10-26