Mojo trait
SMemTileWriter
Base trait for tile writing mechanisms in matrix multiplication.
This trait defines the interface for writing tiles from shared memory to global memory, abstracting over different hardware mechanisms.
Implemented traits
AnyType,
Copyable,
ImplicitlyCopyable,
ImplicitlyDestructible,
Movable,
RegisterPassable,
TrivialRegisterPassable
Required methods
__init__
__init__(out self: _Self, *, copy: _Self)
Create a new instance of the value by copying an existing one.
Args:
- copy (
_Self): The value to copy.
Returns:
_Self
__init__(out self: _Self, *, deinit take: _Self)
Create a new instance of the value by moving the value of another.
Args:
- take (
_Self): The value to move.
Returns:
_Self
write_tile
write_tile(self: _Self, src: LayoutTensor[_Self._dtype, src.layout, MutAnyOrigin, address_space=AddressSpace.SHARED, element_layout=src.element_layout, layout_int_type=src.layout_int_type, linear_idx_type=src.linear_idx_type, masked=src.masked, alignment=128], coords: Tuple[Int, Int])
Write a tile from shared memory to global memory.
Args:
- src (
LayoutTensor): Source tile in shared memory (must be 128-byte aligned). - coords (
Tuple): Tile coordinates (row, column) in the destination matrix.
Provided methods
copy
copy(self: _Self) -> _Self
Explicitly construct a copy of self, a convenience method for Self(copy=self) when the type is inconvenient to write out.
Returns:
_Self: A copy of this value.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!