pub enum LayoutSku {
Rcr,
Rrr,
}Expand description
Layout SKU. Describes the row/column orientation of A, B, C, and D for matrix-multiply-shaped kernels.
Intentionally NOT #[non_exhaustive] — the GEMM layout space
is essentially closed in practice (row-major / column-major
permutations of A, B, C/D); the two wired variants cover the
dispatch space baracuda-cutlass selects against. New variants
would be a deliberate breaking change with a major-version bump.
Variants§
Rcr
A row-major [M, K], B column-major [K, N], C/D row-major [M, N].
Useful when a row-major weight tensor stored as [N, K] is
reinterpreted as logical column-major B = [K, N] without a
transpose copy.
Rrr
A row-major [M, K], B row-major [K, N], C/D row-major [M, N].
The natural shape for activation-row-major @ weight-row-major matmul (the typical ML graph layout). No transpose pass needed before launch — both operands stored in their native row-major form.