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

[ZeRO-1] use bf16 allgather #7043

Merged
merged 1 commit into from
May 20, 2024

Conversation

hgt312
Copy link
Collaborator

@hgt312 hgt312 commented May 10, 2024

align to internal changes

@jeffhataws jeffhataws merged commit cc2f89b into pytorch:r2.1_aws_neuron May 20, 2024
1 check passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants