mirror of
https://github.com/deepseek-ai/DeepInfer.git
synced 2026-10-30 07:47:10 +08:00
Two bindings a prefill/decode split needs from the scheduler: - add_prompted_sequence(seq): put a sequence whose KV blocks and state were filled by the caller (received from a prompt-only engine) straight into the prompted queue, without cloning. The sequence holds prompt + first generated token, with everything but the last token computed; the checks assert that layout. - free_finished_seqs(): expose the existing helper that moves finished sequences out of running and returns them without freeing blocks or state, so a prompt-only engine can export the prefilled KV first and free the sequence itself afterwards. No scheduling behaviour changes for engines that do not call them. Co-Authored-By: Claude Fable 5.1 <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_01LFPG29daR1FhchJBTjiEKr |
||
|---|---|---|
| .. | ||
| input_metadata | ||
| block_manager.hpp | ||
| flops_estimator.hpp | ||
| mtp_varlen.hpp | ||
| outputs.hpp | ||
| prepare_sample.cpp | ||
| prepare_sample.hpp | ||
| process_output.hpp | ||
| pybind_output.cpp | ||
| pybind_scheduler.cpp | ||
| pybind_sequence.cpp | ||
| pybind.cpp | ||
| pybind.hpp | ||
| scheduler.hpp | ||
| sequence.hpp | ||
| utils.hpp | ||
| zos.cpp | ||
| zos.hpp | ||