pytorch
840fb74e - 86990 range mps support (#91075)

Commit
1 year ago
86990 range mps support (#91075) Fixes #86990 - Added range_mps_out to RangeFactories.mm - Updated native_functions.yaml - Added tests in test_mps.py I did observe that despite [the documentation for torch.range](https://pytorch.org/docs/stable/generated/torch.range.html), the existing implementations do not adjust their return type based off the arguments passed to them. The MPS implementation provided here behaves the same way as the existing CPU and CUDA implementations in this regard, hence the conversion to float32 in the test cases. Pull Request resolved: https://github.com/pytorch/pytorch/pull/91075 Approved by: https://github.com/kulinseth, https://github.com/DenisVieriu97
Committer
Parents
Loading