Struct CpuContext 

pub struct CpuContext { /* private fields */ }

Reusable CPU execution context backed by an owned rayon thread pool.

Examples

use tenferro_tensor::cpu::CpuContext;

let ctx = CpuContext::with_threads(1);
let seen = ctx.install(|| rayon::current_num_threads());
assert_eq!(seen, 1);

Implementations

impl CpuContext

pub fn from_env() -> Self

Create a CPU context from RAYON_NUM_THREADS, or fall back to the process-visible CPU count.

Examples

use tenferro_tensor::cpu::CpuContext;

let ctx = CpuContext::from_env();
let _ = ctx.num_threads();

pub fn try_from_env() -> Result<Self>

Try to create a CPU context from RAYON_NUM_THREADS.

Examples

use tenferro_tensor::cpu::CpuContext;

let ctx = CpuContext::try_from_env().unwrap();
let _ = ctx.num_threads();

pub fn with_threads(num_threads: usize) -> Self

Create a CPU context with a fixed rayon thread-pool size.

Examples

use tenferro_tensor::cpu::CpuContext;

let ctx = CpuContext::with_threads(2);
assert_eq!(ctx.num_threads(), 2);

pub fn num_threads(&self) -> usize

Return the number of threads in this context’s owned rayon pool.

Examples

use tenferro_tensor::cpu::CpuContext;

let ctx = CpuContext::with_threads(2);
assert_eq!(ctx.num_threads(), 2);

pub fn install<R>(&self, op: impl FnOnce() -> R + Send) -> R

where R: Send,

Run a closure inside this context’s owned rayon thread pool.

Examples

use tenferro_tensor::cpu::CpuContext;

let ctx = CpuContext::with_threads(1);
let value = ctx.install(|| 1 + 1);
assert_eq!(value, 2);

Trait Implementations

impl Clone for CpuContext

fn clone(&self) -> CpuContext

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.