Skip to main content

candle_graph/trace/
mod.rs

1//! TensorFlow-Profiler-style trace protocol (`candle-graph/trace/10`).
2//!
3//! Trace-only evidence: spans, timed ops, tensor snapshots, memory events, gradient facts,
4//! and span edges. No static Rust analysis — importers aggregate JSONL into
5//! [`document::TraceDocument`].
6
7pub mod document;
8pub mod events;
9pub mod health;
10pub mod memory;
11pub mod schema;
12
13pub use document::{parse_trace, write_jsonl, TraceDocument};
14pub use events::{
15    DeviceIntervalEvent, DeviceMemoryEvent, EdgeEvent, GradientEvent, MemoryEvent, OpEvent,
16    SpanEndEvent, SpanStartEvent, TensorEvent, TensorStatsEvent, TerminalEvent, TraceEvent,
17};
18pub use health::{analyze_health, EvidenceCoverage, HealthIssue, HealthSeverity, TraceHealth};
19pub use memory::{category_for_step, dense_tensor_bytes, MemoryAction, MemoryCategory};
20pub use schema::{
21    ComparisonIdentity, GradientState, RunOutcome, SpanKind, SpanRecord, TimingMode, TraceRunMeta,
22    TraceSummary, SCHEMA,
23};