Add bench/svd

Author: termoshtt

ndarray-linalg/Cargo.toml

@@ -58,6 +58,10 @@ harness = false
 name = "eigh"
 harness = false
 
+[[bench]]
+name = "svd"
+harness = false
+
 [[bench]]
 name = "solve"
 harness = false

ndarray-linalg/benches/svd.rs

@@ -0,0 +1,60 @@
+use criterion::*;
+use ndarray::*;
+use ndarray_linalg::*;
+
+fn svd_small(c: &mut Criterion) {
+    let mut group = c.benchmark_group("svd");
+    for &n in &[4, 8, 16, 32, 64, 128] {
+        group.bench_with_input(BenchmarkId::new("C", n), &n, |b, n| {
+            let a: Array2<f64> = random((*n, *n));
+            b.iter(|| {
+                let _ = a.svd(false, false).unwrap();
+            })
+        });
+        group.bench_with_input(BenchmarkId::new("F", n), &n, |b, n| {
+            let a: Array2<f64> = random((*n, *n).f());
+            b.iter(|| {
+                let _ = a.svd(false, false).unwrap();
+            })
+        });
+        group.bench_with_input(BenchmarkId::new("u/C", n), &n, |b, n| {
+            let a: Array2<f64> = random((*n, *n));
+            b.iter(|| {
+                let _ = a.svd(true, false).unwrap();
+            })
+        });
+        group.bench_with_input(BenchmarkId::new("u/F", n), &n, |b, n| {
+            let a: Array2<f64> = random((*n, *n).f());
+            b.iter(|| {
+                let _ = a.svd(true, false).unwrap();
+            })
+        });
+        group.bench_with_input(BenchmarkId::new("vt/C", n), &n, |b, n| {
+            let a: Array2<f64> = random((*n, *n));
+            b.iter(|| {
+                let _ = a.svd(false, true).unwrap();
+            })
+        });
+        group.bench_with_input(BenchmarkId::new("vt/F", n), &n, |b, n| {
+            let a: Array2<f64> = random((*n, *n).f());
+            b.iter(|| {
+                let _ = a.svd(false, true).unwrap();
+            })
+        });
+        group.bench_with_input(BenchmarkId::new("uvt/C", n), &n, |b, n| {
+            let a: Array2<f64> = random((*n, *n));
+            b.iter(|| {
+                let _ = a.svd(false, true).unwrap();
+            })
+        });
+        group.bench_with_input(BenchmarkId::new("uvt/F", n), &n, |b, n| {
+            let a: Array2<f64> = random((*n, *n).f());
+            b.iter(|| {
+                let _ = a.svd(false, true).unwrap();
+            })
+        });
+    }
+}
+
+criterion_group!(svd, svd_small);
+criterion_main!(svd);