fix(profile): capture one profiling step

Oseltamivir · Oseltamivir · commit f9d6523bb456 · 2026-05-21T13:38:00.000-07:00
diff --git a/utils/bench_serving/benchmark_serving.py b/utils/bench_serving/benchmark_serving.py
@@ -538,7 +538,7 @@ async def warmup_limited_req_fn():
                                          api_url=base_url + "/start_profile",
                                          prompt_len=test_prompt_len,
                                          output_len=test_output_len,
-                                         extra_body={"num_steps": 2, "merge_profiles": True, "profile_by_stage": True},
+                                         extra_body={"num_steps": 1, "merge_profiles": True, "profile_by_stage": True},
                                          logprobs=logprobs,
                                          best_of=best_of,
                                          multi_modal_content=test_mm_content,
