-
Notifications
You must be signed in to change notification settings - Fork 26.3k
Enabled BFloat16 storage #21523
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
Enabled BFloat16 storage #21523
Conversation
Enabled BFloat16 storage gh-metadata: pytorch pytorch 21523 gh/izdeby/9/head
Enabled BFloat16 storage gh-metadata: pytorch pytorch 21523 gh/izdeby/9/head
Enabled BFloat16 storage gh-metadata: pytorch pytorch 21523 gh/izdeby/9/head
|
This LGTM. I'll let Greg do final approval |
gchanan
left a comment
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.
testing? I'm presuming we have tests for bool around conversion, moving to cuda and back, etc. We should have those here.
Enabled BFloat16 storage gh-metadata: pytorch pytorch 21523 gh/izdeby/9/head
Enabled BFloat16 storage gh-metadata: pytorch pytorch 21523 gh/izdeby/9/head
Enabled BFloat16 storage gh-metadata: pytorch pytorch 21523 gh/izdeby/9/head
Enabled BFloat16 storage gh-metadata: pytorch pytorch 21523 gh/izdeby/9/head
Enabled BFloat16 storage gh-metadata: pytorch pytorch 21523 gh/izdeby/9/head
Enabled BFloat16 storage gh-metadata: pytorch pytorch 21523 gh/izdeby/9/head
Enabled BFloat16 storage gh-metadata: pytorch pytorch 21523 gh/izdeby/9/head
Enabled BFloat16 storage gh-metadata: pytorch pytorch 21523 gh/izdeby/9/head
Enabled BFloat16 storage gh-metadata: pytorch pytorch 21523 gh/izdeby/9/head
Enabled BFloat16 storage gh-metadata: pytorch pytorch 21523 gh/izdeby/9/head
Enabled BFloat16 storage gh-metadata: pytorch pytorch 21523 gh/izdeby/9/head
Enabled BFloat16 storage gh-metadata: pytorch pytorch 21523 gh/izdeby/9/head
|
I am curious if this is currently being gated by just the two build failures above or even a reviewer. |
|
@soumyarooproy i think Iurii will land this along with other BFloat16 diffs at once. If you see the PR description (the first comment), this is a stack of diffs that he is building up. |
Enabled BFloat16 storage gh-metadata: pytorch pytorch 21523 gh/izdeby/9/head
Enabled BFloat16 storage gh-metadata: pytorch pytorch 21523 gh/izdeby/9/head
Summary: Pull Request resolved: pytorch/pytorch#21523 ghimport-source-id: 698b3cbd6b21c09b9ff8bf8011980df8e35c33b0 Test Plan: Imported from OSS Differential Revision: D15819368 Pulled By: izdeby fbshipit-source-id: f6b3bba7b3ca8ee677bd80a231dbb3920c07d61c
|
Thanks for adding this! I'm having trouble using it from the python API. Have we not yet implemented casting to and from |
|
Hey, @nottombrown, what are you trying to do? i added conversion method in this PR. |
Stack from ghstack:
Enabled storage to support BFloat16 type.
Tested via unit tests
Differential Revision: D15819368