Skip to main content

Module parallel

Module parallel 

Source
Expand description

Parallel BLAS operations on ndarray types.

This module provides parallelized BLAS operations using Rayon, gated behind the parallel feature flag.

§Example

use ndarray::array;
use oxiblas_ndarray::parallel::matmul_par;

// A (2x3) * B (3x2) = C (2x2)
let a = array![[1.0f64, 2.0, 3.0], [4.0, 5.0, 6.0]];
let b = array![[7.0f64, 8.0], [9.0, 10.0], [11.0, 12.0]];

let c = matmul_par(&a, &b);
assert_eq!(c, array![[58.0, 64.0], [139.0, 154.0]]);

Functions§

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