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
38
39
40
41
42
43
44
45
46
47
48

#[rustfmt::skip]
pub trait MatrixMultiply where Self: Sized {
    fn gemm(m: usize, k: usize, n: usize,
        a: &[Self], rsa: usize, csa: usize,
        b: &[Self], rsb: usize, csb: usize,
        c: &mut [Self], rsc: usize, csc: usize);
}

#[allow(unused)]
mod implements {
    use super::MatrixMultiply;

    #[rustfmt::skip]
    impl MatrixMultiply for f32 {
        #[inline]
        fn gemm(m: usize, k: usize, n: usize,
            a: &[Self], rsa: usize, csa: usize,
            b: &[Self], rsb: usize, csb: usize,
            c: &mut [Self], rsc: usize, csc: usize) {

                #[cfg(not(feature = "matrixmultiply"))]
                unimplemented!("Activate the matrixmultiply feature");
            
                #[cfg(feature = "matrixmultiply")]
                unsafe {
                    matrixmultiply::sgemm(m, k, n, 1., a.as_ptr(), rsa as isize, csa as isize, b.as_ptr(), rsb as isize, csb as isize, 1., c.as_mut_ptr(), rsc as isize, csc as isize);
                }
                
        }
    }

    #[rustfmt::skip]
    impl MatrixMultiply for f64 {
        #[inline]
        fn gemm(m: usize, k: usize, n: usize,
            a: &[Self], rsa: usize, csa: usize,
            b: &[Self], rsb: usize, csb: usize,
            c: &mut [Self], rsc: usize, csc: usize) 
        {
            #[cfg(not(feature = "matrixmultiply"))]
            unimplemented!("Activate the matrixmultiply feature");
            #[cfg(feature = "matrixmultiply")]
            unsafe {
                matrixmultiply::dgemm(m, k, n, 1., a.as_ptr(), rsa as isize, csa as isize, b.as_ptr(), rsb as isize, csb as isize, 1., c.as_mut_ptr(), rsc as isize, csc as isize);
            }           
        }
    }
}