pytorch
692b525b - [MPS] Extend unary ops to int64 (#86615)

Commit
3 years ago
[MPS] Extend unary ops to int64 (#86615) Most of them are already supported for `int64` except for: - rounding operations (`floor`, `ceil` and `round`), which are no-ops for integral types anyway - sign operation, when it can be emulated by clamping it tensor to [-1, 1] range Test new types by test MPS Fixes https://github.com/pytorch/pytorch/issues/86319 Pull Request resolved: https://github.com/pytorch/pytorch/pull/86615 Approved by: https://github.com/DenisVieriu97, https://github.com/huydhn
Author
Committer
Parents
Loading