use std::sync::Arc; use std::sync::atomic::Ordering; use std::time::Instant; use tokio::runtime::Builder; use tokio::sync; use clap::{Arg, App}; use num_format::{Locale, ToFormattedString}; #[path = "../bench.rs"] mod bench; // ================================================== struct Partner { sem: sync::Semaphore, next: usize, } async fn partner_main(idx: usize, others: Arc>>, exp: Arc ) -> u64 { let this = &others[idx]; let mut count:u64 = 0; loop { this.sem.acquire().await.forget(); others[this.next].sem.add_permits(1); count += 1; if exp.clock_mode && exp.stop.load(Ordering::Relaxed) { break; } if !exp.clock_mode && count >= exp.stop_count { break; } } exp.threads_left.fetch_sub(1, Ordering::SeqCst); count } // ================================================== fn main() { let options = App::new("Cycle Tokio") .args(&bench::args()) .arg(Arg::with_name("ringsize") .short("r").long("ringsize") .takes_value(true).default_value("1").help("Number of threads in a cycle")) .get_matches(); let ring_size = options.value_of("ringsize").unwrap().parse::().unwrap(); let nthreads = options.value_of("nthreads").unwrap().parse::().unwrap(); let nprocs = options.value_of("nprocs").unwrap().parse::().unwrap(); let tthreads = nthreads * ring_size; let exp = Arc::new(bench::BenchData::new(options, tthreads, None)); let s = (1000000 as u64).to_formatted_string(&Locale::en); assert_eq!(&s, "1,000,000"); let thddata : Arc>> = Arc::new( (0..tthreads).map(|i| { let pi = (i + nthreads) % tthreads; Arc::new(Partner{ sem: sync::Semaphore::new(0), next: pi, }) }).collect() ); let mut global_counter :u64 = 0; let mut duration : std::time::Duration = std::time::Duration::from_secs(0); let runtime = Builder::new_multi_thread() .worker_threads(nprocs) .enable_all() .build() .unwrap(); runtime.block_on(async { let threads: Vec<_> = (0..tthreads).map(|i| { tokio::spawn(partner_main(i, thddata.clone(), exp.clone())) }).collect(); println!("Starting"); let start = Instant::now(); for i in 0..nthreads { thddata[i].sem.add_permits(1); } duration = exp.wait(&start).await; println!("\nDone"); for i in 0..tthreads { thddata[i].sem.add_permits(1); } for t in threads { global_counter += t.await.unwrap(); } }); println!("Duration (ms) : {}", (duration.as_millis()).to_formatted_string(&Locale::en)); println!("Number of processors : {}", (nprocs).to_formatted_string(&Locale::en)); println!("Number of threads : {}", (tthreads).to_formatted_string(&Locale::en)); println!("Cycle size (# thrds) : {}", (ring_size).to_formatted_string(&Locale::en)); println!("Total Operations(ops): {:>15}", (global_counter).to_formatted_string(&Locale::en)); println!("Ops per second : {:>15}", (((global_counter as f64) / duration.as_secs() as f64) as u64).to_formatted_string(&Locale::en)); println!("ns per ops : {:>15}", ((duration.as_nanos() as f64 / global_counter as f64) as u64).to_formatted_string(&Locale::en)); println!("Ops per threads : {:>15}", (global_counter / tthreads as u64).to_formatted_string(&Locale::en)); println!("Ops per procs : {:>15}", (global_counter / nprocs as u64).to_formatted_string(&Locale::en)); println!("Ops/sec/procs : {:>15}", ((((global_counter as f64) / nprocs as f64) / duration.as_secs() as f64) as u64).to_formatted_string(&Locale::en)); 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)); }