Fix start_profile does not support with_stack and record_shapes (#6043)

2025-05-12 14:11:32 +08:00
parent 6ea05950b1
commit 3f2702ae51
3 changed files with 11 additions and 1 deletions
--- a/python/sglang/srt/entrypoints/http_server.py
+++ b/python/sglang/srt/entrypoints/http_server.py
@@ -338,7 +338,11 @@ async def start_profile_async(obj: Optional[ProfileReqInput] = None):
        obj = ProfileReqInput()

    await _global_state.tokenizer_manager.start_profile(
-        obj.output_dir, obj.num_steps, obj.activities
+        output_dir=obj.output_dir,
+        num_steps=obj.num_steps,
+        activities=obj.activities,
+        with_stack=obj.with_stack,
+        record_shapes=obj.record_shapes,
    )
    return Response(
        content="Start profiling.\n",