Skip to content

Commit

Permalink
formatting
Browse files Browse the repository at this point in the history
  • Loading branch information
LouisChourakiSonos committed Dec 9, 2024
1 parent 66c131e commit 806925c
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 4 deletions.
4 changes: 2 additions & 2 deletions libcli/src/profile.rs
Original file line number Diff line number Diff line change
Expand Up @@ -96,14 +96,14 @@ pub fn profile(
&state.session_state.resolved_symbols,
)?;
session_handler.before_plan_eval(&mut state.session_state)?;

let mut entire = Duration::default();
while iters < bench_limits.max_loops && entire < bench_limits.max_time {
entire += rec_profiler_metal(&mut state, dg, inputs, &prefix)?.1;

iters += 1;
}

session_handler.after_plan_eval(&mut state.session_state)?;
entire
}
Expand Down
4 changes: 2 additions & 2 deletions libcli/src/terminal.rs
Original file line number Diff line number Diff line change
Expand Up @@ -370,7 +370,7 @@ pub fn render_summaries(
options: &DisplayParams,
) -> TractResult<()> {
let total = annotations.tags.values().sum::<NodeTags>();

if options.tmp_mem_usage {
if let Some(summary) = &annotations.memory_summary {
println!("{}", White.bold().paint("Memory summary"));
Expand All @@ -392,7 +392,7 @@ pub fn render_summaries(
White.bold().paint(format!("{:<43}", "Most time consuming operations")),
White.bold().paint(format!("{:<17}", "CPU")),
White.bold().paint(if options.has_accelerator { "Accelerator" } else { "" }),
);
);

for (op, (cpu_dur, accel_dur, n)) in annotations
.tags
Expand Down

0 comments on commit 806925c

Please sign in to comment.