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
Add complex support for torch.{acosh, asinh, atanh}
#50387
Conversation
[ghstack-poisoned]
torch.{cosh, sinh, tanh}
[ghstack-poisoned]
ghstack-source-id: 5b80363d49813698688e378d2d2ef9a284f11739 Pull Request resolved: #50387
test/test_torch.py
Outdated
('acosh', '', lambda t, d: _small_3d(t, d) + 1, lambda t, d: [], 1e-3, 1e-2, 1e-5, torch.testing.get_all_fp_dtypes()), | ||
('asinh', '', _small_3d, lambda t, d: [], 1e-3, 1e-2, 1e-5, torch.testing.get_all_fp_dtypes()), | ||
('atanh', '', _small_3d, lambda t, d: [], 1e-3, 1e-2, 1e-5, torch.testing.get_all_fp_dtypes()), | ||
('acosh', '', lambda t, d: _small_3d(t, d) + 1, lambda t, d: [], 1e-3, 1e-2, 1e-5, |
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.
These were recently removed (they're redundant with the OpInfo tests). You can just revert changes to test_torch.py for this PR.
@@ -336,23 +336,23 @@ static void cosh_kernel(TensorIterator& iter) { | |||
} | |||
|
|||
static void acosh_kernel(TensorIterator& iter) { |
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.
CUDA dispatches also need an update.
torch.{cosh, sinh, tanh}
torch.{acosh, asinh, atanh}
[ghstack-poisoned]
ghstack-source-id: fe59a5244e702853f1e372c935a0b1bd9486bd30 Pull Request resolved: #50387
[ghstack-poisoned]
ghstack-source-id: dac14477cb366ff0ea3a7953ec3a1a1879e902c6 Pull Request resolved: #50387
Looks like this will have to skip test_fn_grad_acosh_cuda_complex128 - TestGradientsCUDA on pytorch_windows_vs2019_py36_cuda10.1_test2, too. I'm not totally sure what's going on there. |
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.
Cool! Looks like this just needs one more skip.
For that skip would you file an issue and copy the issue's URL into a comment over the skip?
[ghstack-poisoned]
Codecov Report
@@ Coverage Diff @@
## gh/anjali411/81/base #50387 +/- ##
=====================================================
Coverage 80.66% 80.66%
=====================================================
Files 1913 1913
Lines 208091 208091
=====================================================
+ Hits 167849 167854 +5
+ Misses 40242 40237 -5 |
Differential Revision: [D25947496](https://our.internmc.facebook.com/intern/diff/D25947496) [ghstack-poisoned]
ghstack-source-id: 9235bedd932e31e10a8c70f6c3fe132fab6ae069 Pull Request resolved: #50387
@anjali411 merged this pull request in 5d64658. |
Stack from ghstack:
torch.{acosh, asinh, atanh}
#50387 Add complex support fortorch.{acosh, asinh, atanh}
Differential Revision: D25947496