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」频道最新
- Anthropic 营收三年暴涨 4000 倍,上半年已入账 163 亿美元 — daniel_mac8 · 2026-10-27
- Tenstorrent 芯片跑通蛋白质生成模型,宣称无 HBM 短缺之忧 — gajesh · 2026-10-27
- YC 系创企 Hopper 出击语音推理：TTFT 80ms、价格为 GPT-4.1 的 15% — ycombinator · 2026-10-27
- a16z 数据:A100 租赁价格 2026 年仍在涨,旧卡并未贬值 — VraserX · 2026-10-27
- Anthropic S-1 曝超 8000 亿美元未入表的未来算力承诺 — GaryMarcus · 2026-10-27
- Anthropic 约 11GW 算力合同中 NVIDIA GPU 占约 42% — ns123abc · 2026-10-27