PyTorch 推出 CUDA Tile:Python DSL 写 tile 级内核自动吃满 Tensor Core

PyTorch · x · 2026-11-10

PyTorch 官方介绍 CUDA Tile:一种基于 Python 的 DSL 和以 tile 为中心的原语,可自动面向 Tensor Core 等高级 GPU 硬件特性。它能把 PyTorch 中分开计算多项的直接实现融合成单个 tiled kernel 并把矩阵保留在寄存器中。官方博客给出示例、文档与样例代码,展示如何写出清晰易读且高性能的内核。

原文链接 →

「Infra」频道最新

更多「Infra」频道 AI 资讯 →