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/managers/io_struct.py
+++ b/python/sglang/srt/managers/io_struct.py
@@ -836,6 +836,8 @@ class ProfileReqInput:
    # the caller doesn't need to run stop_profile.
    num_steps: Optional[int] = None
    activities: Optional[List[Literal["CPU", "GPU", "MEM", "CUDA_PROFILER"]]] = None
+    with_stack: Optional[bool] = None
+    record_shapes: Optional[bool] = None


 class ProfileReqType(Enum):