For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /get-started.md).
Mojo struct
TMADestination
struct TMADestination[dtype: DType, smem_elems: Int]
Pairs a shared memory TileTensor with a barrier for TMA operations.
The stored TileTensor uses a flat row_major[smem_elems]() layout.
TMA only uses .ptr.
Parameters
- dtype (
DType): Element dtype of the shared memory tile. - smem_elems (
Int): Number of elements in the flat shared memory buffer used by the TileTensor.
Fields
- mbar (
MBarType): - smem (
TMADestination[dtype, smem_elems].SmemType):
Implemented traits
AnyType,
Copyable,
Deinitable,
ImplicitlyCopyable,
Movable,
RegisterPassable,
TrivialRegisterPassable
comptime members
SmemType
comptime SmemType = TileTensor[dtype, Layout[*?, *?], MutAnyOrigin, address_space=AddressSpace.SHARED]
Methods
__init__
def __init__(mbar: Pointer[SharedMemBarrier, MutAnyOrigin, address_space=AddressSpace.SHARED], smem: TileTensor[dtype, Layout[*?, *?], MutAnyOrigin, address_space=AddressSpace.SHARED]) -> Self