Struct Mamba1CacheConfig 

pub struct Mamba1CacheConfig {
    pub batch: usize,
    pub state_rank: usize,
    pub conv_kernel: usize,
    pub d_inner: usize,
}

Configuration / factory for a single Mamba1Cache.

Fields

batch: usize

Batch size.

state_rank: usize

State rank — the latent dimension of the SSM hidden state. Corresponds to state_rank in Mamba1Config.

conv_kernel: usize

Causal convolution window length. Corresponds to conv_kernel in Mamba1Config.

d_inner: usize

Inner (expanded) channel width d_inner.

Implementations

impl Mamba1CacheConfig

pub fn new(batch: usize, d_inner: usize) -> Self

Create a new instance of the config.

Arguments

Required Arguments

batch

Batch size.

d_inner

Inner (expanded) channel width d_inner.

Default Arguments

state_rank

State rank — the latent dimension of the SSM hidden state. Corresponds to state_rank in Mamba1Config.

• Defaults to 16

conv_kernel

Causal convolution window length. Corresponds to conv_kernel in Mamba1Config.

• Defaults to 4

impl Mamba1CacheConfig

pub fn with_state_rank(self, state_rank: usize) -> Self

Sets the value for the field state_rank.

State rank — the latent dimension of the SSM hidden state. Corresponds to state_rank in Mamba1Config.

• Defaults to 16

pub fn with_conv_kernel(self, conv_kernel: usize) -> Self

Sets the value for the field conv_kernel.

Causal convolution window length. Corresponds to conv_kernel in Mamba1Config.

• Defaults to 4

impl Mamba1CacheConfig

pub fn new_from_block_config(batch: usize, block_config: Mamba1Config) -> Self

Derive cache shapes from a Mamba-1 block configuration plus a batch size.

pub fn init(&self, device: &Device) -> Mamba1Cache

Allocate zero-initialised cache tensors on device.

Zero initialisation is correct because the convolution window represents “no previous tokens” (identity padding) and the SSM state represents the standard zero initial condition h₀ = 0.

Trait Implementations

impl Clone for Mamba1CacheConfig

fn clone(&self) -> Self

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Config for Mamba1CacheConfig

fn load_binary(data: &[u8]) -> Result<Self, ConfigError>

Loads the configuration from a binary buffer.

impl Debug for Mamba1CacheConfig

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for Mamba1CacheConfig

fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>

where D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Display for Mamba1CacheConfig

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

Formats the value using the given formatter.

impl Serialize for Mamba1CacheConfig

fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>

where S: Serializer,

Serialize this value into the given Serde serializer.