source: benchmark/readyQ/yield.rs @ 8df19af

Last change on this file since 8df19af was 9cf2b0f, checked in by Thierry Delisle <tdelisle@…>, 2 years ago

Updated yield benchmarks to be more consistents.

  • Property mode set to 100644
File size: 3.2 KB
Line 
1use std::sync::Arc;
2use std::sync::atomic::Ordering;
3use std::time::Instant;
4
5use tokio::runtime::Builder;
6use tokio::sync;
7use tokio::task;
8
9use clap::App;
10use num_format::{Locale, ToFormattedString};
11
12#[path = "../bench.rs"]
13mod bench;
14
15// ==================================================
16struct Yielder {
17        sem: sync::Semaphore,
18}
19
20async fn yield_main(idx: usize, others: Arc<Vec<Arc<Yielder>>>, exp: Arc<bench::BenchData> ) -> u64 {
21        let this = &others[idx];
22        this.sem.acquire().await.forget();
23
24        let mut count:u64 = 0;
25        loop {
26                task::yield_now().await;
27                count += 1;
28
29                if  exp.clock_mode && exp.stop.load(Ordering::Relaxed) { break; }
30                if !exp.clock_mode && count >= exp.stop_count { break; }
31        }
32
33        exp.threads_left.fetch_sub(1, Ordering::SeqCst);
34        count
35}
36
37// ==================================================
38fn main() {
39        let options = App::new("Yield Tokio")
40                .args(&bench::args())
41                .get_matches();
42
43        let nthreads  = options.value_of("nthreads").unwrap().parse::<usize>().unwrap();
44        let nprocs    = options.value_of("nprocs").unwrap().parse::<usize>().unwrap();
45
46        let exp = Arc::new(bench::BenchData::new(options, nthreads, None));
47
48        let s = (1000000 as u64).to_formatted_string(&Locale::en);
49        assert_eq!(&s, "1,000,000");
50
51        let thddata : Arc<Vec<Arc<Yielder>>> = Arc::new(
52                (0..nthreads).map(|_i| {
53                        Arc::new(Yielder{
54                                sem: sync::Semaphore::new(0),
55                        })
56                }).collect()
57        );
58
59        let mut global_counter :u64 = 0;
60        let mut duration : std::time::Duration = std::time::Duration::from_secs(0);
61        let runtime = Builder::new_multi_thread()
62                .worker_threads(nprocs)
63                .enable_all()
64                .build()
65                .unwrap();
66
67        runtime.block_on(async {
68                let threads: Vec<_> = (0..nthreads).map(|i| {
69                        tokio::spawn(yield_main(i, thddata.clone(), exp.clone()))
70                }).collect();
71                println!("Starting");
72
73                let start = Instant::now();
74
75                for i in 0..nthreads {
76                        thddata[i].sem.add_permits(1);
77                }
78
79                duration = exp.wait(&start).await;
80
81                println!("\nDone");
82
83                for i in 0..nthreads {
84                        thddata[i].sem.add_permits(1);
85                }
86
87                for t in threads {
88                        global_counter += t.await.unwrap();
89                }
90        });
91
92        println!("Duration (ms)        : {}", (duration.as_millis()).to_formatted_string(&Locale::en));
93        println!("Number of processors : {}", (nprocs).to_formatted_string(&Locale::en));
94        println!("Number of threads    : {}", (nthreads).to_formatted_string(&Locale::en));
95        println!("Total Operations(ops): {:>15}", (global_counter).to_formatted_string(&Locale::en));
96        println!("Ops per second       : {:>15}", (((global_counter as f64) / duration.as_secs() as f64) as u64).to_formatted_string(&Locale::en));
97        println!("ns per ops           : {:>15}", ((duration.as_nanos() as f64 / global_counter as f64) as u64).to_formatted_string(&Locale::en));
98        println!("Ops per threads      : {:>15}", (global_counter / nthreads as u64).to_formatted_string(&Locale::en));
99        println!("Ops per procs        : {:>15}", (global_counter / nprocs as u64).to_formatted_string(&Locale::en));
100        println!("Ops/sec/procs        : {:>15}", ((((global_counter as f64) / nprocs as f64) / duration.as_secs() as f64) as u64).to_formatted_string(&Locale::en));
101        println!("ns per ops/procs     : {:>15}", ((duration.as_nanos() as f64 / (global_counter as f64 / nprocs as f64)) as u64).to_formatted_string(&Locale::en));
102}
Note: See TracBrowser for help on using the repository browser.