Module stage

Source

Structs§

ColMajorTilingOrder
Tiles laid out in column-major order.
ColMajorTilingOrderExpand
ContiguousTilingLayout
Each tile is stored contiguously in shared memory. Global memory loads may require remapping to match this layout.
FullReaderFamily
Full reader family for any precision
FullStageToTileReader
Reads any tile from the stage memory
FullStageToTileReaderExpand
NoEvent
Use when there is no event listening to do
NoEventExpand
NumStages
Number of stages in one shared memory, i.e. buffers for double buffering
OrderedTilingOrder
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.
OrderedTilingOrderExpand
PartialReaderFamily
Partial reader family for any precision
PartialStageToTileReader
Reads tile from the stage memory for the specified stage ident only
PartialStageToTileReaderExpand
PlaneMatmulFamily
Plane Matmul family for any precision
RowMajorTilingOrder
Tiles laid out in row-major order.
RowMajorTilingOrderExpand
StageMemory
Wrapper over the shared memory used for staging, abstracting its layout
StridedTilingLayout
Tiles follow a strided layout that often mirrors global memory layout. Not all tiles are contiguous in shared memory, but mapping is more direct.
UnitMatmulFamily
Unit Matmul family for any precision

Enums§

PartitionBuffering
StageEvent
Events that occur during the process of loading tiles to registers and executing inner Tile Matmuls
TilingOrderEnum
Enum for the available traits

Traits§

ReaderFamily
Reader family for any precision
StageConfig
Configuration for the Stage matmul (SMM) level
StageEventListener
Function that is called at each StageEvent
StageMatmul
Provides matrix multiplication operations at the stage level.
StageMatmulFamily
A family of StageMatmul implementations that operate with any precision.
StageToTileReader
Read the tile at (row, col) from stage memory
TilingLayout
Describes how tiles are arranged in shared memory.
TilingOrder
Determines the order in which tiles are stored in shared memory, if TilingLayout is contiguous