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

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

fn criterion_benchmark(c: &mut Criterion) {
    let ou = OU::default();

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

    c.bench_function("ou-msd-cpu-f32", |b| {
        b.iter(|| {
            let _ = ou
                .msd(
                    black_box(duration),
                    black_box(particles),
                    black_box(time_step),
                )
                .unwrap();
        })
    });

    #[cfg(feature = "cuda")]
    c.bench_function("ou-msd-cuda-f32", |b| {
        b.iter(|| {
            let _ = ou
                .msd_gpu(
                    black_box(duration),
                    black_box(particles),
                    black_box(time_step),
                )
                .unwrap();
        })
    });
}

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