microgemm 0.3.1

General matrix multiplication with custom configuration in Rust. Supports no_std and no_alloc environments.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29

use microgemm::{kernels::GenericKernel4x4, Kernel as _, MatMut, MatRef, PackSizes};

const M: usize = 15;
const K: usize = 16;
const N: usize = 22;

const KERNEL: GenericKernel4x4<f32> = GenericKernel4x4::<f32>::new();

const PACK_SIZES: PackSizes = PackSizes {
    mc: 2 * GenericKernel4x4::<f32>::MR,
    kc: 16,
    nc: 3 * GenericKernel4x4::<f32>::NR,
};

fn main() {
    let mut packing_buf = [0.0; PACK_SIZES.buf_len()];
    let (alpha, beta) = (2.0, -3.0);

    let a = [3.0; M * K];
    let b = [4.0; K * N];
    let mut c = [5.0; M * N];

    let a = MatRef::row_major(M, K, &a);
    let b = MatRef::row_major(K, N, &b);
    let mut c = MatMut::row_major(M, N, &mut c);

    KERNEL.gemm(alpha, a, b, beta, &mut c, PACK_SIZES, &mut packing_buf);
    println!("{:?}", c.as_slice());
}