diff --git a/python/sglang/srt/managers/data_parallel_controller.py b/python/sglang/srt/managers/data_parallel_controller.py index 7b92d1a17..3256e07f0 100644 --- a/python/sglang/srt/managers/data_parallel_controller.py +++ b/python/sglang/srt/managers/data_parallel_controller.py @@ -33,6 +33,7 @@ from sglang.srt.managers.io_struct import ( BatchTokenizedEmbeddingReqInput, BatchTokenizedGenerateReqInput, BlockReqInput, + ProfileReq, TokenizedEmbeddingReqInput, TokenizedGenerateReqInput, WatchLoadUpdateReq, @@ -218,6 +219,7 @@ class DataParallelController: (BatchTokenizedGenerateReqInput, self.dispatch_batch_generate), (BatchTokenizedEmbeddingReqInput, self.dispatch_batch_embedding), (BlockReqInput, self.send_to_all_workers), + (ProfileReq, self.send_to_all_workers), (WatchLoadUpdateReq, self.handle_load_update_req), (ActiveRanksOutput, self.update_active_ranks), ]