pytorch
30cf0e70 - [MPS] Copy fixes for MPS backend (#95321)

Commit
1 year ago
[MPS] Copy fixes for MPS backend (#95321) * [MPS] Handle broadcasting by expanding src tensor in Copy.mm (#95272) Fixes #ISSUE_NUMBER Pull Request resolved: https://github.com/pytorch/pytorch/pull/95272 Approved by: https://github.com/DenisVieriu97 * [MPS] Fix copy_cast_mps() on tensors with storage offset (#95093) - The copy_cast path requires storage_offset to be applied before casting - This should fix some correctness issues in transformer models Fixes #94980 Pull Request resolved: https://github.com/pytorch/pytorch/pull/95093 Approved by: https://github.com/kulinseth --------- Co-authored-by: Ramin Azarmehr <razarmehr@apple.com>
Author
Parents
Loading