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, var 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_witness(_Self, nn::mha_sm100::AccumulatorTile, dtype), get_witness(_Self, nn::mha_sm100::AccumulatorTile, vec_output_layout), MutableAnyOrigin, address_space=AddressSpace(5), element_layout=get_witness(_Self, nn::mha_sm100::AccumulatorTile, element_layout)]) -> LayoutTensor[get_witness(_Self, nn::mha_sm100::AccumulatorTile, dtype), get_witness(_Self, nn::mha_sm100::AccumulatorTile, rows_of_frags_layout), MutableAnyOrigin, address_space=AddressSpace(5)]

allocate_register_tile

allocate_register_tile(self: _Self) -> LayoutTensor[get_witness(_Self, nn::mha_sm100::AccumulatorTile, dtype), get_witness(_Self, nn::mha_sm100::AccumulatorTile, vec_output_layout), MutableAnyOrigin, address_space=AddressSpace(5), element_layout=get_witness(_Self, nn::mha_sm100::AccumulatorTile, element_layout)]

copy_from

copy_from(self: _Self, src: LayoutTensor[get_witness(_Self, nn::mha_sm100::AccumulatorTile, dtype), get_witness(_Self, nn::mha_sm100::AccumulatorTile, vec_output_layout), MutableAnyOrigin, address_space=AddressSpace(5), element_layout=get_witness(_Self, nn::mha_sm100::AccumulatorTile, element_layout)])

copy_to

copy_to(self: _Self, dst: LayoutTensor[get_witness(_Self, nn::mha_sm100::AccumulatorTile, dtype), get_witness(_Self, nn::mha_sm100::AccumulatorTile, vec_output_layout), MutableAnyOrigin, address_space=AddressSpace(5), element_layout=get_witness(_Self, nn::mha_sm100::AccumulatorTile, element_layout)])

Was this page helpful?