Categorical Foundations for Cute Layouts(arxiv.org)
arxiv.org
Categorical Foundations for Cute Layouts
https://arxiv.org/abs/2601.05972
1 comments
From the abstract: NVIDIA's CUTLASS library provides a robust and expressive set of methods for describing and manipulating multi-dimensional tensor data on the GPU. These methods are conceptually grounded in the abstract notion of a CuTe layout and a rich algebra of such layouts, including operations such as composition, logical product, and logical division. In this paper, we present a categorical framework for understanding this layout algebra by focusing on a naturally occurring class of tractable layouts.