Skip to content

Conversation

@TianyiZhao1437
Copy link
Collaborator

@TianyiZhao1437 TianyiZhao1437 commented Nov 11, 2025

📋 PR Title Format

The PR title should follow the format:

type(scope): concise message (max 50 chars)

Where:

  • type is one of: feat, fix, docs, refactor, perf, test, chore.
  • scope is optional and describes the part of the codebase affected (e.g., auth, ui, api).
  • concise message is a short description of the change (max 50 chars).

📝 Change Type

Please select the type of change this PR introduces (choose one or more):

  • feat: New feature.
  • fix: Bug fix.
  • docs: Documentation only changes.
  • refactor: A code change that neither fixes a bug nor adds a feature.
  • perf: Performance improvement.
  • test: Adding missing tests or correcting existing tests.
  • chore: Maintenance tasks (e.g., updating dependencies).

💡 Description

Support TP in scheduler mode.
TODO: Currently we assume node has the same type of GPUs. Need to support different types of GPUs on a single node in the future.

Key Changes

  1. Launch tp_rank=0 in the main process and launch tp_rank>0 each in a subprocess.
  2. Scheduler is now aware of GPU numbers of each node. Scheduler assigns tp_size in layer allocation.
  3. Update deprecated SGL_ENABLE_JIT_DEEPGEMM environment to SGLANG_ENABLE_JIT_DEEPGEMM.

🔗 Related Issues

List any issues this PR closes or relates to:

  • Closes #IssueNumber (e.g., Closes #123)

✅ Checklist

Please ensure the following points are addressed before merging:

  • I have performed a self-review of my own code.
  • I have added/updated tests that prove my fix or feature works (if applicable).
  • I have updated the documentation (if necessary).
  • My code follows the project's style guidelines.

@TianyiZhao1437 TianyiZhao1437 requested a review from a team November 11, 2025 12:58
@TianyiZhao1437 TianyiZhao1437 linked an issue Nov 12, 2025 that may be closed by this pull request
@TianyiZhao1437 TianyiZhao1437 merged commit b76a93a into GradientHQ:main Nov 12, 2025
5 checks passed
@TianyiZhao1437 TianyiZhao1437 deleted the dev-tp-scheduler branch November 12, 2025 11:21
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.

[Feature]: Support TP on gpu

2 participants