From 50d1edaa7fcff16b49d2d9797599a4015a1926c3 Mon Sep 17 00:00:00 2001 From: Liangsheng Yin Date: Tue, 14 Jul 2026 16:35:04 -0700 Subject: [PATCH] [misc] Move SchedulerRecvSkipper into scheduler_components (#31222) --- python/sglang/srt/managers/scheduler.py | 4 +++- python/sglang/srt/managers/scheduler_components/dp_attn.py | 4 +++- .../recv_skipper.py} | 0 test/registered/unit/managers/test_scheduler_recv_skipper.py | 2 +- 4 files changed, 7 insertions(+), 3 deletions(-) rename python/sglang/srt/managers/{scheduler_recv_skipper.py => scheduler_components/recv_skipper.py} (100%) diff --git a/python/sglang/srt/managers/scheduler.py b/python/sglang/srt/managers/scheduler.py index f04458c2c..4d5078e61 100644 --- a/python/sglang/srt/managers/scheduler.py +++ b/python/sglang/srt/managers/scheduler.py @@ -207,6 +207,9 @@ from sglang.srt.managers.scheduler_components.pool_stats_observer import ( from sglang.srt.managers.scheduler_components.profiler_manager import ( SchedulerProfilerManager, ) +from sglang.srt.managers.scheduler_components.recv_skipper import ( + SchedulerRecvSkipper, +) from sglang.srt.managers.scheduler_components.request_receiver import ( SchedulerRequestReceiver, ) @@ -215,7 +218,6 @@ from sglang.srt.managers.scheduler_components.weight_updater import ( ) from sglang.srt.managers.scheduler_input_blocker import SchedulerInputBlocker from sglang.srt.managers.scheduler_pp_mixin import SchedulerPPMixin -from sglang.srt.managers.scheduler_recv_skipper import SchedulerRecvSkipper from sglang.srt.managers.utils import ( EmbeddingBatchResult, GenerationBatchResult, diff --git a/python/sglang/srt/managers/scheduler_components/dp_attn.py b/python/sglang/srt/managers/scheduler_components/dp_attn.py index 10f18df75..0851f7605 100644 --- a/python/sglang/srt/managers/scheduler_components/dp_attn.py +++ b/python/sglang/srt/managers/scheduler_components/dp_attn.py @@ -11,7 +11,9 @@ from sglang.srt.distributed.parallel_state import get_tp_group from sglang.srt.distributed.parallel_state_wrapper import ParallelState from sglang.srt.environ import envs from sglang.srt.managers.schedule_batch import ScheduleBatch -from sglang.srt.managers.scheduler_recv_skipper import SchedulerRecvSkipper +from sglang.srt.managers.scheduler_components.recv_skipper import ( + SchedulerRecvSkipper, +) from sglang.srt.mem_cache.allocator import BaseTokenToKVPoolAllocator from sglang.srt.mem_cache.base_prefix_cache import BasePrefixCache from sglang.srt.mem_cache.memory_pool import ReqToTokenPool diff --git a/python/sglang/srt/managers/scheduler_recv_skipper.py b/python/sglang/srt/managers/scheduler_components/recv_skipper.py similarity index 100% rename from python/sglang/srt/managers/scheduler_recv_skipper.py rename to python/sglang/srt/managers/scheduler_components/recv_skipper.py diff --git a/test/registered/unit/managers/test_scheduler_recv_skipper.py b/test/registered/unit/managers/test_scheduler_recv_skipper.py index df3e14e3d..b57c6749c 100644 --- a/test/registered/unit/managers/test_scheduler_recv_skipper.py +++ b/test/registered/unit/managers/test_scheduler_recv_skipper.py @@ -6,7 +6,7 @@ from sglang.test.test_utils import CustomTestCase, maybe_stub_sgl_kernel maybe_stub_sgl_kernel() -from sglang.srt.managers.scheduler_recv_skipper import ( # noqa: E402 +from sglang.srt.managers.scheduler_components.recv_skipper import ( # noqa: E402 SchedulerRecvSkipper, ) from sglang.srt.model_executor.forward_batch_info import ForwardMode # noqa: E402