[misc] Move SchedulerRecvSkipper into scheduler_components (#31222)
This commit is contained in:
@@ -207,6 +207,9 @@ from sglang.srt.managers.scheduler_components.pool_stats_observer import (
|
|||||||
from sglang.srt.managers.scheduler_components.profiler_manager import (
|
from sglang.srt.managers.scheduler_components.profiler_manager import (
|
||||||
SchedulerProfilerManager,
|
SchedulerProfilerManager,
|
||||||
)
|
)
|
||||||
|
from sglang.srt.managers.scheduler_components.recv_skipper import (
|
||||||
|
SchedulerRecvSkipper,
|
||||||
|
)
|
||||||
from sglang.srt.managers.scheduler_components.request_receiver import (
|
from sglang.srt.managers.scheduler_components.request_receiver import (
|
||||||
SchedulerRequestReceiver,
|
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_input_blocker import SchedulerInputBlocker
|
||||||
from sglang.srt.managers.scheduler_pp_mixin import SchedulerPPMixin
|
from sglang.srt.managers.scheduler_pp_mixin import SchedulerPPMixin
|
||||||
from sglang.srt.managers.scheduler_recv_skipper import SchedulerRecvSkipper
|
|
||||||
from sglang.srt.managers.utils import (
|
from sglang.srt.managers.utils import (
|
||||||
EmbeddingBatchResult,
|
EmbeddingBatchResult,
|
||||||
GenerationBatchResult,
|
GenerationBatchResult,
|
||||||
|
|||||||
@@ -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.distributed.parallel_state_wrapper import ParallelState
|
||||||
from sglang.srt.environ import envs
|
from sglang.srt.environ import envs
|
||||||
from sglang.srt.managers.schedule_batch import ScheduleBatch
|
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.allocator import BaseTokenToKVPoolAllocator
|
||||||
from sglang.srt.mem_cache.base_prefix_cache import BasePrefixCache
|
from sglang.srt.mem_cache.base_prefix_cache import BasePrefixCache
|
||||||
from sglang.srt.mem_cache.memory_pool import ReqToTokenPool
|
from sglang.srt.mem_cache.memory_pool import ReqToTokenPool
|
||||||
|
|||||||
@@ -6,7 +6,7 @@ from sglang.test.test_utils import CustomTestCase, maybe_stub_sgl_kernel
|
|||||||
|
|
||||||
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,
|
SchedulerRecvSkipper,
|
||||||
)
|
)
|
||||||
from sglang.srt.model_executor.forward_batch_info import ForwardMode # noqa: E402
|
from sglang.srt.model_executor.forward_batch_info import ForwardMode # noqa: E402
|
||||||
|
|||||||
Reference in New Issue
Block a user