PyTorch 的 Helion DSL 开始支持通过 Pallas 编写 TPU kernel
PyTorch · x · 2026-07-24
PyTorch 表示，面向自定义 kernel 的 DSL Helion 正在向 异构硬件 kernel 编写 扩展。
- Meta 和 Google 展示了 Helion 如何通过 Pallas 编译到 TPU。
- 目标是让开发者在 PyTorch 生态里，用一套统一方式编写性能更好的 TPU kernel。
- 文章用 FlashAttention 作为案例。
- 编译器会在多种代码生成策略之间 自动调优，并根据不同输入形状选择合适的 pipeline，以最大化计算与内存访问的重叠。
「Infra」频道最新
- Engram 随机读取不适合 SSD，CPU 内存低延迟共享或成正解 — bookwormengr · 2026-09-11
- 劝退帖:推理工程是真金白银的硬活,八成尝鲜者已悄然放弃 — abhijithneil · 2026-09-11
- Hugging Face《Ultra Scale Playbook》:GPU 集群训练 LLM 的免费技术书 — mdancho84 · 2026-09-11
- 从聊天到 Agent,推理延迟正在成为生产级瓶颈 — Euphoric_Sea632 · 2026-09-11
- 推理服务关键指标盘点:流式速度 TPOT 与可用性如何影响体验 — abhijithneil · 2026-09-11
- PlanetScale 推出分片 Postgres,768 台服务器组成 1PB 单库 — dhruv2038 · 2026-09-11