From fe92f3563c693c6e4b81359f49e6902a931ed66c Mon Sep 17 00:00:00 2001
From: Yilong Zhao <74357408+happierpig@users.noreply.github.com>
Date: Fri, 3 Apr 2026 20:47:09 -0700
Subject: [PATCH] dp: add profile req hook (#22083)

---
 python/sglang/srt/managers/data_parallel_controller.py | 2 ++
 1 file changed, 2 insertions(+)

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),
             ]