Skip to main content

Module engine

Module engine 

Source
Expand description

Runtime-neutral mock inference schedulers and attention-DP composition.

Modules§

generalized
Runtime-neutral mock-engine composition.

Structs§

Admission
Request admission exposed at pass start.
CacheTierAttribution
First-admission provenance for prompt tokens reused across cache tiers.
CommandEffects
Effects of a scheduler command.
EngineConfig
Serializable configuration for one scheduler rank.
EngineFactory
Constructs engines with one process-local timing provider.
ForwardPassMetrics
Per-pass scheduling statistics.
HandoffId
Stable identifier for one replay-local prefill-to-decode handoff attempt.
HandoffTransferTiming
Source-provided inputs for calculating KV transfer delay.
KvBlock
One runtime-neutral KV block identity.
KvEvent
Ordered runtime-neutral KV event emitted by one rank.
Metrics
Rank-local scheduler and G1 metrics.
NativeHostOffloadConfig
Physical controls for framework-native G1-to-host offload.
Output
One client-visible output.
PassCompletionEffects
Effects released at the modeled pass completion boundary.
PassStartEffects
Effects visible as soon as an engine pass starts.
PressureEvent
One scheduler-owned KV pressure action emitted with pass-start effects.
PressureState
Runtime-neutral scheduler and KV occupancy around one pressure action.
Request
Runtime-neutral request accepted by the rank engine.
RunningMean
A generic running mean calculator with a fixed-size sliding window. Maintains a running sum and count to compute the mean in O(1) time.
SchedulerRank
One preserved vLLM/SGLang scheduler rank behind the neutral contract.
SglangConfig
Serializable SGLang scheduler controls.
StoredBlocks
A consecutive set of newly visible blocks.
TrtllmConfig
Serializable TensorRT-LLM scheduler controls.

Enums§

Backend
Scheduler semantics selected for an AISimulate rank.
Command
Commands supported by the standalone scheduler.
CommandResult
Result of applying a scheduler command.
KvEventData
Runtime-neutral KV event payload.
LifecycleEvent
Asynchronous scheduler lifecycle fact consumed by the Replayer’s handoff coordinator.
PreemptionMode
Decode preemption victim selection.
PressureKind
Scheduler action taken to relieve KV pressure.
SglangSchedulePolicy
SGLang waiting-queue ordering.
TimingModelConfig
Serializable timing-provider selection.
TransferTimingMode
Prompt footprint used when the Replayer models KV transfer time.
TrtllmCapacityPolicy
TensorRT-LLM capacity scheduler policy.
WorkerType
Scheduler role.

Traits§

TimingModel
Runtime latency model injected at the engine boundary.

Functions§

normalize_conditional_accept_rates
Parse and normalize MTP/EAGLE conditional draft-token acceptance rates.
prefill_handoff_delay_ms
Compute the client-visible prefill-to-decode handoff delay.

Type Aliases§

Engine
A single-rank or attention-DP engine.