microgemm 0.2.0-alpha

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
30
31
32
33
34
35
36
37

use crate::Kernel;
use num_traits::{One, Zero};

#[derive(Debug, Clone)]
pub struct PackSizes {
    pub mc: usize,
    pub kc: usize,
    pub nc: usize,
}

impl PackSizes {
    pub const fn buf_len(&self) -> usize {
        self.mc * self.kc + self.kc * self.nc
    }
    pub(crate) fn check<T, K>(&self, _: &K)
    where
        T: One + Zero + Copy,
        K: Kernel<Scalar = T> + ?Sized,
    {
        let mr = K::MR;
        let nr = K::NR;
        assert!(mr <= self.mc);
        assert!(nr <= self.nc);
        assert_eq!(self.mc % mr, 0);
        assert_eq!(self.nc % nr, 0);
    }
    pub(crate) fn split_buf<'buf, T>(&self, buf: &'buf mut [T]) -> (&'buf mut [T], &'buf mut [T]) {
        let (apack, bpack) = buf.split_at_mut(self.mc * self.kc);
        (apack, bpack)
    }
}

impl AsRef<PackSizes> for PackSizes {
    fn as_ref(&self) -> &PackSizes {
        self
    }
}