eig, eigvals, eigvalsh

Author: termoshtt

ndarray-linalg/Cargo.toml

@@ -54,6 +54,10 @@ approx = { version = "0.3.2", features = ["num-complex"] }
 name = "truncated_eig"
 harness = false
 
+[[bench]]
+name = "eig"
+harness = false
+
 [[bench]]
 name = "eigh"
 harness = false

ndarray-linalg/benches/eig.rs

@@ -0,0 +1,36 @@
+use criterion::*;
+use ndarray::*;
+use ndarray_linalg::*;
+
+fn eig_small(c: &mut Criterion) {
+    let mut group = c.benchmark_group("eig");
+    for &n in &[4, 8, 16, 32, 64, 128] {
+        group.bench_with_input(BenchmarkId::new("vecs/C", n), &n, |b, n| {
+            let a: Array2<f64> = random((*n, *n));
+            b.iter(|| {
+                let (_e, _vecs) = a.eig().unwrap();
+            })
+        });
+        group.bench_with_input(BenchmarkId::new("vecs/F", n), &n, |b, n| {
+            let a: Array2<f64> = random((*n, *n).f());
+            b.iter(|| {
+                let (_e, _vecs) = a.eig().unwrap();
+            })
+        });
+        group.bench_with_input(BenchmarkId::new("vals/C", n), &n, |b, n| {
+            let a: Array2<f64> = random((*n, *n));
+            b.iter(|| {
+                let _result = a.eigvals().unwrap();
+            })
+        });
+        group.bench_with_input(BenchmarkId::new("vals/F", n), &n, |b, n| {
+            let a: Array2<f64> = random((*n, *n).f());
+            b.iter(|| {
+                let _result = a.eigvals().unwrap();
+            })
+        });
+    }
+}
+
+criterion_group!(eig, eig_small);
+criterion_main!(eig);

ndarray-linalg/benches/eigh.rs

@@ -5,16 +5,28 @@ use ndarray_linalg::*;
 fn eigh_small(c: &mut Criterion) {
     let mut group = c.benchmark_group("eigh");
     for &n in &[4, 8, 16, 32, 64, 128] {
-        group.bench_with_input(BenchmarkId::new("C", n), &n, |b, n| {
+        group.bench_with_input(BenchmarkId::new("vals/C", n), &n, |b, n| {
             let a: Array2<f64> = random((*n, *n));
             b.iter(|| {
-                let (_e, _vecs) = a.eigh(UPLO::Upper).unwrap();
+                let _result = a.eigvalsh(UPLO::Upper).unwrap();
             })
         });
-        group.bench_with_input(BenchmarkId::new("F", n), &n, |b, n| {
+        group.bench_with_input(BenchmarkId::new("vals/F", n), &n, |b, n| {
             let a: Array2<f64> = random((*n, *n).f());
             b.iter(|| {
-                let (_e, _vecs) = a.eigh(UPLO::Upper).unwrap();
+                let _result = a.eigvalsh(UPLO::Upper).unwrap();
+            })
+        });
+        group.bench_with_input(BenchmarkId::new("vecs/C", n), &n, |b, n| {
+            let a: Array2<f64> = random((*n, *n));
+            b.iter(|| {
+                let _result = a.eigvalsh(UPLO::Upper).unwrap();
+            })
+        });
+        group.bench_with_input(BenchmarkId::new("vecs/F", n), &n, |b, n| {
+            let a: Array2<f64> = random((*n, *n).f());
+            b.iter(|| {
+                let _result = a.eigvalsh(UPLO::Upper).unwrap();
             })
         });
     }