CpuPlan

Enum CpuPlan

pub enum CpuPlan<T: ScalarBase> {
    BatchedGemm {
        m: usize,
        n: usize,
        k: usize,
        _marker: PhantomData<T>,
    },
    Reduce {
        axis: usize,
        op: ReduceOp,
        _marker: PhantomData<T>,
    },
    Trace {
        paired: Vec<(u32, u32)>,
        _marker: PhantomData<T>,
    },
    Permute {
        perm: Vec<usize>,
        _marker: PhantomData<T>,
    },
    AntiTrace {
        paired: Vec<(u32, u32)>,
        _marker: PhantomData<T>,
    },
    AntiDiag {
        paired: Vec<(u32, u32)>,
        _marker: PhantomData<T>,
    },
    ElementwiseUnary {
        op: UnaryOp,
        _marker: PhantomData<T>,
    },
    Contract {
        _marker: PhantomData<T>,
    },
    ElementwiseMul {
        _marker: PhantomData<T>,
    },
}

Expand description

CPU plan — concrete enum, no type erasure.

Created by CpuBackend::plan and consumed by CpuBackend::execute.

Variants§

BatchedGemm

Plan for batched GEMM.

Fields

§m: usize

Number of rows.

§n: usize

Number of columns.

§k: usize

Contraction dimension.

§_marker: PhantomData<T>

Reduce

Plan for reduction.

Fields

§axis: usize

Axis to reduce over.

§op: ReduceOp

Reduction operation.

§_marker: PhantomData<T>

Trace

Plan for trace.

Fields

§paired: Vec<(u32, u32)>

Paired modes.

§_marker: PhantomData<T>

Permute

Plan for permutation.

Fields

§perm: Vec<usize>

Permutation mapping.

§_marker: PhantomData<T>

AntiTrace

Plan for anti-trace (AD backward).

Fields

§paired: Vec<(u32, u32)>

Paired modes.

§_marker: PhantomData<T>

AntiDiag

Plan for anti-diag (AD backward).

Fields

§paired: Vec<(u32, u32)>

Paired modes.

§_marker: PhantomData<T>

ElementwiseUnary

Plan for element-wise unary operation.

Fields

§op: UnaryOp

Unary operation.

§_marker: PhantomData<T>

Contract

Plan for fused contraction (extended op).

Fields

§_marker: PhantomData<T>

ElementwiseMul

Plan for element-wise multiplication (extended op).

Fields

§_marker: PhantomData<T>

Auto Trait Implementations§

impl<T> Freeze for CpuPlan<T>

impl<T> RefUnwindSafe for CpuPlan<T>
where T: RefUnwindSafe,

impl<T> Send for CpuPlan<T>

impl<T> Sync for CpuPlan<T>

impl<T> Unpin for CpuPlan<T>
where T: Unpin,

impl<T> UnwindSafe for CpuPlan<T>
where T: UnwindSafe,

Blanket Implementations§

impl<T> Any for T
where T: 'static + ?Sized,

fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more

impl<T> Borrow<T> for T
where T: ?Sized,

fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more

impl<T> BorrowMut<T> for T
where T: ?Sized,

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

Mutably borrows from an owned value. Read more

impl<T> From<T> for T

fn from(t: T) -> T

Returns the argument unchanged.

impl<T, U> Into<U> for T
where U: From<T>,

fn into(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<U> for T
where U: Into<T>,

type Error = Infallible

The type returned in the event of a conversion error.

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

Performs the conversion.

impl<T, U> TryInto<U> for T
where U: TryFrom<T>,

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.

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

Performs the conversion.