For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /max/get-started.md).
Mojo trait
TileLoader
Base trait for tile loading mechanisms in matrix multiplication.
This trait defines the interface for loading tiles from global memory to shared memory, abstracting over different hardware mechanisms.
Implemented traits
AnyType,
Copyable,
ImplicitlyCopyable,
ImplicitlyDeletable,
Movable,
RegisterPassable,
TrivialRegisterPassable
Required methods
__init__
def __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
def __init__(out self: _Self, *, deinit move: _Self)
Create a new instance of the value by moving the value of another.
Args:
- move (
_Self): The value to move.
Returns:
_Self
load_tile
def load_tile(self: _Self, dst: TileTensor[address_space=AddressSpace.SHARED, linear_idx_type=dst.linear_idx_type, element_size=dst.element_size], mem_barrier: UnsafePointer[SharedMemBarrier, address_space=AddressSpace.SHARED], coords: Tuple[Int, Int])
Load a tile from global memory to shared memory.
Args:
- dst (
TileTensor[address_space=AddressSpace.SHARED, linear_idx_type=dst.linear_idx_type, element_size=dst.element_size]): Destination tile in shared memory (must be 128-byte aligned). - mem_barrier (
UnsafePointer[SharedMemBarrier, address_space=AddressSpace.SHARED]): Memory barrier for synchronization. - coords (
Tuple[Int, Int]): Tile coordinates (row, column) in the source matrix.
Provided methods
copy
def 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!