faer-core 0.17.1

Basic linear algebra routines
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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94

use criterion::{criterion_group, criterion_main, Criterion};
use faer_core::{
    c32, c64, is_vectorizable, mul::matmul_with_conj_gemm_dispatch as matmul_with_conj,
    ComplexField, Conj, Mat, Parallelism,
};
use std::time::Duration;

pub fn matmul<E: ComplexField>(criterion: &mut Criterion) {
    let name = core::any::type_name::<E>();
    for n in [2, 3, 4, 5, 6, 8, 32, 64, 128, 256, 512, 1024] {
        let mut acc = Mat::<E>::zeros(n, n);
        let a = Mat::<E>::zeros(n, n);
        let b = Mat::<E>::zeros(n, n);
        criterion.bench_function(&format!("matmul-{name}-st-{n}"), |bencher| {
            bencher.iter(|| {
                matmul_with_conj(
                    acc.as_mut(),
                    a.as_ref(),
                    Conj::No,
                    b.as_ref(),
                    Conj::No,
                    None,
                    E::faer_one(),
                    Parallelism::None,
                    false,
                );
            });
        });
        criterion.bench_function(&format!("matmul-{name}-mt-{n}"), |bencher| {
            bencher.iter(|| {
                matmul_with_conj(
                    acc.as_mut(),
                    a.as_ref(),
                    Conj::No,
                    b.as_ref(),
                    Conj::No,
                    None,
                    E::faer_one(),
                    Parallelism::Rayon(0),
                    false,
                );
            });
        });

        if is_vectorizable::<E>() {
            criterion.bench_function(&format!("gemm-{name}-mt-{n}"), |bencher| {
                bencher.iter(|| {
                    matmul_with_conj(
                        acc.as_mut(),
                        a.as_ref(),
                        Conj::No,
                        b.as_ref(),
                        Conj::No,
                        None,
                        E::faer_one(),
                        Parallelism::Rayon(0),
                        true,
                    );
                });
            });
            criterion.bench_function(&format!("gemm-{name}-st-{n}"), |bencher| {
                bencher.iter(|| {
                    matmul_with_conj(
                        acc.as_mut(),
                        a.as_ref(),
                        Conj::No,
                        b.as_ref(),
                        Conj::No,
                        None,
                        E::faer_one(),
                        Parallelism::None,
                        true,
                    );
                });
            });
        }
    }
}

criterion_group!(
    name = benches;
    config = Criterion::default()
        .warm_up_time(Duration::from_secs(1))
        .measurement_time(Duration::from_secs(1))
        .sample_size(10);
    targets =
        matmul::<num_complex::Complex64>,
        matmul::<c64>,
        matmul::<f64>,
        matmul::<num_complex::Complex32>,
        matmul::<c32>,
        matmul::<f32>,
);
criterion_main!(benches);