一行代码,Google 想用它撬开英伟达 CUDA 筑了十几年的墙。Google 与 Meta 联合开源的 TorchTPU,给 TPU 补上了原生 PyTorch 后端,模型迁移的动作被压缩成把 .to(“cuda”) 改成 .to(“tpu”)。听着像营销话术,但放在 PyTorch 长期被 CUDA 生态绑死的背景下,这一行代码的分量并不轻。
PyTorch 的用户群有多大,CUDA 的护城河就有多深。过去要把训练任务搬到 TPU,团队得面对 JAX 或 TensorFlow 的另一套写法,重写模型、重调分布式、重跑基准,隐性成本足以劝退大多数中小团队。TorchTPU 的价值不在于让 TPU 变快,而在于把“换硬件”从工程项目降级成一次配置修改。Google 出芯片和运行时,Meta 出框架和社区,两家在算力紧缺、成本高企的当下有共同的算盘:让开发者少一个留在英伟达生态里的理由。
但一行迁移眼下更像入口,而不是承诺。算子覆盖、混合精度、多卡通信、性能调优,这些才决定模型能不能真跑起来,而它们往往藏在那行代码之后。TorchTPU 现在最实际的意义,是给 PyTorch 用户一个低成本试错的机会:先把模型跑通,再谈跑得好不好。对英伟达来说,这是芯片对手与框架主家第一次联手正面敲门。门会不会开,要看接下来几个版本的后端成熟度,也要看社区愿不愿意真把 .to(“tpu”) 写进生产代码。

