Support with_stack and record_shapes in profiler (#4740)

Co-authored-by: Lianmin Zheng <lianminzheng@gmail.com>
2025-03-28 14:01:42 +08:00
parent 265e756494
commit 8c04f0f2e1
2 changed files with 11 additions and 2 deletions
--- a/python/sglang/srt/managers/io_struct.py
+++ b/python/sglang/srt/managers/io_struct.py
@@ -675,6 +675,8 @@ class ProfileReq:
    output_dir: Optional[str] = None
    num_steps: Optional[int] = None
    activities: Optional[List[str]] = None
+    with_stack: Optional[bool] = None
+    record_shapes: Optional[bool] = None


@dataclass