Skip to main content

/

Docs

Nightly

v24.6

Log in

Mojo struct

TMATensorTile

struct TMATensorTile[dtype: DType, layout: Layout, desc_layout: Layout = $1]

Fields​

  • ​descriptor (TMADescriptor):

Implemented traits​

AnyType, Copyable, ExplicitlyCopyable, Movable, UnknownDestructibility

Methods​

__init__​

@implicit __init__(out self, descriptor: TMADescriptor)

__copyinit__​

__copyinit__(out self, other: Self)

async_copy​

async_copy(self, dst: LayoutTensor[dtype, layout, rank, address_space=3, element_layout=element_layout, layout_bitwidth=layout_bitwidth, masked=masked, alignment=alignment], mem_barrier: TMABarrier, coords: Tuple[UInt, UInt])

async_multicast_load​

async_multicast_load(self, dst: LayoutTensor[dtype, layout, rank, address_space=3, element_layout=element_layout, layout_bitwidth=layout_bitwidth, masked=masked, alignment=alignment], mem_barrier: TMABarrier, coords: Tuple[UInt, UInt], multicast_mask: SIMD[uint16, 1])

async_store​

async_store(self, src: LayoutTensor[dtype, layout, layout.rank(), address_space=3, element_layout=element_layout, layout_bitwidth=layout_bitwidth, masked=masked, alignment=alignment], coords: Tuple[UInt, UInt])

async_reduce​

async_reduce[reduction_kind: StringLiteral](self, src: LayoutTensor[dtype, layout, layout.rank(), address_space=3, element_layout=element_layout, layout_bitwidth=layout_bitwidth, masked=masked, alignment=alignment], coords: Tuple[UInt, UInt])