-
Notifications
You must be signed in to change notification settings - Fork 26.3k
[BE] Remove warn about using Half on CPUs #139961
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
Conversation
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/139961
Note: Links to docs will display an error until the docs builds have been completed. ✅ No FailuresAs of commit 9f7adcd with merge base 59cf4bc ( This comment was automatically generated by Dr. CI and updates every 15 minutes. |
|
@pytorchbot merge |
Merge startedYour change will be merged once all checks pass (ETA 0-4 Hours). Learn more about merging in the wiki. Questions? Feedback? Please reach out to the PyTorch DevX Team |
It's strange that it has not been implemented for the type yet Pull Request resolved: #139959 Approved by: https://github.com/jgong5, https://github.com/janeyx99 ghstack dependencies: #139961
Becuase if invoked with long type it crahses deep in MPSGraph framework and to keep parity with CPU Add test that validates that if dtype is not floating, both CPU and MPS implementations will error out Fix function name for `mse_loss_out_mps` as `__func__` for any structured op implementation is `impl` Fixes #139723 Pull Request resolved: #139960 Approved by: https://github.com/kimishpatel ghstack dependencies: #139961, #139959
Was added by pytorch#33021, but modern CPUs right now are quite capable of handling half precision types. Alternatively one can guard the warning with `#ifdef x86_64` Pull Request resolved: pytorch#139961 Approved by: https://github.com/jgong5
It's strange that it has not been implemented for the type yet Pull Request resolved: pytorch#139959 Approved by: https://github.com/jgong5, https://github.com/janeyx99 ghstack dependencies: pytorch#139961
Becuase if invoked with long type it crahses deep in MPSGraph framework and to keep parity with CPU Add test that validates that if dtype is not floating, both CPU and MPS implementations will error out Fix function name for `mse_loss_out_mps` as `__func__` for any structured op implementation is `impl` Fixes pytorch#139723 Pull Request resolved: pytorch#139960 Approved by: https://github.com/kimishpatel ghstack dependencies: pytorch#139961, pytorch#139959
Stack from ghstack (oldest at bottom):
Was added by #33021, but modern CPUs right now are quite capable of handling half precision types.
Alternatively one can guard the warning with
#ifdef x86_64cc @jgong5 @mingfeima @XiaobingSuper @sanchitintel @ashokei @jingxu10