Skip to content
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.

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

checkpointing: move to subfolder #105

Merged
merged 1 commit into from
Feb 10, 2025
Merged

Conversation

d4l3k
Copy link
Member

@d4l3k d4l3k commented Feb 10, 2025

This restructures torchft.checkpointing to move CheckpointServer to it's own file and rename it HTTPTransport in preparation for the new PGTransport.

This new structure will make it easier to add in changes for PGTransport as well as parallel/streaming HTTP transfers.

It also has a small fix to fsdp_test due to tests running in a different order now.

Test plan:

pytest

@d4l3k d4l3k requested review from fegin, H-Huang and LucasLLC February 10, 2025 22:37
@facebook-github-bot facebook-github-bot added the CLA Signed This label is managed by the Meta Open Source bot. label Feb 10, 2025
@d4l3k d4l3k merged commit 8f0d125 into main Feb 10, 2025
6 checks passed
@d4l3k d4l3k deleted the d4l3k/checkpoint_restructure branch February 10, 2025 22:49
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 Meta Open Source bot.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants