diff --git a/python/sglang/srt/entrypoints/engine.py b/python/sglang/srt/entrypoints/engine.py index 372bf41a6..02991a6ad 100644 --- a/python/sglang/srt/entrypoints/engine.py +++ b/python/sglang/srt/entrypoints/engine.py @@ -324,6 +324,24 @@ class Engine(EngineBase): loop = asyncio.get_event_loop() loop.run_until_complete(self.tokenizer_manager.stop_profile()) + def start_expert_distribution_record(self): + loop = asyncio.get_event_loop() + loop.run_until_complete( + self.tokenizer_manager.start_expert_distribution_record() + ) + + def stop_expert_distribution_record(self): + loop = asyncio.get_event_loop() + loop.run_until_complete( + self.tokenizer_manager.stop_expert_distribution_record() + ) + + def dump_expert_distribution_record(self): + loop = asyncio.get_event_loop() + loop.run_until_complete( + self.tokenizer_manager.dump_expert_distribution_record() + ) + def get_server_info(self): loop = asyncio.get_event_loop() internal_states = loop.run_until_complete(