8000 [Checkpoint] Update import for spmd checkpoint by wz337 · Pull Request #620 · pytorch/PiPPy · GitHub
[go: up one dir, main page]
Skip to content

[Checkpoint] Update import for spmd checkpoint #620

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 14 commits into from
Nov 18, 2022
Merged

[Checkpoint] Update import for spmd checkpoint #620

merged 14 commits into from
Nov 18, 2022

Conversation

wz337
Copy link
Contributor
@wz337 wz337 commented Nov 17, 2022

Update import for spmd checkpoint files, as we have moved distributed checkpointing from torch.distributed._shard.checkpoint to torch.distributed.checkpoint in PyTorch (pytorch/pytorch#88698).

Test:
CI

@wz337 wz337 marked this pull request as ready for review November 17, 2022 22:49
wanchaol
wanchaol approved these changes Nov 17, 2022
View reviewed changes
Copy link
Contributor
@wanchaol wanchaol left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

gogogo

@wz337 wz337 merged commit 40a40af into pytorch:main Nov 18, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@wanchaol wanchaol wanchaol approved these changes

@aazzolini aazzolini Awaiting requested review from aazzolini

@awgu awgu Awaiting requested review from awgu

@mrshenli mrshenli Awaiting requested review from mrshenli

Assignees
No one assigned
Labels
cla signed
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
3 participants
@wz337 @wanchaol @facebook-github-bot
0