Skip to main content

Mojo trait

TileWriter

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, 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, src: LayoutTensor[_Self._dtype, layout, MutableAnyOrigin, address_space=AddressSpace(3), alignment=128], coords: Tuple[UInt, UInt])

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.

Was this page helpful?