Mojo trait
RegTileWriter
Base trait for tile writing mechanisms in matrix multiplication.
This trait defines the interface for writing register tiles to memory (either shared memory or global memory).
Implemented traits
AnyType,
UnknownDestructibility
Aliases
__del__is_trivial
alias __del__is_trivial
A flag (often compiler generated) to indicate whether the implementation of __del__ is trivial.
The implementation of __del__ is considered to be trivial if:
- The struct has a compiler-generated trivial destructor and all its fields
have a trivial
__del__method.
In practice, it means that the __del__ can be considered as no-op.
Required methods
write_tile
write_tile(self: _Self, c_reg_tile: LayoutTensor[_dtype, layout, MutableAnyOrigin, address_space=AddressSpace(5), element_layout=element_layout, layout_int_type=layout_int_type, linear_idx_type=linear_idx_type, masked=masked, alignment=alignment], coords: Tuple[UInt, UInt])
Write a register tile to memory.
Args:
- c_reg_tile (
LayoutTensor): Source register tile containing accumulator values. - coords (
Tuple): Tile coordinates (row, column) in the destination matrix.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!