Support profile args in Engine API (#6539)

4da55336 · fzyzcjy · GitHub · ac964d2e · 4da55336
Unverified Commit 4da55336 authored Sep 14, 2025 by fzyzcjy Committed by GitHub Sep 14, 2025
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

python/sglang/srt/entrypoints/engine.py python/sglang/srt/entrypoints/engine.py +2 -2

No files found.
--- a/python/sglang/srt/entrypoints/engine.py
+++ b/python/sglang/srt/entrypoints/engine.py
@@ -364,9 +364,9 @@ class Engine(EngineBase):
        loop = asyncio.get_event_loop()
        return loop.run_until_complete(self.tokenizer_manager.flush_cache())

-    def start_profile(self):
+    def start_profile(self, **kwargs):
        loop = asyncio.get_event_loop()
-        loop.run_until_complete(self.tokenizer_manager.start_profile())
+        loop.run_until_complete(self.tokenizer_manager.start_profile(**kwargs))

    def stop_profile(self):
        loop = asyncio.get_event_loop()