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

Fix dtensor momentum bug #80

Closed
wants to merge 1 commit into from
Closed

Fix dtensor momentum bug #80

wants to merge 1 commit into from

Conversation

runame
Copy link
Contributor

@runame runame commented Jan 13, 2025

Fixes #79.

This bug was introduced in #68 (see here).

@runame runame added the bug Something isn't working label Jan 13, 2025
@runame runame requested a review from tsunghsienlee January 13, 2025 15:13
@runame runame self-assigned this Jan 13, 2025
@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 Jan 13, 2025
@facebook-github-bot
Copy link
Contributor

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

@facebook-github-bot
Copy link
Contributor

@tsunghsienlee merged this pull request in 3fd883a.

@runame runame deleted the fix-dtensor-momentum branch January 17, 2025 15:15
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
bug Something isn't working CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. Merged
Projects
None yet
Development

Successfully merging this pull request may close these issues.

RuntimeError: aten._foreach_add_.List: got mixed torch.Tensor and DTensor when using momentum
3 participants