Skip to content

Conversation

@shengfukevin
Copy link
Contributor

Fixed support to replay all2all.

Depend on PyTorch pytorch/pytorch#149485.

Test plan:
constructed 4 rank case to invoke torch.distributed.all_to_all() and torch.distributed.all_to_all_single(), then dump trace and replay.

@facebook-github-bot facebook-github-bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Mar 31, 2025
@facebook-github-bot
Copy link
Contributor

@shengfukevin has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator.

Summary:
Fixed support to replay all2all.

Depend on PyTorch pytorch/pytorch#149485.


Test Plan: constructed 4 rank case to invoke torch.distributed.all_to_all() and torch.distributed.all_to_all_single(), then dump trace and replay.

Differential Revision: D72196033

Pulled By: shengfukevin
@facebook-github-bot
Copy link
Contributor

@shengfukevin has updated the pull request. You must reimport the pull request before landing.

@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D72196033

@facebook-github-bot
Copy link
Contributor

@shengfukevin merged this pull request in 0fd6eff.

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

Labels

CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported Merged

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants