janis/executor
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

cpu pinning

Browse source
This commit is contained in:
Janis 2025-01-31 00:23:56 +01:00
parent fd0cd86a8d
commit 25fa558a17
1 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
src/lib.rs
View file

@ -509,6 +509,8 @@ impl ThreadPool {
return self.pool_state.num_threads.load(Ordering::Acquire); return self.pool_state.num_threads.load(Ordering::Acquire);
} }
let cpus = core_affinity::get_core_ids().unwrap();
let _guard = self.pool_state.lock.lock(); let _guard = self.pool_state.lock.lock();
let current_size = self.pool_state.num_threads.load(Ordering::Acquire); let current_size = self.pool_state.num_threads.load(Ordering::Acquire);
@ -528,7 +530,9 @@ impl ThreadPool {
let new_threads = &self.threads[current_size..new_size]; let new_threads = &self.threads[current_size..new_size];
for (i, _) in new_threads.iter().enumerate() { for (i, _) in new_threads.iter().enumerate() {
let core = cpus[i];
std::thread::spawn(move || { std::thread::spawn(move || {
core_affinity::set_for_current(core);
WorkerThread::worker_loop(&self, current_size + i); WorkerThread::worker_loop(&self, current_size + i);
}); });
} }
@ -1317,7 +1321,7 @@ mod tests {
let now = std::time::Instant::now(); let now = std::time::Instant::now();
run_in_scope(pool, |s| { run_in_scope(pool, |s| {
for &p in core::iter::repeat_n(PRIMES, 0x1000).flatten() { for &p in core::iter::repeat_n(PRIMES, 0x200).flatten() {
s.spawn(move |_| { s.spawn(move |_| {
// std::thread::sleep(Duration::from_micros(p as u64)); // std::thread::sleep(Duration::from_micros(p as u64));
// spin for // spin for