nordic-dev.net/rust
nordic-dev.net/rust
2
0
Fork 0
mirror of https://github.com/rust-lang/rust.git synced 2024-11-25 16:24:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

Added an environment variable to control how many threads to use.

Browse Source
This commit is contained in:
Eric Holk 2011-06-20 18:01:38 -07:00 committed by Graydon Hoare
parent 4bc773465f
commit cb00befff0
1 changed files with 18 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
src/rt/rust.cpp
View File

@ -71,6 +71,19 @@ command_line_args : public dom_owned<command_line_args>
}
};
int get_num_threads()
{
char *env = getenv("RUST_THREADS");
if(env) {
int num = atoi(env);
if(num > 0)
return num;
}
// TODO: in this case, determine the number of CPUs present on the
// machine.
return 1;
}
/**
* Main entry point into the Rust runtime. Here we create a Rust service,
* initialize the kernel, create the root domain and run it.
@ -95,7 +108,11 @@ rust_start(uintptr_t main_fn, int argc, char **argv, void* crate_map) {
dom->root_task->start(main_fn, (uintptr_t)args->args);
int ret = dom->start_main_loops(8);
int num_threads = get_num_threads();
DLOG(dom, dom, "Using %d worker threads.", num_threads);
int ret = dom->start_main_loops(num_threads);
delete args;
kernel->destroy_domain(dom);
kernel->join_all_domains();