pytorch
4d3352ed - [MPS] Remove casts from reduction/cumsum/sort ops starting with macOS 13.3 (#95817)

Commit
1 year ago
[MPS] Remove casts from reduction/cumsum/sort ops starting with macOS 13.3 (#95817) MPS in macOS13.3 has added support for int64 in reduction ops / cumsum / sort / argsort. This change removes the hard-coded casts and error messages prior macOS 13.3, allowing the op to run natively with int64. Pull Request resolved: https://github.com/pytorch/pytorch/pull/95817 Approved by: https://github.com/kulinseth
Author
Committer
Parents
Loading