Enum TropicalPlan 

pub enum TropicalPlan<T: Scalar> {
    BatchedGemm {
        batch_dims: Vec<usize>,
        m: usize,
        n: usize,
        k: usize,
        _marker: PhantomData<T>,
    },
    Reduce {
        reduced_axes: Vec<usize>,
        _marker: PhantomData<T>,
    },
    Trace {
        paired_axes: Vec<(usize, usize)>,
        free_axes: Vec<usize>,
        _marker: PhantomData<T>,
    },
    AntiTrace {
        paired_axes: Vec<(usize, usize)>,
        free_axes: Vec<usize>,
        _marker: PhantomData<T>,
    },
    AntiDiag {
        paired_axes: Vec<(usize, usize)>,
        free_axes: Vec<usize>,
        _marker: PhantomData<T>,
    },
    MakeContiguous {
        _marker: PhantomData<T>,
    },
}

Execution plan for tropical primitive operations on CPU.

Analogous to CpuPlan but for tropical algebras. The plan captures pre-computed kernel selection information.

Examples

use tenferro_device::LogicalMemorySpace;
use tenferro_prims::{CpuBackend, CpuContext, SemiringCoreDescriptor, TensorSemiringCore};
use tenferro_tensor::{MemoryOrder, Tensor};
use tenferro_ext_tropical::{MaxPlus, MaxPlusAlgebra, TropicalPlan};

let mut ctx = CpuContext::new(1);
let col = MemoryOrder::ColumnMajor;
let mem = LogicalMemorySpace::MainMemory;
let a = Tensor::<MaxPlus<f64>>::zeros(&[3, 4], mem, col);
let mut c = Tensor::<MaxPlus<f64>>::zeros(&[3], mem, col);
let desc = SemiringCoreDescriptor::ReduceAdd {
    modes_a: vec![0, 1],
    modes_c: vec![0],
};
let plan =
    <CpuBackend as TensorSemiringCore<MaxPlusAlgebra<f64>>>::plan(
        &mut ctx,
        &desc,
        &[&[3, 4], &[3]],
    )
        .unwrap();
<CpuBackend as TensorSemiringCore<MaxPlusAlgebra<f64>>>::execute(
    &mut ctx,
    &plan,
    MaxPlus::one(),
    &[&a],
    MaxPlus::zero(),
    &mut c,
)
.unwrap();

Variants

BatchedGemm

Plan for batched GEMM under tropical algebra.

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>

Reduce

Plan for reduction under tropical algebra.

Fields

reduced_axes: Vec<usize>

Axes to reduce over (positions in input).

_marker: PhantomData<T>

Trace

Plan for trace under tropical algebra.

Fields

paired_axes: Vec<(usize, usize)>

Paired axis positions in input.

free_axes: Vec<usize>

Free axis positions in input (corresponding to output modes).

_marker: PhantomData<T>

AntiTrace

Plan for anti-trace (AD backward).

Fields

paired_axes: Vec<(usize, usize)>

Paired axis positions in output.

free_axes: Vec<usize>

Free axis positions in output (corresponding to input modes).

_marker: PhantomData<T>

AntiDiag

Plan for anti-diag (AD backward).

Fields

paired_axes: Vec<(usize, usize)>

Paired axis positions in output.

free_axes: Vec<usize>

Free axis positions in output (corresponding to input modes).

_marker: PhantomData<T>

MakeContiguous

Plan for making a tensor contiguous.

Fields

_marker: PhantomData<T>

Trait Implementations

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

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

Formats the value using the given formatter.