Skip to content

Conversation

@d4l3k
Copy link
Member

@d4l3k d4l3k commented Dec 6, 2024

This adds a way to switch between dynamic and fixed with spares world sizes.

This works with DDP currently but isn't implemented by ManagedProcessGroup since we don't implement zero grad there yet.

Test plan:

Added manager tests that step through the logic for members in the world and out.

pytest torchft/manager_test.py

@facebook-github-bot facebook-github-bot added the CLA Signed This label is managed by the Meta Open Source bot. label Dec 6, 2024
@d4l3k d4l3k merged commit ab66c7c into main Dec 6, 2024
6 checks passed
@d4l3k d4l3k deleted the d4l3k/spares branch December 6, 2024 18:06
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