pytorch
6660df9f - [MPS] Fix `copy_kernel_mps` (#78428)

Commit
3 years ago
[MPS] Fix `copy_kernel_mps` (#78428) By passing `storage_offset` of source and destination Tensors This fixes following simple usecase: ``` python3` -c "import torch;x=torch.zeros(3, 3, device='mps'); x[1, 1]=1;print(x)" ``` Add test to validate it would not regress in the future Pull Request resolved: https://github.com/pytorch/pytorch/pull/78428 Approved by: https://github.com/kulinseth (cherry picked from commit 437ecfc4612b73ada1f99de94f3c79de6b08f99a)
Author
Committer
Parents
Loading