Skip to content

Add asynchronous prefill scheduling - #26

Closed
lmcafee-nvidia wants to merge 12 commits into
defer-req-res-overlapfrom
async-sched-prefill
Closed

Add asynchronous prefill scheduling#26
lmcafee-nvidia wants to merge 12 commits into
defer-req-res-overlapfrom
async-sched-prefill

Conversation

@lmcafee-nvidia

Copy link
Copy Markdown
Owner

Summary

Add a dedicated asynchronous prefill path so dynamic inference can remain under async scheduling while requests move between prefill and decode.

Architecture

  • Route async controller iterations through phase orders tailored to their work:
    • prefill: sample or verify the previous output, resolve it, prepare and admit requests, then launch the next forward
    • decode: prepare, sample the previous output, launch the next forward, then resolve the previous output
    • MTP decode: sample and verify drafts, rewind rejected state, prepare, launch the next forward, then resolve
  • Make non-chunked request admission part of the async prefill flow, with non-mutating admission probes and the existing CUDA-graph admission rules.
  • Keep the first admitted prefill as a primer-only forward while preserving one base-model forward per controller iteration.
  • Preserve GPU-resident sampled-token handoff for decode and defer bookkeeping publication until prepared metadata is ready.
  • Keep expert-parallel dummy ranks in the same MTP-before-base-forward order as ranks with active work.
  • Extend request preparation and resolution to handle prefill-to-decode transitions and mixed survivor/admission batches without returning to the legacy path.

Phlip79 and others added 8 commits July 20, 2026 18:13
Signed-off-by: Philip Petrakian <ppetrakian@nvidia.com>
Signed-off-by: Lawrence McAfee <lmcafee@nvidia.com>
… failures (NVIDIA#5881)

Signed-off-by: Ajay Balasa <abalasa@nvidia.com>
)

Signed-off-by: Evgenii Zheltonozhskii <zheltonozhskiy@gmail.com>
Co-authored-by: Maanu Grover <maanug@nvidia.com>
Co-authored-by: claude[bot] <209825114+claude[bot]@users.noreply.github.com>
…IA#5570)

Signed-off-by: Yongqiang Wang <yongqiang.seagull@gmail.com>
Co-authored-by: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
Signed-off-by: Achyuthan Sivasankar <achyuthan.sivasankar@gmail.com>
Signed-off-by: svcnvidia-nemo-ci <svcnvidia-nemo-ci@nvidia.com>
Co-authored-by: svcnvidia-nemo-ci <svcnvidia-nemo-ci@nvidia.com>
Signed-off-by: Jingyue Wu <jingyuew@nvidia.com>
Signed-off-by: Philip Petrakian <ppetrakian@nvidia.com>
tdene and others added 4 commits July 21, 2026 04:45
Signed-off-by: Teodor-Dumitru Ene <teodord.ene@gmail.com>
Signed-off-by: Jingyue Wu <jingyuew@nvidia.com>
Signed-off-by: svcnvidia-nemo-ci <svcnvidia-nemo-ci@nvidia.com>
Co-authored-by: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
Co-authored-by: svcnvidia-nemo-ci <svcnvidia-nemo-ci@nvidia.com>
Signed-off-by: Lawrence McAfee <lmcafee@nvidia.com>
@lmcafee-nvidia

Copy link
Copy Markdown
Owner Author

Superseded by NVIDIA#5939 now that the overlap prerequisite has merged into upstream main.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

9 participants