For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /get-started.md).

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, Deinitable, ImplicitlyCopyable, Movable, RegisterPassable, TrivialRegisterPassable

Required methods

write_tile

def write_tile(self, src: TileTensor[_Self._dtype, Engine=src.Engine, address_space=AddressSpace.SHARED, linear_idx_type=src.linear_idx_type], coords: Tuple[Int, Int])

Write a tile from shared memory to global memory.

Args:

• ​src (TileTensor[_Self._dtype, Engine=src.Engine, address_space=AddressSpace.SHARED, linear_idx_type=src.linear_idx_type]): Source tile in shared memory (must be 128-byte aligned).
• ​coords (Tuple[Int, Int]): Tile coordinates (row, column) in the destination matrix.