Struct JitRecurrent 

pub struct JitRecurrent<J: RecurrentJit> {
    pub jit: J,
    pub last_timing: StepTiming,
    /* private fields */
}

Wraps a recurrent JIT and its host-side LSTM state. Owns the head read buffer so callers can borrow the non-state output between steps.

Fields

jit: J

last_timing: StepTiming

Implementations

impl<J: RecurrentJit> JitRecurrent<J>

pub fn new(jit: J, state: LstmState, head_len: usize) -> Result<Self>

Construct from a prepared JIT, the initial state, and the declared head length in f32 elements. The JIT’s output buffer is read once and its size is checked against head_len + |h| + |c| to catch build-closure layout drift at construction time instead of letting silently mis-split output corrupt downstream values.

pub fn state(&self) -> &LstmState

pub fn state_mut(&mut self) -> &mut LstmState

pub fn head_len(&self) -> usize

pub fn step<F>(&mut self, pack_inputs: F) -> Result<&[f32]>

where F: FnOnce(&mut J) -> Result<()>,

One recurrent step. pack_inputs writes per-step non-state inputs (audio chunk, token id, encoder frame, …) into the JIT. The wrapper then packs state, executes, splits the output into [head | h_tail | c_tail], and updates host state. Returns the head slice borrowed from the wrapper-owned buffer.

pub fn reset(&mut self)

Zero the host-side state. The next step writes the fresh state into the JIT.