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
[MPS] Add pow.Scalar #95201
[MPS] Add pow.Scalar #95201
Conversation
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/95201
Note: Links to docs will display an error until the docs builds have been completed. ✅ No FailuresAs of commit 3858a8a: This comment was automatically generated by Dr. CI and updates every 15 minutes. |
elif (op.name in self.FP16_LOW_PRECISION_LIST) and dtype == torch.float16: | ||
elif op.name in self.FP16_LOW_PRECISION_LIST and dtype == torch.float16: | ||
atol = 1e-2 | ||
rtol = 1e-2 | ||
elif (op.name == "masked.mean"): | ||
elif op.name == "masked.mean": | ||
atol = 7e-4 | ||
rtol = 2e-3 | ||
elif (op.name == "native_layer_norm"): | ||
elif op.name == "native_layer_norm": |
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.
Make them more pythonic and consistent
@pytorchbot label "accept2run" |
6c50dfa
to
66e2cc2
Compare
66e2cc2
to
4155ac2
Compare
1a8f4ff
to
a65dffd
Compare
@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 |
1. Adds `pow.Scalar`. 2. Modifies testing `atol` and `rtol` to get pow output match tests pass. 3. Xfails numerically incorrect dtypes. Pull Request resolved: pytorch/pytorch#95201 Approved by: https://github.com/kulinseth
1. Adds `pow.Scalar`. 2. Modifies testing `atol` and `rtol` to get pow output match tests pass. 3. Xfails numerically incorrect dtypes. Pull Request resolved: pytorch/pytorch#95201 Approved by: https://github.com/kulinseth
1. Adds `pow.Scalar`. 2. Modifies testing `atol` and `rtol` to get pow output match tests pass. 3. Xfails numerically incorrect dtypes. Pull Request resolved: pytorch/pytorch#95201 Approved by: https://github.com/kulinseth
This reverts commit f33180f.
pow.Scalar
.atol
andrtol
to get pow output match tests pass.