pytorch
54bfd41a - Fix torch.angle on aarch64 (#59832)

Commit
3 years ago
Fix torch.angle on aarch64 (#59832) Summary: angle should return 0 for positive values, pi for negative and keep nans in place, which can be accomplished using two blendv functions. Fixes number of unary test failures on M1/aarch64 Pull Request resolved: https://github.com/pytorch/pytorch/pull/59832 Reviewed By: kimishpatel Differential Revision: D29046402 Pulled By: malfet fbshipit-source-id: cb93ad2de140f7a54796387fc11053c507a1d4e9
Author
Parents
Loading