[e5aba4a] | 1 | use std::sync::Arc; |
---|
| 2 | use std::sync::atomic::Ordering; |
---|
| 3 | use std::time::Instant; |
---|
| 4 | |
---|
| 5 | use tokio::runtime::Builder; |
---|
| 6 | use tokio::sync; |
---|
| 7 | use tokio::task; |
---|
| 8 | |
---|
| 9 | use clap::App; |
---|
| 10 | use num_format::{Locale, ToFormattedString}; |
---|
| 11 | |
---|
| 12 | #[path = "../bench.rs"] |
---|
| 13 | mod bench; |
---|
| 14 | |
---|
| 15 | // ================================================== |
---|
| 16 | struct Yielder { |
---|
| 17 | sem: sync::Semaphore, |
---|
| 18 | } |
---|
| 19 | |
---|
| 20 | async 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 | // ================================================== |
---|
| 38 | fn main() { |
---|
[d0b9247] | 39 | let options = App::new("Yield Tokio") |
---|
[e5aba4a] | 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 | |
---|
[821c534] | 46 | let exp = Arc::new(bench::BenchData::new(options, nthreads, None)); |
---|
[e5aba4a] | 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( |
---|
[821c534] | 52 | (0..nthreads).map(|_i| { |
---|
[e5aba4a] | 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 | |
---|
[9cf2b0f] | 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)); |
---|
[e5aba4a] | 102 | } |
---|