Enum Mamba3Caches 

pub enum Mamba3Caches {
    DoubleSsd(Mamba3DoubleSsdCaches),
    SingleSsd(Mamba3SingleSsdCaches),
}

A pathway-tagged bundle of per-layer caches, so a single dispatch entry can accept / return either cache family.

The caches selection infers whether Double-SSD or Single-SSD is used.
If none is specified, this defaults to Self::SingleSsd.

See also crate::mamba3::ssd_path::Mamba3SsdPath.

Variants

DoubleSsd(Mamba3DoubleSsdCaches)

Caches for the double-ssd pathway.

SingleSsd(Mamba3SingleSsdCaches)

Caches for the single-ssd pathway.

Implementations

impl Mamba3Caches

pub fn double_ssd(self) -> Option<Mamba3DoubleSsdCaches>

Unwrap to the double-SSD caches, or None if this is the single-SSD variant.

pub fn single_ssd(self) -> Option<Mamba3SingleSsdCaches>

Unwrap to the single-SSD caches, or None if this is the double-SSD variant.

pub fn caches_len(&self) -> usize

Number of per-layer caches (independent of pathway).

pub fn from_vec(vec: Vec<Mamba3Cache>) -> Self

Collect per-layer caches into a pathway-tagged bundle. The pathway is inferred from the first element (an empty vec implies single-SSD).

pub fn into_options(self) -> Vec<Option<Mamba3Cache>>

Wrap each per-layer cache in Some so the loop can take it without cloning (Burn tensors are reference-counted).

pub fn from_options(options: Vec<Option<Mamba3Cache>>) -> Self

Inverse of Self::into_options: unwrap each slot and re-bundle.

Trait Implementations

impl CacheStack for Mamba3Caches

type Cache = Mamba3Cache

The per-layer cache element.

fn slot_count(&self) -> usize

Number of per-(virtual-)layer slots.

fn into_slots(self) -> Vec<Option<Mamba3Cache>>

Move each slot into an Option so the loop can take without cloning.

fn from_slots(slots: Vec<Option<Mamba3Cache>>) -> Self

Inverse of Self::into_slots.

fn cache_to_inner(c: Mamba3Cache) -> Mamba3Cache

Move one cache slot to the inner (non-autodiff) backend.

fn cache_from_inner(c: Mamba3Cache) -> Mamba3Cache

Lift one cache slot back from the inner backend, as a fresh graph root. The inverse of Self::cache_to_inner; see its notes.

impl Clone for Mamba3Caches

fn clone(&self) -> Mamba3Caches

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for Mamba3Caches

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

Formats the value using the given formatter.

impl From<Mamba3DoubleSsdCaches> for Mamba3Caches

fn from(caches: Mamba3DoubleSsdCaches) -> Self

Converts to this type from the input type.

impl From<Mamba3SingleSsdCaches> for Mamba3Caches

fn from(caches: Mamba3SingleSsdCaches) -> Self

Converts to this type from the input type.