Enum CpuPlan

Source

pub enum CpuPlan<T: Scalar> {
    BatchedGemm {
        batch_dims: Vec<usize>,
        m: usize,
        n: usize,
        k: usize,
        _marker: PhantomData<T>,
    },
    ReduceAdd {
        reduced_axes: Vec<usize>,
        _marker: PhantomData<T>,
    },
    Trace {
        free_axes: Vec<usize>,
        components: Vec<Vec<usize>>,
        comp_dims: Vec<usize>,
        _marker: PhantomData<T>,
    },
    AntiTrace {
        paired_axes: Vec<(usize, usize)>,
        free_axes: Vec<usize>,
        components: Vec<Vec<usize>>,
        comp_dims: Vec<usize>,
        _marker: PhantomData<T>,
    },
    AntiDiag {
        paired_axes: Vec<(usize, usize)>,
        free_axes: Vec<usize>,
        components: Vec<Vec<usize>>,
        comp_dims: Vec<usize>,
        generative_comps: Vec<usize>,
        _marker: PhantomData<T>,
    },
    Contract {
        modes_a: Vec<u32>,
        modes_b: Vec<u32>,
        modes_c: Vec<u32>,
        gemm_spec: Option<ContractGemmSpec>,
        _marker: PhantomData<T>,
    },
    ElementwiseBinary {
        op: SemiringBinaryOp,
        _marker: PhantomData<T>,
    },
    MakeContiguous {
        _marker: PhantomData<T>,
    },
}

Expand description

CPU plan — concrete enum, no type erasure.

Created by the family planners on crate::CpuBackend and consumed by the semiring family executors.

Variants§

§

BatchedGemm

Plan for batched GEMM.

Fields

§batch_dims: Vec<usize>

Batch dimension sizes.

§m: usize

Number of rows.

§n: usize

Number of columns.

§k: usize

Contraction dimension.

§_marker: PhantomData<T>

§

ReduceAdd

Plan for semiring-add reduction.

Fields

§reduced_axes: Vec<usize>

Axes to reduce over (positions in input tensor).

§_marker: PhantomData<T>

§

Trace

Plan for trace.

Fields

§free_axes: Vec<usize>

Output axis positions mapping.

§components: Vec<Vec<usize>>

Connected components of paired axes (union-find groups).

§comp_dims: Vec<usize>

Dimension of each component (all axes in a component share the same dim).

§_marker: PhantomData<T>

§

AntiTrace

Plan for anti-trace (AD backward).

Fields

§paired_axes: Vec<(usize, usize)>

Paired axis positions in output tensor.

§free_axes: Vec<usize>

Input axis positions mapping.

§components: Vec<Vec<usize>>

Connected components of paired axes (union-find groups).

§comp_dims: Vec<usize>

Dimension of each component.

§_marker: PhantomData<T>

§

AntiDiag

Plan for anti-diag (AD backward).

Fields

§paired_axes: Vec<(usize, usize)>

Paired axis positions in output tensor.

§free_axes: Vec<usize>

Input axis positions mapping.

§components: Vec<Vec<usize>>

Connected components of paired axes (union-find groups).

§comp_dims: Vec<usize>

Dimension of each component.

§generative_comps: Vec<usize>

Indices of generative components (no overlap with free axes).

§_marker: PhantomData<T>

§

Contract

Plan for fused contraction.

Fields

§modes_a: Vec<u32>

Mode labels for input A.

§modes_b: Vec<u32>

Mode labels for input B.

§modes_c: Vec<u32>

Mode labels for output C.

§gemm_spec: Option<ContractGemmSpec>

Cached GEMM mode analysis (None if not a valid GEMM pattern).

§_marker: PhantomData<T>

§

ElementwiseBinary

Plan for optional semiring binary fast paths.

Fields

§op: SemiringBinaryOp

The semiring binary operation to apply.

§_marker: PhantomData<T>

§

MakeContiguous

Plan for making a tensor contiguous.

Fields

§_marker: PhantomData<T>

Trait Implementations§

Source §

impl<T: Clone + Scalar> Clone for CpuPlan<T>

Source §

fn clone(&self) -> CpuPlan<T>

Returns a duplicate of the value. Read more

1.0.0 · Source§

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

Performs copy-assignment from source. Read more

Source §

impl<T: Debug + Scalar> Debug for CpuPlan<T>

Source §

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

Formats the value using the given formatter. Read more

Auto Trait Implementations§

§

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

Blanket Implementations§

Source §

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

Source §

fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more

Source §

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

Source §

fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more

Source §

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

Source §

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

Mutably borrows from an owned value. Read more

§

impl<T> ByRef<T> for T

§

fn by_ref(&self) -> &T

Source §

impl<T> CloneToUninit for T
where T: Clone,

Source §

unsafe fn clone_to_uninit(&self, dest: *mut u8)

🔬This is a nightly-only experimental API. (clone_to_uninit)

Performs copy-assignment from self to dest. Read more

§

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

§

fn rand<T>(&self, rng: &mut (impl Rng + ?Sized)) -> T
where Self: Distribution<T>,

Source §

impl<T> From<T> for T

Source §

fn from(t: T) -> T

Returns the argument unchanged.

Source §

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

Source §

fn into(self) -> U

Calls U::from(self).

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

Source §

impl<T> IntoEither for T

Source §

fn into_either(self, into_left: bool) -> Either<Self, Self>

Converts self into a Left variant of Either<Self, Self> if into_left is true. Converts self into a Right variant of Either<Self, Self> otherwise. Read more

Source §

fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
where F: FnOnce(&Self) -> bool,

Converts self into a Left variant of Either<Self, Self> if into_left(&self) returns true. Converts self into a Right variant of Either<Self, Self> otherwise. Read more

§