Support profiling only prefill or decode without the other (#14182)

2025-12-01 14:46:30 +08:00
parent a2960bdd6b
commit e8ba5a668c
5 changed files with 11 additions and 1 deletions
--- a/python/sglang/srt/entrypoints/http_server.py
+++ b/python/sglang/srt/entrypoints/http_server.py
@@ -693,6 +693,7 @@ async def start_profile_async(obj: Optional[ProfileReqInput] = None):
        profile_by_stage=obj.profile_by_stage,
        merge_profiles=obj.merge_profiles,
        profile_prefix=obj.profile_prefix,
+        profile_stages=obj.profile_stages,
    )
    return Response(
        content="Start profiling.\n",