pytorch
9b1ff809 - CPU-Strided-Complex Fixes for real and imag ops (#29607)

Commit
5 years ago
CPU-Strided-Complex Fixes for real and imag ops (#29607) Summary: In-tree changes to pytorch to support complex numbers are being submitted here. Out-of-tree support for complex numbers is here: [pytorch-cpu-strided-complex extension](https://gitlab.com/pytorch-complex/pytorch-cpu-strided-complex) - [x] Replaced std:real(a) with a.real() in kernel level code. - [x] Fixed Vec256_base implementation of complex ops so that it works correctly on Non-AVX devices. - [ ] Clean up CopyKernel after https://github.com/pytorch/pytorch/issues/29612 is approved. zasdfgbnm is fixing this issue in https://github.com/pytorch/pytorch/issues/29612. This should be added first. cc: iotamudelta, ezyang, bddppq Pull Request resolved: https://github.com/pytorch/pytorch/pull/29607 Differential Revision: D18451046 Pulled By: ezyang fbshipit-source-id: b9dcd8e25e91cab13bd131b070d027b090cdedc9
Author
Parents
Loading