37 lines
1.2 KiB
Rust
37 lines
1.2 KiB
Rust
//! Diamond pattern benchmarks (fan-out then fan-in)
|
|
|
|
use criterion::Criterion;
|
|
use dagx::{task_fn, DagRunner, TaskHandle};
|
|
|
|
|
|
pub fn bench_diamond(c: &mut Criterion) {
|
|
let rt = tokio::runtime::Runtime::new().unwrap();
|
|
|
|
c.bench_function("diamond_pattern_50", |b| {
|
|
b.iter(|| {
|
|
rt.block_on(async {
|
|
let dag = DagRunner::new();
|
|
|
|
for i in 0..50 {
|
|
let source: TaskHandle<_> =
|
|
dag.add_task(task_fn::<(), _, _>(move |_: ()| i)).into();
|
|
|
|
let left = dag
|
|
.add_task(task_fn::<i32, _, _>(|&x: &i32| x * 2))
|
|
.depends_on(source);
|
|
let right = dag
|
|
.add_task(task_fn::<i32, _, _>(|&x: &i32| x * 3))
|
|
.depends_on(source);
|
|
|
|
dag.add_task(task_fn::<(i32, i32), _, _>(|(l, r): (&i32, &i32)| l + r))
|
|
.depends_on((&left, &right));
|
|
}
|
|
|
|
dag.run(|fut| async move { tokio::spawn(fut).await.unwrap() })
|
|
.await
|
|
.unwrap();
|
|
})
|
|
});
|
|
});
|
|
}
|