Skip to content

Support expert parallel in Transformers backend #26162

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom
Merged
merged 5 commits into from
Oct 4, 2025
Merged

Support expert parallel in Transformers backend #26162

merged 5 commits into from
Oct 4, 2025

Conversation

hmellor
Copy link
Member

@hmellor hmellor commented Oct 3, 2025 •
edited by github-actions bot
Loading

This PR solves 2 problems that enables expert parallel in the Transformers backend:

  • Ensures that the dtypes of topk_ids (torch.int32) and topk_weights (torch.float32) match what is expected from FusedMoE.select_experts
  • Gathers the topk_ids which are passed directly from Transformers as is done for the hidden_states and topk_weights automatically inside FusedMoE

@hmellor hmellor moved this to In Progress in Transformers backend Oct 3, 2025
@hmellor
Update doc
0061256 
Signed-off-by: Harry Mellor <[email protected]>
@hmellor hmellor marked this pull request as ready for review October 3, 2025 11:22
@mergify mergify bot added the documentation Improvements or additions to documentation label Oct 3, 2025
@hmellor hmellor requested a review from Isotr0py October 3, 2025 13:37
@Isotr0py Isotr0py enabled auto-merge (squash) October 4, 2025 02:47
@github-actions github-actions bot added the ready ONLY add when PR is ready to merge/full CI is needed label Oct 4, 2025
@Isotr0py Isotr0py merged commit d3d649e into vllm-project:main Oct 4, 2025
54 checks passed
@github-project-automation github-project-automation bot moved this from In Progress to Done in Transformers backend Oct 4, 2025
@hmellor hmellor deleted the transformers-backend-ep branch October 4, 2025 07:37
tomeras91 pushed a commit to tomeras91/vllm that referenced this pull request Oct 6, 2025
@hmellor @Isotr0py @tomeras91
Support expert parallel in Transformers backend (vllm-project#26162)
505ce80 
Signed-off-by: Harry Mellor <[email protected]>
Co-authored-by: Isotr0py <[email protected]>
Signed-off-by: Tomer Asida <[email protected]>
karan pushed a commit to karan/vllm that referenced this pull request Oct 6, 2025
@hmellor @Isotr0py @karan
Support expert parallel in Transformers backend (vllm-project#26162)
c03bb7f 
Signed-off-by: Harry Mellor <[email protected]>
Co-authored-by: Isotr0py <[email protected]>
Signed-off-by: Karan Goel <[email protected]>
southfreebird pushed a commit to southfreebird/vllm that referenced this pull request Oct 7, 2025
@hmellor @Isotr0py @southfreebird
Support expert parallel in Transformers backend (vllm-project#26162)
6320814 
Signed-off-by: Harry Mellor <[email protected]>
Co-authored-by: Isotr0py <[email protected]>
xuebwang-amd pushed a commit to xuebwang-amd/vllm that referenced this pull request Oct 10, 2025
@hmellor @Isotr0py @xuebwang-amd
Support expert parallel in Transformers backend (vllm-project#26162)
4c10627 
Signed-off-by: Harry Mellor <[email protected]>
Co-authored-by: Isotr0py <[email protected]>
Signed-off-by: xuebwang-amd <[email protected]>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

@Isotr0py Isotr0py Isotr0py approved these changes

Assignees

No one assigned

Labels

documentation Improvements or additions to documentation ready ONLY add when PR is ready to merge/full CI is needed

Projects

Transformers backend
Status: Done

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

2 participants

@hmellor @Isotr0py