core-allocator 0.2.2

A crate to help set core affinity
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

use crate::CpuSet;
#[allow(dead_code)]
pub fn to_io_error<T>(err: nix::Result<T>) -> std::io::Result<T> {
    match err {
        Ok(x) => Ok(x),
        Err(errno) => Err(errno.into()),
    }
}

/// Returns previous CpuSet
#[cfg(not(target_os = "macos"))]
pub fn bind_to_cpu_set(cpuset: CpuSet) -> std::io::Result<CpuSet> {
    let pid = nix::unistd::gettid();

    // debug!("taskset -pc {} {}", set, pid);
    let previous = nix::sched::sched_getaffinity(pid)?;
    to_io_error(nix::sched::sched_setaffinity(pid, &cpuset))?;
    Ok(previous)
}

#[cfg(target_os = "macos")]
pub fn bind_to_cpu_set(_cpuset: CpuSet) -> std::io::Result<CpuSet> {
    Ok(CpuSet)
}

pub fn to_cpu_set(cores: impl IntoIterator<Item=usize>) -> CpuSet {
    let mut set = CpuSet::new();
    let mut is_set = false;
    for i in cores {
        set.set(i as _).unwrap();
        is_set = true;
    }
    if !is_set {
        for i in 0..CpuSet::count() {
            set.set(i as _).unwrap();
        }
    }
    set
}