pytorch
29cde007 - [MPS] Add `random_` overload (#98333)

Commit
2 years ago
[MPS] Add `random_` overload (#98333) That simply calls `torch.random_(from=0, to=None)` Also, fix optional upper bound calculation for all `dtypes` but int64: As one can see from https://pytorch.org/docs/stable/generated/torch.Tensor.random_.html `from` boundary is inclusive, but `to` is exclusive, i.e. if `to` is omitted for `torch.int8` dtype, it should be set to `128` and to `2` for torch.bool. Add test for `torch.random_` Fixes https://github.com/pytorch/pytorch/issues/98118 Pull Request resolved: https://github.com/pytorch/pytorch/pull/98333 Approved by: https://github.com/kulinseth
Author
Committer
Parents
Loading