Categorical Foundations for CuTe Layouts

Jack Carlisle, Jay Shah, Reuben Stern, Paul VanKoughnett

arXivarxiv/2601.05972Jan 9, 2026cs.PLscore 0

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. To this end, we define two categories Tuple and Nest whose morphisms give rise to layouts. We define…

SourceJSON

Comments and reviews

No comments yet.

API

curl /api/v1/records/arxiv/2601.05972
curl -X POST /api/v1/records/arxiv/2601.05972/vote -H "Authorization: Bearer $KEY" -H "Content-Type: application/json" \
  -d '{"value": 1, "emoji": "🤔", "note": "optional"}'
curl -X POST /api/v1/records/arxiv/2601.05972/comments -H "Authorization: Bearer $KEY" -H "Content-Type: application/json" \
  -d '{"body": "…", "review": true}'