Skip to main content

gemm_par_ndarray

oxiblas_ndarray::parallel

Function gemm_par_ndarray 

Source 
pub fn gemm_par_ndarray<T>(
    alpha: T,
    a: &Array2<T>,
    b: &Array2<T>,
    beta: T,
    c: &mut Array2<T>,
)
where
    T: Zeroable + Clone + Field + GemmKernel,
Expand description

Parallel general matrix-matrix multiplication: C = alpha * A * B + beta * C

Uses Rayon to parallelize the GEMM computation across available threads.

§Arguments

  • alpha - Scalar multiplier for A * B
  • a - Left matrix (m x k)
  • b - Right matrix (k x n)
  • beta - Scalar multiplier for C
  • c - Output matrix (m x n), modified in place

§Panics

Panics if matrix dimensions are incompatible.