Skip to content

unified scheduler prototype#19

Open
QiJune wants to merge 2 commits intomainfrom
unified_scheduler
Open

unified scheduler prototype#19
QiJune wants to merge 2 commits intomainfrom
unified_scheduler

Conversation

@QiJune
Copy link
Copy Markdown
Owner

@QiJune QiJune commented Mar 2, 2026

@coderabbitai summary

Description

Test Coverage

PR Checklist

Please review the following before submitting your PR:

  • PR description clearly explains what and why. If using CodeRabbit's summary, please make sure it makes sense.

  • PR Follows TRT-LLM CODING GUIDELINES to the best of your knowledge.

  • Test cases are provided for new code paths (see test instructions)

  • Any new dependencies have been scanned for license and vulnerabilities

  • CODEOWNERS updated if ownership changes

  • Documentation updated as needed

  • Update tava architecture diagram if there is a significant design change in PR.

  • The reviewers assigned automatically/manually are appropriate for the PR.

  • Please check this after reviewing the above items as appropriate for this PR.

GitHub Bot Help

To see a list of available CI bot commands, please comment /bot help.

@QiJune QiJune force-pushed the unified_scheduler branch 6 times, most recently from 7f4123d to e66bd0a Compare March 9, 2026 11:06
@QiJune QiJune force-pushed the unified_scheduler branch 5 times, most recently from 4730546 to f505332 Compare March 20, 2026 20:45
…cheduler

Pure-Python scheduler with fused single-pass capacity + token-budget
scheduling. Drop-in replacement for SimpleScheduler — same
schedule_request() interface, same py_executor code path.

Key changes:
- Fused single-pass scheduling via TokenBudgetTracker (eliminates
  separate microbatch pass)
- Batched block decrements and preview/commit reservation
- Cached C++ property calls to reduce boundary crossings
- mypyc compilation support (setup_mypyc.py)

Gated by SchedulerConfig(use_python_scheduler=True).

Signed-off-by: junq <22017000+QiJune@users.noreply.github.com>
@QiJune QiJune force-pushed the unified_scheduler branch from f505332 to 3558278 Compare March 20, 2026 22:05
Signed-off-by: junq <22017000+QiJune@users.noreply.github.com>
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.

1 participant