Struct ProfiledScirs2Exec 

pub struct ProfiledScirs2Exec { /* private fields */ }

Profiling-enabled executor wrapper

Implementations

impl ProfiledScirs2Exec

pub fn new() -> Self

Create a new profiled executor

pub fn with_memory_pool() -> Self

Create with memory pooling enabled

pub fn executor(&self) -> &Scirs2Exec

Access the underlying executor

pub fn executor_mut(&mut self) -> &mut Scirs2Exec

Access the underlying executor mutably

Trait Implementations

impl Default for ProfiledScirs2Exec

fn default() -> Self

Returns the “default value” for a type.

impl TlAutodiff for ProfiledScirs2Exec

type Tape = <Scirs2Exec as TlAutodiff>::Tape

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 ProfiledScirs2Exec

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.

impl TlProfiledExecutor for ProfiledScirs2Exec

fn profiler(&self) -> Option<&Profiler>

Get the current profiler

fn profiler_mut(&mut self) -> Option<&mut Profiler>

Get mutable access to profiler

fn enable_profiling(&mut self)

Enable profiling

fn disable_profiling(&mut self)

Disable profiling

fn get_profile_data(&self) -> Option<&ProfileData>

Get profiling data