Enum StatePairing 

pub enum StatePairing {
    Real,
    ComplexInterleaved {
        num_pairs: usize,
    },
    ComplexHalfHalf {
        num_pairs: usize,
    },
    QuaternionBlocks {
        num_blocks: usize,
    },
}

How the state_rank axis of a (realified) SSM state groups into complex / quaternionic coordinates — the realification layout the Mamba-3 rotation applies to B/C (and hence to the state). Consumed by StateMoments::pr_complex.

Construct it with Mamba3::state_pairing() (the single source of truth, mirroring exactly what rotate_bc_forward applies) — never re-derive the layout by hand.

Variants

Real

No rotation pairing: every coordinate is real (rope_fraction = 0); StateMoments::pr_complex then equals StateMoments::pr.

ComplexInterleaved

Complex pairs, interleaved (NeoX-style; Mamba-3 SISO): coordinate 2a is the real and 2a + 1 the imaginary part of complex coordinate a, for a < num_pairs; the tail [2·num_pairs, state_rank) stays real.

Fields

num_pairs: usize

Number of rotated complex pairs (rope_dim / 2).

ComplexHalfHalf

Complex pairs, half-and-half (GPT-J-style; Mamba-3 MIMO): coordinate a is the real and state_rank/2 + a the imaginary part, for a < num_pairs; the remainder of both halves stays real.

Fields

num_pairs: usize

Number of rotated complex pairs (rope_dim / 2).

QuaternionBlocks

Quaternion blocks (Mamba-3 Quaternion4D): coordinates [4j, 4j + 4) form quaternion coordinate j (components ordered w, x, y, z), for j < num_blocks; the tail stays real.

Fields

num_blocks: usize

Number of rotated quaternion blocks (rope_width / 4).

Trait Implementations

impl Clone for StatePairing

fn clone(&self) -> StatePairing

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Copy for StatePairing

impl Debug for StatePairing

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

Formats the value using the given formatter.

impl Eq for StatePairing

impl PartialEq for StatePairing

fn eq(&self, other: &StatePairing) -> bool

Equality operator ==.

fn ne(&self, other: &Rhs) -> bool

Inequality operator !=.

impl StructuralPartialEq for StatePairing