queue.rs 623 Bytes
Newer Older
1
2
3
// SPDX-FileCopyrightText: Copyright (c) 2024-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
// SPDX-License-Identifier: Apache-2.0

4
pub use dynamo_kv_router::queue::DEFAULT_MAX_BATCHED_TOKENS;
5

6
7
use crate::kv_router::sequence::RuntimeSequencePublisher;
use crate::local_model::runtime_config::ModelRuntimeConfig;
8
use dynamo_kv_router::scheduling::policy::RouterSchedulingPolicy;
9

10
/// Concrete `SchedulerQueue` wired to the runtime publisher and config types.
11
12
13
14
15
pub type SchedulerQueue = dynamo_kv_router::queue::SchedulerQueue<
    RuntimeSequencePublisher,
    ModelRuntimeConfig,
    RouterSchedulingPolicy,
>;