Skip to content

let allgather and alltoall execute in parallel when both attention and MOE used TP #7723

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
Open
taozhiwei wants to merge 2 commits into
base: master
Choose a base branch
from
Open

let allgather and alltoall execute in parallel when both attention and MOE used TP #7723

taozhiwei wants to merge 2 commits into from
+154 −23

Conversation

@taozhiwei
Copy link
Contributor

@taozhiwei taozhiwei commented Dec 11, 2025

let allgather and alltoall execute in parallel when both attention and MOE used TP.

@taozhiwei taozhiwei force-pushed the myfeature2 branch 2 times, most recently from 14be0fd to bccd2bc Compare December 11, 2025 11:25
@delock
Copy link
Collaborator

delock commented Dec 14, 2025

Hi @taozhiwei, do you have setup info and performance data about this optimization?

@wunongruanyu
Copy link

wunongruanyu commented Dec 15, 2025

1

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

Reviewers

@tohtana tohtana Awaiting requested review from tohtana tohtana is a code owner

@tjruwase tjruwase Awaiting requested review from tjruwase tjruwase is a code owner

@loadams loadams Awaiting requested review from loadams loadams is a code owner

At least 1 approving review is required to merge this pull request.

Assignees

No one assigned

Labels

None yet

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

4 participants

@taozhiwei @delock @wunongruanyu @sfc-gh-truwase