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.
Fields
- mbar (
MBarType): - smem (
TMADestination[dtype, smem_elems].SmemType):
Implemented traits
AnyType,
Copyable,
ImplicitlyCopyable,
ImplicitlyDestructible,
Movable,
RegisterPassable,
TrivialRegisterPassable
comptime members
SmemType
comptime SmemType = TileTensor[dtype, Layout[ComptimeInt[smem_elems], ComptimeInt[1]], MutAnyOrigin, address_space=AddressSpace.SHARED]
Methods
__init__
__init__(mbar: UnsafePointer[SharedMemBarrier, MutAnyOrigin, address_space=AddressSpace.SHARED], smem: TileTensor[dtype, Layout[ComptimeInt[smem_elems], ComptimeInt[1]], MutAnyOrigin, address_space=AddressSpace.SHARED]) -> Self
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!