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

[NVIDIA] Rename fp8 custom dtype to fp32_max_grad #3984

Merged
merged 2 commits into from
Jul 8, 2024

Conversation

kaixih
Copy link
Contributor

@kaixih kaixih commented Jun 11, 2024

This PR renames the original fm32 to fp32_max_grad to express the idea of the dtype is used for storing fp32 values and using max for the gradient accumulation.

cc. @nouiz

@IvyZX
Copy link
Collaborator

IvyZX commented Jul 2, 2024

LGTM - please rebase and clear the CI test checks then I can merge it.

@kaixih
Copy link
Contributor Author

kaixih commented Jul 2, 2024

@IvyZX Thanks for the comments. Just resolved the conflict. PTAL.

@codecov-commenter
Copy link

Codecov Report

Attention: Patch coverage is 0% with 6 lines in your changes missing coverage. Please review.

Project coverage is 0.00%. Comparing base (31adb00) to head (798cfe7).
Report is 59 commits behind head on main.

Files Patch % Lines
flax/linen/fp8_ops.py 0.00% 6 Missing ⚠️
Additional details and impacted files
@@          Coverage Diff           @@
##            main   #3984    +/-   ##
======================================
  Coverage   0.00%   0.00%            
======================================
  Files        106     107     +1     
  Lines      13582   13761   +179     
======================================
- Misses     13582   13761   +179     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

@kaixih
Copy link
Contributor Author

kaixih commented Jul 8, 2024

@IvyZX Any updates? Do I need to do anything to trigger the merge?

@copybara-service copybara-service bot merged commit b3785dc into google:main Jul 8, 2024
16 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants