tilelang.layout.cute

CuTe layout IR objects and layout-algebra Python API, in TileLang.

Attributes

PyIntTuple
IntTupleLike
ModeLike

Classes

Swizzle
IntTuple
IntTupleConst
IntTuplePrimExpr
IntTupleScaledBasis
IntTupleTuple
ScaledBasis	A ScaledBasis wrapper.
Layout
ComposedLayout

Functions

to_python(t)	Unpack an IntTuple into Python: a branch becomes a nested tuple, and each leaf becomes a plain int (static), a PrimExpr (dynamic), or a ScaledBasis (basis stride).
from_python(value)	Convert a Python value to an FFI IntTuple (the inverse of to_python()): a nested tuple/list becomes an IntTupleTuple branch; an int becomes IntTupleConst; a PrimExpr becomes IntTuplePrimExpr; a ScaledBasis becomes IntTupleScaledBasis; an already-built IntTuple passes through.
E(mode)
product(shape)
flatten_to_tuple(value)
rank(layout)
flatten(layout)
size(layout)
coalesce(layout[, max_extent])
right_inverse(layout)
composition(lhs, rhs)
make_layout(shape[, stride])
make_column_major_layout(shape)
make_row_major_layout(shape)
make_identity_layout(shape)

Module Contents

tilelang.layout.cute.PyIntTuple

tilelang.layout.cute.IntTupleLike

tilelang.layout.cute.ModeLike

tilelang.layout.cute.to_python(t)

Unpack an IntTuple into Python: a branch becomes a nested tuple, and each leaf becomes a plain int (static), a PrimExpr (dynamic), or a ScaledBasis (basis stride).

Parameters:

t (IntTuple)

Return type:

PyIntTuple

tilelang.layout.cute.from_python(value)

Convert a Python value to an FFI IntTuple (the inverse of to_python()): a nested tuple/list becomes an IntTupleTuple branch; an int becomes IntTupleConst; a PrimExpr becomes IntTuplePrimExpr; a ScaledBasis becomes IntTupleScaledBasis; an already-built IntTuple passes through.

Parameters:

value (IntTupleLike)

Return type:

IntTuple

class tilelang.layout.cute.Swizzle

Bases: tvm.ir.base.Node, tvm.runtime.Scriptable

b_bits: int

m_base: int

s_shift: int

property is_swizzled: bool

Return type:

bool

recast(old_bits, new_bits)

Parameters:

• old_bits (int)

• new_bits (int)

Return type:

Swizzle

class tilelang.layout.cute.IntTuple

Bases: tvm.ir.base.Node, tvm.runtime.Scriptable

__add__(other)

Parameters:

other (IntTupleLike)

Return type:

IntTuple

__radd__(other)

Parameters:

other (IntTupleLike)

Return type:

IntTuple

__mul__(other)

Parameters:

other (IntTupleLike)

Return type:

IntTuple

__rmul__(other)

Parameters:

other (IntTupleLike)

Return type:

IntTuple

class tilelang.layout.cute.IntTupleConst

Bases: IntTuple

value: int

class tilelang.layout.cute.IntTuplePrimExpr

Bases: IntTuple

value: object

class tilelang.layout.cute.IntTupleScaledBasis

Bases: IntTuple

value: IntTuple

basis: list

class tilelang.layout.cute.IntTupleTuple

Bases: IntTuple

fields: list

class tilelang.layout.cute.ScaledBasis(value, mode)

A ScaledBasis wrapper.

Parameters:

• value (PyIntTuple)

• mode (ModeLike)

property value: PyIntTuple

Return type:

PyIntTuple

property mode: tuple

Return type:

tuple

__repr__()

Return type:

str

tilelang.layout.cute.E(mode)

Parameters:

mode (ModeLike)

Return type:

ScaledBasis

tilelang.layout.cute.product(shape)

Parameters:

shape (IntTupleLike)

Return type:

PyIntTuple

tilelang.layout.cute.flatten_to_tuple(value)

Parameters:

value (IntTupleLike)

Return type:

tuple

class tilelang.layout.cute.Layout

Bases: tvm.ir.base.Node, tvm.runtime.Scriptable

property shape

property stride

__getitem__(idx)

Parameters:

idx (int)

Return type:

Layout

__call__(coord)

Parameters:

coord (IntTupleLike)

static from_tilelang(layout)

Return type:

Layout | None

tilelang.layout.cute.rank(layout)

Parameters:

layout (Layout)

Return type:

int

tilelang.layout.cute.flatten(layout)

Parameters:

layout (Layout)

Return type:

Layout

tilelang.layout.cute.size(layout)

Parameters:

layout (Layout | ComposedLayout)

Return type:

PyIntTuple

tilelang.layout.cute.coalesce(layout, max_extent=None)

Parameters:

• layout (Layout)

• max_extent (int | None)

Return type:

Layout

tilelang.layout.cute.right_inverse(layout)

Parameters:

layout (Layout)

Return type:

Layout

tilelang.layout.cute.composition(lhs, rhs)

Parameters:

• lhs (Layout)

• rhs (Layout)

Return type:

Layout

tilelang.layout.cute.make_layout(shape, stride=None)

Parameters:

shape (PyIntTuple)

Return type:

Layout

tilelang.layout.cute.make_column_major_layout(shape)

Parameters:

shape (PyIntTuple)

Return type:

Layout

tilelang.layout.cute.make_row_major_layout(shape)

Parameters:

shape (PyIntTuple)

Return type:

Layout

tilelang.layout.cute.make_identity_layout(shape)

Parameters:

shape (PyIntTuple)

Return type:

Layout

class tilelang.layout.cute.ComposedLayout

Bases: tvm.ir.base.Node, tvm.runtime.Scriptable

swizzle: Swizzle

offset: int

layout: Layout

recast(old_bits, new_bits)

Parameters:

• old_bits (int)

• new_bits (int)

Return type:

ComposedLayout

static from_tilelang(layout, buffer=None)

Parameters:

buffer (tilelang._typing.BufferLikeType)

Return type:

ComposedLayout | None