Skip to content

Commit 4eeae2c

Browse files
committed
Add bench of LU
1 parent c2095c9 commit 4eeae2c

File tree

3 files changed

+30
-2
lines changed

3 files changed

+30
-2
lines changed

ndarray-linalg/Cargo.toml

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -58,5 +58,9 @@ harness = false
5858
name = "eigh"
5959
harness = false
6060

61+
[[bench]]
62+
name = "solve"
63+
harness = false
64+
6165
[package.metadata.docs.rs]
6266
rustdoc-args = ["--html-in-header", "katex-header.html"]

ndarray-linalg/benches/eigh.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,13 +5,13 @@ use ndarray_linalg::*;
55
fn eigh_small(c: &mut Criterion) {
66
let mut group = c.benchmark_group("eigh");
77
for &n in &[4, 8, 16, 32, 64, 128] {
8-
group.bench_with_input(BenchmarkId::new("eigh", n), &n, |b, n| {
8+
group.bench_with_input(BenchmarkId::new("C", n), &n, |b, n| {
99
let a: Array2<f64> = random((*n, *n));
1010
b.iter(|| {
1111
let (_e, _vecs) = a.eigh(UPLO::Upper).unwrap();
1212
})
1313
});
14-
group.bench_with_input(BenchmarkId::new("eigh_t", n), &n, |b, n| {
14+
group.bench_with_input(BenchmarkId::new("F", n), &n, |b, n| {
1515
let a: Array2<f64> = random((*n, *n).f());
1616
b.iter(|| {
1717
let (_e, _vecs) = a.eigh(UPLO::Upper).unwrap();

ndarray-linalg/benches/solve.rs

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,24 @@
1+
use criterion::*;
2+
use ndarray::*;
3+
use ndarray_linalg::*;
4+
5+
fn factorize_small(c: &mut Criterion) {
6+
let mut group = c.benchmark_group("factorize");
7+
for &n in &[4, 8, 16, 32, 64, 128] {
8+
group.bench_with_input(BenchmarkId::new("C", n), &n, |b, n| {
9+
let a: Array2<f64> = random((*n, *n));
10+
b.iter(|| {
11+
let _lu = a.factorize().unwrap();
12+
})
13+
});
14+
group.bench_with_input(BenchmarkId::new("F", n), &n, |b, n| {
15+
let a: Array2<f64> = random((*n, *n).f());
16+
b.iter(|| {
17+
let _lu = a.factorize().unwrap();
18+
})
19+
});
20+
}
21+
}
22+
23+
criterion_group!(factorize, factorize_small);
24+
criterion_main!(factorize);

0 commit comments

Comments
 (0)