From 2aa69953085de5963a9542ec7561b6f835402b8e Mon Sep 17 00:00:00 2001 From: Shangming Cai Date: Mon, 25 May 2026 23:52:58 +0800 Subject: [PATCH] [CI] Enable EPD CI for EPD architecture enhancements (#26281) --- python/sglang/srt/disaggregation/encode_receiver.py | 3 +++ test/registered/disaggregation/test_epd_disaggregation.py | 2 +- 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/python/sglang/srt/disaggregation/encode_receiver.py b/python/sglang/srt/disaggregation/encode_receiver.py index 0e06caf6d..72f3982cf 100644 --- a/python/sglang/srt/disaggregation/encode_receiver.py +++ b/python/sglang/srt/disaggregation/encode_receiver.py @@ -441,6 +441,7 @@ class WaitingImageRequest: recv_req: TokenizedGenerateReqInput, mm_processor, encoder_urls, + model_type, host_name, receive_count, ): @@ -451,6 +452,7 @@ class WaitingImageRequest: self.thread = None self.mm_processor = mm_processor self.encoder_urls = encoder_urls + self.model_type = model_type self.host_name = host_name self.receive_count = receive_count self.num_items_assigned = recv_req.num_items_assigned @@ -926,6 +928,7 @@ class MMReceiverBase(ABC): recv_req=recv_req, mm_processor=self.mm_processor, encoder_urls=self.encode_urls, + model_type=self.model_type, host_name=self.hostname, receive_count=self.tp_size, ) diff --git a/test/registered/disaggregation/test_epd_disaggregation.py b/test/registered/disaggregation/test_epd_disaggregation.py index a115db259..0c370fcda 100644 --- a/test/registered/disaggregation/test_epd_disaggregation.py +++ b/test/registered/disaggregation/test_epd_disaggregation.py @@ -36,7 +36,7 @@ DEFAULT_OMNI_MODEL = "Qwen/Qwen3-Omni-30B-A3B-Instruct" QWEN35_27B_MODEL = "Qwen/Qwen3.5-27B" -register_cuda_ci(est_time=97, suite="nightly-4-gpu", nightly=True) +register_cuda_ci(est_time=97, stage="base-c", runner_config="4-gpu-h100") @unittest.skipIf(