Expand description
Components for matrix multiplication
Modules§
Structs§
- Complete
Stage Tiling - Aggregation of [StageTiling]s for all components.
- Formatted
Config Error - Lhs
- Matmul
Problem - Description of a matmul problem to solve, regardless of actual data
- Matmul
Selection - Matmul
Size - Matmul
Size Expand - Out
- Quantized
- ReplaceES
- Rhs
- Tiling
Dimensions - Dimensions for stage.
Enums§
- Ident
- Identifier for all three tensors in a matmul
- Input
Ident - Identifier for the two input tensors in a matmul.
- Matrix
Layout - Layout of a 2D structure such as a tensor, shared memory or slice, used within any matmul kernel level
- Matrix
Layout Expand
Traits§
- Matmul
Config - A config for a matmul
- Matmul
Config Factory - Provides configuration for a matmul kernel at any level
- Matmul
Launch - Provides launch entry point to solve a matmul
- Matmul
Precision - Matrix multiplication precisions.
- Matmul
Spec - Matrix multiplication spec definiting each element types used in the computation as well as how the arguments are passed to the kernel.
- Tensor
Ident
Functions§
- as_
cmma_ layout - Maps the matmul MatrixLayout to cmma’s MatrixLayout, for use in Cmma API.
Type Aliases§
- Args
- EA
- EI
- EO
- ES
- Input
Arg - Input argument
- Input
Runtime Arg - Input runtime argument
- Invalid
Config Error - Output
Arg - Output argument
- Output
Runtime Arg - Output runtime argument