Update bench/test to use random number

Author: termoshtt

intel-mkl-sys/Cargo.toml

@@ -8,7 +8,9 @@ edition = "2018"
 path = "../intel-mkl-src"
 
 [dev-dependencies]
-criterion = "*"
+criterion = "0.3"
+rand = "0.7"
+approx = "0.3"
 
 [[bench]]
 name = "cos"

intel-mkl-sys/benches/cos.rs

@@ -2,22 +2,58 @@
 extern crate criterion;
 
 use criterion::Criterion;
+use rand::distributions::{Distribution, Uniform};
 
 fn criterion_benchmark(c: &mut Criterion) {
+    // f32
     for &n in &[100, 1000, 10000] {
-        c.bench_function(&format!("cos{}", n), |b| {
-            let in_ = vec![0.0_f64; n];
-            let mut out = vec![0.0_f64; n];
+        let in_ = {
+            let mut rng = rand::thread_rng();
+            let between = Uniform::from(0.0..2.0 * std::f32::consts::PI);
+            let mut buf = vec![0.0; n];
+            for val in buf.iter_mut() {
+                *val = between.sample(&mut rng);
+            }
+            buf
+        };
+
+        let mut out = vec![0.0_f32; n];
+        c.bench_function(&format!("cos32_n{}", n), |b| {
+            b.iter(|| {
+                for i in 0..n {
+                    out[i] = in_[i].cos();
+                }
+            })
+        });
+        c.bench_function(&format!("vcos32_n{}", n), |b| {
+            b.iter(|| unsafe {
+                intel_mkl_sys::vsCos(n as i32, in_.as_ptr(), out.as_mut_ptr());
+            })
+        });
+    }
+
+    // f64
+    for &n in &[100, 1000, 10000] {
+        let in_ = {
+            let mut rng = rand::thread_rng();
+            let between = Uniform::from(0.0..2.0 * std::f64::consts::PI);
+            let mut buf = vec![0.0; n];
+            for val in buf.iter_mut() {
+                *val = between.sample(&mut rng);
+            }
+            buf
+        };
+
+        let mut out = vec![0.0_f64; n];
+        c.bench_function(&format!("cos64_n{}", n), |b| {
             b.iter(|| {
                 for i in 0..n {
                     out[i] = in_[i].cos();
                 }
             })
         });
 
-        c.bench_function(&format!("vcos{}", n), |b| {
-            let in_ = vec![0.0_f64; n];
-            let mut out = vec![0.0_f64; n];
+        c.bench_function(&format!("vcos64_n{}", n), |b| {
             b.iter(|| unsafe {
                 intel_mkl_sys::vdCos(n as i32, in_.as_ptr(), out.as_mut_ptr());
             })

intel-mkl-sys/src/lib.rs

@@ -25,14 +25,30 @@ include!("mkl.rs");
 #[cfg(test)]
 mod tests {
     use super::*;
+    use approx::ulps_eq;
+    use rand::distributions::{Distribution, Uniform};
     use std::ffi::c_void;
 
+    fn gen_rand_array(n: usize) -> Vec<f64> {
+        let mut rng = rand::thread_rng();
+        let between = Uniform::from(0.0..2.0 * std::f64::consts::PI);
+        let mut buf = vec![0.0; n];
+        for val in buf.iter_mut() {
+            *val = between.sample(&mut rng);
+        }
+        buf
+    }
+
     #[test]
     fn cos() {
-        let a = vec![0.0_f64; 1024];
-        let mut b = vec![0.0_f64; 1024];
+        let n = 1024;
+        let a = gen_rand_array(n);
+        let mut b = vec![0.0_f64; n];
         unsafe {
-            vdCos(1024_i32, a.as_ptr(), b.as_mut_ptr());
+            vdCos(n as i32, a.as_ptr(), b.as_mut_ptr());
+        }
+        for i in 0..n {
+            ulps_eq!(b[i], a[i].cos(), max_ulps = 4, epsilon = std::f64::EPSILON);
         }
     }