oa::ml::matrix::mamba3_siso_step
Advance one Mamba-3 SISO token and update four explicit recurrent states. Operand layouts are the length-one forms accepted by mamba3_siso. State layouts are SSM [B,H,P,N], cumulative angle [B,H,A], previous key [B,H,N], and previous value [B,H,P]. The states are mutated only after the complete operation has been validated and recorded. Errors Returns an error unless config.sequence_length == 1, every operand and state has the configured same-engine F32 shape, or recording fails.
Function Documentation
pub fn mamba3_siso_step(
c: &Matrix,
b: &Matrix,
x: &Matrix,
z: &Matrix,
adt: &Matrix,
dt: &Matrix,
trap: &Matrix,
angle: &Matrix,
c_bias: &Matrix,
b_bias: &Matrix,
d: &Matrix,
ssm_state: &mut Matrix,
angle_state: &mut Matrix,
k_state: &mut Matrix,
v_state: &mut Matrix,
config: SsmConfig
) -> Result<Matrix>Advance one Mamba-3 SISO token and update four explicit recurrent states. Operand layouts are the length-one forms accepted by mamba3_siso. State layouts are SSM [B,H,P,N], cumulative angle [B,H,A], previous key [B,H,N], and previous value [B,H,P]. The states are mutated only after the complete operation has been validated and recorded. Errors Returns an error unless config.sequence_length == 1, every operand and state has the configured same-engine F32 shape, or recording fails.