Struct ParallelScirs2Exec 

pub struct ParallelScirs2Exec {
    pub config: ParallelConfig,
    pub stats: Option<ParallelStats>,
    /* private fields */
}

Parallel executor using Rayon for multi-threaded execution.

Fields

config: ParallelConfig

Configuration for parallel execution

stats: Option<ParallelStats>

Statistics from last execution

Implementations

impl ParallelScirs2Exec

pub fn new() -> Self

Create a new parallel executor with default configuration.

pub fn with_config(config: ParallelConfig) -> Self

Create a parallel executor with custom configuration.

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

Set the number of threads to use.

pub fn num_threads(&self) -> usize

pub fn set_pooling(&mut self, enable: bool)

Enable or disable memory pooling.

pub fn pool_stats(&self) -> Option<PoolStats>

Get pool statistics if pooling is enabled.

pub fn execution_stats(&self) -> Option<&ParallelStats>

Get statistics from the last execution.

pub fn add_tensor(&mut self, name: impl Into<String>, tensor: Scirs2Tensor)

Add a named tensor to the executor.

pub fn get_tensor(&self, name: &str) -> Option<&Scirs2Tensor>

Get a tensor by name.

Trait Implementations

impl Default for ParallelScirs2Exec

fn default() -> Self

Returns the “default value” for a type.

impl TlAutodiff for ParallelScirs2Exec

Available on non-crate feature parallel only.

type Tape = ForwardTape

fn forward(&mut self, graph: &EinsumGraph) -> Result<Self::Tensor, Self::Error>

Execute forward pass on an EinsumGraph.

fn backward( &mut self, graph: &EinsumGraph, loss_grad: &Self::Tensor, ) -> Result<Self::Tape, Self::Error>

Execute backward pass to compute gradients.

impl TlExecutor for ParallelScirs2Exec

type Tensor = ArrayBase<OwnedRepr<f64>, Dim<IxDynImpl>>

type Error = ExecutorError

fn einsum( &mut self, spec: &str, inputs: &[Self::Tensor], ) -> Result<Self::Tensor, Self::Error>

Execute an einsum operation on input tensors.

fn elem_op( &mut self, op: ElemOp, x: &Self::Tensor, ) -> Result<Self::Tensor, Self::Error>

Apply an element-wise unary operation.

fn elem_op_binary( &mut self, op: ElemOp, x: &Self::Tensor, y: &Self::Tensor, ) -> Result<Self::Tensor, Self::Error>

Apply an element-wise binary operation.

fn reduce( &mut self, op: ReduceOp, x: &Self::Tensor, axes: &[usize], ) -> Result<Self::Tensor, Self::Error>

Reduce a tensor along specified axes.