From 0f97e2792e83e9f16c1c7355a0b5ec48b03fd136 Mon Sep 17 00:00:00 2001 From: "Wu, Chunyuan" Date: Fri, 28 Mar 2025 16:35:04 +0800 Subject: [PATCH] rename profiling trace --- python/sglang/bench_one_batch.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/sglang/bench_one_batch.py b/python/sglang/bench_one_batch.py index c8ca6cb5bb7e..5c4d15569faa 100644 --- a/python/sglang/bench_one_batch.py +++ b/python/sglang/bench_one_batch.py @@ -357,7 +357,7 @@ def latency_test_run_once( if profile: profiler.stop() - profile_filename = f"{profile_filename_prefix}_batch{batch_size}_input{input_len}_output{output_len}_prefill.trace.json.gz" + profile_filename = f"Trace_prefill_{profile_filename_prefix}_batch{batch_size}_input{input_len}_output{output_len}.trace.json.gz" parent_dir = os.path.dirname(os.path.abspath(profile_filename)) os.makedirs(parent_dir, exist_ok=True) profiler.export_chrome_trace(profile_filename) @@ -402,7 +402,7 @@ def latency_test_run_once( if i == output_len / 2 and profile: profiler.stop() - profile_filename = f"{profile_filename_prefix}_batch{batch_size}_input{input_len}_output{output_len}_decode.trace.json.gz" + profile_filename = f"Trace_decode_{profile_filename_prefix}_batch{batch_size}_input{input_len}_output{output_len}.trace.json.gz" parent_dir = os.path.dirname(os.path.abspath(profile_filename)) os.makedirs(parent_dir, exist_ok=True) profiler.export_chrome_trace(profile_filename)