From 2c6274664532d9f82e8ce3cb221c7bb3d0c19e71 Mon Sep 17 00:00:00 2001 From: Daniel Vega-Myhre Date: Tue, 30 Sep 2025 17:21:59 -0700 Subject: [PATCH] [moe training] update llama4 bench script to handle torchtitan new log format --- benchmarks/float8/training/parse_torchtitan_logs.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/benchmarks/float8/training/parse_torchtitan_logs.py b/benchmarks/float8/training/parse_torchtitan_logs.py index 0ad939983e..9d8bcef120 100644 --- a/benchmarks/float8/training/parse_torchtitan_logs.py +++ b/benchmarks/float8/training/parse_torchtitan_logs.py @@ -23,7 +23,7 @@ def main(args: Namespace): print(" Calculating training performance metrics") print("=====================================================") - log_pattern = re.compile(r"step: (\d+).*?memory: ([\d.]+)GiB.*?tps: ([\d,]+)") + log_pattern = re.compile(r"step:.*?(\d+).*?memory:.*?([\d.]+)GiB.*?tps:.*?([\d,]+)") assert os.path.exists(args.log_file), f"{args.log_file} does not exist"