pytorch
9e089db3 - [MPS] Enable `arange` for `int8` and `uint8` dtypes (#101303)

Commit
2 years ago
[MPS] Enable `arange` for `int8` and `uint8` dtypes (#101303) Not sure, why it was not enabled previously. Sort types in `AT_DISPATCH_MPS_TYPES` by group (floats first then integers) and size. Test implicitly in `test_bernoulli`. <!-- copilot:poem --> ### <samp>🤖 Generated by Copilot at 80c7ed7</samp> > _`Char` and `Byte` types_ > _MPS can dispatch them now_ > _Winter of tensors_ Pull Request resolved: https://github.com/pytorch/pytorch/pull/101303 Approved by: https://github.com/huydhn, https://github.com/ZainRizvi, https://github.com/atalman, https://github.com/kulinseth
Author
Committer
Parents
Loading