Structs§
- ColMajor
Tiling Order - Tiles laid out in column-major order.
- ColMajor
Tiling Order Expand - Contiguous
Tiling Layout - Each tile is stored contiguously in shared memory. Global memory loads may require remapping to match this layout.
- Full
Reader Family - Full reader family for any precision
- Full
Stage ToTile Reader - Reads any tile from the stage memory
- Full
Stage ToTile Reader Expand - NoEvent
- Use when there is no event listening to do
- NoEvent
Expand - NumStages
- Number of stages in one shared memory, i.e. buffers for double buffering
- Ordered
Tiling Order - Tiles are laid out in column-major order across a fixed number of rows, with all tiles from those rows placed contiguously side by side.
- Ordered
Tiling Order Expand - Partial
Reader Family - Partial reader family for any precision
- Partial
Stage ToTile Reader - Reads tile from the stage memory for the specified stage ident only
- Partial
Stage ToTile Reader Expand - Plane
Matmul Family - Plane Matmul family for any precision
- RowMajor
Tiling Order - Tiles laid out in row-major order.
- RowMajor
Tiling Order Expand - Stage
Memory - Wrapper over the shared memory used for staging, abstracting its layout
- Strided
Tiling Layout - Tiles follow a strided layout that often mirrors global memory layout. Not all tiles are contiguous in shared memory, but mapping is more direct.
- Unit
Matmul Family - Unit Matmul family for any precision
Enums§
- Partition
Buffering - Stage
Event - Events that occur during the process of loading tiles to registers and executing inner Tile Matmuls
- Tiling
Order Enum - Enum for the available traits
Traits§
- Reader
Family - Reader family for any precision
- Stage
Config - Configuration for the Stage matmul (SMM) level
- Stage
Event Listener - Function that is called at each StageEvent
- Stage
Matmul - Provides matrix multiplication operations at the stage level.
- Stage
Matmul Family - A family of StageMatmul implementations that operate with any precision.
- Stage
ToTile Reader - Read the tile at (row, col) from stage memory
- Tiling
Layout - Describes how tiles are arranged in shared memory.
- Tiling
Order - Determines the order in which tiles are stored in shared memory, if TilingLayout is contiguous