pub enum MatmulAvailabilityError {
CubeCountTooBig(CubeCount),
CubeDimTooBig(CubeDim),
PlaneDimUnsupported {
plane_dim: u32,
},
TypesUnavailable {
lhs: StorageType,
rhs: StorageType,
output: StorageType,
},
CmmaInstructionUnavailable {
lhs: StorageType,
rhs: StorageType,
output: StorageType,
size: Option<TileSize>,
},
TileSizeNotFound,
LayoutUnsupported {
lhs: MatrixLayout,
rhs: MatrixLayout,
},
BarrierUnavailable,
TmaUnavailable,
DynamicLineSizeUnavailable,
PlaneOpsUnavailable,
}Expand description
A specific feature required for matmul is not available in the current runtime or hardware.
Variants§
CubeCountTooBig(CubeCount)
The requested cube count exceeds what the runtime or hardware supports.
CubeDimTooBig(CubeDim)
The requested cube dimensions are too large for the current runtime or hardware.
PlaneDimUnsupported
The requested plane dimension is not supported.
The required data types for input or output are not supported.
The required CMMA instruction is not supported for the given element types and tile size.
TileSizeNotFound
Impossible to find a supported tile size for the problem.
LayoutUnsupported
The layout of the matmul is unsupported
Barrier synchronization is not available in the runtime.
TMA (Tensor Memory Access) is not available in the runtime.
Dynamic selection of line size is unsupported in the current runtime.
Plane operations like plane_sum are unavailable