Struct TensorCheckpointer 

pub struct TensorCheckpointer { /* private fields */ }

Manages a bounded, rolling window of tensor computation checkpoints with rollback support.

Implementations

impl TensorCheckpointer

pub fn new(config: CheckpointConfig) -> Self

Create a new checkpointer with the given configuration.

pub fn create_checkpoint( &mut self, label: &str, tensor_data: Vec<Vec<f64>>, metadata: HashMap<String, String>, ) -> u64

Create a checkpoint with the given label, tensor data, and metadata.

If the number of stored checkpoints exceeds max_checkpoints, the oldest checkpoint is evicted (FIFO).

Returns the unique id of the newly created checkpoint.

pub fn rollback(&mut self, checkpoint_id: u64) -> Result<Checkpoint, String>

Roll back to the checkpoint with the given checkpoint_id.

Returns the checkpoint data and removes all checkpoints that were created after the target checkpoint. The target checkpoint itself is also consumed (removed) so callers can restore from the returned data.

Increments rollbacks_performed.

pub fn latest_checkpoint(&self) -> Option<&Checkpoint>

Return a reference to the most recent checkpoint, if any.

pub fn get_checkpoint(&self, id: u64) -> Option<&Checkpoint>

Look up a checkpoint by its id.

pub fn checkpoint_count(&self) -> usize

Number of checkpoints currently stored.

pub fn should_auto_checkpoint(&self) -> bool

Returns true if the number of ticks since the last checkpoint is greater than or equal to auto_checkpoint_interval.

pub fn tick(&mut self)

Advance the internal tick clock by one.

pub fn list_checkpoints(&self) -> Vec<(u64, String, u64)>

List all checkpoints as (id, label, tick) triples, ordered oldest to newest.

pub fn clear_all(&mut self)

Remove all stored checkpoints.

pub fn stats(&self) -> CheckpointerStats

Compute aggregate statistics.