impl ThreadPoolBuilder

fn build_scoped<W, F, R>(self: Self, wrapper: W, with_pool: F) -> Result<R, ThreadPoolBuildError>
where
    W: Fn(ThreadBuilder) + Sync,
    F: FnOnce(&ThreadPool) -> R

Creates a scoped ThreadPool initialized using this configuration.

This is a convenience function for building a pool using std::thread::scope to spawn threads in a spawn_handler. The threads in this pool will start by calling wrapper, which should do initialization and continue by calling ThreadBuilder::run().

Examples

A scoped pool may be useful in combination with scoped thread-local variables.

# use rayon_core as rayon;

scoped_tls::scoped_thread_local!(static POOL_DATA: Vec<i32>);

fn main() -> Result<(), rayon::ThreadPoolBuildError> {
    let pool_data = vec![1, 2, 3];

    // We haven't assigned any TLS data yet.
    assert!(!POOL_DATA.is_set());

    rayon::ThreadPoolBuilder::new()
        .build_scoped(
            // Borrow `pool_data` in TLS for each thread.
            |thread| POOL_DATA.set(&pool_data, || thread.run()),
            // Do some work that needs the TLS data.
            |pool| pool.install(|| assert!(POOL_DATA.is_set())),
        )?;

    // Once we've returned, `pool_data` is no longer borrowed.
    drop(pool_data);
    Ok(())
}

impl ThreadPoolBuilder

fn new() -> Self

Creates and returns a valid rayon thread pool builder, but does not initialize it.

impl<S> ThreadPoolBuilder<S>

fn spawn_handler<F>(self: Self, spawn: F) -> ThreadPoolBuilder<CustomSpawn<F>>
where
    F: FnMut(ThreadBuilder) -> Result<()>

Sets a custom function for spawning threads.

Note that the threads will not exit until after the pool is dropped. It is up to the caller to wait for thread termination if that is important for any invariants. For instance, threads created in std::thread::scope will be joined before that scope returns, and this will block indefinitely if the pool is leaked. Furthermore, the global thread pool doesn't terminate until the entire process exits!

Examples

A minimal spawn handler just needs to call run() from an independent thread.

# use rayon_core as rayon;
fn main() -> Result<(), rayon::ThreadPoolBuildError> {
    let pool = rayon::ThreadPoolBuilder::new()
        .spawn_handler(|thread| {
            std::thread::spawn(|| thread.run());
            Ok(())
        })
        .build()?;

    pool.install(|| println!("Hello from my custom thread!"));
    Ok(())
}

The default spawn handler sets the name and stack size if given, and propagates any errors from the thread builder.

# use rayon_core as rayon;
fn main() -> Result<(), rayon::ThreadPoolBuildError> {
    let pool = rayon::ThreadPoolBuilder::new()
        .spawn_handler(|thread| {
            let mut b = std::thread::Builder::new();
            if let Some(name) = thread.name() {
                b = b.name(name.to_owned());
            }
            if let Some(stack_size) = thread.stack_size() {
                b = b.stack_size(stack_size);
            }
            b.spawn(|| thread.run())?;
            Ok(())
        })
        .build()?;

    pool.install(|| println!("Hello from my fully custom thread!"));
    Ok(())
}

This can also be used for a pool of scoped threads like crossbeam::scope, or std::thread::scope introduced in Rust 1.63, which is encapsulated in build_scoped.

# use rayon_core as rayon;
fn main() -> Result<(), rayon::ThreadPoolBuildError> {
    std::thread::scope(|scope| {
        let pool = rayon::ThreadPoolBuilder::new()
            .spawn_handler(|thread| {
                let mut builder = std::thread::Builder::new();
                if let Some(name) = thread.name() {
                    builder = builder.name(name.to_string());
                }
                if let Some(size) = thread.stack_size() {
                    builder = builder.stack_size(size);
                }
                builder.spawn_scoped(scope, || {
                    // Add any scoped initialization here, then run!
                    thread.run()
                })?;
                Ok(())
            })
            .build()?;

        pool.install(|| println!("Hello from my custom scoped thread!"));
        Ok(())
    })
}

fn thread_name<F>(self: Self, closure: F) -> Self
where
    F: FnMut(usize) -> String + 'static

Sets a closure which takes a thread index and returns the thread's name.

fn num_threads(self: Self, num_threads: usize) -> Self

Sets the number of threads to be used in the rayon threadpool.

If you specify a non-zero number of threads using this function, then the resulting thread-pools are guaranteed to start at most this number of threads.

If num_threads is 0, or you do not call this function, then the Rayon runtime will select the number of threads automatically. At present, this is based on the RAYON_NUM_THREADS environment variable (if set), or the number of logical CPUs (otherwise). In the future, however, the default behavior may change to dynamically add or remove threads as needed.

Future compatibility warning: Given the default behavior may change in the future, if you wish to rely on a fixed number of threads, you should use this function to specify that number. To reproduce the current default behavior, you may wish to use std::thread::available_parallelism to query the number of CPUs dynamically.

Old environment variable: RAYON_NUM_THREADS is a one-to-one replacement of the now deprecated RAYON_RS_NUM_CPUS environment variable. If both variables are specified, RAYON_NUM_THREADS will be preferred.

fn use_current_thread(self: Self) -> Self

Use the current thread as one of the threads in the pool.

The current thread is guaranteed to be at index 0, and since the thread is not managed by rayon, the spawn and exit handlers do not run for that thread.

