[MPS] Add igamma/igammac ops
#161927
[MPS] Add igamma/igammac ops
#161927
Conversation
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/161927
Note: Links to docs will display an error until the docs builds have been completed. ❗ 1 Active SEVsThere are 1 currently active SEVs. If your PR is affected, please view them below: ✅ No FailuresAs of commit b84fff5 with merge base 5561e45 ( This comment was automatically generated by Dr. CI and updates every 15 minutes. |
pytorch-bot
bot
added
ciflow/mps
Attention! native_functions.yaml was changedIf you are adding a new function or defaulted argument to native_functions.yaml, you cannot use it from pre-existing Python frontend code until our FC window passes (two weeks). Split your PR into two PRs, one which adds the new C++ functionality, and one that makes use of it from Python, and land them two weeks apart. See https://github.com/pytorch/pytorch/wiki/PyTorch's-Python-Frontend-Backward-and-Forward-Compatibility-Policy#forwards-compatibility-fc for more info. Caused by: |
| return expm1f(a); | ||
| } | ||
|
|
||
| // NOTE: The following code was ported directly from the CUDA implementation in |
There was a problem hiding this comment.
I could make the CUDA and Metal impls share the same code. It would have to be peppered with a lot of ifdefs, but maybe it would be worth doing?
|
@pytorchbot merge -f "No need to wait for trunc" |
Merge startedYour change will be merged immediately since you used the force (-f) flag, bypassing any CI checks (ETA: 1-5 minutes). Please use Learn more about merging in the wiki. Questions? Feedback? Please reach out to the PyTorch DevX Team |
Fixes pytorch#161725 Pull Request resolved: pytorch#161927 Approved by: https://github.com/malfet
Fixes pytorch#161725 Pull Request resolved: pytorch#161927 Approved by: https://github.com/malfet
4 participants
Stack from ghstack (oldest at bottom):
igamma/igammacops #161927Fixes #161725
cc @voznesenskym @penguinwu @EikanWang @jgong5 @Guobing-Chen @XiaobingSuper @zhuhaozhe @blzheng @wenzhe-nrv @jiayisunx @ipiszy @chenyang78 @kadeng @muchulee8 @amjames @chauhang @aakhundov @coconutruben