Struct GemmOp 

pub struct GemmOp {
    pub m: usize,
    pub n: usize,
    pub k: usize,
}

C[m, n] = A[m, k] · B[n, k]^T (row-major, B already transposed to head-major). Matches the Backend::gemm signature used by Linear.

Fields

m: usize

n: usize

k: usize

Implementations

impl GemmOp

pub fn expected_output_len(&self) -> Result<usize, String>

Validate buffer sizes and the signed indices used by Metal/cuBLAS. These are arithmetic limits, not a guarantee of available device memory.

Examples found in repository

examples/backend_numerics/config.rs (line 92)

    fn expected_output_len(&self) -> Result<usize, String> {
        match *self {
            Self::MetalContext {
                tokens,
                intermediate,
                k,
            } => MetalContextOp {
                tokens,
                intermediate,
                k,
            }
            .expected_output_len(),
            Self::RmsNorm { tokens, dim, eps } => {
                RmsNormOp { tokens, dim, eps }.expected_output_len()
            }
            Self::Gemm { m, n, k } => GemmOp { m, n, k }.expected_output_len(),
            Self::SiluMul {
                tokens,
                intermediate,
            } => SiluMulOp {
                tokens,
                intermediate,
            }
            .expected_output_len(),
        }
    }

Trait Implementations

impl OpUnderTest for GemmOp

fn name(&self) -> &str

Display name (used in test failure messages).

fn run_cpu(&self, seed: u64) -> Output

Run on CPU (reference). Always available.