From e9be0ec15dfd39201116560ec5e325ebd2d186b1 Mon Sep 17 00:00:00 2001 From: Huy Do Date: Mon, 10 Feb 2025 18:02:04 -0800 Subject: [PATCH] Fix typo --- benchmarks/benchmark_latency.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/benchmarks/benchmark_latency.py b/benchmarks/benchmark_latency.py index 96d0eeee6f387..ff5e252c1ad26 100644 --- a/benchmarks/benchmark_latency.py +++ b/benchmarks/benchmark_latency.py @@ -44,7 +44,7 @@ def save_to_pytorch_benchmark_format( }, } - if os.environ.get("SAVE_IN_PYTORCH_BENCHMARK_FORMAT", False): + if os.environ.get("SAVE_TO_PYTORCH_BENCHMARK_FORMAT", False): # Don't use json suffix here as we don't want convert-results-json-to-markdown.py # to pick it up output_file = f"{os.path.splitext(args.output_json)[0]}.pytorch"