Crate axonml_profile 

Performance profiling for AxonML.

Unified Profiler combining MemoryProfiler (allocation tracking, peak/ current stats), ComputeProfiler (per-op timing + FLOP counting), TimelineProfiler (event-based recording with start/stop spans). RAII ProfileGuard + profile_scope! macro for zero-overhead-when-disabled instrumentation. Global singleton via global_profiler(). BottleneckAnalyzer detects memory-bound, compute-bound, I/O-bound, launch-overhead, and synchronization bottlenecks. ProfileReport exports to Text/JSON/Markdown/HTML.

File

crates/axonml-profile/src/lib.rs

Author

Andrew Jewell Sr. — AutomataNexus LLC ORCID: 0009-0005-2158-7060

Updated

April 14, 2026 11:15 PM EST

Disclaimer

Use at own risk. This software is provided “as is”, without warranty of any kind, express or implied. The author and AutomataNexus shall not be held liable for any damages arising from the use of this software.

Re-exports

pub use bottleneck::Bottleneck;

pub use bottleneck::BottleneckAnalyzer;

pub use bottleneck::BottleneckType;

pub use compute::ComputeProfiler;

pub use compute::OperationStats;

pub use compute::ProfiledOp;

pub use error::ProfileError;

pub use error::ProfileResult;

pub use memory::AllocationRecord;

pub use memory::MemoryProfiler;

pub use memory::MemoryStats;

pub use report::ProfileReport;

pub use report::ReportFormat;

pub use timeline::Event;

pub use timeline::EventType;

pub use timeline::TimelineProfiler;

Modules

bottleneck

Bottleneck Detection — Rule-Based Compute and Memory Analyzer

compute

Compute Profiling — Per-Operation Timing, FLOPs, and Bandwidth

error

Profile Error Types — Failures for Compute and Memory Profiling

memory

Memory Profiling — Named Allocation Tracking and Leak Detection

report

Report Generation — Text, JSON, Markdown, and HTML Profile Reports

timeline

Timeline Profiling — Time-Stamped Event Recorder

Macros

profile_scope

Creates a profile guard for automatic scope-based profiling.

Structs

ProfileGuard

RAII guard for automatic profiling scope.

Profiler

Unified profiler combining memory, compute, and timeline profiling.

Functions

global_profiler

Gets the global profiler instance.

record_alloc

Records a memory allocation using the global profiler.

record_free

Records a memory deallocation using the global profiler.

start

Starts profiling an operation using the global profiler.

stop

Stops profiling an operation using the global profiler.