1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
//! Core recording methods for BrickProfiler.
//!
//! Extracted from mod.rs to keep file sizes manageable.
//! Contains: start/stop (legacy string API), record_elapsed, record_elapsed_with_bytes,
//! set_brick_bottleneck, stats lookup, all_stats, all_brick_stats, brick_names, reset.
use super::BrickProfiler;
use crate::brick::exec_graph::{BrickBottleneck, BrickId, BrickStats};
impl BrickProfiler {
/// Start timing a brick. Returns timer handle.
///
/// IMPORTANT: For GPU operations, call sync AFTER the operation
/// completes but BEFORE calling stop().
#[must_use]
pub fn start(&self, name: &str) -> super::BrickTimer {
super::BrickTimer { name: name.to_string(), start: std::time::Instant::now() }
}
/// Stop timing and record the sample.
///
/// # Arguments
/// - `timer`: Timer handle from `start()`
/// - `elements`: Number of elements (tokens) processed
pub fn stop(&mut self, timer: super::BrickTimer, elements: u64) {
if !self.enabled {
return;
}
let elapsed = timer.start.elapsed();
let elapsed_ns = elapsed.as_nanos() as u64;
// PAR-200: Try fast path first if name matches a known BrickId
if let Some(brick_id) = BrickId::from_str(&timer.name) {
let stats = &mut self.brick_stats[brick_id as usize];
stats.add_sample(elapsed_ns, elements);
} else {
// Fall back to dynamic stats
let name = timer.name;
let stats =
self.dynamic_stats.entry(name.clone()).or_insert_with(|| BrickStats::new(&name));
stats.add_sample(elapsed_ns, elements);
}
// Update totals
self.total_tokens += elements;
self.total_ns += elapsed_ns;
}
/// Record a pre-measured duration for a brick.
///
/// PAR-073: This method allows timing with raw `Instant` calls, avoiding
/// borrow conflicts when profiling CUDA operations that also need `&mut self`.
///
/// # Arguments
/// - `name`: Brick name
/// - `elapsed`: Duration of the operation (from `Instant::elapsed()`)
/// - `elements`: Number of elements (tokens) processed
///
/// # Example
/// ```rust,ignore
/// let start = std::time::Instant::now();
/// cuda_stream.synchronize()?;
/// self.some_cuda_operation()?;
/// cuda_stream.synchronize()?;
/// let elapsed = start.elapsed();
/// self.profiler.record_elapsed("SomeBrick", elapsed, 1);
/// ```
pub fn record_elapsed(&mut self, name: &str, elapsed: std::time::Duration, elements: u64) {
if !self.enabled {
return;
}
let elapsed_ns = elapsed.as_nanos() as u64;
// PAR-200: Try fast path first if name matches a known BrickId
if let Some(brick_id) = BrickId::from_str(name) {
let stats = &mut self.brick_stats[brick_id as usize];
stats.add_sample(elapsed_ns, elements);
} else {
// Fall back to dynamic stats
let stats =
self.dynamic_stats.entry(name.to_string()).or_insert_with(|| BrickStats::new(name));
stats.add_sample(elapsed_ns, elements);
}
// Update totals
self.total_tokens += elements;
self.total_ns += elapsed_ns;
}
/// PMAT-451: Record elapsed time with byte metrics for compression workloads.
///
/// # Arguments
/// - `name`: Brick name
/// - `elapsed`: Duration of the operation
/// - `elements`: Number of elements (pages) processed
/// - `input_bytes`: Original uncompressed size
/// - `output_bytes`: Compressed output size
///
/// # Example
/// ```rust,ignore
/// let start = std::time::Instant::now();
/// let compressed = zstd_compress(&page_data);
/// let elapsed = start.elapsed();
/// profiler.record_elapsed_with_bytes(
/// "ZstdCompress",
/// elapsed,
/// 1,
/// page_data.len() as u64,
/// compressed.len() as u64,
/// );
/// ```
pub fn record_elapsed_with_bytes(
&mut self,
name: &str,
elapsed: std::time::Duration,
elements: u64,
input_bytes: u64,
output_bytes: u64,
) {
if !self.enabled {
return;
}
let elapsed_ns = elapsed.as_nanos() as u64;
// PAR-200: Try fast path first if name matches a known BrickId
if let Some(brick_id) = BrickId::from_str(name) {
let stats = &mut self.brick_stats[brick_id as usize];
stats.add_sample_with_bytes(elapsed_ns, elements, input_bytes, output_bytes);
} else {
// Fall back to dynamic stats
let stats =
self.dynamic_stats.entry(name.to_string()).or_insert_with(|| BrickStats::new(name));
stats.add_sample_with_bytes(elapsed_ns, elements, input_bytes, output_bytes);
}
// Update totals
self.total_tokens += elements;
self.total_ns += elapsed_ns;
}
/// PMAT-451: Set bottleneck classification for a brick.
pub fn set_brick_bottleneck(&mut self, name: &str, bottleneck: BrickBottleneck) {
// PAR-200: Try fast path first
if let Some(brick_id) = BrickId::from_str(name) {
self.brick_stats[brick_id as usize].set_bottleneck(bottleneck);
} else if let Some(stats) = self.dynamic_stats.get_mut(name) {
stats.set_bottleneck(bottleneck);
}
}
/// Get statistics for a specific brick by name.
///
/// First checks known BrickId types (O(1)), then falls back to dynamic stats.
#[must_use]
pub fn stats(&self, name: &str) -> Option<&BrickStats> {
// Try fast path first
if let Some(brick_id) = BrickId::from_str(name) {
let stats = &self.brick_stats[brick_id as usize];
if stats.count > 0 {
return Some(stats);
}
}
// Fall back to dynamic stats
self.dynamic_stats.get(name)
}
/// Get all brick statistics (legacy API, returns dynamic stats only).
///
/// For full statistics including known bricks, use `all_brick_stats()` instead.
#[must_use]
#[deprecated(since = "0.12.0", note = "Use all_brick_stats() for complete statistics")]
pub fn all_stats(&self) -> &std::collections::HashMap<String, BrickStats> {
&self.dynamic_stats
}
/// Get all brick statistics including both known and dynamic bricks.
pub fn all_brick_stats(&self) -> impl Iterator<Item = &BrickStats> {
self.brick_stats.iter().filter(|s| s.count > 0).chain(self.dynamic_stats.values())
}
/// Get all brick names.
#[must_use]
pub fn brick_names(&self) -> Vec<String> {
let mut names: Vec<String> = self
.brick_stats
.iter()
.enumerate()
.filter(|(_, s)| s.count > 0)
.map(|(i, _)| {
let brick_id = BrickId::ALL[i];
brick_id.name().to_string()
})
.collect();
names.extend(self.dynamic_stats.keys().cloned());
names
}
/// Reset all statistics.
pub fn reset(&mut self) {
for stats in &mut self.brick_stats {
stats.count = 0;
stats.total_ns = 0;
stats.min_ns = u64::MAX;
stats.max_ns = 0;
stats.total_elements = 0;
stats.total_bytes = 0;
stats.total_compressed_bytes = 0;
}
self.dynamic_stats.clear();
self.pending.clear();
self.total_tokens = 0;
self.total_ns = 0;
}
}