Struct caffe2_threadpool::ThreadPool

#[repr(align(64))]
pub struct ThreadPool { /* private fields */ }

| A threadpool with the given number of | threads. | | ———– | @note | | the kCacheLineSize alignment is present | only for cache performance, and is not | strictly enforced (for example, when | the object is created on the heap). Thus, | in order to avoid misaligned intrinsics, | no SSE instructions shall be involved | in the ThreadPool implementation. | ––––– | @note | | alignas is disabled because some compilers | do not deal with TORCH_API and alignas | annotations at the same time. |

Implementations

impl ThreadPool

pub fn get_min_work_size(&self) -> usize

pub fn default_thread_pool(&mut self) -> Box<ThreadPool>

pub fn new(num_threads: i32) -> Self

pub fn get_num_threads(&self) -> i32

| Returns the number of threads currently | in use |

pub fn set_num_threads(&mut self, num_threads: usize)

| Sets the number of threads | | # of threads should not be bigger than the | number of big cores

pub fn set_min_work_size(&mut self, size: usize)

| Sets the minimum work size (range) for | which to invoke the threadpool; work sizes | smaller than this will just be run on the | main (calling) thread

pub fn run(f: fn(a: i32, b: usize), range: usize)

pub fn with_pool(f: fn(x: *mut WorkersPool))

| Run an arbitrary function in a thread-safe | manner accessing the Workers Pool |

Trait Implementations

impl Default for ThreadPool

fn default() -> Self

Returns the “default value” for a type.

impl Send for ThreadPool

impl Sync for ThreadPool