diff --git a/crates/ipu-codegen/src/package/benchmark.rs b/crates/ipu-codegen/src/package/benchmark.rs index c1f9e8a..e622e05 100644 --- a/crates/ipu-codegen/src/package/benchmark.rs +++ b/crates/ipu-codegen/src/package/benchmark.rs @@ -27,6 +27,7 @@ pub struct ExpansionTiming { pub mid_values: usize, pub expand_ms: f64, pub tile_lists_ms: f64, + pub footprint_ms: f64, pub shards: usize, pub kernels: usize, pub local_copies: usize, @@ -129,6 +130,9 @@ pub fn benchmark_mid_expansion( )?; let expand_ms = start.elapsed().as_secs_f64() * 1000.0; let start = Instant::now(); + let _footprint = crate::estimate::program_footprint(&expanded)?; + let footprint_ms = start.elapsed().as_secs_f64() * 1000.0; + let start = Instant::now(); let low = crate::low::lower_to_tiles(&expanded, config.diagnostic_checkpoints); let tile_lists_ms = start.elapsed().as_secs_f64() * 1000.0; let timing = ExpansionTiming { @@ -137,6 +141,7 @@ pub fn benchmark_mid_expansion( mid_values: mid.values.len(), expand_ms, tile_lists_ms, + footprint_ms, shards: low.shards.len(), kernels: low.kernel_runs.len(), local_copies: low.local_copies.len(),