Modules§
- as_
cmma_ layout - cube_
count - from_
cmma_ layout - launch
- Shared utilities for launching kernels.
- stage
- tile
- Tile module.
Structs§
- Formatted
Config Error - Error that arises from invalid configurations
- Global
Partition Size - Number of global matmul blocks computed by a single cube.
- Matmul
Problem Size - Matmul
Problem Size Expand - Partition
Size - Partition
Size Expand - Plane
Flow Counts - Represents how many planes are used for main computation and for loading-only tasks.
- Specialized
Cube Dim - Plane-flow configuration carried by
CubeDimResource::Specialized. Holds the counts for main-flow vs load-only planes and the partition rule used at runtime. - Stage
Size - Stage
Size Expand - Swizzle
Modes - Tile
Size - Tile
Size Expand
Enums§
- Cube
DimResource - Number of compute primitives required by some component, specified as either units, planes, or a specialized plane-flow split.
- Input
Binding - Matmul
Dim - Matrix dimension specifier for matmul operations.
- Matrix
Layout - Layout of a 2D structure such as a tensor, shared memory or slice, used within any matmul kernel level
- Matrix
Layout Expand - Plane
Flow Partition Rule - How planes are partitioned by id between the main flow and load-only roles.
- Stage
Ident
Functions§
- as_
cmma_ layout - Maps the matmul MatrixLayout to cmma’s MatrixLayout, for use in Cmma API.
- from_
cmma_ layout - Maps the cmma’s MatrixLayout to matmul MatrixLayout.
Type Aliases§
- Invalid
Config Error - Error that arises from invalid configurations