PyTorch's CUDA Tile: a Python DSL for tiling kernels onto Tensor Cores

PyTorch · x · 2026-11-10

PyTorch introduces CUDA Tile, a Python-based DSL with tile-centric primitives that automatically targets Tensor Cores and other advanced GPU features. It can take straightforward PyTorch code computing multiple terms separately and fuse it into one tiled kernel keeping matrices in registers. The post includes examples, docs and sample code.

Original post →

More from Infra

Infra channel →