Struct Checkpoint 

pub struct Checkpoint<R: Recomputable> { /* private fields */ }

A single checkpointed computation segment.

Stores the segment implementation and its saved input so that the output can be recomputed at any time. The output itself is not stored; calling recompute always re-runs the forward pass.

Implementations

impl<R: Recomputable> Checkpoint<R>

pub fn new(recomputable: R, input: R::Input) -> Self

Create a new checkpoint, saving input for later recomputation.

The recomputable segment is stored alongside the input so that recompute can call recomputable.forward(&saved_input).

pub fn recompute(&self) -> R::Output

Recompute and return the output from the saved input.

This is the key operation: instead of loading a cached output tensor, we replay the forward pass from the checkpointed input.

pub fn memory_bytes(&self) -> usize

Bytes consumed by the saved checkpoint (input only, not the output).