-
-
Notifications
You must be signed in to change notification settings - Fork 871
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
Enable or disable bf16 support based on availability #1116
Enable or disable bf16 support based on availability #1116
Conversation
Also if you're feeling ambitious, adding a test for normalize config that it sets bf16 appropriately based on a mock of is_torch_bf16_gpu_available. |
@winglian I've changed logging level and added a unit test. |
Btw the test fails on Mac since |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Perfect. Thanks! No need to worry about mocking the device for now.
Issue #1090
Added support for
auto
in bf16 setting of config-files. Now sets bf16 to True/False depending on GPU-support ifbf16: auto
.