ndarray/ndarray-rand/benches/bench.rs at 0214fc5ee672eba6fdb2271c283383b8878aefa8 · rust-ndarray/ndarray · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
#![feature(test)]

extern crate test;

use ndarray::Array;
use ndarray_rand::RandomExt;
use rand_distr::Normal;
use rand_distr::Uniform;

use test::Bencher;

#[bench]
fn uniform_f32(b: &mut Bencher)
{
    let m = 100;
    b.iter(|| Array::random((m, m), Uniform::new(-1f32, 1.).unwrap()));
}

#[bench]
#[cfg(feature = "half")]
fn norm_f16(b: &mut Bencher)
{
    let m = 100;
    b.iter(|| Array::random((m, m), Normal::new(half::f16::ZERO, half::f16::ONE).unwrap()));
}

#[bench]
#[cfg(feature = "half")]
fn norm_bf16(b: &mut Bencher)
{
    let m = 100;
    b.iter(|| Array::random((m, m), Normal::new(half::bf16::ZERO, half::bf16::ONE).unwrap()));
}

#[bench]
fn norm_f32(b: &mut Bencher)
{
    let m = 100;
    b.iter(|| Array::random((m, m), Normal::new(0f32, 1.).unwrap()));
}

#[bench]
fn norm_f64(b: &mut Bencher)
{
    let m = 100;
    b.iter(|| Array::random((m, m), Normal::new(0f64, 1.).unwrap()));
}