corealloc 0.6.0

Helper CLI utility decide on core allocation on the current system
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58

use clap::{crate_version, value_t, App, Arg};

mod topology;
use topology::*;

fn main() {
    let _r = env_logger::try_init();
    let args = std::env::args();
    let matches = App::new("Core allocation")
        .version(crate_version!())
        .author("Gerd Zellweger <mail@gerdzellweger.com>")
        .about("Figures out which cores IDs to pin my application on.")
        .arg(
            Arg::with_name("cores")
                .short("c")
                .long("cores")
                .help("How many cores do you want?")
                .required(true)
                .takes_value(true),
        )
        .arg(
            Arg::with_name("tm")
                .long("thread-mapping")
                .short("t")
                .takes_value(true)
                .possible_values(&["sequential", "interleave", "roundrobin"])
                .help("Do you want to interleave over sockets or allocate sequential"),
        )
        .arg(
            Arg::with_name("noht")
                .long("no-ht")
                .short("n")
                .takes_value(false)
                .help("Avoid using hyper-threads"),
        )
        .get_matches_from(args);

    let cores = value_t!(matches, "cores", usize).unwrap_or_else(|e| e.exit());
    let no_ht = matches.is_present("noht");
    let thread_mapping = match matches.value_of("tm").unwrap_or("interleave") {
        "interleave" => ThreadMapping::Interleave,
        "sequential" => ThreadMapping::Sequential,
        "roundrobin" => ThreadMapping::RoundRobin,
        _ => unreachable!(),
    };
    let mt = MachineTopology::new();

    let cpus = mt.allocate(thread_mapping, cores, !no_ht);
    if cores > cpus.len() {
        eprintln!("Don't overprovision cores");
        std::process::exit(1);
    }

    for cpu in cpus {
        print!("{} ", cpu.cpu);
    }
    println!("");
}