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 core::ops::Range;

use num_traits::Zero;

use crate::{MatMut, MatRef};

// write colmajor "registers" back to c
pub(crate) fn registers_to_c<T>(
    registers: &[T],
    c: &mut MatMut<T>,
    c_rows: Range<usize>,
    c_cols: Range<usize>,
) where
    T: Copy,
{
    crate::packing::block::ColMajor(registers).copy_to(c, c_rows, c_cols);
}

// read submatrix from c to colmajor "registers"
pub(crate) fn registers_from_c<T>(
    registers: &mut [T],
    c: MatRef<T>,
    c_rows: Range<usize>,
    c_cols: Range<usize>,
) where
    T: Copy + Zero,
{
    crate::packing::block::ColMajor(registers).init_from(c, c_rows, c_cols);
}