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

Add more Float8 description #284

Merged
merged 1 commit into from
Apr 29, 2024
Merged

Conversation

drisspg
Copy link
Contributor

@drisspg drisspg commented Apr 29, 2024

Summary

Add more the possible options in the configs and add a note on how to get the dependency at the top of the file.

@facebook-github-bot facebook-github-bot added the CLA Signed This label is managed by the Meta Open Source bot. label Apr 29, 2024
@drisspg drisspg requested review from wanchaol and lessw2020 April 29, 2024 19:37
@@ -37,7 +37,7 @@ steps = 10
data_parallel_degree = -1
tensor_parallel_degree = 1
pipeline_parallel_degree = 1
fp8_linear = ""
fp8_linear = '' # ['', 'dynamic'] See torchtitan/float8_linear.py for more detail
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

IMO we should put details about a config into the helper message in config_manager.py. Here we can simply have # ['', 'dynamic'] as the possible options, similar to the norm_type config. But I'm ok with also putting them here.

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

yeah that makes sense, I was trying to find the best place for this one I can update this PR

@drisspg drisspg force-pushed the add-more-fp8-context branch from 8504e2c to 5ce52a6 Compare April 29, 2024 20:00
Copy link
Contributor

@tianyu-l tianyu-l left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

lgtm! thanks!

@drisspg drisspg merged commit a843abf into pytorch:main Apr 29, 2024
4 checks passed
@drisspg drisspg deleted the add-more-fp8-context branch April 29, 2024 20:46
tianyu-l pushed a commit to tianyu-l/torchtitan_intern24 that referenced this pull request Aug 16, 2024
# Summary

Add more the possible options in the configs and add a note on how to
get the dependency at the top of the file.
philippguevorguian pushed a commit to YerevaNN/YNNtitan that referenced this pull request Aug 17, 2024
# Summary

Add more the possible options in the configs and add a note on how to
get the dependency at the top of the file.
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