tilelang.layout.swizzle_mode

Shared memory swizzle mode.

Classes

SwizzleMode

Module Contents

class tilelang.layout.swizzle_mode.SwizzleMode

Bases: tvm_ffi.dataclasses.Enum

NONE: ClassVar[SwizzleMode]

SWIZZLE_32B: ClassVar[SwizzleMode]

SWIZZLE_64B: ClassVar[SwizzleMode]

SWIZZLE_128B: ClassVar[SwizzleMode]

is_none()

Return type:

bool

is_swizzle_32b()

Return type:

bool

is_swizzle_64b()

Return type:

bool

is_swizzle_128b()

Return type:

bool

wgmma_layout_type()

WGMMA descriptor layout_type_ field (none->0, 32B->3, 64B->2, 128B->1).

Return type:

int

tcgen05_layout_type()

TCGEN05 descriptor swizzle field (none->0, 32B->6, 64B->4, 128B->2).

Return type:

int

swizzle_byte_size()

Swizzle size in bytes (none->1, else 32/64/128).

Return type:

int

swizzle_atom_size()

Swizzle size in 16-byte vectors (none->1, else 2/4/8).

Return type:

int

smem_alignment()

Required shared-memory base alignment in bytes (none->128, else 256/512/1024).

Return type:

int

static from_ordinal(ordinal)

Parameters:

ordinal (int)

Return type:

SwizzleMode