Fix latency benchmark script (#2035)

2023-12-11 11:19:08 -08:00 · 2023-12-11 11:19:08 -08:00 · 5dd80d3777
commit 5dd80d3777
parent beeee69bc9
1 changed files with 1 additions and 1 deletions
--- a/benchmarks/benchmark_latency.py
+++ b/benchmarks/benchmark_latency.py
@ -72,7 +72,7 @@ def main(args: argparse.Namespace):
    # Benchmark.
    latencies = []
    for _ in tqdm(range(args.num_iters), desc="Profiling iterations"):
-        latencies.append(run_to_completion(profile=False))
+        latencies.append(run_to_completion(profile_dir=None))
    print(f'Avg latency: {np.mean(latencies)} seconds')