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?
Thank you! We'll create more content like this.
Thank you for helping us improve!