Skip to main content

Module continuous_engine

Module continuous_engine 

Source
Expand description

Continuous Batching Engine

Iteration-level continuous batching: each step processes a mixed batch of prefill and decode requests selected by the scheduler. Multiple callers can submit requests concurrently. An iteration_lock makes publication atomic with planning, while the single background driver owns device-wave execution.

Structsยง

ContinuousBatchEngine
Continuous batching inference engine.
SequenceState
State of a running sequence in the continuous batch.