Struct ModelTracer 

pub struct ModelTracer { /* private fields */ }

Unified model tracer that coordinates all trace types.

Example

let config = ModelTracerConfig::lightweight();
let mut tracer = ModelTracer::new(config);

tracer.begin_forward(position);
// ... forward pass with trace hooks ...
if let Some(anomaly) = tracer.end_forward() {
    log::warn!("Anomaly: {}", anomaly);
}

Implementations

impl ModelTracer

pub fn new(config: ModelTracerConfig) -> Self

Create a new tracer with the given configuration.

pub fn config(&self) -> &ModelTracerConfig

Get the configuration.

pub fn current_logit_trace(&self) -> Option<&LogitEvolutionTrace>

Get a reference to the current logit trace (if any).

pub fn set_current_logit_trace(&mut self, trace: Option<LogitEvolutionTrace>)

Set the current logit trace (for testing purposes).

pub fn begin_forward(&mut self, position: usize)

Begin a forward pass at the given position.

pub fn record_layer_activation(&mut self, trace: LayerActivationTrace)

Record layer activation (called by executor after each layer).

pub fn record_attention(&mut self, trace: AttentionWeightTrace)

Record attention weights (called by attention brick).

pub fn record_logits(&mut self, layer_idx: usize, logits: &[f32])

Record logit state at a layer (called by lm_head or probe).

pub fn record_kv_state(&mut self, trace: KvCacheStateTrace)

Record KV cache state (called after each generation step).

pub fn record_quant_error(&mut self, trace: QuantizationErrorTrace)

Record quantization error for a brick.

pub fn end_forward(&mut self) -> Option<String>

Complete forward pass and check for anomalies.

Returns a description of the first anomaly detected, if any.

pub fn summary(&self) -> ModelTracerSummary

Get summary statistics.

pub fn summary_to_json(&self) -> String

Export summary as JSON for artifact validation.

pub fn clear(&mut self)

Clear all accumulated traces (free memory).