Note that the current thread won't run the main work-stealing loop, so jobs spawned into the thread-pool will generally not be picked up automatically by this thread unless you yield to rayon in some way, like via [yield_now()], [yield_local()], or [scope()].

Local thread-pools

Using this in a local thread-pool means the registry will be leaked. In future versions there might be a way of cleaning up the current-thread state.

fn panic_handler<H>(self: Self, panic_handler: H) -> Self
where
    H: Fn(Box<dyn Any + Send>) + Send + Sync + 'static

Normally, whenever Rayon catches a panic, it tries to propagate it to someplace sensible, to try and reflect the semantics of sequential execution. But in some cases, particularly with the spawn() APIs, there is no obvious place where we should propagate the panic to. In that case, this panic handler is invoked.

If no panic handler is set, the default is to abort the process, under the principle that panics should not go unobserved.

If the panic handler itself panics, this will abort the process. To prevent this, wrap the body of your panic handler in a call to std::panic::catch_unwind().

fn stack_size(self: Self, stack_size: usize) -> Self

Sets the stack size of the worker threads

fn breadth_first(self: Self) -> Self

(DEPRECATED) Suggest to worker threads that they execute spawned jobs in a "breadth-first" fashion.

Typically, when a worker thread is idle or blocked, it will attempt to execute the job from the top of its local deque of work (i.e., the job most recently spawned). If this flag is set to true, however, workers will prefer to execute in a breadth-first fashion -- that is, they will search for jobs at the bottom of their local deque. (At present, workers always steal from the bottom of other workers' deques, regardless of the setting of this flag.)

If you think of the tasks as a tree, where a parent task spawns its children in the tree, then this flag loosely corresponds to doing a breadth-first traversal of the tree, whereas the default would be to do a depth-first traversal.

Note that this is an "execution hint". Rayon's task execution is highly dynamic and the precise order in which independent tasks are executed is not intended to be guaranteed.

This breadth_first() method is now deprecated per RFC #1, and in the future its effect may be removed. Consider using scope_fifo() for a similar effect.

fn start_handler<H>(self: Self, start_handler: H) -> Self
where
    H: Fn(usize) + Send + Sync + 'static

Sets a callback to be invoked on thread start.

The closure is passed the index of the thread on which it is invoked. Note that this same closure may be invoked multiple times in parallel. If this closure panics, the panic will be passed to the panic handler. If that handler returns, then startup will continue normally.

fn exit_handler<H>(self: Self, exit_handler: H) -> Self
where
    H: Fn(usize) + Send + Sync + 'static

Sets a callback to be invoked on thread exit.

The closure is passed the index of the thread on which it is invoked. Note that this same closure may be invoked multiple times in parallel. If this closure panics, the panic will be passed to the panic handler. If that handler returns, then the thread will exit normally.

impl<S> ThreadPoolBuilder<S>

fn build(self: Self) -> Result<ThreadPool, ThreadPoolBuildError>

Creates a new ThreadPool initialized using this configuration.

fn build_global(self: Self) -> Result<(), ThreadPoolBuildError>

Initializes the global thread pool. This initialization is optional. If you do not call this function, the thread pool will be automatically initialized with the default configuration. Calling build_global is not recommended, except in two scenarios:

• You wish to change the default configuration.
• You are running a benchmark, in which case initializing may yield slightly more consistent results, since the worker threads will already be ready to go even in the first iteration. But this cost is minimal.

Initialization of the global thread pool happens exactly once. Once started, the configuration cannot be changed. Therefore, if you call build_global a second time, it will return an error. An Ok result indicates that this is the first initialization of the thread pool.

impl Default for ThreadPoolBuilder

fn default() -> Self

impl<S = DefaultSpawn> RefUnwindSafe for ThreadPoolBuilder<S>

impl<S = DefaultSpawn> Send for ThreadPoolBuilder<S>

impl<S = DefaultSpawn> Sync for ThreadPoolBuilder<S>

impl<S = DefaultSpawn> UnwindSafe for ThreadPoolBuilder<S>

impl<S> Debug for ThreadPoolBuilder<S>

fn fmt(self: &Self, f: &mut Formatter<'_>) -> Result

impl<S> Freeze for ThreadPoolBuilder<S>

impl<S> Unpin for ThreadPoolBuilder<S>

impl<S> UnsafeUnpin for ThreadPoolBuilder<S>

impl<T> Any for ThreadPoolBuilder<S>

fn type_id(self: &Self) -> TypeId

impl<T> Borrow for ThreadPoolBuilder<S>

fn borrow(self: &Self) -> &T

impl<T> BorrowMut for ThreadPoolBuilder<S>

fn borrow_mut(self: &mut Self) -> &mut T

impl<T> From for ThreadPoolBuilder<S>

fn from(t: T) -> T

Returns the argument unchanged.

impl<T> Pointable for ThreadPoolBuilder<S>

unsafe fn init(init: <T as Pointable>::Init) -> usize

unsafe fn deref<'a>(ptr: usize) -> &'a T

unsafe fn deref_mut<'a>(ptr: usize) -> &'a mut T

unsafe fn drop(ptr: usize)

impl<T, U> Into for ThreadPoolBuilder<S>

fn into(self: Self) -> U

Calls U::from(self).

That is, this conversion is whatever the implementation of [From]<T> for U chooses to do.

impl<T, U> TryFrom for ThreadPoolBuilder<S>

fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>

impl<T, U> TryInto for ThreadPoolBuilder<S>

fn try_into(self: Self) -> Result<U, <U as TryFrom<T>>::Error>