diffusionx 0.12.0

A multi-threaded crate for random number generation and stochastic process simulation, with optional GPU acceleration.
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

use criterion::{Criterion, criterion_group, criterion_main};
use diffusionx::simulation::{continuous::Levy, prelude::*};
use std::hint::black_box;

fn criterion_benchmark(_: &mut Criterion) {
    let levy = Levy::new(0.0f32, 1.7f32).unwrap();

    let duration = 100.0f32;
    let time_step = 0.01;
    let particles = 10_000;
    let order = 0.7f32;

    let mut criterion = Criterion::default().sample_size(10);

    criterion.bench_function("levy-frac-raw-moment-cpu-f32", |b| {
        b.iter(|| {
            let _ = levy
                .frac_raw_moment(
                    black_box(duration),
                    black_box(order),
                    black_box(particles),
                    black_box(time_step),
                )
                .unwrap();
        })
    });

    #[cfg(feature = "cuda")]
    criterion.bench_function("levy-frac-raw-moment-cuda-f32", |b| {
        b.iter(|| {
            let _ = levy
                .frac_raw_moment_gpu(
                    black_box(duration),
                    black_box(order),
                    black_box(particles),
                    black_box(time_step),
                )
                .unwrap();
        })
    });
}

criterion_group!(benches, criterion_benchmark);
criterion_main!(benches);