pytorch
e23cbd63 - [complex32] jiterator support

Commit
2 years ago
[complex32] jiterator support Reference #74537 Support for jiterating with `c10::complex<Half>`. Note that computation will take place in `complex<float>` by allowing implicit casting in JITerated code (similar to Half and BFloat16 which upcast to float for computation). We add `complex32` support for `sigmoid` and `sigmoid_backward` in this PR. This is tested with `test_ops.py::test_dtypes and test_ops.py::test_complex_half_reference_testing` Pull Request resolved: https://github.com/pytorch/pytorch/pull/75656 Approved by: https://github.com/ngimel
Author
Committer
Parents
Loading