Skip to main content

Mojo trait

AccumulatorTile

Implemented traits

AnyType, Copyable, Movable, UnknownDestructibility

Aliases

dtype

alias dtype

element_layout

alias element_layout

rows_of_frags_layout

alias rows_of_frags_layout

vec_output_layout

alias vec_output_layout

Methods

__copyinit__

__copyinit__(out self: _Self, existing: _Self, /)

Create a new instance of the value by copying an existing one.

Args:

  • existing (_Self): The value to copy.

__moveinit__

__moveinit__(out self: _Self, owned existing: _Self, /)

Create a new instance of the value by moving the value of another.

Args:

  • existing (_Self): The value to move.

rows_of_frags

static rows_of_frags(src: LayoutTensor[get_vtable_entry(:trait<_nn::_mha_sm100::_AccumulatorTile> _Self, "dtype"), get_vtable_entry(:trait<_nn::_mha_sm100::_AccumulatorTile> _Self, "vec_output_layout"), MutableAnyOrigin, address_space=AddressSpace(5), element_layout=get_vtable_entry(:trait<_nn::_mha_sm100::_AccumulatorTile> _Self, "element_layout")]) -> LayoutTensor[get_vtable_entry(:trait<_nn::_mha_sm100::_AccumulatorTile> _Self, "dtype"), get_vtable_entry(:trait<_nn::_mha_sm100::_AccumulatorTile> _Self, "rows_of_frags_layout"), MutableAnyOrigin, address_space=AddressSpace(5)]

allocate_register_tile

allocate_register_tile(self: _Self) -> LayoutTensor[get_vtable_entry(:trait<_nn::_mha_sm100::_AccumulatorTile> _Self, "dtype"), get_vtable_entry(:trait<_nn::_mha_sm100::_AccumulatorTile> _Self, "vec_output_layout"), MutableAnyOrigin, address_space=AddressSpace(5), element_layout=get_vtable_entry(:trait<_nn::_mha_sm100::_AccumulatorTile> _Self, "element_layout")]

copy_from

copy_from(self: _Self, src: LayoutTensor[get_vtable_entry(:trait<_nn::_mha_sm100::_AccumulatorTile> _Self, "dtype"), get_vtable_entry(:trait<_nn::_mha_sm100::_AccumulatorTile> _Self, "vec_output_layout"), MutableAnyOrigin, address_space=AddressSpace(5), element_layout=get_vtable_entry(:trait<_nn::_mha_sm100::_AccumulatorTile> _Self, "element_layout")])

copy_to

copy_to(self: _Self, dst: LayoutTensor[get_vtable_entry(:trait<_nn::_mha_sm100::_AccumulatorTile> _Self, "dtype"), get_vtable_entry(:trait<_nn::_mha_sm100::_AccumulatorTile> _Self, "vec_output_layout"), MutableAnyOrigin, address_space=AddressSpace(5), element_layout=get_vtable_entry(:trait<_nn::_mha_sm100::_AccumulatorTile> _Self, "element_layout")])

Was this page helpful?