Struct Profiler 

pub struct Profiler {
    pub metrics: Vec<OperationMetrics>,
    /* private fields */
}

Performance profiler for tracking operation metrics

Fields

metrics: Vec<OperationMetrics>

Collected metrics

Implementations

impl Profiler

pub fn new() -> Self

Create a new profiler

pub fn enable_memory_tracking(&mut self)

Enable memory tracking

pub fn enable_flops_counting(&mut self)

Enable FLOPS counting

pub fn add_hook<F>(&mut self, hook: F)

where F: Fn(&OperationMetrics) + Send + Sync + 'static,

Add a profiling hook

pub fn start_operation( &mut self, name: &str, inputs: &[&Tensor], ) -> TorshResult<()>

Start profiling an operation

pub fn finish_operation(&mut self, outputs: &[&Tensor]) -> TorshResult<()>

Finish profiling an operation

pub fn get_metrics(&self, operation_name: &str) -> Vec<&OperationMetrics>

Get metrics for a specific operation

pub fn get_summary(&self, operation_name: &str) -> Option<OperationSummary>

Get summary statistics for an operation

pub fn clear(&mut self)

Clear all collected metrics

pub fn export_csv(&self) -> String

Export metrics to CSV format

Trait Implementations

impl Default for Profiler

fn default() -> Self

Returns the “default value” for a type.