Enable grouping for benchmarking

Author: termoshtt

benches/eigh.rs

@@ -1,37 +1,25 @@
-#[macro_use]
-extern crate criterion;
-
-use criterion::Criterion;
+use criterion::*;
 use ndarray::*;
 use ndarray_linalg::*;
 
-macro_rules! impl_eigh {
-    ($n:expr) => {
-        paste::item! {
-            fn [<eigh $n>](c: &mut Criterion) {
-                c.bench_function(&format!("eigh{}", $n), |b| {
-                    let a: Array2<f64> = random(($n, $n));
-                    b.iter(|| {
-                        let (_e, _vecs) = a.eigh(UPLO::Upper).unwrap();
-                    })
-                });
-                c.bench_function(&format!("eigh{}_t", $n), |b| {
-                    let a: Array2<f64> = random(($n, $n).f());
-                    b.iter(|| {
-                        let (_e, _vecs) = a.eigh(UPLO::Upper).unwrap();
-                    })
-                });
-            }
-        }
-    };
-}
+fn eigh_small(c: &mut Criterion) {
+    let mut group = c.benchmark_group("eigh");
 
-impl_eigh!(4);
-impl_eigh!(8);
-impl_eigh!(16);
-impl_eigh!(32);
-impl_eigh!(64);
-impl_eigh!(128);
+    for &n in &[4, 8, 16, 32, 64, 128] {
+        group.bench_function(&format!("eigh{}", n), |b| {
+            let a: Array2<f64> = random((n, n));
+            b.iter(|| {
+                let (_e, _vecs) = a.eigh(UPLO::Upper).unwrap();
+            })
+        });
+        group.bench_function(&format!("eigh{}_t", n), |b| {
+            let a: Array2<f64> = random((n, n).f());
+            b.iter(|| {
+                let (_e, _vecs) = a.eigh(UPLO::Upper).unwrap();
+            })
+        });
+    }
+}
 
-criterion_group!(eigh, eigh4, eigh8, eigh16, eigh32, eigh64, eigh128);
+criterion_group!(eigh, eigh_small);
 criterion_main!(eigh);