pytorch
bdbf188c - [MPS] Exclude int64 dtype from reduction ops (#91272)

Commit
1 year ago
[MPS] Exclude int64 dtype from reduction ops (#91272) Reduction ops don't support int64 data type. This PR takes care to assert when int64 is used for min / max reductions ops. All other integer dtypes are casted to int32. Pull Request resolved: https://github.com/pytorch/pytorch/pull/91272 Approved by: https://github.com/razarmehr, https://github.com/malfet
Author
Committer
Parents
Loading