Adding a benchmark for the new message passing code
This commit is contained in:
parent
dc3862bf58
commit
40559ea839
76
src/test/bench/msgsend-ring-new.rs
Normal file
76
src/test/bench/msgsend-ring-new.rs
Normal file
@ -0,0 +1,76 @@
|
||||
// This test creates a bunch of tasks that simultaneously send to each
|
||||
// other in a ring. The messages should all be basically
|
||||
// independent. It's designed to hammer the global kernel lock, so
|
||||
// that things will look really good once we get that lock out of the
|
||||
// message path.
|
||||
|
||||
import newcomm::*;
|
||||
import future::future;
|
||||
|
||||
use std;
|
||||
import std::time;
|
||||
|
||||
fn thread_ring(i: uint,
|
||||
count: uint,
|
||||
num_chan: chan<uint>,
|
||||
num_port: port<uint>) {
|
||||
// Send/Receive lots of messages.
|
||||
for uint::range(0u, count) {|j|
|
||||
num_chan.send(i * j);
|
||||
num_port.recv();
|
||||
};
|
||||
}
|
||||
|
||||
fn main(args: [str]) {
|
||||
let args = if os::getenv("RUST_BENCH").is_some() {
|
||||
["", "100", "10000"]
|
||||
} else if args.len() <= 1u {
|
||||
["", "100", "1000"]
|
||||
} else {
|
||||
args
|
||||
};
|
||||
|
||||
let num_tasks = option::get(uint::from_str(args[1]));
|
||||
let msg_per_task = option::get(uint::from_str(args[2]));
|
||||
|
||||
let num_port = port();
|
||||
let mut num_chan = chan(num_port);
|
||||
|
||||
let start = time::precise_time_s();
|
||||
|
||||
// create the ring
|
||||
let mut futures = [];
|
||||
|
||||
for uint::range(1u, num_tasks) {|i|
|
||||
let get_chan = port();
|
||||
let get_chan_chan = chan(get_chan);
|
||||
{
|
||||
let num_chan = num_chan.clone();
|
||||
futures += [future::spawn {|move num_chan, move get_chan_chan|
|
||||
let p = port();
|
||||
get_chan_chan.send(chan(p));
|
||||
thread_ring(i, msg_per_task, num_chan, p)
|
||||
}];
|
||||
}
|
||||
|
||||
num_chan = get_chan.recv();
|
||||
};
|
||||
|
||||
// do our iteration
|
||||
thread_ring(0u, msg_per_task, num_chan, num_port);
|
||||
|
||||
// synchronize
|
||||
for futures.each {|f| f.get() };
|
||||
|
||||
let stop = time::precise_time_s();
|
||||
|
||||
// all done, report stats.
|
||||
let num_msgs = num_tasks * msg_per_task;
|
||||
let elapsed = (stop - start);
|
||||
let rate = (num_msgs as float) / elapsed;
|
||||
|
||||
io::println(#fmt("Sent %? messages in %? seconds",
|
||||
num_msgs, elapsed));
|
||||
io::println(#fmt(" %? messages / second", rate));
|
||||
io::println(#fmt(" %? μs / message", 1000000. / rate));
|
||||
}
|
Loading…
Reference in New Issue
Block a user